molde 0.1.10__tar.gz → 0.1.12__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (69) hide show
  1. {molde-0.1.10 → molde-0.1.12}/PKG-INFO +1 -1
  2. {molde-0.1.10 → molde-0.1.12}/README.md +26 -26
  3. {molde-0.1.10 → molde-0.1.12}/molde/__init__.py +12 -12
  4. {molde-0.1.10 → molde-0.1.12}/molde/__main__.py +81 -81
  5. {molde-0.1.10 → molde-0.1.12}/molde/actors/__init__.py +5 -5
  6. {molde-0.1.10 → molde-0.1.12}/molde/actors/common_symbols_actor.py +148 -164
  7. {molde-0.1.10 → molde-0.1.12}/molde/actors/ghost_actor.py +12 -12
  8. {molde-0.1.10 → molde-0.1.12}/molde/actors/lines_actor.py +31 -31
  9. {molde-0.1.10 → molde-0.1.12}/molde/actors/round_points_actor.py +7 -7
  10. {molde-0.1.10 → molde-0.1.12}/molde/actors/square_points_actor.py +32 -32
  11. molde-0.1.12/molde/colors/__init__.py +2 -0
  12. {molde-0.1.10 → molde-0.1.12}/molde/colors/color.py +150 -150
  13. {molde-0.1.10 → molde-0.1.12}/molde/colors/color_names.py +124 -124
  14. {molde-0.1.10 → molde-0.1.12}/molde/interactor_styles/__init__.py +2 -2
  15. {molde-0.1.10 → molde-0.1.12}/molde/interactor_styles/arcball_camera_style.py +288 -288
  16. {molde-0.1.10 → molde-0.1.12}/molde/interactor_styles/box_selection_style.py +87 -87
  17. {molde-0.1.10 → molde-0.1.12}/molde/main_window.ui +864 -864
  18. {molde-0.1.10 → molde-0.1.12}/molde/pickers/__init__.py +2 -2
  19. {molde-0.1.10 → molde-0.1.12}/molde/pickers/cell_area_picker.py +61 -61
  20. {molde-0.1.10 → molde-0.1.12}/molde/pickers/cell_property_area_picker.py +84 -84
  21. {molde-0.1.10 → molde-0.1.12}/molde/poly_data/__init__.py +13 -13
  22. {molde-0.1.10 → molde-0.1.12}/molde/poly_data/lines_data.py +23 -23
  23. {molde-0.1.10 → molde-0.1.12}/molde/poly_data/vertices_data.py +24 -24
  24. {molde-0.1.10 → molde-0.1.12}/molde/render_widgets/__init__.py +2 -2
  25. {molde-0.1.10 → molde-0.1.12}/molde/render_widgets/animated_render_widget.py +164 -164
  26. {molde-0.1.10 → molde-0.1.12}/molde/render_widgets/common_render_widget.py +429 -429
  27. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/__init__.py +122 -122
  28. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/common.qss +15 -15
  29. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/create_color_page.py +61 -61
  30. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/mainwindow.ui +646 -646
  31. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qcheckbox.qss +23 -23
  32. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qinputs.qss +81 -81
  33. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qlayouts.qss +23 -23
  34. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qmenubar.qss +12 -12
  35. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qprogressbar.qss +11 -11
  36. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qpushbutton.qss +89 -89
  37. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qradiobutton.qss +30 -30
  38. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qscrollbar.qss +29 -29
  39. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qslider.qss +61 -61
  40. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qtablewidget.qss +27 -27
  41. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qtabwidget.qss +28 -28
  42. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qtoolbar.qss +63 -63
  43. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qtoolbuttons.qss +14 -14
  44. {molde-0.1.10 → molde-0.1.12}/molde/stylesheets/qtreewidget.qss +25 -25
  45. {molde-0.1.10 → molde-0.1.12}/molde/ui_files/messages/new_loading_window.ui +73 -73
  46. {molde-0.1.10 → molde-0.1.12}/molde/utils/__init__.py +8 -8
  47. {molde-0.1.10 → molde-0.1.12}/molde/utils/format_sequences.py +44 -44
  48. {molde-0.1.10 → molde-0.1.12}/molde/utils/poly_data_utils.py +66 -66
  49. {molde-0.1.10 → molde-0.1.12}/molde/utils/tree_info.py +52 -52
  50. {molde-0.1.10 → molde-0.1.12}/molde/windows/loading_window.py +189 -189
  51. {molde-0.1.10 → molde-0.1.12}/pyproject.toml +28 -28
  52. molde-0.1.10/molde/colors/__init__.py +0 -2
  53. {molde-0.1.10 → molde-0.1.12}/molde/fonts/IBMPlexMono-Bold.ttf +0 -0
  54. {molde-0.1.10 → molde-0.1.12}/molde/fonts/IBMPlexMono-Regular.ttf +0 -0
  55. {molde-0.1.10 → molde-0.1.12}/molde/icons/arrow_down_dark_theme.svg +0 -0
  56. {molde-0.1.10 → molde-0.1.12}/molde/icons/arrow_down_disabled_dark_theme.svg +0 -0
  57. {molde-0.1.10 → molde-0.1.12}/molde/icons/arrow_down_disabled_light_theme.svg +0 -0
  58. {molde-0.1.10 → molde-0.1.12}/molde/icons/arrow_down_light_theme.svg +0 -0
  59. {molde-0.1.10 → molde-0.1.12}/molde/icons/arrow_left_dark_theme.svg +0 -0
  60. {molde-0.1.10 → molde-0.1.12}/molde/icons/arrow_left_light_theme.svg +0 -0
  61. {molde-0.1.10 → molde-0.1.12}/molde/icons/arrow_right_dark_theme.svg +0 -0
  62. {molde-0.1.10 → molde-0.1.12}/molde/icons/arrow_right_light_theme.svg +0 -0
  63. {molde-0.1.10 → molde-0.1.12}/molde/icons/arrow_up_dark_theme.svg +0 -0
  64. {molde-0.1.10 → molde-0.1.12}/molde/icons/arrow_up_disabled_dark_theme.svg +0 -0
  65. {molde-0.1.10 → molde-0.1.12}/molde/icons/arrow_up_disabled_light_theme.svg +0 -0
  66. {molde-0.1.10 → molde-0.1.12}/molde/icons/arrow_up_light_theme.svg +0 -0
  67. {molde-0.1.10 → molde-0.1.12}/molde/icons/check_box_image.svg +0 -0
  68. {molde-0.1.10 → molde-0.1.12}/molde/poly_data/arrows.py +0 -0
  69. {molde-0.1.10 → molde-0.1.12}/molde/poly_data/simple_shapes.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: molde
3
- Version: 0.1.10
3
+ Version: 0.1.12
4
4
  Summary:
5
5
  Author: André Fernandes
6
6
  Author-email: fpf.andre@gmail.com
@@ -1,26 +1,26 @@
1
- <div align="center">
2
- <img height="90" src="logos/mold_horizontal_logo_dark_300dpi.png" alt="Logo" align="center">
3
- </div>
4
-
5
- # MOLDE
6
- **M**OPT **L**ightweight **De**esign (MOLDE) is a library created to standardize and make easier the creation and maintainance of internal products of the Multidisciplinary Optimization Group (MOPT).
7
-
8
- Here are the colors, icons, fonts and common interface components, that may be usefull for multiple MOPT projects.
9
-
10
- This is heavily inspired by [Carbon Design System](https://carbondesignsystem.com/), [Atlassian Design System](https://atlassian.design/), [Fast Design System](https://www.fast.design/), [Material Design](https://m3.material.io/) and [Bold Design System](https://bold.bridge.ufsc.br/pt/).
11
-
12
- # Colors
13
- The color palettes are available [here](https://andrefpf.github.io/molde/).
14
-
15
- # Icons
16
- A lot of free to use icons, from Material Design, are available [here](https://fonts.google.com/icons).
17
- Other icons may be necessary, and they will be made to match the same style.
18
- For consistency avoid using icons from other places.
19
-
20
- # Typeface
21
- The font choose for this system is [IBM Plex](https://www.ibm.com/plex/).
22
- The `.ttf` files for every version of the font, including monospaced, can be found [here](https://github.com/IBM/plex/tree/master/packages).
23
-
24
- # Component
25
- The sylesheets of the Qt components are designed to make it similar to [Atlassian components](https://atlassian.design/components/).
26
- Always check for the Atlassian component usage guide in case of any doubt.
1
+ <div align="center">
2
+ <img height="90" src="logos/mold_horizontal_logo_dark_300dpi.png" alt="Logo" align="center">
3
+ </div>
4
+
5
+ # MOLDE
6
+ **M**OPT **L**ightweight **De**esign (MOLDE) is a library created to standardize and make easier the creation and maintainance of internal products of the Multidisciplinary Optimization Group (MOPT).
7
+
8
+ Here are the colors, icons, fonts and common interface components, that may be usefull for multiple MOPT projects.
9
+
10
+ This is heavily inspired by [Carbon Design System](https://carbondesignsystem.com/), [Atlassian Design System](https://atlassian.design/), [Fast Design System](https://www.fast.design/), [Material Design](https://m3.material.io/) and [Bold Design System](https://bold.bridge.ufsc.br/pt/).
11
+
12
+ # Colors
13
+ The color palettes are available [here](https://andrefpf.github.io/molde/).
14
+
15
+ # Icons
16
+ A lot of free to use icons, from Material Design, are available [here](https://fonts.google.com/icons).
17
+ Other icons may be necessary, and they will be made to match the same style.
18
+ For consistency avoid using icons from other places.
19
+
20
+ # Typeface
21
+ The font choose for this system is [IBM Plex](https://www.ibm.com/plex/).
22
+ The `.ttf` files for every version of the font, including monospaced, can be found [here](https://github.com/IBM/plex/tree/master/packages).
23
+
24
+ # Component
25
+ The sylesheets of the Qt components are designed to make it similar to [Atlassian components](https://atlassian.design/components/).
26
+ Always check for the Atlassian component usage guide in case of any doubt.
@@ -1,12 +1,12 @@
1
- from pathlib import Path
2
- from qtpy.uic import loadUi
3
- from .colors import Color
4
-
5
- MOLDE_DIR = Path(__file__).parent
6
- UI_DIR = MOLDE_DIR / "ui_files/"
7
-
8
- def load_ui(uifile: str | Path, baseinstance, working_directory: str| Path = None):
9
- from PySide6.QtCore import QDir
10
- if working_directory is not None:
11
- working_directory = QDir(working_directory)
12
- return loadUi(uifile, baseinstance, working_directory)
1
+ from pathlib import Path
2
+ from qtpy.uic import loadUi
3
+ from .colors import Color
4
+
5
+ MOLDE_DIR = Path(__file__).parent
6
+ UI_DIR = MOLDE_DIR / "ui_files/"
7
+
8
+ def load_ui(uifile: str | Path, baseinstance, working_directory: str| Path = None):
9
+ from PySide6.QtCore import QDir
10
+ if working_directory is not None:
11
+ working_directory = QDir(working_directory)
12
+ return loadUi(uifile, baseinstance, working_directory)
@@ -1,81 +1,81 @@
1
- import os, sys
2
- from PySide6.QtWidgets import QMainWindow, QApplication, QMessageBox, QLineEdit, QTableWidgetItem, QPushButton, QLabel, QSlider
3
- from PySide6.QtGui import QColor, QFont
4
- from PySide6.QtCore import Qt
5
- from time import time
6
-
7
- from vtkmodules.vtkFiltersSources import vtkCylinderSource
8
- from vtkmodules.vtkRenderingCore import vtkPolyDataMapper, vtkActor
9
-
10
- from molde import MOLDE_DIR, load_ui
11
- from molde import stylesheets
12
- from molde.render_widgets.common_render_widget import CommonRenderWidget
13
-
14
-
15
- class Example(QMainWindow):
16
- def __init__(self, parent=None) -> None:
17
- super().__init__()
18
- load_ui(MOLDE_DIR / "stylesheets/mainwindow.ui", self, MOLDE_DIR / "stylesheets")
19
- self.current_theme = "light"
20
-
21
- self.change_theme_button.clicked.connect(self.change_theme)
22
- # self.render_widget: CommonRenderWidget
23
- # self.render_widget.create_axes()
24
- # self.render_widget.create_scale_bar()
25
- # self.render_widget.create_color_bar()
26
- # self.render_widget.set_info_text("Hola\nque\ntal?")
27
-
28
- cylinder = vtkCylinderSource()
29
- cylinder.SetResolution(8)
30
- cylinder_mapper = vtkPolyDataMapper()
31
- cylinder_mapper.SetInputConnection(cylinder.GetOutputPort())
32
- cylinder_actor = vtkActor()
33
- cylinder_actor.SetMapper(cylinder_mapper)
34
- # self.render_widget.add_actors(cylinder_actor)
35
-
36
- self.botao1 = QPushButton()
37
- self.label = QLabel("Olha o sapooo")
38
- self.label.setProperty("type", "logo")
39
- # font = QFont()
40
- # font.setFamily("Bauhaus 93")
41
- # self.label.setFont(font)
42
- print(self.label.fontInfo().family())
43
-
44
- self.slider = QSlider(Qt.Orientation.Horizontal)
45
- self.slider.setValue(50)
46
- self.botao1.setText("Olha a faca")
47
- self.toolbar_2.addWidget(self.label)
48
- self.toolbar_2.setDisabled(False)
49
-
50
- item = QTableWidgetItem("fr")
51
- item.setBackground(QColor("#FF0000"))
52
- self.tableWidget.setItem(0, 0, item)
53
- self.show()
54
-
55
- def change_theme(self):
56
- if self.current_theme == "light":
57
- self.current_theme = "dark"
58
- else:
59
- self.current_theme = "light"
60
-
61
- # self.render_widget.set_theme(self.current_theme)
62
- stylesheets.set_theme(self.current_theme)
63
-
64
- def closeEvent(self, event):
65
- close = QMessageBox.question(
66
- self,
67
- "QUIT",
68
- "Would you like to close the application?",
69
- QMessageBox.Yes | QMessageBox.No
70
- )
71
- QApplication.quit()
72
-
73
-
74
- if __name__ == "__main__":
75
- # Make the window scale evenly for every monitor
76
- os.environ["QT_AUTO_SCREEN_SCALE_FACTOR"] = "1"
77
-
78
- app = QApplication(sys.argv)
79
- e = Example()
80
- e.change_theme()
81
- sys.exit(app.exec())
1
+ import os, sys
2
+ from PySide6.QtWidgets import QMainWindow, QApplication, QMessageBox, QLineEdit, QTableWidgetItem, QPushButton, QLabel, QSlider
3
+ from PySide6.QtGui import QColor, QFont
4
+ from PySide6.QtCore import Qt
5
+ from time import time
6
+
7
+ from vtkmodules.vtkFiltersSources import vtkCylinderSource
8
+ from vtkmodules.vtkRenderingCore import vtkPolyDataMapper, vtkActor
9
+
10
+ from molde import MOLDE_DIR, load_ui
11
+ from molde import stylesheets
12
+ from molde.render_widgets.common_render_widget import CommonRenderWidget
13
+
14
+
15
+ class Example(QMainWindow):
16
+ def __init__(self, parent=None) -> None:
17
+ super().__init__()
18
+ load_ui(MOLDE_DIR / "stylesheets/mainwindow.ui", self, MOLDE_DIR / "stylesheets")
19
+ self.current_theme = "light"
20
+
21
+ self.change_theme_button.clicked.connect(self.change_theme)
22
+ # self.render_widget: CommonRenderWidget
23
+ # self.render_widget.create_axes()
24
+ # self.render_widget.create_scale_bar()
25
+ # self.render_widget.create_color_bar()
26
+ # self.render_widget.set_info_text("Hola\nque\ntal?")
27
+
28
+ cylinder = vtkCylinderSource()
29
+ cylinder.SetResolution(8)
30
+ cylinder_mapper = vtkPolyDataMapper()
31
+ cylinder_mapper.SetInputConnection(cylinder.GetOutputPort())
32
+ cylinder_actor = vtkActor()
33
+ cylinder_actor.SetMapper(cylinder_mapper)
34
+ # self.render_widget.add_actors(cylinder_actor)
35
+
36
+ self.botao1 = QPushButton()
37
+ self.label = QLabel("Olha o sapooo")
38
+ self.label.setProperty("type", "logo")
39
+ # font = QFont()
40
+ # font.setFamily("Bauhaus 93")
41
+ # self.label.setFont(font)
42
+ print(self.label.fontInfo().family())
43
+
44
+ self.slider = QSlider(Qt.Orientation.Horizontal)
45
+ self.slider.setValue(50)
46
+ self.botao1.setText("Olha a faca")
47
+ self.toolbar_2.addWidget(self.label)
48
+ self.toolbar_2.setDisabled(False)
49
+
50
+ item = QTableWidgetItem("fr")
51
+ item.setBackground(QColor("#FF0000"))
52
+ self.tableWidget.setItem(0, 0, item)
53
+ self.show()
54
+
55
+ def change_theme(self):
56
+ if self.current_theme == "light":
57
+ self.current_theme = "dark"
58
+ else:
59
+ self.current_theme = "light"
60
+
61
+ # self.render_widget.set_theme(self.current_theme)
62
+ stylesheets.set_theme(self.current_theme)
63
+
64
+ def closeEvent(self, event):
65
+ close = QMessageBox.question(
66
+ self,
67
+ "QUIT",
68
+ "Would you like to close the application?",
69
+ QMessageBox.Yes | QMessageBox.No
70
+ )
71
+ QApplication.quit()
72
+
73
+
74
+ if __name__ == "__main__":
75
+ # Make the window scale evenly for every monitor
76
+ os.environ["QT_AUTO_SCREEN_SCALE_FACTOR"] = "1"
77
+
78
+ app = QApplication(sys.argv)
79
+ e = Example()
80
+ e.change_theme()
81
+ sys.exit(app.exec())
@@ -1,5 +1,5 @@
1
- from .ghost_actor import GhostActor
2
- from .lines_actor import LinesActor
3
- from .round_points_actor import RoundPointsActor
4
- from .square_points_actor import SquarePointsActor
5
- from .common_symbols_actor import CommonSymbolsActorFixedSize, CommonSymbolsActorVariableSize
1
+ from .ghost_actor import GhostActor
2
+ from .lines_actor import LinesActor
3
+ from .round_points_actor import RoundPointsActor
4
+ from .square_points_actor import SquarePointsActor
5
+ from .common_symbols_actor import CommonSymbolsActorFixedSize, CommonSymbolsActorVariableSize
@@ -1,164 +1,148 @@
1
- from dataclasses import dataclass
2
-
3
- from molde import Color
4
- from molde.utils import transform_polydata
5
- from vtkmodules.vtkCommonCore import (
6
- vtkDoubleArray,
7
- vtkIntArray,
8
- vtkPoints,
9
- vtkUnsignedCharArray,
10
- )
11
- from vtkmodules.vtkCommonDataModel import vtkPolyData
12
- from vtkmodules.vtkRenderingCore import vtkActor, vtkDistanceToCamera, vtkGlyph3DMapper, vtkRenderer
13
-
14
- Triple = tuple[float, float, float]
15
-
16
-
17
- @dataclass
18
- class Symbol:
19
- shape_name: str
20
- position: Triple
21
- orientation: Triple
22
- color: Color
23
- scale: float
24
-
25
-
26
- class CommonSymbolsActor(vtkActor):
27
- def __init__(self, *args, **kwargs):
28
- self._shapes: dict[str, vtkPolyData] = dict()
29
- self._symbols: list[Symbol] = list()
30
-
31
- def register_shape(
32
- self,
33
- name: str,
34
- shape: vtkPolyData,
35
- position: Triple = (0, 0, 0),
36
- rotation: Triple = (0, 0, 0),
37
- scale: Triple = (1, 1, 1),
38
- ):
39
- self._shapes[name] = transform_polydata(
40
- shape,
41
- position,
42
- rotation,
43
- scale,
44
- )
45
-
46
- def add_symbol(
47
- self,
48
- shape_name: str,
49
- position: Triple,
50
- orientation: Triple,
51
- color: Triple,
52
- scale: float = 1,
53
- ):
54
- symbol = Symbol(
55
- shape_name,
56
- position,
57
- orientation,
58
- color,
59
- scale,
60
- )
61
- self._symbols.append(symbol)
62
-
63
- def clear_shapes(self):
64
- self._shapes.clear()
65
-
66
- def clear_symbols(self):
67
- self._symbols.clear()
68
-
69
- def clear_all(self):
70
- self.clear_shapes()
71
- self.clear_symbols()
72
-
73
- def common_build(self):
74
- self.data = vtkPolyData()
75
- points = vtkPoints()
76
-
77
- self.mapper: vtkGlyph3DMapper = vtkGlyph3DMapper()
78
- self.SetMapper(self.mapper)
79
-
80
- sources = vtkIntArray()
81
- sources.SetName("sources")
82
-
83
- rotations = vtkDoubleArray()
84
- rotations.SetNumberOfComponents(3)
85
- rotations.SetName("rotations")
86
-
87
- scales = vtkDoubleArray()
88
- scales.SetName("scales")
89
-
90
- colors = vtkUnsignedCharArray()
91
- colors.SetNumberOfComponents(3)
92
- colors.SetName("colors")
93
-
94
- shape_name_to_index = dict()
95
- for index, (name, shape) in enumerate(self._shapes.items()):
96
- shape_name_to_index[name] = index
97
- self.mapper.SetSourceData(index, shape)
98
-
99
- for symbol in self._symbols:
100
- points.InsertNextPoint(symbol.position)
101
- rotations.InsertNextTuple(symbol.orientation)
102
- colors.InsertNextTuple(symbol.color.to_rgb())
103
- scales.InsertNextValue(symbol.scale)
104
- sources.InsertNextValue(shape_name_to_index[symbol.shape_name])
105
-
106
- self.data.SetPoints(points)
107
- self.data.GetPointData().AddArray(sources)
108
- self.data.GetPointData().AddArray(rotations)
109
- self.data.GetPointData().AddArray(scales)
110
- self.data.GetPointData().SetScalars(colors)
111
-
112
- def set_zbuffer_offsets(self, factor: float, units: float):
113
- """
114
- This functions is usefull to make a object appear in front of the others.
115
- If the object should never be hidden, the parameters should be set to
116
- factor = 1 and offset = -66000.
117
- """
118
- self.mapper.SetResolveCoincidentTopologyToPolygonOffset()
119
- self.mapper.SetRelativeCoincidentTopologyLineOffsetParameters(factor, units)
120
- self.mapper.SetRelativeCoincidentTopologyPolygonOffsetParameters(factor, units)
121
- self.mapper.SetRelativeCoincidentTopologyPointOffsetParameter(units)
122
- self.mapper.Update()
123
-
124
-
125
- class CommonSymbolsActorFixedSize(CommonSymbolsActor):
126
- def build(self):
127
- self.common_build()
128
-
129
- self.mapper.SetInputData(self.data)
130
- self.mapper.SetSourceIndexArray("sources")
131
- self.mapper.SetOrientationArray("rotations")
132
- self.mapper.SetScaleArray("scales")
133
- self.mapper.SourceIndexingOn()
134
- self.mapper.ScalarVisibilityOn()
135
- self.mapper.SetScaleModeToScaleByMagnitude()
136
- self.mapper.SetScalarModeToUsePointData()
137
- self.mapper.SetOrientationModeToDirection()
138
- self.mapper.Update()
139
-
140
-
141
- class CommonSymbolsActorVariableSize(CommonSymbolsActor):
142
- def __init__(self, renderer: vtkRenderer):
143
- super().__init__()
144
- self.renderer = renderer
145
-
146
- def build(self):
147
- self.common_build()
148
-
149
- distance_to_camera = vtkDistanceToCamera()
150
- distance_to_camera.SetInputData(self.data)
151
- distance_to_camera.SetScreenSize(40)
152
- distance_to_camera.SetRenderer(self.renderer)
153
-
154
- self.mapper.SetInputConnection(distance_to_camera.GetOutputPort())
155
- self.mapper.SetSourceIndexArray("sources")
156
- self.mapper.SetOrientationArray("rotations")
157
- self.mapper.SetScaleArray("DistanceToCamera")
158
- self.mapper.SourceIndexingOn()
159
- self.mapper.ScalarVisibilityOn()
160
- self.mapper.SetScaleModeToScaleByMagnitude()
161
- self.mapper.SetScalarModeToUsePointData()
162
- self.mapper.SetOrientationModeToDirection()
163
-
164
- self.mapper.Update()
1
+ from dataclasses import dataclass
2
+ from typing import Callable
3
+
4
+ from vtkmodules.vtkCommonCore import (
5
+ vtkDoubleArray,
6
+ vtkIntArray,
7
+ vtkPoints,
8
+ vtkUnsignedCharArray,
9
+ )
10
+ from vtkmodules.vtkCommonDataModel import vtkPolyData
11
+ from vtkmodules.vtkRenderingCore import (
12
+ vtkActor,
13
+ vtkDistanceToCamera,
14
+ vtkGlyph3DMapper,
15
+ vtkRenderer,
16
+ )
17
+
18
+ from molde import Color
19
+
20
+ Triple = tuple[float, float, float]
21
+
22
+
23
+ @dataclass
24
+ class Symbol:
25
+ shape_function: Callable
26
+ position: Triple
27
+ orientation: Triple
28
+ color: Color
29
+ scale: float
30
+
31
+
32
+ class CommonSymbolsActor(vtkActor):
33
+ def __init__(self, *args, **kwargs):
34
+ self._symbols: list[Symbol] = list()
35
+
36
+ def add_symbol(
37
+ self,
38
+ shape_function: Callable,
39
+ position: Triple,
40
+ orientation: Triple,
41
+ color: Triple,
42
+ scale: float = 1,
43
+ ):
44
+ symbol = Symbol(
45
+ shape_function,
46
+ position,
47
+ orientation,
48
+ color,
49
+ scale,
50
+ )
51
+ self._symbols.append(symbol)
52
+
53
+ def clear_symbols(self):
54
+ self._symbols.clear()
55
+
56
+ def common_build(self):
57
+ self.data = vtkPolyData()
58
+ points = vtkPoints()
59
+
60
+ self.mapper: vtkGlyph3DMapper = vtkGlyph3DMapper()
61
+ self.SetMapper(self.mapper)
62
+
63
+ sources = vtkIntArray()
64
+ sources.SetName("sources")
65
+
66
+ rotations = vtkDoubleArray()
67
+ rotations.SetNumberOfComponents(3)
68
+ rotations.SetName("rotations")
69
+
70
+ scales = vtkDoubleArray()
71
+ scales.SetName("scales")
72
+
73
+ colors = vtkUnsignedCharArray()
74
+ colors.SetNumberOfComponents(4)
75
+ colors.SetName("colors")
76
+
77
+ shape_function_to_index = dict()
78
+ for symbol in self._symbols:
79
+ if symbol.shape_function not in shape_function_to_index:
80
+ index = len(shape_function_to_index)
81
+ shape_function_to_index[symbol.shape_function] = index
82
+ self.mapper.SetSourceData(index, symbol.shape_function())
83
+
84
+ points.InsertNextPoint(symbol.position)
85
+ rotations.InsertNextTuple(symbol.orientation)
86
+ colors.InsertNextTuple(symbol.color.to_rgba())
87
+ scales.InsertNextValue(symbol.scale)
88
+ sources.InsertNextValue(shape_function_to_index[symbol.shape_function])
89
+
90
+ self.data.SetPoints(points)
91
+ self.data.GetPointData().AddArray(sources)
92
+ self.data.GetPointData().AddArray(rotations)
93
+ self.data.GetPointData().AddArray(scales)
94
+ self.data.GetPointData().SetScalars(colors)
95
+
96
+ def set_zbuffer_offsets(self, factor: float, units: float):
97
+ """
98
+ This functions is usefull to make a object appear in front of the others.
99
+ If the object should never be hidden, the parameters should be set to
100
+ factor = 1 and offset = -66000.
101
+ """
102
+ self.mapper.SetResolveCoincidentTopologyToPolygonOffset()
103
+ self.mapper.SetRelativeCoincidentTopologyLineOffsetParameters(factor, units)
104
+ self.mapper.SetRelativeCoincidentTopologyPolygonOffsetParameters(factor, units)
105
+ self.mapper.SetRelativeCoincidentTopologyPointOffsetParameter(units)
106
+ self.mapper.Update()
107
+
108
+
109
+ class CommonSymbolsActorFixedSize(CommonSymbolsActor):
110
+ def build(self):
111
+ self.common_build()
112
+
113
+ self.mapper.SetInputData(self.data)
114
+ self.mapper.SetSourceIndexArray("sources")
115
+ self.mapper.SetOrientationArray("rotations")
116
+ self.mapper.SetScaleArray("scales")
117
+ self.mapper.SourceIndexingOn()
118
+ self.mapper.ScalarVisibilityOn()
119
+ self.mapper.SetScaleModeToScaleByMagnitude()
120
+ self.mapper.SetScalarModeToUsePointData()
121
+ self.mapper.SetOrientationModeToDirection()
122
+ self.mapper.Update()
123
+
124
+
125
+ class CommonSymbolsActorVariableSize(CommonSymbolsActor):
126
+ def __init__(self, renderer: vtkRenderer):
127
+ super().__init__()
128
+ self.renderer = renderer
129
+
130
+ def build(self):
131
+ self.common_build()
132
+
133
+ distance_to_camera = vtkDistanceToCamera()
134
+ distance_to_camera.SetInputData(self.data)
135
+ distance_to_camera.SetScreenSize(40)
136
+ distance_to_camera.SetRenderer(self.renderer)
137
+
138
+ self.mapper.SetInputConnection(distance_to_camera.GetOutputPort())
139
+ self.mapper.SetSourceIndexArray("sources")
140
+ self.mapper.SetOrientationArray("rotations")
141
+ self.mapper.SetScaleArray("DistanceToCamera")
142
+ self.mapper.SourceIndexingOn()
143
+ self.mapper.ScalarVisibilityOn()
144
+ self.mapper.SetScaleModeToScaleByMagnitude()
145
+ self.mapper.SetScalarModeToUsePointData()
146
+ self.mapper.SetOrientationModeToDirection()
147
+
148
+ self.mapper.Update()
@@ -1,12 +1,12 @@
1
- from vtkmodules.vtkRenderingCore import vtkActor
2
-
3
-
4
- class GhostActor(vtkActor):
5
- def make_ghost(self):
6
- self.GetProperty().LightingOff()
7
- offset = -66000
8
- mapper = self.GetMapper()
9
- mapper.SetResolveCoincidentTopologyToPolygonOffset()
10
- mapper.SetRelativeCoincidentTopologyLineOffsetParameters(0, offset)
11
- mapper.SetRelativeCoincidentTopologyPolygonOffsetParameters(0, offset)
12
- mapper.SetRelativeCoincidentTopologyPointOffsetParameter(offset)
1
+ from vtkmodules.vtkRenderingCore import vtkActor
2
+
3
+
4
+ class GhostActor(vtkActor):
5
+ def make_ghost(self):
6
+ self.GetProperty().LightingOff()
7
+ offset = -66000
8
+ mapper = self.GetMapper()
9
+ mapper.SetResolveCoincidentTopologyToPolygonOffset()
10
+ mapper.SetRelativeCoincidentTopologyLineOffsetParameters(0, offset)
11
+ mapper.SetRelativeCoincidentTopologyPolygonOffsetParameters(0, offset)
12
+ mapper.SetRelativeCoincidentTopologyPointOffsetParameter(offset)