Source code for omni.isaac.lab.utils.timer

# Copyright (c) 2022-2024, The Isaac Lab Project Developers.
# All rights reserved.
#
# SPDX-License-Identifier: BSD-3-Clause

"""Sub-module for a timer class that can be used for performance measurements."""

from __future__ import annotations

import time
from contextlib import ContextDecorator
from typing import Any, ClassVar


[docs]class TimerError(Exception): """A custom exception used to report errors in use of :class:`Timer` class.""" pass
[docs]class Timer(ContextDecorator): """A timer for performance measurements. A class to keep track of time for performance measurement. It allows timing via context managers and decorators as well. It uses the `time.perf_counter` function to measure time. This function returns the number of seconds since the epoch as a float. It has the highest resolution available on the system. As a regular object: .. code-block:: python import time from omni.isaac.lab.utils.timer import Timer timer = Timer() timer.start() time.sleep(1) print(1 <= timer.time_elapsed <= 2) # Output: True time.sleep(1) timer.stop() print(2 <= stopwatch.total_run_time) # Output: True As a context manager: .. code-block:: python import time from omni.isaac.lab.utils.timer import Timer with Timer() as timer: time.sleep(1) print(1 <= timer.time_elapsed <= 2) # Output: True Reference: https://gist.github.com/sumeet/1123871 """ timing_info: ClassVar[dict[str, float]] = dict() """Dictionary for storing the elapsed time per timer instances globally. This dictionary logs the timer information. The keys are the names given to the timer class at its initialization. If no :attr:`name` is passed to the constructor, no time is recorded in the dictionary. """
[docs] def __init__(self, msg: str | None = None, name: str | None = None): """Initializes the timer. Args: msg: The message to display when using the timer class in a context manager. Defaults to None. name: The name to use for logging times in a global dictionary. Defaults to None. """ self._msg = msg self._name = name self._start_time = None self._stop_time = None self._elapsed_time = None
def __str__(self) -> str: """A string representation of the class object. Returns: A string containing the elapsed time. """ return f"{self.time_elapsed:0.6f} seconds" """ Properties """ @property def time_elapsed(self) -> float: """The number of seconds that have elapsed since this timer started timing. Note: This is used for checking how much time has elapsed while the timer is still running. """ return time.perf_counter() - self._start_time @property def total_run_time(self) -> float: """The number of seconds that elapsed from when the timer started to when it ended.""" return self._elapsed_time """ Operations """
[docs] def start(self): """Start timing.""" if self._start_time is not None: raise TimerError("Timer is running. Use .stop() to stop it") self._start_time = time.perf_counter()
[docs] def stop(self): """Stop timing.""" if self._start_time is None: raise TimerError("Timer is not running. Use .start() to start it") self._stop_time = time.perf_counter() self._elapsed_time = self._stop_time - self._start_time self._start_time = None if self._name: Timer.timing_info[self._name] = self._elapsed_time
""" Context managers """ def __enter__(self) -> Timer: """Start timing and return this `Timer` instance.""" self.start() return self def __exit__(self, *exc_info: Any): """Stop timing.""" self.stop() # print message if self._msg is not None: print(self._msg, f": {self._elapsed_time:0.6f} seconds") """ Static Methods """
[docs] @staticmethod def get_timer_info(name: str) -> float: """Retrieves the time logged in the global dictionary based on name. Args: name: Name of the the entry to be retrieved. Raises: TimerError: If name doesn't exist in the log. Returns: A float containing the time logged if the name exists. """ if name not in Timer.timing_info: raise TimerError(f"Timer {name} does not exist") return Timer.timing_info.get(name)