ffmpeg C 视频解码

#include <stdio.h> #include <stdlib.h> #include "libavcodec/avcodec.h" #include "libavformat/avformat.h" #include "libswscale/swscale.h" #include "libavutil/avutil.h" int main(){ //获取输入输出文件名 char* src = "./test.mp4"; char* desc = "./test.yuv"; //封装格式上下文，统领全局的结构体，保存了视频文件封装格式的相关信息 AVFormatContext * avf_ctx = avformat_alloc_context(); int video_index = -1; //设置日志级别 av_log_set_level(AV_LOG_INFO); //1.注册所有组件 av_register_all(); //2.打开视频文件 int ret = avformat_open_input(&avf_ctx,src,NULL,NULL); if(ret < 0){ av_log(NULL,AV_LOG_ERROR,"Can't open the file %s \n",src); return -1; } //3.获取视频文件信息 ret = avformat_find_stream_info(avf_ctx,NULL); if(ret < 0){ av_log(NULL,AV_LOG_ERROR,"can't find the stream"); goto _end; } //打印视频文件信息 av_dump_format(avf_ctx,0,src,0); for(int i = 0; i < avf_ctx->nb_streams; i++){ // av_log(NULL,AV_LOG_INFO,"第%d次",i); if(avf_ctx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO){ video_index = i; break; } } if(video_index == -1){ av_log(NULL,AV_LOG_ERROR,"can't find the video stream"); goto _end; } //只有知道视频的编码方式，才能够根据编码方式去找到解码器 //获取视频流中的编解码上下文 AVCodecContext* avc_cxt = avf_ctx->streams[video_index]->codec; //根据视频中的编码Id找到对应的解码器 AVCodec *avc = avcodec_find_decoder(avc_cxt->codec_id); if(avc == NULL){ av_log(NULL,AV_LOG_ERROR,"can't find the decoder"); goto _end; } // 打开解码器 ret = avcodec_open2(avc_cxt,avc,NULL); if(ret < 0){ av_log(NULL,AV_LOG_ERROR,"can't open the codec"); goto _end; } // 输出视频信息 av_log(NULL,AV_LOG_INFO,"视频格式： %s \n",avf_ctx->iformat->name); av_log(NULL,AV_LOG_INFO,"视频时长： %lld s\n",avf_ctx->duration/1000000); av_log(NULL,AV_LOG_INFO,"视频宽x高： %d x %d \n",avc_cxt->width,avc_cxt->height); av_log(NULL,AV_LOG_INFO,"视频的解码器的名称：%s \n",avc->name); //准备读取 // AVPacket用于存储一帧一帧的压缩数据（H264） //缓冲区，开辟空间 AVPacket * avp = (AVPacket*)av_malloc(sizeof(AVPacket)); //AVFrame用于存储解码后的像素数据(YUV) //内存分配 AVFrame * avframe = av_frame_alloc(); //YUV420 AVFrame *pFrameYUV = av_frame_alloc(); //只有指定了AVFrame的像素格式、画面大小才能真正分配内存 //缓冲区分配内存 int8_t *buf = (int8_t*)av_malloc(avpicture_get_size(AV_PIX_FMT_YUV420P,avc_cxt->width,avc_cxt->height)); //初始化缓冲区 avpicture_fill((AVPicture*)pFrameYUV , buf ,AV_PIX_FMT_YUV420P , avc_cxt->width,avc_cxt->height); //用于转码（缩放）的参数，转之前的宽高，转之后的宽高，格式等 struct SwsContext *sws_ctx = sws_getContext(avc_cxt->width,avc_cxt->height,avc_cxt->pix_fmt, avc_cxt->width,avc_cxt->height,AV_PIX_FMT_YUV420P, SWS_BICUBIC,NULL,NULL,NULL); int got_pic; FILE *out_yuv = fopen(desc,"wb+"); int frame_count = 0; while(av_read_frame(avf_ctx,avp) >= 0){ //只要视频压缩数据（根据流的索引位置判断） if(avp->stream_index == video_index){ ret = avcodec_decode_video2(avc_cxt,avframe,&got_pic,avp); if(ret < 0){ av_log(NULL,AV_LOG_ERROR,"decode is error"); goto _end; } //如果got_pic == 0 表示解码完成 非0表示正在解码 if(got_pic){ //AVFrame转为像素格式YUV420，宽高 //2 6输入、输出数据 //3 7输入、输出画面一行的数据的大小 AVFrame 转换是一行一行转换的 //4 输入数据第一列要转码的位置 从0开始 //5 输入画面的高度 sws_scale(sws_ctx,avframe->data,avframe->linesize,0,avc_cxt->height,pFrameYUV->data,pFrameYUV->linesize); // sws_scale(sws_ctx, pFrame->data, pFrame->linesize, 0, pCodecCtx->height, // pFrameYUV->data, pFrameYUV->linesize); //输出到YUV文件 //AVFrame像素帧写入文件 //data解码后的图像像素数据（音频采样数据） //Y 亮度 UV 色度（压缩了） 人对亮度更加敏感 //U V 个数是Y的1/4 int y_size = avc_cxt->width * avc_cxt->height; fwrite(pFrameYUV->data[0], 1 ,y_size,out_yuv); fwrite(pFrameYUV->data[1], 1 ,y_size / 4,out_yuv); fwrite(pFrameYUV->data[2], 1 ,y_size / 4,out_yuv); frame_count++; av_log(NULL,AV_LOG_INFO,"解析完第%d帧\n",frame_count); } } //释放avpacket av_free_packet(avp); } _end: fclose(out_yuv); av_frame_free(&avframe); if(avc_cxt != NULL){ avcodec_close(avc_cxt); } avformat_free_context(avf_ctx); return 0; }