add close option

gustaf/helpers/notebook.py

@@ -121,22 +121,34 @@ def show(
         if len(args) != 0 or len(kwargs) != 0:
             self._logd(f"*args ({args}) and **kwargs ({kwargs}) ignored")
 
+        # this converts vedo plotter to k3d plot.
+        # after this, vedo plotter has no relevance to what you see
         self[self._at_get_location(at)] = self.vedo_plotters[at].show(
             list_of_showables,
             interactive=interactive,
             camera=camera,
             axes=axes,
         )
 
-    def display(self):
+    def display(self, close=True):
         """Display the plotter.
 
         This is needed in case the plotter is the last thing in a cell. In that
         case the IPython will try to call this function to display this.
         """
         display(self)
 
+        # we add option to close here, as we set default_autoclose = False
+        if close:
+            self.clear()
+            self.close()
+
     def clear(self, *args, **kwargs):
         """Clear the plotters."""
-        for renderer in self.vedo_plotters:
-            renderer.clear(*args, deep=True, **kwargs)
+        for v_plotter in self.vedo_plotters:
+            v_plotter.clear(*args, deep=True, **kwargs)
+
+    def close(self):
+        """Closes all vedo.Plotters"""
+        for v_plotter in self.vedo_plotters:
+            v_plotter.close()

gustaf/show.py

@@ -59,10 +59,7 @@ def __call__(self, *args, **kwargs):
     is_ipython = False
 
 
-def show(
-    *args,
-    **kwargs,
-):
+def show(*args, **kwargs):
     """`vedo.show` wrapper. Each args represent one section of window. In other
     words len(args) == N, where N corresponds to the parameter for vedo.show().
 
@@ -222,7 +219,7 @@ def cam_tuple_to_list(dict_cam):
             )
 
     if is_ipython:
-        plt.display()
+        plt.display(close=close)
         return None
 
     if interact and not offs: