From 3aa00b69bf01f6b2a2f920642f8faa6a52bbb1c4 Mon Sep 17 00:00:00 2001
From: Stan Seibert <stan@mtrr.org>
Date: Thu, 19 Jan 2012 17:49:01 -0500
Subject: Pull memoization decorator out to tools module.

---
 chroma/tools.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/chroma/tools.py b/chroma/tools.py
index ecfe976..1e111ff 100644
--- a/chroma/tools.py
+++ b/chroma/tools.py
@@ -127,3 +127,29 @@ def offset(points, x):
         offset_points.append((a + j*(b-a)))
 
     return np.array(offset_points)
+
+def memoize_method_with_dictionary_arg(func):
+    def lookup(*args):
+        # based on function by Michele Simionato
+        # http://www.phyast.pitt.edu/~micheles/python/
+        # Modified to work for class method with dictionary argument
+
+        assert len(args) == 2
+        # create hashable arguments by replacing dictionaries with tuples of items
+        dict_items = args[1].items()
+        dict_items.sort()
+        hashable_args = (args[0], tuple(dict_items))
+        try:
+            return func._memoize_dic[hashable_args]
+        except AttributeError:
+            # _memoize_dic doesn't exist yet.
+
+            result = func(*args)
+            func._memoize_dic = {hashable_args: result}
+            return result
+        except KeyError:
+            result = func(*args)
+            func._memoize_dic[hashable_args] = result
+            return result
+    return lookup
+
-- 
cgit