46 lines
1.8 KiB
C++
46 lines
1.8 KiB
C++
|
||
#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";
|
||
};
|
||
|
||
} |