initial commit

wangrong

2025-01-03 2c4b8838f89bebca97615c5f0bac2c58e0349771

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
 
 
#include "vp_image_des_node.h" 
 
 
namespace vp_nodes { 
         
    vp_image_des_node::vp_image_des_node(std::string node_name,  
                                    int channel_index,   
                                    std::string location,  
                                    int interval, 
                                    vp_objects::vp_size resolution_w_h, 
                                    bool osd, 
                                    std::string gst_encoder_name): 
                                    vp_des_node(node_name, channel_index), 
                                    location(location), 
                                    interval(interval), 
                                    resolution_w_h(resolution_w_h), 
                                    osd(osd), 
                                    gst_encoder_name(gst_encoder_name) { 
        // make sure not greater than 1 minute (too long) and not lower than 1 second (since it's too quick, use video stream instead directly) 
        assert(interval >= 1 && interval <= 60); 
        if (vp_utils::ends_with(location, ".jpeg") || vp_utils::ends_with(location, ".jpg")) { 
            // save to file 
            gst_template_file = vp_utils::string_format(gst_template_file, interval, gst_encoder_name.c_str(), location.c_str()); 
            to_file = true; 
        } 
        else if (location.find(":") != std::string::npos) { 
            // push to remote 
            auto parts = vp_utils::string_split(location, ':'); 
            assert(parts.size() == 2); 
            auto host = parts[0];  // ip 
            auto port = std::stoi(parts[1]);  // try to get port 
 
            gst_template_udp = vp_utils::string_format(gst_template_udp, interval, gst_encoder_name.c_str(), host.c_str(), port); 
             
            to_file = false; 
        } 
        else { 
            // error 
            throw "invalid input parameter for `location`!"; 
        } 
 
        auto s = to_file ? gst_template_file : gst_template_udp; 
        VP_INFO(vp_utils::string_format("[%s] [%s]", node_name.c_str(), s.c_str())); 
        this->initialized(); 
    } 
     
    vp_image_des_node::~vp_image_des_node() { 
        deinitialized(); 
    } 
 
    std::shared_ptr<vp_objects::vp_meta> vp_image_des_node::handle_frame_meta(std::shared_ptr<vp_objects::vp_frame_meta> meta) { 
        VP_DEBUG(vp_utils::string_format("[%s] received frame meta, channel_index=>%d, frame_index=>%d", node_name.c_str(), meta->channel_index, meta->frame_index)); 
         
        cv::Mat resize_frame; 
        if (this->resolution_w_h.width != 0 && this->resolution_w_h.height != 0) {                  
            cv::resize((osd && !meta->osd_frame.empty()) ? meta->osd_frame : meta->frame, resize_frame, cv::Size(resolution_w_h.width, resolution_w_h.height)); 
        } 
        else { 
            resize_frame = (osd && !meta->osd_frame.empty()) ? meta->osd_frame : meta->frame; 
        } 
 
        if (!image_writer.isOpened()) { 
            if (to_file) { 
                assert(image_writer.open(this->gst_template_file, cv::CAP_GSTREAMER, 0, meta->fps, {resize_frame.cols, resize_frame.rows})); 
            } 
            else { 
                assert(image_writer.open(this->gst_template_udp, cv::CAP_GSTREAMER, 0, meta->fps, {resize_frame.cols, resize_frame.rows})); 
            } 
        } 
 
        image_writer.write(resize_frame); 
 
        // for general works defined in base class 
        return vp_des_node::handle_frame_meta(meta); 
    } 
 
    std::string vp_image_des_node::to_string() { 
        return to_file ? location : "udp://" + location + "/jpg"; 
    } 
}