| # 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() |