fix drawer

KOS-UJ · Sep 14, 2023 · 459a1a0 · 459a1a0
1 parent 39552c6
commit 459a1a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/conmech/plotting/drawer.py b/conmech/plotting/drawer.py
@@ -24,7 +24,7 @@ def __init__(self, state, config: Config):
         self.state = state
         self.config = config
         self.mesh = state.body.mesh
-        self.node_size = 2 + (300 / len(self.mesh.initial_nodes))
+        self.node_size = 2 + (300 / len(self.mesh.nodes))
         self.line_width = self.node_size / 2
         self.deformed_mesh_color = "k"
         self.original_mesh_color = "0.7"
@@ -123,7 +123,7 @@ def set_axes_limits(self, axes, foundation):
     def draw_meshes(self, axes):
         if self.original_mesh_color is not None:
             self.draw_mesh(
-                self.mesh.initial_nodes,
+                self.mesh.nodes,
                 axes,
                 label="Original",
                 node_color=self.original_mesh_color,