美文网首页
QT开发 FFmpeg录制音频 wav格式

QT开发 FFmpeg录制音频 wav格式

作者: 我有一只小毛驴_从来都不骑 | 来源:发表于2021-08-10 18:54 被阅读0次
     // 获取输入格式对象
        AVInputFormat *fmt = av_find_input_format("avfoundation");
        if (!fmt) {
            qDebug() << "获取输入格式对象失败" << FMT_NAME;
            return;
        }
    
        // 格式上下文(将来可以利用上下文操作设备)
        AVFormatContext *ctx = nullptr;
        // 打开设备
        int ret = avformat_open_input(&ctx,":2", fmt, nullptr);
        if (ret < 0) {
            char errbuf[1024];
            av_strerror(ret, errbuf, sizeof (errbuf));
            qDebug() << "打开设备失败" << errbuf;
            return;
        }
        // 文件名
        QString filename = "/Users/apple/Desktop/";
        filename += QDateTime::currentDateTime().toString("MM_dd_HH_mm_ss");
        filename += ".wav";
        QFile file(filename);
    
        // 打开文件
        if (!file.open(QFile::WriteOnly)) {
            qDebug() << "文件打开失败" << filename;
    
            // 关闭设备
            avformat_close_input(&ctx);
            return;
        }
    
        // 获取输入流
        AVStream *stream = ctx->streams[0];
        // 获取音频参数
        AVCodecParameters *params = stream->codecpar;
    
        // 写入WAV文件头
        WAVHeader header;
        header.sampleRate = params->sample_rate;
        // 2
        header.bitsPerSample = av_get_bits_per_sample(params->codec_id);
        header.numChannels = params->channels;
        if (params->codec_id >= AV_CODEC_ID_PCM_F32BE) {
            header.audioFormat = AUDIO_FORMAT_FLOAT;
        }
        header.blockAlign = header.bitsPerSample * header.numChannels >> 3;
        header.byteRate = header.sampleRate * header.blockAlign;
    //    header.dataChunkDataSize = 0;
        file.write((char *) &header, sizeof (WAVHeader));
    
        // 数据包
        AVPacket *pkt = av_packet_alloc();
        while (!isInterruptionRequested()) {
            // 不断采集数据
            ret = av_read_frame(ctx, pkt);
    
            if (ret == 0) { // 读取成功
                // 将数据写入文件
                file.write((const char *) pkt->data, pkt->size);
    
                // 计算录音时长
               // header.dataChunkDataSize += pkt->size;
               // unsigned long long ms = 1000.0 * header.dataChunkDataSize / header.byteRate;
    
                // 释放资源
                av_packet_unref(pkt);
            } else if (ret == AVERROR(EAGAIN)) { // 资源临时不可用
                continue;
            } else { // 其他错误
                char errbuf[1024];
                av_strerror(ret, errbuf, sizeof (errbuf));
                qDebug() << "av_read_frame error" << errbuf << ret;
                break;
            }
        }
    
    //    qDebug() << file.size() << header.dataChunkDataSize;
    // 写入dataChunkDataSize
    //    header.dataChunkDataSize = size - sizeof (WAVHeader);
        file.seek(sizeof (WAVHeader) - sizeof (header.dataChunkDataSize));
        file.write((char *) &header.dataChunkDataSize, sizeof (header.dataChunkDataSize));
    
        // 写入riffChunkDataSize
        header.riffChunkDataSize = file.size()
                                   - sizeof (header.riffChunkId)
                                   - sizeof (header.riffChunkDataSize);
        file.seek(sizeof (header.riffChunkId));
        file.write((char *) &header.riffChunkDataSize, sizeof (header.riffChunkDataSize));
    
        // 释放资源
        av_packet_free(&pkt);
    
        // 关闭文件
        file.close();
    
        // 关闭设备
        avformat_close_input(&ctx);
    
        qDebug() << this << "正常结束----------";
    

    相关文章

      网友评论

          本文标题:QT开发 FFmpeg录制音频 wav格式

          本文链接:https://www.haomeiwen.com/subject/nzdfbltx.html