//==============================================================================
//
//  MPEGTS Depacketizer
//
//  Created by Getroot
//  Copyright (c) 2020 AirenSoft. All rights reserved.
//
//==============================================================================
#pragma once

#include <base/ovlibrary/ovlibrary.h>
#include <base/mediarouter/media_type.h>
#include <base/info/media_track.h>

#include "mpegts_common.h"
#include "mpegts_packet.h"
#include "mpegts_section.h"
#include "mpegts_pes.h"

/*  PES Depacketization Process

	Packet 1: [TS Header][Adaptation field][PES Header |    Payload     ] : uint_starting_indicator = 1
	Packet 2: [TS Header][                    Payload                   ]
	Packet 3: [TS Header][Adaptation field : stuffing[    Payload       ]

	Packet 1: [TS Header][Adaptation field][PES Header |    Payload     ] : uint_starting_indicator = 1
	(Assemble packet 1,2,3)
	(Create New ES 1)
*/

namespace mpegts
{
	enum class PacketType : uint8_t
	{
		UNKNOWN = 0,
		SUPPORTED_SECTION = 1,
		UNSUPPORTED_SECTION = 2,
		PES = 3
	};

	class MpegTsDepacketizer
	{
	public:
		MpegTsDepacketizer();
		~MpegTsDepacketizer();

		bool AddPacket(const std::shared_ptr<const ov::Data> &packet);
		bool AddPacket(const std::shared_ptr<Packet> &packet);

		bool IsTrackInfoAvailable();
		bool IsESAvailable();

		const std::shared_ptr<PAT> GetFirstPAT();
		const std::shared_ptr<PAT> GetPAT(uint8_t program_number);
		bool GetPMTList(uint16_t program_num, std::vector<std::shared_ptr<Section>> *pmt_list);
		bool GetTrackList(std::map<uint16_t, std::shared_ptr<MediaTrack>> *track_list);

		const std::shared_ptr<Pes> PopES();

	private:
		PacketType GetPacketType(const std::shared_ptr<Packet> &packet);

		bool ParseSection(const std::shared_ptr<Packet> &packet);
		bool ParsePes(const std::shared_ptr<Packet> &packet);
		
		const std::shared_ptr<Section> GetSectionDraft(uint16_t pid);	
		// incompleted section will be inserted
		bool SaveSectionDraft(const std::shared_ptr<Section> &section);
		// process completed section and remove, extract a table
		bool CompleteSection(const std::shared_ptr<Section> &section);

		const std::shared_ptr<Pes> GetPesDraft(uint16_t pid);	
		// incompleted section will be inserted
		bool SavePesDraft(const std::shared_ptr<Pes> &pes);
		// process completed section and remove, extract a elementary stream (es)
		bool CompletePes(const std::shared_ptr<Pes> &pes);

		bool CreateTracks();
		bool ExtractH264TrackInfo(const std::shared_ptr<Pes> &pes);
		bool ExtractAACTrackInfo(const std::shared_ptr<Pes> &pes);
		
		// PID : Section
		std::shared_mutex _section_draft_map_lock;
		std::map<uint16_t, std::shared_ptr<Section>> _section_draft_map;
		// PID : PES
		// there is only one pes saved per pid
		std::shared_mutex _pes_draft_map_lock;
		std::map<uint16_t, std::shared_ptr<Pes>> _pes_draft_map;

		// PID : Last continuity counter
		std::map<uint16_t, uint8_t> _last_continuity_counter_map;

		// PAT
		bool _pat_list_completed = false;
		// program number + packet identifier list
		// Program number : PAT Section
		std::map<uint8_t, std::shared_ptr<Section>> _pat_map;

		// PMT
		bool _pmt_list_completed = false;
		// Program Num : PMT Section
		std::vector<uint16_t> _completed_pmt_list;
		// PID, ES INFO
		std::multimap<uint16_t, std::shared_ptr<Section>> _pmt_map;

		// ES Info : from PMT
		std::map<uint16_t, std::shared_ptr<ESInfo>> _es_info_map;

		// extract from es and combine with es_info to make MediaTrack
		bool _track_list_completed = false;
		std::map<uint16_t, std::shared_ptr<MediaTrack>> _media_tracks;
		
		std::shared_mutex _es_list_lock;
		std::queue<std::shared_ptr<Pes>> _es_list;
		
		// PMT, PES quickly search PMT, PES
		// PID : PacketType 
		// PMT's PID comes from PAT
		// PES's PID comes from PMT/ES_INFO
		std::map<uint16_t, PacketType>	_packet_type_table;

		std::shared_ptr<ov::Data> _buffer = std::make_shared<ov::Data>();
	};
}