格式转换

学习了一部分ffmpeg的知识，但是没有目标的学习似乎好像学不到什么东西。尤其是编程方面。想找一些例子来做，看着别人的博客却提不起兴趣来做。当自己有了一个目标的时候才会充满动力，在了解了一些音视频的知识之后，才知道原来各种音视频的格式不是靠改个后缀名就能改变其编码格式的。于是便想实现一个格式转换的程序，来完成格式的转换。当然我也是只是测试了视频的转换，并没有测试音频的转换。

对于格式的转换其实就是对于要转换的文件首先进行解封装，然后再通过要转换的格式的编解码器进行封装，最后将数据写入目标文件。

其流程图如下：

img

格式转换代码流程图如下：

img

实现部分

先从输入文件中读取数据以及数据的信息：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
av_register_all();
AVFormatContext *ifmt_ctx = avformat_alloc_context();
if (avformat_open_input(&ifmt_ctx,input_file,NULL,NULL) != 0)
{
    std::cout << "fail to open the input file " << input_file << std::endl;
    return -1;
}
if (avformat_find_stream_info(ifmt_ctx,NULL) < 0)
{
    std::cout << "fail to find the stream information " << std::endl;
    return -1;
}
//显示解码信息
av_dump_format(ifmt_ctx,0,input_file,0);

根据输出文件的格式，得到输出文件的解码格式，如果输出文件没有扩展名，则会出错：

1
2
3
4
5
6
7
8
9
10
11
12
AVOutputFormat *out_fmt = av_guess_format(NULL,output_file,NULL);
  if (out_fmt == NULL)
  {
      std::cout << "fail to find the format" << std::endl;
      return -1;
  }
  AVFormatContext *ofmt_ctx = avformat_alloc_context();
  if (avformat_alloc_output_context2(&ofmt_ctx,out_fmt,out_fmt->name,output_file) < 0)
  {
      std::cout << "fail to alloc the output context" << std::endl;
      return -1;
  }

将各种流的编解码参数设置到输出Context中：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
for(int i=0;i < ifmt_ctx->nb_streams;i++){
        //找到解码器
        AVStream *input_stream = ifmt_ctx->streams[i];
        AVCodec *pCodec = avcodec_find_decoder(input_stream->codecpar->codec_id);
        //定义输出参数
        AVStream *out_stream = avformat_new_stream(ofmt_ctx,pCodec);
        if (out_stream == NULL)
        {
            std::cout << "fail to create the out stream " << std::endl;
            return -1;
        }
        //复制解码参数
        if (avcodec_parameters_copy(out_stream->codecpar,input_stream->codecpar) < 0)
        {
            std::cout << "fail to copy parameters " << std::endl;
            return -1;
        }
        out_stream->codecpar->codec_tag = 0;
        if (ofmt_ctx->oformat->flags & AVFMT_GLOBALHEADER)
            out_stream->codec->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;
    }

    //输出信息
    av_dump_format(ofmt_ctx,0,output_file,1);

打开输出文件，并写入数据，写写入头信息，最后写入尾部信息：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
if (!(out_fmt->flags &AVFMT_NOFILE)) {
       if (avio_open(&ofmt_ctx->pb, output_file, AVIO_FLAG_WRITE) < 0) {
           std::cout << "fail to open the avio context" << std::endl;
           return -1;
       }
   }
   //写入信息头
   if (avformat_write_header(ofmt_ctx,NULL) < 0)
   {
       std::cout << "fail to write header" << std::endl;
       return -1;
   }
   AVPacket *pkt = av_packet_alloc();
   av_init_packet(pkt);
   int frame_index = 0;
   //转换avi需要使用bsf，如果不使用这个会转换失败
   AVBitStreamFilterContext *bsf = av_bitstream_filter_init("h264_mp4toannexb");

   while(av_read_frame(ifmt_ctx,pkt) >=0)
   {
       AVStream *in_steam,*out_stream;
       in_steam = ifmt_ctx->streams[pkt->stream_index];
       out_stream = ofmt_ctx->streams[pkt->stream_index];
       pkt->pts = av_rescale_q_rnd(pkt->pts,in_steam->time_base,out_stream->time_base,(AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
       pkt->dts = av_rescale_q_rnd(pkt->dts,in_steam->time_base,out_stream->time_base,(AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
       pkt->duration = av_rescale_q(pkt->duration,in_steam->time_base,out_stream->time_base);
       pkt->pos = -1;
       if (pkt->stream_index == 0)
       {
           AVPacket *fpkt = av_packet_alloc();
           int test = av_bitstream_filter_filter(bsf,in_steam->codec,NULL,&fpkt->data,&fpkt->size,
                   pkt->data,pkt->size,pkt->flags | AV_PKT_FLAG_KEY);
           pkt->data = fpkt->data;
           pkt->size = fpkt->size;
           av_packet_free(&fpkt);
       }

       if (av_interleaved_write_frame(ofmt_ctx,pkt) < 0){
           std::cout << "fail to write" << frame_index <<" frame" << std::endl;
           continue;
       }
       av_packet_unref(pkt);
       frame_index ++;
   }
   //写入文件尾部信息
   if (av_write_trailer(ofmt_ctx) < 0)
   {
       std::cout << "fail to write trailer " << std::endl;
       return -1;
   }

释放所有申请的资源，关闭打开的环境：

1
2
3
4
5
6
av_bitstream_filter_close(bsf);
    if (ofmt_ctx && !(out_fmt->flags & AVFMT_NOFILE))
        avio_close(ofmt_ctx->pb);
    av_packet_free(&pkt);
    avformat_close_input(&ifmt_ctx);
    avformat_free_context(ifmt_ctx);

所有代码如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
#include <iostream>
#ifdef __cplusplus
extern "C" {
#endif
#include <libavcodec/avcodec.h>
#include <libavutil/avutil.h>
#include <libswresample/swresample.h>
#include <libavformat/avformat.h>
#include <libavfilter/avfilter.h>
#ifdef __cplusplus
}
#endif
int main(int argc,char* argv[])
{
    if (argc != 2)
    {
        std::cout << "usage :convert <input> <output>" << std::endl;
    }
    const char* input_file = argv[1];
    const char* output_file = argv[2];
    av_register_all();
    AVFormatContext *ifmt_ctx = avformat_alloc_context();
    if (avformat_open_input(&ifmt_ctx,input_file,NULL,NULL) != 0)
    {
        std::cout << "fail to open the input file " << input_file << std::endl;
        return -1;
    }
    if (avformat_find_stream_info(ifmt_ctx,NULL) < 0)
    {
        std::cout << "fail to find the stream information " << std::endl;
        return -1;
    }
    //显示解码信息
    av_dump_format(ifmt_ctx,0,input_file,0);
    //输出设置
    AVOutputFormat *out_fmt = av_guess_format(NULL,output_file,NULL);
    if (out_fmt == NULL)
    {
        std::cout << "fail to find the format" << std::endl;
        return -1;
    }
    AVFormatContext *ofmt_ctx = avformat_alloc_context();
    if (avformat_alloc_output_context2(&ofmt_ctx,out_fmt,out_fmt->name,output_file) < 0)
    {
        std::cout << "fail to alloc the output context" << std::endl;
        return -1;
    }


    for(int i=0;i < ifmt_ctx->nb_streams;i++){
        //找到解码器
        AVStream *input_stream = ifmt_ctx->streams[i];
        AVCodec *pCodec = avcodec_find_decoder(input_stream->codecpar->codec_id);
        //定义输出参数
        AVStream *out_stream = avformat_new_stream(ofmt_ctx,pCodec);
        if (out_stream == NULL)
        {
            std::cout << "fail to create the out stream " << std::endl;
            return -1;
        }
        //复制解码参数
        if (avcodec_parameters_copy(out_stream->codecpar,input_stream->codecpar) < 0)
        {
            std::cout << "fail to copy parameters " << std::endl;
            return -1;
        }
        out_stream->codecpar->codec_tag = 0;
        if (ofmt_ctx->oformat->flags & AVFMT_GLOBALHEADER)
            out_stream->codec->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;
    }

    //输出信息
    av_dump_format(ofmt_ctx,0,output_file,1);
    if (!(out_fmt->flags &AVFMT_NOFILE)) {
        if (avio_open(&ofmt_ctx->pb, output_file, AVIO_FLAG_WRITE) < 0) {
            std::cout << "fail to open the avio context" << std::endl;
            return -1;
        }
    }
    //写入信息头
    if (avformat_write_header(ofmt_ctx,NULL) < 0)
    {
        std::cout << "fail to write header" << std::endl;
        return -1;
    }
    AVPacket *pkt = av_packet_alloc();
    av_init_packet(pkt);
    int frame_index = 0;
    //转换avi需要使用bsf
    AVBitStreamFilterContext *bsf = av_bitstream_filter_init("h264_mp4toannexb");

    while(av_read_frame(ifmt_ctx,pkt) >=0)
    {
        AVStream *in_steam,*out_stream;
        in_steam = ifmt_ctx->streams[pkt->stream_index];
        out_stream = ofmt_ctx->streams[pkt->stream_index];
        pkt->pts = av_rescale_q_rnd(pkt->pts,in_steam->time_base,out_stream->time_base,(AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt->dts = av_rescale_q_rnd(pkt->dts,in_steam->time_base,out_stream->time_base,(AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt->duration = av_rescale_q(pkt->duration,in_steam->time_base,out_stream->time_base);
        pkt->pos = -1;
        if (pkt->stream_index == 0)
        {
            AVPacket *fpkt = av_packet_alloc();
            int test = av_bitstream_filter_filter(bsf,in_steam->codec,NULL,&fpkt->data,&fpkt->size,
                    pkt->data,pkt->size,pkt->flags | AV_PKT_FLAG_KEY);
            pkt->data = fpkt->data;
            pkt->size = fpkt->size;
            av_packet_free(&fpkt);
        }

        if (av_interleaved_write_frame(ofmt_ctx,pkt) < 0){
            std::cout << "fail to write" << frame_index <<" frame" << std::endl;
            continue;
        }
        av_packet_unref(pkt);
        frame_index ++;
    }
    //写入文件尾部信息
    if (av_write_trailer(ofmt_ctx) < 0)
    {
        std::cout << "fail to write trailer " << std::endl;
        return -1;
    }

    av_bitstream_filter_close(bsf);
    if (ofmt_ctx && !(out_fmt->flags & AVFMT_NOFILE))
        avio_close(ofmt_ctx->pb);
    av_packet_free(&pkt);
    avformat_close_input(&ifmt_ctx);
    avformat_free_context(ifmt_ctx);
    return 0;
}
赏

谢谢你请我吃糖果

扫一扫，分享到微信

微信分享二维码

tag:

    缺失模块。
    1、请确保node版本大于6.2
    2、在博客根目录（注意不是yilia根目录）执行以下命令：
    npm i hexo-generator-json-content --save

    3、在根目录_config.yml里添加配置： 

      jsonContent:
    meta: false
    pages: false
    posts:
      title: true
      date: true
      path: true
      text: false
      raw: false
      content: false
      slug: false
      updated: false
      comments: false
      link: false
      permalink: false
      excerpt: false
      categories: false
      tags: true

I am what I am