public/py-tools
1
0
Fork 0
Code Issues Pull Requests Actions Releases 8 Activity

better stopwatch

Browse Source
This commit is contained in:
Stefan Harmuth 2022-11-29 19:36:55 +01:00
parent 4f45d1f32d
commit d92686dd28
1 changed files with 19 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
tools/stopwatch.py
View File

@ -2,16 +2,23 @@ from time import perf_counter_ns
from .types import IntOrNone from .types import IntOrNone
def ns_to_string(ns: int) -> str: def get_time_string_from_ns(ns: int) -> str:
# mis, ns = ns // 1000, ns % 1000
# ms, mis = mis // 1000, mis % 1000
# s, ms = ms // 1000, ms % 1000
# m, s = s // 60, s % 60
# h, m = m // 60, m % 60
# d, h = h // 24, h % 24
units = ['ns', 'µs', 'ms', 's'] units = ['ns', 'µs', 'ms', 's']
unit = 0 unit = 0
while ns > 1_000: while ns > 1_000:
ns /= 1_000 if unit > 3:
unit += 1
if unit == len(units) - 1:
break break
ns /= 1000
unit += 1
return f"{ns:1.2f}{units[unit]}" return "%1.2f%s" % (ns, units[unit])
class StopWatch: class StopWatch:
@ -26,26 +33,26 @@ class StopWatch:
self.started = perf_counter_ns() self.started = perf_counter_ns()
self.stopped = None self.stopped = None
def stop(self) -> int: def stop(self) -> float:
self.stopped = perf_counter_ns() self.stopped = perf_counter_ns()
return self.elapsed() return self.elapsed()
reset = start reset = start
def elapsed(self) -> int: def elapsed(self) -> float:
if self.stopped is None: if self.stopped is None:
return perf_counter_ns() - self.started return perf_counter_ns() - self.started
else: else:
return self.stopped - self.started return self.stopped - self.started
def avg_elapsed(self, divider: int) -> int: def elapsed_string(self) -> str:
return self.elapsed() // divider # in ns precision loosing some rounding error probably will not hurt return get_time_string_from_ns(self.elapsed())
def elapsed_string(self): def avg_elapsed(self, divider: int) -> float:
return ns_to_string(self.elapsed()) return self.elapsed() / divider
def avg_string(self, divider: int) -> str: def avg_string(self, divider: int) -> str:
return ns_to_string(self.avg_elapsed(divider)) return get_time_string_from_ns(int(self.avg_elapsed(divider)))
def __str__(self): def __str__(self):
return self.avg_string(1) return self.avg_string(1)