内容填充

Visualizer.h

@@ -0,0 +1,30 @@
+#ifndef VISUALIZER_H
+#define VISUALIZER_H
+
+#include <opencv2/opencv.hpp>
+#include <vector>
+#include <string>
+#include "ArmorDetector.h"
+
+class Visualizer {
+public:
+    Visualizer();
+    ~Visualizer();
+    
+    // Draw light bars on frame
+    cv::Mat& draw_light_bars(cv::Mat& frame, const std::vector<LightBar>& light_bars, const std::string& target_color);
+    
+    // Draw armor plate on frame
+    cv::Mat& draw_armor_plate(cv::Mat& frame, const ArmorPlate& armor_plate);
+    
+    // Draw offset text
+    cv::Mat& draw_offset_text(cv::Mat& frame, const cv::Point2f* display_center, const std::string& target_color, bool is_predicted = false);
+    
+    // Draw crosshair at center
+    cv::Mat& draw_crosshair(cv::Mat& frame, const cv::Point2f& center, const cv::Scalar& color = cv::Scalar(0, 255, 0), int size = 10);
+    
+    // Draw ballistic point
+    cv::Mat& draw_ballistic_point(cv::Mat& frame, const cv::Point2f* ballistic_point);
+};
+
+#endif // VISUALIZER_H