984fad243d
This reverts commit cb935f3456.
Discussion in D68708 advises that green dragon is being briskly
refurbished, and it's good to have this patch up testing it.
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
# 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()
|