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

molde/__init__.py

@@ -1,6 +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,74 +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 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())
+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,5 +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 .common_symbols_actor import CommonSymbolsActorFixedSize, CommonSymbolsActorVariableSize
+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

@@ -1,148 +1,164 @@
-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()
+from dataclasses import dataclass
+
+from molde import Color
+from molde.utils import transform_polydata
+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,
+        position: Triple = (0, 0, 0),
+        rotation: Triple = (0, 0, 0),
+        scale: Triple = (1, 1, 1),
+    ):
+        self._shapes[name] = transform_polydata(
+            shape,
+            position,
+            rotation,
+            scale,
+        )
+
+    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)
+
+    def set_zbuffer_offsets(self, factor: float, units: float):
+        """
+        This functions is usefull to make a object appear in front of the others.
+        If the object should never be hidden, the parameters should be set to
+        factor = 1 and offset = -66000.
+        """
+        self.mapper.SetResolveCoincidentTopologyToPolygonOffset()
+        self.mapper.SetRelativeCoincidentTopologyLineOffsetParameters(factor, units)
+        self.mapper.SetRelativeCoincidentTopologyPolygonOffsetParameters(factor, units)
+        self.mapper.SetRelativeCoincidentTopologyPointOffsetParameter(units)
+        self.mapper.Update()
+
+
+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()
+
+        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 *