Fix a bug when the user wants to reuse the visualization component

orbingol · Jan 9, 2018 · 107c9b5 · 107c9b5
1 parent a006a92
commit 107c9b5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/geomdl/BSpline.py b/geomdl/BSpline.py
@@ -220,6 +220,7 @@ def render(self, cpcolor="blue", curvecolor="black"):
             self.evaluate()
 
         # Run the visualization component
+        self._vis_component.clear()
         self._vis_component.add(self.ctrlpts, "Control Points", cpcolor)
         self._vis_component.add(self.curvepts, "Curve", curvecolor)
         self._vis_component.render()
@@ -1067,6 +1068,7 @@ def render(self, cpcolor="blue", surfcolor="green"):
         ctrlpts = utils.make_quad(self.ctrlpts, self._control_points_size_v, self._control_points_size_u)
 
         # Run the visualization component
+        self._vis_component.clear()
         self._vis_component.add(ctrlpts, "Control Points", cpcolor)
         self._vis_component.add(self.surfpts, "Surface", surfcolor)
         self._vis_component.render()

diff --git a/geomdl/visualization/VisBase.py b/geomdl/visualization/VisBase.py
@@ -18,6 +18,12 @@ def __init__(self):
         self._colors = []
         self._names = []
 
+    def clear(self):
+        if self._points:
+            self._points[:] = []
+            self._colors[:] = []
+            self._names[:] = []
+
     def add(self, ptsarr=(), name=None, color=None):
         if not ptsarr or not color or not name:
             return