#ifndef _FFMPEG_FORMAT_CONTEXT_IN_H_
#define _FFMPEG_FORMAT_CONTEXT_IN_H_

#include <stdint.h>
#include <memory>
#include <deque>
#include "PsToEs.hpp"

struct AVFormatContext;
struct AVDictionary;
struct AVStream;
struct AVCodecContext;
struct AVPacket;
struct AVFrame;
struct AVCodec;
struct AVIOContext;
struct AVCodecParserContext;

typedef int(* read_packet)(void *opaque,uint8_t *buf, int buf_size);

namespace ffwrapper{

	class VideoProp;

	class FormatIn
	{
	public:
		explicit FormatIn(bool hw=true);
		explicit FormatIn(const VideoProp &prop, bool hw=true);
		virtual ~FormatIn();
		
	public:

		virtual int open(const char *filename, AVDictionary **options);
		virtual const bool IsHEVC()const;
		virtual const bool IsAVC1()const;
		virtual int readPacket(AVPacket *pkt_out);
		virtual bool isVideoPkt(AVPacket *pkt);
		virtual bool isAudioPkt(AVPacket *pkt);
		bool notVideoAudio(AVPacket *pkt);

		int openWithCustomIO(void *opaque, read_packet fn, AVDictionary **options=NULL);
		bool openCodec(AVDictionary **options);
		int decode(AVFrame* frame, AVPacket *pkt);

	protected:
		bool allocCodec(AVCodec *dec, AVStream *s, AVDictionary **options);
	public:
		AVStream *getStream(int type = -1);
		AVCodecContext *getCodecContext(int type = 0);
		AVFormatContext *getFromatContext(){return ctx_;}
		const double getFPS()const{return fps_;}
	protected:
	 	AVFormatContext 	*ctx_;
	 	AVCodecContext 		*dec_ctx_;
	 	int 				vs_idx_;
		int 				as_idx_;

		VideoProp 			*prop_;
		int 				fps_;
	private:
		AVIOContext			*io_ctx_;
		uint8_t 			*read_io_buff_;
		const int 			read_io_buff_size_;
	};

	class FormatInGB : public FormatIn{
	public:
		FormatInGB();
		explicit FormatInGB(const VideoProp &prop);
		~FormatInGB();

		virtual int open(const char *filename, AVDictionary **options) override;
		virtual const bool IsHEVC()const override;
		virtual const bool IsAVC1()const override;
		virtual int readPacket(AVPacket *pkt_out) override;

		virtual bool isVideoPkt(AVPacket *pkt) override;
		virtual bool isAudioPkt(AVPacket *pkt) override;

	private:
		GB28181API* 			gb28181_;
		AVCodecParserContext* 	parser_ctx_;
		unsigned char*  		buffer_;
		int 					buffer_size_;

		std::deque<AVPacket*>   q_pkt_;
	};
}

#endif