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
 
 
#include <string>
#include <opencv2/core/core.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/videoio.hpp>
#include "vp_src_node.h"
 
namespace vp_nodes {
    // udp source node, receive video stream via udp(rtp) protocal.
    // example:
    // udp://127.0.0.1:6000
    class vp_udp_src_node: public vp_src_node
    {
    private:
        std::string gst_template = "udpsrc port=%d ! application/x-rtp,mdeia=video ! rtph264depay ! h264parse ! %s ! videoconvert ! appsink";
        cv::VideoCapture udp_capture;
    protected:
        // re-implemetation
        virtual void handle_run() override;
    public:
        vp_udp_src_node(std::string node_name, 
                        int channel_index, 
                        int port, 
                        float resize_ratio = 1.0,
                        std::string gst_decoder_name = "avdec_h264",
                        int skip_interval = 0);
        ~vp_udp_src_node();
 
        virtual std::string to_string() override;
 
        // port to listen
        int port;
        // set avdec_h264 as the default decoder, we can use hardware decoder instead.
        std::string gst_decoder_name = "avdec_h264";
        // 0 means no skip
        int skip_interval = 0;
    };
}