#pragma once

#include "stdafx.h"
#include "EmulationSettings.h"
#include "Snapshotable.h"

struct ButtonState
{
	bool Up = false;
	bool Down = false;
	bool Left = false;
	bool Right = false;

	bool A = false;
	bool B = false;

	bool Select = false;
	bool Start = false;

	uint8_t ToByte()
	{
		//"Button status for each controller is returned as an 8-bit report in the following order: A, B, Select, Start, Up, Down, Left, Right."
		return (uint8_t)A | ((uint8_t)B << 1) | ((uint8_t)Select << 2) | ((uint8_t)Start << 3) |
				((uint8_t)Up << 4) | ((uint8_t)Down << 5) | ((uint8_t)Left << 6) | ((uint8_t)Right << 7);
	}

	void FromByte(uint8_t stateData)
	{
		A = (stateData & 0x01) == 0x01;
		B = (stateData & 0x02) == 0x02;
		Select = (stateData & 0x04) == 0x04;
		Start = (stateData & 0x08) == 0x08;
		Up = (stateData & 0x10) == 0x10;
		Down = (stateData & 0x20) == 0x20;
		Left = (stateData & 0x40) == 0x40;
		Right = (stateData & 0x80) == 0x80;
	}
};

class BaseControlDevice : public Snapshotable
{
protected:
	uint8_t _port;
	uint8_t _currentState;
	vector<KeyMapping> _keyMappings;
	uint32_t _turboSpeed = 0;
	bool _famiconDevice = false;

	uint8_t GetPort();
	void AddKeyMappings(KeyMappingSet keyMappings);

	//Defined in controller-specific code and called when we need to read a device's state
	virtual uint8_t RefreshState() = 0;	
	virtual uint8_t ProcessNetPlayState(uint32_t netplayState);

	virtual void StreamState(bool saving);

public:
	//Used by controller-specific code to get the current state (buttons, position, etc)
	uint8_t GetControlState();

	BaseControlDevice(uint8_t port);
	virtual ~BaseControlDevice();

	//Called when reading $4016/7
	virtual uint8_t GetPortOutput() = 0;

	virtual uint32_t GetNetPlayState() = 0;

	//Used by standard controllers when $4017.1 is set
	virtual void RefreshStateBuffer();
};