ffmpeg之h264格式转yuv

h264格式转yuv具体步骤

• 初始化FFmpeg库：通过av_register_all()来初始化须要的组件。
  
• 打开输入文件并查找解码器：使用avformat_open_input和 avcodec_find_decoder 打开H.264文件，并查找视频流。
  
• 分配并配置解码上下文：使用 avcodec_alloc_context3 分配解码上下文，并设置须要的参数。
  
• 使用av_parser_init初始化分析器上下文
  
• 打开解码器：通过 avcodec_open2 打开解码器。
  
• 读取和解码帧：使用 av_parser_parse2 数据经过分析器分析，并使用 avcodec_send_packet 和 avcodec_receive_frame 解码帧。
  
• 生存YUV帧到文件：将解码后的YUV帧写入输出文件。
  
• 清理资源：开释所有分配的资源，确保没有内存泄漏。
  

具体代码如下：

1. #include "ffmpeg.h"
2. #include <QFile>
3. #include <QDebug>
4. extern "C"
5. {
6. #include <libavcodec/avcodec.h>
7. #include <libavutil/avutil.h>
8. #include <libavutil/imgutils.h>
10. }
12. #define ERROR_BUF(ret) \
13.     char errbuf[1024]; \
14.     av_strerror(ret,errbuf,sizeof(errbuf));
16. //输入缓冲区的大小
17. #define IN_DATA_SIZE 4096
19. ffmpegs::ffmpegs()
20. {
22. }
24. static int decode(AVCodecContext *ctx,AVPacket *pkt,AVFrame *frame,QFile &outFile)
25. {
26.     //发送压缩数据到解码器
27.     int ret = avcodec_send_packet(ctx,pkt);
28.     if(ret < 0)
29.     {
30.         ERROR_BUF(ret);
31.         qDebug() << "avcodec_send_packet error" << errbuf;
32.         return ret;
33.     }
35.     while(true)
36.     {
37.         //获取解码后的数据
38.         ret = avcodec_receive_frame(ctx,frame);
39.         if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
40.         {
41.             return 0;
42.         } else if (ret < 0)
43.         {
44.             ERROR_BUF(ret);
45.             qDebug() << "avcodec_receive_frame error" << errbuf;
46.             return ret;
47.         }
49.         // 将解码后的数据写入文件
50.         //写入y平面
51.         outFile.write((char *) frame->data[0], frame->linesize[0] * ctx->height);
52.         //写入u平面
53.         outFile.write((char *) frame->data[1], frame->linesize[1] * ctx->height >> 1);
54.         //写入v平面
55.         outFile.write((char *) frame->data[2], frame->linesize[2] * ctx->height >> 2);
56.     }
57. }
59. void ffmpegs::h264Decode(const char *inFileName, videodecodeSpec &out)
60. {
61.     //返回结果
62.     int ret = 0;
64.     //用来存放读取的文件数据（h264）
65.     //加上AV_INPUT_BUFFER_PADDING_SIZE是为了防止某些优化过的reader一次性读取过多导致越界
66.     char inDataArray[IN_DATA_SIZE + AV_INPUT_BUFFER_PADDING_SIZE];
67.     char *inData = inDataArray;
69.     //每次从输入文件中读取的长度（h264）
70.     //输入缓冲区中，剩下等待进行解码的有效数据长度
71.     int inLen = 0;
72.     int inEnd = 0;
74.     //文件
75.     QFile inFile(inFileName);
76.     QFile outFile(out.filename);
78.     //解码器
79.     AVCodec *codec = nullptr;
80.     //上下文
81.     AVCodecContext *ctx = nullptr;
82.     //编解码器上下文
83.     AVCodecParserContext *parserCtx = nullptr;
85.     //存放编码前的数据(h264)
86.     AVPacket *pkt = nullptr;
87.     //存放编码后的数据(yuv)
88.     AVFrame *frame = nullptr;
90.     //获取编码器
91.     //codec = avcodec_find_decoder_by_name("h264");
92.     codec = avcodec_find_decoder(AV_CODEC_ID_H264);
93.     if(!codec)
94.     {
95.         qDebug() << "decodec not found";
96.         return;
97.     }
99.     //初始化解析器上下文
100.     //    parserCtx = av_parser_init(codec->id);//根据编码器ID进行初始化
101.     parserCtx = av_parser_init(AV_CODEC_ID_H264);//根据编码器ID进行初始化
102.     if(!parserCtx)
103.     {
104.         qDebug() << "av_parser_init error";
105.         return;
106.     }
108.     //创建上下文
109.     ctx = avcodec_alloc_context3(codec);
110.     if(!ctx)
111.     {
112.         qDebug() << "avcodec_alloc_context3 error";
113.         goto end;
115.     }
117.     //创建AVPacket
118.     pkt = av_packet_alloc();
119.     if(!pkt)
120.     {
121.         qDebug() << "av_packet_alloc error";
122.         goto end;
123.     }
125.     //创建AVFrame
126.     frame = av_frame_alloc();
127.     if(!frame)
128.     {
129.         qDebug() << "av_frame_alloc error";
130.         goto end;
131.     }
133.     //打开解码器
134.     ret = avcodec_open2(ctx,codec,nullptr);
135.     if(ret < 0)
136.     {
137.         ERROR_BUF(ret);
138.         qDebug() << "avcodec_open2 error" << errbuf;
139.         goto end;
140.     }
142.     //打开文件
143.     if(!inFile.open(QFile::ReadOnly))
144.     {
145.         qDebug() << "file open error:" << inFileName;
146.         goto end;
147.     }
148.     if(!outFile.open(QFile::WriteOnly))
149.     {
150.         qDebug() << "file open error:" << out.filename;
151.         goto end;
152.     }
154.     //读取文件数据
155.     do
156.     {
157.         inLen = inFile.read(inDataArray, IN_DATA_SIZE);
158.         //设置是否到了文件尾部
159.         inEnd = !inLen;
161.         //让inData指向数组的首元素
162.         inData = inDataArray;
164.         //只要输入缓冲区中还有等待进行解码的数据
165.         while(inLen > 0 || inEnd)
166.         {
167.             //经过解析器解析
168.             //内部调用的核心是：ff_aac_ac3_parse
169.             //到了文件尾部(虽然没有读取任何数据，但也要调用av_parser_parse2）
170.             //经过解析器解析
171.             ret = av_parser_parse2(parserCtx,ctx,&pkt->data,&pkt->size,
172.                                    (uint8_t *)inData,inLen,
173.                                    AV_NOPTS_VALUE,AV_NOPTS_VALUE,0);
174.             if (ret < 0)
175.             {
176.                 ERROR_BUF(ret);
177.                 qDebug() << "av_parser_parse2 error" << errbuf;
178.                 goto end;
179.             }
181.             //跳过已经解析过的数据
182.             inData += ret;
183.             //减去已经解析过的数据大小
184.             inLen -= ret;
186.             qDebug() << inEnd << pkt->size << ret;
188.             //解码
189.             if(pkt->size > 0)
190.             {
191.                 if(decode(ctx,pkt,frame,outFile) < 0)
192.                 {
193.                     goto end;
194.                 }
195.             }
197.             //如果到了文件末尾
198.             if(inEnd) break;
199.         }
200.     }while(!inEnd);
202.     //刷新缓冲区
203.     //    pkt->data = nullptr;
204.     //    pkt->size = 0;
205.     //    decode(ctx,pkt,frame,outFile);
206.     //刷新缓冲区
207.     decode(ctx,nullptr,frame,outFile);
209.     // 设置输出参数
210.     out.width = ctx->width;
211.     out.height = ctx->height;
212.     out.pixFmt = ctx->pix_fmt;
213.     out.fps = ctx->time_base.num;
215. end:
216.     inFile.close();
217.     outFile.close();
218.     av_packet_free(&pkt);
219.     av_frame_free(&frame);
220.     av_parser_close(parserCtx);
221.     avcodec_free_context(&ctx);
222. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。