add Node.copy()

fplll · Aug 20, 2018 · 6e191ce · 6e191ce
1 parent 4f568a2
commit 6e191ce
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/fpylll/tools/bkz_stats.py b/src/fpylll/tools/bkz_stats.py
@@ -647,6 +647,17 @@ def __sub__(self, rhs):
                 print("Skipping missing node '%s'"%lchild.label)
         return diff
 
+    def copy(self, deepcopy=True):
+        """
+        Return a (deep)copy of this node.
+
+        :param deepcopy: If ``False`` child nodes and data dictionaries are not copied, this is
+            usually not what the user wants.
+        """
+        if deepcopy:
+            return copy.deepcopy(self)
+        else:
+            return copy.copy(self)
 
 class TimeTreeTracer(Tracer):
     """