mirror of
https://github.com/f4exb/sdrangel.git
synced 2024-11-02 16:01:14 -04:00
427 lines
14 KiB
C++
427 lines
14 KiB
C++
///////////////////////////////////////////////////////////////////////////////////
|
|
// Copyright (C) 2017 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 //
|
|
// //
|
|
// 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/filerecord.h"
|
|
#include "dsp/dspcommands.h"
|
|
#include "device/devicesourceapi.h"
|
|
#include "device/devicesinkapi.h"
|
|
#include "plutosdr/deviceplutosdrparams.h"
|
|
#include "plutosdr/deviceplutosdrbox.h"
|
|
|
|
#include "plutosdrinput.h"
|
|
#include "plutosdrinputthread.h"
|
|
|
|
#define PLUTOSDR_BLOCKSIZE (1024*1024) //complex samples per buffer
|
|
|
|
MESSAGE_CLASS_DEFINITION(PlutoSDRInput::MsgFileRecord, Message)
|
|
|
|
PlutoSDRInput::PlutoSDRInput(DeviceSourceAPI *deviceAPI) :
|
|
m_deviceAPI(deviceAPI),
|
|
m_fileSink(0),
|
|
m_deviceDescription("PlutoSDR"),
|
|
m_running(false),
|
|
m_plutoRxBuffer(0),
|
|
m_plutoSDRInputThread(0)
|
|
{
|
|
char recFileNameCStr[30];
|
|
sprintf(recFileNameCStr, "test_%d.sdriq", m_deviceAPI->getDeviceUID());
|
|
m_fileSink = new FileRecord(std::string(recFileNameCStr));
|
|
m_deviceAPI->addSink(m_fileSink);
|
|
}
|
|
|
|
|
|
PlutoSDRInput::~PlutoSDRInput()
|
|
{
|
|
m_deviceAPI->removeSink(m_fileSink);
|
|
delete m_fileSink;
|
|
}
|
|
|
|
bool PlutoSDRInput::start()
|
|
{
|
|
if (!m_deviceShared.m_deviceParams->getBox()) {
|
|
return false;
|
|
}
|
|
|
|
if (m_running) stop();
|
|
|
|
applySettings(m_settings, true);
|
|
|
|
// start / stop streaming is done in the thread.
|
|
|
|
if ((m_plutoSDRInputThread = new PlutoSDRInputThread(PLUTOSDR_BLOCKSIZE, m_deviceShared.m_deviceParams->getBox(), &m_sampleFifo)) == 0)
|
|
{
|
|
qFatal("PlutoSDRInput::start: cannot create thread");
|
|
stop();
|
|
return false;
|
|
}
|
|
else
|
|
{
|
|
qDebug("PlutoSDRInput::start: thread created");
|
|
}
|
|
|
|
m_plutoSDRInputThread->setLog2Decimation(m_settings.m_log2Decim);
|
|
m_plutoSDRInputThread->startWork();
|
|
|
|
m_deviceShared.m_thread = m_plutoSDRInputThread;
|
|
m_running = true;
|
|
|
|
return true;
|
|
}
|
|
|
|
void PlutoSDRInput::stop()
|
|
{
|
|
if (m_plutoSDRInputThread != 0)
|
|
{
|
|
m_plutoSDRInputThread->stopWork();
|
|
delete m_plutoSDRInputThread;
|
|
m_plutoSDRInputThread = 0;
|
|
}
|
|
|
|
m_deviceShared.m_thread = 0;
|
|
m_running = false;
|
|
}
|
|
|
|
const QString& PlutoSDRInput::getDeviceDescription() const
|
|
{
|
|
return m_deviceDescription;
|
|
}
|
|
int PlutoSDRInput::getSampleRate() const
|
|
{
|
|
return (m_settings.m_devSampleRate / (1<<m_settings.m_log2Decim));
|
|
}
|
|
|
|
quint64 PlutoSDRInput::getCenterFrequency() const
|
|
{
|
|
return m_settings.m_centerFrequency;
|
|
}
|
|
|
|
|
|
bool PlutoSDRInput::handleMessage(const Message& message)
|
|
{
|
|
if (MsgConfigurePlutoSDR::match(message))
|
|
{
|
|
MsgConfigurePlutoSDR& conf = (MsgConfigurePlutoSDR&) message;
|
|
qDebug() << "PlutoSDRInput::handleMessage: MsgConfigurePlutoSDR";
|
|
|
|
if (!applySettings(conf.getSettings(), conf.getForce()))
|
|
{
|
|
qDebug("PlutoSDRInput::handleMessage config error");
|
|
}
|
|
|
|
return true;
|
|
}
|
|
else if (MsgFileRecord::match(message))
|
|
{
|
|
MsgFileRecord& conf = (MsgFileRecord&) message;
|
|
qDebug() << "PlutoSDRInput::handleMessage: MsgFileRecord: " << conf.getStartStop();
|
|
|
|
if (conf.getStartStop()) {
|
|
m_fileSink->startRecording();
|
|
} else {
|
|
m_fileSink->stopRecording();
|
|
}
|
|
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
bool PlutoSDRInput::openDevice()
|
|
{
|
|
if (!m_sampleFifo.setSize(96000 * 4))
|
|
{
|
|
qCritical("PlutoSDRInput::openDevice: could not allocate SampleFifo");
|
|
return false;
|
|
}
|
|
else
|
|
{
|
|
qDebug("PlutoSDRInput::openDevice: allocated SampleFifo");
|
|
}
|
|
|
|
// look for Tx buddy and get reference to common parameters
|
|
if (m_deviceAPI->getSinkBuddies().size() > 0) // then sink
|
|
{
|
|
qDebug("PlutoSDRInput::openDevice: look at Tx buddy");
|
|
|
|
DeviceSinkAPI *sinkBuddy = m_deviceAPI->getSinkBuddies()[0];
|
|
m_deviceShared = *((DevicePlutoSDRShared *) sinkBuddy->getBuddySharedPtr()); // copy parameters
|
|
|
|
if (m_deviceShared.m_deviceParams == 0)
|
|
{
|
|
qCritical("PlutoSDRInput::openDevice: cannot get device parameters from Tx buddy");
|
|
return false; // the device params should have been created by the buddy
|
|
}
|
|
else
|
|
{
|
|
qDebug("PlutoSDRInput::openDevice: getting device parameters from Tx buddy");
|
|
}
|
|
}
|
|
// There is no buddy then create the first PlutoSDR common parameters
|
|
// open the device this will also populate common fields
|
|
else
|
|
{
|
|
qDebug("PlutoSDRInput::openDevice: open device here");
|
|
|
|
m_deviceShared.m_deviceParams = new DevicePlutoSDRParams();
|
|
char serial[256];
|
|
strcpy(serial, qPrintable(m_deviceAPI->getSampleSourceSerial()));
|
|
m_deviceShared.m_deviceParams->open(serial);
|
|
}
|
|
|
|
m_deviceAPI->setBuddySharedPtr(&m_deviceShared); // propagate common parameters to API
|
|
|
|
// acquire the channel
|
|
DevicePlutoSDRBox *plutoBox = m_deviceShared.m_deviceParams->getBox();
|
|
plutoBox->openRx();
|
|
m_plutoRxBuffer = plutoBox->createRxBuffer(PLUTOSDR_BLOCKSIZE, false);
|
|
|
|
return true;
|
|
}
|
|
|
|
void PlutoSDRInput::closeDevice()
|
|
{
|
|
if (m_deviceShared.m_deviceParams->getBox() == 0) { // was never open
|
|
return;
|
|
}
|
|
|
|
if (m_deviceAPI->getSinkBuddies().size() == 0)
|
|
{
|
|
m_deviceShared.m_deviceParams->close();
|
|
delete m_deviceShared.m_deviceParams;
|
|
m_deviceShared.m_deviceParams = 0;
|
|
}
|
|
}
|
|
|
|
void PlutoSDRInput::suspendBuddies()
|
|
{
|
|
// suspend Tx buddy's thread
|
|
|
|
for (unsigned int i = 0; i < m_deviceAPI->getSinkBuddies().size(); i++)
|
|
{
|
|
DeviceSinkAPI *buddy = m_deviceAPI->getSinkBuddies()[i];
|
|
DevicePlutoSDRShared *buddyShared = (DevicePlutoSDRShared *) buddy->getBuddySharedPtr();
|
|
|
|
if (buddyShared->m_thread) {
|
|
buddyShared->m_thread->stopWork();
|
|
}
|
|
}
|
|
}
|
|
|
|
void PlutoSDRInput::resumeBuddies()
|
|
{
|
|
// resume Tx buddy's thread
|
|
|
|
for (unsigned int i = 0; i < m_deviceAPI->getSinkBuddies().size(); i++)
|
|
{
|
|
DeviceSinkAPI *buddy = m_deviceAPI->getSinkBuddies()[i];
|
|
DevicePlutoSDRShared *buddyShared = (DevicePlutoSDRShared *) buddy->getBuddySharedPtr();
|
|
|
|
if (buddyShared->m_thread) {
|
|
buddyShared->m_thread->startWork();
|
|
}
|
|
}
|
|
}
|
|
|
|
bool PlutoSDRInput::applySettings(const PlutoSDRInputSettings& settings, bool force)
|
|
{
|
|
bool forwardChangeOwnDSP = false;
|
|
bool forwardChangeOtherDSP = false;
|
|
bool suspendOwnThread = false;
|
|
bool ownThreadWasRunning = false;
|
|
bool suspendAllOtherThreads = false; // All others means Tx in fact
|
|
bool doCalibration = false;
|
|
bool firFilterSet = false;
|
|
DevicePlutoSDRBox *plutoBox = m_deviceShared.m_deviceParams->getBox();
|
|
|
|
// determine if buddies threads or own thread need to be suspended
|
|
|
|
// change of global baseband sample rate affecting all buddies can occur if
|
|
// - device to host sample rate is changed
|
|
// - rate governor is changed
|
|
// - FIR filter decimation is changed
|
|
// - FIR filter is enabled or disabled
|
|
if ((m_settings.m_devSampleRate != settings.m_devSampleRate) ||
|
|
(m_settings.m_rateGovernor != settings.m_rateGovernor) ||
|
|
(m_settings.m_lpfFIRlog2Decim != settings.m_lpfFIRlog2Decim) ||
|
|
(m_settings.m_lpfFIREnable != settings.m_lpfFIREnable) || force)
|
|
{
|
|
suspendAllOtherThreads = true;
|
|
suspendOwnThread = true;
|
|
}
|
|
else
|
|
{
|
|
suspendOwnThread = true;
|
|
}
|
|
|
|
if (suspendAllOtherThreads)
|
|
{
|
|
const std::vector<DeviceSinkAPI*>& sinkBuddies = m_deviceAPI->getSinkBuddies();
|
|
std::vector<DeviceSinkAPI*>::const_iterator itSink = sinkBuddies.begin();
|
|
|
|
for (; itSink != sinkBuddies.end(); ++itSink)
|
|
{
|
|
DevicePlutoSDRShared *buddySharedPtr = (DevicePlutoSDRShared *) (*itSink)->getBuddySharedPtr();
|
|
|
|
if (buddySharedPtr->m_thread) {
|
|
buddySharedPtr->m_thread->stopWork();
|
|
buddySharedPtr->m_threadWasRunning = true;
|
|
}
|
|
else
|
|
{
|
|
buddySharedPtr->m_threadWasRunning = false;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (suspendOwnThread)
|
|
{
|
|
if (m_plutoSDRInputThread && m_plutoSDRInputThread->isRunning())
|
|
{
|
|
m_plutoSDRInputThread->stopWork();
|
|
ownThreadWasRunning = true;
|
|
}
|
|
}
|
|
|
|
// TODO: apply settings (all cases)
|
|
|
|
// Change affecting device sample rate chain potentially affecting other buddies device/host sample rate
|
|
if ((m_settings.m_devSampleRate != settings.m_devSampleRate) ||
|
|
(m_settings.m_rateGovernor != settings.m_rateGovernor) ||
|
|
(m_settings.m_lpfFIRlog2Decim != settings.m_lpfFIRlog2Decim) ||
|
|
(m_settings.m_lpfFIREnable != settings.m_lpfFIREnable) || force)
|
|
{
|
|
std::vector<std::string> params;
|
|
params.push_back(QString(tr("in_voltage_sampling_frequency=%1").arg(settings.m_devSampleRate)).toStdString());
|
|
QString rateGovStr;
|
|
PlutoSDRInputSettings::translateGovernor(settings.m_rateGovernor, rateGovStr);
|
|
params.push_back(QString(tr("trx_rate_governor=%1").arg(rateGovStr)).toStdString());
|
|
plutoBox->set_params(DevicePlutoSDRBox::DEVICE_PHY, params); // set end point frequency and rate governor first
|
|
|
|
// Unless unconditionally forced to change settings change FIR settings if FIR is enabled AND
|
|
// - FIR was not enabled before OR
|
|
// - Host interface sample rate has changed OR
|
|
// - Decimation chain rate governor has changed OR
|
|
// - FIR decimation has changed
|
|
if ((settings.m_lpfFIREnable && (!m_settings.m_lpfFIREnable
|
|
|| (m_settings.m_devSampleRate != settings.m_devSampleRate)
|
|
|| (m_settings.m_rateGovernor != settings.m_rateGovernor)
|
|
|| (m_settings.m_lpfFIRlog2Decim != settings.m_lpfFIRlog2Decim))) || force)
|
|
{
|
|
plutoBox->setFIR(DevicePlutoSDRBox::USE_RX, (1<<settings.m_lpfFIRlog2Decim), settings.m_lpfFIRBW, settings.m_lpfFIRGain);
|
|
firFilterSet = true;
|
|
}
|
|
|
|
// Reset the filter if gain or bandwidth changes without change to any of the sample rates and not done at previous step
|
|
if (((settings.m_lpfFIRBW != m_settings.m_lpfFIRBW) ||
|
|
(settings.m_lpfFIRGain != m_settings.m_lpfFIRGain)) && !firFilterSet)
|
|
{
|
|
plutoBox->setFIR(DevicePlutoSDRBox::USE_RX, (1<<settings.m_lpfFIRlog2Decim), settings.m_lpfFIRBW, settings.m_lpfFIRGain);
|
|
firFilterSet = true;
|
|
}
|
|
|
|
if ((m_settings.m_lpfFIREnable != settings.m_lpfFIREnable) || force)
|
|
{
|
|
params.clear();
|
|
params.push_back(QString(tr("in_voltage_filter_fir_en=%1").arg(settings.m_lpfFIREnable ? 1 : 0)).toStdString());
|
|
plutoBox->set_params(DevicePlutoSDRBox::DEVICE_PHY, params); // eventually enable/disable FIR
|
|
}
|
|
|
|
plutoBox->getRxSampleRates(m_deviceSampleRates); // pick up possible new rates
|
|
qDebug() << "PlutoSDRInput::applySettings: BBPLL: " << m_deviceSampleRates.m_bbRate
|
|
<< " ADC: " << m_deviceSampleRates.m_addaConnvRate
|
|
<< " -HB3-> " << m_deviceSampleRates.m_hb3Rate
|
|
<< " -HB2-> " << m_deviceSampleRates.m_hb2Rate
|
|
<< " -HB1-> " << m_deviceSampleRates.m_hb1Rate
|
|
<< " -FIR-> " << m_deviceSampleRates.m_firRate;
|
|
|
|
forwardChangeOtherDSP = true;
|
|
forwardChangeOwnDSP = (m_settings.m_devSampleRate != settings.m_devSampleRate);
|
|
}
|
|
|
|
if ((m_settings.m_log2Decim != settings.m_log2Decim) || force)
|
|
{
|
|
if (m_plutoSDRInputThread != 0)
|
|
{
|
|
m_plutoSDRInputThread->setLog2Decimation(settings.m_log2Decim);
|
|
qDebug() << "PlutoSDRInput::applySettings: set soft decimation to " << (1<<m_settings.m_log2Decim);
|
|
}
|
|
|
|
forwardChangeOwnDSP = true;
|
|
}
|
|
|
|
if ((m_settings.m_centerFrequency != settings.m_centerFrequency) || force)
|
|
{
|
|
std::vector<std::string> params;
|
|
params.push_back(QString(tr("in_voltage_sampling_frequency=%1").arg(settings.m_centerFrequency)).toStdString());
|
|
plutoBox->set_params(DevicePlutoSDRBox::DEVICE_PHY, params);
|
|
|
|
forwardChangeOwnDSP = true;
|
|
}
|
|
|
|
m_settings = settings;
|
|
|
|
// TODO: calibration
|
|
if (doCalibration)
|
|
{
|
|
qDebug("PlutoSDRInput::applySettings: doCalibration");
|
|
}
|
|
|
|
if (suspendAllOtherThreads)
|
|
{
|
|
const std::vector<DeviceSinkAPI*>& sinkBuddies = m_deviceAPI->getSinkBuddies();
|
|
std::vector<DeviceSinkAPI*>::const_iterator itSink = sinkBuddies.begin();
|
|
|
|
for (; itSink != sinkBuddies.end(); ++itSink)
|
|
{
|
|
DevicePlutoSDRShared *buddySharedPtr = (DevicePlutoSDRShared *) (*itSink)->getBuddySharedPtr();
|
|
|
|
if (buddySharedPtr->m_threadWasRunning) {
|
|
buddySharedPtr->m_thread->startWork();
|
|
}
|
|
}
|
|
}
|
|
|
|
if (suspendOwnThread)
|
|
{
|
|
if (ownThreadWasRunning) {
|
|
m_plutoSDRInputThread->startWork();
|
|
}
|
|
}
|
|
|
|
// TODO: forward changes to other (Tx) DSP
|
|
if (forwardChangeOtherDSP)
|
|
{
|
|
qDebug("PlutoSDRInput::applySettings: forwardChangeOtherDSP");
|
|
}
|
|
|
|
if (forwardChangeOwnDSP)
|
|
{
|
|
qDebug("PlutoSDRInput::applySettings: forward change to self");
|
|
|
|
int sampleRate = m_settings.m_devSampleRate/(1<<m_settings.m_log2Decim);
|
|
DSPSignalNotification *notif = new DSPSignalNotification(sampleRate, m_settings.m_centerFrequency);
|
|
m_fileSink->handleMessage(*notif); // forward to file sink
|
|
m_deviceAPI->getDeviceInputMessageQueue()->push(notif);
|
|
}
|
|
|
|
return false;
|
|
}
|