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

molde/__init__.py

@@ -1,5 +1,6 @@
-from pathlib import Path
-
-from .colors import Color
-
-MOLDE_DIR = Path(__file__).parent
+from pathlib import Path
+
+from .colors import Color
+
+MOLDE_DIR = Path(__file__).parent
+UI_DIR = MOLDE_DIR / "ui_files/"

molde/__main__.py

@@ -1,63 +1,74 @@
-import os, sys
-from PyQt5.QtWidgets import QMainWindow, QApplication, QMessageBox, QLineEdit, QTableWidgetItem, QPushButton, QLabel
-from PyQt5.QtGui import QColor
-from PyQt5 import uic
-from PyQt5.QtCore import Qt
-from time import time
-
-from molde import MOLDE_DIR
-from molde import stylesheets
-from molde.render_widgets.common_render_widget import CommonRenderWidget
-
-
-class Example(QMainWindow):
-    def __init__(self, parent=None) -> None:
-        super().__init__(parent)
-        uic.loadUi(MOLDE_DIR / "stylesheets/mainwindow.ui", self)
-        self.current_theme = "light"
-
-        self.change_theme_button.clicked.connect(self.change_theme)
-        self.render_widget: CommonRenderWidget
-        self.render_widget.create_axes()
-        self.render_widget.create_scale_bar()
-        self.render_widget.create_color_bar()
-        self.render_widget.set_info_text("Hola\nque\ntal?")
-
-        self.botao1 = QPushButton()
-        self.label = QLabel("Olha o sapooo")
-        self.botao1.setText("Olha a faca")
-        self.toolbar_2.addWidget(self.label)
-        self.toolbar_2.addWidget(self.botao1)
-
-        item = QTableWidgetItem("fr")
-        item.setBackground(QColor("#FF0000"))  
-        self.tableWidget.setItem(0, 0, item) 
-        self.show()
-        
-    def change_theme(self):
-        if self.current_theme == "light":
-            self.current_theme = "dark"
-        else:
-            self.current_theme = "light"
-        
-        self.render_widget.set_theme(self.current_theme)
-        stylesheets.set_theme(self.current_theme)
-    
-    def closeEvent(self, event):
-        close = QMessageBox.question(
-            self, 
-            "QUIT", 
-            "Would you like to close the application?", 
-            QMessageBox.Yes | QMessageBox.No
-        )
-        QApplication.quit()
-
-
-if __name__ == "__main__":
-    # Make the window scale evenly for every monitor
-    os.environ["QT_AUTO_SCREEN_SCALE_FACTOR"] = "1"
-
-    app = QApplication(sys.argv)
-    e = Example()
-    e.change_theme()
-    sys.exit(app.exec())
+import os, sys
+from PyQt5.QtWidgets import QMainWindow, QApplication, QMessageBox, QLineEdit, QTableWidgetItem, QPushButton, QLabel
+from PyQt5.QtGui import QColor
+from PyQt5 import uic
+from PyQt5.QtCore import Qt
+from time import time
+
+from vtkmodules.vtkFiltersSources import vtkCylinderSource
+from vtkmodules.vtkRenderingCore import vtkPolyDataMapper, vtkActor
+
+from molde import MOLDE_DIR
+from molde import stylesheets
+from molde.render_widgets.common_render_widget import CommonRenderWidget
+
+
+class Example(QMainWindow):
+    def __init__(self, parent=None) -> None:
+        super().__init__(parent)
+        uic.loadUi(MOLDE_DIR / "stylesheets/mainwindow.ui", self)
+        self.current_theme = "light"
+
+        self.change_theme_button.clicked.connect(self.change_theme)
+        self.render_widget: CommonRenderWidget
+        self.render_widget.create_axes()
+        self.render_widget.create_scale_bar()
+        self.render_widget.create_color_bar()
+        self.render_widget.set_info_text("Hola\nque\ntal?")
+
+        cylinder = vtkCylinderSource()
+        cylinder.SetResolution(8)
+        cylinder_mapper = vtkPolyDataMapper()
+        cylinder_mapper.SetInputConnection(cylinder.GetOutputPort())
+        cylinder_actor = vtkActor()
+        cylinder_actor.SetMapper(cylinder_mapper)
+        self.render_widget.add_actors(cylinder_actor)
+
+        self.botao1 = QPushButton()
+        self.label = QLabel("Olha o sapooo")
+        self.botao1.setText("Olha a faca")
+        self.toolbar_2.addWidget(self.label)
+        self.toolbar_2.addWidget(self.botao1)
+
+        item = QTableWidgetItem("fr")
+        item.setBackground(QColor("#FF0000"))  
+        self.tableWidget.setItem(0, 0, item) 
+        self.show()
+        
+    def change_theme(self):
+        if self.current_theme == "light":
+            self.current_theme = "dark"
+        else:
+            self.current_theme = "light"
+        
+        self.render_widget.set_theme(self.current_theme)
+        stylesheets.set_theme(self.current_theme)
+    
+    def closeEvent(self, event):
+        close = QMessageBox.question(
+            self, 
+            "QUIT", 
+            "Would you like to close the application?", 
+            QMessageBox.Yes | QMessageBox.No
+        )
+        QApplication.quit()
+
+
+if __name__ == "__main__":
+    # Make the window scale evenly for every monitor
+    os.environ["QT_AUTO_SCREEN_SCALE_FACTOR"] = "1"
+
+    app = QApplication(sys.argv)
+    e = Example()
+    e.change_theme()
+    sys.exit(app.exec())

molde/actors/__init__.py

@@ -1,4 +1,5 @@
-from .ghost_actor import GhostActor
-from .lines_actor import LinesActor
-from .round_points_actor import RoundPointsActor
-from .square_points_actor import SquarePointsActor
+from .ghost_actor import GhostActor
+from .lines_actor import LinesActor
+from .round_points_actor import RoundPointsActor
+from .square_points_actor import SquarePointsActor
+from .common_symbols_actor import CommonSymbolsActorFixedSize, CommonSymbolsActorVariableSize

molde/actors/common_symbols_actor.py

@@ -0,0 +1,148 @@
+from dataclasses import dataclass
+
+from molde import Color
+from vtkmodules.vtkCommonCore import (
+    vtkDoubleArray,
+    vtkIntArray,
+    vtkPoints,
+    vtkUnsignedCharArray,
+)
+from vtkmodules.vtkCommonDataModel import vtkPolyData
+from vtkmodules.vtkRenderingCore import vtkActor, vtkDistanceToCamera, vtkGlyph3DMapper, vtkRenderer
+
+Triple = tuple[float, float, float]
+
+
+@dataclass
+class Symbol:
+    shape_name: str
+    position: Triple
+    orientation: Triple
+    color: Color
+    scale: float
+
+
+class CommonSymbolsActor(vtkActor):
+    def __init__(self, *args, **kwargs):
+        self._shapes: dict[str, vtkPolyData] = dict()
+        self._symbols: list[Symbol] = list()
+
+    def register_shape(self, name: str, shape: vtkPolyData):
+        self._shapes[name] = shape
+
+    def add_symbol(
+        self,
+        shape_name: str,
+        position: Triple,
+        orientation: Triple,
+        color: Triple,
+        scale: float = 1,
+    ):
+        symbol = Symbol(
+            shape_name,
+            position,
+            orientation,
+            color,
+            scale,
+        )
+        self._symbols.append(symbol)
+
+    def clear_shapes(self):
+        self._shapes.clear()
+
+    def clear_symbols(self):
+        self._symbols.clear()
+
+    def clear_all(self):
+        self.clear_shapes()
+        self.clear_symbols()
+
+    def common_build(self):
+        self.data = vtkPolyData()
+        points = vtkPoints()
+
+        self.mapper: vtkGlyph3DMapper = vtkGlyph3DMapper()
+        self.SetMapper(self.mapper)
+
+        sources = vtkIntArray()
+        sources.SetName("sources")
+
+        rotations = vtkDoubleArray()
+        rotations.SetNumberOfComponents(3)
+        rotations.SetName("rotations")
+
+        scales = vtkDoubleArray()
+        scales.SetName("scales")
+
+        colors = vtkUnsignedCharArray()
+        colors.SetNumberOfComponents(3)
+        colors.SetName("colors")
+
+        shape_name_to_index = dict()
+        for index, (name, shape) in enumerate(self._shapes.items()):
+            shape_name_to_index[name] = index
+            self.mapper.SetSourceData(index, shape)
+
+        for symbol in self._symbols:
+            points.InsertNextPoint(symbol.position)
+            rotations.InsertNextTuple(symbol.orientation)
+            colors.InsertNextTuple(symbol.color.to_rgb())
+            scales.InsertNextValue(symbol.scale)
+            sources.InsertNextValue(shape_name_to_index[symbol.shape_name])
+
+        self.data.SetPoints(points)
+        self.data.GetPointData().AddArray(sources)
+        self.data.GetPointData().AddArray(rotations)
+        self.data.GetPointData().AddArray(scales)
+        self.data.GetPointData().SetScalars(colors)
+
+
+class CommonSymbolsActorFixedSize(CommonSymbolsActor):
+    def build(self):
+        self.common_build()
+
+        self.mapper.SetInputData(self.data)
+        self.mapper.SetSourceIndexArray("sources")
+        self.mapper.SetOrientationArray("rotations")
+        self.mapper.SetScaleArray("scales")
+        self.mapper.SourceIndexingOn()
+        self.mapper.ScalarVisibilityOn()
+        self.mapper.SetScaleModeToScaleByMagnitude()
+        self.mapper.SetScalarModeToUsePointData()
+        self.mapper.SetOrientationModeToDirection()
+        self.mapper.Update()
+
+
+class CommonSymbolsActorVariableSize(CommonSymbolsActor):
+    def __init__(self, renderer: vtkRenderer):
+        super().__init__()
+        self.renderer = renderer
+
+    def build(self):
+        self.common_build()
+
+        distance_to_camera = vtkDistanceToCamera()
+        distance_to_camera.SetInputData(self.data)
+        distance_to_camera.SetScreenSize(40)
+        distance_to_camera.SetRenderer(self.renderer)
+
+        self.mapper.SetInputConnection(distance_to_camera.GetOutputPort())
+        self.mapper.SetSourceIndexArray("sources")
+        self.mapper.SetOrientationArray("rotations")
+        self.mapper.SetScaleArray("DistanceToCamera")
+        self.mapper.SourceIndexingOn()
+        self.mapper.ScalarVisibilityOn()
+        self.mapper.SetScaleModeToScaleByMagnitude()
+        self.mapper.SetScalarModeToUsePointData()
+        self.mapper.SetOrientationModeToDirection()
+
+        # shows the actor in front of everything else
+        # offset = -66000
+        offset = 0
+        factor = 1.3
+        self.mapper.SetResolveCoincidentTopologyToPolygonOffset()
+        self.mapper.SetRelativeCoincidentTopologyLineOffsetParameters(factor, offset)
+        self.mapper.SetRelativeCoincidentTopologyPolygonOffsetParameters(factor, offset)
+        self.mapper.SetRelativeCoincidentTopologyPointOffsetParameter(offset)
+
+        self.mapper.Update()

molde/actors/ghost_actor.py

@@ -1,12 +1,12 @@
-from vtkmodules.vtkRenderingCore import vtkActor
-
-
-class GhostActor(vtkActor):
-    def make_ghost(self):
-        self.GetProperty().LightingOff()
-        offset = -66000
-        mapper = self.GetMapper()
-        mapper.SetResolveCoincidentTopologyToPolygonOffset()
-        mapper.SetRelativeCoincidentTopologyLineOffsetParameters(0, offset)
-        mapper.SetRelativeCoincidentTopologyPolygonOffsetParameters(0, offset)
-        mapper.SetRelativeCoincidentTopologyPointOffsetParameter(offset)
+from vtkmodules.vtkRenderingCore import vtkActor
+
+
+class GhostActor(vtkActor):
+    def make_ghost(self):
+        self.GetProperty().LightingOff()
+        offset = -66000
+        mapper = self.GetMapper()
+        mapper.SetResolveCoincidentTopologyToPolygonOffset()
+        mapper.SetRelativeCoincidentTopologyLineOffsetParameters(0, offset)
+        mapper.SetRelativeCoincidentTopologyPolygonOffsetParameters(0, offset)
+        mapper.SetRelativeCoincidentTopologyPointOffsetParameter(offset)

molde/actors/lines_actor.py

@@ -1,31 +1,31 @@
-from vtkmodules.vtkRenderingCore import vtkActor, vtkPolyDataMapper
-
-from molde.poly_data import LinesData
-
-
-class LinesActor(vtkActor):
-    def __init__(self, lines_list) -> None:
-        super().__init__()
-        self.lines_list = lines_list
-
-        self.build()
-
-    def build(self):
-        data = LinesData(self.lines_list)
-        mapper = vtkPolyDataMapper()
-        mapper.SetInputData(data)
-        self.SetMapper(mapper)
-        self.GetProperty().SetLineWidth(3)
-
-    def set_width(self, width):
-        self.GetProperty().SetLineWidth(width)
-
-    def appear_in_front(self, cond: bool):
-        # this offset is the Z position of the camera buffer.
-        # if it is -66000 the object stays in front of everything.
-        offset = -66000 if cond else 0
-        mapper = self.GetMapper()
-        mapper.SetResolveCoincidentTopologyToPolygonOffset()
-        mapper.SetRelativeCoincidentTopologyLineOffsetParameters(0, offset)
-        mapper.SetRelativeCoincidentTopologyPolygonOffsetParameters(0, offset)
-        mapper.SetRelativeCoincidentTopologyPointOffsetParameter(offset)
+from vtkmodules.vtkRenderingCore import vtkActor, vtkPolyDataMapper
+
+from molde.poly_data import LinesData
+
+
+class LinesActor(vtkActor):
+    def __init__(self, lines_list) -> None:
+        super().__init__()
+        self.lines_list = lines_list
+
+        self.build()
+
+    def build(self):
+        data = LinesData(self.lines_list)
+        mapper = vtkPolyDataMapper()
+        mapper.SetInputData(data)
+        self.SetMapper(mapper)
+        self.GetProperty().SetLineWidth(3)
+
+    def set_width(self, width):
+        self.GetProperty().SetLineWidth(width)
+
+    def appear_in_front(self, cond: bool):
+        # this offset is the Z position of the camera buffer.
+        # if it is -66000 the object stays in front of everything.
+        offset = -66000 if cond else 0
+        mapper = self.GetMapper()
+        mapper.SetResolveCoincidentTopologyToPolygonOffset()
+        mapper.SetRelativeCoincidentTopologyLineOffsetParameters(0, offset)
+        mapper.SetRelativeCoincidentTopologyPolygonOffsetParameters(0, offset)
+        mapper.SetRelativeCoincidentTopologyPointOffsetParameter(offset)

molde/actors/round_points_actor.py

@@ -1,7 +1,7 @@
-from .square_points_actor import SquarePointsActor
-
-
-class RoundPointsActor(SquarePointsActor):
-    def __init__(self, points) -> None:
-        super().__init__(points)
-        self.GetProperty().RenderPointsAsSpheresOn()
+from .square_points_actor import SquarePointsActor
+
+
+class RoundPointsActor(SquarePointsActor):
+    def __init__(self, points) -> None:
+        super().__init__(points)
+        self.GetProperty().RenderPointsAsSpheresOn()

molde/actors/square_points_actor.py

@@ -1,32 +1,32 @@
-from vtkmodules.vtkRenderingCore import vtkActor, vtkPolyDataMapper
-
-from molde.poly_data import VerticesData
-
-
-class SquarePointsActor(vtkActor):
-    def __init__(self, points_list) -> None:
-        super().__init__()
-        self.points_list = points_list
-        self.build()
-
-    def build(self):
-        data = VerticesData(self.points_list)
-        mapper = vtkPolyDataMapper()
-        mapper.SetInputData(data)
-        self.SetMapper(mapper)
-
-        self.GetProperty().SetPointSize(20)
-        self.GetProperty().LightingOff()
-
-    def set_size(self, size):
-        self.GetProperty().SetPointSize(size)
-
-    def appear_in_front(self, cond: bool):
-        # this offset is the Z position of the camera buffer.
-        # if it is -66000 the object stays in front of everything.
-        offset = -66000 if cond else 0
-        mapper = self.GetMapper()
-        mapper.SetResolveCoincidentTopologyToPolygonOffset()
-        mapper.SetRelativeCoincidentTopologyLineOffsetParameters(0, offset)
-        mapper.SetRelativeCoincidentTopologyPolygonOffsetParameters(0, offset)
-        mapper.SetRelativeCoincidentTopologyPointOffsetParameter(offset)
+from vtkmodules.vtkRenderingCore import vtkActor, vtkPolyDataMapper
+
+from molde.poly_data import VerticesData
+
+
+class SquarePointsActor(vtkActor):
+    def __init__(self, points_list) -> None:
+        super().__init__()
+        self.points_list = points_list
+        self.build()
+
+    def build(self):
+        data = VerticesData(self.points_list)
+        mapper = vtkPolyDataMapper()
+        mapper.SetInputData(data)
+        self.SetMapper(mapper)
+
+        self.GetProperty().SetPointSize(20)
+        self.GetProperty().LightingOff()
+
+    def set_size(self, size):
+        self.GetProperty().SetPointSize(size)
+
+    def appear_in_front(self, cond: bool):
+        # this offset is the Z position of the camera buffer.
+        # if it is -66000 the object stays in front of everything.
+        offset = -66000 if cond else 0
+        mapper = self.GetMapper()
+        mapper.SetResolveCoincidentTopologyToPolygonOffset()
+        mapper.SetRelativeCoincidentTopologyLineOffsetParameters(0, offset)
+        mapper.SetRelativeCoincidentTopologyPolygonOffsetParameters(0, offset)
+        mapper.SetRelativeCoincidentTopologyPointOffsetParameter(offset)

molde/colors/__init__.py

@@ -1,2 +1,2 @@
-from .color import Color
+from .color import Color
 from .color_names import *