把jpeg图片转换为任意视频格式

在一些嵌入式系统中,视频以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.}


你可能感兴趣的:(把jpeg图片转换为任意视频格式)