f98ee40f4b
This is an ongoing series of commits that are reformatting our Python code. This catches the last of the python files to reformat. Since they where so few I bunched them together. Reformatting is done with `black`. If you end up having problems merging this commit because you have made changes to a python file, the best way to handle that is to run git checkout --ours <yourfile> and then reformat it with black. If you run into any problems, post to discourse about it and we will try to help. RFC Thread below: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Reviewed By: jhenderson, #libc, Mordante, sivachandra Differential Revision: https://reviews.llvm.org/D150784
52 lines
1.4 KiB
Python
52 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()
|