Skip to content

Commit cb15a19

Browse files
committed
feat: add parameters to Renderer object
1 parent e64f98b commit cb15a19

File tree

1 file changed

+38
-20
lines changed

1 file changed

+38
-20
lines changed

src/pynite_plotly/rendering.py

Lines changed: 38 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -14,23 +14,12 @@ class Renderer:
1414

1515
scalar = None
1616

17-
def __init__(self, model):
18-
self.model = model
19-
20-
# Default settings for rendering
21-
self._annotation_size = 5
22-
self._deformed_shape = False
23-
self._deformed_scale = 30
24-
self._render_nodes = True
25-
self._render_loads = True
26-
self._color_map = None
27-
self._combo_name = "Combo 1"
28-
self._case = None
29-
self._labels = True
30-
self._scalar_bar = False
31-
self._scalar_bar_text_size = 24
32-
self.theme = "default"
33-
self.colors = dict(
17+
def __init__(
18+
self,
19+
model,
20+
combo_name: str,
21+
annotation_size: int = 5,
22+
colors: dict = dict(
3423
annotation_text="black",
3524
annotation_point="grey",
3625
point_label_text="green",
@@ -42,7 +31,36 @@ def __init__(self, model):
4231
dist_load="green",
4332
moment_load="green",
4433
area_load="green",
45-
)
34+
),
35+
deformed_scale: float = 30.0,
36+
deformed_shape: bool = False,
37+
labels: bool = True,
38+
line_widths = dict(
39+
member=4,
40+
loads=2,
41+
deformed_member=2,
42+
spring=3
43+
),
44+
title: str = "Pynite - Simple Finite Element Analysis for Python",
45+
height: int = 800,
46+
width: int = 800,
47+
):
48+
self.model = model
49+
50+
# Default settings for rendering
51+
self._annotation_size = annotation_size
52+
self._deformed_shape = deformed_shape
53+
self._deformed_scale = deformed_scale
54+
self._render_nodes = True
55+
self._render_loads = True
56+
self._color_map = None
57+
self._combo_name = combo_name
58+
self._case = None
59+
self._labels = labels
60+
self._scalar_bar = False
61+
self._scalar_bar_text_size = 24
62+
self.theme = "default"
63+
self.colors = colors
4664
self.line_widths = dict(
4765
member=4,
4866
loads=2,
@@ -56,8 +74,8 @@ def __init__(self, model):
5674

5775

5876
self._layout = default_layout(self._title)
59-
self.window_width = 800
60-
self.window_height = 800
77+
self.window_width = width
78+
self.window_height = height
6179

6280
# self.plotter.set_background('white') # Setting background color
6381
# # self.plotter.add_logo_widget('./Resources/Full Logo No Buffer - Transparent.png')

0 commit comments

Comments
 (0)