YUV编码为JPEG

将大雷神的代码改了一下适合最新版本API并能够完美运行

/** 
 * 最简单的基于FFmpeg的图像编码器 
 * Simplest FFmpeg Picture Encoder 
 *  
 * 雷霄骅 Lei Xiaohua 
 * [email protected] 
 * 中国传媒大学/数字电视技术 
 * Communication University of China / Digital TV Technology 
 * http://blog.csdn.net/leixiaohua1020 
 *  
 * 本程序实现了YUV420P像素数据编码为JPEG图片。是最简单的FFmpeg编码方面的教程。 
 * 通过学习本例子可以了解FFmpeg的编码流程。 
 */  
  
#include <stdio.h>  

#define __STDC_CONSTANT_MACROS  
  
#ifdef _WIN32  
//Windows  
extern "C"  
{  
#include "libavcodec/avcodec.h"  
#include "libavformat/avformat.h"  
#include "libavutil/imgutils.h" 
};  
#else  
//Linux...  
#ifdef __cplusplus  
extern "C"  
{  
#endif  
#include <libavcodec/avcodec.h>  
#include <libavformat/avformat.h>  
#ifdef __cplusplus  
};  
#endif  
#endif
static int MyWriteJPEG(AVFrame* pFrame, int width, int height, int iIndex)  
{  
    // 输出文件路径  
    #define MAX_PATH 32
    char out_file[MAX_PATH] = {0};  
    sprintf_s(out_file, sizeof(out_file), "%d.jpg", iIndex);   
 av_register_all();
    // 分配AVFormatContext对象  
    AVFormatContext* pFormatCtx = avformat_alloc_context();  
      
    // 设置输出文件格式  
    pFormatCtx->oformat = av_guess_format("mjpeg", NULL, NULL);

    // 创建并初始化一个和该url相关的AVIOContext  
    if( avio_open(&pFormatCtx->pb, out_file, AVIO_FLAG_READ_WRITE) < 0) {  
        printf("Couldn't open output file.");  
        return -1;  
    }  
      
    // 构建一个新stream  
    AVStream* pAVStream = avformat_new_stream(pFormatCtx, 0);  
    if( pAVStream == NULL ) {  
        return -1;  
    }  
      
    // 设置该stream的信息  
    //AVCodecContext* pCodecCtx = pAVStream->codec; 
	// Get a pointer to the codec context for the video stream  
    // 得到视频流编码的上下文指针  
    AVCodecContext* pCodecCtx = avcodec_alloc_context3(NULL);  
    if (pCodecCtx == NULL)  
    {  
        printf("Could not allocate AVCodecContext\n");  
        return -1;  
    }  
    avcodec_parameters_to_context(pCodecCtx, pAVStream->codecpar); 

      
    pCodecCtx->codec_id = pFormatCtx->oformat->video_codec;  
    pCodecCtx->codec_type = AVMEDIA_TYPE_VIDEO;  
    pCodecCtx->pix_fmt = AV_PIX_FMT_YUVJ420P;  
    pCodecCtx->width = width;  
    pCodecCtx->height = height;  
    pCodecCtx->time_base.num = 1;  
    pCodecCtx->time_base.den = 25;  
      
    // Begin Output some information  
    av_dump_format(pFormatCtx, 0, out_file, 1);  
    // End Output some information  
      
    // 查找解码器  
    AVCodec* pCodec = avcodec_find_encoder(pCodecCtx->codec_id);  
    if( !pCodec ) {  
        printf("Codec not found.");  
        return -1;  
    }  
    // 设置pCodecCtx的解码器为pCodec  
    if( avcodec_open2(pCodecCtx, pCodec, NULL) < 0 ) {  
        printf("Could not open codec.");  
        return -1;  
    }  

    //Write Header  
    avformat_write_header(pFormatCtx, NULL);  
      
    int y_size = pCodecCtx->width * pCodecCtx->height;  
      
    //Encode  
    // 给AVPacket分配足够大的空间  
    AVPacket pkt;  
    av_new_packet(&pkt, y_size * 3);  
      
    //   
    int got_picture = 0;  
    //int ret = avcodec_encode_video2(pCodecCtx, &pkt, pFrame, &got_picture);  
	//编码该帧  
    //avcodec_send_packet(pCodecCtx, &pkt);
	//int ret = avcodec_receive_frame(pCodecCtx, pFrame); 
	avcodec_send_frame(pCodecCtx, pFrame);
	int ret = avcodec_receive_packet(pCodecCtx, &pkt);
    
    //if( got_picture == 1 ) {  
        //pkt.stream_index = pAVStream->index;  
        ret = av_write_frame(pFormatCtx, &pkt);  
    //}  
	av_packet_unref(&pkt);   
  
    //Write Trailer  
    av_write_trailer(pFormatCtx);  
  
    printf("Encode Successful.\n");  
  
    if( pAVStream ) {  
        avcodec_close(pCodecCtx);
		av_free(pFrame);
    }  
    avio_close(pFormatCtx->pb);  
    avformat_free_context(pFormatCtx);  
      
    return 0;  
}
int main(int argc, char* argv[])  
{  
	/*AVFormatContext* pFormatCtx = NULL;

	const char* out_file = "cuc_view_encode.jpg";    //Output file  

    if (avio_open(&pFormatCtx->pb,out_file, AVIO_FLAG_READ_WRITE) < 0){  
        printf("Couldn't open output file.");  
        return -1;  
    }*/
	/*FILE *in_file = fopen("cuc_view_480x272.yuv", "rb");	//视频YUV源文件 
	//读入YUV
	AVFrame* picture=NULL;
	uint8_t* picture_buf1=new uint8_t[1024];

	int y_size = 480*272;
	if (fread(picture_buf1, 1, y_size*3/2, in_file) < 0)
	{
		printf("文件读取错误");
		return -1;
	}
	convertYUVToJPG(picture_buf1);*/

	AVFormatContext* pFormatCtx=NULL;
	AVOutputFormat* fmt=NULL;
	AVStream* video_st=NULL;
	AVCodecContext* pCodecCtx=NULL;
	AVCodec* pCodec=NULL;

	uint8_t* picture_buf=NULL;
	AVFrame* picture=NULL;
	int size;

	FILE *in_file = fopen("cuc_view_480x272.yuv", "rb");	//视频YUV源文件 
	//FILE *in_file = fopen("G:\\foreman_cif.h264", "rb");	//视频YUV源文件 
	int in_w=480,in_h=272;									//宽高
	const char* out_file = "cuc_view_encode.jpg";					//输出文件路径

	av_register_all();

	//方法1.组合使用几个函数
	pFormatCtx = avformat_alloc_context();
	//猜格式。用MJPEG编码
	fmt = av_guess_format("mjpeg", NULL, NULL);
	pFormatCtx->oformat = fmt;
	//注意：输出路径
	if (avio_open(&pFormatCtx->pb,out_file, AVIO_FLAG_READ_WRITE) < 0)
	{
		printf("输出文件打开失败");
		return -1;
	}

	//方法2.更加自动化一些
	//avformat_alloc_output_context2(&pFormatCtx, NULL, NULL, out_file);
	//fmt = pFormatCtx->oformat;

	video_st = avformat_new_stream(pFormatCtx, 0);
	if (video_st==NULL)
	{
		return -1;
	}
	//pCodecCtx = video_st->codec;
	pCodecCtx = avcodec_alloc_context3(NULL);  
    if (pCodecCtx == NULL)  
    {  
        printf("Could not allocate AVCodecContext\n");  
        return -1;  
    }  
    avcodec_parameters_to_context(pCodecCtx, video_st->codecpar);

	pCodecCtx->codec_id = fmt->video_codec;
	pCodecCtx->codec_type = AVMEDIA_TYPE_VIDEO;
	pCodecCtx->pix_fmt = AV_PIX_FMT_YUVJ420P;

	pCodecCtx->width = in_w;  
	pCodecCtx->height = in_h;

	pCodecCtx->time_base.num = 1;  
	pCodecCtx->time_base.den = 25;   
	//输出格式信息
	av_dump_format(pFormatCtx, 0, out_file, 1);

	pCodec = avcodec_find_encoder(pCodecCtx->codec_id);
	if (!pCodec)
	{
		printf("没有找到合适的编码器！");
		return -1;
	}
	if (avcodec_open2(pCodecCtx, pCodec,NULL) < 0)
	{
		printf("编码器打开失败！");
		return -1;
	}
	picture = av_frame_alloc();
	size = av_image_get_buffer_size(pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height, 1);  
	picture_buf = (uint8_t *)av_malloc(size);
	if (!picture_buf)
	{
		return -1;
	}
	//avpicture_fill((AVPicture *)picture, picture_buf, pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height);
	av_image_fill_arrays(picture->data, picture->linesize, picture_buf, AV_PIX_FMT_YUVJ420P, pCodecCtx->width, pCodecCtx->height, 1);
	//写文件头
	avformat_write_header(pFormatCtx,NULL);

	AVPacket pkt;
	int y_size = pCodecCtx->width * pCodecCtx->height;
	av_new_packet(&pkt,y_size*3);
	//读入YUV
	if (fread(picture_buf, 1, y_size*3/2, in_file) < 0)
	{
		printf("文件读取错误");
		return -1;
	}
	MyWriteJPEG(picture, pCodecCtx->width, pCodecCtx->height, 1);  
	system("pause");
	return 0;
}