#include "..\easy2d.h"
#include "..\EasyX\easyx.h"
#include <conio.h>

// 按键监听回调函数的容器
static std::vector<KeyMsg*> s_vListeners;

// 虚拟键值的定义
const VK_KEY KeyMsg::A = 'A';
const VK_KEY KeyMsg::B = 'B';
const VK_KEY KeyMsg::C = 'C';
const VK_KEY KeyMsg::D = 'D';
const VK_KEY KeyMsg::E = 'E';
const VK_KEY KeyMsg::F = 'F';
const VK_KEY KeyMsg::G = 'G';
const VK_KEY KeyMsg::H = 'H';
const VK_KEY KeyMsg::I = 'I';
const VK_KEY KeyMsg::J = 'J';
const VK_KEY KeyMsg::K = 'K';
const VK_KEY KeyMsg::L = 'L';
const VK_KEY KeyMsg::M = 'M';
const VK_KEY KeyMsg::N = 'N';
const VK_KEY KeyMsg::O = 'O';
const VK_KEY KeyMsg::P = 'P';
const VK_KEY KeyMsg::Q = 'Q';
const VK_KEY KeyMsg::R = 'R';
const VK_KEY KeyMsg::S = 'S';
const VK_KEY KeyMsg::T = 'T';
const VK_KEY KeyMsg::U = 'U';
const VK_KEY KeyMsg::V = 'V';
const VK_KEY KeyMsg::W = 'W';
const VK_KEY KeyMsg::X = 'X';
const VK_KEY KeyMsg::Y = 'Y';
const VK_KEY KeyMsg::Z = 'Z';
const VK_KEY KeyMsg::NUM_0 = '0';
const VK_KEY KeyMsg::NUM_1 = '1';
const VK_KEY KeyMsg::NUM_2 = '2';
const VK_KEY KeyMsg::NUM_3 = '3';
const VK_KEY KeyMsg::NUM_4 = '4';
const VK_KEY KeyMsg::NUM_5 = '5';
const VK_KEY KeyMsg::NUM_6 = '6';
const VK_KEY KeyMsg::NUM_7 = '7';
const VK_KEY KeyMsg::NUM_8 = '8';
const VK_KEY KeyMsg::NUM_9 = '9';
const VK_KEY KeyMsg::NUMPAD_0 = VK_NUMPAD0;
const VK_KEY KeyMsg::NUMPAD_1 = VK_NUMPAD1;
const VK_KEY KeyMsg::NUMPAD_2 = VK_NUMPAD2;
const VK_KEY KeyMsg::NUMPAD_3 = VK_NUMPAD3;
const VK_KEY KeyMsg::NUMPAD_4 = VK_NUMPAD4;
const VK_KEY KeyMsg::NUMPAD_5 = VK_NUMPAD5;
const VK_KEY KeyMsg::NUMPAD_6 = VK_NUMPAD6;
const VK_KEY KeyMsg::NUMPAD_7 = VK_NUMPAD7;
const VK_KEY KeyMsg::NUMPAD_8 = VK_NUMPAD8;
const VK_KEY KeyMsg::NUMPAD_9 = VK_NUMPAD9;
const VK_KEY KeyMsg::Enter = VK_RETURN;
const VK_KEY KeyMsg::Space = VK_SPACE;
const VK_KEY KeyMsg::Ctrl = VK_CONTROL;
const VK_KEY KeyMsg::LCtrl = VK_LCONTROL;
const VK_KEY KeyMsg::RCtrl = VK_RCONTROL;
const VK_KEY KeyMsg::Shift = VK_SHIFT;
const VK_KEY KeyMsg::LShift = VK_LSHIFT;
const VK_KEY KeyMsg::RShift = VK_RSHIFT;
const VK_KEY KeyMsg::Up = VK_UP;
const VK_KEY KeyMsg::Down = VK_DOWN;
const VK_KEY KeyMsg::Left = VK_LEFT;
const VK_KEY KeyMsg::Right = VK_RIGHT;
const VK_KEY KeyMsg::Esc = VK_ESCAPE;
const VK_KEY KeyMsg::F1 = VK_F1;
const VK_KEY KeyMsg::F2 = VK_F2;
const VK_KEY KeyMsg::F3 = VK_F3;
const VK_KEY KeyMsg::F4 = VK_F4;
const VK_KEY KeyMsg::F5 = VK_F5;
const VK_KEY KeyMsg::F6 = VK_F6;
const VK_KEY KeyMsg::F7 = VK_F7;
const VK_KEY KeyMsg::F8 = VK_F8;
const VK_KEY KeyMsg::F9 = VK_F9;
const VK_KEY KeyMsg::F10 = VK_F10;
const VK_KEY KeyMsg::F11 = VK_F11;
const VK_KEY KeyMsg::F12 = VK_F12;

static VK_KEY convert(int ascii);

KeyMsg::KeyMsg(TString name, const KEY_CALLBACK & callback) :
	m_sName(name),
	m_callback(callback),
	m_pParentScene(nullptr),
	m_bRunning(true),
	m_bWaiting(false)
{
}

KeyMsg::~KeyMsg()
{
}

void KeyMsg::onKbHit(VK_KEY key)
{
	m_callback(key);
}

void KeyMsg::start()
{
	m_bRunning = true;
}

void KeyMsg::stop()
{
	m_bRunning = false;
}

void KeyMsg::wait()
{
	m_bWaiting = true;
}

void KeyMsg::notify()
{
	m_bWaiting = false;
}

void KeyMsg::__exec()
{
	if (_kbhit())						// 检测有无按键消息
	{
		VK_KEY key = convert(_getch());	// 获取键盘消息

		for (auto l : s_vListeners)		// 分发该消息
		{
			if (!l->m_bWaiting && l->m_bRunning)
			{
				l->onKbHit(key);		// 执行按键回调函数
			}
		}
	}
}

void KeyMsg::addListener(TString name, const KEY_CALLBACK & callback)
{
	// 创建新的监听对象
	auto listener = new KeyMsg(name, callback);
	// 绑定在场景上
	listener->m_pParentScene = EApp::getLoadingScene();
	// 添加新的按键回调函数
	s_vListeners.push_back(listener);
}

void KeyMsg::startListener(TString name)
{
	// 查找名称相同的监听器
	for (auto l : s_vListeners)
	{
		if (l->m_sName == name && l->m_pParentScene == EApp::getCurrentScene())
		{
			l->start();
		}
	}
}

void KeyMsg::stopListener(TString name)
{
	// 查找名称相同的监听器
	for (auto l : s_vListeners)
	{
		if (l->m_sName == name && l->m_pParentScene == EApp::getCurrentScene())
		{
			l->stop();
		}
	}
}

void KeyMsg::delListener(TString name)
{
	// 创建迭代器
	std::vector<KeyMsg*>::iterator iter;
	// 循环遍历所有监听器
	for (iter = s_vListeners.begin(); iter != s_vListeners.end();)
	{
		// 查找相同名称的监听器
		if ((*iter)->m_sName == name && (*iter)->m_pParentScene == EApp::getCurrentScene())
		{
			// 删除该定时器
			delete (*iter);
			iter = s_vListeners.erase(iter);
		}
		else
		{
			iter++;
		}
	}
}

void KeyMsg::notifyAllSceneListeners(Scene * scene)
{
	for (auto l : s_vListeners)
	{
		if (l->m_pParentScene == scene)
		{
			l->notify();
		}
	}
}

void KeyMsg::waitAllSceneListeners(Scene * scene)
{
	for (auto l : s_vListeners)
	{
		if (l->m_pParentScene == scene)
		{
			l->wait();
		}
	}
}

void KeyMsg::clearAllSceneListeners(Scene * scene)
{
	// 创建迭代器
	std::vector<KeyMsg*>::iterator iter;
	// 循环遍历所有监听器
	for (iter = s_vListeners.begin(); iter != s_vListeners.end();)
	{
		// 查找相同名称的监听器
		if ((*iter)->m_pParentScene == scene)
		{
			// 删除该定时器
			delete (*iter);
			iter = s_vListeners.erase(iter);
		}
		else
		{
			iter++;
		}
	}
}

void KeyMsg::clearAllListeners()
{
	// 删除所有监听器
	for (auto l : s_vListeners)
	{
		delete l;
	}
	// 清空容器
	s_vListeners.clear();
}

bool KeyMsg::isKeyDown(VK_KEY key)
{
	// 获取 key 的按下情况
	return (GetAsyncKeyState(key) & 0x8000);
}


VK_KEY convert(int ascii)
{
	// 字母键
	if (ascii >= 'a' && ascii <= 'z')
	{
		return VK_KEY(ascii - ('a' - 'A'));
	}
	else if (ascii >= 'A' && ascii <= 'Z')
	{
		return VK_KEY(ascii);
	}
	// 数字键
	else if (ascii >= '0' && ascii <= '9')
	{
		return VK_KEY(ascii);
	}
	// 回车、空格、Esc键
	else if (ascii == 0x0D || ascii == 0x20 || ascii == 0x1B)
	{
		return VK_KEY(ascii);
	}
	// 功能键
	else if (ascii == 0 || ascii == 0xE0)
	{
		switch (_getch())
		{
		case 72:
			return KeyMsg::Up;
		case 75:
			return KeyMsg::Left;
		case 77:
			return KeyMsg::Right;
		case 80:
			return KeyMsg::Down;
		case 59:
			return KeyMsg::F1;
		case 60:
			return KeyMsg::F2;
		case 61:
			return KeyMsg::F3;
		case 62:
			return KeyMsg::F4;
		case 63:
			return KeyMsg::F5;
		case 64:
			return KeyMsg::F6;
		case 65:
			return KeyMsg::F7;
		case 66:
			return KeyMsg::F8;
		case 67:
			return KeyMsg::F9;
		case 133:
			return KeyMsg::F10;
		case 134:
			return KeyMsg::F11;
		default:
			return 0;
		}
	}
	return 0;
}