Ffmpeg视频开发教程(四)——最新版FFMPEG开发包实现250张图片编码为h264文件保存

本文实现将250张图片编码为h264视频文件保存。使用的图片文件名字是1.jpg  2.jpg   3.jpg  .........以此类推。图片文件可以自己生成，也可以从我上传的位置下载：https://download.csdn.net/download/zhangamxqun/10438476    。注意image文件夹调试的话放.vcxproj同级目录，单独运行放exe同级目录，否则找不到图片。

网上类似功能的文章很多，本文的特点主要在于，本文是基于最新版FFMPEG实现的（ffmpeg-20180508），本文的方法也不同于网上的大多数方法，本文的方法可以方便的修改为直接从内存输入数据，而不一定要一个一个文件的读取。本文的方法输入的就是jpeg数据，可以分文件输入，其实也可以连续输入，输出是h264的数据包，非常方便。

本文程序的处理流程：

本文程序的环境搭建参考我的第一篇FFMPEG教程：https://blog.csdn.net/zhangamxqun/article/details/80304494

所以下面就不再讲述环境搭建，直接上核心代码。

/** 
实现最新版FFMPEG把JPEG图像数据编码为264数据，作者自己测试正确可用 
作者：明天继续
使用的ffmpeg版本：ffmpeg-20180508-293a6e8-win32 
开发工具：vs2012 
**/  
  
#include "stdafx.h"  
#include <Windows.h>
#include <wingdi.h>
#include <stdio.h>
#include <stdlib.h>
#include <string>
extern "C" {  
#include <libavcodec/avcodec.h>  
#include <libavformat/avformat.h>  
#include <libswscale/swscale.h>
#include <libavutil/opt.h>
#include <libavutil/imgutils.h>
 
}  
  
#define INBUF_SIZE 4096
#pragma comment(lib,"../../ffmpeg-20180508-293a6e8-win32-dev/lib/avcodec.lib")  
#pragma comment(lib,"../../ffmpeg-20180508-293a6e8-win32-dev/lib/avformat.lib")  
#pragma comment(lib,"../../ffmpeg-20180508-293a6e8-win32-dev/lib/avfilter.lib")  
#pragma comment(lib,"../../ffmpeg-20180508-293a6e8-win32-dev/lib/avutil.lib")  
#pragma comment(lib,"../../ffmpeg-20180508-293a6e8-win32-dev/lib/swscale.lib") 
  
#if _MSC_VER  
#define snprintf _snprintf_s  
#define PRIx64       "I64x"  
#define PRIX64       "I64X"  
#endif  



//编码frame 中的数据，pkt用于临时存储编码结果
static void encode_yuv_to_264(AVCodecContext *enc_ctx, AVFrame *frame, AVPacket *pkt,
				   FILE *outfile)
{
	int ret;

	/* send the frame to the encoder */

	ret = avcodec_send_frame(enc_ctx, frame);
	if (ret < 0) {
		fprintf(stderr, "Error sending a frame for encoding\n");
		exit(1);
	}

	while (ret >= 0) {
		ret = avcodec_receive_packet(enc_ctx, pkt);
		if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
			return;
		else if (ret < 0) {
			fprintf(stderr, "Error during encoding\n");
			exit(1);
		}

		fwrite(pkt->data, 1, pkt->size, outfile);
		av_packet_unref(pkt);
	}
}


//解码jpeg数据，并编码为h264保存到文件
static void decode_jpg_and_encode_264(AVCodecContext *dec_ctx, //jpeg解码器上下文
									  AVFrame *frameyuv, //存放中间解码获取的yuv数据
									  AVPacket *pktjpg,//输入数据中解析出的jpeg数据包
									  AVCodecContext *codech264,//h264的编码器上下文
									  AVPacket* packh264,//临时存储编码出的h264数据包
                                      FILE* file, //输出文件
									  int64_t pts)//时间戳
{

    int ret;

    ret = avcodec_send_packet(dec_ctx, pktjpg);
    if (ret < 0) {
        fprintf(stderr, "Error sending a packet for decoding\n");
        exit(1);
    }

    while (ret >= 0) {
        ret = avcodec_receive_frame(dec_ctx, frameyuv);//解码一帧数据
        if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
            return;
        else if (ret < 0) {
            fprintf(stderr, "Error during decoding\n");
            exit(1);
        }

        printf("saving frame %3d\n", dec_ctx->frame_number);
        fflush(stdout);
        /* 图片是解码器分配的内存，不需要释放 */        
		frameyuv->pts = pts;

		 encode_yuv_to_264(codech264, frameyuv, packh264, file);

    }
}


long getFileSize(const char* sFile)
{
	FILE* f = fopen(sFile, "rb");
	fseek(f,0,SEEK_END);
	long s = ftell(f);
	fclose(f);
	return s;
}

//测试把yuv编码为h264
void test_Jpeg_to_Yuv_To_H264()
{
	const char *out_filename,//保存的h24文件名字
		*codec_name;//编码器的名字，ffmpeg.exe查看所有支持的编码器名字：ffmpeg -encoders
    const AVCodec *codec_264;//264的编码器
    AVCodecContext *codec_context_264= NULL;//264编码器上下文
    int i, ret, x, y;
    FILE *file_264;//264文件指针
    AVFrame *frame_yuv;//保存yuv数据帧
    AVPacket *pkt_264,//临时保存编码出的264包数据
		*pkt_jpg;//临时保存jpeg包数据


	av_register_all();  

	//保存的h24文件名字
    out_filename = "out.h264";

	//编码为h264
    codec_name = "libx264";

    /* 找到h264的编码器 */
    codec_264 = avcodec_find_encoder_by_name(codec_name);
    if (!codec_264) {
        fprintf(stderr, "Codec '%s' not found\n", codec_name);
        exit(1);
    }

	//生成h264的编码器上下文
    codec_context_264 = avcodec_alloc_context3(codec_264);
    if (!codec_context_264) {
        fprintf(stderr, "Could not allocate video codec context\n");
        exit(1);
    }

	//分配264包的临时存储空间
    pkt_264 = av_packet_alloc();
    if (!pkt_264)
        exit(1);

	//分配jpeg包的临时存储
	pkt_jpg = av_packet_alloc();

    /* 设置比特率 */
    codec_context_264->bit_rate = 400000;
    /* 设置需要编码的jpeg文件的分辨率，注意此处的分辨率应该和你要使用的jpeg数据的分辨率一致*/
    codec_context_264->width = 160;
    codec_context_264->height = 128;
    /* 设置帧率 每秒25帧*/
    codec_context_264->time_base.num = 1;
	codec_context_264->time_base.den = 25;
    codec_context_264->framerate.num = 25;
	codec_context_264->framerate.den = 1;


    codec_context_264->gop_size = 10;
    codec_context_264->max_b_frames = 1;
    codec_context_264->pix_fmt = AV_PIX_FMT_YUV420P;//设置编码器像素格式

    if (codec_264->id == AV_CODEC_ID_H264)
        av_opt_set(codec_context_264->priv_data, "preset", "slow", 0);

    /* 打开解码器 */
    ret = avcodec_open2(codec_context_264, codec_264, NULL);
    if (ret < 0) {
        fprintf(stderr, "Could not open codec\n");
        exit(1);
    }

	//打开要输出的264文件
    file_264 = fopen(out_filename, "wb");
    if (!file_264) {
        fprintf(stderr, "Could not open %s\n", out_filename);
        exit(1);
    }

	//分配保存yuv帧的临时存储
    frame_yuv = av_frame_alloc();
    if (!frame_yuv) {
        fprintf(stderr, "Could not allocate video frame\n");
        exit(1);
    }

	jpeg解码
	//jpeg解码器解析上下文
	AVCodecParserContext *parser_jpg;
	//jpeg解码器
	const AVCodec *codec_jpg;
	//jpeg的解码上下文
	AVCodecContext *codecContext_jpg= NULL;
	//查找jpeg的解码器
	codec_jpg = avcodec_find_decoder(AV_CODEC_ID_MJPEG);
	//初始化jpeg的解码器解析上下文
	parser_jpg = av_parser_init(codec_jpg->id);
	//jpeg的解码器上下文
	codecContext_jpg = avcodec_alloc_context3(codec_jpg);
	//打开jpeg解码器
	if (avcodec_open2(codecContext_jpg, codec_jpg, NULL) < 0) {
		fprintf(stderr, "Could not open codec\n");
		exit(1);
	}
    /* 循环编码250帧图片，帧率为25，所以生成的视频长度约为10秒，生成的h264文件可以用vlc软件播放 */
    for (i = 1; i <= 250; i++) {
        fflush(stdout);
		//读取jpeg图像数据
		std::string imageData;
		//250张图片放在程序路径下的image/文件夹下，命名规则是1.jpg  2.jpg  3.jpg  ...... 250.jpg
		char jpeName[256]={0};
		snprintf(jpeName, sizeof(jpeName), "image/%d.%s", i,"jpg");
		long filesize = getFileSize(jpeName);
		if (filesize > imageData.size())
		{
			imageData.resize(filesize);
		}		
		FILE* fileImage = fopen(jpeName, "rb");
		fread(&(imageData[0]),1,filesize,fileImage);
		fclose(fileImage);

		//把读取的jpeg数据解析为包		
		uint8_t *tempdata = (uint8_t *)(imageData.data());
		size_t   tempdata_size = filesize;
		while (tempdata_size > 0) {
			ret = av_parser_parse2(parser_jpg, codecContext_jpg, &pkt_jpg->data, &pkt_jpg->size,
				tempdata, tempdata_size, AV_NOPTS_VALUE, AV_NOPTS_VALUE, 0);
			if (ret < 0) {
				fprintf(stderr, "Error while parsing\n");
				exit(1);
			}
			tempdata      += ret;
			tempdata_size -= ret;

			if (pkt_jpg->size)
				decode_jpg_and_encode_264(codecContext_jpg, frame_yuv, pkt_jpg, codec_context_264,pkt_264,file_264,i);
		}
    }
	//处理最后可能残余的数据
    encode_yuv_to_264(codec_context_264, NULL, pkt_264, file_264);

    fclose(file_264);

    avcodec_free_context(&codec_context_264);
	av_frame_free(&frame_yuv);
    av_packet_free(&pkt_264);
	av_packet_free(&pkt_jpg);
	av_parser_close(parser_jpg);
	avcodec_free_context(&codecContext_jpg);

}

  
int _tmain(int argc, _TCHAR* argvec[])  
{  
   test_Jpeg_to_Yuv_To_H264();
    return 0;
}