My tools of the trade for python programming.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

77 lines
1.7 KiB

# $Id: timer.py,v 1.2 2010-09-27 19:54:26 wirawan Exp $
#
# timer.py
# Simple timer and possibly other timing-related routine
#
# Wirawan Purwanto
# Created: 20081022
#
# 20081022: Created as pyqmc.utils.timer .
# 20100927: Moved to wpylib.timer .
#
"""
Simple timer utility.
This module is part of wpylib project.
"""
import time
class timer:
'''A small timer class.'''
def start(self):
self.tm1 = time.clock()
def stop(self):
self.tm2 = time.clock()
return (self.tm2 - self.tm1)
def length(self):
return self.tm2 - self.tm1
class block_timer(object):
"""Timer for a code block in python.
For use with python 2.5+ 'with' statement.
See: http://preshing.com/20110924/timing-your-code-using-pythons-with-statement
To use this:
with block_timer() as t:
<code>
"""
@staticmethod
def writeout_file(out):
return lambda tm: out.write("Total execution time = %s secs\n" % (tm,))
@staticmethod
def writeout_dict(rec, key):
def wrt_dict(tm):
rec[key] = tm
return wrt_dict
#return lambda tm: rec.__setitem__(key, tm)
@staticmethod
def bt_file(fobj):
if isinstance(fobj, basestring):
from wpylib.iofmt.text_output import text_output
out = text_output(fobj)
else:
out = fobj
return block_timer(report=block_timer.writeout_file(out))
@staticmethod
def bt_dict(rec, key):
return block_timer(report=block_timer.writeout_dict(rec, key))
def __init__(self, report=None):
if report == None: report = block_timer.writeout_file(sys.stdout)
self.report = report
def __enter__(self):
self.start = time.clock()
return self
def __exit__(self, *args):
self.end = time.clock()
self.interval = self.end - self.start
self.report(self.interval)