#ifndef RTSPCAPTUREELEMENT_H
#define RTSPCAPTUREELEMENT_H

#include <basic/pipe/TimerElement.h>
#include <opencv2/opencv.hpp>
#include "../StructureApp/HiredisTool.h"
#include <atomic>
struct CvCapture_FFMPEG;

    /**
     * 使用ffmpeg封装的视频采集流水元素
     * 输入，rtsp地址或文件（mp4或avi）路径，输出opencv中的cv::Mat
     * 支持GPU硬解码
     */
    class RtspCaptureElement: public TimerElement {
    public:
        RtspCaptureElement(const std::string& path,const std::string& camId, int fps = 30, int reOpenTime = -1, int gpuIndex = -1);
        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();

        //
        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;

        //几张图丢一张,目前是4张丢一张
        const int m_nPicsPickOne = 8;

        //每台摄像机保存到Redis的图片数量的最大值
        const int M_CAM_PIC_MAX_COUNT = 50;
    };

#endif // VIDEOCAPTUREELEMENT_H