优化项目结构
This commit is contained in:
30
inc/Visualizer.h
Normal file
30
inc/Visualizer.h
Normal file
@@ -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
|
||||
Reference in New Issue
Block a user