///////////////////////////////////////////////////////////////////////////////////
// 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 "audionetsink.h"
#include "util/udpsink.h"
#include "util/rtpsink.h"

const int AudioNetSink::m_udpBlockSize = 512;

AudioNetSink::AudioNetSink(QObject *parent, bool stereo) :
    m_type(SinkUDP),
    m_udpBufferAudioMono(0),
    m_udpBufferAudioStereo(0),
    m_rtpBufferAudio(0)
{
    if (stereo) {
        m_udpBufferAudioStereo = new UDPSink<AudioSample>(parent, m_udpBlockSize);
    } else {
        m_udpBufferAudioMono = new UDPSink<int16_t>(parent, m_udpBlockSize);
    }

    m_rtpBufferAudio = new RTPSink("127.0.0.1", 9999, stereo ? RTPSink::PayloadL16Stereo : RTPSink::PayloadL16Mono);
}

AudioNetSink::~AudioNetSink()
{
    if (m_udpBufferAudioMono) {
        delete m_udpBufferAudioMono;
    }

    if (m_udpBufferAudioStereo) {
        delete m_udpBufferAudioStereo;
    }
    if (m_rtpBufferAudio) {
        delete m_rtpBufferAudio;
    }
}

bool AudioNetSink::isRTPCapable() const
{
        return m_rtpBufferAudio->isValid();
}

bool AudioNetSink::selectType(SinkType type)
{
    if (type == SinkUDP)
    {
        m_type = SinkUDP;
        return true;
    }
    else if (type == SinkRTP)
    {
        m_type = SinkRTP;
        return true;
    }
    else
    {
        return false;
    }
}

void AudioNetSink::setDestination(const QString& address, uint16_t port)
{
    if (m_udpBufferAudioMono) {
        m_udpBufferAudioMono->setDestination(address, port);
    }
    if (m_udpBufferAudioStereo) {
        m_udpBufferAudioStereo->setDestination(address, port);
    }
    if (m_rtpBufferAudio) {
        m_rtpBufferAudio->setDestination(address, port);
    }
}

void AudioNetSink::addDestination(const QString& address, uint16_t port)
{
    if (m_rtpBufferAudio) {
        m_rtpBufferAudio->addDestination(address, port);
    }
}

void AudioNetSink::deleteDestination(const QString& address, uint16_t port)
{
    if (m_rtpBufferAudio) {
        m_rtpBufferAudio->deleteDestination(address, port);
    }
}

void AudioNetSink::write(qint16 sample)
{
    if (m_udpBufferAudioMono == 0) {
        return;
    }

    if (m_type == SinkUDP) {
        m_udpBufferAudioMono->write(sample);
    } else if (m_type == SinkRTP) {
        m_rtpBufferAudio->write((uint8_t *) &sample);
    }
}

void AudioNetSink::write(const AudioSample& sample)
{
    if (m_udpBufferAudioStereo == 0) {
        return;
    }

    if (m_type == SinkUDP) {
        m_udpBufferAudioStereo->write(sample);
    } else if (m_type == SinkRTP) {
        m_rtpBufferAudio->write((uint8_t *) &sample);
    }
}

void AudioNetSink::moveToThread(QThread *thread)
{
    if (m_udpBufferAudioMono) {
        m_udpBufferAudioMono->moveToThread(thread);
    }

    if (m_udpBufferAudioStereo) {
        m_udpBufferAudioMono->moveToThread(thread);
    }

    if (m_rtpBufferAudio) {
        m_rtpBufferAudio->moveToThread(thread);
    }
}