98 lines
2.8 KiB
Python
Executable File
98 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import sys
|
|
import cProfile # pylint: disable=import-error
|
|
import pprofile # pylint: disable=import-error
|
|
import pyprof2calltree # pylint: disable=import-error
|
|
|
|
from common.params import Params
|
|
from tools.lib.logreader import LogReader
|
|
from selfdrive.test.profiling.lib import SubMaster, PubMaster, SubSocket, ReplayDone
|
|
from selfdrive.test.process_replay.process_replay import CONFIGS
|
|
from selfdrive.car.toyota.values import CAR as TOYOTA
|
|
from selfdrive.car.honda.values import CAR as HONDA
|
|
from selfdrive.car.volkswagen.values import CAR as VW
|
|
|
|
BASE_URL = "https://commadataci.blob.core.windows.net/openpilotci/"
|
|
|
|
CARS = {
|
|
'toyota': ("0982d79ebb0de295|2021-01-03--20-03-36/6", TOYOTA.RAV4),
|
|
'honda': ("0982d79ebb0de295|2021-01-08--10-13-10/6", HONDA.CIVIC),
|
|
"vw": ("ef895f46af5fd73f|2021-05-22--14-06-35/6", VW.AUDI_A3_MK3),
|
|
}
|
|
|
|
|
|
def get_inputs(msgs, process, fingerprint):
|
|
for config in CONFIGS:
|
|
if config.proc_name == process:
|
|
sub_socks = list(config.pub_sub.keys())
|
|
trigger = sub_socks[0]
|
|
break
|
|
|
|
# some procs block on CarParams
|
|
for msg in msgs:
|
|
if msg.which() == 'carParams':
|
|
m = msg.as_builder()
|
|
m.carParams.carFingerprint = fingerprint
|
|
Params().put("CarParams", m.carParams.copy().to_bytes())
|
|
break
|
|
|
|
sm = SubMaster(msgs, trigger, sub_socks)
|
|
pm = PubMaster()
|
|
if 'can' in sub_socks:
|
|
can_sock = SubSocket(msgs, 'can')
|
|
else:
|
|
can_sock = None
|
|
return sm, pm, can_sock
|
|
|
|
|
|
def profile(proc, func, car='toyota'):
|
|
segment, fingerprint = CARS[car]
|
|
segment = segment.replace('|', '/')
|
|
rlog_url = f"{BASE_URL}{segment}/rlog.bz2"
|
|
msgs = list(LogReader(rlog_url)) * int(os.getenv("LOOP", "1"))
|
|
|
|
os.environ['FINGERPRINT'] = fingerprint
|
|
|
|
def run(sm, pm, can_sock):
|
|
try:
|
|
if can_sock is not None:
|
|
func(sm, pm, can_sock)
|
|
else:
|
|
func(sm, pm)
|
|
except ReplayDone:
|
|
pass
|
|
|
|
# Statistical
|
|
sm, pm, can_sock = get_inputs(msgs, proc, fingerprint)
|
|
with pprofile.StatisticalProfile()(period=0.00001) as pr:
|
|
run(sm, pm, can_sock)
|
|
pr.dump_stats(f'cachegrind.out.{proc}_statistical')
|
|
|
|
# Deterministic
|
|
sm, pm, can_sock = get_inputs(msgs, proc, fingerprint)
|
|
with cProfile.Profile() as pr:
|
|
run(sm, pm, can_sock)
|
|
pyprof2calltree.convert(pr.getstats(), f'cachegrind.out.{proc}_deterministic')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from selfdrive.controls.controlsd import main as controlsd_thread
|
|
from selfdrive.controls.radard import radard_thread
|
|
from selfdrive.locationd.paramsd import main as paramsd_thread
|
|
from selfdrive.controls.plannerd import main as plannerd_thread
|
|
|
|
procs = {
|
|
'radard': radard_thread,
|
|
'controlsd': controlsd_thread,
|
|
'paramsd': paramsd_thread,
|
|
'plannerd': plannerd_thread,
|
|
}
|
|
|
|
proc = sys.argv[1]
|
|
if proc not in procs:
|
|
print(f"{proc} not available")
|
|
sys.exit(0)
|
|
else:
|
|
profile(proc, procs[proc])
|