# Profiling: Times operations and provides the user with statistical information

import sys

import datetime

import util

def now(): return datetime.datetime.now()

class Profiler:

    def __init__(self, start_now=False):

        self.total = datetime.timedelta()

        if start_now: self.start()

    def start(self): self.start = now()

    def stop(self):

        assert self.start != None

        self.total += now() - self.start

    def msg(self): return 'Took '+str(self.total)+' sec'

class ItersProfiler(Profiler):

    def __init__(self, iter_text='iteration', **settings):

        Profiler.__init__(self, **settings)

        self.iter_text = iter_text

        self.iter_ct = None

    def add_iters(self, iter_ct):

        if self.iter_ct == None: self.iter_ct = 0

        self.iter_ct += iter_ct

    def stop(self, iter_ct=None):

        Profiler.stop(self)

        if iter_ct != None: self.add_iters(iter_ct)

    def msg(self):

        msg = Profiler.msg(self)

        if self.iter_ct != None:

            msg += ('/'+str(self.iter_ct)+' '+self.iter_text+'(s)'+' = '

            +util.to_si(float(self.total.total_seconds())/self.iter_ct)

            +'s/'+self.iter_text)

        return msg