#ifndef MS_RTC_RTP_RTP_STREAM_RECV_HPP
#define MS_RTC_RTP_RTP_STREAM_RECV_HPP

#include "RTC/NackGenerator.hpp"
#include "RTC/RTCP/XrDelaySinceLastRr.hpp"
#include "RTC/RTP/RtpStream.hpp"
#include "RTC/RateCalculator.hpp"
#include "handles/TimerHandle.hpp"
#include <vector>

namespace RTC
{
	namespace RTP
	{
		class RtpStreamRecv : public RTP::RtpStream,
		                      public RTC::NackGenerator::Listener,
		                      public TimerHandle::Listener
		{
		public:
			class Listener : public RTP::RtpStream::Listener
			{
			public:
				virtual void OnRtpStreamSendRtcpPacket(
				  RTP::RtpStreamRecv* rtpStream, RTC::RTCP::Packet* packet) = 0;
				virtual void OnRtpStreamNeedWorstRemoteFractionLost(
				  RTP::RtpStreamRecv* rtpStream, uint8_t& worstRemoteFractionLost) = 0;
			};

		public:
			class TransmissionCounter
			{
			public:
				TransmissionCounter(uint8_t spatialLayers, uint8_t temporalLayers, size_t windowSize);
				void Update(const RTP::Packet* packet);
				uint32_t GetBitrate(uint64_t nowMs);
				uint32_t GetBitrate(uint64_t nowMs, uint8_t spatialLayer, uint8_t temporalLayer);
				uint32_t GetSpatialLayerBitrate(uint64_t nowMs, uint8_t spatialLayer);
				uint32_t GetLayerBitrate(uint64_t nowMs, uint8_t spatialLayer, uint8_t temporalLayer);
				size_t GetPacketCount() const;
				size_t GetBytes() const;

			private:
				std::vector<std::vector<RTC::RtpDataCounter>> spatialLayerCounters;
			};

		public:
			RtpStreamRecv(
			  RTP::RtpStreamRecv::Listener* listener,
			  RTP::RtpStream::Params& params,
			  unsigned int sendNackDelayMs,
			  bool useRtpInactivityCheck);
			~RtpStreamRecv() override;

			flatbuffers::Offset<FBS::RtpStream::Stats> FillBufferStats(
			  flatbuffers::FlatBufferBuilder& builder) override;
			bool ReceivePacket(RTP::Packet* packet);
			bool ReceiveRtxPacket(RTP::Packet* packet);
			RTC::RTCP::ReceiverReport* GetRtcpReceiverReport();
			RTC::RTCP::ReceiverReport* GetRtxRtcpReceiverReport();
			void ReceiveRtcpSenderReport(RTC::RTCP::SenderReport* report);
			void ReceiveRtxRtcpSenderReport(RTC::RTCP::SenderReport* report);
			void ReceiveRtcpXrDelaySinceLastRr(RTC::RTCP::DelaySinceLastRr::SsrcInfo* ssrcInfo);
			void RequestKeyFrame();
			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
			{
				return this->transmissionCounter.GetBitrate(nowMs, spatialLayer, temporalLayer);
			}
			uint32_t GetSpatialLayerBitrate(uint64_t nowMs, uint8_t spatialLayer) override
			{
				return this->transmissionCounter.GetSpatialLayerBitrate(nowMs, spatialLayer);
			}
			uint32_t GetLayerBitrate(uint64_t nowMs, uint8_t spatialLayer, uint8_t temporalLayer) override
			{
				return this->transmissionCounter.GetLayerBitrate(nowMs, spatialLayer, temporalLayer);
			}
			bool HasRtpInactivityCheckEnabled() const
			{
				return this->useRtpInactivityCheck;
			}

		private:
			void CalculateJitter(uint32_t rtpTimestamp);
			void UpdateScore();

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

			/* Pure virtual methods inherited from TimerHandle. */
		protected:
			void OnTimer(TimerHandle* timer) override;

			/* Pure virtual methods inherited from RTC::NackGenerator. */
		protected:
			void OnNackGeneratorNackRequired(const std::vector<uint16_t>& seqNumbers) override;
			void OnNackGeneratorKeyFrameRequired() override;

		private:
			// Passed by argument.
			unsigned int sendNackDelayMs{ 0u };
			bool useRtpInactivityCheck{ false };
			// Others.
			// Packets expected at last interval.
			uint32_t expectedPrior{ 0u };
			// Packets expected at last interval for score calculation.
			uint32_t expectedPriorScore{ 0u };
			// Packets received at last interval.
			uint32_t receivedPrior{ 0u };
			// Packets received at last interval for score calculation.
			uint32_t receivedPriorScore{ 0u };
			// The middle 32 bits out of 64 in the NTP timestamp received in the most
			// recent sender report.
			uint32_t lastSrTimestamp{ 0u };
			// Wallclock time representing the most recent sender report arrival.
			uint64_t lastSrReceived{ 0u };
			// Relative transit time for prev packet.
			int32_t transit{ 0u };
			uint8_t firSeqNumber{ 0u };
			int32_t reportedPacketsLost{ 0 };
			std::unique_ptr<RTC::NackGenerator> nackGenerator;
			TimerHandle* inactivityCheckPeriodicTimer{ nullptr };
			bool inactive{ false };
			// Valid media + valid RTX.
			TransmissionCounter transmissionCounter;
			// Just valid media.
			RTC::RtpDataCounter mediaTransmissionCounter;
			// Template dependency structure for Dependency Descriptor.
			std::unique_ptr<RTP::Codecs::DependencyDescriptor::TemplateDependencyStructure>
			  templateDependencyStructure;
		};
	} // namespace RTP
} // namespace RTC

#endif