//
//  FFmpegH264Encoder.h
//  FFmpegRTSPServer
//
//  Created by Mina Saad on 9/22/15.
//  Copyright (c) 2015 Mina Saad. All rights reserved.
//

#ifndef MESAI_FFMPEGH264_ENCODER_H
#define MESAI_FFMPEGH264_ENCODER_H

#include <string>
#include <queue>
#include <pthread.h>
#include <functional>

#include "IEncoder.h"

extern "C" {
	
	#include <stdlib.h>
	#include <stdio.h>
	#include <string.h>
	#include <math.h>
	#include <libavutil/opt.h>
	#include <libavutil/mathematics.h>
	#include <libavformat/avformat.h>
	#include <libswscale/swscale.h>
	#include <libswresample/swresample.h>
	#include <libavutil/imgutils.h>
	#include <libavcodec/avcodec.h>

}

namespace MESAI
{
	class FrameStructure {
		public:
			uint8_t * dataPointer;
			int dataSize;
			int frameID;
			~FrameStructure()
			{
				delete dataPointer;
			}
	};

	class FFmpegH264Encoder : public IEncoder
	{
	public:
		FFmpegH264Encoder();
		virtual ~FFmpegH264Encoder();
		
		virtual void setCallbackFunctionFrameIsReady(std::function<void()> func);
		
		bool SetupVideo(std::string filename, int Width, int Height, int FPS, int GOB, int BitPerSecond);
		void CloseVideo();
		bool SetupCodec(const char *filename, int codec_id);
		void CloseCodec();
		

		void SendNewFrame(uint8_t * RGBFrame);		
		bool WriteFrameRGB(uint8_t * RGBFrame);
		bool WriteFrameYUV420(AVFrame * YUVFrame);
		virtual char ReleaseFrame();

		void run();	
		virtual char GetFrame(u_int8_t** FrameBuffer, unsigned int *FrameSize);

	private:


		std::queue<uint8_t*> inqueue;
		pthread_mutex_t inqueue_mutex;
		std::queue<FrameStructure *> outqueue;
		pthread_mutex_t outqueue_mutex;


		int m_sws_flags;
		int	m_AVIMOV_FPS;
		int	m_AVIMOV_GOB;
		int	m_AVIMOV_BPS;
		int m_frame_count;
		int	m_AVIMOV_WIDTH;
		int	m_AVIMOV_HEIGHT;
		std::string m_filename;

	    double m_video_time;

	    AVCodecContext *m_c;
		AVStream *m_video_st;
		AVOutputFormat *m_fmt;
		AVFormatContext *m_oc;
		AVCodec *m_video_codec;
	    AVFrame * m_src_picture, * m_dst_picture;
	    SwsContext *sws_ctx;
	    int bufferSize;

		std::function<void()> onFrame;

	};
}
#endif