#ifndef MS_RTC_CODECS_AV1_HPP
#define MS_RTC_CODECS_AV1_HPP

#include "common.hpp"
#include "RTC/Codecs/PayloadDescriptorHandler.hpp"
#include "RTC/RtpPacket.hpp"

namespace RTC
{
	namespace Codecs
	{
		class AV1
		{
		public:
			struct PayloadDescriptor : public RTC::Codecs::PayloadDescriptor
			{
				struct EncodingData
				{
					uint32_t maxSpatialLayer{ 0 };
					uint32_t maxTemporalLayer{ 0 };
				};

				struct Encoder : public RTC::Codecs::PayloadDescriptor::Encoder
				{
					~Encoder() override = default;
					explicit Encoder(EncodingData encodingData) : encodingData(encodingData)
					{
					}
					void Encode(AV1::PayloadDescriptor* payloadDescriptor) const;

					EncodingData encodingData;
				};

				PayloadDescriptor(std::unique_ptr<Codecs::DependencyDescriptor>& dependencyDescriptor);
				/* Pure virtual methods inherited from RTC::Codecs::PayloadDescriptor. */
				~PayloadDescriptor() override = default;

				void Dump(int indentation = 0) const override;
				void UpdateListener(Codecs::DependencyDescriptor::Listener* listener)
				{
					if (this->dependencyDescriptor)
					{
						this->dependencyDescriptor->UpdateListener(listener);
					}
				}
				void Encode();
				void Restore() const;
				void UpdateActiveDecodeTargets(uint16_t spatialLayer, uint16_t temporalLayer);
				std::unique_ptr<Codecs::PayloadDescriptor::Encoder> GetEncoder() const
				{
					if (this->encoder.has_value())
					{
						return std::make_unique<Encoder>(this->encoder.value());
					}
					else
					{
						return nullptr;
					}
				}

				void CreateEncoder(EncodingData encodingData)
				{
					this->encoder = Encoder(encodingData);
				}

				// Fields in Dependency Descriptor extension.
				bool startOfFrame{ false };
				bool endOfFrame{ false };
				uint16_t frameNumber{ 0 };
				uint8_t spatialLayer{ 0 };
				uint8_t temporalLayer{ 0 };
				std::unique_ptr<Codecs::DependencyDescriptor> dependencyDescriptor{ nullptr };
				// Parsed values.
				bool isKeyFrame{ false };
				std::optional<Encoder> encoder{ std::nullopt };
			};

		public:
			static AV1::PayloadDescriptor* Parse(
			  std::unique_ptr<Codecs::DependencyDescriptor>& dependencyDescriptor);
			static void ProcessRtpPacket(
			  RTC::RtpPacket* packet,
			  std::unique_ptr<RTC::Codecs::DependencyDescriptor::TemplateDependencyStructure>&
			    templateDependencyStructure);

		public:
			class EncodingContext : public RTC::Codecs::EncodingContext
			{
			public:
				explicit EncodingContext(RTC::Codecs::EncodingContext::Params& params)
				  : RTC::Codecs::EncodingContext(params)
				{
				}
				~EncodingContext() override = default;

				/* Pure virtual methods inherited from RTC::Codecs::EncodingContext. */
			public:
				void SyncRequired() override
				{
					this->syncRequired = true;
				}

			public:
				bool syncRequired{ false };
			};

			class PayloadDescriptorHandler : public RTC::Codecs::PayloadDescriptorHandler
			{
			public:
				explicit PayloadDescriptorHandler(PayloadDescriptor* payloadDescriptor);
				~PayloadDescriptorHandler() override = default;

			public:
				void Dump(int indentation = 0) const override
				{
					this->payloadDescriptor->Dump(indentation);
				}
				bool Process(
				  RTC::Codecs::EncodingContext* encodingContext, RTC::RtpPacket* packet, bool& marker) override;
				void RtpPacketCloned(RtpPacket* packet) override
				{
					this->payloadDescriptor->UpdateListener(packet);
				};
				std::unique_ptr<RTC::Codecs::PayloadDescriptor::Encoder> GetEncoder() const override
				{
					return this->payloadDescriptor->GetEncoder();
				}
				void Encode(RtpPacket* packet, Codecs::PayloadDescriptor::Encoder* encoder) override;
				void Restore(RtpPacket* packet) override;
				uint8_t GetSpatialLayer() const override
				{
					return this->payloadDescriptor->spatialLayer;
				}
				uint8_t GetTemporalLayer() const override
				{
					return this->payloadDescriptor->temporalLayer;
				}
				bool IsKeyFrame() const override
				{
					return this->payloadDescriptor->isKeyFrame;
				}

			private:
				std::unique_ptr<PayloadDescriptor> payloadDescriptor;
			};
		};
	} // namespace Codecs
} // namespace RTC

#endif