From a85b5557edca861974896e3d832a424af8fd33b7 Mon Sep 17 00:00:00 2001 From: Willem Melching Date: Tue, 13 Jul 2021 12:01:27 +0200 Subject: [PATCH] fix ui.py after longitudinal refactor --- tools/replay/ui.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tools/replay/ui.py b/tools/replay/ui.py index 611307847..d778e48c2 100755 --- a/tools/replay/ui.py +++ b/tools/replay/ui.py @@ -56,7 +56,7 @@ def ui_thread(addr, frame_address): frame = messaging.sub_sock('roadCameraState', addr=addr, conflate=True) sm = messaging.SubMaster(['carState', 'longitudinalPlan', 'carControl', 'radarState', 'liveCalibration', 'controlsState', - 'liveTracks', 'modelV2', 'liveMpc', 'liveParameters', 'lateralPlan', 'roadCameraState'], addr=addr) + 'liveTracks', 'modelV2', 'liveParameters', 'lateralPlan', 'roadCameraState'], addr=addr) img = np.zeros((480, 640, 3), dtype='uint8') imgff = None @@ -155,9 +155,11 @@ def ui_thread(addr, frame_address): plot_arr[-1, name_to_arr_idx['v_override']] = sm['carControl'].cruiseControl.speedOverride plot_arr[-1, name_to_arr_idx['v_cruise']] = sm['carState'].cruiseState.speed plot_arr[-1, name_to_arr_idx['a_ego']] = sm['carState'].aEgo - plot_arr[-1, name_to_arr_idx['a_target']] = sm['longitudinalPlan'].aTarget plot_arr[-1, name_to_arr_idx['accel_override']] = sm['carControl'].cruiseControl.accelOverride + if len(sm['longitudinalPlan'].accels): + plot_arr[-1, name_to_arr_idx['a_target']] = sm['longitudinalPlan'].accels[0] + if sm.rcv_frame['modelV2']: plot_model(sm['modelV2'], img, calibration, top_down)