/**
 * 拉流engine
 * */

#ifndef INC_VIDEOENGINE_H
#define INC_VIDEOENGINE_H

#include <arpa/inet.h>
#include "AppCommon.h"


 //编译器这部分代码按C语言（而不是C++）的方式进行编译
#ifdef __cplusplus
extern "C"
{
#endif // #ifdef __cplusplus

#include "libavutil/imgutils.h"
#include "libavutil/samplefmt.h"
#include "libavformat/avformat.h"
#include "libavcodec/avcodec.h"

#ifdef __cplusplus
}
#endif // #ifdef __cplusplus

#include "EngineBase.h"
#include "EngineFactory.h"


typedef struct
{
    int64_t i64Timeout;
} TimeoutContext;

//#define RTSP_PULL_CAMERA_VIDEO_STREAM
class VideoEngine : public ai_matrix::EngineBase
{
public:
    VideoEngine();
    ~VideoEngine();

    APP_ERROR Init() override;
    APP_ERROR DeInit() override;
    APP_ERROR Process() override;

protected:
    AVFormatContext *CreateFormatContext();
    APP_ERROR GetStreamInfo();
    APP_ERROR ConnectCamera(); //连接相机
    void ResetCamera();        //复位相机连接
    //static int InterruptCallback(void *pData);

private:
    AVFormatContext *pFormatCtx_ = nullptr;
    AVCodecParameters *pCodecParameters_ = nullptr;
    int iVideoStream_ = 0;
    bool bIsAvc_ = false;
    int iAudioStream_ = -1;

    ai_matrix::DataSourceConfig dataSourceConfig_;
    DataSourceInfo frameInfo_;
    bool bConnectFlag_ = true; //默认不重连相机
    std::string strPort0_;
    std::string strPort1_;

};

ENGINE_REGIST(VideoEngine)

#endif