FFmpeg连载7-mp3转码aac及AVAudioFifo的使用

at 1年前  ca FFmpeg  pv 1072  by touch  

前言

如今以抖音、快手为代表的短视频秀无处不在，比如它们一个很普通的功能就是使用流行音乐替换作为视频的背景音乐。

而在视频中音频一般都是以AAC的形成存在，但流行音乐大多以mp3的格式传播， 因此需要完成背景音乐替换这个功能，其中的一个步骤就需要完成mp3转aac这样的一个音频转转码的过程。

按照我们以往的经验，转码的大致流程应该是这样的：

解封装->提取音频流->解码成PCM->重新编码成AAC

流程是这样没错，但是内部的出来细节是怎样的呢？是mp3解码出来后的AVFrame可以通过函数avcodec_send_frame送进aac编码器即可吗？

很明显这是不行的，因为mp3每帧是1152个采样点，而aac每帧是1024个采样点。它们每帧的采样点数不同，所以不能直接通过avcodec_send_frame进行编码。

AVAudioFifo

AVAudioFifo是一个音频缓冲区，是一个先进先出的队列。

使用它可以很方便地储存我们的音频缓冲数据，例如在mp3转码aac的过程中，因为它们的采样点数不同，我们就可以把mp3解码出来的 pcm数据放入到AVAudioFifo中去，然后每次从AVAudioFifo中获取1024个采样点送进aac编码器，这样的做法让我们的音频转码变得非常的方便灵活。

AVAudioFifo让我们在采样层面做操作，而不用关心底层的字节层面；而且它支持多种格式的单次采样，如支持planar或packed的采样格式，支持不同的通道数等等。

AVAudioFifo的API使用也非常简单，主要包含分配释放、获取可读写空间长度、写入音频数据、读取音频数据等相关函数。

首先是分配和释放操作：

  1. //分配一个AVAudioFifo。
  2. //sample_fmt指定采样格式
  3. //nb_samples则指定AVAudioFifo的缓冲区大小，可以通过av_audio_fifo_realloc重新分配
  4. AVAudioFifo *av_audio_fifo_alloc(enum AVSampleFormat sample_fmt, int channels,int nb_samples);
  5.  
  6. //重新分配缓冲区大小
  7. //成功返回0，失败返回负的错误值
  8. int av_audio_fifo_realloc(AVAudioFifo *af, int nb_samples);
  9.  
  10. //释放AVAudioFifo
  11. void av_audio_fifo_free(AVAudioFifo *af);

查询操作：

  1. //返回fifo中当前存储的采样数量int av_audio_fifo_size(AVAudioFifo *af);
  2.  //返回fifo中当前可写的采样数量，即尚未使用的空间数量int av_audio_fifo_space(AVAudioFifo *af);
  3.  // 以上两个函数的返回值之和等于AVAudioFifo的缓冲区大小

读取操作：

  1. //将采样写入到AVAudioFifo
  2. //成功则返回实际写入的采样数，如果写入成功，返回值必定等于nb_samples，失败返回负的错误值
  3. int av_audio_fifo_write(AVAudioFifo *af, void **data, int nb_samples);
  4.  
  5.  //peek：读取数据，但读到的数据并不会从fifo中删除
  6.  int av_audio_fifo_peek(AVAudioFifo *af, void **data, int nb_samples);
  7.  
  8.  //从指定的偏移位置peek数据
  9.  int av_audio_fifo_peek_at(AVAudioFifo *af, void **data, int nb_samples, int offset);
  10.  
  11.  //读取数据，读到的数据会从fifo中删除
  12.  int av_audio_fifo_read(AVAudioFifo *af, void **data, int nb_samples);
  13.  
  14.  //从fifo中删除nb_samples个采样
  15.  int av_audio_fifo_drain(AVAudioFifo *af, int nb_samples);
  16.  
  17.  //删除fifo中的所有采样，清空
  18.  void av_audio_fifo_reset(AVAudioFifo *af);

音频转码

有了AVAudioFifo，那么我们音频的转码流程就变成了以下这样子：

解封装 -> 提取音频流 -> 解码成PCM->将PCM数据写入AVAudioFifo -> 每次从AVAudioFifo获取1024个采样点送进aac编码器 -> 重新编码成AAC

如果到了最后没有可输入的PCM数据了，但是AVAudioFifo中可读取的采样点数依然不满足aac的1024个采样点的话，可以通过填充静音的方式补充...

上代码：

  1. #include <iostream>
  2.  
  3. extern "C" {
  4. #include <libavformat/avformat.h>
  5. #include <libavcodec/avcodec.h>
  6. #include <libavutil/audio_fifo.h>
  7. #include <libavutil/channel_layout.h>
  8. }
  9.  
  10. class Mp3ToAAC {
  11. public:
  12.     void mp3_to_aac(const char *mp3, const char *aac) {
  13.         avFormatContext = avformat_alloc_context();
  14.         int ret = avformat_open_input(&avFormatContext, mp3, nullptr, nullptr);
  15.         if (ret < 0) {
  16.             std::cout << "打开mp3输入流失败" << std::endl;
  17.             return;
  18.         }
  19.         av_dump_format(avFormatContext, 0, mp3, 0);
  20.  
  21.         std::cout << "流的类型:" << avFormatContext->streams[0]->codecpar->codec_type << std::endl;
  22.  
  23.         int audio_index = av_find_best_stream(avFormatContext, AVMEDIA_TYPE_AUDIO, -1, -1, nullptr, 0);
  24.         if (audio_index < 0) {
  25.             std::cout << "没有找到音频流，换一个方式查找" << std::endl;
  26.             for (int i = 0; i < avFormatContext->nb_streams; ++i) {
  27.                 if (AVMEDIA_TYPE_AUDIO == avFormatContext->streams[i]->codecpar->codec_type) {
  28.                     audio_index = i;
  29.                     std::cout << "找到音频流，audio_index:" << audio_index << std::endl;
  30.                     break;
  31.                 }
  32.             }
  33.  
  34.             if (audio_index < 0) {
  35.                 return;
  36.             }
  37.         }
  38.  
  39.         // 初始化输出
  40.         out_format_context = avformat_alloc_context();
  41.         const AVOutputFormat *avOutputFormat = av_guess_format(nullptr, aac, nullptr);
  42.         out_format_context->oformat = avOutputFormat;
  43.         const AVCodec *aac_encoder = avcodec_find_encoder(AV_CODEC_ID_AAC);
  44.         encode_CodecContext = avcodec_alloc_context3(aac_encoder);
  45.  
  46.         encode_CodecContext->channel_layout = AV_CH_LAYOUT_STEREO;
  47.         encode_CodecContext->channels = av_get_channel_layout_nb_channels(encode_CodecContext->channel_layout);
  48.         // 如果解码出来的pcm不是44100的则需要进行重采样，重采样需要主要音频时长不变
  49.         encode_CodecContext->sample_rate = 44100;
  50.         // 比如使用 22050的采样率进行编码，编码后的时长明显是比实际音频长的
  51. //        encode_CodecContext->sample_rate = 22050;
  52.         encode_CodecContext->codec_type = AVMEDIA_TYPE_AUDIO;
  53.         encode_CodecContext->sample_fmt = AV_SAMPLE_FMT_FLTP;
  54.         encode_CodecContext->profile = FF_PROFILE_AAC_LOW;
  55.         //ffmpeg默认的aac是不带adts，而fdk_aac默认带adts，这里我们强制不带
  56.         encode_CodecContext->flags = AV_CODEC_FLAG_GLOBAL_HEADER;
  57.         // 打开编码器
  58.         ret = avcodec_open2(encode_CodecContext, aac_encoder, nullptr);
  59.         if (ret < 0) {
  60.             char error[1024];
  61.             av_strerror(ret, error, 1024);
  62.             std::cout << "编码器打开失败：" << error << std::endl;
  63.             return;
  64.         }
  65.  
  66.         AVStream *aac_stream = avformat_new_stream(out_format_context, aac_encoder);
  67.         aac_index = aac_stream->index;
  68.         avcodec_parameters_from_context(aac_stream->codecpar, encode_CodecContext);
  69.         ret = avio_open(&out_format_context->pb, aac, AVIO_FLAG_WRITE);
  70.         if (ret < 0) {
  71.             std::cout << "输出流打开失败" << std::endl;
  72.             return;
  73.         }
  74.         ret = avformat_write_header(out_format_context, nullptr);
  75.         if (ret < 0) {
  76.             std::cout << "文件头写入失败" << std::endl;
  77.             return;
  78.         }
  79.  
  80.         // 解码相关
  81.         const AVCodec *decoder = avcodec_find_decoder(avFormatContext->streams[audio_index]->codecpar->codec_id);
  82.         avCodecContext = avcodec_alloc_context3(decoder);
  83.         avcodec_parameters_to_context(avCodecContext, avFormatContext->streams[audio_index]->codecpar);
  84.         ret = avcodec_open2(avCodecContext, decoder, nullptr);
  85.         if (ret < 0) {
  86.             std::cout << "解码器打开失败" << std::endl;
  87.             return;
  88.         }
  89.         // 分配frame和pack
  90.         AVPacket *avPacket = av_packet_alloc();
  91.         avFrame = av_frame_alloc();
  92.         out_pack = av_packet_alloc();
  93.  
  94.         encode_frame = av_frame_alloc();
  95.         encode_frame->nb_samples = encode_CodecContext->frame_size;
  96.         encode_frame->sample_rate = encode_CodecContext->sample_rate;
  97.         encode_frame->channel_layout = encode_CodecContext->channel_layout;
  98.         encode_frame->channels = encode_CodecContext->channels;
  99.         encode_frame->format = encode_CodecContext->sample_fmt;
  100.         av_frame_get_buffer(encode_frame, 0);
  101.  
  102.         // 初始化audiofifo
  103.         audiofifo = av_audio_fifo_alloc(encode_CodecContext->sample_fmt, encode_CodecContext->channels,
  104.                                         encode_CodecContext->frame_size);
  105.         while (true) {
  106.             ret = av_read_frame(avFormatContext, avPacket);
  107.             if (ret < 0) {
  108.                 std::cout << "read end" << std::endl;
  109.                 break;
  110.             } else if (avPacket->stream_index == audio_index) {
  111.                 decode_to_pcm(avPacket);
  112.             }
  113.             av_packet_unref(avPacket);
  114.         }
  115.  
  116.         // todo 需要冲刷数据，不然可能会漏掉几帧数据 [aac @ 0x125e05f50] 2 frames left in the queue on closing
  117.  
  118.         ret = av_write_trailer(out_format_context);
  119.         if (ret < 0) {
  120.             std::cout << "文件尾写入失败" << std::endl;
  121.         }
  122.     }
  123.  
  124.     ~Mp3ToAAC() {
  125.         if (nullptr != avFormatContext) {
  126.             avformat_close_input(&avFormatContext);
  127.         }
  128.         avcodec_free_context(&avCodecContext);
  129.  
  130.         if (nullptr != out_format_context) {
  131.             avformat_close_input(&out_format_context);
  132.         }
  133.         avcodec_free_context(&encode_CodecContext);
  134.     }
  135.  
  136. private:
  137.     // 解码
  138.     AVFormatContext *avFormatContext = nullptr;
  139.     AVCodecContext *avCodecContext = nullptr;
  140.     AVFrame *avFrame = nullptr;
  141.  
  142.     // 编码
  143.     AVFormatContext *out_format_context = nullptr;
  144.     AVCodecContext *encode_CodecContext = nullptr;
  145.     AVPacket *out_pack = nullptr;
  146.     AVFrame *encode_frame = nullptr;
  147.     AVAudioFifo *audiofifo = nullptr;
  148.  
  149.     int aac_index = 0;
  150.     int64_t cur_pts = 0;
  151.  
  152.     void encode_to_aac() {
  153.         av_frame_make_writable(encode_frame);
  154.         // todo 如果是冲刷最后几帧数据，不够的可以填充静音  av_samples_set_silence
  155.         while (av_audio_fifo_size(audiofifo) > encode_CodecContext->frame_size) {
  156.             int ret = av_audio_fifo_read(audiofifo, reinterpret_cast<void **>(encode_frame->data),
  157.                                          encode_CodecContext->frame_size);
  158.             if (ret < 0) {
  159.                 std::cout << "audiofifo 读取数据失败" << std::endl;
  160.                 return;
  161.             }
  162.             cur_pts += encode_frame->nb_samples;
  163.             encode_frame->pts = cur_pts;
  164.             ret = avcodec_send_frame(encode_CodecContext, encode_frame);
  165.             if (ret < 0) {
  166.                 std::cout << "发送编码失败" << std::endl;
  167.                 return;
  168.             }
  169.             while (true) {
  170.                 ret = avcodec_receive_packet(encode_CodecContext, out_pack);
  171.                 if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
  172.                     std::cout << "avcodec_receive_packet end:" << ret << std::endl;
  173.                     break;
  174.                 } else if (ret < 0) {
  175.                     std::cout << "avcodec_receive_packet fail:" << ret << std::endl;
  176.                     return;
  177.                 } else {
  178.                     out_pack->stream_index = aac_index;
  179.                     ret = av_write_frame(out_format_context, out_pack);
  180.                     if (ret < 0) {
  181.                         std::cout << "av_write_frame fail:" << ret << std::endl;
  182.                         return;
  183.                     } else {
  184.                         std::cout << "av_write_frame success:" << ret << std::endl;
  185.                     }
  186.                 }
  187.             }
  188.         }
  189.     }
  190.  
  191.     void decode_to_pcm(const AVPacket *avPacket) {
  192.         int ret = avcodec_send_packet(avCodecContext, avPacket);
  193.         if (ret < 0) {
  194.             char error[1024];
  195.             av_strerror(ret, error, 1024);
  196.             std::cout << "发送解码失败:" << error << std::endl;
  197.         }
  198.         while (true) {
  199.             ret = avcodec_receive_frame(avCodecContext, avFrame);
  200.             if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
  201.                 std::cout << "avcodec_receive_frame end:" << ret << std::endl;
  202.                 break;
  203.             } else if (ret < 0) {
  204.                 std::cout << "获取解码数据失败" << std::endl;
  205.                 return;
  206.             } else {
  207.                 std::cout << "获取解码数据成功 nb_samples:" << avFrame->nb_samples << std::endl;
  208.                 /**
  209.                  * mp3解码出来的pcm无法直接编码成aac，
  210.                  * 因为mp3每帧是1152个采样点，而aac每帧需要1024个采样点
  211.                  * 解决方案是使用AVAudioFifo缓冲起来
  212.                  */
  213.  
  214.                 int cache_size = av_audio_fifo_size(audiofifo);
  215.                 std::cout << "cache_size:" << cache_size << std::endl;
  216.                 av_audio_fifo_realloc(audiofifo, cache_size + avFrame->nb_samples);
  217.                 av_audio_fifo_write(audiofifo, reinterpret_cast<void **>(avFrame->data), avFrame->nb_samples);
  218.                 encode_to_aac();
  219.             }
  220.         }
  221.     }
  222. };

作者：思想觉悟

链接：juejin.cn/post/70902033


版权声明

本文仅代表作者观点，不代表码农殇立场。
本文系作者授权码农殇发表，未经许可，不得转载。

 

扫一扫在手机阅读、分享本文

相关阅读
冀ICP备14009681号-2 Powered By 码农殇 Theme By zb脚本
您是本站第8028名访客 今日有0篇新文章