mirror of
https://github.com/f4exb/sdrangel.git
synced 2024-11-03 07:21:14 -05:00
251 lines
8.0 KiB
C++
251 lines
8.0 KiB
C++
#include <QTcpServer>
|
|
#include <QTcpSocket>
|
|
#include <QThread>
|
|
#include "tcpsrc.h"
|
|
#include "tcpsrcgui.h"
|
|
#include "dsp/dspcommands.h"
|
|
|
|
MESSAGE_CLASS_DEFINITION(TCPSrc::MsgTCPSrcConfigure, Message)
|
|
MESSAGE_CLASS_DEFINITION(TCPSrc::MsgTCPSrcConnection, Message)
|
|
MESSAGE_CLASS_DEFINITION(TCPSrc::MsgTCPSrcSpectrum, Message)
|
|
|
|
TCPSrc::TCPSrc(MessageQueue* uiMessageQueue, TCPSrcGUI* tcpSrcGUI, SampleSink* spectrum)
|
|
{
|
|
m_inputSampleRate = 96000;
|
|
m_sampleFormat = FormatSSB;
|
|
m_outputSampleRate = 48000;
|
|
m_rfBandwidth = 32000;
|
|
m_tcpPort = 9999;
|
|
m_nco.setFreq(0, m_inputSampleRate);
|
|
m_interpolator.create(16, m_inputSampleRate, m_rfBandwidth / 2.1);
|
|
m_sampleDistanceRemain = m_inputSampleRate / m_outputSampleRate;
|
|
m_uiMessageQueue = uiMessageQueue;
|
|
m_tcpSrcGUI = tcpSrcGUI;
|
|
m_spectrum = spectrum;
|
|
m_spectrumEnabled = false;
|
|
m_nextSSBId = 0;
|
|
m_nextS16leId = 0;
|
|
|
|
m_last = 0;
|
|
m_this = 0;
|
|
m_scale = 20000;
|
|
m_sampleBufferSSB.resize(tcpFftLen);
|
|
TCPFilter = new fftfilt(0.001, 16.0 / 48.0, tcpFftLen);
|
|
// if (!TCPFilter) segfault;
|
|
}
|
|
|
|
TCPSrc::~TCPSrc()
|
|
{
|
|
}
|
|
|
|
void TCPSrc::configure(MessageQueue* messageQueue, SampleFormat sampleFormat, Real outputSampleRate, Real rfBandwidth, int tcpPort, int boost)
|
|
{
|
|
Message* cmd = MsgTCPSrcConfigure::create(sampleFormat, outputSampleRate, rfBandwidth, tcpPort, boost);
|
|
cmd->submit(messageQueue, this);
|
|
}
|
|
|
|
void TCPSrc::setSpectrum(MessageQueue* messageQueue, bool enabled)
|
|
{
|
|
Message* cmd = MsgTCPSrcSpectrum::create(enabled);
|
|
cmd->submit(messageQueue, this);
|
|
}
|
|
|
|
void TCPSrc::feed(SampleVector::const_iterator begin, SampleVector::const_iterator end, bool positiveOnly)
|
|
{
|
|
Complex ci;
|
|
cmplx* sideband;
|
|
Real l, r;
|
|
|
|
m_sampleBuffer.clear();
|
|
|
|
// Rtl-Sdr uses full 16-bit scale; FCDPP does not
|
|
int rescale = 20000 * m_boost * m_boost;
|
|
|
|
for(SampleVector::const_iterator it = begin; it < end; ++it) {
|
|
Complex c(it->real() / 32768.0, it->imag() / 32768.0);
|
|
c *= m_nco.nextIQ();
|
|
|
|
if(m_interpolator.interpolate(&m_sampleDistanceRemain, c, &ci)) {
|
|
m_sampleBuffer.push_back(Sample(ci.real() * rescale, ci.imag() * rescale));
|
|
m_sampleDistanceRemain += m_inputSampleRate / m_outputSampleRate;
|
|
}
|
|
}
|
|
|
|
if((m_spectrum != NULL) && (m_spectrumEnabled))
|
|
m_spectrum->feed(m_sampleBuffer.begin(), m_sampleBuffer.end(), positiveOnly);
|
|
|
|
for(int i = 0; i < m_s16leSockets.count(); i++)
|
|
m_s16leSockets[i].socket->write((const char*)&m_sampleBuffer[0], m_sampleBuffer.size() * 4);
|
|
|
|
if((m_sampleFormat == FormatSSB) && (m_ssbSockets.count() > 0)) {
|
|
for(SampleVector::const_iterator it = m_sampleBuffer.begin(); it != m_sampleBuffer.end(); ++it) {
|
|
Complex cj(it->real() / 30000.0, it->imag() / 30000.0);
|
|
int n_out = TCPFilter->runSSB(cj, &sideband, true);
|
|
if (n_out) {
|
|
for (int i = 0; i < n_out; i+=2) {
|
|
l = (sideband[i].real() + sideband[i].imag()) * 0.7 * 32000.0;
|
|
r = (sideband[i+1].real() + sideband[i+1].imag()) * 0.7 * 32000.0;
|
|
m_sampleBufferSSB.push_back(Sample(l, r));
|
|
}
|
|
for(int i = 0; i < m_ssbSockets.count(); i++)
|
|
m_ssbSockets[i].socket->write((const char*)&m_sampleBufferSSB[0], n_out * 2);
|
|
m_sampleBufferSSB.clear();
|
|
}
|
|
}
|
|
}
|
|
|
|
if((m_sampleFormat == FormatNFM) && (m_ssbSockets.count() > 0)) {
|
|
for(SampleVector::const_iterator it = m_sampleBuffer.begin(); it != m_sampleBuffer.end(); ++it) {
|
|
Complex cj(it->real() / 30000.0, it->imag() / 30000.0);
|
|
int n_out = TCPFilter->runFilt(cj, &sideband);
|
|
if (n_out) {
|
|
Real sum = 1.0;
|
|
for (int i = 0; i < n_out; i+=2) {
|
|
l = m_this.real() * (m_last.imag() - sideband[i].imag())
|
|
- m_this.imag() * (m_last.real() - sideband[i].real());
|
|
m_last = sideband[i];
|
|
r = m_last.real() * (m_this.imag() - sideband[i+1].imag())
|
|
- m_last.imag() * (m_this.real() - sideband[i+1].real());
|
|
m_this = sideband[i+1];
|
|
m_sampleBufferSSB.push_back(Sample(l * m_scale, r * m_scale));
|
|
sum += m_this.real() * m_this.real() + m_this.imag() * m_this.imag();
|
|
}
|
|
m_scale = 0.8 * m_scale + 0.2 * 20000 * tcpFftLen / sum;
|
|
for(int i = 0; i < m_ssbSockets.count(); i++)
|
|
m_ssbSockets[i].socket->write((const char*)&m_sampleBufferSSB[0], n_out * 2);
|
|
m_sampleBufferSSB.clear();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void TCPSrc::start()
|
|
{
|
|
m_tcpServer = new QTcpServer();
|
|
connect(m_tcpServer, SIGNAL(newConnection()), this, SLOT(onNewConnection()));
|
|
m_tcpServer->listen(QHostAddress::Any, m_tcpPort);
|
|
}
|
|
|
|
void TCPSrc::stop()
|
|
{
|
|
closeAllSockets(&m_ssbSockets);
|
|
closeAllSockets(&m_s16leSockets);
|
|
|
|
if(m_tcpServer->isListening())
|
|
m_tcpServer->close();
|
|
delete m_tcpServer;
|
|
}
|
|
|
|
bool TCPSrc::handleMessage(Message* cmd)
|
|
{
|
|
if(DSPSignalNotification::match(cmd)) {
|
|
DSPSignalNotification* signal = (DSPSignalNotification*)cmd;
|
|
qDebug("%d samples/sec, %lld Hz offset", signal->getSampleRate(), signal->getFrequencyOffset());
|
|
m_inputSampleRate = signal->getSampleRate();
|
|
m_nco.setFreq(-signal->getFrequencyOffset(), m_inputSampleRate);
|
|
m_interpolator.create(16, m_inputSampleRate, m_rfBandwidth / 2.1);
|
|
m_sampleDistanceRemain = m_inputSampleRate / m_outputSampleRate;
|
|
cmd->completed();
|
|
return true;
|
|
} else if(MsgTCPSrcConfigure::match(cmd)) {
|
|
MsgTCPSrcConfigure* cfg = (MsgTCPSrcConfigure*)cmd;
|
|
m_sampleFormat = cfg->getSampleFormat();
|
|
m_outputSampleRate = cfg->getOutputSampleRate();
|
|
m_rfBandwidth = cfg->getRFBandwidth();
|
|
if(cfg->getTCPPort() != m_tcpPort) {
|
|
m_tcpPort = cfg->getTCPPort();
|
|
if(m_tcpServer->isListening())
|
|
m_tcpServer->close();
|
|
m_tcpServer->listen(QHostAddress::Any, m_tcpPort);
|
|
}
|
|
m_boost = cfg->getBoost();
|
|
m_interpolator.create(16, m_inputSampleRate, m_rfBandwidth / 2.1);
|
|
m_sampleDistanceRemain = m_inputSampleRate / m_outputSampleRate;
|
|
TCPFilter->create_filter(0.048f / 48.0f, m_rfBandwidth / 2.0 / m_outputSampleRate);
|
|
cmd->completed();
|
|
return true;
|
|
} else if(MsgTCPSrcSpectrum::match(cmd)) {
|
|
MsgTCPSrcSpectrum* spc = (MsgTCPSrcSpectrum*)cmd;
|
|
m_spectrumEnabled = spc->getEnabled();
|
|
cmd->completed();
|
|
return true;
|
|
} else {
|
|
if(m_spectrum != NULL)
|
|
return m_spectrum->handleMessage(cmd);
|
|
else return false;
|
|
}
|
|
}
|
|
|
|
void TCPSrc::closeAllSockets(Sockets* sockets)
|
|
{
|
|
for(int i = 0; i < sockets->count(); ++i) {
|
|
MsgTCPSrcConnection* msg = MsgTCPSrcConnection::create(false, sockets->at(i).id, QHostAddress(), 0);
|
|
msg->submit(m_uiMessageQueue, (PluginGUI*)m_tcpSrcGUI);
|
|
sockets->at(i).socket->close();
|
|
}
|
|
}
|
|
|
|
void TCPSrc::onNewConnection()
|
|
{
|
|
while(m_tcpServer->hasPendingConnections()) {
|
|
QTcpSocket* connection = m_tcpServer->nextPendingConnection();
|
|
connect(connection, SIGNAL(disconnected()), this, SLOT(onDisconnected()));
|
|
|
|
switch(m_sampleFormat) {
|
|
|
|
case FormatNFM:
|
|
case FormatSSB: {
|
|
quint32 id = (FormatSSB << 24) | m_nextSSBId;
|
|
MsgTCPSrcConnection* msg = MsgTCPSrcConnection::create(true, id, connection->peerAddress(), connection->peerPort());
|
|
m_nextSSBId = (m_nextSSBId + 1) & 0xffffff;
|
|
m_ssbSockets.push_back(Socket(id, connection));
|
|
msg->submit(m_uiMessageQueue, (PluginGUI*)m_tcpSrcGUI);
|
|
break;
|
|
}
|
|
|
|
case FormatS16LE: {
|
|
quint32 id = (FormatS16LE << 24) | m_nextS16leId;
|
|
MsgTCPSrcConnection* msg = MsgTCPSrcConnection::create(true, id, connection->peerAddress(), connection->peerPort());
|
|
m_nextS16leId = (m_nextS16leId + 1) & 0xffffff;
|
|
m_s16leSockets.push_back(Socket(id, connection));
|
|
msg->submit(m_uiMessageQueue, (PluginGUI*)m_tcpSrcGUI);
|
|
break;
|
|
}
|
|
|
|
default:
|
|
delete connection;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void TCPSrc::onDisconnected()
|
|
{
|
|
quint32 id;
|
|
QTcpSocket* socket = NULL;
|
|
|
|
for(int i = 0; i < m_ssbSockets.count(); i++) {
|
|
if(m_ssbSockets[i].socket == sender()) {
|
|
id = m_ssbSockets[i].id;
|
|
socket = m_ssbSockets[i].socket;
|
|
m_ssbSockets.removeAt(i);
|
|
break;
|
|
}
|
|
}
|
|
if(socket == NULL) {
|
|
for(int i = 0; i < m_s16leSockets.count(); i++) {
|
|
if(m_s16leSockets[i].socket == sender()) {
|
|
id = m_s16leSockets[i].id;
|
|
socket = m_s16leSockets[i].socket;
|
|
m_s16leSockets.removeAt(i);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if(socket != NULL) {
|
|
MsgTCPSrcConnection* msg = MsgTCPSrcConnection::create(false, id, QHostAddress(), 0);
|
|
msg->submit(m_uiMessageQueue, (PluginGUI*)m_tcpSrcGUI);
|
|
socket->deleteLater();
|
|
}
|
|
}
|