mirror of
https://github.com/f4exb/sdrangel.git
synced 2024-12-24 19:00:36 -05:00
219 lines
6.9 KiB
C++
219 lines
6.9 KiB
C++
///////////////////////////////////////////////////////////////////////////////////
|
|
// Copyright (C) 2019 Edouard Griffiths, F4EXB //
|
|
// //
|
|
// This program is free software; you can redistribute it and/or modify //
|
|
// it under the terms of the GNU General Public License as published by //
|
|
// the Free Software Foundation as version 3 of the License, or //
|
|
// (at your option) any later version. //
|
|
// //
|
|
// This program is distributed in the hope that it will be useful, //
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of //
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the //
|
|
// GNU General Public License V3 for more details. //
|
|
// //
|
|
// You should have received a copy of the GNU General Public License //
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>. //
|
|
///////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include <QDebug>
|
|
|
|
#include <boost/crc.hpp>
|
|
#include <boost/cstdint.hpp>
|
|
#include "dsp/devicesamplesource.h"
|
|
#include "dsp/hbfilterchainconverter.h"
|
|
#include "dsp/spectrumvis.h"
|
|
#include "dsp/fftfilt.h"
|
|
#include "util/db.h"
|
|
|
|
#include "localsinkworker.h"
|
|
#include "localsinksink.h"
|
|
|
|
LocalSinkSink::LocalSinkSink() :
|
|
m_deviceSource(nullptr),
|
|
m_sinkWorker(nullptr),
|
|
m_spectrumSink(nullptr),
|
|
m_running(false),
|
|
m_gain(1.0),
|
|
m_centerFrequency(0),
|
|
m_frequencyOffset(0),
|
|
m_sampleRate(48000),
|
|
m_deviceSampleRate(48000)
|
|
{
|
|
m_sampleFifo.setSize(SampleSinkFifo::getSizePolicy(4000000));
|
|
// m_fftFilter = new fftfilt(0.1f, 0.4f, 1<<m_settings.m_log2FFT);
|
|
m_fftFilter = new fftfilt(1<<m_settings.m_log2FFT);
|
|
applySettings(m_settings, QList<QString>(), true);
|
|
}
|
|
|
|
LocalSinkSink::~LocalSinkSink()
|
|
{
|
|
delete m_fftFilter;
|
|
}
|
|
|
|
void LocalSinkSink::feed(const SampleVector::const_iterator& begin, const SampleVector::const_iterator& end)
|
|
{
|
|
if (m_settings.m_dsp && m_settings.m_fftOn)
|
|
{
|
|
fftfilt::cmplx *rf;
|
|
int rf_out;
|
|
|
|
for (SampleVector::const_iterator it = begin; it != end; ++it)
|
|
{
|
|
Complex c(it->real(), it->imag());
|
|
rf_out = m_fftFilter->runAsym(c, &rf, true); // filter RF
|
|
|
|
if (rf_out > 0)
|
|
{
|
|
m_spectrumBuffer.resize(rf_out);
|
|
std::transform(
|
|
rf,
|
|
rf + rf_out,
|
|
m_spectrumBuffer.begin(),
|
|
[this](const fftfilt::cmplx& c) -> Sample {
|
|
return Sample(c.real()*m_gain, c.imag()*m_gain);
|
|
}
|
|
);
|
|
if (m_running && m_deviceSource) {
|
|
m_deviceSource->getSampleFifo()->write(m_spectrumBuffer.begin(), m_spectrumBuffer.begin() + rf_out);
|
|
}
|
|
if (m_spectrumSink) {
|
|
m_spectrumSink->feed(m_spectrumBuffer.begin(), m_spectrumBuffer.begin() + rf_out, false);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else if (m_settings.m_dsp && (m_settings.m_gaindB != 0))
|
|
{
|
|
m_spectrumBuffer.resize(end - begin);
|
|
std::transform(
|
|
begin,
|
|
end,
|
|
m_spectrumBuffer.begin(),
|
|
[this](const Sample& s) -> Sample {
|
|
Complex c(s.real(), s.imag());
|
|
c *= m_gain;
|
|
return Sample(c.real(), c.imag());
|
|
}
|
|
);
|
|
if (m_running && m_deviceSource) {
|
|
m_deviceSource->getSampleFifo()->write(m_spectrumBuffer.begin(), m_spectrumBuffer.end());
|
|
}
|
|
if (m_spectrumSink) {
|
|
m_spectrumSink->feed(m_spectrumBuffer.begin(), m_spectrumBuffer.end(), false);
|
|
}
|
|
m_spectrumBuffer.clear();
|
|
}
|
|
else
|
|
{
|
|
if (m_running && m_deviceSource) {
|
|
m_deviceSource->getSampleFifo()->write(begin, end);
|
|
}
|
|
if (m_spectrumSink) {
|
|
m_spectrumSink->feed(begin, end, false);
|
|
}
|
|
}
|
|
|
|
// m_sampleFifo.write(begin, end);
|
|
}
|
|
|
|
void LocalSinkSink::start(DeviceSampleSource *deviceSource)
|
|
{
|
|
qDebug("LocalSinkSink::start: deviceSource: %p", deviceSource);
|
|
|
|
if (m_running) {
|
|
stop();
|
|
}
|
|
|
|
m_deviceSource = deviceSource;
|
|
// TODO: We'll see later if a worker is really needed
|
|
// m_sinkWorker = new LocalSinkWorker();
|
|
// m_sinkWorker->moveToThread(&m_sinkWorkerThread);
|
|
// m_sinkWorker->setSampleFifo(&m_sampleFifo);
|
|
|
|
// if (deviceSource) {
|
|
// m_sinkWorker->setDeviceSampleFifo(deviceSource->getSampleFifo());
|
|
// }
|
|
|
|
// QObject::connect(
|
|
// &m_sampleFifo,
|
|
// &SampleSinkFifo::dataReady,
|
|
// m_sinkWorker,
|
|
// &LocalSinkWorker::handleData,
|
|
// Qt::QueuedConnection
|
|
// );
|
|
|
|
// startWorker();
|
|
m_running = true;
|
|
}
|
|
|
|
void LocalSinkSink::stop()
|
|
{
|
|
qDebug("LocalSinkSink::stop");
|
|
|
|
// TODO: We'll see later if a worker is really needed
|
|
// QObject::disconnect(
|
|
// &m_sampleFifo,
|
|
// &SampleSinkFifo::dataReady,
|
|
// m_sinkWorker,
|
|
// &LocalSinkWorker::handleData
|
|
// );
|
|
|
|
// if (m_sinkWorker != 0)
|
|
// {
|
|
// stopWorker();
|
|
// m_sinkWorker->deleteLater();
|
|
// m_sinkWorker = nullptr;
|
|
// }
|
|
|
|
m_running = false;
|
|
m_deviceSource = nullptr;
|
|
}
|
|
|
|
void LocalSinkSink::startWorker()
|
|
{
|
|
m_sinkWorker->startStop(true);
|
|
m_sinkWorkerThread.start();
|
|
}
|
|
|
|
void LocalSinkSink::stopWorker()
|
|
{
|
|
m_sinkWorker->startStop(false);
|
|
m_sinkWorkerThread.quit();
|
|
m_sinkWorkerThread.wait();
|
|
}
|
|
|
|
void LocalSinkSink::applySettings(const LocalSinkSettings& settings, const QList<QString>& settingsKeys, bool force)
|
|
{
|
|
qDebug() << "LocalSinkSink::applySettings:" << settings.getDebugString(settingsKeys, force) << " force: " << force;
|
|
|
|
if (settingsKeys.contains("gaindB") || force) {
|
|
m_gain = CalcDb::powerFromdB(settings.m_gaindB/2.0); // Amplitude gain
|
|
}
|
|
|
|
if (settingsKeys.contains("log2FFT") || force)
|
|
{
|
|
delete m_fftFilter;
|
|
m_fftFilter = new fftfilt(1<<settings.m_log2FFT);
|
|
m_fftFilter->create_filter(m_settings.m_fftBands, true, m_settings.m_fftWindow);
|
|
}
|
|
|
|
if (settingsKeys.contains("fftWindow")
|
|
|| settingsKeys.contains("fftBands")
|
|
|| settingsKeys.contains("reverseFilter")
|
|
|| force)
|
|
{
|
|
m_fftFilter->create_filter(settings.m_fftBands, !settings.m_reverseFilter, settings.m_fftWindow);
|
|
}
|
|
|
|
if (force) {
|
|
m_settings = settings;
|
|
} else {
|
|
m_settings.applySettings(settingsKeys, settings);
|
|
}
|
|
}
|
|
|
|
void LocalSinkSink::setSampleRate(int sampleRate)
|
|
{
|
|
m_sampleFifo.setSize(SampleSinkFifo::getSizePolicy(sampleRate));
|
|
}
|