molde 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

molde/poly_data/__init__.py

@@ -1,2 +1,13 @@
 from .lines_data import LinesData
 from .vertices_data import VerticesData
+from .arrows import (
+    create_arrow_source,
+    create_long_arrow_source,
+    create_double_arrow_source,
+    create_outwards_arrow_source,
+)
+
+from .simple_shapes import (
+    create_cone_source,
+    create_cube_source,
+)

molde/poly_data/arrows.py

@@ -0,0 +1,54 @@
+from vtkmodules.vtkFiltersSources import vtkArrowSource
+from vtkmodules.vtkFiltersCore import vtkAppendPolyData
+from molde.utils.poly_data_utils import transform_polydata
+
+
+def create_arrow_source():
+    source = vtkArrowSource()
+    source.SetTipLength(0.25)
+    source.Update()
+
+    return transform_polydata(
+        source.GetOutput(),
+        position=(-1, 0, 0),
+    )
+
+
+def create_long_arrow_source():
+    source = vtkArrowSource()
+    source.SetTipResolution(4)
+    source.SetShaftResolution(4)
+    source.SetTipLength(0.85)
+    source.Update()
+
+    return transform_polydata(
+        source.GetOutput(),
+        position=(-1, 0, 0),
+    )
+
+
+def create_double_arrow_source():
+    arrow1 = vtkArrowSource()
+    arrow1.SetTipLength(0.45)
+    arrow1.Update()
+
+    arrow2 = vtkArrowSource()
+    arrow2.SetTipLength(0.3)
+    arrow2.Update()
+
+    source = vtkAppendPolyData()
+    source.AddInputData(arrow1.GetOutput())
+    source.AddInputData(arrow2.GetOutput())
+    source.Update()
+
+    return transform_polydata(
+        source.GetOutput(),
+        position=(-1, 0, 0),
+    )
+
+
+def create_outwards_arrow_source():
+    source = vtkArrowSource()
+    source.SetTipLength(0.25)
+    source.Update()
+    return source.GetOutput()

molde/poly_data/simple_shapes.py

@@ -0,0 +1,22 @@
+from molde.utils.poly_data_utils import transform_polydata
+from vtkmodules.vtkFiltersSources import (
+    vtkConeSource,
+    vtkCubeSource,
+)
+
+def create_cone_source():
+    source = vtkConeSource()
+    source.SetHeight(1)
+    source.SetRadius(0.5)
+    source.SetResolution(12)
+    source.Update()
+    return transform_polydata(
+        source.GetOutput(),
+        position=(-0.5, 0, 0),
+    )
+
+def create_cube_source():
+    source = vtkCubeSource()
+    source.SetBounds(0, 1, 0, 1, 0, 1)
+    source.Update()
+    return source.GetOutput()

molde/ui_files/messages/new_loading_window.ui

@@ -0,0 +1,73 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<ui version="4.0">
+ <class>loading_window</class>
+ <widget class="QWidget" name="loading_window">
+  <property name="windowModality">
+   <enum>Qt::ApplicationModal</enum>
+  </property>
+  <property name="geometry">
+   <rect>
+    <x>0</x>
+    <y>0</y>
+    <width>377</width>
+    <height>145</height>
+   </rect>
+  </property>
+  <property name="windowTitle">
+   <string>Loading Window</string>
+  </property>
+  <layout class="QVBoxLayout" name="verticalLayout">
+   <item>
+    <spacer name="verticalSpacer_2">
+     <property name="orientation">
+      <enum>Qt::Vertical</enum>
+     </property>
+     <property name="sizeHint" stdset="0">
+      <size>
+       <width>20</width>
+       <height>40</height>
+      </size>
+     </property>
+    </spacer>
+   </item>
+   <item>
+    <widget class="QLabel" name="progress_label">
+     <property name="text">
+      <string>Loading ...</string>
+     </property>
+     <property name="alignment">
+      <set>Qt::AlignCenter</set>
+     </property>
+    </widget>
+   </item>
+   <item>
+    <widget class="QProgressBar" name="progress_bar">
+     <property name="value">
+      <number>10</number>
+     </property>
+     <property name="textVisible">
+      <bool>true</bool>
+     </property>
+     <property name="textDirection">
+      <enum>QProgressBar::TopToBottom</enum>
+     </property>
+    </widget>
+   </item>
+   <item>
+    <spacer name="verticalSpacer">
+     <property name="orientation">
+      <enum>Qt::Vertical</enum>
+     </property>
+     <property name="sizeHint" stdset="0">
+      <size>
+       <width>20</width>
+       <height>40</height>
+      </size>
+     </property>
+    </spacer>
+   </item>
+  </layout>
+ </widget>
+ <resources/>
+ <connections/>
+</ui>

molde/utils/__init__.py

@@ -1,5 +1,5 @@
 from .format_sequences import format_long_sequence
-from .poly_data_utils import set_polydata_colors, set_polydata_property
+from .poly_data_utils import set_polydata_colors, set_polydata_property, read_obj_file, read_stl_file, transform_polydata
 from .tree_info import TreeInfo
 
 

molde/utils/poly_data_utils.py

@@ -1,5 +1,10 @@
+from pathlib import Path
+
 from vtkmodules.vtkCommonCore import vtkUnsignedCharArray, vtkUnsignedIntArray
 from vtkmodules.vtkCommonDataModel import vtkPolyData
+from vtkmodules.vtkCommonTransforms import vtkTransform
+from vtkmodules.vtkFiltersGeneral import vtkTransformPolyDataFilter
+from vtkmodules.vtkIOGeometry import vtkOBJReader, vtkSTLReader
 
 
 def set_polydata_colors(data: vtkPolyData, color: tuple) -> vtkUnsignedCharArray:
@@ -15,11 +20,47 @@ def set_polydata_colors(data: vtkPolyData, color: tuple) -> vtkUnsignedCharArray
     return cell_colors
 
 
-def set_polydata_property(data: vtkPolyData, property_data: int, property_name: str) -> vtkUnsignedIntArray:
+def set_polydata_property(
+    data: vtkPolyData, property_data: int, property_name: str
+) -> vtkUnsignedIntArray:
     n_cells = data.GetNumberOfCells()
     cell_identifier = vtkUnsignedIntArray()
     cell_identifier.SetName(property_name)
     cell_identifier.SetNumberOfTuples(n_cells)
     cell_identifier.Fill(property_data)
     data.GetCellData().AddArray(cell_identifier)
-    return cell_identifier
+    return cell_identifier
+
+
+def read_obj_file(path: str | Path) -> vtkPolyData:
+    reader = vtkOBJReader()
+    reader.SetFileName(str(path))
+    reader.Update()
+    return reader.GetOutput()
+
+
+def read_stl_file(path: str | Path) -> vtkPolyData:
+    reader = vtkSTLReader()
+    reader.SetFileName(str(path))
+    reader.Update()
+    return reader.GetOutput()
+
+
+def transform_polydata(
+    polydata: vtkPolyData,
+    position=(0, 0, 0),
+    rotation=(0, 0, 0),
+    scale=(1, 1, 1),
+) -> vtkPolyData:
+    transform = vtkTransform()
+    transform.Translate(position)
+    transform.Scale(scale)
+    transform.RotateX(rotation[0])
+    transform.RotateY(rotation[1])
+    transform.RotateZ(rotation[2])
+    transform.Update()
+    transformation = vtkTransformPolyDataFilter()
+    transformation.SetTransform(transform)
+    transformation.SetInputData(polydata)
+    transformation.Update()
+    return transformation.GetOutput()

molde/windows/loading_window.py

@@ -0,0 +1,189 @@
+import logging
+import re
+from time import sleep
+
+from PyQt5.QtCore import Qt
+from PyQt5.QtWidgets import QApplication, QLabel, QProgressBar, QWidget
+from PyQt5 import uic
+
+from molde import UI_DIR
+
+
+# Catches every message that contains something like [n/N]
+PROGRESS_FRACTION_REGEX = re.compile(r"\[\d+/\d+\]")
+
+# Catches every message that contains something like n%
+PROGRESS_PERCENTAGE_REGEX = re.compile(r"\d+%")
+
+
+class LoadingWindow(QWidget):
+    """
+    This function is intended to be called for functions that take
+    a long time to run and should run together with a progress bar.
+
+    The indended use is explained in the following example:
+    ```
+    def long_function(param_a, param_b=0):
+        ...
+        return value_c
+    ```
+
+    To run this function with a progress bar you just need to write
+    ```
+    value_c = LoadingWindow(long_function, param_a, param_b=1234)
+    ```
+
+    Disclaimers:
+    This is a monstruosity, but it works.
+    I hope no one ever needs to modify it.
+    I am trying my best to explain every part of this code.
+
+    An easier way to do it would be using a QWorker and a QThread,
+    but it does not work for our Application.
+    Qt manages the windows in the main thread, therefore the function
+    to be loaded would need to run in a secondary thread.
+    The problem is GMSH (wow, what a surprise).
+
+    GMSH, for some reason, refuses to run in secondary thread.
+    And GMSH is an important part of our software, that of course
+    need a progress bar when it is running.
+    So this is an attempt to run everything (both GMSH and QT) in the
+    main thread without conflicts.
+
+    I also don't want to mix the interface code with the engine code
+    because it can easily became a mess and make the creation of
+    automated tests really hard.
+    """
+
+    def __init__(self, _function, application, window_icon):
+        super().__init__()
+
+        ui_path = UI_DIR / "messages/new_loading_window.ui"
+        uic.loadUi(ui_path, self)
+
+        self._function = _function
+        self._application = application
+        self._window_icon = window_icon
+    
+        self._config_window()
+
+    def _config_window(self):
+        self.setWindowIcon(self._window_icon)
+        self.setWindowTitle("Loading")
+        self.setWindowFlags(Qt.Window | Qt.CustomizeWindowHint | Qt.WindowTitleHint | Qt.WindowMinimizeButtonHint)
+        self.setWindowModality(Qt.ApplicationModal)
+        self.update_position()
+        self.progress_bar.setValue(0)
+
+    def _define_qt_variables(self):
+        self.progress_bar: QProgressBar
+        self.progress_label: QLabel
+
+    def update_position(self):
+        '''
+        Place the window on the center of the screen.
+        '''
+        desktop_geometry = self._application.primaryScreen().geometry()
+        pos_x = int((desktop_geometry.width() - self.width())/2)
+        pos_y = int((desktop_geometry.height() - self.height())/2)
+        self.setGeometry(pos_x, pos_y, self.width(), self.height())
+
+    def run(self, *args, **kwargs):
+        self.show()
+
+        # Changes the cursor to wait
+        QApplication.setOverrideCursor(Qt.WaitCursor)
+
+        # Creates a handler to update progress_bar and progress_label
+        # every time a logging containing [n/N] appears
+        progress_handler = ProgressBarLogUpdater(logging.DEBUG, loading_window=self)
+        logging.getLogger().addHandler(progress_handler)
+
+        # Waits the loading bar to appear and uptates pyqt
+        sleep(0.1)
+        QApplication.processEvents()
+
+        try:
+            # Calls the actual function
+            return_value = self._function(*args, **kwargs)
+
+        finally:
+            """
+            This piece of code will run even if the function have errors.
+            Because the error is not handled here it will propagate to the
+            function that called it.
+            The error should be threated there, here we are just mitigating
+            things related to the loading window.
+            """
+
+            # Restores the previous cursor
+            QApplication.restoreOverrideCursor()
+
+            # Removes the ProgressBarLogUpdater
+            logging.getLogger().removeHandler(progress_handler)
+
+            self.hide()
+
+        return return_value
+
+    def __call__(self, *args, **kwargs):
+        return self.run(self._function, *args, **kwargs)
+
+
+class ProgressBarLogUpdater(logging.Handler):
+    """
+    This class is an log observer. It is meant to watch logs
+    and use it to update an instance of the loading window.
+    """
+
+    def __init__(self, level=0, *, loading_window: LoadingWindow) -> None:
+        super().__init__(level)
+        self.loading_window = loading_window
+
+    def emit(self, record):
+        """
+        This function is fired when something is logged.
+        If the log have a marker like [n/N] or "..." in its message it
+        will update the LoadingWindow associated with this class.
+        """
+
+        # Updates QT to prevent freezing
+        QApplication.processEvents()
+
+        percent = self.get_percentage(record.msg)
+
+        if percent is not None:
+            self.loading_window.progress_label.setText(record.msg)
+            self.loading_window.progress_bar.setValue(percent)
+        
+        elif "..." in record.msg:
+            self.loading_window.progress_label.setText(record.msg)
+
+        # Updates QT to show the window modifications
+        QApplication.processEvents()
+
+    def get_percentage(self, message: str):
+        """
+        Uses regex to check if the message have a marker like [2/10]
+        If it does, it extracts the step (2) and the max_step (10) and
+        calculates the percentage (20%).
+        Otherwise it just returns None.
+        """
+
+        if not isinstance(message, str):
+            return
+
+        matches = PROGRESS_FRACTION_REGEX.findall(message)
+        if matches:
+            first_match: str = matches[0]
+            step, max_step = first_match.strip("[]").split("/")
+            percentage = 100 * int(step) // int(max_step)
+            return percentage
+
+        matches = PROGRESS_PERCENTAGE_REGEX.findall(message)
+        if matches:
+            first_match: str = matches[0]
+            percentage = int(first_match.strip("%"))
+            return percentage
+
+        return None

{molde-0.1.1.dist-info → molde-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: molde
-Version: 0.1.1
+Version: 0.1.3
 Summary: 
 Author: André Fernandes
 Author-email: fpf.andre@gmail.com

{molde-0.1.1.dist-info → molde-0.1.3.dist-info}/RECORD

@@ -1,6 +1,7 @@
-molde/__init__.py,sha256=uutAOqWJe5XvTpDdU92HBNYga5Pa9zLaMowJBf-9P9U,92
-molde/__main__.py,sha256=CEHF8UXXsTxr2APu7lZX1MXHYR984bjQvgOzYFzh8Zc,2097
-molde/actors/__init__.py,sha256=8STcYmULmRKe8RhYih_ENSHR-zl7Rlowwfue5ZOT4yo,176
+molde/__init__.py,sha256=EDwvJl_tqHnoLsB4XI2iVUmtxdrBvrpoxyLkmwaSiSA,124
+molde/__main__.py,sha256=ZB6W4bZTvbgez28mzZ4QEDWjApbndHoL6QeZSw1RG1g,2565
+molde/actors/__init__.py,sha256=XgHvW7jKs4OxL9q-PLNi4ses4CBrXXsnOszZTNJ9CTA,271
+molde/actors/common_symbols_actor.py,sha256=86IkbRvHh9N7QunT3fv_iYahuQ9cJ6Nzhyptu2Jxw4k,5231
 molde/actors/ghost_actor.py,sha256=9LPRfWCjwHhLrZTAUhmmMNt3Tdb0fDAjjOn6BT-mspI,505
 molde/actors/lines_actor.py,sha256=T-9arZMV6kSWMmII1UFlGE-6NeAX1_zpmrUhaGNNbd8,1122
 molde/actors/round_points_actor.py,sha256=FZKq46MZ7IJspE9qdr4xHDoG_6qi4ZE7qzWjV47fv0Y,229
@@ -24,13 +25,16 @@ molde/icons/arrow_up_disabled_light_theme.svg,sha256=mXbCteiX6LX1GPcvFjXa-_cZauF
 molde/icons/arrow_up_light_theme.svg,sha256=bpon_eUC97Q3nYlHvLPGBDoToUlZq2VHqGJsLMI6Cs4,179
 molde/icons/check_box_image.svg,sha256=5rKzmsoHYtp52jPY4TpWK0jd7mPLPTteYfbLUF2RJbY,335
 molde/interactor_styles/__init__.py,sha256=Ovu5F9EycKEDQ9GWYirJhvWaBEjjOZxNsacq3jv3SY4,126
-molde/interactor_styles/arcball_camera_style.py,sha256=0PNZbmMZhj-uKSWvuenMmhYU_6JJEMZZ28iVILc9di8,9605
+molde/interactor_styles/arcball_camera_style.py,sha256=vOgv9yli0Ei3tpFUkifFSIl0bzd_ab2QXu1vBYX-XIU,10680
 molde/interactor_styles/box_selection_style.py,sha256=qQCjvXxryOH1jwBc_jEX0N5q7vanmkFA6r7JJ0ybACk,2619
+molde/main_window.ui,sha256=9ZAsT7Y-bNo2mbB5UX04QgsDlmJh3HTtRVAVtX9m-Hs,26493
 molde/pickers/__init__.py,sha256=Jyc2t5A9vaypAlQ1U2dXC-XHHDG4jkbw2FoqnnLxh00,109
 molde/pickers/cell_area_picker.py,sha256=Dhvb-H1l0WqZoqs5Arp-9tGQBxdlNhPdVRCKRKJd--w,2114
 molde/pickers/cell_property_area_picker.py,sha256=cN2DK9No0BNFruTJKA3SCYUzXrfO8LSnHTbSL8c6W7Q,2953
-molde/poly_data/__init__.py,sha256=Yv9vfHFVKnHvspwPYtrjs4501mIKDY80jA3jhiGj4zM,76
+molde/poly_data/__init__.py,sha256=smsqssy-HCKf1rVQD4cMaGdXRL93Oh9y616i0hDsq5I,312
+molde/poly_data/arrows.py,sha256=amd5iVn6jnSixpKFGNWSUdK1LEA8DIftH8KtxCQVOag,1283
 molde/poly_data/lines_data.py,sha256=DScG3EJSP713HbYdL8Dd8sZy-cQ7xXIlStdZXG5mBPs,737
+molde/poly_data/simple_shapes.py,sha256=h43qE-PDxY9d3pZ6B0_XQ0hBFYGHxGppACQsJrJDGis,560
 molde/poly_data/vertices_data.py,sha256=Ks6drjDTCd1rFZ9WkTo-rr-lQ_HiW3cLz4YIKKd4V90,697
 molde/render_widgets/__init__.py,sha256=gTCyI5MXmk7hl4yfrR2BpHPQl9bgYQdNZt2IiSj1Y4o,112
 molde/render_widgets/animated_render_widget.py,sha256=auh818MYT28-yfN4JkK4H-VQi2KSEo1-7fCJg_oH34M,5516
@@ -53,10 +57,12 @@ molde/stylesheets/qtabwidget.qss,sha256=ygmcTp0S4bzV1X-QcVNKZn88YaLl6MIjCzqqlTHn
 molde/stylesheets/qtoolbar.qss,sha256=o6WLggd0X1goQk8Ih0YrbxNjnUDwJZFrbzr0pTaltuU,1085
 molde/stylesheets/qtoolbuttons.qss,sha256=9IkxWx_zwGStuFZoEqGv8xcETnagvQGaFkKNJbLWnhk,286
 molde/stylesheets/qtreewidget.qss,sha256=nkwjNWFshp-nk_Ug_dLes2ZOU3ERYe8_NRHZt16zLoM,582
-molde/utils/__init__.py,sha256=Vi1vpnt_iKSZ9JjAoeyhddSja3p4THoj_KlXIOAbeuc,271
+molde/ui_files/messages/new_loading_window.ui,sha256=h9iIzdqKBoCIfaIwBI8tDMhWm-q8kmkjPSMHY0Eq-cA,1799
+molde/utils/__init__.py,sha256=_hyuh9B44qON6n_K1pOVBVD8wCfFOnEYPkSIW0b_rAg,321
 molde/utils/format_sequences.py,sha256=ikyjXNXWgFHzg3vrv1lQA5VDJO5A678lHZT7HsMvMCg,1389
-molde/utils/poly_data_utils.py,sha256=cG88mGDtbJWU3Dmnnm9xgYsXJn027YAgs1TAlFiD_qo,1035
+molde/utils/poly_data_utils.py,sha256=IJ3TpuSCtinQhPH2uJ1rH7SAenYhgAcay6Cowjvez9o,2191
 molde/utils/tree_info.py,sha256=DyMshZuQKF_UVk-bchhHfvoRCiGloj49p4ABkZWgkKg,1812
-molde-0.1.1.dist-info/METADATA,sha256=goxGic1DLGy7xGlZVE9zKGrwfQ5BwAc0ABGQ5KJp_O0,2038
-molde-0.1.1.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
-molde-0.1.1.dist-info/RECORD,,
+molde/windows/loading_window.py,sha256=DnHEwL3n_702dIGJiG6U--jvUnVKbwhh24buVudDaAc,6477
+molde-0.1.3.dist-info/METADATA,sha256=DFKzHbZ_PnVl6QwVfciEA3epnuaMl2f_EcahsuI33xw,2038
+molde-0.1.3.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
+molde-0.1.3.dist-info/RECORD,,