#include "pch.h"
#include "Shared/BaseControlManager.h"
#include "Shared/Emulator.h"
#include "Shared/EmuSettings.h"
#include "Shared/KeyManager.h"
#include "Shared/ControllerHub.h"
#include "Shared/Interfaces/IKeyManager.h"
#include "Shared/Interfaces/IInputProvider.h"
#include "Shared/Interfaces/IInputRecorder.h"
#include "Shared/SystemActionManager.h"
#include "Shared/EventType.h"
#include "Shared/MessageManager.h"
#include "Utilities/Serializer.h"

BaseControlManager::BaseControlManager(Emulator* emu, CpuType cpuType)
{
	_emu = emu;
	_cpuType = cpuType;
	_pollCounter = 0;
	AddSystemControlDevice(_emu->GetSystemActionManager());
	UpdateControlDevices();
}

BaseControlManager::~BaseControlManager()
{
}

void BaseControlManager::AddSystemControlDevice(shared_ptr<BaseControlDevice> device)
{
	_controlDevices.clear();
	_systemDevices.push_back(device);
	UpdateControlDevices();
}

void BaseControlManager::RegisterInputProvider(IInputProvider* provider)
{
	auto lock = _deviceLock.AcquireSafe();
	_inputProviders.push_back(provider);
}

void BaseControlManager::UnregisterInputProvider(IInputProvider* provider)
{
	auto lock = _deviceLock.AcquireSafe();
	vector<IInputProvider*>& vec = _inputProviders;
	vec.erase(std::remove(vec.begin(), vec.end(), provider), vec.end());
}

void BaseControlManager::RegisterInputRecorder(IInputRecorder* provider)
{
	auto lock = _deviceLock.AcquireSafe();
	_inputRecorders.push_back(provider);
}

void BaseControlManager::UnregisterInputRecorder(IInputRecorder* provider)
{
	auto lock = _deviceLock.AcquireSafe();
	vector<IInputRecorder*>& vec = _inputRecorders;
	vec.erase(std::remove(vec.begin(), vec.end(), provider), vec.end());
}

vector<ControllerData> BaseControlManager::GetPortStates()
{
	vector<ControllerData> states;
	for(shared_ptr<BaseControlDevice>& device : _controlDevices) {
		states.push_back({
			device->GetControllerType(),
			device->GetRawState(),
			device->GetPort()
		});
	}
	return states;
}

shared_ptr<BaseControlDevice> BaseControlManager::GetControlDevice(uint8_t port, uint8_t subPort)
{
	auto lock = _deviceLock.AcquireSafe();

	for(size_t i = 0; i < _controlDevices.size(); i++) {
		if(_controlDevices[i] && _controlDevices[i]->GetPort() == port) {
			shared_ptr<IControllerHub> hub = std::dynamic_pointer_cast<IControllerHub>(_controlDevices[i]);
			if(hub) {
				return hub->GetController(subPort);
			} else {
				return subPort == 0 ? _controlDevices[i] : nullptr;
			}
		}
	}
	return nullptr;
}

shared_ptr<BaseControlDevice> BaseControlManager::GetControlDeviceByIndex(uint8_t index)
{
	auto lock = _deviceLock.AcquireSafe();

	int counter = 0;
	for(size_t i = 0; i < _controlDevices.size(); i++) {
		if(_controlDevices[i] && _controlDevices[i]->GetPort() <= 2) {
			shared_ptr<IControllerHub> hub = std::dynamic_pointer_cast<IControllerHub>(_controlDevices[i]);
			if(hub) {
				int portCount = hub->GetHubPortCount();
				for(int j = 0; j < portCount; j++) {
					shared_ptr<BaseControlDevice> device = hub->GetController(j);
					if(counter == index) {
						return device;
					}
					counter++;
				}
			} else {
				if(counter == index) {
					return _controlDevices[i];
				}
				counter++;
			}
		}
	}
	return nullptr;
}

void BaseControlManager::RefreshHubState()
{
	auto lock = _deviceLock.AcquireSafe();

	for(size_t i = 0; i < _controlDevices.size(); i++) {
		if(_controlDevices[i] && _controlDevices[i]->GetPort() <= 2) {
			shared_ptr<IControllerHub> hub = std::dynamic_pointer_cast<IControllerHub>(_controlDevices[i]);
			if(hub) {
				hub->RefreshHubState();
			}
		}
	}
}

vector<shared_ptr<BaseControlDevice>> BaseControlManager::GetControlDevices()
{
	return _controlDevices;
}

void BaseControlManager::Serialize(Serializer& s)
{
	SV(_pollCounter);
}

void BaseControlManager::RegisterControlDevice(shared_ptr<BaseControlDevice> controlDevice)
{
	controlDevice->Init();
	_controlDevices.push_back(controlDevice);
}

void BaseControlManager::ClearDevices()
{
	_controlDevices.clear();

	for(shared_ptr<BaseControlDevice> device : _systemDevices) {
		RegisterControlDevice(device);
	}
}

void BaseControlManager::UpdateInputState()
{
	KeyManager::RefreshKeyState();

	auto lock = _deviceLock.AcquireSafe();

	//string log = "F: " + std::to_string(_emu->GetFrameCount()) + " C:" + std::to_string(_pollCounter) + " ";
	for(shared_ptr<BaseControlDevice>& device : _controlDevices) {
		device->ClearState();
		device->SetStateFromInput();

		for(size_t i = 0; i < _inputProviders.size(); i++) {
			IInputProvider* provider = _inputProviders[i];
			if(provider->SetInput(device.get())) {
				break;
			}
		}

		device->OnAfterSetState();
		//log += "|" + device->GetTextState();
	}

	_emu->ProcessEvent(EventType::InputPolled, _cpuType);

	if(!_emu->IsRunAheadFrame()) {
		for(IInputRecorder* recorder : _inputRecorders) {
			recorder->RecordInput(_controlDevices);
		}
	}

	//MessageManager::Log(log);

	_pollCounter++;
}

void BaseControlManager::ProcessEndOfFrame()
{
	if(!_wasInputRead) {
		_lagCounter++;
	}
	_wasInputRead = false;
}

void BaseControlManager::SetInputReadFlag()
{
	//Used for lag counter - any frame where the input is read does not count as lag
	_wasInputRead = true;
}

uint32_t BaseControlManager::GetLagCounter()
{
	return _lagCounter;
}

void BaseControlManager::ResetLagCounter()
{
	_lagCounter = 0;
}

bool BaseControlManager::HasControlDevice(ControllerType type)
{
	auto lock = _deviceLock.AcquireSafe();

	for(shared_ptr<BaseControlDevice>& device : _controlDevices) {
		if(device->HasControllerType(type)) {
			return true;
		}
	}
	return false;
}

uint32_t BaseControlManager::GetPollCounter()
{
	return _pollCounter;
}

void BaseControlManager::SetPollCounter(uint32_t value)
{
	_pollCounter = value;
}