ffmpeg 音频数据采集

#include <QtCore/QCoreApplication> #include <stdio.h> #include <QString> extern "C" { #include "libavcodec/avcodec.h" #include "libavformat/avformat.h" #include "libswscale/swscale.h" #include "libavdevice/avdevice.h" } #include <qDebug> #include <QTextCodec> //Show Dshow Device void show_dshow_device(){ AVFormatContext *pFormatCtx = avformat_alloc_context(); AVDictionary* options = NULL; av_dict_set(&options,"list_devices","true",0); AVInputFormat *iformat = av_find_input_format("dshow"); printf("========Device Info=============\n"); avformat_open_input(&pFormatCtx,"video=dummy",iformat,&options); printf("================================\n"); } //Show Dshow Device Option void show_dshow_device_option(){ AVFormatContext *pFormatCtx = avformat_alloc_context(); AVDictionary* options = NULL; av_dict_set(&options,"list_options","true",0); AVInputFormat *iformat = av_find_input_format("dshow"); printf("========Device Option Info======\n"); avformat_open_input(&pFormatCtx,"video=Integrated Camera",iformat,&options); printf("================================\n"); } //Show AVFoundation Device void show_avfoundation_device(){ AVFormatContext *pFormatCtx = avformat_alloc_context(); AVDictionary* options = NULL; av_dict_set(&options,"list_devices","true",0); AVInputFormat *iformat = av_find_input_format("avfoundation"); printf("==AVFoundation Device Info===\n"); avformat_open_input(&pFormatCtx,"",iformat,&options); printf("=============================\n"); } #define QStringLiteral(str) QString::fromUtf8(str, sizeof(str) - 1) int main(int argc, char *argv[]) { QCoreApplication a(argc, argv); QTextCodec::setCodecForLocale(QTextCodec::codecForName("UTF-8"));//支持中文 AVFormatContext *pFormatCtx; int i, audioindex; AVCodecContext *aCodecCtx; AVCodec *aCodec = NULL; AVFrame *aFrame; av_register_all(); // avformat_network_init(); //初始化网络 avdevice_register_all();//Register Device //Show Dshow Device show_dshow_device(); //Show Device Options // show_dshow_device_option(); //Show VFW Options // show_vfw_device(); pFormatCtx = avformat_alloc_context(); QString audioDeviceName = QStringLiteral("audio=麦克风 (Realtek High Definition Audio)"); // QString audioDeviceName = QString("audio=麦克风 (Realtek High Definition Audio)"); AVInputFormat *ifmt = av_find_input_format("dshow"); QByteArray qbArray = audioDeviceName.toUtf8(); char* audioName = qbArray.data(); if(avformat_open_input(&pFormatCtx,audioName,ifmt,NULL)!=0){ fprintf(stderr,"Couldn't open input stream.（无法打开输入流）\n"); return -1; } audioindex = -1; aCodecCtx = NULL; for(i=0; i<pFormatCtx->nb_streams; i++) if(pFormatCtx->streams[i]->codec->codec_type==AVMEDIA_TYPE_AUDIO) { audioindex=i; break; } if(audioindex==-1) { printf("Didn't find a video stream.（没有找到视频流）\n"); return -1; } aCodecCtx = pFormatCtx->streams[audioindex]->codec; aCodec = avcodec_find_decoder(aCodecCtx->codec_id); if(aCodec == NULL) { printf("audio Codec not found.\n"); return -1; } if(avcodec_open2(aCodecCtx, aCodec,NULL)<0) { printf("Could not open video codec.\n"); return -1; } aFrame=avcodec_alloc_frame(); int ret, got_frame; AVPacket *packet=(AVPacket *)av_malloc(sizeof(AVPacket)); FILE *fp_pcm=fopen("output.pcm","wb"); ///这里打印出音频的信息 qDebug()<<"audio info:"; qDebug()<<"audio info:"<<aCodecCtx->sample_fmt<<aCodecCtx->bit_rate<<aCodecCtx->sample_rate<<aCodecCtx->channels; float Time = 0; for(int i=0;;i++) { if (Time > 30) break; //就采集30秒 if(av_read_frame(pFormatCtx, packet) < 0) { break; } if(packet->stream_index==audioindex) { ret = avcodec_decode_audio4(aCodecCtx, aFrame, &got_frame, packet); // qDebug()<<i<<ret<<audioindex; if(ret < 0) { fprintf(stderr,"Audio Error.\n"); return -1; } if (got_frame) { int pcmSize = av_samples_get_buffer_size(NULL,aCodecCtx->channels, aFrame->nb_samples,aCodecCtx->sample_fmt, 1); uint8_t * pcmBuffer = aFrame->data[0]; float useTime = aFrame->nb_samples * 1.0 / aCodecCtx->sample_rate; Time += useTime; qDebug()<<i<<Time<<useTime; fwrite(pcmBuffer,1,pcmSize,fp_pcm); //写入文件 } } av_free_packet(packet); } fclose(fp_pcm); if (aFrame != NULL) { av_free(aFrame); aFrame = NULL; } if (aCodecCtx != NULL) avcodec_close(aCodecCtx); avformat_close_input(&pFormatCtx); avformat_free_context(pFormatCtx); return a.exec(); }