46 lines
1.9 KiB
C++
46 lines
1.9 KiB
C++
#pragma once
|
||
|
||
#include "vp_des_node.h"
|
||
|
||
namespace vp_nodes {
|
||
// image des node, save image to local file or push image to remote via udp.
|
||
class vp_image_des_node: public vp_des_node
|
||
{
|
||
private:
|
||
// gstreamer template for saving image to file (jpeg encoding only, filename MUST end with 'jpg/jpeg')
|
||
std::string gst_template_file = "appsrc ! videoconvert ! videoscale ! videorate ! video/x-raw,framerate=1/%d ! %s ! multifilesink location=%s";
|
||
// gstreamer template for pushing image to remote via udp （jpeg encoding only)
|
||
std::string gst_template_udp = "appsrc ! videoconvert ! videoscale ! videorate ! video/x-raw,format=I420,framerate=1/%d ! %s ! rtpjpegpay ! udpsink host=%s port=%d";
|
||
|
||
// gstreamer wrapper by opencv
|
||
cv::VideoWriter image_writer;
|
||
|
||
// save/push ONE image every `interval` seconds
|
||
int interval;
|
||
// save/push image to where
|
||
// for example, `./%d.jpg` for file (end with `jpg/jpeg`), `192.168.1.90:8000` for udp (split by `:`)
|
||
std::string location = "./%d.jpg";
|
||
|
||
bool osd = true;
|
||
vp_objects::vp_size resolution_w_h;
|
||
|
||
bool to_file = true;
|
||
// set jpegenc as the default encoder, we can use hardware encoder instead.
|
||
std::string gst_encoder_name = "jpegenc";
|
||
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_image_des_node(std::string node_name,
|
||
int channel_index,
|
||
std::string location = "./%d.jpg",
|
||
int interval = 5,
|
||
vp_objects::vp_size resolution_w_h = {},
|
||
bool osd = true,
|
||
std::string gst_encoder_name = "jpegenc");
|
||
~vp_image_des_node();
|
||
virtual std::string to_string() override;
|
||
};
|
||
|
||
} |