import time class ScopeTimer: def __init__(self, name): self.name = name def __enter__(self): self.start = time.time() return self def __exit__(self, *args): self.end = time.time() self.interval = self.end - self.start print("{} {:.3E}".format(self.name, self.interval)) class Timer: def __init__(self): self.times = [] def tick(self): self.times.append(time.time()) def get_average_and_reset(self): if len(self.times) < 2: return -1 avg = (self.times[-1] - self.times[0]) / (len(self.times) - 1) self.times = [self.times[-1]] return avg def get_last_iteration(self): if len(self.times) < 2: return 0 return self.times[-1] - self.times[-2] class TickTock: def __init__(self): self.time_pairs = [] self.current_time = None def tick(self): self.current_time = time.time() def tock(self): assert self.current_time is not None, self.current_time self.time_pairs.append([self.current_time, time.time()]) self.current_time = None def get_average_and_reset(self): if len(self.time_pairs) == 0: return -1 deltas = [t2 - t1 for t1, t2 in self.time_pairs] avg = sum(deltas) / len(deltas) self.time_pairs = [] return avg def get_last_iteration(self): if len(self.time_pairs) == 0: return -1 return self.time_pairs[-1][1] - self.time_pairs[-1][0]