Changeset 99

Timestamp:

07/21/06 11:57:12 (2 years ago)

Author:

mk

Message:

Added automatic timing for indices computation methods.

Files:

• branches/mk/cheesecake/cheesecake_index.py (modified) (2 diffs)
• branches/mk/cheesecake/util.py (modified) (2 diffs)

branches/mk/cheesecake/cheesecake_index.py

@@ -36 +36 @@
-from util import mkdirs
-from util import StdoutRedirector
+from util import time_function
-from config import get_pkg_config
-from codeparser import CodeParser
@@ -316 +317 @@
-        if 'name' not in dict:
-            setattr(cls, 'name', name)
+
+        if 'compute_with' in dict:
+            orig_compute_with = cls.compute_with
+
+            def _timed_compute_with(self, cheesecake):
+                (ret, self.time_taken) = time_function(lambda: orig_compute_with(self, cheesecake))
+                self.cheesecake.log.debug("Index %s computed in %.2f seconds." % (self.name, self.time_taken))
+                return ret
+
+            setattr(cls, 'compute_with', _timed_compute_with)
-
-    def __repr__(cls):

branches/mk/cheesecake/util.py

@@ -3 +3 @@
-import sys
-import tarfile
+import time
-import zipfile
-
@@ -177 +178 @@
-        if not os.path.exists(path):
-            os.mkdir(path)
+
+def time_function(function):
+    """Measure function execution time.
+
+    Return (return value, time taken) tuple.
+
+    >>> def fun(x):
+    ...     return x*2
+    >>> ret, time_taken = time_function(lambda: fun(5))
+    >>> ret
+    10
+    """
+    start = time.time()
+    ret = function()
+    end = time.time()
+    return ret, end-start