#pragma once
#include "stdafx.h"
#include<iostream>
using namespace std;
extern char *VideoPath;
extern int CameraIndex;
extern int fps;
extern "C"
{
#include "libavcodec/avcodec.h"
#include "libavformat/avformat.h"
#include "libswscale/swscale.h"
#include <libavdevice/avdevice.h>  

};
class ffmpegDeCode
{
public:
	ffmpegDeCode();
	~ffmpegDeCode();
	bool DecoderInit(const char *sdpurl);
	void DecodeFrame();
	void DecodeLastFrame();
	void StopDecodeFrame();
	void GetopencvFrame();
	
	//////////解码部分////////
	AVFrame	*pFrame;//解码后的帧数据
	AVFormatContext	*pFormatCtx  ;//视频流编码上下文的指针
	int videoindex;//视频流索引
	AVCodecContext	*pCodecCtx;//用于编解码器的类型（视频，音频...）及设置
	AVCodec			*pCodec   ;//解码器指针
	long m_framenums;//帧序号
	SwsContext *img_convert_ctx;//根据编码信息设置渲染格式
	AVPacket *packet;////是存储压缩编码数据相关信息的结构体,数据流分包(一帧的分包)
	//1个AVPacket的data通常对应一个NAL。
	IplImage * pCVFrame ;
	char datatime[60];
	char Recordname[100];
	//////录像部分/////////
	AVStream *i_video_stream;
	AVFormatContext *opFormatCtx;
	AVStream *o_video_stream;
	void WriteVideoHead();
	void  WriteVideoTail();
	int m_recordframenums;
	int m_Toatlframenums;
	int last_pts ; 
	int last_dts;
	int64_t pts, dts; 
	int count;//录像文件计数
	
};

#include "stdafx.h"
#include"stdio.h"
#include"ffmpegDeCode.h"
#include "sqlite/sqlite3.h"
#include <fstream>
#include <windows.h> 
SYSTEMTIME sys; //系统时间
char systime[50];

ofstream outfile;//打印异常信息
ffmpegDeCode :: ffmpegDeCode()
{
	pFrame = NULL/**pFrameRGB = NULL*/;
	pFormatCtx  = NULL;
	videoindex=0;
	pCodecCtx   = NULL;
	pCodec      = NULL;
	m_framenums = 0;
	img_convert_ctx = NULL;
	packet = NULL;
	pCVFrame = NULL;
	//////////////
	i_video_stream= NULL;
	opFormatCtx= NULL;
	o_video_stream= NULL;
	m_recordframenums=1;
	last_pts = 0; 
	last_dts = 0;
	pts=0;
	dts=0;
	count=1;
	m_Toatlframenums=fps*60*60;//fps*3600
	memset(datatime, 0, 100*sizeof(char));
	memset(Recordname, 0, 100*sizeof(char));
	
}
ffmpegDeCode :: ~ffmpegDeCode()
{   
	WriteVideoTail();
	if(pCVFrame)
		cvReleaseImage(&pCVFrame);
	if(pCodecCtx)
	avcodec_close(pCodecCtx);
	if(pFormatCtx)
	avformat_close_input(&pFormatCtx);
	m_framenums=0;
}
void ffmpegDeCode::StopDecodeFrame()//WriteVideoTail()手动
{
	WriteVideoTail();//关闭
	if(pCVFrame)
	cvReleaseImage(&pCVFrame);
	if(pCodecCtx)
	{
	avcodec_close(pCodecCtx);
	pCodecCtx=NULL;
	}
	if(pFormatCtx)
	{
	avformat_close_input(&pFormatCtx);
	pFormatCtx=NULL;
	}
	m_framenums=0;
	
}


/*ffmpeg及相关变量初始化*/
bool ffmpegDeCode::DecoderInit(const char *sdpurl)
{
	
	//1)ffmpeg注册复用器，编码器等的函数。
	av_register_all();
	avformat_network_init();//需要播放网络视频
	//avdevice_register_all();//打开摄像头时需要
    pFormatCtx = avformat_alloc_context();
	//打开视频文件,通过参数filepath来获得文件名。这个函数读取文件的头部并且把信息保存到我们给的AVFormatContext结构体中。
	//最后2个参数用来指定特殊的文件格式，缓冲大小和格式参数，但如果把它们设置为空NULL或者0，libavformat将自动检测这些参数。
	/////2)打开文件或摄像头或网络流
	if(avformat_open_input(&pFormatCtx,sdpurl,NULL,NULL)!=0)
	//avformat_open_input(&pFormatCtx,url,NULL,&avdic)
	{
		GetLocalTime( &sys ); 
	    sprintf(systime,"\n%4d-%02d-%02d-%02d:%02d:%02d.%03d",sys.wYear,sys.wMonth,sys.wDay,sys.wHour,sys.wMinute, sys.wSecond,sys.wMilliseconds); 
		outfile<<systime<<"-->Couldn't open stream！";
		StopDecodeFrame();
		return false;
	}
	//3）查找文件的流信息,avformat_open_input函数只是检测了文件的头部，接着要检查在文件中的流的信息
	if(av_find_stream_info(pFormatCtx)<0)
	{
		GetLocalTime( &sys ); 
	    sprintf(systime,"\n%4d-%02d-%02d-%02d:%02d:%02d.%03d",sys.wYear,sys.wMonth,sys.wDay,sys.wHour,sys.wMinute, sys.wSecond,sys.wMilliseconds); 
		outfile<<systime<<"-->Couldn't find stream information！";
		StopDecodeFrame();
		return false;
	}
	//遍历文件的各个流，找到第一个视频流，并记录该流的编码信息
	videoindex=-1;
	for(int i=0; i<pFormatCtx->nb_streams; i++) 
	{
		if(pFormatCtx->streams[i]->codec->codec_type==AVMEDIA_TYPE_VIDEO)
		{
			videoindex=i;
			i_video_stream = pFormatCtx->streams[i];
			break;
		}
	}
	if(videoindex==-1)
	{
		GetLocalTime( &sys ); 
	    sprintf(systime,"\n%4d-%02d-%02d-%02d:%02d:%02d.%03d",sys.wYear,sys.wMonth,sys.wDay,sys.wHour,sys.wMinute, sys.wSecond,sys.wMilliseconds); 
		outfile<<systime<<"-->Couldn't find a video stream！";
		StopDecodeFrame();
		return false;
	}
	pCodecCtx=pFormatCtx->streams[videoindex]->codec;
	//4）在库里面查找支持该格式的解码器
	pCodec=avcodec_find_decoder(pCodecCtx->codec_id);
	if(pCodec==NULL)
	{
		GetLocalTime( &sys ); 
	    sprintf(systime,"\n%4d-%02d-%02d-%02d:%02d:%02d.%03d",sys.wYear,sys.wMonth,sys.wDay,sys.wHour,sys.wMinute, sys.wSecond,sys.wMilliseconds); 
		outfile<<systime<<"-->Couldn't find Codec！";
		StopDecodeFrame();
		return false;
	}
	//5）打开解码器
	if(avcodec_open2(pCodecCtx, pCodec,NULL)<0)//avcodec_open2(pCodecCtx, pCodec,NULL)
	{
		GetLocalTime( &sys ); 
	    sprintf(systime,"\n%4d-%02d-%02d-%02d:%02d:%02d.%03d",sys.wYear,sys.wMonth,sys.wDay,sys.wHour,sys.wMinute, sys.wSecond,sys.wMilliseconds); 
		outfile<<systime<<"-->Couldn't  open codec！";
		StopDecodeFrame();
		return false;
	}
	int y_size = pCodecCtx->width * pCodecCtx->height;
	packet=(AVPacket *)malloc(sizeof(AVPacket));//分配（一帧的分包）内存
	av_new_packet(packet, y_size);
	if (NULL==pCVFrame)
	{
		pCVFrame=cvCreateImage(cvSize(pCodecCtx->width, pCodecCtx->height),8,3); 
	}
	return true;
}


/*将解码后的RGB转化成opencv的帧结构*/
void ffmpegDeCode::GetopencvFrame()
{
	
	AVFrame	*pFrameRGB = avcodec_alloc_frame();//分配一个帧指针，指向解码后的转换帧
	uint8_t * out_bufferRGB=new uint8_t[avpicture_get_size( PIX_FMT_RGB24, 
		pCodecCtx->width, pCodecCtx->height)];//给out_bufferRGB帧分配内存;
	avpicture_fill((AVPicture *)pFrameRGB, 
		out_bufferRGB, PIX_FMT_RGB24, pCodecCtx->width, pCodecCtx->height);
	//(RGB转换成opencv的BGR)
	uchar *p = NULL;
	p = pFrame->data[1];
	pFrame->data[1] = pFrame->data[2];
	pFrame->data[2] = p;
	//设置图像转换上下文
    sws_scale(img_convert_ctx,
                        pFrame->data, pFrame->linesize, 0, pCodecCtx->height,
                        pFrameRGB->data, pFrameRGB->linesize);//格式转换，YUV420转换成RGB
	memcpy(pCVFrame->imageData,out_bufferRGB,pCodecCtx->width*pCodecCtx->height*3);
	pCVFrame->widthStep=pCodecCtx->width*3;// (3通道的)
	pCVFrame->origin=0;
	delete[] out_bufferRGB;
	av_free(pFrameRGB);
}
void ffmpegDeCode::DecodeFrame()
{
	
	int decode_result=0;
	int got_picture=0;//
	pFrame=avcodec_alloc_frame();//分配一个帧指针，指向解码后的帧
	///一个AVPacket最多只包含一个AVFrame，而一个AVFrame可能包含好几个AVPacket，
	///AVPacket是种数据流分包的概念。记录一些音视频相关的属性值，如pts,dts等
	if(pFormatCtx==NULL)
		exit(1);
	if(av_read_frame(pFormatCtx,packet)<0)
		exit(1);
	//SavedSensorData();//解析传感器数据
		if(packet->stream_index==videoindex)
		{
			decode_result = avcodec_decode_video2(pCodecCtx, pFrame, &got_picture, packet);
			////////分时段录像//////////////////
			if(m_framenums==30)//避免开始传感器数据不正常或没有
		    WriteVideoHead();//写视频头，第一段视频头
			if(m_framenums>=30)
			{
			packet->flags|=AV_PKT_FLAG_KEY; 
			pts = packet->pts; 
			packet->pts += last_pts; 
			dts = packet->dts; 
			packet->dts += last_dts; 
			packet->stream_index = 0;
			m_recordframenums++;
			av_interleaved_write_frame(opFormatCtx,packet);
			if(m_recordframenums==m_Toatlframenums)
			{
				WriteVideoTail();
				WriteVideoHead();
			}
			}
		//////////////////////
			if(decode_result<0)
			{
				GetLocalTime( &sys ); 
				sprintf(systime,"\n%4d-%02d-%02d-%02d:%02d:%02d.%03d",sys.wYear,sys.wMonth,sys.wDay,sys.wHour,sys.wMinute, sys.wSecond,sys.wMilliseconds); 
				outfile<<systime<<"\n-->解码错误！\n"; 
				return ;
			}
			if(got_picture)//got_picture标志解码出完整一帧
			{
			////////////////////
			//printf("帧号: %d,帧大小：%d,关键帧标志：%d,显示时间戳:%d,解码时间戳%d\n",
			//	m_framenums,packet->size,packet->data,packet->pts,packet->dts);
			////printf("帧号: %d,帧大小：%d,关键帧标志：%d,显示时间戳:%d,解码时间戳%d\n",
			//	//m_framenums,pFrame->data,pFrame->key_frame);
			//////////////////////
			m_framenums++;
			img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt, 
				pCodecCtx->width, pCodecCtx->height, PIX_FMT_RGB24, SWS_BICUBIC, NULL, NULL, NULL); 
			if(img_convert_ctx != NULL)  
			{  
				GetopencvFrame();
				sws_freeContext(img_convert_ctx);
			}  
		}
	}
    if(pFrame)
	av_free(pFrame);
	if(packet)
	av_free_packet(packet);
}
void ffmpegDeCode ::WriteVideoHead()
{
	
    sprintf(Recordname,"%sCam%d-%s.mp4",VideoPath,CameraIndex+1,datatime);
	avformat_alloc_output_context2(&opFormatCtx, NULL, NULL, Recordname);
	//int y_size = pCodecCtx->width * pCodecCtx->height;
	//packet2=(AVPacket *)malloc(sizeof(AVPacket));//分配（一帧的分包）内存
	//av_new_packet(packet2, y_size);
	o_video_stream = avformat_new_stream(opFormatCtx, NULL); 
	AVCodecContext *c; 
	c = o_video_stream->codec; 
	//c->bit_rate = 900000; 
	c->bit_rate = i_video_stream->codec->bit_rate; 
	c->codec_id = i_video_stream->codec->codec_id; 
	c->codec_type = i_video_stream->codec->codec_type; 
	c->time_base.num = i_video_stream->time_base.num; 
	c->time_base.den = i_video_stream->time_base.den; 
	//fprintf(stderr, "time_base.num = %d time_base.den = %d\n", c->time_base.num, c->time_base.den); 
	c->width = i_video_stream->codec->width; 
	c->height = i_video_stream->codec->height; 
	c->pix_fmt = i_video_stream->codec->pix_fmt; 
	//printf("%d %d %d", c->width, c->height, c->pix_fmt); 
	c->flags = i_video_stream->codec->flags; 
	c->flags |= CODEC_FLAG_GLOBAL_HEADER; 
	c->me_range = i_video_stream->codec->me_range; 
	c->max_qdiff = i_video_stream->codec->max_qdiff;

	c->qmin = i_video_stream->codec->qmin; 
	c->qmax = i_video_stream->codec->qmax;
	c->qcompress = i_video_stream->codec->qcompress; 
	avio_open(&opFormatCtx->pb,Recordname, AVIO_FLAG_WRITE);
	if(avformat_write_header(opFormatCtx, NULL)>=0)
	{
		GetLocalTime( &sys ); 
		sprintf(systime,"\n%4d-%02d-%02d-%02d:%02d:%02d.%03d",sys.wYear,sys.wMonth,sys.wDay,sys.wHour,sys.wMinute, sys.wSecond,sys.wMilliseconds); 
		outfile<<systime<<"-->写入视频头文件成功！"; 
	}

}
void ffmpegDeCode ::WriteVideoTail()
{
	last_dts += dts; 
	last_pts += pts;
	if(opFormatCtx)
	{
	if(av_write_trailer(opFormatCtx)>=0)
	{
	GetLocalTime( &sys ); 
	sprintf(systime,"\n%4d-%02d-%02d-%02d:%02d:%02d.%03d",sys.wYear,sys.wMonth,sys.wDay,sys.wHour,sys.wMinute, sys.wSecond,sys.wMilliseconds); 
	outfile<<systime<<"-->写入视频尾文件成功！";
	}
	avcodec_close(opFormatCtx->streams[0]->codec); 
	av_freep(&opFormatCtx->streams[0]->codec); 
	av_freep(&opFormatCtx->streams[0]);
	avio_close(opFormatCtx->pb); 
	av_free(opFormatCtx);
	}
	/*if(packet2)
		av_free_packet(packet2);*/
	m_recordframenums=1;
	last_pts = 0; 
	last_dts = 0;
}

#include "stdafx.h"
#include "ffmpegDecode.h"
#include "highgui.h"
#include "cv.h"
#include "cxcore.h"
#include <windows.h>
char *VideoPath="D:\\";
int CameraIndex=3;//测试数据存在Cam4_SensorData
int fps=25;
char *sdpurl1 = "test1.sdp";//test.sdp  D:\\DataSaved\\HC_Record_ch04_20160410145430.mp4
ffmpegDeCode deCode1;
int _tmain(int argc, _TCHAR* argv[])
{

	///////////初始化///////
   if(!deCode1.DecoderInit(sdpurl1))
   {
	   cout<<"播放失败!"<<endl; 
   }
	cvNamedWindow("Decode1"); 
	//////解码播放//////////
	while(1)
	{
		deCode1.DecodeFrame();
		cout<<"------->第1路第"<<deCode1.m_framenums<<"帧"<<endl; 
		if(deCode1.pCVFrame)
		{
			cvShowImage("Decode1",deCode1.pCVFrame); 
		    cvWaitKey(5);//标准1000/fps
			
		}
	}
	return 0;
}