Changeset 554 for branches

Timestamp:

09/05/12 17:55:08 (4 years ago)

Author:

mmckerns

Message:

added Monitor with built-in timer to branches/decorate

File:

• branches/decorate/timer.py (modified) (1 diff)

branches/decorate/timer.py

@@ -6 +6 @@
         t = time()
         res = f(*args, **kwds)
-        print "Timed: %s" % (time() - t)
+        print "Timed: %s" % (time() - t) #XXX: write to a monitor instead ?
         return res
     return func
 
 
+class Monitor(object):
+    """
+Instances of objects that can be passed as monitors.
+Typically, a Monitor logs a list of parameters and the
+corresponding costs, retrievable by accessing the Monitor's
+member variables.
+
+example usage...
+    >>> sow = Monitor()
+    >>> sow([1,2],3)
+    >>> sow([4,5],6)
+    >>> sow.x
+    [[1, 2], [4, 5]]
+    >>> sow.y
+    [3, 6]
+
+    """
+    def __init__(self, **kwds):#, all=True):
+        self._x = []
+        self._y = []
+        self._id = []
+        self._info = []
+        self._t = []
+        self.__t0 = [self.__now()]
+        self._timed = kwds.get('timed', False)
+       #self._all = all
+
+    def __len__(self):
+        return len(self.x)
+
+    def info(self, message):
+        self._info.append("%s" % "".join(["",str(message)]))
+        return
+
+    def timer(self, on='?'): #XXX: str-driven args... a hallmark of bad design
+        if on == '?':
+            return self._timed
+        if on in [False, None, 'off']: #turn timing off
+            self._timed = False
+        elif isinstance(on, bool) or on == 'on': #turn timing on
+            self._timed = True
+        elif on == 'toggle': #toggle timing
+            self._timed = not self._timed
+        elif on in ['reset', 'restart']: #'restart' the StartTime
+            self.__t0.append(self.__now())
+        return
+
+    def __now(self):
+        from time import time as now
+        return now()  
+
+    def __deadtime(self):
+        from numpy import NaN
+        return [NaN] * (len(self._y) - len(self._t))
+
+    def __call__(self, x, y, id=None, **kwds):#, best=0):
+        if self._timed:
+            self._t.extend(self.__deadtime())
+            self._t.append(self.__now())
+        from mystic.tools import listify
+        self._x.append(listify(x)) #XXX: better to save as-is?
+        self._y.append(listify(y)) #XXX: better to save as-is?
+        self._id.append(id)
+       #if not self._all and list_or_tuple_or_ndarray(x):
+       #    self._x[-1] = self._x[-1][best]
+       #if not self._all and list_or_tuple_or_ndarray(y):
+       #    self._y[-1] = self._y[-1][best]
+
+    def get_x(self):
+        return self._x
+
+    def get_y(self):
+        return self._y
+
+    def get_id(self):
+        return self._id
+
+    def get_info(self):
+        return self._info
+
+    def get_t(self):
+        from numpy import array
+        return list(array(self._t + self.__deadtime()) - self._t0)
+
+    def _get_t0(self):
+        return self.__t0[-1]
+
+    def _set_t0(self, t): #XXX: allow setting t0 ?
+        self.__t0.append(t)
+        return
+
+    x = property(get_x, doc = "Params")
+    y = property(get_y, doc = "Costs")
+    id = property(get_id, doc = "Id")
+    t = property(get_t, doc = "Time")
+    _t0 = property(_get_t0, _set_t0, doc = "StartTime")
+    pass
+