FFmpeg入门测试

#include <iostream>
#include <stdio.h>

#define __STDC_CONSTANT_MACROS

extern "C"
{
    #include "libavutil/imgutils.h"   //av_image使用
    #include "libavcodec/avcodec.h"   //编解码处理
    #include "libavformat/avformat.h" //文件封装处理
    #include "libswscale/swscale.h"//图像变换
    #include "libavdevice/avdevice.h"
};

void PrintErrStr(const char *str)
{
	printf(str);
	printf("\n\n");
	system("pause");
}

int Decode(AVCodecContext *dec_ctx, AVFrame *frame, AVPacket *pkt)//解码
{
	 int ret = avcodec_send_packet(dec_ctx, pkt);// 先发送包数据到解码上下文中
	 if (ret < 0)
	 {
		 PrintErrStr("发送数据包进行解码时出错.");
		 return ret;
	 }
	 return avcodec_receive_frame(dec_ctx, frame);// 然后从解码上下文中读取帧数据到frame对象中
}

int main()
{
	AVFormatContext *pFormatCtx = NULL;//主要用于处理封装格式（FLV/MKV/RMVB等）
	AVCodecContext *pVideoCodecCtx = NULL;
	AVCodecContext *pAudioCodecCtx = NULL;
	AVCodec *pVideoCodec = NULL;
	AVCodec *pAudioCodec = NULL;
	AVFrame *pFrame;   //存储非压缩的数据（视频对应RGB/YUV像素数据，音频对应PCM采样数据）
	AVPacket packet;   //存储压缩数据（视频对应H.264等码流数据，音频对应AAC/MP3等码流数据）

	int iHour, iMinute, iSecond, iTotalSeconds;//HH:MM:SS
	int videoStreamIndex, audioStreamIndex; //音视频流索引
	int ret, FrameBytes;
	const char In_FileName[] = "F:\\TestMov\\Titanic.ts";//输入文件
	const char Out_h264_FileName[] = "F:\\TestMov\\output.h264";//输出h264文件
	const char Out_rgb24_FileName[] = "F:\\TestMov\\output.rgb24";//输出h264文件

	FILE *fp_Out_h264_File = fopen(Out_h264_FileName, "wb+");
	FILE *fp_Out_rgb24_File = fopen(Out_rgb24_FileName, "wb+");

	int OutImgW = 640;
	int OutImgH = 272;
	int OutImg_linesize[3] = {OutImgW * 3,0,0};

	avdevice_register_all();//注册所有组件
	if (avformat_open_input(&pFormatCtx, In_FileName, NULL, NULL) != 0)//打开输入视频文件
	{
		PrintErrStr("打开输入文件失败。");
		return -1;
	}

	if (avformat_find_stream_info(pFormatCtx, NULL) < 0)//获取文件信息
	{
		PrintErrStr("没有发现流信息。");
		return -1;
	}

	videoStreamIndex = -1;
	audioStreamIndex = -1;
	for (unsigned int i = 0; i < pFormatCtx->nb_streams; i++)//寻找流索引
	{
		if (pFormatCtx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO)//查找视频流索引
		{
			videoStreamIndex = i;
		}
		else
		if (pFormatCtx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_AUDIO)//查找音频流索引
		{
			audioStreamIndex = i;
		}
	}
	if ((videoStreamIndex == -1)||(audioStreamIndex == -1))
	{
		PrintErrStr("没有发现视频或音频流。");
		return -1;
	}

	pVideoCodec = avcodec_find_decoder(pFormatCtx->streams[videoStreamIndex]->codecpar->codec_id);//查找视频解码器
	if (pVideoCodec == NULL)
	{
		PrintErrStr("没有找到视频解码器。");
		return -1;
	}
	pAudioCodec = avcodec_find_decoder(pFormatCtx->streams[audioStreamIndex]->codecpar->codec_id);//查找音频解码器
	if (pVideoCodec == NULL)
	{
		PrintErrStr("没有找到音频解码器。");
		return -1;
	}

	pVideoCodecCtx = avcodec_alloc_context3(pVideoCodec);//申请视频解码空间
	if (pVideoCodecCtx == NULL)
	{
		PrintErrStr("无法分配视频解码器。");
		return -1;
	}
	pAudioCodecCtx = avcodec_alloc_context3(pAudioCodec);//申请音频解码空间
	if (pAudioCodecCtx == NULL)
	{
		PrintErrStr("无法分配音频解码器。");
		return -1;
	}
	avcodec_parameters_to_context(pVideoCodecCtx, pFormatCtx->streams[videoStreamIndex]->codecpar);
	avcodec_parameters_to_context(pAudioCodecCtx, pFormatCtx->streams[audioStreamIndex]->codecpar);
	 
	if (avcodec_open2(pVideoCodecCtx, pVideoCodec, NULL) < 0) //打开视频解码器
	{  
		PrintErrStr("没有打开视频解码器。");
		return -1;
	}
	if (avcodec_open2(pAudioCodecCtx, pAudioCodec, NULL) < 0) //打开音频解码器
	{
		PrintErrStr("没有打开音频解码器。");
		return -1;
	}

	pFrame = av_frame_alloc();//申请解压帧缓存
	if (pFrame == NULL)
	{
		PrintErrStr("申请解压帧缓存失败。");
		return -1;
	}

	FrameBytes = av_image_get_buffer_size(AV_PIX_FMT_RGB24, pVideoCodecCtx->width, pVideoCodecCtx->height, 1);//获取视频帧字节数
	uint8_t *pFrameBuffer = (uint8_t *)av_malloc(FrameBytes);//创建动态视频帧数组

	while (av_read_frame(pFormatCtx, &packet) >= 0)//从文件中读取一个packet 
	{
		if (packet.stream_index == videoStreamIndex)//如果是视频帧 
		{
			fwrite(packet.data, packet.size, 1, fp_Out_h264_File);//写视频包文件
			ret = Decode(pVideoCodecCtx, pFrame, &packet);//解码 packet->pFrame
			if (ret == 0) //已得到解码的图像在pFrame里
			{
				struct SwsContext *pSwsCtx = NULL;
				pSwsCtx = sws_getContext( pVideoCodecCtx->width, pVideoCodecCtx->height, pVideoCodecCtx->pix_fmt,//源宽度高度和格式YUV420 
					                      OutImgW, OutImgH, AV_PIX_FMT_RGB24,//转换到目标的宽度高度和格式
					                      1,//缩放算法 SWS_FAST_BILINEAR
					                      NULL, NULL, NULL);//初始化转换
				if (!pSwsCtx)
				{
					PrintErrStr("无法初始化转换Frame帧。");
					return -1;
				} 
				sws_scale(pSwsCtx, pFrame->data, pFrame->linesize, 0, pVideoCodecCtx->height, &pFrameBuffer, OutImg_linesize);
				fwrite(pFrameBuffer, FrameBytes, 1, fp_Out_rgb24_File);//写RGB视频文件
				sws_freeContext(pSwsCtx);//释放转换
			}
		}
		av_packet_unref(&packet);
	}

	//输出信息
	puts("[文件信息]");
	iTotalSeconds = (int)pFormatCtx->duration / 1000000; //S
	iHour = iTotalSeconds / 3600;//小时
	iMinute = iTotalSeconds % 3600 / 60;//分钟
	iSecond = iTotalSeconds % 60;//秒
	printf("播放时间：%02d:%02d:%02d\n", iHour, iMinute, iSecond);
	printf("流 个 数：%d\n", pFormatCtx->nb_streams);
	printf("封装格式：%s\n", pFormatCtx->iformat->long_name);
	printf("\n");

	puts("[视频信息]");
	printf("编码格式：%s\n", pVideoCodec->long_name);
	printf("视频码率：%I64d kb/s\n", pVideoCodecCtx->bit_rate / 1000);
	printf("分 辨 率：%d * %d\n", pVideoCodecCtx->width, pVideoCodecCtx->height);
	printf("\n");

	puts("[音频信息]");
	printf("编码格式：%s\n", pAudioCodec->long_name);
	printf("音频码率：%I64d kb/s\n", pAudioCodecCtx->bit_rate / 1000);
	printf("通 道 数：%d\n", pAudioCodecCtx->channels);
	printf("采 样 率：%d\n", pAudioCodecCtx->sample_rate);
	printf("\n");

	fclose(fp_Out_h264_File);
	fclose(fp_Out_rgb24_File);
	av_free(pFrameBuffer);
	av_frame_free(&pFrame);
	avcodec_close(pVideoCodecCtx);
	avcodec_close(pAudioCodecCtx);
	avformat_close_input(&pFormatCtx);

	printf("\n\n");
	system("pause");
}