本文介绍一个简单的基于FFmpeg的转码器。它可以将一种视频格式(包括封转格式和编码格式)转换为另一种视频格式。转码器在视音频编解码处理的程序中,属于一个比较复杂的东西。因为它结合了视频的解码和编码。一个视频播放器,一般只包含解码功能;一个视频编码工具,一般只包含编码功能;而一个视频转码器,则需要先对视频进行解码,然后再对视频进行编码,因而相当于解码器和编码器的结合。下图例举了一个视频的转码流程。输入视频的封装格式是FLV,视频编码标准是H.264,音频编码标准是AAC;输出视频的封装格式是AVI,视频编码标准是MPEG2,音频编码标准是MP3。从流程中可以看出,首先从输入视频中分离出视频码流和音频压缩码流,然后分别将视频码流和音频码流进行解码,获取到非压缩的像素数据/音频采样数据,接着将非压缩的像素数据/音频采样数据重新进行编码,获得重新编码后的视频码流和音频码流,最后将视频码流和音频码流重新封装成一个文件。

 

最简单的基于FFMPEG的转码程序_视频 

本文介绍的视频转码器正是使用FFMPEG类库从编程的角度实现了上述流程。该例子是从FFmpeg的例子改编的,平台是VC2010,类库版本是2014.5.6。

流程图(2014.9.29更新)

下面附两张使用FFmpeg转码视频的流程图。图中使用浅绿色标出了视频的编码、解码函数。从代码中可以看出,使用了AVFilter的不少东西,因此建议先学习AVFilter的内容后再看这个转码器的源代码。

PS:实际上,转码器不是一定依赖AVFilter的。因此打算有时间对这个转码器进行进一步的简化,使学习的人无需AVFilter的基础也可以理解转码器。

最简单的基于FFMPEG的转码程序_音频_02

简单介绍一下流程中各个函数的意义:
open_input_file():打开输入文件,并初始化相关的结构体。
open_output_file():打开输出文件,并初始化相关的结构体。
init_filters():初始化AVFilter相关的结构体。
av_read_frame():从输入文件中读取一个AVPacket。
avcodec_decode_video2():解码一个视频AVPacket(存储H.264等压缩码流数据)为AVFrame(存储YUV等非压缩的像素数据)。
avcodec_decode_video4():解码一个音频AVPacket(存储MP3等压缩码流数据)为AVFrame(存储PCM采样数据)。
filter_encode_write_frame():编码一个AVFrame。
flush_encoder():输入文件读取完毕后,输出编码器中剩余的AVPacket。

以上函数中open_input_file(),open_output_file(),init_filters()中的函数在其他文章中都有所叙述,在这里不再重复:

open_input_file()可参考:100行代码实现最简单的基于FFMPEG+SDL的视频播放器(SDL1.x)

open_output_file()可参考:最简单的基于FFMPEG的视频编码器(YUV编码为H.264)

init_filters()可参考:最简单的基于FFmpeg的AVfilter例子(水印叠加)

在这里介绍一下其中编码的函数filter_encode_write_frame()。filter_encode_write_frame()函数的流程如下图所示,它完成了视频/音频的编码功能。

PS:视频和音频的编码流程中除了编码函数avcodec_encode_video2()和avcodec_encode_audio2()不一样之外,其他部分几乎完全一样。

最简单的基于FFMPEG的转码程序_vs2010_03

简单介绍一下filter_encode_write_frame()中各个函数的意义:

av_buffersrc_add_frame():将解码后的AVFrame加入Filtergraph。

av_buffersink_get_buffer_ref():从Filtergraph中取一个AVFrame。

avcodec_encode_video2():编码一个视频AVFrame为AVPacket。

avcodec_encode_audio2():编码一个音频AVFrame为AVPacket。

av_interleaved_write_frame():将编码后的AVPacket写入文件。

 

代码

贴上代码

 

[cpp]  view plain copy 
 
  1. /* 
  2.  *最简单的基于FFmpeg的转码器 
  3.  *Simplest FFmpeg Transcoder 
  4.  * 
  5.  *雷霄骅 Lei Xiaohua 
  6.  *leixiaohua1020@126.com 
  7.  *中国传媒大学/数字电视技术 
  8.  *Communication University of China / DigitalTV Technology
  9.  *
  10.  *本程序实现了视频格式之间的转换。是一个最简单的视频转码程序。 
  11.  * 
  12.  */  
  13.    
  14. #include "stdafx.h"  
  15. extern "C"  
  16. {  
  17. #include "libavcodec/avcodec.h"  
  18. #include "libavformat/avformat.h"  
  19. #include "libavfilter/avfiltergraph.h"  
  20. #include "libavfilter/avcodec.h"  
  21. #include "libavfilter/buffersink.h"  
  22. #include "libavfilter/buffersrc.h"  
  23. #include "libavutil/avutil.h"  
  24. #include "libavutil/opt.h"  
  25. #include "libavutil/pixdesc.h"  
  26. };  
  27.    
  28.    
  29.    
  30. static AVFormatContext *ifmt_ctx;  
  31. static AVFormatContext *ofmt_ctx;  
  32. typedef struct FilteringContext{  
  33.     AVFilterContext*buffersink_ctx;  
  34.     AVFilterContext*buffersrc_ctx;  
  35.     AVFilterGraph*filter_graph;  
  36. } FilteringContext;  
  37. static FilteringContext *filter_ctx;  
  38. static int open_input_file(const char *filename)  
  39. {  
  40.     int ret;  
  41.     unsigned int i;  
  42.     ifmt_ctx =NULL;  
  43.     if ((ret = avformat_open_input(&ifmt_ctx,filename, NULL, NULL)) < 0) {  
  44.        av_log(NULL, AV_LOG_ERROR, "Cannot openinput file\n");  
  45.         return ret;  
  46.     }  
  47.     if ((ret = avformat_find_stream_info(ifmt_ctx, NULL))< 0) {  
  48.        av_log(NULL, AV_LOG_ERROR, "Cannot findstream information\n");  
  49.         return ret;  
  50.     }  
  51.     for (i = 0; i < ifmt_ctx->nb_streams; i++) {  
  52.         AVStream*stream;  
  53.        AVCodecContext *codec_ctx;  
  54.         stream =ifmt_ctx->streams[i];  
  55.         codec_ctx =stream->codec;  
  56.         /* Reencode video & audio and remux subtitles etc. */  
  57.         if (codec_ctx->codec_type == AVMEDIA_TYPE_VIDEO  
  58.                 ||codec_ctx->codec_type == AVMEDIA_TYPE_AUDIO) {  
  59.             /* Open decoder */  
  60.             ret =avcodec_open2(codec_ctx,  
  61.                    avcodec_find_decoder(codec_ctx->codec_id), NULL);  
  62.             if (ret < 0) {  
  63.                av_log(NULL, AV_LOG_ERROR, "Failed toopen decoder for stream #%u\n", i);  
  64.                 return ret;  
  65.             }  
  66.         }  
  67.     }  
  68.    av_dump_format(ifmt_ctx, 0, filename, 0);  
  69.     return 0;  
  70. }  
  71. static int open_output_file(const char *filename)  
  72. {  
  73.     AVStream*out_stream;  
  74.     AVStream*in_stream;  
  75.     AVCodecContext*dec_ctx, *enc_ctx;  
  76.     AVCodec*encoder;  
  77.     int ret;  
  78.     unsigned int i;  
  79.     ofmt_ctx =NULL;  
  80.    avformat_alloc_output_context2(&ofmt_ctx, NULL, NULL, filename);  
  81.     if (!ofmt_ctx) {  
  82.        av_log(NULL, AV_LOG_ERROR, "Could notcreate output context\n");  
  83.         return AVERROR_UNKNOWN;  
  84.     }  
  85.     for (i = 0; i < ifmt_ctx->nb_streams; i++) {  
  86.         out_stream= avformat_new_stream(ofmt_ctx, NULL);  
  87.         if (!out_stream) {  
  88.            av_log(NULL, AV_LOG_ERROR, "Failedallocating output stream\n");  
  89.             return AVERROR_UNKNOWN;  
  90.         }  
  91.         in_stream =ifmt_ctx->streams[i];  
  92.         dec_ctx =in_stream->codec;  
  93.         enc_ctx =out_stream->codec;  
  94.         if (dec_ctx->codec_type == AVMEDIA_TYPE_VIDEO  
  95.                 ||dec_ctx->codec_type == AVMEDIA_TYPE_AUDIO) {  
  96.             /* in this example, we choose transcoding to same codec */  
  97.             encoder= avcodec_find_encoder(dec_ctx->codec_id);  
  98.             /* In this example, we transcode to same properties(picture size, 
  99.             * sample rate etc.). These properties can be changed for output 
  100.             * streams easily using filters */  
  101.             if (dec_ctx->codec_type == AVMEDIA_TYPE_VIDEO) {  
  102.                enc_ctx->height = dec_ctx->height;  
  103.                enc_ctx->width = dec_ctx->width;  
  104.                enc_ctx->sample_aspect_ratio = dec_ctx->sample_aspect_ratio;  
  105.                 /* take first format from list of supported formats */  
  106.                enc_ctx->pix_fmt = encoder->pix_fmts[0];  
  107.                 /* video time_base can be set to whatever is handy andsupported by encoder */  
  108.                enc_ctx->time_base = dec_ctx->time_base;  
  109.             } else {  
  110.                enc_ctx->sample_rate = dec_ctx->sample_rate;  
  111.                enc_ctx->channel_layout = dec_ctx->channel_layout;  
  112.                enc_ctx->channels = av_get_channel_layout_nb_channels(enc_ctx->channel_layout);  
  113.                 /* take first format from list of supported formats */  
  114.                enc_ctx->sample_fmt = encoder->sample_fmts[0];  
  115.                 AVRationaltime_base={1, enc_ctx->sample_rate};  
  116.                enc_ctx->time_base = time_base;  
  117.             }  
  118.             /* Third parameter can be used to pass settings to encoder*/  
  119.             ret =avcodec_open2(enc_ctx, encoder, NULL);  
  120.             if (ret < 0) {  
  121.                av_log(NULL, AV_LOG_ERROR, "Cannot openvideo encoder for stream #%u\n", i);  
  122.                 return ret;  
  123.             }  
  124.         } else if(dec_ctx->codec_type == AVMEDIA_TYPE_UNKNOWN) {  
  125.            av_log(NULL, AV_LOG_FATAL, "Elementarystream #%d is of unknown type, cannot proceed\n", i);  
  126.             return AVERROR_INVALIDDATA;  
  127.         } else {  
  128.             /* if this stream must be remuxed */  
  129.             ret =avcodec_copy_context(ofmt_ctx->streams[i]->codec,  
  130.                    ifmt_ctx->streams[i]->codec);  
  131.             if (ret < 0) {  
  132.                av_log(NULL, AV_LOG_ERROR, "Copyingstream context failed\n");  
  133.                 return ret;  
  134.             }  
  135.         }  
  136.         if (ofmt_ctx->oformat->flags &AVFMT_GLOBALHEADER)  
  137.            enc_ctx->flags |= CODEC_FLAG_GLOBAL_HEADER;  
  138.     }  
  139.    av_dump_format(ofmt_ctx, 0, filename, 1);  
  140.     if (!(ofmt_ctx->oformat->flags &AVFMT_NOFILE)) {  
  141.         ret =avio_open(&ofmt_ctx->pb, filename, AVIO_FLAG_WRITE);  
  142.         if (ret < 0) {  
  143.            av_log(NULL, AV_LOG_ERROR, "Could notopen output file '%s'", filename);  
  144.             return ret;  
  145.         }  
  146.     }  
  147.     /* init muxer, write output file header */  
  148.     ret =avformat_write_header(ofmt_ctx, NULL);  
  149.     if (ret < 0) {  
  150.         av_log(NULL,AV_LOG_ERROR, "Error occurred when openingoutput file\n");  
  151.         return ret;  
  152.     }  
  153.     return 0;  
  154. }  
  155. static intinit_filter(FilteringContext* fctx, AVCodecContext *dec_ctx,  
  156.        AVCodecContext *enc_ctx, const char *filter_spec)  
  157. {  
  158.     char args[512];  
  159.     int ret = 0;  
  160.     AVFilter*buffersrc = NULL;  
  161.     AVFilter*buffersink = NULL;  
  162.     AVFilterContext*buffersrc_ctx = NULL;  
  163.     AVFilterContext*buffersink_ctx = NULL;  
  164.     AVFilterInOut*outputs = avfilter_inout_alloc();  
  165.     AVFilterInOut*inputs  = avfilter_inout_alloc();  
  166.     AVFilterGraph*filter_graph = avfilter_graph_alloc();  
  167.     if (!outputs || !inputs || !filter_graph) {  
  168.         ret =AVERROR(ENOMEM);  
  169.         goto end;  
  170.     }  
  171.     if (dec_ctx->codec_type == AVMEDIA_TYPE_VIDEO) {  
  172.         buffersrc =avfilter_get_by_name("buffer");  
  173.         buffersink= avfilter_get_by_name("buffersink");  
  174.         if (!buffersrc || !buffersink) {  
  175.            av_log(NULL, AV_LOG_ERROR, "filteringsource or sink element not found\n");  
  176.             ret = AVERROR_UNKNOWN;  
  177.             goto end;  
  178.         }  
  179.        _snprintf(args, sizeof(args),  
  180.                 "video_size=%dx%d:pix_fmt=%d:time_base=%d/%d:pixel_aspect=%d/%d",  
  181.                dec_ctx->width, dec_ctx->height, dec_ctx->pix_fmt,  
  182.                 dec_ctx->time_base.num,dec_ctx->time_base.den,  
  183.                dec_ctx->sample_aspect_ratio.num,  
  184.                dec_ctx->sample_aspect_ratio.den);  
  185.         ret =avfilter_graph_create_filter(&buffersrc_ctx, buffersrc, "in",  
  186.                args, NULL, filter_graph);  
  187.         if (ret < 0) {  
  188.            av_log(NULL, AV_LOG_ERROR, "Cannotcreate buffer source\n");  
  189.             goto end;  
  190.         }  
  191.         ret =avfilter_graph_create_filter(&buffersink_ctx, buffersink, "out",  
  192.                NULL, NULL, filter_graph);  
  193.         if (ret < 0) {  
  194.            av_log(NULL, AV_LOG_ERROR, "Cannotcreate buffer sink\n");  
  195.             goto end;  
  196.         }  
  197.         ret =av_opt_set_bin(buffersink_ctx, "pix_fmts",  
  198.                (uint8_t*)&enc_ctx->pix_fmt, sizeof(enc_ctx->pix_fmt),  
  199.                AV_OPT_SEARCH_CHILDREN);  
  200.         if (ret < 0) {  
  201.            av_log(NULL, AV_LOG_ERROR, "Cannot setoutput pixel format\n");  
  202.             goto end;  
  203.         }  
  204.     } else if(dec_ctx->codec_type == AVMEDIA_TYPE_AUDIO) {  
  205.         buffersrc = avfilter_get_by_name("abuffer");  
  206.         buffersink= avfilter_get_by_name("abuffersink");  
  207.         if (!buffersrc || !buffersink) {  
  208.            av_log(NULL, AV_LOG_ERROR, "filteringsource or sink element not found\n");  
  209.             ret =AVERROR_UNKNOWN;  
  210.             goto end;  
  211.         }  
  212.         if (!dec_ctx->channel_layout)  
  213.            dec_ctx->channel_layout =  
  214.                av_get_default_channel_layout(dec_ctx->channels);  
  215.        _snprintf(args, sizeof(args),  
  216.                 "time_base=%d/%d:sample_rate=%d:sample_fmt=%s:channel_layout=0x%I64x",  
  217.                dec_ctx->time_base.num, dec_ctx->time_base.den,dec_ctx->sample_rate,  
  218.                av_get_sample_fmt_name(dec_ctx->sample_fmt),  
  219.                dec_ctx->channel_layout);  
  220.         ret =avfilter_graph_create_filter(&buffersrc_ctx, buffersrc, "in",  
  221.                args, NULL, filter_graph);  
  222.         if (ret < 0) {  
  223.            av_log(NULL, AV_LOG_ERROR, "Cannotcreate audio buffer source\n");  
  224.             goto end;  
  225.         }  
  226.         ret =avfilter_graph_create_filter(&buffersink_ctx, buffersink, "out",  
  227.                NULL, NULL, filter_graph);  
  228.         if (ret < 0) {  
  229.            av_log(NULL, AV_LOG_ERROR, "Cannotcreate audio buffer sink\n");  
  230.             goto end;  
  231.         }  
  232.         ret = av_opt_set_bin(buffersink_ctx, "sample_fmts",  
  233.                (uint8_t*)&enc_ctx->sample_fmt, sizeof(enc_ctx->sample_fmt),  
  234.                AV_OPT_SEARCH_CHILDREN);  
  235.         if (ret < 0) {  
  236.            av_log(NULL, AV_LOG_ERROR, "Cannot setoutput sample format\n");  
  237.             goto end;  
  238.         }  
  239.         ret =av_opt_set_bin(buffersink_ctx, "channel_layouts",  
  240.                (uint8_t*)&enc_ctx->channel_layout,  
  241.                 sizeof(enc_ctx->channel_layout),AV_OPT_SEARCH_CHILDREN);  
  242.         if (ret < 0) {  
  243.            av_log(NULL, AV_LOG_ERROR, "Cannot setoutput channel layout\n");  
  244.             goto end;  
  245.         }  
  246.         ret =av_opt_set_bin(buffersink_ctx, "sample_rates",  
  247.                (uint8_t*)&enc_ctx->sample_rate, sizeof(enc_ctx->sample_rate),  
  248.                AV_OPT_SEARCH_CHILDREN);  
  249.         if (ret < 0) {  
  250.            av_log(NULL, AV_LOG_ERROR, "Cannot setoutput sample rate\n");  
  251.             goto end;  
  252.         }  
  253.     } else {  
  254.         ret =AVERROR_UNKNOWN;  
  255.         goto end;  
  256.     }  
  257.     /* Endpoints for the filter graph. */  
  258.    outputs->name       =av_strdup("in");  
  259.    outputs->filter_ctx = buffersrc_ctx;  
  260.    outputs->pad_idx    = 0;  
  261.    outputs->next       = NULL;  
  262.    inputs->name       = av_strdup("out");  
  263.    inputs->filter_ctx = buffersink_ctx;  
  264.    inputs->pad_idx    = 0;  
  265.    inputs->next       = NULL;  
  266.     if (!outputs->name || !inputs->name) {  
  267.         ret =AVERROR(ENOMEM);  
  268.         goto end;  
  269.     }  
  270.     if ((ret = avfilter_graph_parse_ptr(filter_graph,filter_spec,  
  271.                    &inputs, &outputs, NULL)) < 0)  
  272.         goto end;  
  273.     if ((ret = avfilter_graph_config(filter_graph, NULL))< 0)  
  274.         goto end;  
  275.     /* Fill FilteringContext */  
  276.    fctx->buffersrc_ctx = buffersrc_ctx;  
  277.    fctx->buffersink_ctx = buffersink_ctx;  
  278.     fctx->filter_graph= filter_graph;  
  279. end:  
  280.    avfilter_inout_free(&inputs);  
  281.    avfilter_inout_free(&outputs);  
  282.     return ret;  
  283. }  
  284. static int init_filters(void)  
  285. {  
  286.     const char*filter_spec;  
  287.     unsigned int i;  
  288.     int ret;  
  289.     filter_ctx =(FilteringContext *)av_malloc_array(ifmt_ctx->nb_streams, sizeof(*filter_ctx));  
  290.     if (!filter_ctx)  
  291.         return AVERROR(ENOMEM);  
  292.     for (i = 0; i < ifmt_ctx->nb_streams; i++) {  
  293.        filter_ctx[i].buffersrc_ctx  =NULL;  
  294.         filter_ctx[i].buffersink_ctx= NULL;  
  295.        filter_ctx[i].filter_graph   =NULL;  
  296.         if(!(ifmt_ctx->streams[i]->codec->codec_type == AVMEDIA_TYPE_AUDIO  
  297.                 ||ifmt_ctx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO))  
  298.             continue;  
  299.         if (ifmt_ctx->streams[i]->codec->codec_type== AVMEDIA_TYPE_VIDEO)  
  300.            filter_spec = "null"; /* passthrough (dummy) filter for video */  
  301.         else  
  302.            filter_spec = "anull"; /* passthrough (dummy) filter for audio */  
  303.         ret =init_filter(&filter_ctx[i], ifmt_ctx->streams[i]->codec,  
  304.                ofmt_ctx->streams[i]->codec, filter_spec);  
  305.         if (ret)  
  306.             return ret;  
  307.     }  
  308.     return 0;  
  309. }  
  310. static intencode_write_frame(AVFrame *filt_frame, unsignedint stream_index, int*got_frame) {  
  311.     int ret;  
  312.     int got_frame_local;  
  313.     AVPacketenc_pkt;  
  314.     int (*enc_func)(AVCodecContext *, AVPacket *, const AVFrame *, int*) =  
  315.        (ifmt_ctx->streams[stream_index]->codec->codec_type ==  
  316.         AVMEDIA_TYPE_VIDEO) ? avcodec_encode_video2 : avcodec_encode_audio2;  
  317.     if (!got_frame)  
  318.         got_frame =&got_frame_local;  
  319.     av_log(NULL,AV_LOG_INFO, "Encoding frame\n");  
  320.     /* encode filtered frame */  
  321.     enc_pkt.data =NULL;  
  322.     enc_pkt.size =0;  
  323.     av_init_packet(&enc_pkt);  
  324.     ret =enc_func(ofmt_ctx->streams[stream_index]->codec, &enc_pkt,  
  325.            filt_frame, got_frame);  
  326.    av_frame_free(&filt_frame);  
  327.     if (ret < 0)  
  328.         return ret;  
  329.     if (!(*got_frame))  
  330.         return 0;  
  331.     /* prepare packet for muxing */  
  332.    enc_pkt.stream_index = stream_index;  
  333.     enc_pkt.dts =av_rescale_q_rnd(enc_pkt.dts,  
  334.            ofmt_ctx->streams[stream_index]->codec->time_base,  
  335.            ofmt_ctx->streams[stream_index]->time_base,  
  336.            (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));  
  337.     enc_pkt.pts =av_rescale_q_rnd(enc_pkt.pts,  
  338.            ofmt_ctx->streams[stream_index]->codec->time_base,  
  339.            ofmt_ctx->streams[stream_index]->time_base,  
  340.            (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));  
  341.    enc_pkt.duration = av_rescale_q(enc_pkt.duration,  
  342.            ofmt_ctx->streams[stream_index]->codec->time_base,  
  343.            ofmt_ctx->streams[stream_index]->time_base);  
  344.     av_log(NULL,AV_LOG_DEBUG, "Muxing frame\n");  
  345.     /* mux encoded frame */  
  346.     ret =av_interleaved_write_frame(ofmt_ctx, &enc_pkt);  
  347.     return ret;  
  348. }  
  349. static intfilter_encode_write_frame(AVFrame *frame, unsignedint stream_index)  
  350. {  
  351.     int ret;  
  352.     AVFrame*filt_frame;  
  353.     av_log(NULL,AV_LOG_INFO, "Pushing decoded frame tofilters\n");  
  354.     /* push the decoded frame into the filtergraph */  
  355.     ret =av_buffersrc_add_frame_flags(filter_ctx[stream_index].buffersrc_ctx,  
  356.             frame,0);  
  357.     if (ret < 0) {  
  358.        av_log(NULL, AV_LOG_ERROR, "Error whilefeeding the filtergraph\n");  
  359.         return ret;  
  360.     }  
  361.     /* pull filtered frames from the filtergraph */  
  362.     while (1) {  
  363.         filt_frame= av_frame_alloc();  
  364.         if (!filt_frame) {  
  365.             ret =AVERROR(ENOMEM);  
  366.             break;  
  367.         }  
  368.        av_log(NULL, AV_LOG_INFO, "Pullingfiltered frame from filters\n");  
  369.         ret =av_buffersink_get_frame(filter_ctx[stream_index].buffersink_ctx,  
  370.                filt_frame);  
  371.         if (ret < 0) {  
  372.             /* if nomore frames for output - returns AVERROR(EAGAIN) 
  373.             * if flushed and no more frames for output - returns AVERROR_EOF 
  374.             * rewrite retcode to 0 to show it as normal procedure completion 
  375.             */  
  376.             if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)  
  377.                 ret= 0;  
  378.            av_frame_free(&filt_frame);  
  379.             break;  
  380.         }  
  381.        filt_frame->pict_type = AV_PICTURE_TYPE_NONE;  
  382.         ret =encode_write_frame(filt_frame, stream_index, NULL);  
  383.         if (ret < 0)  
  384.             break;  
  385.     }  
  386.     return ret;  
  387. }  
  388. static int flush_encoder(unsigned intstream_index)  
  389. {  
  390.     int ret;  
  391.     int got_frame;  
  392.     if(!(ofmt_ctx->streams[stream_index]->codec->codec->capabilities&  
  393.                CODEC_CAP_DELAY))  
  394.         return 0;  
  395.     while (1) {  
  396.        av_log(NULL, AV_LOG_INFO, "Flushingstream #%u encoder\n", stream_index);  
  397.         ret =encode_write_frame(NULL, stream_index, &got_frame);  
  398.         if (ret < 0)  
  399.             break;  
  400.         if (!got_frame)  
  401.             return 0;  
  402.     }  
  403.     return ret;  
  404. }  
  405.    
  406. int_tmain(int argc, _TCHAR* argv[])  
  407. {  
  408.     int ret;  
  409.     AVPacketpacket;  
  410.     AVFrame *frame= NULL;  
  411.     enum AVMediaType type;  
  412.     unsigned intstream_index;  
  413.     unsigned int i;  
  414.     int got_frame;  
  415.     int (*dec_func)(AVCodecContext *, AVFrame *, int *, const AVPacket*);  
  416.     if (argc != 3) {  
  417.        av_log(NULL, AV_LOG_ERROR, "Usage: %s<input file> <output file>\n", argv[0]);  
  418.         return 1;  
  419.     }  
  420.    av_register_all();  
  421.    avfilter_register_all();  
  422.     if ((ret = open_input_file(argv[1])) < 0)  
  423.         goto end;  
  424.     if ((ret = open_output_file(argv[2])) < 0)  
  425.         goto end;  
  426.     if ((ret = init_filters()) < 0)  
  427.         goto end;  
  428.     /* read all packets */  
  429.     while (1) {  
  430.         if ((ret= av_read_frame(ifmt_ctx, &packet)) < 0)  
  431.             break;  
  432.        stream_index = packet.stream_index;  
  433.         type =ifmt_ctx->streams[packet.stream_index]->codec->codec_type;  
  434.        av_log(NULL, AV_LOG_DEBUG, "Demuxergave frame of stream_index %u\n",  
  435.                stream_index);  
  436.         if (filter_ctx[stream_index].filter_graph) {  
  437.            av_log(NULL, AV_LOG_DEBUG, "Going toreencode&filter the frame\n");  
  438.             frame =av_frame_alloc();  
  439.             if (!frame) {  
  440.                 ret = AVERROR(ENOMEM);  
  441.                 break;  
  442.             }  
  443.            packet.dts = av_rescale_q_rnd(packet.dts,  
  444.                    ifmt_ctx->streams[stream_index]->time_base,  
  445.                    ifmt_ctx->streams[stream_index]->codec->time_base,  
  446.                     (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));  
  447.            packet.pts = av_rescale_q_rnd(packet.pts,  
  448.                    ifmt_ctx->streams[stream_index]->time_base,  
  449.                    ifmt_ctx->streams[stream_index]->codec->time_base,  
  450.                    (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));  
  451.            dec_func = (type == AVMEDIA_TYPE_VIDEO) ? avcodec_decode_video2 :  
  452.                avcodec_decode_audio4;  
  453.             ret =dec_func(ifmt_ctx->streams[stream_index]->codec, frame,  
  454.                    &got_frame, &packet);  
  455.             if (ret < 0) {  
  456.                av_frame_free(&frame);  
  457.                av_log(NULL, AV_LOG_ERROR, "Decodingfailed\n");  
  458.                 break;  
  459.             }  
  460.             if (got_frame) {  
  461.                frame->pts = av_frame_get_best_effort_timestamp(frame);  
  462.                 ret= filter_encode_write_frame(frame, stream_index);  
  463.                av_frame_free(&frame);  
  464.                 if (ret< 0)  
  465.                    goto end;  
  466.             } else {  
  467.                av_frame_free(&frame);  
  468.             }  
  469.         } else {  
  470.             /* remux this frame without reencoding */  
  471.            packet.dts = av_rescale_q_rnd(packet.dts,  
  472.                    ifmt_ctx->streams[stream_index]->time_base,  
  473.                    ofmt_ctx->streams[stream_index]->time_base,  
  474.                     (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));  
  475.            packet.pts = av_rescale_q_rnd(packet.pts,  
  476.                    ifmt_ctx->streams[stream_index]->time_base,  
  477.                    ofmt_ctx->streams[stream_index]->time_base,  
  478.                     (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));  
  479.             ret =av_interleaved_write_frame(ofmt_ctx, &packet);  
  480.             if (ret < 0)  
  481.                 goto end;  
  482.         }  
  483.        av_free_packet(&packet);  
  484.     }  
  485.     /* flush filters and encoders */  
  486.     for (i = 0; i < ifmt_ctx->nb_streams; i++) {  
  487.         /* flush filter */  
  488.         if (!filter_ctx[i].filter_graph)  
  489.             continue;  
  490.         ret =filter_encode_write_frame(NULL, i);  
  491.         if (ret < 0) {  
  492.            av_log(NULL, AV_LOG_ERROR, "Flushingfilter failed\n");  
  493.             goto end;  
  494.         }  
  495.         /* flush encoder */  
  496.         ret = flush_encoder(i);  
  497.         if (ret < 0) {  
  498.            av_log(NULL, AV_LOG_ERROR, "Flushingencoder failed\n");  
  499.             goto end;  
  500.         }  
  501.     }  
  502.    av_write_trailer(ofmt_ctx);  
  503. end:  
  504.    av_free_packet(&packet);  
  505.    av_frame_free(&frame);  
  506.     for (i = 0; i < ifmt_ctx->nb_streams; i++) {  
  507.        avcodec_close(ifmt_ctx->streams[i]->codec);  
  508.         if (ofmt_ctx && ofmt_ctx->nb_streams >i && ofmt_ctx->streams[i] &&ofmt_ctx->streams[i]->codec)  
  509.            avcodec_close(ofmt_ctx->streams[i]->codec);  
  510.         if(filter_ctx && filter_ctx[i].filter_graph)  
  511.            avfilter_graph_free(&filter_ctx[i].filter_graph);  
  512.     }  
  513.    av_free(filter_ctx);  
  514.    avformat_close_input(&ifmt_ctx);  
  515.     if (ofmt_ctx &&!(ofmt_ctx->oformat->flags & AVFMT_NOFILE))  
  516.         avio_close(ofmt_ctx->pb);  
  517.    avformat_free_context(ofmt_ctx);  
  518.     if (ret < 0)  
  519.        av_log(NULL, AV_LOG_ERROR, "Erroroccurred\n");  
  520.     return (ret? 1:0);  
  521. }  

 

程序运行截图:

最简单的基于FFMPEG的转码程序_ffmpeg_04

默认情况下运行程序,会将“cuc_ieschool.ts”转换为“cuc_ieschool.avi”。调试的时候,可以修改“配置属性->调试->命令参数”中的参数,即可改变转码的输入输出文件。

最简单的基于FFMPEG的转码程序_音频_05