From 6c377d6e91c0cdd4630d2d99ca15cbf80e5f8d35 Mon Sep 17 00:00:00 2001
From: David Halter <davidhalter88@gmail.com>
Date: Sun, 25 Mar 2012 12:59:04 +0200
Subject: [PATCH] cache and executions work now recursive

---
 evaluate.py | 27 +++++++++++++++++++++------
 1 file changed, 21 insertions(+), 6 deletions(-)

diff --git a/evaluate.py b/evaluate.py
index bed30c3c..afc01e01 100644
--- a/evaluate.py
+++ b/evaluate.py
@@ -42,15 +42,28 @@ class Execution(Exec):
     This class is used to evaluate functions and their returns.
     """
 
+    cache = {}
     def get_return_types(self):
         """
         Get the return vars of a function.
         """
+        # check cache
+        try:
+            return Execution.cache[self.base]
+        except KeyError:
+            # cache is not only here as a cache, but also to prevent an
+            # endless recursion.
+            Execution.cache[self.base] = []
+        def remove_executions(scope):
+            if isinstance(scope, Execution):
+                # there maybe executions of executions
+                stmts = scope.get_return_types()
+            else:
+                stmts = scope.returns
+            return stmts
         result = []
-        if isinstance(self.base, Execution):
-            stmts = self.base.get_return_types()
-        else:
-            stmts = self.base.returns
+        stmts = remove_executions(self.base)
+        print 'stmts=', stmts, self.base, repr(self)
 
         #n += self.function.get_set_vars()
         # these are the statements of the return functions
@@ -61,9 +74,11 @@ class Execution(Exec):
                 result.append(Instance(stmt))
             else:
                 print 'addstmt', stmt
-                result += follow_statement(stmt)
+                for followed in follow_statement(stmt):
+                    result += remove_executions(followed)
 
             print 'ret', stmt
+        Execution.cache[self.base] = result
         return result
 
     def __repr__(self):
@@ -174,7 +189,6 @@ def follow_paths(path, results):
         print 'enter', results, len(results)
         if len(results):
             for i, r in enumerate(results):
-                print 1
                 results_new += follow_path(iter_paths[i], r)
     except StopIteration:
         return results
@@ -204,6 +218,7 @@ def follow_path(path, input):
                                     search_func=input.get_instance_vars)
             elif isinstance(input, Execution):
                 #try:
+                    print '\n\n\n\n\nbefexec', input
                     stmts = input.get_return_types()
                     print 'exec', stmts
                     for s in stmts: