在一些嵌入式系统中,视频以jpeg图像格式传出来,而且数据量非常大。这种情况下如果要对jpeg图像进行解码并且重新编码成264之类的格式再 放入视频容器中会消耗大量的CPU资源,若不是性能特别犀利的CPU是无法承受的,这个时候有一个折衷的办法,就是直接将jpeg连续保存起来(也就是保 存成mjpeg视频格式),然后直接放入视频容器中。
这个方面的代码实现网上资料比较少,可以说几乎没有,基本都是直接用FFmpeg的命令行来实现的,这一方法虽然可行但毕竟不太适合软件的集成,是故本人做了些研究,并将代码放出来供各位参考指正。
在贴出代码之前特别感谢雷霄骅老师的博客,其关于FFmpeg的博文提供了非常多的参考,如果对FFmpeg有深入了解需求的可以去看看,会有很多收获。
下面上代码:
环境为:VS2013 C++ 控制台程序(空项目)+ FFmpeg(配置相关可参考本人之前博文)
view sourceprint?
001.
#include <stdio.h>
002.
003.
extern
"C"
//包含C文件头
004.
{
005.
#include
"libavformat/avformat.h"
006.
};
007.
#define DATASIZE
1024
*
1024
008.
009.
AVStream *add_vidio_stream(AVFormatContext *oc,
enum
AVCodecID codec_id)
//用以初始化一个用于输出的AVFormatContext结构体
010.
{
011.
AVStream *st;
012.
AVCodec *codec;
013.
014.
st = avformat_new_stream(oc, NULL);
015.
if
(!st)
016.
{
017.
printf("Could not alloc stream
018.
");
019.
exit(
1
);
020.
}
021.
codec = avcodec_find_encoder(codec_id);
//查找mjpeg解码器
022.
if
(!codec)
023.
{
024.
printf("codec not found
025.
");
026.
exit(
1
);
027.
}
028.
avcodec_get_context_defaults3(st->codec, codec);
//申请AVStream->codec(AVCodecContext对象)空间并设置默认值(由avcodec_get_context_defaults3()设置
029.
030.
st->codec->bit_rate =
400000
;
//设置采样参数,即比特率
031.
st->codec->width =
1080
;
//设置视频宽高,这里跟图片的宽高保存一致即可
032.
st->codec->height =
1800
;
033.
st->codec->time_base.den =
10
;
//设置帧率
034.
st->codec->time_base.num =
1
;
035.
036.
st->codec->pix_fmt = PIX_FMT_YUV420P;
//设置像素格式
037.
st->codec->codec_tag =
0
;
038.
if
(oc->oformat->flags & AVFMT_GLOBALHEADER)
//一些格式需要视频流数据头分开
039.
st->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;
040.
return
st;
041.
}
042.
043.
void
main()
044.
{
045.
AVFormatContext *ofmt_ctx = NULL;
//其包含码流参数较多,是一个贯穿始终的数据结构,很多函数都要用到它作为参数
046.
const
char
*out_filename =
"out.mkv"
;
//输出文件路径,在这里也可以将mkv改成别的ffmpeg支持的格式,如mp4,flv,avi之类的
047.
int
ret;
//返回标志
048.
049.
av_register_all();
//初始化解码器和复用器
050.
avformat_alloc_output_context2(&ofmt_ctx, NULL, NULL, out_filename);
//初始化一个用于输出的AVFormatContext结构体,视频帧率和宽高在此函数里面设置
051.
if
(!ofmt_ctx)
052.
{
053.
printf("Could not create output context
054.
");
055.
return
;
056.
}
057.
058.
AVStream *out_stream = add_vidio_stream(ofmt_ctx, AV_CODEC_ID_MJPEG);
//创造输出视频流
059.
av_dump_format(ofmt_ctx,
0
, out_filename,
1
);
//该函数会打印出视频流的信息,如果看着不开心可以不要
060.
061.
if
(!(ofmt_ctx->oformat->flags & AVFMT_NOFILE))
//打开输出视频文件
062.
{
063.
ret = avio_open(&ofmt_ctx->pb, out_filename, AVIO_FLAG_WRITE);
064.
if
(ret <
0
) {
065.
printf(
"Could not open output file '%s'"
, out_filename);
066.
return
;
067.
}
068.
}
069.
070.
if
(avformat_write_header(ofmt_ctx, NULL) <
0
)
//写文件头(Write file header)
071.
{
072.
printf("Error occurred when opening output file
073.
");
074.
return
;
075.
}
076.
077.
int
frame_index =
0
;
//放入视频的图像计数
078.
unsigned
char
*mydata =
new
unsigned
char
[DATASIZE];
079.
AVPacket pkt;
080.
av_init_packet(&pkt);
081.
pkt.flags |= AV_PKT_FLAG_KEY;
082.
pkt.stream_index = out_stream->index;
//获取视频信息,为压入帧图像做准备
083.
while
(frame_index<
100
)
//将图像压入视频中
084.
{
085.
FILE *file;
//打开一张jpeg图像并读取其数据,在这里图像最大为1M,如果超过1M,则需要修改1024*1024这里
086.
fopen_s(&file,
"1.jpg"
,
"rb"
);
087.
pkt.size = fread(mydata,
1
, DATASIZE, file);
088.
pkt.data = mydata;
089.
fclose(file);
090.
if
(av_interleaved_write_frame(ofmt_ctx, &pkt) <
0
)
//写入图像到视频
091.
{
092.
printf("Error muxing packet
093.
");
094.
break
;
095.
}
096.
printf("Write %8d frames to output file
097.
", frame_index);
//打印出当前压入的帧数
098.
frame_index++;
099.
}
100.
av_free_packet(&pkt);
//释放掉帧数据包对象
101.
av_write_trailer(ofmt_ctx);
//写文件尾(Write file trailer)
102.
delete[]mydata;
//释放数据对象
103.
if
(ofmt_ctx && !(ofmt_ctx->oformat->flags & AVFMT_NOFILE))
104.
avio_close(ofmt_ctx->pb);
//关闭视频文件
105.
avformat_free_context(ofmt_ctx);
//释放输出视频相关数据结构
106.
return
;
107.
}