mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-02-18 23:23:53 +08:00
59 lines
1.1 KiB
Python
Executable File
59 lines
1.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import io
|
|
import os
|
|
import time
|
|
import pstats
|
|
import cProfile
|
|
from contextlib import contextmanager
|
|
|
|
from panda import Panda, PandaDFU
|
|
from panda.tests.hitl.helpers import get_random_can_messages
|
|
|
|
|
|
PROFILE = "PROFILE" in os.environ
|
|
|
|
@contextmanager
|
|
def print_time(desc):
|
|
if PROFILE:
|
|
pr = cProfile.Profile()
|
|
pr.enable()
|
|
start = time.perf_counter()
|
|
yield
|
|
end = time.perf_counter()
|
|
print(f"{end - start:.3f}s - {desc}")
|
|
if PROFILE:
|
|
pr.disable()
|
|
s = io.StringIO()
|
|
ps = pstats.Stats(pr, stream=s).sort_stats("cumtime")
|
|
ps.print_stats()
|
|
print(s.getvalue())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
with print_time("Panda()"):
|
|
p = Panda()
|
|
|
|
with print_time("PandaDFU.list()"):
|
|
PandaDFU.list()
|
|
|
|
fxn = [
|
|
'reset',
|
|
'reconnect',
|
|
'up_to_date',
|
|
'health',
|
|
#'flash',
|
|
]
|
|
for f in fxn:
|
|
with print_time(f"Panda.{f}()"):
|
|
getattr(p, f)()
|
|
|
|
p.set_can_loopback(True)
|
|
|
|
for n in range(6):
|
|
msgs = get_random_can_messages(int(10**n))
|
|
with print_time(f"Panda.can_send_many() - {len(msgs)} msgs"):
|
|
p.can_send_many(msgs)
|
|
|
|
with print_time("Panda.can_recv()"):
|
|
m = p.can_recv()
|