audiorecorder.cpp Example File | Qt Multimedia 5.4

- Qt 5.4
- Qt Multimedia
- Audio Recorder Example
- audiorecorder.cpp Example File
- Qt 5.4.2 Reference Documentation

# audiorecorder.cpp Example File

audiorecorder/audiorecorder.cpp

    /****************************************************************************
    **
    ** Copyright (C) 2013 Digia Plc and/or its subsidiary(-ies).
    ** Contact: http://www.qt-project.org/legal
    **
    ** This file is part of the examples of the Qt Toolkit.
    **
    ** $QT_BEGIN_LICENSE:BSD$
    ** You may use this file under the terms of the BSD license as follows:
    **
    ** "Redistribution and use in source and binary forms, with or without
    ** modification, are permitted provided that the following conditions are
    ** met:
    **   * Redistributions of source code must retain the above copyright
    **     notice, this list of conditions and the following disclaimer.
    **   * Redistributions in binary form must reproduce the above copyright
    **     notice, this list of conditions and the following disclaimer in
    **     the documentation and/or other materials provided with the
    **     distribution.
    **   * Neither the name of Digia Plc and its Subsidiary(-ies) nor the names
    **     of its contributors may be used to endorse or promote products derived
    **     from this software without specific prior written permission.
    **
    **
    ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
    ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
    ** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
    ** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
    ** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
    ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
    ** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
    ** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
    ** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
    ** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
    ** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
    **
    ** $QT_END_LICENSE$
    **
    ****************************************************************************/

    #include <QAudioProbe>
    #include <QAudioRecorder>
    #include <QDir>
    #include <QFileDialog>
    #include <QMediaRecorder>

    #include "audiorecorder.h"
    #include "qaudiolevel.h"

    #include "ui_audiorecorder.h"

    static qreal getPeakValue(const QAudioFormat &format);
    static QVector<qreal> getBufferLevels(const QAudioBuffer &buffer);

    template <class T>
    static QVector<qreal> getBufferLevels(const T *buffer, int frames, int channels);

    AudioRecorder::AudioRecorder(QWidget *parent) :
        QMainWindow(parent),
        ui(new Ui::AudioRecorder),
        outputLocationSet(false)
    {
        ui->setupUi(this);

        audioRecorder = new QAudioRecorder(this);
        probe = new QAudioProbe;
        connect(probe, SIGNAL(audioBufferProbed(QAudioBuffer)),
                this, SLOT(processBuffer(QAudioBuffer)));
        probe->setSource(audioRecorder);

        //audio devices
        ui->audioDeviceBox->addItem(tr("Default"), QVariant(QString()));
        foreach (const QString &device, audioRecorder->audioInputs()) {
            ui->audioDeviceBox->addItem(device, QVariant(device));
        }

        //audio codecs
        ui->audioCodecBox->addItem(tr("Default"), QVariant(QString()));
        foreach (const QString &codecName, audioRecorder->supportedAudioCodecs()) {
            ui->audioCodecBox->addItem(codecName, QVariant(codecName));
        }

        //containers
        ui->containerBox->addItem(tr("Default"), QVariant(QString()));
        foreach (const QString &containerName, audioRecorder->supportedContainers()) {
            ui->containerBox->addItem(containerName, QVariant(containerName));
        }

        //sample rate
        ui->sampleRateBox->addItem(tr("Default"), QVariant(0));
        foreach (int sampleRate, audioRecorder->supportedAudioSampleRates()) {
            ui->sampleRateBox->addItem(QString::number(sampleRate), QVariant(
                    sampleRate));
        }

        //channels
        ui->channelsBox->addItem(tr("Default"), QVariant(-1));
        ui->channelsBox->addItem(QStringLiteral("1"), QVariant(1));
        ui->channelsBox->addItem(QStringLiteral("2"), QVariant(2));
        ui->channelsBox->addItem(QStringLiteral("4"), QVariant(4));

        //quality
        ui->qualitySlider->setRange(0, int(QMultimedia::VeryHighQuality));
        ui->qualitySlider->setValue(int(QMultimedia::NormalQuality));

        //bitrates:
        ui->bitrateBox->addItem(tr("Default"), QVariant(0));
        ui->bitrateBox->addItem(QStringLiteral("32000"), QVariant(32000));
        ui->bitrateBox->addItem(QStringLiteral("64000"), QVariant(64000));
        ui->bitrateBox->addItem(QStringLiteral("96000"), QVariant(96000));
        ui->bitrateBox->addItem(QStringLiteral("128000"), QVariant(128000));

        connect(audioRecorder, SIGNAL(durationChanged(qint64)), this,
                SLOT(updateProgress(qint64)));
        connect(audioRecorder, SIGNAL(statusChanged(QMediaRecorder::Status)), this,
                SLOT(updateStatus(QMediaRecorder::Status)));
        connect(audioRecorder, SIGNAL(stateChanged(QMediaRecorder::State)),
                this, SLOT(onStateChanged(QMediaRecorder::State)));
        connect(audioRecorder, SIGNAL(error(QMediaRecorder::Error)), this,
                SLOT(displayErrorMessage()));
    }

    AudioRecorder::~AudioRecorder()
    {
        delete audioRecorder;
        delete probe;
    }

    void AudioRecorder::updateProgress(qint64 duration)
    {
        if (audioRecorder->error() != QMediaRecorder::NoError || duration < 2000)
            return;

        ui->statusbar->showMessage(tr("Recorded %1 sec").arg(duration / 1000));
    }

    void AudioRecorder::updateStatus(QMediaRecorder::Status status)
    {
        QString statusMessage;

        switch (status) {
        case QMediaRecorder::RecordingStatus:
            statusMessage = tr("Recording to %1").arg(audioRecorder->actualLocation().toString());
            break;
        case QMediaRecorder::PausedStatus:
            clearAudioLevels();
            statusMessage = tr("Paused");
            break;
        case QMediaRecorder::UnloadedStatus:
        case QMediaRecorder::LoadedStatus:
            clearAudioLevels();
            statusMessage = tr("Stopped");
        default:
            break;
        }

        if (audioRecorder->error() == QMediaRecorder::NoError)
            ui->statusbar->showMessage(statusMessage);
    }

    void AudioRecorder::onStateChanged(QMediaRecorder::State state)
    {
        switch (state) {
        case QMediaRecorder::RecordingState:
            ui->recordButton->setText(tr("Stop"));
            ui->pauseButton->setText(tr("Pause"));
            break;
        case QMediaRecorder::PausedState:
            ui->recordButton->setText(tr("Stop"));
            ui->pauseButton->setText(tr("Resume"));
            break;
        case QMediaRecorder::StoppedState:
            ui->recordButton->setText(tr("Record"));
            ui->pauseButton->setText(tr("Pause"));
            break;
        }

        ui->pauseButton->setEnabled(audioRecorder->state() != QMediaRecorder::StoppedState);
    }

    static QVariant boxValue(const QComboBox *box)
    {
        int idx = box->currentIndex();
        if (idx == -1)
            return QVariant();

        return box->itemData(idx);
    }

    void AudioRecorder::toggleRecord()
    {
        if (audioRecorder->state() == QMediaRecorder::StoppedState) {
            audioRecorder->setAudioInput(boxValue(ui->audioDeviceBox).toString());

            QAudioEncoderSettings settings;
            settings.setCodec(boxValue(ui->audioCodecBox).toString());
            settings.setSampleRate(boxValue(ui->sampleRateBox).toInt());
            settings.setBitRate(boxValue(ui->bitrateBox).toInt());
            settings.setChannelCount(boxValue(ui->channelsBox).toInt());
            settings.setQuality(QMultimedia::EncodingQuality(ui->qualitySlider->value()));
            settings.setEncodingMode(ui->constantQualityRadioButton->isChecked() ?
                                     QMultimedia::ConstantQualityEncoding :
                                     QMultimedia::ConstantBitRateEncoding);

            QString container = boxValue(ui->containerBox).toString();

            audioRecorder->setEncodingSettings(settings, QVideoEncoderSettings(), container);
            audioRecorder->record();
        }
        else {
            audioRecorder->stop();
        }
    }

    void AudioRecorder::togglePause()
    {
        if (audioRecorder->state() != QMediaRecorder::PausedState)
            audioRecorder->pause();
        else
            audioRecorder->record();
    }

    void AudioRecorder::setOutputLocation()
    {
        QString fileName = QFileDialog::getSaveFileName();
        audioRecorder->setOutputLocation(QUrl::fromLocalFile(fileName));
        outputLocationSet = true;
    }

    void AudioRecorder::displayErrorMessage()
    {
        ui->statusbar->showMessage(audioRecorder->errorString());
    }

    void AudioRecorder::clearAudioLevels()
    {
        for (int i = 0; i < audioLevels.size(); ++i)
            audioLevels.at(i)->setLevel(0);
    }

    // This function returns the maximum possible sample value for a given audio format
    qreal getPeakValue(const QAudioFormat& format)
    {
        // Note: Only the most common sample formats are supported
        if (!format.isValid())
            return qreal(0);

        if (format.codec() != "audio/pcm")
            return qreal(0);

        switch (format.sampleType()) {
        case QAudioFormat::Unknown:
            break;
        case QAudioFormat::Float:
            if (format.sampleSize() != 32) // other sample formats are not supported
                return qreal(0);
            return qreal(1.00003);
        case QAudioFormat::SignedInt:
            if (format.sampleSize() == 32)
                return qreal(INT_MAX);
            if (format.sampleSize() == 16)
                return qreal(SHRT_MAX);
            if (format.sampleSize() == 8)
                return qreal(CHAR_MAX);
            break;
        case QAudioFormat::UnSignedInt:
            if (format.sampleSize() == 32)
                return qreal(UINT_MAX);
            if (format.sampleSize() == 16)
                return qreal(USHRT_MAX);
            if (format.sampleSize() == 8)
                return qreal(UCHAR_MAX);
            break;
        }

        return qreal(0);
    }

    // returns the audio level for each channel
    QVector<qreal> getBufferLevels(const QAudioBuffer& buffer)
    {
        QVector<qreal> values;

        if (!buffer.format().isValid() || buffer.format().byteOrder() != QAudioFormat::LittleEndian)
            return values;

        if (buffer.format().codec() != "audio/pcm")
            return values;

        int channelCount = buffer.format().channelCount();
        values.fill(0, channelCount);
        qreal peak_value = getPeakValue(buffer.format());
        if (qFuzzyCompare(peak_value, qreal(0)))
            return values;

        switch (buffer.format().sampleType()) {
        case QAudioFormat::Unknown:
        case QAudioFormat::UnSignedInt:
            if (buffer.format().sampleSize() == 32)
                values = getBufferLevels(buffer.constData<quint32>(), buffer.frameCount(), channelCount);
            if (buffer.format().sampleSize() == 16)
                values = getBufferLevels(buffer.constData<quint16>(), buffer.frameCount(), channelCount);
            if (buffer.format().sampleSize() == 8)
                values = getBufferLevels(buffer.constData<quint8>(), buffer.frameCount(), channelCount);
            for (int i = 0; i < values.size(); ++i)
                values[i] = qAbs(values.at(i) - peak_value / 2) / (peak_value / 2);
            break;
        case QAudioFormat::Float:
            if (buffer.format().sampleSize() == 32) {
                values = getBufferLevels(buffer.constData<float>(), buffer.frameCount(), channelCount);
                for (int i = 0; i < values.size(); ++i)
                    values[i] /= peak_value;
            }
            break;
        case QAudioFormat::SignedInt:
            if (buffer.format().sampleSize() == 32)
                values = getBufferLevels(buffer.constData<qint32>(), buffer.frameCount(), channelCount);
            if (buffer.format().sampleSize() == 16)
                values = getBufferLevels(buffer.constData<qint16>(), buffer.frameCount(), channelCount);
            if (buffer.format().sampleSize() == 8)
                values = getBufferLevels(buffer.constData<qint8>(), buffer.frameCount(), channelCount);
            for (int i = 0; i < values.size(); ++i)
                values[i] /= peak_value;
            break;
        }

        return values;
    }

    template <class T>
    QVector<qreal> getBufferLevels(const T *buffer, int frames, int channels)
    {
        QVector<qreal> max_values;
        max_values.fill(0, channels);

        for (int i = 0; i < frames; ++i) {
            for (int j = 0; j < channels; ++j) {
                qreal value = qAbs(qreal(buffer[i * channels + j]));
                if (value > max_values.at(j))
                    max_values.replace(j, value);
            }
        }

        return max_values;
    }

    void AudioRecorder::processBuffer(const QAudioBuffer& buffer)
    {
        if (audioLevels.count() != buffer.format().channelCount()) {
            qDeleteAll(audioLevels);
            audioLevels.clear();
            for (int i = 0; i < buffer.format().channelCount(); ++i) {
                QAudioLevel *level = new QAudioLevel(ui->centralwidget);
                audioLevels.append(level);
                ui->levelsLayout->addWidget(level);
            }
        }

        QVector<qreal> levels = getBufferLevels(buffer);
        for (int i = 0; i < levels.count(); ++i)
            audioLevels.at(i)->setLevel(levels.at(i));
    }