add samples

wangrong

2025-03-19 291afb9c81aae1d15d9e3edcea618d0a2e1fc5a6

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#include <opencv2/imgproc.hpp> 
 
#include "vp_plate_osd_node.h" 
 
namespace vp_nodes { 
     
    vp_plate_osd_node::vp_plate_osd_node(std::string node_name, std::string font, bool display_his):  
                                vp_node(node_name), display_his(display_his) { 
        if (!font.empty()) { 
            ft2 = cv::freetype::createFreeType2(); 
            ft2->loadFontData(font, 0);    
        }        
        this->initialized(); 
    } 
     
    vp_plate_osd_node::~vp_plate_osd_node() { 
        deinitialized(); 
    } 
 
    std::shared_ptr<vp_objects::vp_meta> vp_plate_osd_node::handle_frame_meta(std::shared_ptr<vp_objects::vp_frame_meta> meta) { 
        // operations on osd_frame 
        if (meta->osd_frame.empty()) { 
            meta->osd_frame = meta->frame.clone(); 
        } 
 
        auto& canvas = meta->osd_frame; 
        if (display_his) { 
            cv::rectangle(canvas, cv::Rect(0, canvas.rows - height_his, canvas.cols, height_his), cv::Scalar(240, 200, 220), cv::FILLED); 
        } 
         
        // scan targets 
        for (auto& i : meta->targets) { 
            // plate color and plate text 
            auto color_and_text = vp_utils::string_split(i->primary_label, '_'); 
            if(color_and_text.size() != 2) { 
                continue; 
            } 
            auto& color = color_and_text[0]; 
            auto& text = color_and_text[1]; 
            auto text_2_display = text_colors.at(color) + " " + text; 
            // tracked ? 
            if (i->track_id != -1) { 
                text_2_display = "#" + std::to_string(i->track_id) + " " + text_2_display; 
            } 
             
            cv::rectangle(canvas, cv::Rect(i->x, i->y, i->width, i->height), draw_colors.at(color), 2); 
            if (ft2 != nullptr) { 
                ft2->putText(canvas, text_2_display, cv::Point(i->x, i->y), 20, draw_colors.at(color), cv::FILLED, cv::LINE_AA, true); 
            } 
            else {                
                // ignore 
            }    
 
            if (display_his) { 
                // put it to cache 
                auto plate = meta->frame(cv::Rect(i->x, i->y, i->width, i->height)); 
                cv::Mat resized_plate; 
                cv::resize(plate, resized_plate, cv::Size(height_his, int((float(height_his) / plate.cols) * plate.rows)));     
                plates_his.push_back(resized_plate);   
            }             
        } 
 
        // display history at the bottom of canvas 
        if (display_his) { 
            // remove previous history plates if need, since no space to draw 
            auto width_need = plates_his.size() * (height_his + gap_his) + gap_his; 
            while(width_need >= canvas.cols) { 
                plates_his.erase(plates_his.begin()); 
                width_need = plates_his.size() * (height_his + gap_his) + gap_his; 
            } 
 
            // draw history plates at the bottom of screen 
            for (int i = 0; i < plates_his.size(); i++) { 
                auto& p = plates_his[i]; 
 
                auto roi = canvas(cv::Rect((gap_his + height_his) * i + gap_his, canvas.rows - height_his / 2 - p.rows / 2 , height_his, p.rows)); 
                plates_his[i].copyTo(roi); 
            } 
        } 
 
        return meta; 
    } 
}