sdrangel/plugins/channelrx/localsink/localsinksink.cpp

///////////////////////////////////////////////////////////////////////////////////
// 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, 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->runFilt(c, &rf); // 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, bool force)
{
    qDebug() << "LocalSinkSink::applySettings:"
            << " m_localDeviceIndex: " << settings.m_localDeviceIndex
            << " m_streamIndex: " << settings.m_streamIndex
            << " m_dsp: " << settings.m_dsp
            << " m_gaindB: " << settings.m_gaindB
            << " m_fftOn: " << settings.m_fftOn
            << " force: " << force;

    if ((settings.m_gaindB != m_settings.m_gaindB) || force) {
        m_gain = CalcDb::powerFromdB(settings.m_gaindB/2.0); // Amplitude gain
    }

    if ((settings.m_fftOn != m_settings.m_fftOn) || force)
    {
        if (settings.m_fftOn) {
            m_fftFilter->create_filter(m_settings.m_fftBands, true, FFTWindow::Function::Rectangle);
        }
    }

    m_settings = settings;
}

void LocalSinkSink::setSampleRate(int sampleRate)
{
    m_sampleFifo.setSize(SampleSinkFifo::getSizePolicy(sampleRate));
}