mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-02-19 22:53:57 +08:00
d9bf9f0a40
* enable some more flake8 checks * some more quick ones * bump opendbc * e401 * e711 e712 * e115 e116 * e222 * e301 * remove that * e129 * e701 e702 * e125 e131 * e227 * e306 * e262 * W503 * e713 * e704 * e731 * bump opendbc * fix some e722
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import time
|
|
|
|
class Profiler():
|
|
def __init__(self, enabled=False):
|
|
self.enabled = enabled
|
|
self.cp = {}
|
|
self.cp_ignored = []
|
|
self.iter = 0
|
|
self.start_time = time.time()
|
|
self.last_time = self.start_time
|
|
self.tot = 0.
|
|
|
|
def reset(self, enabled=False):
|
|
self.enabled = enabled
|
|
self.cp = {}
|
|
self.cp_ignored = []
|
|
self.iter = 0
|
|
self.start_time = time.time()
|
|
self.last_time = self.start_time
|
|
|
|
def checkpoint(self, name, ignore=False):
|
|
# ignore flag needed when benchmarking threads with ratekeeper
|
|
if not self.enabled:
|
|
return
|
|
tt = time.time()
|
|
if name not in self.cp:
|
|
self.cp[name] = 0.
|
|
if ignore:
|
|
self.cp_ignored.append(name)
|
|
self.cp[name] += tt - self.last_time
|
|
if not ignore:
|
|
self.tot += tt - self.last_time
|
|
self.last_time = tt
|
|
|
|
def display(self):
|
|
if not self.enabled:
|
|
return
|
|
self.iter += 1
|
|
print("******* Profiling *******")
|
|
for n, ms in sorted(self.cp.items(), key=lambda x: -x[1]):
|
|
if n in self.cp_ignored:
|
|
print("%30s: %9.2f percent: %3.0f IGNORED" % (n, ms*1000.0, ms/self.tot*100))
|
|
else:
|
|
print("%30s: %9.2f percent: %3.0f" % (n, ms*1000.0, ms/self.tot*100))
|
|
print("Iter clock: %2.6f TOTAL: %2.2f" % (self.tot/self.iter, self.tot))
|