mirror of
https://github.com/f4exb/sdrangel.git
synced 2024-09-10 07:06:34 -04:00
181 lines
7.7 KiB
C++
181 lines
7.7 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 //
|
|
// (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/>. //
|
|
///////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef PLUGINS_SAMPLESINK_REMOTEOUTPUT_UDPSINKFEC_H_
|
|
#define PLUGINS_SAMPLESINK_REMOTEOUTPUT_UDPSINKFEC_H_
|
|
|
|
#include <channel/remotedatablock.h>
|
|
#include <string.h>
|
|
#include <cstddef>
|
|
|
|
#include <QObject>
|
|
#include <QHostAddress>
|
|
#include <QString>
|
|
|
|
#include "dsp/dsptypes.h"
|
|
#include "util/CRC64.h"
|
|
|
|
class QThread;
|
|
class RemoteOutputSender;
|
|
|
|
class UDPSinkFEC : public QObject
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
static const uint32_t m_udpSize = 512; //!< Size of UDP block in number of bytes
|
|
static const uint32_t m_nbOriginalBlocks = 128; //!< Number of original blocks in a protected block sequence
|
|
|
|
/**
|
|
* Construct UDP sink
|
|
*/
|
|
UDPSinkFEC();
|
|
|
|
/** Destroy UDP sink */
|
|
~UDPSinkFEC();
|
|
|
|
void init();
|
|
void startSender();
|
|
void stopSender();
|
|
|
|
/**
|
|
* Write IQ samples
|
|
*/
|
|
void write(const SampleVector::iterator& begin, uint32_t sampleChunkSize, bool isTx);
|
|
|
|
/** Return the last error, or return an empty string if there is no error. */
|
|
std::string error()
|
|
{
|
|
std::string ret(m_error);
|
|
m_error.clear();
|
|
return ret;
|
|
}
|
|
|
|
void setDeviceIndex(uint32_t deviceIndex) { m_deviceIndex = deviceIndex; }
|
|
/** Set sample rate given in S/s */
|
|
void setSampleRate(uint32_t sampleRate);
|
|
|
|
void setNbBlocksFEC(uint32_t nbBlocksFEC);
|
|
void setNbTxBytes(uint32_t nbTxBytes) { m_nbTxBytes = nbTxBytes; }
|
|
void setRemoteAddress(const QString& address, uint16_t port);
|
|
|
|
/** Return true if the stream is OK, return false if there is an error. */
|
|
operator bool() const
|
|
{
|
|
return m_error.empty();
|
|
}
|
|
|
|
private:
|
|
std::string m_error;
|
|
|
|
uint32_t m_sampleRate; //!< sample rate in Hz
|
|
uint32_t m_nbSamples; //!< total number of samples sent int the last frame
|
|
QHostAddress m_ownAddress;
|
|
|
|
CRC64 m_crc64;
|
|
RemoteMetaDataFEC m_currentMetaFEC; //!< Meta data for current frame
|
|
uint32_t m_nbBlocksFEC; //!< Variable number of FEC blocks
|
|
uint32_t m_nbTxBytes;
|
|
float m_txDelayRatio; //!< Delay in ratio of nominal frame period
|
|
RemoteDataFrame *m_dataFrame;
|
|
RemoteSuperBlock m_superBlock; //!< current super block being built
|
|
int m_txBlockIndex; //!< Current index in blocks to transmit in the Tx row
|
|
uint16_t m_frameCount; //!< transmission frame count
|
|
int m_sampleIndex; //!< Current sample index in protected block data
|
|
uint32_t m_deviceIndex; //!< Index of current device set
|
|
|
|
RemoteOutputSender *m_remoteOutputSender;
|
|
QThread *m_senderThread;
|
|
QString m_remoteAddress;
|
|
uint16_t m_remotePort;
|
|
|
|
uint32_t getNbSampleBits();
|
|
|
|
inline void convertSampleToData(const SampleVector::iterator& begin, int nbSamples, bool isTx)
|
|
{
|
|
if (sizeof(Sample) == m_nbTxBytes * 2) // 16 -> 16 or 24 ->24: direct copy
|
|
{
|
|
memcpy((void *) &m_superBlock.m_protectedBlock.buf[m_sampleIndex*m_nbTxBytes*2],
|
|
(const void *) &(*(begin)),
|
|
nbSamples * sizeof(Sample));
|
|
}
|
|
else if (isTx)
|
|
{
|
|
if (m_nbTxBytes == 4) // just convert type int16_t -> int32_t (always 16 bit wide)
|
|
{
|
|
for (int i = 0; i < nbSamples; i++)
|
|
{
|
|
*((int32_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2]) = (begin+i)->m_real;
|
|
*((int32_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2 + m_nbTxBytes]) = (begin+i)->m_imag;
|
|
}
|
|
}
|
|
else if (m_nbTxBytes == 2) //just convert type int32_t -> int16_t (always 16 bit wide)
|
|
{
|
|
for (int i = 0; i < nbSamples; i++)
|
|
{
|
|
*((int16_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2]) = (begin+i)->m_real;
|
|
*((int16_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2 + m_nbTxBytes]) = (begin+i)->m_imag;
|
|
}
|
|
}
|
|
else if (m_nbTxBytes == 1) // 16 or 24 -> 8
|
|
{
|
|
for (int i = 0; i < nbSamples; i++)
|
|
{
|
|
*((int8_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2]) = (int8_t) ((begin+i)->m_real / (1<<8));
|
|
*((int8_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2 + m_nbTxBytes]) = (int8_t) ((begin+i)->m_imag / (1<<8));
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (m_nbTxBytes == 4) // 16 -> 24
|
|
{
|
|
for (int i = 0; i < nbSamples; i++)
|
|
{
|
|
*((int32_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2]) = (begin+i)->m_real * (1<<8);
|
|
*((int32_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2 + m_nbTxBytes]) = (begin+i)->m_imag * (1<<8);
|
|
}
|
|
}
|
|
else if (m_nbTxBytes == 2) // 24 -> 16
|
|
{
|
|
for (int i = 0; i < nbSamples; i++)
|
|
{
|
|
*((int16_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2]) = (begin+i)->m_real / (1<<8);
|
|
*((int16_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2 + m_nbTxBytes]) = (begin+i)->m_imag / (1<<8);
|
|
}
|
|
}
|
|
else if ((m_nbTxBytes == 1) && (sizeof(Sample) == 4)) // 16 -> 8
|
|
{
|
|
for (int i = 0; i < nbSamples; i++)
|
|
{
|
|
*((int8_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2]) = (int8_t) ((begin+i)->m_real / (1<<8));
|
|
*((int8_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2 + m_nbTxBytes]) = (int8_t) ((begin+i)->m_imag / (1<<8));
|
|
}
|
|
}
|
|
else if ((m_nbTxBytes == 1) && (sizeof(Sample) == 4)) // 24 -> 8
|
|
{
|
|
for (int i = 0; i < nbSamples; i++)
|
|
{
|
|
*((int8_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2]) = (int8_t) ((begin+i)->m_real / (1<<16));
|
|
*((int8_t*) &m_superBlock.m_protectedBlock.buf[(m_sampleIndex+ i)*m_nbTxBytes*2 + m_nbTxBytes]) = (int8_t) ((begin+i)->m_imag / (1<<16));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
#endif /* PLUGINS_SAMPLESINK_REMOTEOUTPUT_UDPSINKFEC_H_ */
|