diff options
Diffstat (limited to 'examples/multimedia/spectrum/spectrumanalyser.cpp')
-rw-r--r-- | examples/multimedia/spectrum/spectrumanalyser.cpp | 150 |
1 files changed, 50 insertions, 100 deletions
diff --git a/examples/multimedia/spectrum/spectrumanalyser.cpp b/examples/multimedia/spectrum/spectrumanalyser.cpp index 4a7b06809..c309c6a73 100644 --- a/examples/multimedia/spectrum/spectrumanalyser.cpp +++ b/examples/multimedia/spectrum/spectrumanalyser.cpp @@ -1,75 +1,31 @@ -/**************************************************************************** -** -** Copyright (C) 2017 The Qt Company Ltd. -** Contact: https://www.qt.io/licensing/ -** -** This file is part of the examples of the Qt Toolkit. -** -** $QT_BEGIN_LICENSE:BSD$ -** Commercial License Usage -** Licensees holding valid commercial Qt licenses may use this file in -** accordance with the commercial license agreement provided with the -** Software or, alternatively, in accordance with the terms contained in -** a written agreement between you and The Qt Company. For licensing terms -** and conditions see https://www.qt.io/terms-conditions. For further -** information use the contact form at https://www.qt.io/contact-us. -** -** BSD License Usage -** Alternatively, 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 The Qt Company Ltd 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$ -** -****************************************************************************/ +// Copyright (C) 2017 The Qt Company Ltd. +// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause #include "spectrumanalyser.h" -#include "utils.h" #include "fftreal_wrapper.h" +#include "utils.h" -#include <qmath.h> -#include <qmetatype.h> #include <QAudioFormat> +#include <QMetaType> #include <QThread> +#include <QtMath> SpectrumAnalyserThread::SpectrumAnalyserThread(QObject *parent) - : QObject(parent) + : QObject(parent) #ifndef DISABLE_FFT - , m_fft(new FFTRealWrapper) + , + m_fft(new FFTRealWrapper) #endif - , m_numSamples(SpectrumLengthSamples) - , m_windowFunction(DefaultWindowFunction) - , m_window(SpectrumLengthSamples, 0.0) - , m_input(SpectrumLengthSamples, 0.0) - , m_output(SpectrumLengthSamples, 0.0) - , m_spectrum(SpectrumLengthSamples) + , + m_numSamples(SpectrumLengthSamples), + m_windowFunction(DefaultWindowFunction), + m_window(SpectrumLengthSamples, 0.0), + m_input(SpectrumLengthSamples, 0.0), + m_output(SpectrumLengthSamples, 0.0), + m_spectrum(SpectrumLengthSamples) #ifdef SPECTRUM_ANALYSER_SEPARATE_THREAD - , m_thread(new QThread(this)) + , + m_thread(new QThread(this)) #endif { #ifdef SPECTRUM_ANALYSER_SEPARATE_THREAD @@ -96,7 +52,7 @@ void SpectrumAnalyserThread::setWindowFunction(WindowFunction type) void SpectrumAnalyserThread::calculateWindow() { - for (int i=0; i<m_numSamples; ++i) { + for (int i = 0; i < m_numSamples; ++i) { DataType x = 0.0; switch (m_windowFunction) { @@ -114,17 +70,16 @@ void SpectrumAnalyserThread::calculateWindow() } } -void SpectrumAnalyserThread::calculateSpectrum(const QByteArray &buffer, - int inputFrequency, - int bytesPerFrame) +void SpectrumAnalyserThread::calculateSpectrum(const QByteArray &buffer, int inputFrequency, + int bytesPerFrame) { #ifndef DISABLE_FFT Q_ASSERT(buffer.size() == m_numSamples * bytesPerFrame); // Initialize data array const char *ptr = buffer.constData(); - for (int i=0; i<m_numSamples; ++i) { - const qint16 pcmSample = *reinterpret_cast<const qint16*>(ptr); + for (int i = 0; i < m_numSamples; ++i) { + const qint16 pcmSample = *reinterpret_cast<const qint16 *>(ptr); // Scale down to range [-1.0, 1.0] const DataType realSample = pcmToReal(pcmSample); const DataType windowedSample = realSample * m_window[i]; @@ -136,16 +91,16 @@ void SpectrumAnalyserThread::calculateSpectrum(const QByteArray &buffer, m_fft->calculateFFT(m_output.data(), m_input.data()); // Analyze output to obtain amplitude and phase for each frequency - for (int i=2; i<=m_numSamples/2; ++i) { + for (int i = 2; i <= m_numSamples / 2; ++i) { // Calculate frequency of this complex sample m_spectrum[i].frequency = qreal(i * inputFrequency) / (m_numSamples); const qreal real = m_output[i]; qreal imag = 0.0; - if (i>0 && i<m_numSamples/2) - imag = m_output[m_numSamples/2 + i]; + if (i > 0 && i < m_numSamples / 2) + imag = m_output[m_numSamples / 2 + i]; - const qreal magnitude = qSqrt(real*real + imag*imag); + const qreal magnitude = qSqrt(real * real + imag * imag); qreal amplitude = SpectrumAnalyserMultiplier * qLn(magnitude); // Bound amplitude to [0.0, 1.0] @@ -159,21 +114,21 @@ void SpectrumAnalyserThread::calculateSpectrum(const QByteArray &buffer, emit calculationComplete(m_spectrum); } - //============================================================================= // SpectrumAnalyser //============================================================================= SpectrumAnalyser::SpectrumAnalyser(QObject *parent) - : QObject(parent) - , m_thread(new SpectrumAnalyserThread(this)) - , m_state(Idle) + : QObject(parent), + m_thread(new SpectrumAnalyserThread(this)), + m_state(Idle) #ifdef DUMP_SPECTRUMANALYSER - , m_count(0) + , + m_count(0) #endif { - connect(m_thread, &SpectrumAnalyserThread::calculationComplete, - this, &SpectrumAnalyser::calculationComplete); + connect(m_thread, &SpectrumAnalyserThread::calculationComplete, this, + &SpectrumAnalyser::calculationComplete); } SpectrumAnalyser::~SpectrumAnalyser() = default; @@ -194,21 +149,18 @@ void SpectrumAnalyser::setOutputPath(const QString &outputDir) void SpectrumAnalyser::setWindowFunction(WindowFunction type) { - const bool b = QMetaObject::invokeMethod(m_thread, "setWindowFunction", - Qt::AutoConnection, - Q_ARG(WindowFunction, type)); + const bool b = QMetaObject::invokeMethod(m_thread, "setWindowFunction", Qt::AutoConnection, + Q_ARG(WindowFunction, type)); Q_ASSERT(b); Q_UNUSED(b); // suppress warnings in release builds } -void SpectrumAnalyser::calculate(const QByteArray &buffer, - const QAudioFormat &format) +void SpectrumAnalyser::calculate(const QByteArray &buffer, const QAudioFormat &format) { // QThread::currentThread is marked 'for internal use only', but // we're only using it for debug output here, so it's probably OK :) - SPECTRUMANALYSER_DEBUG << "SpectrumAnalyser::calculate" - << QThread::currentThread() - << "state" << m_state; + SPECTRUMANALYSER_DEBUG << "SpectrumAnalyser::calculate" << QThread::currentThread() << "state" + << m_state; if (isReady()) { Q_ASSERT(format.sampleFormat() == QAudioFormat::Int16); @@ -217,15 +169,16 @@ void SpectrumAnalyser::calculate(const QByteArray &buffer, #ifdef DUMP_SPECTRUMANALYSER m_count++; - const QString pcmFileName = m_outputDir.filePath(QString("spectrum_%1.pcm").arg(m_count, 4, 10, QChar('0'))); + const QString pcmFileName = + m_outputDir.filePath(QStringLiteral("spectrum_%1.pcm").arg(m_count, 4, 10, QChar('0'))); QFile pcmFile(pcmFileName); pcmFile.open(QIODevice::WriteOnly); const int bufferLength = m_numSamples * bytesPerFrame; pcmFile.write(buffer, bufferLength); m_textStream << "TimeDomain " << m_count << "\n"; - const qint16* input = reinterpret_cast<const qint16*>(buffer); - for (int i=0; i<m_numSamples; ++i) { + const qint16 *input = reinterpret_cast<const qint16 *>(buffer); + for (int i = 0; i < m_numSamples; ++i) { m_textStream << i << "\t" << *input << "\n"; input += format.channels(); } @@ -238,22 +191,18 @@ void SpectrumAnalyser::calculate(const QByteArray &buffer, // calculation will be done in the child thread. // Once the calculation is finished, a calculationChanged signal will be // emitted by m_thread. - const bool b = QMetaObject::invokeMethod(m_thread, "calculateSpectrum", - Qt::AutoConnection, - Q_ARG(QByteArray, buffer), - Q_ARG(int, format.sampleRate()), - Q_ARG(int, bytesPerFrame)); + const bool b = QMetaObject::invokeMethod( + m_thread, "calculateSpectrum", Qt::AutoConnection, Q_ARG(QByteArray, buffer), + Q_ARG(int, format.sampleRate()), Q_ARG(int, bytesPerFrame)); Q_ASSERT(b); Q_UNUSED(b); // suppress warnings in release builds #ifdef DUMP_SPECTRUMANALYSER m_textStream << "FrequencySpectrum " << m_count << "\n"; FrequencySpectrum::const_iterator x = m_spectrum.begin(); - for (int i=0; i<m_numSamples; ++i, ++x) - m_textStream << i << "\t" - << x->frequency << "\t" - << x->amplitude<< "\t" - << x->phase << "\n"; + for (int i = 0; i < m_numSamples; ++i, ++x) + m_textStream << i << "\t" << x->frequency << "\t" << x->amplitude << "\t" << x->phase + << "\n"; #endif } } @@ -269,7 +218,6 @@ void SpectrumAnalyser::cancelCalculation() m_state = Cancelled; } - //----------------------------------------------------------------------------- // Private slots //----------------------------------------------------------------------------- @@ -281,3 +229,5 @@ void SpectrumAnalyser::calculationComplete(const FrequencySpectrum &spectrum) emit spectrumChanged(spectrum); m_state = Idle; } + +#include "moc_spectrumanalyser.cpp" |