first commit

nodes/vp_image_src_node.h

@@ -0,0 +1,46 @@
+
+#pragma once
+
+#include "vp_src_node.h"
+
+namespace vp_nodes {
+    // image src node, read image from local files or receive image from remote via udp.
+    class vp_image_src_node: public vp_src_node
+    {
+    private:
+        // gstreamer template for reading image from file (jpeg encoding only, filename MUST end with 'jpg/jpeg')
+        std::string gst_template_file = "multifilesrc location=%s loop=%s ! jpegparse ! %s ! videorate ! video/x-raw,framerate=1/%d ! videoconvert ! appsink";
+        // gstreamer template for receiving image from remote via udp （jpeg encoding only)
+        std::string gst_template_udp = "udpsrc port=%d ! application/x-rtp,encoding-name=jpeg ! rtpjpegdepay ! jpegparse ! %s ! videorate ! video/x-raw,framerate=1/%d ! videoconvert ! appsink";
+
+        cv::VideoCapture image_capture;
+
+        // `port number` for udp mode, `directory path` for file mode. auto-detect which mode should apply according to this value.
+        std::string port_or_location = "./%d.jpg";
+        // frequency to read/receive image
+        int interval = 1;
+
+        // restart reading if images read completely (only used for file mode)
+        bool cycle = true;
+
+        bool from_file = true;
+    protected:
+        // re-implemetation
+        virtual void handle_run() override;
+
+    public:
+        vp_image_src_node(std::string node_name, 
+                        int channel_index, 
+                        std::string port_or_location,
+                        int interval = 1,
+                        float resize_ratio = 1.0, 
+                        bool cycle = true,
+                        std::string gst_decoder_name = "jpegdec");
+        ~vp_image_src_node();
+        virtual std::string to_string() override;
+
+        // set jpegdec as the default decoder, we can use hardware decoder instead.
+        std::string gst_decoder_name = "jpegdec";
+    };
+    
+}