nbv_sim/src/active_grasp/timer.py

import time


class Timer:
    timers = dict()

    def __init__(self, name):
        self.name = name
        self.timers.setdefault(name, 0)

    def __enter__(self):
        self.start()
        return self

    def __exit__(self, *exc_info):
        self.stop()

    @classmethod
    def reset(cls):
        cls.timers = dict()

    def start(self):
        self.tic = time.perf_counter()

    def stop(self):
        elapsed_time = time.perf_counter() - self.tic
        self.timers[self.name] += elapsed_time
        # with open(f"{self.name}.txt", "a") as f:
        #     f.write(f"{elapsed_time}\n")