timing.py

'''
Author: Kuan-Hsun Chen
Referring to the existing solution on stackoverflow.com for time mesurement.
http://stackoverflow.com/questions/1557571/how-to-get-time-of-a-python-program-execution
'''

import atexit
from time import clock

tstart = clock()
tend = tstart

def secondsToStr(t):
    return "%d:%02d:%02d.%03d" % \
        reduce(lambda ll,b : divmod(ll[0],b) + ll[1:],
            [(t*1000,),1000,60,60])

line = "="*40
def log(s, elapsed=None):
    print line
    print secondsToStr(clock()), '-', s
    if elapsed:
        print "Elapsed time:", elapsed
    print line
    print

def tlog_start(s, mode):
    global tstart
    tstart = clock()
    if(mode):
        print secondsToStr(clock()), '-', s

def tlog_end(s, stamp, mode):
    global tend
    global tstart
    tend = clock()
    if(mode):
        print secondsToStr(clock()), '-', s
    telapsed= tend - tstart
    if(mode):
        print secondsToStr(telapsed), '-', "elapsed"
    if telapsed < 60*10:
        stamp.append(telapsed)

def endlog():
    end = clock()
    elapsed = end-start
    log("End Program", secondsToStr(elapsed))

def now():
    return secondsToStr(clock())

start = clock()
atexit.register(endlog)
log("Start Program")