TheRock/compiler/amd-llvm/cross-project-tests/debuginfo-tests/dexter/dex/utils/Timer.py

# DExTer : Debugging Experience Tester
# ~~~~~~   ~         ~~         ~   ~~
#
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
"""RAII-style timer class to be used with a 'with' statement to get wall clock
time for the contained code.
"""

import sys
import time


def _indent(indent):
    return "| " * indent


class Timer(object):
    fn = sys.stdout.write
    display = False
    indent = 0

    def __init__(self, name=None):
        self.name = name
        self.start = self.now

    def __enter__(self):
        Timer.indent += 1
        if Timer.display and self.name:
            indent = _indent(Timer.indent - 1) + " _"
            Timer.fn("{}\n".format(_indent(Timer.indent - 1)))
            Timer.fn("{} start {}\n".format(indent, self.name))
        return self

    def __exit__(self, *args):
        if Timer.display and self.name:
            indent = _indent(Timer.indent - 1) + "|_"
            Timer.fn(
                "{} {} time taken: {:0.1f}s\n".format(indent, self.name, self.elapsed)
            )
            Timer.fn("{}\n".format(_indent(Timer.indent - 1)))
        Timer.indent -= 1

    @property
    def elapsed(self):
        return self.now - self.start

    @property
    def now(self):
        return time.time()