IOS 音视频 ffmpeg 录音并播放

// // test.c // av_recoder // // Created by 蔡磊 on 2023/4/25. // #include "test.h" void test(void){ printf("this is a c foundtion test\n"); av_log_set_level(AV_LOG_DEBUG); av_log(NULL, AV_LOG_DEBUG, "this is av log\n"); } void changeRecordStatu(int status){ rec_status = status; } // 打开编码器 AVCodecContext* openAvCodec(void){ // 找到编码器 // avcodec_find_encoder(AV_CODEC_ID_AAC); const AVCodec *codec = avcodec_find_encoder_by_name("libfdk-aac"); AVCodecContext * codecContext = avcodec_alloc_context3(codec); codecContext->sample_fmt = AV_SAMPLE_FMT_FLTP; codecContext->channel_layout =AV_CH_LAYOUT_STEREO; codecContext->sample_rate = 96000; codecContext->codec_type = AVMEDIA_TYPE_AUDIO; codecContext->ch_layout.nb_channels = 1; codecContext->bit_rate = 64000; // aac_lc 128k aac_he 64k aac_he_v2 32k // codecContext->profile = FF_PROFILE_AAC_HE_V2; // 打开编码器 int ret = avcodec_open2(codecContext, codec, NULL); if(ret) { printf("open avcodec failed \n"); char errbuf[128] = { 0 }; const char *errbuf_ptr = errbuf; if (av_strerror(ret, errbuf, sizeof(errbuf)) < 0) printf(errbuf); } return codecContext; } SwrContext* init_swr(void){ // 创建重采样上下文 SwrContext *swr_ctx = NULL; swr_ctx = swr_alloc_set_opts(NULL, AV_CH_LAYOUT_STEREO, AV_SAMPLE_FMT_S16, 44100, AV_CH_LAYOUT_STEREO, AV_SAMPLE_FMT_FLT, 44100, 0, NULL); if(!swr_ctx) { } if(swr_init(swr_ctx)<0) { } return swr_ctx; } void record(void){ // if(recording){ // return; // } char errors[1024]; int ret; int count = 0; AVFormatContext *fmt_ctx = avformat_alloc_context(); char *deviceName = ":0"; // 注册设备 avdevice_register_all(); // get format const AVInputFormat *inputFormat = av_find_input_format("avfoundation"); // 重采样上下文 SwrContext* swr_ctx = init_swr(); // aac 编码器上下文 AVCodecContext *codeContext = openAvCodec(); AVFrame *frame = av_frame_alloc(); frame->format = AV_SAMPLE_FMT_FLTP; frame->channel_layout = AV_CH_LAYOUT_STEREO; frame->nb_samples = 512; // 单通道每秒采样数 av_frame_get_buffer(frame, 0); if(!frame || !frame->buf[0]) { } AVPacket *newPkt = av_packet_alloc(); if(!newPkt) { } // 打开设备 if((ret = avformat_open_input(&fmt_ctx, deviceName, inputFormat, NULL) < 0)) { av_strerror(ret, errors, 1024); fprintf(stderr,"Failed to open audio device,[%d]%s\n",ret,errors); return; } // 读取数据 AVPacket *avPacket = av_packet_alloc(); // 创建文件夹保存数据 char *out = "/Users/cailei/Downloads/audio.aac"; FILE *outfile = fopen(out, "wb+"); rec_status = 1; uint8_t **src_data = NULL; int src_linesize = 0; uint8_t **out_data = NULL; int out_linesize = 0; // 创建输入缓冲区 av_samples_alloc_array_and_samples(&src_data, &src_linesize, 1, 512, AV_SAMPLE_FMT_FLT, 0); // 创建输出缓冲区 av_samples_alloc_array_and_samples(&out_data, &out_linesize, 1, 512, AV_SAMPLE_FMT_S16, 0); while(((ret = av_read_frame(fmt_ctx, avPacket)) == 0 || ret == -35) && rec_status) { if(ret == -35) { usleep(100); continue; } printf("pkt size is %d \n",avPacket->size); // 进行内存拷贝 memcpy((void*)src_data[0], (void*)avPacket->data, avPacket->size); count++; // 重采样开始 swr_convert(swr_ctx, out_data, 512, (const uint8_t **)src_data, 512); // 将重采样的数据拷贝到avframe中 memcpy((void*)frame->data[0], (void*)out_data[0], out_linesize); // 将数据送到编码器 ret = avcodec_send_frame(codeContext, frame); while(ret >= 0) { // 获取编码后的音频数据 ret = avcodec_receive_packet(codeContext, newPkt); if(ret < 0) { if(ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) { // 编码器没有数据，或者编码器数据不够编码的数据了 break; }else if(ret < 0) { printf("Error, encoding audio frame\n"); exit(-1); } } fwrite(newPkt,1,newPkt->size, outfile); fflush(outfile); } av_packet_unref(avPacket); } av_packet_free(&avPacket); fclose(outfile); // 释放输入输出缓冲区 if(src_data) { av_freep(&src_data[0]); } av_freep(src_data); if(out_data) { av_freep(&out_data[0]); } av_freep(out_data); // 释放重采样上下文 swr_free(&swr_ctx); avformat_close_input(&fmt_ctx); }