Cryptic (deep)copy of TextPath

deep-jkl · deep-jkl · commit ce1ead100599 · 2021-08-29T20:32:38.000+02:00
This time, revalidation is enabled. Added test for member deepcopy().
diff --git a/lib/matplotlib/tests/test_textpath.py b/lib/matplotlib/tests/test_textpath.py
@@ -15,17 +15,21 @@ def test_set_size():
 
 
 def test_deepcopy():
-    # Should not raise any error
     path = TextPath((0, 0), ".")
     path_copy = deepcopy(path)
     assert isinstance(path_copy, TextPath)
     assert path is not path_copy
     assert path.vertices is not path_copy.vertices
     assert path.codes is not path_copy.codes
+    path = TextPath((0, 0), ".")
+    path_copy = path.deepcopy({})
+    assert isinstance(path_copy, TextPath)
+    assert path is not path_copy
+    assert path.vertices is not path_copy.vertices
+    assert path.codes is not path_copy.codes
 
 
 def test_copy():
-    # Should not raise any error
     path = TextPath((0, 0), ".")
     path_copy = copy(path)
     assert path is not path_copy
diff --git a/lib/matplotlib/textpath.py b/lib/matplotlib/textpath.py
@@ -1,3 +1,4 @@
+from copy import deepcopy
 from collections import OrderedDict
 import functools
 import logging
@@ -429,3 +430,23 @@ def _revalidate_path(self):
             self._cached_vertices = tr.transform(self._vertices)
             self._cached_vertices.flags.writeable = False
             self._invalid = False
+
+    def __deepcopy__(self, memo):
+        # taken from https://stackoverflow.com/a/15774013
+        self._revalidate_path()
+        cls = self.__class__
+        new_instance = cls.__new__(cls)
+        memo[id(self)] = new_instance
+        for k, v in self.__dict__.items():
+            setattr(new_instance, k, deepcopy(v, memo))
+        return new_instance
+
+    deepcopy = __deepcopy__
+
+    def __copy__(self):
+        # taken from https://stackoverflow.com/a/15774013
+        self._revalidate_path()
+        cls = self.__class__
+        new_instance = cls.__new__(cls)
+        new_instance.__dict__.update(self.__dict__)
+        return new_instance
