#ifndef RTSPCAPTUREELEMENT_H
|
#define RTSPCAPTUREELEMENT_H
|
|
#include <basic/pipe/TimerElement.h>
|
#include <opencv2/opencv.hpp>
|
#include "../StructureApp/HiredisTool.h"
|
#include <atomic>
|
|
class RtspAnalysManager;
|
|
struct CvCapture_FFMPEG;
|
|
/**
|
* 使用ffmpeg封装的视频采集流水元素
|
* 输入，rtsp地址或文件（mp4或avi）路径，输出opencv中的cv::Mat
|
* 支持GPU硬解码
|
* 主要完成视频帧的获取以及向CvCapture_FFMPEG传送数据
|
*/
|
class RtspCaptureElement : public TimerElement {
|
public:
|
RtspCaptureElement(const std::string &path, const std::string &camId, int fps = 30, int reOpenTime = -1,
|
int gpuIndex = -1, RtspAnalysManager *manager = nullptr);
|
|
//保存视频接口,从RtspAnalysManager发起调用
|
void SaveVideo(const std::string &strImageName);
|
|
//设置保存视频的最小和最大长度,单位是秒,实际的运行情况有一些差距，需要完善
|
void SetVideoMinMaxSeconds(const int minSeconds, const int maxSeconds);
|
|
private:
|
//线程执行函数
|
virtual void timerFunc() override;
|
|
//线程的一些变量初始化,线程启动的时候调用
|
virtual void threadInitial() override;
|
|
//线程结束函数的时候调用,完成一些资源的释放
|
virtual void threadClosing() override;
|
|
//打开视频
|
void openVideo();
|
|
//根据timeStamp创建路径
|
std::string MakeDir(const std::string &timeStamp);
|
|
private:
|
//用来抓取视频的Ffmpeg的封装类
|
CvCapture_FFMPEG *m_capture;
|
//保存视频流的路径,类似于rtsp://admin:a1234567@192.168.1.201:554/h264/ch2/main/av_stream
|
std::string m_path;
|
|
// Redis的工具类
|
// HiredisTool m_redisTool;
|
|
//对保存到Redis的图片进行计数
|
std::atomic<int> m_picCount{0};
|
//GPU的索引
|
int m_gpuIndex;
|
|
//打开视频流失败的时候，sleep一段时间
|
int m_reopenTime;
|
|
//摄像机ID
|
std::string m_camId;
|
|
//用来保存录像视频的路径
|
std::string m_cutPath;
|
|
//几张图丢一张,目前是8张丢一张
|
const int m_nPicsPickOne = 8;
|
|
RtspAnalysManager *m_pManager;
|
};
|
|
#endif // VIDEOCAPTUREELEMENT_H
|
