mirror of
https://github.com/f4exb/sdrangel.git
synced 2024-11-16 05:11:49 -05:00
312 lines
11 KiB
C++
312 lines
11 KiB
C++
///////////////////////////////////////////////////////////////////////////////////
|
|
// Copyright (C) 2018 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 //
|
|
// //
|
|
// 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 "rtpsink.h"
|
|
#include "dsp/dsptypes.h"
|
|
#include <algorithm>
|
|
|
|
RTPSink::RTPSink(QUdpSocket *udpSocket, int sampleRate, bool stereo) :
|
|
m_payloadType(stereo ? RTPSink::PayloadL16Stereo : RTPSink::PayloadL16Mono),
|
|
m_sampleRate(sampleRate),
|
|
m_sampleBytes(0),
|
|
m_packetSamples(0),
|
|
m_bufferSize(0),
|
|
m_sampleBufferIndex(0),
|
|
m_byteBuffer(0),
|
|
m_destport(9998),
|
|
m_mutex(QMutex::Recursive)
|
|
{
|
|
m_rtpSessionParams.SetOwnTimestampUnit(1.0 / (double) m_sampleRate);
|
|
m_rtpTransmissionParams.SetRTCPMultiplexing(true); // do not allocate another socket for RTCP
|
|
m_rtpTransmissionParams.SetUseExistingSockets(udpSocket, udpSocket);
|
|
|
|
int status = m_rtpTransmitter.Init();
|
|
if (status < 0) {
|
|
qCritical("RTPSink::RTPSink: cannot initialize transmitter: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
m_valid = false;
|
|
} else {
|
|
qDebug("RTPSink::RTPSink: initialized transmitter: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
}
|
|
|
|
m_rtpTransmitter.Create(m_rtpSessionParams.GetMaximumPacketSize(), &m_rtpTransmissionParams);
|
|
qDebug("RTPSink::RTPSink: created transmitter: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
|
|
status = m_rtpSession.Create(m_rtpSessionParams, &m_rtpTransmitter);
|
|
if (status < 0) {
|
|
qCritical("RTPSink::RTPSink: cannot create session: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
m_valid = false;
|
|
} else {
|
|
qDebug("RTPSink::RTPSink: created session: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
}
|
|
|
|
setPayloadInformation(m_payloadType, m_sampleRate);
|
|
m_valid = true;
|
|
|
|
uint32_t endianTest32 = 1;
|
|
uint8_t *ptr = (uint8_t*) &endianTest32;
|
|
m_endianReverse = (*ptr == 1);
|
|
}
|
|
|
|
RTPSink::~RTPSink()
|
|
{
|
|
qrtplib::RTPTime delay = qrtplib::RTPTime(10.0);
|
|
m_rtpSession.BYEDestroy(delay, "Time's up", 9);
|
|
|
|
if (m_byteBuffer) {
|
|
delete[] m_byteBuffer;
|
|
}
|
|
}
|
|
|
|
void RTPSink::setPayloadInformation(PayloadType payloadType, int sampleRate)
|
|
{
|
|
uint32_t timestampinc;
|
|
QMutexLocker locker(&m_mutex);
|
|
|
|
qDebug("RTPSink::setPayloadInformation: %d sampleRate: %d", payloadType, sampleRate);
|
|
|
|
switch (payloadType)
|
|
{
|
|
case PayloadL16Stereo:
|
|
m_sampleBytes = 4;
|
|
m_rtpSession.SetDefaultPayloadType(96);
|
|
timestampinc = m_sampleRate / 100;
|
|
break;
|
|
case PayloadL16Mono:
|
|
default:
|
|
m_sampleBytes = 2;
|
|
m_rtpSession.SetDefaultPayloadType(96);
|
|
timestampinc = m_sampleRate / 50;
|
|
break;
|
|
}
|
|
|
|
m_packetSamples = m_sampleRate/50; // 20ms packet samples
|
|
m_bufferSize = m_packetSamples * m_sampleBytes;
|
|
|
|
if (m_byteBuffer) {
|
|
delete[] m_byteBuffer;
|
|
}
|
|
|
|
m_byteBuffer = new uint8_t[m_bufferSize];
|
|
m_sampleBufferIndex = 0;
|
|
m_payloadType = payloadType;
|
|
|
|
int status = m_rtpSession.SetTimestampUnit(1.0 / (double) m_sampleRate);
|
|
|
|
if (status < 0) {
|
|
qCritical("RTPSink::setPayloadInformation: cannot set timestamp unit: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
} else {
|
|
qDebug("RTPSink::setPayloadInformation: timestamp unit set to %f: %s",
|
|
1.0 / (double) m_sampleRate,
|
|
qrtplib::RTPGetErrorString(status).c_str());
|
|
}
|
|
|
|
status = m_rtpSession.SetDefaultMark(false);
|
|
|
|
if (status < 0) {
|
|
qCritical("RTPSink::setPayloadInformation: cannot set default mark: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
} else {
|
|
qDebug("RTPSink::setPayloadInformation: set default mark to false: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
}
|
|
|
|
status = m_rtpSession.SetDefaultTimestampIncrement(timestampinc);
|
|
|
|
if (status < 0) {
|
|
qCritical("RTPSink::setPayloadInformation: cannot set default timestamp increment: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
} else {
|
|
qDebug("RTPSink::setPayloadInformation: set default timestamp increment to %d: %s", timestampinc, qrtplib::RTPGetErrorString(status).c_str());
|
|
}
|
|
|
|
int maximumPacketSize = m_bufferSize+20; // was +40
|
|
|
|
while (maximumPacketSize < RTP_MINPACKETSIZE) {
|
|
maximumPacketSize += m_bufferSize;
|
|
}
|
|
|
|
status = m_rtpSession.SetMaximumPacketSize(maximumPacketSize);
|
|
|
|
if (status < 0) {
|
|
qCritical("RTPSink::setPayloadInformation: cannot set maximum packet size: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
} else {
|
|
qDebug("RTPSink::setPayloadInformation: set maximum packet size to %d bytes: %s", maximumPacketSize, qrtplib::RTPGetErrorString(status).c_str());
|
|
}
|
|
}
|
|
|
|
void RTPSink::setDestination(const QString& address, uint16_t port)
|
|
{
|
|
m_rtpSession.ClearDestinations();
|
|
m_rtpSession.DeleteDestination(qrtplib::RTPAddress(m_destip, m_destport));
|
|
m_destip.setAddress(address);
|
|
m_destport = port;
|
|
|
|
int status = m_rtpSession.AddDestination(qrtplib::RTPAddress(m_destip, m_destport));
|
|
|
|
if (status < 0) {
|
|
qCritical("RTPSink::setDestination: cannot set destination address: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
}
|
|
}
|
|
|
|
void RTPSink::deleteDestination(const QString& address, uint16_t port)
|
|
{
|
|
QHostAddress destip(address);
|
|
|
|
int status = m_rtpSession.DeleteDestination(qrtplib::RTPAddress(destip, port));
|
|
|
|
if (status < 0) {
|
|
qCritical("RTPSink::deleteDestination: cannot delete destination address: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
}
|
|
}
|
|
|
|
void RTPSink::addDestination(const QString& address, uint16_t port)
|
|
{
|
|
QHostAddress destip(address);
|
|
|
|
int status = m_rtpSession.AddDestination(qrtplib::RTPAddress(destip, port));
|
|
|
|
if (status < 0) {
|
|
qCritical("RTPSink::addDestination: cannot add destination address: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
} else {
|
|
qDebug("RTPSink::addDestination: destination address set to %s:%d: %s",
|
|
address.toStdString().c_str(),
|
|
port,
|
|
qrtplib::RTPGetErrorString(status).c_str());
|
|
}
|
|
}
|
|
|
|
void RTPSink::write(const uint8_t *sampleByte)
|
|
{
|
|
QMutexLocker locker(&m_mutex);
|
|
|
|
if (m_sampleBufferIndex < m_packetSamples)
|
|
{
|
|
writeNetBuf(&m_byteBuffer[m_sampleBufferIndex*m_sampleBytes],
|
|
sampleByte,
|
|
elemLength(m_payloadType),
|
|
m_sampleBytes,
|
|
m_endianReverse);
|
|
m_sampleBufferIndex++;
|
|
}
|
|
else
|
|
{
|
|
int status = m_rtpSession.SendPacket((const void *) m_byteBuffer, (std::size_t) m_bufferSize);
|
|
|
|
if (status < 0) {
|
|
qCritical("RTPSink::write: cannot write packet: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
}
|
|
|
|
writeNetBuf(&m_byteBuffer[0], sampleByte, elemLength(m_payloadType), m_sampleBytes, m_endianReverse);
|
|
m_sampleBufferIndex = 1;
|
|
}
|
|
}
|
|
|
|
void RTPSink::write(const uint8_t *sampleByteL, const uint8_t *sampleByteR)
|
|
{
|
|
QMutexLocker locker(&m_mutex);
|
|
|
|
if (m_sampleBufferIndex < m_packetSamples)
|
|
{
|
|
writeNetBuf(&m_byteBuffer[m_sampleBufferIndex*m_sampleBytes],
|
|
sampleByteL,
|
|
elemLength(m_payloadType),
|
|
m_sampleBytes,
|
|
m_endianReverse);
|
|
writeNetBuf(&m_byteBuffer[m_sampleBufferIndex*m_sampleBytes + elemLength(m_payloadType)],
|
|
sampleByteR,
|
|
elemLength(m_payloadType),
|
|
m_sampleBytes,
|
|
m_endianReverse);
|
|
m_sampleBufferIndex++;
|
|
}
|
|
else
|
|
{
|
|
int status = m_rtpSession.SendPacket((const void *) m_byteBuffer, (std::size_t) m_bufferSize);
|
|
|
|
if (status < 0) {
|
|
qCritical("RTPSink::write: cannot write packet: %s", qrtplib::RTPGetErrorString(status).c_str());
|
|
}
|
|
|
|
writeNetBuf(&m_byteBuffer[0], sampleByteL, elemLength(m_payloadType), m_sampleBytes, m_endianReverse);
|
|
writeNetBuf(&m_byteBuffer[2], sampleByteR, elemLength(m_payloadType), m_sampleBytes, m_endianReverse);
|
|
m_sampleBufferIndex = 1;
|
|
}
|
|
|
|
}
|
|
|
|
void RTPSink::write(const uint8_t *samples, int nbSamples)
|
|
{
|
|
int samplesIndex = 0;
|
|
QMutexLocker locker(&m_mutex);
|
|
|
|
// fill remainder of buffer and send it
|
|
if (m_sampleBufferIndex + nbSamples > m_packetSamples)
|
|
{
|
|
writeNetBuf(&m_byteBuffer[m_sampleBufferIndex*m_sampleBytes],
|
|
&samples[samplesIndex*m_sampleBytes],
|
|
elemLength(m_payloadType),
|
|
(m_packetSamples - m_sampleBufferIndex)*m_sampleBytes,
|
|
m_endianReverse);
|
|
m_rtpSession.SendPacket((const void *) m_byteBuffer, (std::size_t) m_bufferSize);
|
|
nbSamples -= (m_packetSamples - m_sampleBufferIndex);
|
|
m_sampleBufferIndex = 0;
|
|
}
|
|
|
|
// send complete packets
|
|
while (nbSamples > m_packetSamples)
|
|
{
|
|
writeNetBuf(m_byteBuffer,
|
|
samples,
|
|
elemLength(m_payloadType),
|
|
m_bufferSize,
|
|
m_endianReverse);
|
|
m_rtpSession.SendPacket((const void *) m_byteBuffer, (std::size_t) m_bufferSize);
|
|
samplesIndex += m_packetSamples;
|
|
nbSamples -= m_packetSamples;
|
|
}
|
|
|
|
// copy remainder of input to buffer
|
|
writeNetBuf(&m_byteBuffer[m_sampleBufferIndex*m_sampleBytes],
|
|
&samples[samplesIndex*m_sampleBytes],
|
|
elemLength(m_payloadType),
|
|
nbSamples*m_sampleBytes,m_endianReverse);
|
|
}
|
|
|
|
void RTPSink::writeNetBuf(uint8_t *dest, const uint8_t *src, unsigned int elemLen, unsigned int bytesLen, bool endianReverse)
|
|
{
|
|
for (unsigned int i = 0; i < bytesLen; i += elemLen)
|
|
{
|
|
memcpy(&dest[i], &src[i], elemLen);
|
|
|
|
if (endianReverse) {
|
|
std::reverse(&dest[i], &dest[i+elemLen]);
|
|
}
|
|
}
|
|
}
|
|
|
|
unsigned int RTPSink::elemLength(PayloadType payloadType)
|
|
{
|
|
switch (payloadType)
|
|
{
|
|
case PayloadL16Stereo:
|
|
return sizeof(int16_t);
|
|
break;
|
|
case PayloadL16Mono:
|
|
default:
|
|
return sizeof(int16_t);
|
|
break;
|
|
}
|
|
}
|
|
|