From 58cafc80608001ef3762f16eefa42fa2b484a5f0 Mon Sep 17 00:00:00 2001 From: Dean Lee Date: Wed, 12 Jan 2022 06:16:52 +0800 Subject: [PATCH] AlertManager: cleanup add_many (#23438) old-commit-hash: 678732d0ce93446f0468893c6faf6e1d1e20ee6e --- selfdrive/controls/lib/alertmanager.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/selfdrive/controls/lib/alertmanager.py b/selfdrive/controls/lib/alertmanager.py index bf93b5f47e..4694d74bff 100644 --- a/selfdrive/controls/lib/alertmanager.py +++ b/selfdrive/controls/lib/alertmanager.py @@ -40,12 +40,12 @@ class AlertManager: def add_many(self, frame: int, alerts: List[Alert]) -> None: for alert in alerts: - key = alert.alert_type - self.alerts[key].alert = alert - if not self.alerts[key].active(frame): - self.alerts[key].start_frame = frame - min_end_frame = self.alerts[key].start_frame + alert.duration - self.alerts[key].end_frame = max(frame + 1, min_end_frame) + entry = self.alerts[alert.alert_type] + entry.alert = alert + if not entry.active(frame): + entry.start_frame = frame + min_end_frame = entry.start_frame + alert.duration + entry.end_frame = max(frame + 1, min_end_frame) def process_alerts(self, frame: int, clear_event_type=None) -> Optional[Alert]: current_alert = AlertEntry()