mirror of
https://github.com/vladmandic/automatic
synced 2026-08-25 22:20:46 +02:00
Vladimir Mandic
99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
import os
|
|
import time
|
|
import sys
|
|
|
|
|
|
try:
|
|
default_min_time = float(os.environ.get('SD_MIN_TIMER', '0.1'))
|
|
except Exception:
|
|
default_min_time = 0.1
|
|
|
|
|
|
class Timer:
|
|
def __init__(self, profile=False):
|
|
self.start = time.time()
|
|
self.records = {}
|
|
self.total = 0
|
|
self.profile = profile
|
|
|
|
def elapsed(self, reset=True):
|
|
end = time.time()
|
|
res = end - self.start
|
|
if reset:
|
|
self.start = end
|
|
return res
|
|
|
|
def add(self, name, t):
|
|
if name not in self.records:
|
|
self.records[name] = 0
|
|
self.records[name] += t
|
|
|
|
def rm(self, name):
|
|
if name in self.records:
|
|
del self.records[name]
|
|
|
|
def get(self, name):
|
|
return self.records.get(name, 0)
|
|
|
|
def ts(self, name, t):
|
|
elapsed = time.time() - t
|
|
self.add(name, elapsed)
|
|
|
|
def record(self, category=None, extra_time=0, reset=True):
|
|
e = self.elapsed(reset)
|
|
if category is None:
|
|
category = sys._getframe(1).f_code.co_name # pylint: disable=protected-access
|
|
if category not in self.records:
|
|
self.records[category] = 0
|
|
self.records[category] += e + extra_time
|
|
self.total += e + extra_time
|
|
|
|
def summary(self, min_time=default_min_time, max_time=-1, total=True):
|
|
if self.profile:
|
|
min_time = -1
|
|
self.total = sum(self.records.values())
|
|
res = f"total={self.total:.2f} " if total else ''
|
|
additions = [x for x in self.records.items() if x[1] >= min_time]
|
|
if max_time > 0:
|
|
additions = [x for x in additions if x[1] <= max_time]
|
|
if not total:
|
|
additions = [x for x in additions if x[0] != 'total']
|
|
additions = sorted(additions, key=lambda x: x[1], reverse=True)
|
|
if not additions:
|
|
return res
|
|
res += " ".join([f"{category}={time_taken:.2f}" for category, time_taken in additions])
|
|
return res
|
|
|
|
def max(self):
|
|
if not self.records:
|
|
return 0
|
|
return max(self.records.values())
|
|
|
|
def get_total(self):
|
|
return sum(self.records.values())
|
|
|
|
def dct(self, min_time=default_min_time, no_total=False):
|
|
self.total = sum(self.records.values())
|
|
if no_total:
|
|
self.records.pop('total', None)
|
|
else:
|
|
self.records['total'] = self.total
|
|
if self.profile:
|
|
res = {k: round(v, 4) for k, v in self.records.items()}
|
|
else:
|
|
res = {k: round(v, 2) for k, v in self.records.items() if v >= min_time}
|
|
res = {k: v for k, v in sorted(res.items(), key=lambda x: x[1], reverse=True)} # noqa: C416 # pylint: disable=unnecessary-comprehension
|
|
return res
|
|
|
|
def reset(self):
|
|
self.records.clear()
|
|
self.__init__(self.profile)
|
|
|
|
startup = Timer()
|
|
process = Timer()
|
|
launch = Timer()
|
|
init = Timer()
|
|
load = Timer()
|
|
dynamo = Timer()
|
|
autotune = Timer(profile=True)
|