lgtm

commaai · Dec 22, 2021 · 81eed68 · 81eed68
1 parent f9439e2
commit 81eed68
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 13 deletions.
diff --git a/selfdrive/ui/replay/main.cc b/selfdrive/ui/replay/main.cc
@@ -59,6 +59,10 @@ void keyboardThread(Replay *replay_) {
         qDebug() << "invalid argument";
       }
       getch();  // remove \n from entering seek
+    } else if (c == 'e') {
+      replay_->seekToFlag(FindFlag::nextEngagement);
+    } else if (c == 'd') {
+      replay_->seekToFlag(FindFlag::nextDisEngagement);
     } else if (c == 'm') {
       replay_->seekTo(+60, true);
     } else if (c == 'M') {
@@ -69,10 +73,6 @@ void keyboardThread(Replay *replay_) {
       replay_->seekTo(-10, true);
     } else if (c == 'G') {
       replay_->seekTo(0, true);
-    } else if (c == 'e') {
-      replay_->seekToFlag(FindFlag::nextEngagement);
-    } else if (c == 'd') {
-      replay_->seekToFlag(FindFlag::nextDisEngagement);
     } else if (c == ' ') {
       replay_->pause(!replay_->isPaused());
     }

diff --git a/selfdrive/ui/replay/replay.cc b/selfdrive/ui/replay/replay.cc
@@ -152,15 +152,12 @@ std::optional<uint64_t> Replay::find(FindFlag flag) {
     if (!log.load(route_->at(n).qlog.toStdString(), nullptr, cache_to_local, 0, 3)) continue;
 
     for (const Event *e : log.events) {
-      if (e->mono_time > cur_mono_time_) {
-        if (flag == FindFlag::nextEngagement) {
-          if (e->which == cereal::Event::Which::CONTROLS_STATE && e->event.getControlsState().getEnabled()) {
-            return e->mono_time;
-          }
-        } else if (flag == FindFlag::nextDisEngagement) {
-          if (e->which == cereal::Event::Which::CONTROLS_STATE && !e->event.getControlsState().getEnabled()) {
-            return e->mono_time;
-          }
+      if (e->mono_time > cur_mono_time_ && e->which == cereal::Event::Which::CONTROLS_STATE) {
+        const auto cs = e->event.getControlsState();
+        if (flag == FindFlag::nextEngagement && cs.getEnabled()) {
+          return e->mono_time;
+        } else if (flag == FindFlag::nextDisEngagement && !cs.getEnabled()) {
+          return e->mono_time;
         }
       }
     }