mirror of
https://github.com/f4exb/sdrangel.git
synced 2024-10-01 17:26:40 -04:00
477 lines
16 KiB
C++
477 lines
16 KiB
C++
///////////////////////////////////////////////////////////////////////////////////
|
|
// Copyright (C) 2019 F4EXB //
|
|
// written by Edouard Griffiths //
|
|
// //
|
|
// 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 "dsp/dspcommands.h"
|
|
#include "threadedbasebandsamplesource.h"
|
|
#include "threadedbasebandsamplesink.h"
|
|
#include "devicesamplemimo.h"
|
|
|
|
#include "dspdevicemimoengine.h"
|
|
|
|
DSPDeviceMIMOEngine::DSPDeviceMIMOEngine(uint32_t uid, QObject* parent) :
|
|
QThread(parent),
|
|
m_uid(uid),
|
|
m_state(StNotStarted)
|
|
{
|
|
connect(&m_inputMessageQueue, SIGNAL(messageEnqueued()), this, SLOT(handleInputMessages()), Qt::QueuedConnection);
|
|
connect(&m_syncMessenger, SIGNAL(messageSent()), this, SLOT(handleSynchronousMessages()), Qt::QueuedConnection);
|
|
|
|
moveToThread(this);
|
|
}
|
|
|
|
DSPDeviceMIMOEngine::~DSPDeviceMIMOEngine()
|
|
{
|
|
stop();
|
|
wait();
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::run()
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::run";
|
|
m_state = StIdle;
|
|
exec();
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::start()
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::start";
|
|
QThread::start();
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::stop()
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::stop";
|
|
gotoIdle();
|
|
m_state = StNotStarted;
|
|
QThread::exit();
|
|
}
|
|
|
|
bool DSPDeviceMIMOEngine::initProcess()
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::initGeneration";
|
|
DSPGenerationInit cmd;
|
|
|
|
return m_syncMessenger.sendWait(cmd) == StReady;
|
|
}
|
|
|
|
bool DSPDeviceMIMOEngine::startProcess()
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::startGeneration";
|
|
DSPGenerationStart cmd;
|
|
|
|
return m_syncMessenger.sendWait(cmd) == StRunning;
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::stopProcess()
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::stopGeneration";
|
|
DSPGenerationStop cmd;
|
|
m_syncMessenger.storeMessage(cmd);
|
|
handleSynchronousMessages();
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::setMIMO(DeviceSampleMIMO* mimo)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::setSink";
|
|
SetSampleMIMO cmd(mimo);
|
|
m_syncMessenger.sendWait(cmd);
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::setMIMOSequence(int sequence)
|
|
{
|
|
qDebug("DSPDeviceMIMOEngine::setSinkSequence: seq: %d", sequence);
|
|
m_sampleMIMOSequence = sequence;
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::addChannelSource(ThreadedBasebandSampleSource* source, int index)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::addThreadedSource: "
|
|
<< source->objectName().toStdString().c_str()
|
|
<< " at: "
|
|
<< index;
|
|
AddThreadedBasebandSampleSource cmd(source, index);
|
|
m_syncMessenger.sendWait(cmd);
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::removeChannelSource(ThreadedBasebandSampleSource* source, int index)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::removeThreadedSource: "
|
|
<< source->objectName().toStdString().c_str()
|
|
<< " at: "
|
|
<< index;
|
|
RemoveThreadedBasebandSampleSource cmd(source, index);
|
|
m_syncMessenger.sendWait(cmd);
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::addChannelSink(ThreadedBasebandSampleSink* sink, int index)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::addThreadedSink: "
|
|
<< sink->objectName().toStdString().c_str()
|
|
<< " at: "
|
|
<< index;
|
|
AddThreadedBasebandSampleSink cmd(sink, index);
|
|
m_syncMessenger.sendWait(cmd);
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::removeChannelSink(ThreadedBasebandSampleSink* sink, int index)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::removeThreadedSink: "
|
|
<< sink->objectName().toStdString().c_str()
|
|
<< " at: "
|
|
<< index;
|
|
RemoveThreadedBasebandSampleSink cmd(sink, index);
|
|
m_syncMessenger.sendWait(cmd);
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::addAncillarySink(BasebandSampleSink* sink, int index)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::addSink: "
|
|
<< sink->objectName().toStdString().c_str()
|
|
<< " at: "
|
|
<< index;
|
|
AddBasebandSampleSink cmd(sink, index);
|
|
m_syncMessenger.sendWait(cmd);
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::removeAncillarySink(BasebandSampleSink* sink, int index)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::removeSink: "
|
|
<< sink->objectName().toStdString().c_str()
|
|
<< " at: "
|
|
<< index;
|
|
RemoveBasebandSampleSink cmd(sink, index);
|
|
m_syncMessenger.sendWait(cmd);
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::addSpectrumSink(BasebandSampleSink* spectrumSink)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::addSpectrumSink: " << spectrumSink->objectName().toStdString().c_str();
|
|
AddSpectrumSink cmd(spectrumSink);
|
|
m_syncMessenger.sendWait(cmd);
|
|
}
|
|
|
|
void DSPDeviceMIMOEngine::removeSpectrumSink(BasebandSampleSink* spectrumSink)
|
|
{
|
|
qDebug() << "DSPDeviceSinkEngine::removeSpectrumSink: " << spectrumSink->objectName().toStdString().c_str();
|
|
DSPRemoveSpectrumSink cmd(spectrumSink);
|
|
m_syncMessenger.sendWait(cmd);
|
|
}
|
|
|
|
QString DSPDeviceMIMOEngine::errorMessage()
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::errorMessage";
|
|
GetErrorMessage cmd;
|
|
m_syncMessenger.sendWait(cmd);
|
|
return cmd.getErrorMessage();
|
|
}
|
|
|
|
QString DSPDeviceMIMOEngine::deviceDescription()
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::deviceDescription";
|
|
GetSinkDeviceDescription cmd;
|
|
m_syncMessenger.sendWait(cmd);
|
|
return cmd.getDeviceDescription();
|
|
}
|
|
|
|
/**
|
|
* Routes samples from device source FIFO to sink channels that are registered for the FIFO
|
|
* Routes samples from source channels registered for the FIFO to the device sink FIFO
|
|
*/
|
|
void DSPDeviceMIMOEngine::work(int nbWriteSamples, int nbReadSamples)
|
|
{
|
|
// Sources
|
|
for (unsigned int isource = 0; isource < m_deviceSampleMIMO->getNbSourceFifos(); isource++)
|
|
{
|
|
if (isource >= m_sourceStreamSampleRates.size()) {
|
|
continue;
|
|
}
|
|
|
|
SampleSinkFifo* sampleFifo = m_deviceSampleMIMO->getSampleSinkFifo(isource); // sink FIFO is for Rx
|
|
std::size_t samplesDone = 0;
|
|
bool positiveOnly = false;
|
|
|
|
while ((sampleFifo->fill() > 0) && (m_inputMessageQueue.size() == 0) && (samplesDone < m_sourceStreamSampleRates[isource]))
|
|
{
|
|
SampleVector::iterator part1begin;
|
|
SampleVector::iterator part1end;
|
|
SampleVector::iterator part2begin;
|
|
SampleVector::iterator part2end;
|
|
|
|
std::size_t count = sampleFifo->readBegin(sampleFifo->fill(), &part1begin, &part1end, &part2begin, &part2end);
|
|
|
|
// first part of FIFO data
|
|
if (part1begin != part1end)
|
|
{
|
|
// TODO: DC and IQ corrections
|
|
|
|
// feed data to direct sinks
|
|
if (isource < m_basebandSampleSinks.size())
|
|
{
|
|
for (BasebandSampleSinks::const_iterator it = m_basebandSampleSinks[isource].begin(); it != m_basebandSampleSinks[isource].end(); ++it) {
|
|
(*it)->feed(part1begin, part1end, positiveOnly);
|
|
}
|
|
}
|
|
|
|
// feed data to threaded sinks
|
|
if (isource < m_threadedBasebandSampleSinks.size())
|
|
{
|
|
for (ThreadedBasebandSampleSinks::const_iterator it = m_threadedBasebandSampleSinks[isource].begin(); it != m_threadedBasebandSampleSinks[isource].end(); ++it) {
|
|
(*it)->feed(part1begin, part1end, positiveOnly);
|
|
}
|
|
}
|
|
}
|
|
|
|
// second part of FIFO data (used when block wraps around)
|
|
if(part2begin != part2end)
|
|
{
|
|
// TODO: DC and IQ corrections
|
|
|
|
// feed data to direct sinks
|
|
if (isource < m_basebandSampleSinks.size())
|
|
{
|
|
for (BasebandSampleSinks::const_iterator it = m_basebandSampleSinks[isource].begin(); it != m_basebandSampleSinks[isource].end(); ++it) {
|
|
(*it)->feed(part2begin, part2end, positiveOnly);
|
|
}
|
|
}
|
|
|
|
// feed data to threaded sinks
|
|
if (isource < m_threadedBasebandSampleSinks.size())
|
|
{
|
|
for (ThreadedBasebandSampleSinks::const_iterator it = m_threadedBasebandSampleSinks[isource].begin(); it != m_threadedBasebandSampleSinks[isource].end(); ++it) {
|
|
(*it)->feed(part2begin, part2end, positiveOnly);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
// adjust FIFO pointers
|
|
sampleFifo->readCommit((unsigned int) count);
|
|
samplesDone += count;
|
|
} // while stream FIFO
|
|
} // for stream source
|
|
|
|
// TODO: sinks
|
|
}
|
|
|
|
// notStarted -> idle -> init -> running -+
|
|
// ^ |
|
|
// +-----------------------+
|
|
|
|
DSPDeviceMIMOEngine::State DSPDeviceMIMOEngine::gotoIdle()
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::gotoIdle";
|
|
|
|
switch(m_state) {
|
|
case StNotStarted:
|
|
return StNotStarted;
|
|
|
|
case StIdle:
|
|
case StError:
|
|
return StIdle;
|
|
|
|
case StReady:
|
|
case StRunning:
|
|
break;
|
|
}
|
|
|
|
if (m_deviceSampleMIMO == 0)
|
|
{
|
|
return StIdle;
|
|
}
|
|
|
|
// stop everything
|
|
|
|
std::vector<BasebandSampleSinks>::const_iterator vbit = m_basebandSampleSinks.begin();
|
|
|
|
for (; vbit != m_basebandSampleSinks.end(); ++vbit)
|
|
{
|
|
for (BasebandSampleSinks::const_iterator it = vbit->begin(); it != vbit->end(); ++it) {
|
|
(*it)->stop();
|
|
}
|
|
}
|
|
|
|
std::vector<ThreadedBasebandSampleSinks>::const_iterator vtit = m_threadedBasebandSampleSinks.begin();
|
|
|
|
for (; vtit != m_threadedBasebandSampleSinks.end(); vtit++)
|
|
{
|
|
for (ThreadedBasebandSampleSinks::const_iterator it = vtit->begin(); it != vtit->end(); ++it) {
|
|
(*it)->stop();
|
|
}
|
|
}
|
|
|
|
m_deviceSampleMIMO->stop();
|
|
m_deviceDescription.clear();
|
|
|
|
for (std::vector<uint32_t>::iterator it = m_sourceStreamSampleRates.begin(); it != m_sourceStreamSampleRates.end(); ++it) {
|
|
*it = 0;
|
|
}
|
|
|
|
return StIdle;
|
|
}
|
|
|
|
DSPDeviceMIMOEngine::State DSPDeviceMIMOEngine::gotoInit()
|
|
{
|
|
switch(m_state) {
|
|
case StNotStarted:
|
|
return StNotStarted;
|
|
|
|
case StRunning: // FIXME: assumes it goes first through idle state. Could we get back to init from running directly?
|
|
return StRunning;
|
|
|
|
case StReady:
|
|
return StReady;
|
|
|
|
case StIdle:
|
|
case StError:
|
|
break;
|
|
}
|
|
|
|
if (m_deviceSampleMIMO == 0) {
|
|
return gotoError("No sample MIMO configured");
|
|
}
|
|
|
|
// init: pass sample rate and center frequency to all sample rate and/or center frequency dependent sinks and wait for completion
|
|
|
|
|
|
m_deviceDescription = m_deviceSampleMIMO->getDeviceDescription();
|
|
|
|
qDebug() << "DSPDeviceMIMOEngine::gotoInit: "
|
|
<< " m_deviceDescription: " << m_deviceDescription.toStdString().c_str();
|
|
|
|
for (unsigned int isource = 0; isource < m_deviceSampleMIMO->getNbSourceFifos(); isource++)
|
|
{
|
|
if (isource < m_sourcesCorrections.size())
|
|
{
|
|
m_sourcesCorrections[isource].m_iOffset = 0;
|
|
m_sourcesCorrections[isource].m_qOffset = 0;
|
|
m_sourcesCorrections[isource].m_iRange = 1 << 16;
|
|
m_sourcesCorrections[isource].m_qRange = 1 << 16;
|
|
}
|
|
|
|
if ((isource < m_sourceCenterFrequencies.size()) && (isource < m_sourceStreamSampleRates.size()))
|
|
{
|
|
m_sourceCenterFrequencies[isource] = m_deviceSampleMIMO->getSourceCenterFrequency(isource);
|
|
m_sourceStreamSampleRates[isource] = m_deviceSampleMIMO->getSourceSampleRate(isource);
|
|
|
|
qDebug("DSPDeviceMIMOEngine::gotoInit: m_sourceCenterFrequencies[%d] = %llu", isource, m_sourceCenterFrequencies[isource]);
|
|
qDebug("DSPDeviceMIMOEngine::gotoInit: m_sourceStreamSampleRates[%d] = %d", isource, m_sourceStreamSampleRates[isource]);
|
|
|
|
DSPSignalNotification notif(m_sourceStreamSampleRates[isource], m_sourceCenterFrequencies[isource]);
|
|
|
|
if (isource < m_basebandSampleSinks.size())
|
|
{
|
|
for (BasebandSampleSinks::const_iterator it = m_basebandSampleSinks[isource].begin(); it != m_basebandSampleSinks[isource].end(); ++it)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::gotoInit: initializing " << (*it)->objectName().toStdString().c_str();
|
|
(*it)->handleMessage(notif);
|
|
}
|
|
}
|
|
|
|
if (isource < m_threadedBasebandSampleSinks.size())
|
|
{
|
|
for (ThreadedBasebandSampleSinks::const_iterator it = m_threadedBasebandSampleSinks[isource].begin(); it != m_threadedBasebandSampleSinks[isource].end(); ++it)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::gotoInit: initializing ThreadedSampleSink(" << (*it)->getSampleSinkObjectName().toStdString().c_str() << ")";
|
|
(*it)->handleSinkMessage(notif);
|
|
}
|
|
}
|
|
|
|
// pass data to listeners
|
|
// if (m_deviceSampleSource->getMessageQueueToGUI())
|
|
// {
|
|
// DSPSignalNotification* rep = new DSPSignalNotification(notif); // make a copy for the output queue
|
|
// m_deviceSampleSource->getMessageQueueToGUI()->push(rep);
|
|
// }
|
|
}
|
|
}
|
|
|
|
return StReady;
|
|
}
|
|
|
|
DSPDeviceMIMOEngine::State DSPDeviceMIMOEngine::gotoRunning()
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::gotoRunning";
|
|
|
|
switch(m_state)
|
|
{
|
|
case StNotStarted:
|
|
return StNotStarted;
|
|
|
|
case StIdle:
|
|
return StIdle;
|
|
|
|
case StRunning:
|
|
return StRunning;
|
|
|
|
case StReady:
|
|
case StError:
|
|
break;
|
|
}
|
|
|
|
if (!m_deviceSampleMIMO) {
|
|
return gotoError("DSPDeviceMIMOEngine::gotoRunning: No sample source configured");
|
|
}
|
|
|
|
qDebug() << "DSPDeviceMIMOEngine::gotoRunning: " << m_deviceDescription.toStdString().c_str() << " started";
|
|
|
|
// Start everything
|
|
|
|
if (!m_deviceSampleMIMO->start())
|
|
{
|
|
return gotoError("Could not start sample source");
|
|
}
|
|
|
|
std::vector<BasebandSampleSinks>::const_iterator vbit = m_basebandSampleSinks.begin();
|
|
|
|
for (; vbit != m_basebandSampleSinks.end(); ++vbit)
|
|
{
|
|
for (BasebandSampleSinks::const_iterator it = vbit->begin(); it != vbit->end(); ++it)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::gotoRunning: starting " << (*it)->objectName().toStdString().c_str();
|
|
(*it)->start();
|
|
}
|
|
}
|
|
|
|
std::vector<ThreadedBasebandSampleSinks>::const_iterator vtit = m_threadedBasebandSampleSinks.begin();
|
|
|
|
for (; vtit != m_threadedBasebandSampleSinks.end(); vtit++)
|
|
{
|
|
for (ThreadedBasebandSampleSinks::const_iterator it = vtit->begin(); it != vtit->end(); ++it)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::gotoRunning: starting ThreadedSampleSink(" << (*it)->getSampleSinkObjectName().toStdString().c_str() << ")";
|
|
(*it)->start();
|
|
}
|
|
}
|
|
|
|
qDebug() << "DSPDeviceMIMOEngine::gotoRunning:input message queue pending: " << m_inputMessageQueue.size();
|
|
|
|
return StRunning;
|
|
}
|
|
|
|
DSPDeviceMIMOEngine::State DSPDeviceMIMOEngine::gotoError(const QString& errorMessage)
|
|
{
|
|
qDebug() << "DSPDeviceMIMOEngine::gotoError: " << errorMessage;
|
|
|
|
m_errorMessage = errorMessage;
|
|
m_deviceDescription.clear();
|
|
m_state = StError;
|
|
return StError;
|
|
}
|