implement fcncache

arrowed/compiler.py

@@ -89,7 +89,12 @@ def __call__(self, resolved, *args):
         else:
             return out
 
-def compile(function, paramtypes, env={}, numbaargs={"nopython": True, "nogil": True}, debug=False):
+def compile(function, paramtypes, env={}, numbaargs={"nopython": True, "nogil": True}, fcncache=None, debug=False):
+    if fcncache is not None:
+        fcnkey = (function, tuple(sorted(paramtypes.items())), tuple(sorted(env.items())), tuple(sorted(numbaargs.items())))
+        if fcnkey in fcncache:
+            return fcncache[fcnkey]
+
     # turn the 'function' argument into the syntax tree of a function
     if isinstance(function, string_types):
         sourcefile = "<string>"
@@ -199,7 +204,12 @@ def sym(key):
         #             print(projection.format("         "))
         print("")
 
-    return Compiled(transformed, parameters, env, numbaargs)
+    out = Compiled(transformed, parameters, env, numbaargs)
+
+    if fcncache is not None:
+        fcncache[fcnkey] = out
+
+    return out
 
 ################################################################ functions inserted into code
 

arrowed/schema.py

@@ -81,21 +81,21 @@ def hasbase(self, base):
     def proxy(self, index=0):
         raise TypeError("cannot get a proxy for an unresolved ObjectArrayMap; call the resolved method first or pass a source to this method")
 
-    def compile(self, function, paramtypes={}, env={}, numba={"nopython": True, "nogil": True}, debug=False):
+    def compile(self, function, paramtypes={}, env={}, numba={"nopython": True, "nogil": True}, fcncache=None, debug=False):
         import arrowed.compiler
         paramtypes = paramtypes.copy()
         paramtypes[0] = self
 
-        return arrowed.compiler.compile(function, paramtypes, env=env, numbaargs=numba, debug=debug)
+        return arrowed.compiler.compile(function, paramtypes, env=env, numbaargs=numba, fcncache=fcncache, debug=debug)
 
-    def run(self, function, paramtypes={}, env={}, numba={"nopython": True, "nogil": True}, debug=False, *args):
+    def run(self, function, paramtypes={}, env={}, numba={"nopython": True, "nogil": True}, fcncache=None, debug=False, *args):
         import arrowed.compiler
 
         if not isinstance(function, arrowed.compiler.Compiled):
             base = self
             while base.base is not None:
                 base = base.base
-            function = base.compile(function, paramtypes=paramtypes, env=env, numba=numba, debug=debug)
+            function = base.compile(function, paramtypes=paramtypes, env=env, numba=numba, fcncache=fcncache, debug=debug)
 
         return function(self, *args)
 

arrowed/version.py

@@ -16,7 +16,7 @@
 
 import re
 
-__version__ = "0.0.10"
+__version__ = "0.0.11"
 version = __version__
 version_info = tuple(re.split(r"[-\.]", __version__))