parent
fc233adda7
commit
3538e2060a
|
@ -1,12 +1,44 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
import sys
|
||||
import multiprocessing
|
||||
import subprocess
|
||||
from tempfile import NamedTemporaryFile
|
||||
from tools.lib.route import Route
|
||||
from tools.lib.logreader import LogReader
|
||||
from selfdrive.test.process_replay.compare_logs import save_log
|
||||
from tools.lib.url_file import URLFile
|
||||
|
||||
|
||||
|
||||
def load_segment(segment_name):
|
||||
print(f"Loading {segment_name}")
|
||||
lr = LogReader(segment_name)
|
||||
r = [d for d in lr if d.which() not in ['can', 'sendcan']]
|
||||
print(f"done {segment_name}")
|
||||
return r
|
||||
|
||||
def juggle_file(fn):
|
||||
subprocess.call(f"bin/plotjuggler -d {fn}", shell=True)
|
||||
# subprocess.call(f"/home/batman/PlotJuggler/build/bin/plotjuggler -d {fn}", shell=True)
|
||||
|
||||
def juggle_route(route_name):
|
||||
r = Route(route_name)
|
||||
all_data = []
|
||||
|
||||
pool = multiprocessing.Pool(24)
|
||||
|
||||
all_data = []
|
||||
for d in pool.map(load_segment, r.log_paths()):
|
||||
all_data += d
|
||||
|
||||
tempfile = NamedTemporaryFile(suffix='.rlog')
|
||||
save_log(tempfile.name, all_data, compress=False)
|
||||
del all_data
|
||||
|
||||
juggle_file(tempfile.name)
|
||||
|
||||
def juggle_segment(route_name, segment_nr):
|
||||
|
||||
r = Route(route_name)
|
||||
lp = r.log_paths()[segment_nr]
|
||||
|
||||
|
@ -15,9 +47,11 @@ def juggle_segment(route_name, segment_nr):
|
|||
return
|
||||
|
||||
uf = URLFile(lp)
|
||||
|
||||
subprocess.call(f"bin/plotjuggler -d {uf.name}", shell=True)
|
||||
juggle_file(uf.name)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
juggle_segment(sys.argv[1], int(sys.argv[2]))
|
||||
if len(sys.argv) == 2:
|
||||
juggle_route(sys.argv[1])
|
||||
elif len(sys.argv) == 3:
|
||||
juggle_segment(sys.argv[1], int(sys.argv[2]))
|
||||
|
|
Loading…
Reference in New Issue