#ifndef MS_RTC_RTP_RTP_STREAM_SEND_HPP
#define MS_RTC_RTP_RTP_STREAM_SEND_HPP

#include "RTC/RTP/RetransmissionBuffer.hpp"
#include "RTC/RTP/RtpStream.hpp"
#include "RTC/RTP/SharedPacket.hpp"
#include "RTC/RateCalculator.hpp"

namespace RTC
{
	namespace RTP
	{
		class RtpStreamSend : public RTP::RtpStream
		{
		public:
			// Maximum retransmission buffer size for video (ms).
			static const uint32_t MaxRetransmissionDelayForVideoMs;
			// Maximum retransmission buffer size for audio (ms).
			static const uint32_t MaxRetransmissionDelayForAudioMs;

		public:
			enum class ReceivePacketResult : uint8_t
			{
				DISCARDED,
				ACCEPTED_AND_NOT_STORED,
				ACCEPTED_AND_STORED
			};

		public:
			class Listener : public RTP::RtpStream::Listener
			{
			public:
				virtual void OnRtpStreamRetransmitRtpPacket(
				  RTP::RtpStreamSend* rtpStream, RTP::Packet* packet) = 0;
			};

		public:
			RtpStreamSend(
			  RTP::RtpStreamSend::Listener* listener, RTP::RtpStream::Params& params, std::string& mid);
			~RtpStreamSend() override;

			flatbuffers::Offset<FBS::RtpStream::Stats> FillBufferStats(
			  flatbuffers::FlatBufferBuilder& builder) override;
			void SetRtx(uint8_t payloadType, uint32_t ssrc) override;
			ReceivePacketResult ReceivePacket(RTP::Packet* packet, const RTP::SharedPacket& sharedPacket);
			void ReceiveNack(RTC::RTCP::FeedbackRtpNackPacket* nackPacket);
			void ReceiveKeyFrameRequest(RTC::RTCP::FeedbackPs::MessageType messageType);
			void ReceiveRtcpReceiverReport(RTC::RTCP::ReceiverReport* report);
			void ReceiveRtcpXrReceiverReferenceTime(RTC::RTCP::ReceiverReferenceTime* report);
			RTC::RTCP::SenderReport* GetRtcpSenderReport(uint64_t nowMs);
			RTC::RTCP::DelaySinceLastRr::SsrcInfo* GetRtcpXrDelaySinceLastRrSsrcInfo(uint64_t nowMs);
			RTC::RTCP::SdesChunk* GetRtcpSdesChunk();
			void Pause() override;
			void Resume() override;
			uint32_t GetBitrate(uint64_t nowMs) override
			{
				return this->transmissionCounter.GetBitrate(nowMs);
			}
			uint32_t GetBitrate(uint64_t nowMs, uint8_t spatialLayer, uint8_t temporalLayer) override;
			uint32_t GetSpatialLayerBitrate(uint64_t nowMs, uint8_t spatialLayer) override;
			uint32_t GetLayerBitrate(uint64_t nowMs, uint8_t spatialLayer, uint8_t temporalLayer) override;

		private:
			void FillRetransmissionContainer(uint16_t seq, uint16_t bitmask);
			void UpdateScore(RTC::RTCP::ReceiverReport* report);

			/* Pure virtual methods inherited from RTP::RtpStream. */
		public:
			void UserOnSequenceNumberReset() override;

		private:
			// Packets lost at last interval for score calculation.
			int32_t lostPriorScore{ 0 };
			// Packets sent at last interval for score calculation.
			uint32_t sentPriorScore{ 0u };
			std::string mid;
			uint16_t rtxSeq{ 0u };
			RTC::RtpDataCounter transmissionCounter;
			RTP::RetransmissionBuffer* retransmissionBuffer{ nullptr };
			// The middle 32 bits out of 64 in the NTP timestamp received in the most
			// recent receiver reference timestamp.
			uint32_t lastRrTimestamp{ 0u };
			// Wallclock time representing the most recent receiver reference timestamp
			// arrival.
			uint64_t lastRrReceivedMs{ 0u };
		};
	} // namespace RTP
} // namespace RTC

#endif