From 22519b6cd9b2e211cc112738d507be4de966803d Mon Sep 17 00:00:00 2001
From: Rick Lan <ricklan@gmail.com>
Date: Wed, 12 Jun 2024 12:47:21 +0800
Subject: [PATCH] added chevron extension

---
 selfdrive/ui/qt/onroad/annotated_camera.cc | 5 +++++
 selfdrive/ui/qt/onroad/annotated_camera.h  | 1 +
 2 files changed, 6 insertions(+)

diff --git a/selfdrive/ui/qt/onroad/annotated_camera.cc b/selfdrive/ui/qt/onroad/annotated_camera.cc
index 47de272c3..754bb065d 100644
--- a/selfdrive/ui/qt/onroad/annotated_camera.cc
+++ b/selfdrive/ui/qt/onroad/annotated_camera.cc
@@ -28,6 +28,7 @@ AnnotatedCameraWidget::AnnotatedCameraWidget(VisionStreamType type, QWidget* par
   #ifdef DP
   knight_scanner = new KnightScanner;
   rainbow_path = new RainbowPath;
+  chevron_ext = new ChevronExt;
   #endif
 
 }
@@ -85,6 +86,7 @@ void AnnotatedCameraWidget::updateState(const UIState &s) {
 
   #ifdef DP
   rainbow_path->update_states(s);
+  chevron_ext->update_states(s, is_metric);
   #endif
 }
 
@@ -366,6 +368,9 @@ void AnnotatedCameraWidget::drawLead(QPainter &painter, const cereal::RadarState
   painter.setBrush(redColor(fillAlpha));
   painter.drawPolygon(chevron, std::size(chevron));
 
+  #ifdef DP
+  chevron_ext->paint(painter, d_rel, x, y);
+  #endif
   painter.restore();
 }
 
diff --git a/selfdrive/ui/qt/onroad/annotated_camera.h b/selfdrive/ui/qt/onroad/annotated_camera.h
index 86e95e5fe..e06c143b6 100644
--- a/selfdrive/ui/qt/onroad/annotated_camera.h
+++ b/selfdrive/ui/qt/onroad/annotated_camera.h
@@ -43,6 +43,7 @@ private:
   #ifdef DP
   KnightScanner *knight_scanner;
   RainbowPath *rainbow_path;
+  ChevronExt * chevron_ext;
   #endif
 
 protected: