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

Author: orbingol

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()

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