#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";
    };
    
}