#pragma once
|
|
#include <string>
|
#include <vector>
|
#include <sstream>
|
#include <iostream>
|
|
#include "vp_src_node.h"
|
|
namespace vp_nodes
|
{
|
// rtsp source node, receive video stream via rtsp protocal.
|
// example:
|
// rtsp://admin:admin12345@192.168.77.110:554/
|
class vp_rtsp_ffmpeg_src_node : public vp_src_node
|
{
|
private:
|
/* data */
|
std::string ffprobe_template = "ffprobe -v error -select_streams v:0 -show_entries stream=width,height,r_frame_rate -of csv=p=0 %s";
|
cv::VideoCapture rtsp_capture;
|
|
std::vector<int> initialize_stream_properties();
|
int start_ffmpeg_thread(const std::string &rtsp_url, int width, int height);
|
|
|
protected:
|
// re-implemetation
|
virtual void handle_run() override;
|
|
public:
|
vp_rtsp_ffmpeg_src_node(std::string node_name,
|
int channel_index,
|
std::string rtsp_url,
|
float resize_ratio = 1.0,
|
int skip_interval = 0,
|
bool use_gpu = false,
|
std::string ffmpeg_format = "rawvideo", // other options: avi/mp4/mov/mkv/flv/gif/h264/hevc
|
std::string ffmpeg_pix_fmt = "bgr24"); // yuv420p 带宽小一半，但做detection 时还需要转换成bgr24
|
~vp_rtsp_ffmpeg_src_node();
|
|
virtual std::string to_string() override;
|
|
std::string rtsp_url;
|
|
bool use_gpu = false;
|
|
std::string ffmpeg_format = "rawvideo";
|
std::string ffmpeg_pix_fmt = "bgr24";
|
// 0 means no skip
|
int skip_interval = 0;
|
};
|
}
|
