first commit

nodes/vp_rtsp_des_node.h

@@ -0,0 +1,61 @@
+#pragma once
+
+#include <gst/gst.h>
+#include <gst/rtsp-server/rtsp-server.h>
+#include "vp_des_node.h"
+
+/*
+* ##### Compile Tips #####
+* install additional packages for this node before compiling:
+* sudo apt-get install libgstrtspserver-1.0-dev gstreamer1.0-rtsp
+* https://github.com/GStreamer/gst-rtsp-server
+*/
+
+namespace vp_nodes {
+    // rtsp des node, push video stream via rtsp.
+    // example: rtsp://localhost:7890/test
+    // note, no specialized rtsp server needed since this node itself is a rtsp server, which can be pulled directly by video players such as VLC.
+    class vp_rtsp_des_node: public vp_des_node {
+    private:
+        /* data */
+        std::string gst_template = "appsrc ! videoconvert ! %s bitrate=%d ! h264parse ! rtph264pay ! udpsink host=localhost port=%d";
+        cv::VideoWriter rtsp_writer;
+
+        // start rtsp server
+        void start_rtsp_streaming ();
+        
+        // resolution for rtsp stream
+        vp_objects::vp_size resolution_w_h;
+        int bitrate;
+        // for osd frame 
+        bool osd;
+
+        int udp_buffer_size = 0;
+        // base udp port for stream data exchange internally, base_udp_port + channel_index would be used for each channel, MUST not occupied by others.
+        const int base_udp_port = 7890;
+
+        int rtsp_port = 9000;
+        std::string rtsp_name = "";
+
+        /* gst-rtsp-server variables shared between instances for different channels */
+        static GstRTSPServer* rtsp_server;
+
+        // set x264enc as the default encoder, we can use hardware encoder instead.
+        std::string gst_encoder_name = "x264enc";
+    protected:
+        // re-implementation, return nullptr.
+        virtual std::shared_ptr<vp_objects::vp_meta> handle_frame_meta(std::shared_ptr<vp_objects::vp_frame_meta> meta) override; 
+    public:
+        vp_rtsp_des_node(std::string node_name, 
+                        int channel_index, 
+                        int rtsp_port = 9000, 
+                        std::string rtsp_name = "", 
+                        vp_objects::vp_size resolution_w_h = {}, 
+                        int bitrate = 512,
+                        bool osd = true,
+                        std::string gst_encoder_name = "x264enc");
+        ~vp_rtsp_des_node();
+
+         virtual std::string to_string() override;
+    };
+}