import time class Timer: def __init__(self): self.times = {} # Dictionary to store times per stage def __call__(self, name): print(f"Timing {name}") return self.TimerContextManager(self, name) def print_stats(self): total_time = sum(self.times.values()) # Print table header print("{:<20} {:>10} {:>10}".format("Stage", "Time(s)", "Percent")) for name, t in self.times.items(): percent = (t / total_time) * 100 if total_time > 0 else 0 print("{:<20} {:>10.2f} {:>9.2f}%".format(name, t, percent)) class TimerContextManager: def __init__(self, outer, name): self.outer = outer # Reference to the Timer instance self.name = name self.start_time = None def __enter__(self): self.start_time = time.perf_counter() return self def __exit__(self, exc_type, exc_value, traceback): end_time = time.perf_counter() elapsed = end_time - self.start_time self.outer.times[self.name] = self.outer.times.get(self.name, 0) + elapsed