//  Copyright © 2015 Jean-Luc Deltombe (LX3JL). All rights reserved.

// ulxd -- The universal reflector
// Copyright © 2021 Thomas A. Early N7TAE
//
// 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, either 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 for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

#pragma once

#include "Semaphore.h"
#include "UDPSocket.h"
#include "PacketQueue.h"


////////////////////////////////////////////////////////////////////////////////////////
// define

// frame sizes
#define AMBE_SIZE           9
#define AMBEPLUS_SIZE       9


////////////////////////////////////////////////////////////////////////////////////////
// class

class CPacketStream;

class CCodecStream : public CPacketQueue
{
public:
	// constructor
	CCodecStream(CPacketStream *, uint16_t, uint8_t, uint8_t);

	// destructor
	virtual ~CCodecStream();

	// initialization
	bool Init(uint16_t);
	void Close(void);

	// get
	bool   IsConnected(void) const          { return m_bConnected; }
	uint16_t GetStreamId(void) const          { return m_uiStreamId; }
	double GetPingMin(void) const           { return m_fPingMin; }
	double GetPingMax(void) const           { return m_fPingMax; }
	double GetPingAve(void) const           { return (m_fPingCount != 0) ? m_fPingSum/m_fPingCount : 0; }
	uint32_t GetTotalPackets(void) const      { return m_uiTotalPackets; }
	uint32_t GetTimeoutPackets(void) const    { return m_uiTimeoutPackets; }
	bool   IsEmpty(void) const;

	// task
	void Thread(void);
	void Task(void);


protected:
	// packet decoding helpers
	bool IsValidAmbePacket(const CBuffer &, uint8_t *);

	// packet encoding helpers
	void EncodeAmbePacket(CBuffer *, const uint8_t *);


protected:
	// data
	uint16_t          m_uiStreamId;
	uint16_t          m_uiPort;
	uint8_t           m_uiPid;
	uint8_t           m_uiCodecIn;
	uint8_t           m_uiCodecOut;

	// socket
	CIp             m_Ip;
	CUdpSocket      m_Socket;
	bool            m_bConnected;

	// associated packet stream
	CPacketStream  *m_PacketStream;
	CPacketQueue    m_LocalQueue;

	// thread
	std::atomic<bool> keep_running;
	std::future<void> m_Future;
	CTimer          m_TimeoutTimer;
	CTimer          m_StatsTimer;

	// statistics
	double          m_fPingMin;
	double          m_fPingMax;
	double          m_fPingSum;
	double          m_fPingCount;
	uint32_t          m_uiTotalPackets;
	uint32_t          m_uiTimeoutPackets;
};