diff --git a/selfdrive/test/process_replay/migration.py b/selfdrive/test/process_replay/migration.py index afcf705ff9..3c77dc8835 100644 --- a/selfdrive/test/process_replay/migration.py +++ b/selfdrive/test/process_replay/migration.py @@ -13,6 +13,7 @@ def migrate_all(lr, old_logtime=False, manager_states=False, panda_states=False, msgs = migrate_carParams(msgs, old_logtime) msgs = migrate_gpsLocation(msgs) msgs = migrate_deviceState(msgs) + msgs = migrate_controlsState(msgs) if manager_states: msgs = migrate_managerState(msgs) if panda_states: @@ -68,6 +69,22 @@ def migrate_deviceState(lr): return all_msgs +def migrate_controlsState(lr): + all_msgs = [] + personality = None + for msg in lr: + if msg.which() == 'longitudinalPlan': + personality = msg.longitudinalPlan.personalityDEPRECATED + # Detect old logs with personality in longitidunalPlan + if msg.which() == 'controlsState' and msg.controlsState.personality == 0 and personality is not None: + n = msg.as_builder() + n.controlsState.personality = personality + all_msgs.append(n.as_reader()) + else: + all_msgs.append(msg) + return all_msgs + + def migrate_pandaStates(lr): all_msgs = [] # TODO: safety param migration should be handled automatically