From cc081cdb3cc9c002ed491bea1de74dadae2ca281 Mon Sep 17 00:00:00 2001
From: David Halter <davidhalter88@gmail.com>
Date: Thu, 12 Jul 2012 15:10:57 +0200
Subject: [PATCH] deepcopy removed at one place, tests being 12s faster (from
 20s to 8s)

---
 evaluate.py |  3 ++-
 helpers.py  | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 34 insertions(+), 1 deletion(-)

diff --git a/evaluate.py b/evaluate.py
index e509e18b..949634da 100644
--- a/evaluate.py
+++ b/evaluate.py
@@ -245,7 +245,8 @@ class InstanceElement(object):
         # Copy and modify the array.
         origin = self.var.get_assignment_calls()
         origin.parent_stmt, temp = None, origin.parent_stmt
-        new = copy.deepcopy(origin)
+        # Delete parent, because it isn't used anymore.
+        new = helpers.fast_parent_copy(origin)
         origin.parent_stmt = temp
         new.parent_stmt = InstanceElement(self.instance, temp)
         return new
diff --git a/helpers.py b/helpers.py
index a93edddd..1eaee3a9 100644
--- a/helpers.py
+++ b/helpers.py
@@ -1,5 +1,6 @@
 import parsing
 import debug
+import copy
 
 class RecursionDecorator(object):
     """ A decorator to detect recursions in statements """
@@ -45,3 +46,34 @@ class RecursionNode(object):
         if not other:
             return None
         return self.script == other.script and self.position == other.position
+
+
+def fast_parent_copy(obj):
+    """
+    Much, much faster than deepcopy, but just for the elements in `classes`.
+    """
+    new_elements = {}
+    classes = (parsing.Call, parsing.Scope)
+    def recursion(obj):
+        new_obj = copy.copy(obj)
+        new_elements[obj] = new_obj
+        try:
+            new_obj.parent = new_elements[obj.parent]
+        except KeyError:
+            pass
+
+        #print new_obj.__dict__
+        for key, value in new_obj.__dict__.items():
+            if isinstance(value, list):
+                new_obj.__dict__[key] = list_rec(value)
+        return new_obj
+
+    def list_rec(list_obj):
+        copied_list = list_obj[:]   # lists, tuples, strings, unicode
+        for i, el in enumerate(copied_list):
+            if isinstance(el, classes):
+                copied_list[i] = recursion(el)
+            elif isinstance(el, list):
+                copied_list[i] = list_rec(el)
+        return copied_list
+    return recursion(obj)