Merge pull request #54 from TurBoss/vtk_no_file_fix

Fix 3D plot exceptions without file loaded
kcjengr · Oct 13, 2020 · 443019a · 443019a
2 parents bea56d3 + d5c33f0
commit 443019a
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/qtpyvcp/widgets/display_widgets/vtk_backplot/vtk_backplot.py b/qtpyvcp/widgets/display_widgets/vtk_backplot/vtk_backplot.py
@@ -274,12 +274,13 @@ def draw_lines(self):
 
             self.path_points[self.origin] = list()
 
-            path_actor.poly_data.SetPoints(path_actor.points)
-            path_actor.poly_data.SetLines(path_actor.lines)
-            path_actor.poly_data.GetCellData().SetScalars(path_actor.colors)
-            path_actor.data_mapper.SetInputData(path_actor.poly_data)
-            path_actor.data_mapper.Update()
-            path_actor.SetMapper(path_actor.data_mapper)
+            if path_actor is not None:
+                path_actor.poly_data.SetPoints(path_actor.points)
+                path_actor.poly_data.SetLines(path_actor.lines)
+                path_actor.poly_data.GetCellData().SetScalars(path_actor.colors)
+                path_actor.data_mapper.SetInputData(path_actor.poly_data)
+                path_actor.data_mapper.Update()
+                path_actor.SetMapper(path_actor.data_mapper)
 
     def get_path_actors(self):
         return self.path_actors