#include "stdafx.h"
#include "WindowsKeyManager.h"
#include "../Core/ControlManager.h"

static vector<KeyDefinition> _keyDefinitions = {
	//{ "VK_LBUTTON", 0x01, "Left mouse button" },
	//{ "VK_RBUTTON", 0x02, "Right mouse button" },
	{ "VK_CANCEL", 0x03, "Control-break processing" },
	//{ "VK_MBUTTON", 0x04, "Middle mouse button (three-button mouse)" },
	//{ "VK_XBUTTON1", 0x05, "X1 mouse button" },
	//{ "VK_XBUTTON2", 0x06, "X2 mouse button" },
	{ "-", 0x07, "Undefined" },
	{ "VK_BACK", 0x08, "Backspace" },
	{ "VK_TAB", 0x09, "Tab" },
	//{ "-", 0x0A - 0B, "Reserved" },
	{ "VK_CLEAR", 0x0C, "Clear" },
	{ "VK_RETURN", 0x0D, "Enter" },
	//{ "-", 0x0E - 0F, "Undefined" },
	{ "VK_SHIFT", 0x10, "Shift" },
	{ "VK_CONTROL", 0x11, "Ctrl" },
	{ "VK_MENU", 0x12, "Alt" },
	{ "VK_PAUSE", 0x13, "Pause" },
	{ "VK_CAPITAL", 0x14, "Caps Lock" },
	{ "VK_KANA", 0x15, "IME Kana mode" },
	{ "VK_HANGUEL", 0x15, "IME Hanguel mode" },
	{ "VK_HANGUL", 0x15, "IME Hangul mode" },
	//{ "-", 0x16, "Undefined" },
	{ "VK_JUNJA", 0x17, "IME Junja mode" },
	{ "VK_FINAL", 0x18, "IME final mode" },
	{ "VK_HANJA", 0x19, "IME Hanja mode" },
	{ "VK_KANJI", 0x19, "IME Kanji mode" },
	//{ "-", 0x1A, "Undefined" },
	{ "VK_ESCAPE", 0x1B, "Esc" },
	{ "VK_CONVERT", 0x1C, "IME convert" },
	{ "VK_NONCONVERT", 0x1D, "IME nonconvert" },
	{ "VK_ACCEPT", 0x1E, "IME accept" },
	{ "VK_MODECHANGE", 0x1F, "IME mode change request" },
	{ "VK_SPACE", 0x20, "Spacebar" },
	{ "VK_PRIOR", 0x21, "Page Up" },
	{ "VK_NEXT", 0x22, "Page Down" },
	{ "VK_END", 0x23, "End" },
	{ "VK_HOME", 0x24, "Home" },
	{ "VK_LEFT", 0x25, "Left Arrow" },
	{ "VK_UP", 0x26, "Up Arrow" },
	{ "VK_RIGHT", 0x27, "Right Arrow" },
	{ "VK_DOWN", 0x28, "Down Arrow" },
	{ "VK_SELECT", 0x29, "Select" },
	{ "VK_PRINT", 0x2A, "Print" },
	{ "VK_EXECUTE", 0x2B, "Execute" },
	{ "VK_SNAPSHOT", 0x2C, "Print Screen" },
	{ "VK_INSERT", 0x2D, "Ins" },
	{ "VK_DELETE", 0x2E, "Del" },
	{ "VK_HELP", 0x2F, "Help" },
	{ "0", 0x30, "0" },
	{ "1", 0x31, "1" },
	{ "2", 0x32, "2" },
	{ "3", 0x33, "3" },
	{ "4", 0x34, "4" },
	{ "5", 0x35, "5" },
	{ "6", 0x36, "6" },
	{ "7", 0x37, "7" },
	{ "8", 0x38, "8" },
	{ "9", 0x39, "9" },
	//{ "undefined", 0x3A - 40, "undefined" },
	{ "A", 0x41, "A" },
	{ "B", 0x42, "B" },
	{ "C", 0x43, "C" },
	{ "D", 0x44, "D" },
	{ "E", 0x45, "E" },
	{ "F", 0x46, "F" },
	{ "G", 0x47, "G" },
	{ "H", 0x48, "H" },
	{ "I", 0x49, "I" },
	{ "J", 0x4A, "J" },
	{ "K", 0x4B, "K" },
	{ "L", 0x4C, "L" },
	{ "M", 0x4D, "M" },
	{ "N", 0x4E, "N" },
	{ "O", 0x4F, "O" },
	{ "P", 0x50, "P" },
	{ "Q", 0x51, "Q" },
	{ "R", 0x52, "R" },
	{ "S", 0x53, "S" },
	{ "T", 0x54, "T" },
	{ "U", 0x55, "U" },
	{ "V", 0x56, "V" },
	{ "W", 0x57, "W" },
	{ "X", 0x58, "X" },
	{ "Y", 0x59, "Y" },
	{ "Z", 0x5A, "Z" },
	{ "VK_LWIN", 0x5B, "Left Windows" },
	{ "VK_RWIN", 0x5C, "Right Windows" },
	{ "VK_APPS", 0x5D, "Applications Key" },
	//{ "-", 0x5E, "Reserved" },
	{ "VK_SLEEP", 0x5F, "Computer Sleep" },
	{ "VK_NUMPAD0", 0x60, "Keypad 0" },
	{ "VK_NUMPAD1", 0x61, "Keypad 1" },
	{ "VK_NUMPAD2", 0x62, "Keypad 2" },
	{ "VK_NUMPAD3", 0x63, "Keypad 3" },
	{ "VK_NUMPAD4", 0x64, "Keypad 4" },
	{ "VK_NUMPAD5", 0x65, "Keypad 5" },
	{ "VK_NUMPAD6", 0x66, "Keypad 6" },
	{ "VK_NUMPAD7", 0x67, "Keypad 7" },
	{ "VK_NUMPAD8", 0x68, "Keypad 8" },
	{ "VK_NUMPAD9", 0x69, "Keypad 9" },
	{ "VK_MULTIPLY", 0x6A, "Multiply" },
	{ "VK_ADD", 0x6B, "Add" },
	{ "VK_SEPARATOR", 0x6C, "Separator" },
	{ "VK_SUBTRACT", 0x6D, "Subtract" },
	{ "VK_DECIMAL", 0x6E, "Decimal" },
	{ "VK_DIVIDE", 0x6F, "Divide" },
	{ "VK_F1", 0x70, "F1" },
	{ "VK_F2", 0x71, "F2" },
	{ "VK_F3", 0x72, "F3" },
	{ "VK_F4", 0x73, "F4" },
	{ "VK_F5", 0x74, "F5" },
	{ "VK_F6", 0x75, "F6" },
	{ "VK_F7", 0x76, "F7" },
	{ "VK_F8", 0x77, "F8" },
	{ "VK_F9", 0x78, "F9" },
	{ "VK_F10", 0x79, "F10" },
	{ "VK_F11", 0x7A, "F11" },
	{ "VK_F12", 0x7B, "F12" },
	{ "VK_F13", 0x7C, "F13" },
	{ "VK_F14", 0x7D, "F14" },
	{ "VK_F15", 0x7E, "F15" },
	{ "VK_F16", 0x7F, "F16" },
	{ "VK_F17", 0x80, "F17" },
	{ "VK_F18", 0x81, "F18" },
	{ "VK_F19", 0x82, "F19" },
	{ "VK_F20", 0x83, "F20" },
	{ "VK_F21", 0x84, "F21" },
	{ "VK_F22", 0x85, "F22" },
	{ "VK_F23", 0x86, "F23" },
	{ "VK_F24", 0x87, "F24" },
	//{ "-", 0x88 - 8F, "Unassigned" },
	{ "VK_NUMLOCK", 0x90, "Num Lock" },
	{ "VK_SCROLL", 0x91, "Scroll Lock" },
	//{"-", 0x92-96,"OEM specific"},
	//{ "-", 0x97 - 9F, "Unassigned" },
	{ "VK_LSHIFT", 0xA0, "Left Shift" },
	{ "VK_RSHIFT", 0xA1, "Right Shift" },
	{ "VK_LCONTROL", 0xA2, "Left Control" },
	{ "VK_RCONTROL", 0xA3, "Right Control" },
	{ "VK_LMENU", 0xA4, "Left Menu" },
	{ "VK_RMENU", 0xA5, "Right Menu" },
	{ "VK_BROWSER_BACK", 0xA6, "Browser Back" },
	{ "VK_BROWSER_FORWARD", 0xA7, "Browser Forward" },
	{ "VK_BROWSER_REFRESH", 0xA8, "Browser Refresh" },
	{ "VK_BROWSER_STOP", 0xA9, "Browser Stop" },
	{ "VK_BROWSER_SEARCH", 0xAA, "Browser Search" },
	{ "VK_BROWSER_FAVORITES", 0xAB, "Browser Favorites" },
	{ "VK_BROWSER_HOME", 0xAC, "Browser Start and Home" },
	{ "VK_VOLUME_MUTE", 0xAD, "Volume Mute" },
	{ "VK_VOLUME_DOWN", 0xAE, "Volume Down" },
	{ "VK_VOLUME_UP", 0xAF, "Volume Up" },
	{ "VK_MEDIA_NEXT_TRACK", 0xB0, "Next Track" },
	{ "VK_MEDIA_PREV_TRACK", 0xB1, "Previous Track" },
	{ "VK_MEDIA_STOP", 0xB2, "Stop Media" },
	{ "VK_MEDIA_PLAY_PAUSE", 0xB3, "Play/Pause Media" },
	{ "VK_LAUNCH_MAIL", 0xB4, "Start Mail" },
	{ "VK_LAUNCH_MEDIA_SELECT", 0xB5, "Select Media" },
	{ "VK_LAUNCH_APP1", 0xB6, "Start Application 1" },
	{ "VK_LAUNCH_APP2", 0xB7, "Start Application 2" },
	//{ "-", 0xB8 - B9, "Reserved" },
	{ "VK_OEM_1", 0xBA, "Used for miscellaneous characters; it can vary by keyboard. For the US standard keyboard, the \';:\' key VK_OEM_PLUS" },
	{ "VK_OEM_PLUS", 0xBB, "+" },
	{ "VK_OEM_COMMA", 0xBC, "," },
	{ "VK_OEM_MINUS", 0xBD, "-" },
	{ "VK_OEM_PERIOD", 0xBE, "." },
	{ "VK_OEM_2", 0xBF, "/" },
	{ "VK_OEM_3", 0xC0, "`" },
	//{ "-", 0xC1 - D7, "Reserved" },
	//{ "-", 0xD8 - DA, "Unassigned" },
	{ "VK_OEM_4", 0xDB, "[" },
	{ "VK_OEM_5", 0xDC, "\\" },
	{ "VK_OEM_6", 0xDD, "]" },
	{ "VK_OEM_7", 0xDE, "'" },
	{ "VK_OEM_8", 0xDF, "Used for miscellaneous characters; it can vary by keyboard." },
	//{ "-", 0xE0, "Reserved" },
	//{ "-", 0xE1, "OEM specific" },
	{ "VK_OEM_102", 0xE2, "Either the angle bracket key or the backslash key on the RT 102-key keyboard" },
	//{ "-", 0xE3 - E4, "OEM specific" },
	{ "VK_PROCESSKEY", 0xE5, "IME PROCESS" },
	//{ "-", 0xE6, "OEM specific" },
	{ "VK_PACKET", 0xE7, "Used to pass Unicode characters as if they were keystrokes. The VK_PACKET key is the low word of a 32-bit Virtual Key value used for non-keyboard input methods. For more information, see Remark in KEYBDINPUT, SendInput, WM_KEYDOWN, and WM_KEYUP" },
	//{ "-", 0xE8, "Unassigned" },
	//  {"-",0xE6,"OEM specific"},
	{ "VK_PACKET", 0xE7, "Used to pass Unicode characters as if they were keystrokes. The VK_PACKET key is the low word of a 32-bit Virtual Key value used for non-keyboard input methods. For more information, see Remark in KEYBDINPUT, SendInput, WM_KEYDOWN, and WM_KEYUP" },
	//  {"-",0xE8,"Unassigned"},
	//{ "-", 0xE9 - F5, "OEM specific" },
	{ "VK_ATTN", 0xF6, "Attn" },
	{ "VK_CRSEL", 0xF7, "CrSel" },
	{ "VK_EXSEL", 0xF8, "ExSel" },
	{ "VK_EREOF", 0xF9, "Erase EOF" },
	{ "VK_PLAY", 0xFA, "Play" },
	{ "VK_ZOOM", 0xFB, "Zoom" },
	{ "VK_NONAME", 0xFC, "Reserved" },
	{ "VK_PA1", 0xFD, "PA1" },
	{ "VK_OEM_CLEAR", 0xFE, "Clear" }
};

WindowsKeyManager::WindowsKeyManager(HWND hWnd)
{
	_hWnd = hWnd;

	//Init XInput buttons
	vector<string> buttonNames = { "Up", "Down", "Left", "Right", "Start", "Back", "L3", "R3", "L1", "R1", "?", "?", "A", "B", "X", "Y", "L2", "R2", "RT Up", "RT Down", "RT Up", "RT Left", "RT Right" };
	for(int i = 0; i < 4; i++) {
		for(int j = 0; j < buttonNames.size(); j++) {
			_keyDefinitions.push_back({ "", (uint32_t)(0xFFFF + i * 0x100 + j + 1), "Pad" + std::to_string(i + 1) + " " + buttonNames[j] });
		}
	}

	//Init DirectInput buttons
	vector<string> diButtonNames = { "Y+", "Y-", "X-", "X+", "Y2+", "Y2-", "X2-", "X2+", "Z+", "Z-", "Z2+", "Z2-", "DPad Up", "DPad Down", "DPad Right", "DPad Left" };
	for(int i = 0; i < 16; i++) {
		for(int j = 0; j < diButtonNames.size(); j++) {
			_keyDefinitions.push_back({ "", (uint32_t)(0x11000 + i * 0x100 + j), "Joy" + std::to_string(i + 1) + " " + diButtonNames[j] });
		}

		for(int j = 0; j < 128; j++) {
			_keyDefinitions.push_back({ "", (uint32_t)(0x11000 + i * 0x100 + j + 0x10), "Joy" + std::to_string(i + 1) + " But" + std::to_string(j+1)});
		}
	}

	_directInput.reset(new DirectInput(hWnd));
}

WindowsKeyManager::~WindowsKeyManager()
{
}

void WindowsKeyManager::RefreshState()
{
	_gamePad.RefreshState();
	_directInput->RefreshState();
}

bool WindowsKeyManager::IsKeyPressed(uint32_t key)
{
	if(key >= 0x10000) {
		if(key >= 0x11000) {
			//Directinput key
			uint8_t gamepadPort = (key - 0x11000) / 0x100;
			uint8_t gamepadButton = (key - 0x11000) % 0x100;
			return _directInput->IsButtonPressed(gamepadPort, gamepadButton);
		} else {
			//XInput key
			uint8_t gamepadPort = (key - 0xFFFF) / 0x100;
			uint8_t gamepadButton = (key - 0xFFFF) % 0x100;
			return _gamePad.IsPressed(gamepadPort, gamepadButton);
		}
	} else {
		return (GetAsyncKeyState(key) & 0x8000) == 0x8000;
	}
	return false;
}

bool WindowsKeyManager::IsMouseButtonPressed(MouseButton button)
{
	uint32_t key = 0;
	switch(button) {
		case MouseButton::LeftButton: key = VK_LBUTTON; break;
		case MouseButton::RightButton: key = VK_RBUTTON; break;
		case MouseButton::MiddleButton: key = VK_MBUTTON; break;
	}

	return (GetAsyncKeyState(key) & 0x8000) == 0x8000;
}

uint32_t WindowsKeyManager::GetPressedKey()
{
	_gamePad.RefreshState();
	for(int i = 0; i < XUSER_MAX_COUNT; i++) {
		for(int j = 1; j <= 22; j++) {
			if(_gamePad.IsPressed(i, j)) {
				return 0xFFFF + i * 0x100 + j;
			}
		}
	}

	_directInput->RefreshState();
	for(int i = _directInput->GetJoystickCount() - 1; i >= 0; i--) {
		for(int j = 0; j < 0x29; j++) {
			if(_directInput->IsButtonPressed(i, j)) {
				return 0x11000 + i * 0x100 + j;
			}
		}
	}

	for(int i = 0; i < 0xFF; i++) {
		if((GetAsyncKeyState(i) & 0x8000) == 0x8000) {
			return i;
		}
	}
	return 0;
}

string WindowsKeyManager::GetKeyName(uint32_t key)
{
	for(KeyDefinition keyDef : _keyDefinitions) {
		if(keyDef.keyCode == key) {
			return keyDef.description;
		}
	}
	return "";
}

uint32_t WindowsKeyManager::GetKeyCode(string keyName)
{
	for(KeyDefinition keyDef : _keyDefinitions) {
		if(keyName.compare(keyDef.description) == 0) {
			return keyDef.keyCode;
		}
	}
	return 0;
}