FFMPEG開發(fā)快速入坑——音視頻混流處理

if (m_pFormatCtx->oformat->flags & AVFMT_GLOBALHEADER)
{
    m_ptrVideoEncCtx->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;
}

AVFormatContext*  m_pFormatCtx    = nullptr;      // 媒體格式上下文
bool              m_bGlobalHeader = true;         // 音視頻編解碼器是否需要標記 AV_CODEC_FLAG_GLOBAL_HEADER
AVStream*         m_pVideoStream  = nullptr;      // 視頻流信息
AVStream*         m_pAudioStream  = nullptr;      // 音頻流信息


/**
  * @brief 打開音視頻混流器
  * @param 
  * @return 返回錯誤碼， 0 表示正常; < 0 表示錯誤碼
  *
  */
int32_t MuxerOpen(
  const char*           pszFilePath,        // 要保存的媒體文件,通常是.mp4文件
  const AVCodecContext* pVideoEncCtx,       // 視頻編碼器上下文
  const AVCodecContext* pAudioEncCtx  )     // 音頻編碼器上下文
{
  int res = 0;


  // 創(chuàng)建輸出流格式上下文
  res = avformat_alloc_output_context2(&m_pFormatCtx, nullptr, nullptr, pszFilePath);
  if (nullptr == m_pFormatCtx || res < 0)
  {
    LOGE(" [ERROR] fail to avformat_alloc_output_context2()\n");
    return -1;
  }
  if (m_pFormatCtx->oformat->flags & AVFMT_GLOBALHEADER)
  {
    m_bGlobalHeader = true;
  }


  // 創(chuàng)建寫入的視頻流
  m_pVideoStream = avformat_new_stream(m_pFormatCtx, nullptr);
  if (nullptr == m_pVideoStream)
  {
    LOGE(" [ERROR] fail to create video stream\n");
    avformat_free_context(m_pFormatCtx);
    return -2;
  }
  res = avcodec_parameters_from_context(m_pVideoStream->codecpar, pVideoEncCtx);
  if (res < 0)
  {
    LOGE(" [ERROR] fail to video avcodec_parameters_from_context(), res=%d\n", res);
    avformat_free_context(m_pFormatCtx);
    return -2;
  }
  m_pVideoStream->time_base = pVideoEncCtx->time_base;


  // 創(chuàng)建寫入的音頻流
  m_pAudioStream = avformat_new_stream(m_pFormatCtx, nullptr);
  if (nullptr == m_pAudioStream)
  {
    LOGE(" [ERROR] fail to create video stream\n");
    avformat_free_context(m_pFormatCtx);
    return -2;
  }
  res = avcodec_parameters_from_context(m_pAudioStream->codecpar, pAudioEncCtx);
  if (res < 0)
  {
    LOGE(" [ERROR] fail to audio avcodec_parameters_from_context(), res=%d\n", res);
    avformat_free_context(m_pFormatCtx);
    return -2;
  }
  m_pAudioStream->time_base = pVideoEncCtx->time_base;


  // 打開文件IO上下文
  res = avio_open(&m_pFormatCtx->pb, pszFilePath, AVIO_FLAG_WRITE);
  if (res < 0)
  {
    LOGE(" [ERROR] fail to avio_open(), res=%d\n", res);
    avformat_free_context(m_pFormatCtx);
    return -2;
  }


  //
  // 寫入文件頭信息
  //
  res = avformat_write_header(m_pFormatCtx, nullptr);
  if (res < 0)
  {
    LOGE(" [ERROR] fail to FF_avformat_write_header(), res=%d\n", res);
    avformat_free_context(m_pFormatCtx);
    return -3;
  }

  return 0;
}


/**
  * @brief  關(guān)閉音視頻混流器
  * @param  無
  * @return 無
  * 
  */
void MuxerClose()
{
  // 寫入尾信息
  if (m_pFormatCtx != nullptr)
  {
    av_write_trailer(m_pFormatCtx);
  }

  // 先關(guān)IO上下文
  if (m_pFormatCtx->pb != nullptr)
  {
    avio_closep(&m_pFormatCtx->pb);
    m_pFormatCtx->pb = nullptr;
  }

  // 再釋放媒體格式上下文
  if (m_pFormatCtx != nullptr)
  {
    avformat_free_context(m_pFormatCtx);
    m_pFormatCtx = nullptr;
  }

  // 流文件直接在 avformat_free_context()內(nèi)部已經(jīng)銷毀了
  m_pVideoStream = nullptr;  
  m_pAudioStream = nullptr;
 
}


/**
  * @brief  寫入編碼后的音頻或者視頻數(shù)據(jù)包
  * @param  無
  * @return 無
  * 
  */
int32_t MuxerWrite(bool bVideoPkt, AVPacket* pInPacket)
{
  // 設(shè)置寫入數(shù)據(jù)包的流索引
  if (bVideoPkt)
  {
    pInPacket->stream_index = m_pVideoStream->index;
  }
  else
  {
    pInPacket->stream_index = m_pAudioStream->index;
  }

  // 寫入媒體文件
  int res = av_interleaved_write_frame(m_pFormatCtx, pInPacket);
  return res;
}