利用FFmpeg将Jpeg图片转为任意视频容器格式

http://blog.csdn.net/weixinhum/article/details/45872093

   在一些嵌入式系统中，视频以jpeg图像格式传出来，而且数据量非常大。这种情况下如果要对jpeg图像进行解码并且重新编码成264之类的格式再放入视频容器中会消耗大量的CPU资源，若不是性能特别犀利的CPU是无法承受的，这个时候有一个折衷的办法，就是直接将jpeg连续保存起来（也就是保存成mjpeg视频格式），然后直接放入视频容器中。

    这个方面的代码实现网上资料比较少，可以说几乎没有，基本都是直接用FFmpeg的命令行来实现的，这一方法虽然可行但毕竟不太适合软件的集成，是故本人做了些研究，并将代码放出来供各位参考指正。

    在贴出代码之前特别感谢雷霄骅老师的博客，其关于FFmpeg的博文提供了非常多的参考，如果对FFmpeg有深入了解需求的可以去看看，会有很多收获。

    下面上代码:

    环境为：VS2013 C++ 控制台程序（空项目）+ FFmpeg(配置相关可参考本人之前博文)

[cpp] view plain copy

1. #include <stdio.h>  
2.   
3. extern "C"//包含C文件头  
4. {  
5. #include "libavformat/avformat.h"  
6. };  
7. #define DATASIZE 1024*1024  
8.   
9. AVStream *add_vidio_stream(AVFormatContext *oc, enum AVCodecID codec_id)//用以初始化一个用于输出的AVFormatContext结构体  
10. {  
11.     AVStream *st;  
12.     AVCodec *codec;  
13.   
14.     st = avformat_new_stream(oc, NULL);  
15.     if (!st)  
16.     {  
17.         printf("Could not alloc stream\n");  
18.         exit(1);  
19.     }  
20.     codec = avcodec_find_encoder(codec_id);//查找mjpeg解码器  
21.     if (!codec)  
22.     {  
23.         printf("codec not found\n");  
24.         exit(1);  
25.     }  
26.     avcodec_get_context_defaults3(st->codec, codec);//申请AVStream->codec(AVCodecContext对象)空间并设置默认值(由avcodec_get_context_defaults3()设置  
27.   
28.     st->codec->bit_rate = 400000;//设置采样参数，即比特率    
29.     st->codec->width = 1080;//设置视频宽高，这里跟图片的宽高保存一致即可  
30.     st->codec->height = 1800;  
31.     st->codec->time_base.den = 10;//设置帧率  
32.     st->codec->time_base.num = 1;  
33.   
34.     st->codec->pix_fmt = PIX_FMT_YUV420P;//设置像素格式    
35.     st->codec->codec_tag = 0;  
36.     if (oc->oformat->flags & AVFMT_GLOBALHEADER)//一些格式需要视频流数据头分开  
37.         st->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;  
38.     return st;  
39. }  
40.   
41. void main()  
42. {  
43.     AVFormatContext *ofmt_ctx = NULL;//其包含码流参数较多，是一个贯穿始终的数据结构，很多函数都要用到它作为参数  
44.     const char *out_filename = "out.mkv";//输出文件路径，在这里也可以将mkv改成别的ffmpeg支持的格式，如mp4，flv，avi之类的  
45.     int ret;//返回标志  
46.   
47.     av_register_all();//初始化解码器和复用器  
48.     avformat_alloc_output_context2(&ofmt_ctx, NULL, NULL, out_filename);//初始化一个用于输出的AVFormatContext结构体，视频帧率和宽高在此函数里面设置  
49.     if (!ofmt_ctx)  
50.     {  
51.         printf("Could not create output context\n");  
52.         return;  
53.     }  
54.   
55.     AVStream *out_stream = add_vidio_stream(ofmt_ctx, AV_CODEC_ID_MJPEG);//创造输出视频流  
56.     av_dump_format(ofmt_ctx, 0, out_filename, 1);//该函数会打印出视频流的信息，如果看着不开心可以不要  
57.   
58.     if (!(ofmt_ctx->oformat->flags & AVFMT_NOFILE))//打开输出视频文件  
59.     {  
60.         ret = avio_open(&ofmt_ctx->pb, out_filename, AVIO_FLAG_WRITE);  
61.         if (ret < 0) {  
62.             printf("Could not open output file '%s'", out_filename);  
63.             return;  
64.         }  
65.     }  
66.   
67.     if (avformat_write_header(ofmt_ctx, NULL) < 0)//写文件头（Write file header）  
68.     {  
69.         printf("Error occurred when opening output file\n");  
70.         return;  
71.     }  
72.   
73.     int frame_index = 0;//放入视频的图像计数  
74.     unsigned char *mydata = new unsigned char[DATASIZE];  
75.     AVPacket pkt;  
76.     av_init_packet(&pkt);  
77.     pkt.flags |= AV_PKT_FLAG_KEY;  
78.     pkt.stream_index = out_stream->index;//获取视频信息，为压入帧图像做准备  
79.     while (frame_index<100)//将图像压入视频中  
80.     {  
81.         FILE *file;//打开一张jpeg图像并读取其数据，在这里图像最大为1M,如果超过1M，则需要修改1024*1024这里  
82.         fopen_s(&file, "1.jpg", "rb");  
83.         pkt.size = fread(mydata, 1, DATASIZE, file);  
84.         pkt.data = mydata;  
85.         fclose(file);  
86.         if (av_interleaved_write_frame(ofmt_ctx, &pkt) < 0) //写入图像到视频  
87.         {  
88.             printf("Error muxing packet\n");  
89.             break;  
90.         }  
91.         printf("Write %8d frames to output file\n", frame_index);//打印出当前压入的帧数  
92.         frame_index++;  
93.     }  
94.     av_free_packet(&pkt);//释放掉帧数据包对象  
95.     av_write_trailer(ofmt_ctx);//写文件尾（Write file trailer）  
96.     delete[]mydata;//释放数据对象  
97.     if (ofmt_ctx && !(ofmt_ctx->oformat->flags & AVFMT_NOFILE))  
98.         avio_close(ofmt_ctx->pb);//关闭视频文件  
99.     avformat_free_context(ofmt_ctx);//释放输出视频相关数据结构  
100.     return;  
101. }