#include "stdafx.h"
#include "Emu/Memory/Memory.h"
#include "Emu/System.h"
#include "Emu/SysCalls/Modules.h"

#include "Emu/Io/Keyboard.h"
#include "cellKb.h"

extern Module *sys_io;

int cellKbInit(u32 max_connect)
{
	sys_io->Warning("cellKbInit(max_connect=%d)", max_connect);

	if (Emu.GetKeyboardManager().IsInited())
		return CELL_KB_ERROR_ALREADY_INITIALIZED;
	if (max_connect > 7)
		return CELL_KB_ERROR_INVALID_PARAMETER;

	Emu.GetKeyboardManager().Init(max_connect);

	return CELL_OK;
}

int cellKbEnd()
{
	sys_io->Log("cellKbEnd()");

	if (!Emu.GetKeyboardManager().IsInited())
		return CELL_KB_ERROR_UNINITIALIZED;

	Emu.GetKeyboardManager().Close();
	return CELL_OK;
}

int cellKbClearBuf(u32 port_no)
{
	sys_io->Log("cellKbClearBuf(port_no=%d)", port_no);

	if (!Emu.GetKeyboardManager().IsInited())
		return CELL_KB_ERROR_UNINITIALIZED;

	if (port_no >= Emu.GetKeyboardManager().GetKeyboards().size())
		return CELL_KB_ERROR_INVALID_PARAMETER;

	//?

	return CELL_OK;
}

u16 cellKbCnvRawCode(u32 arrange, u32 mkey, u32 led, u16 rawcode)
{
	sys_io->Log("cellKbCnvRawCode(arrange=%d,mkey=%d,led=%d,rawcode=%d)", arrange, mkey, led, rawcode);

	// CELL_KB_RAWDAT
	if (rawcode <= 0x03 || rawcode == 0x29 || rawcode == 0x35 || (rawcode >= 0x39 && rawcode <= 0x53) || rawcode == 0x65 || rawcode == 0x88 || rawcode == 0x8A || rawcode == 0x8B)
	{
		return rawcode | 0x8000;
	}

	// CELL_KB_NUMPAD
	if (rawcode >= 0x59 && rawcode <= 0x61) return (rawcode - 0x28) | 0x4000; // '1' - '9'
	if (rawcode == 0x62) return 0x30 | 0x4000;                                // '0'
	if (rawcode == 0x53) return 0x00 | 0x4000;                                // 'Num Lock'
	if (rawcode == 0x54) return 0x2F | 0x4000;                                // '/'
	if (rawcode == 0x55) return 0x2A | 0x4000;                                // '*'
	if (rawcode == 0x56) return 0x2D | 0x4000;                                // '-'
	if (rawcode == 0x57) return 0x2B | 0x4000;                                // '+'
	if (rawcode == 0x58) return 0x0A | 0x4000;                                // '\n'

	// ASCII
	if (rawcode >= 0x04 && rawcode <= 0x1D)                                   // 'A' - 'Z'
	{
		rawcode -= 
			(mkey&(CELL_KB_MKEY_L_SHIFT|CELL_KB_MKEY_R_SHIFT)) ? 
			((led&(CELL_KB_LED_CAPS_LOCK)) ? 0 : 0x20) :
			((led&(CELL_KB_LED_CAPS_LOCK)) ? 0x20 : 0);
		return rawcode + 0x5D;
	}				
	if (rawcode >= 0x1E && rawcode <= 0x26) return rawcode + 0x13;            // '1' - '9'
	if (rawcode == 0x27) return 0x30;                                         // '0'
	if (rawcode == 0x28) return 0x0A;                                         // '\n'
	if (rawcode == 0x2B) return 0x09;                                         // '\t'
	if (rawcode == 0x2C) return 0x20;                                         // ' '
	if (rawcode == 0x2D) return 0x2D;                                         // '-'
	if (rawcode == 0x2E) return 0x3D;                                         // '='
	if (rawcode == 0x36) return 0x2C;                                         // ','
	if (rawcode == 0x37) return 0x2E;                                         // '.'
	if (rawcode == 0x38) return 0x2F;                                         // '/'
	if (rawcode == 0x87) return 0x5C;                                         // '\'

	// (TODO: Add more cases)

	return 0x0000;
}

int cellKbGetInfo(vm::ptr<CellKbInfo> info)
{
	sys_io->Log("cellKbGetInfo(info_addr=0x%x)", info.addr());

	if (!Emu.GetKeyboardManager().IsInited())
		return CELL_KB_ERROR_UNINITIALIZED;

	const KbInfo& current_info = Emu.GetKeyboardManager().GetInfo();
	info->max_connect = current_info.max_connect;
	info->now_connect = current_info.now_connect;
	info->info = current_info.info;

	for (u32 i=0; i<CELL_KB_MAX_KEYBOARDS; i++)
	{
		info->status[i] = current_info.status[i];
	}
	
	return CELL_OK;
}

int cellKbRead(u32 port_no, vm::ptr<CellKbData> data)
{
	sys_io->Log("cellKbRead(port_no=%d,info_addr=0x%x)", port_no, data.addr());

	const std::vector<Keyboard>& keyboards = Emu.GetKeyboardManager().GetKeyboards();
	if (!Emu.GetKeyboardManager().IsInited())
		return CELL_KB_ERROR_UNINITIALIZED;

	if (port_no >= keyboards.size())
		return CELL_KB_ERROR_INVALID_PARAMETER;

	KbData& current_data = Emu.GetKeyboardManager().GetData(port_no);
	data->led = current_data.led;
	data->mkey = current_data.mkey;
	data->len = std::min((u32)current_data.len, CELL_KB_MAX_KEYCODES);

	for (s32 i=0; i<current_data.len; i++)
	{
		data->keycode[i] = current_data.keycode[i];
	}

	current_data.len = 0;
	
	return CELL_OK;
}

int cellKbSetCodeType(u32 port_no, u32 type)
{
	sys_io->Log("cellKbSetCodeType(port_no=%d,type=%d)", port_no, type);

	if (!Emu.GetKeyboardManager().IsInited())
		return CELL_KB_ERROR_UNINITIALIZED;
	
	KbConfig& current_config = Emu.GetKeyboardManager().GetConfig(port_no);
	current_config.code_type = type;
	return CELL_OK;
}

int cellKbSetLEDStatus(u32 port_no, u8 led)
{
	UNIMPLEMENTED_FUNC(sys_io);
	return CELL_OK;
}

int cellKbSetReadMode(u32 port_no, u32 rmode)
{
	sys_io->Log("cellKbSetReadMode(port_no=%d,rmode=%d)", port_no, rmode);

	if (!Emu.GetKeyboardManager().IsInited())
		return CELL_KB_ERROR_UNINITIALIZED;
	
	KbConfig& current_config = Emu.GetKeyboardManager().GetConfig(port_no);
	current_config.read_mode = rmode;

	return CELL_OK;
}

int cellKbGetConfiguration(u32 port_no, vm::ptr<CellKbConfig> config)
{
	sys_io->Log("cellKbGetConfiguration(port_no=%d,config_addr=0x%x)", port_no, config.addr());

	if (!Emu.GetKeyboardManager().IsInited())
		return CELL_KB_ERROR_UNINITIALIZED;

	const KbConfig& current_config = Emu.GetKeyboardManager().GetConfig(port_no);
	config->arrange = current_config.arrange;
	config->read_mode = current_config.read_mode;
	config->code_type = current_config.code_type;

	return CELL_OK;
}

void cellKb_init()
{
	sys_io->AddFunc(0x433f6ec0, cellKbInit);
	sys_io->AddFunc(0xbfce3285, cellKbEnd);
	sys_io->AddFunc(0x2073b7f6, cellKbClearBuf);
	sys_io->AddFunc(0x4ab1fa77, cellKbCnvRawCode);
	sys_io->AddFunc(0x2f1774d5, cellKbGetInfo);
	sys_io->AddFunc(0xff0a21b7, cellKbRead);
	sys_io->AddFunc(0xa5f85e4d, cellKbSetCodeType);
	sys_io->AddFunc(0x3f72c56e, cellKbSetLEDStatus);
	sys_io->AddFunc(0xdeefdfa7, cellKbSetReadMode);
	sys_io->AddFunc(0x1f71ecbe, cellKbGetConfiguration);
}