Merge pull request #1022 from JeffreyWardman/enable-using-copy-library

allow mesh classes to handle copy.deepcopy

vedo/pointcloud.py

@@ -827,6 +827,12 @@ def polydata(self, **kwargs):
             c="y")
         return self.dataset
 
+    def __copy__(self):
+        return self.clone(deep=False)
+
+    def __deepcopy__(self, memo):
+        return self.clone(deep=memo)
+
     def copy(self, deep=True):
         """Return a copy of the object. Alias of `clone()`."""
         return self.clone(deep=deep)
@@ -846,7 +852,7 @@ def clone(self, deep=True):
                ![](https://vedo.embl.es/images/basic/mirror.png)
         """
         poly = vtk.vtkPolyData()
-        if deep:
+        if deep or isinstance(deep, dict): # if a memo object is passed this checks as True
             poly.DeepCopy(self.dataset)
         else:
             poly.ShallowCopy(self.dataset)
@@ -856,19 +862,18 @@ def clone(self, deep=True):
         else:
             cloned = Points(poly)
 
-        # new_instance = self.__class__
-        # print("******* cloning", new_instance.__name__)
-        # cloned = new_instance(poly)
-
         cloned.transform = self.transform.clone()
 
         cloned.copy_properties_from(self)
 
         cloned.name = str(self.name)
         cloned.filename = str(self.filename)
         cloned.info = dict(self.info)
-
         cloned.pipeline = utils.OperationNode("clone", parents=[self], shape="diamond", c="#edede9")
+
+        if isinstance(deep, dict):
+            deep[id(self)] = cloned
+
         return cloned
 
     def compute_normals_with_pca(self, n=20, orientation_point=None, invert=False):