调用FFmpeg的c语言sdk实现抽取视频中的音频数据

主要使用函数

• 特征码：Start code

• 解码的一些视频参数，分辨率和帧率：SPS/PPS

• ffmpeg获取SPS/PPS：codec->extradata

实例

#include <stdio.h>
#include <libavutil/log.h>
#include <libavformat/avformat.h>


#define ADTS_HEADER_LEN  7;

void adts_header(char *szAdtsHeader, int dataLen){

    int audio_object_type = 2;
    int sampling_frequency_index = 7;
    int channel_config = 2;

    int adtsLen = dataLen + 7;

    szAdtsHeader[0] = 0xff;         //syncword:0xfff                          高8bits
    szAdtsHeader[1] = 0xf0;         //syncword:0xfff                          低4bits
    szAdtsHeader[1] |= (0 << 3);    //MPEG Version:0 for MPEG-4,1 for MPEG-2  1bit
    szAdtsHeader[1] |= (0 << 1);    //Layer:0                                 2bits 
    szAdtsHeader[1] |= 1;           //protection absent:1                     1bit

    szAdtsHeader[2] = (audio_object_type - 1)<<6;            //profile:audio_object_type - 1                      2bits
    szAdtsHeader[2] |= (sampling_frequency_index & 0x0f)<<2; //sampling frequency index:sampling_frequency_index  4bits 
    szAdtsHeader[2] |= (0 << 1);                             //private bit:0                                      1bit
    szAdtsHeader[2] |= (channel_config & 0x04)>>2;           //channel configuration:channel_config               高1bit

    szAdtsHeader[3] = (channel_config & 0x03)<<6;     //channel configuration:channel_config      低2bits
    szAdtsHeader[3] |= (0 << 5);                      //original：0                               1bit
    szAdtsHeader[3] |= (0 << 4);                      //home：0                                   1bit
    szAdtsHeader[3] |= (0 << 3);                      //copyright id bit：0                       1bit  
    szAdtsHeader[3] |= (0 << 2);                      //copyright id start：0                     1bit
    szAdtsHeader[3] |= ((adtsLen & 0x1800) >> 11);           //frame length：value   高2bits

    szAdtsHeader[4] = (uint8_t)((adtsLen & 0x7f8) >> 3);     //frame length:value    中间8bits
    szAdtsHeader[5] = (uint8_t)((adtsLen & 0x7) << 5);       //frame length:value    低3bits
    szAdtsHeader[5] |= 0x1f;                                 //buffer fullness:0x7ff 高5bits
    szAdtsHeader[6] = 0xfc;
}



int main(int argc, char argv[]){
  int err_code;
  char errors[1024];
  
  char *src_filename = NULL;
  char *dst_filename = NULL;
  
  FILE *dst_fd = NULL;
  int video_stream_index = -1;
  
  // 创建文件上下文，和包管理
  AVFormatContext *fmt_ctx = NULL;
  AVPacket pkt;
  
  // 设置日志等级
  av_log_set_level(AV_LOG_DEBUG);
  
  // 判断参数个数
  if (argc < 3){
    av_log(NULL, AV_LOG_DEBUG, "参数不足");
    return -1
  }
  // 输出输出文件
  src_filename = argv[1];
  dst_filename = argv[2];
  
  //  判断参数不可以为空
  if (src_filename == NULL || dst_filename == NULL){
    av_log(NULL, AV_LOG_ERROR, "传入的参数不可以为空");
    return -1;
  }
  
  // 全局注册函数
  av_register_all();
  
  // 打开目标文件
  dst_fd = fopen(dst_filename, "wb");
  if (!dst_fd){
    av_log(NULL, AV_LOG_DEBUG, "不能打开文件%s", dst_filename);
  	return -1;
  }
  
  // 读取流文件
  if ((err_code=avformat_open_input(&fmt_ctx, src_filename, NULL, NULL))<0){
    // 设置错误信息
    av_strerror(err_code, errors, 1024);
    // 打印日志输出
    av_log(NULL, AV_LOG_DEBUG, "流文件读取失败: %s, %d(%s)\n",
    src_filename,
    err_code,
    errors;);
    return -1;
  }
  av_dump_format(fmt_ctx, 0 , src_filename, 0);
  
  // 数据包处理
  av_init_packet(&pkt);
  pkt.data=NULL;
  pkt.size = 0;
  
  // 找到最好的流
  video_stream_index = av_find_best_stream(fmt_ctx, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);
  if (video_stream_index<0){
    av_log(NULL, AV_LOG_DEBUG, "不能找到流输出", av_get_media_type_string(AVMEDIA_TYPE_VIDEO),
          src_filename);
    return AVERROR(EINVAL);
  }
  
  // 拿到视频的index，去视频流中去取一个一个包文件
  while (av_read_frame(fmt_ctx, &pkt) >= 0){
    if(pkt.stream_index == video_stream_index){
      
      // 处理h264处理函数
      h264_mp4toannexb(fmt_ctx, &pkt, dst_fd);
      
    }
    av_packet_unref(&pkt);
  }
  
  // 关闭流文件
  avformat_close_input(&fmt_ctx);
  if(dst_fd){
    fclose(dst_fd);
  }
  
  return 0;
  
  
}