/*
 * 86Box	A hypervisor and IBM PC system emulator that specializes in
 *		running old operating systems and software designed for IBM
 *		PC systems and compatibles from 1981 through fairly recent
 *		system designs based on the PCI bus.
 *
 *		This file is part of the 86Box distribution.
 *
 *		Windows raw keyboard input handler.
 *
 * Version:	@(#)win_d3d.cc	1.0.0	2017/05/30
 *
 * Author:	Miran Grca, <mgrca8@gmail.com>
 *		Copyright 2016-2017 Miran Grca.
 */

#define UNICODE
#define  _WIN32_WINNT 0x0501
#define BITMAP WINDOWS_BITMAP
#include <windows.h>
#include <windowsx.h>
#undef BITMAP

#include <commctrl.h>
#include <commdlg.h>
#include <process.h>
#include <string.h>
#include <stdio.h>
#include <stdarg.h>
#include <stdlib.h>
#include <stdint.h>

#include "../device.h"
#include "plat_keyboard.h"

#include "win.h"

#ifndef MAPVK_VK_TO_VSC
#define MAPVK_VK_TO_VSC 0
#endif

static uint16_t	scancode_map[65536];

/* This is so we can disambiguate scan codes that would otherwise conflict and get
   passed on incorrectly. */
UINT16 convert_scan_code(UINT16 scan_code)
{
	switch (scan_code)
        {
		case 0xE001:
		return 0xF001;
		case 0xE002:
		return 0xF002;
		case 0xE0AA:
		return 0xF003;
		case 0xE005:
		return 0xF005;
		case 0xE006:
		return 0xF006;
		case 0xE007:
		return 0xF007;
		case 0xE071:
		return 0xF008;
		case 0xE072:
		return 0xF009;
		case 0xE07F:
		return 0xF00A;
		case 0xE0E1:
		return 0xF00B;
		case 0xE0EE:
		return 0xF00C;
		case 0xE0F1:
		return 0xF00D;
		case 0xE0FE:
		return 0xF00E;
		case 0xE0EF:
		return 0xF00F;

		default:
		return scan_code;
	}
}

void get_registry_key_map()
{
	WCHAR *keyName = L"SYSTEM\\CurrentControlSet\\Control\\Keyboard Layout";
	WCHAR *valueName = L"Scancode Map";
	unsigned char buf[32768];
	DWORD bufSize;
	HKEY hKey;
	int j;
	UINT32 *bufEx2;
	int scMapCount;
	UINT16 *bufEx;
	int scancode_unmapped;
	int scancode_mapped;

 	/* First, prepare the default scan code map list which is 1:1.
 	   Remappings will be inserted directly into it.
 	   65536 bytes so scan codes fit in easily and it's easy to find what each maps too,
 	   since each array element is a scan code and provides for E0, etc. ones too. */
	for (j = 0; j < 65536; j++)
		scancode_map[j] = convert_scan_code(j);

	bufSize = 32768;
 	/* Get the scan code remappings from:
 	   HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Control\Keyboard Layout */
	if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, keyName, 0, 1, &hKey) == ERROR_SUCCESS)
        {
		if(RegQueryValueEx(hKey, valueName, NULL, NULL, buf, &bufSize) == ERROR_SUCCESS)
                {
			bufEx2 = (UINT32 *) buf;
			scMapCount = bufEx2[2];
			if ((bufSize != 0) && (scMapCount != 0))
                        {
				bufEx = (UINT16 *) (buf + 12);
				for (j = 0; j < scMapCount*2; j += 2)
 				{
 					/* Each scan code is 32-bit: 16 bits of remapped scan code,
 					   and 16 bits of original scan code. */
  					scancode_unmapped = bufEx[j + 1];
  					scancode_mapped = bufEx[j];

  					scancode_mapped = convert_scan_code(scancode_mapped);

					/* Fixes scan code map logging. */
  					scancode_map[scancode_unmapped] = scancode_mapped;
  				}
			}
		}
		RegCloseKey(hKey);
	}
}

void process_raw_input(LPARAM lParam, int infocus)
{
	uint32_t ri_size = 0;
	UINT size;
	RAWINPUT *raw;
	USHORT scancode;

	if (!infocus)
	{
		return;
	}

	GetRawInputData((HRAWINPUT)lParam, RID_INPUT, NULL, &size, sizeof(RAWINPUTHEADER));

	raw = malloc(size);

	if (raw == NULL)
	{
		return;
	}

	/* Here we read the raw input data for the keyboard */
	ri_size = GetRawInputData((HRAWINPUT)(lParam), RID_INPUT, raw, &size, sizeof(RAWINPUTHEADER));

	if(ri_size != size)
	{
		return;
	}

	/* If the input is keyboard, we process it */
	if (raw->header.dwType == RIM_TYPEKEYBOARD)
	{
		RAWKEYBOARD rawKB = raw->data.keyboard;
		scancode = rawKB.MakeCode;

		/* If it's not a scan code that starts with 0xE1 */
		if (!(rawKB.Flags & RI_KEY_E1))
		{
			if (rawKB.Flags & RI_KEY_E0)
			{
				scancode |= (0xE0 << 8);
			}

			/* Remap it according to the list from the Registry */
			scancode = scancode_map[scancode];

			if ((scancode >> 8) == 0xF0)
			{
				scancode |= 0x100; /* Extended key code in disambiguated format */
			}
			else if ((scancode >> 8) == 0xE0)
			{
				scancode |= 0x80; /* Normal extended key code */
			}

			/* If it's not 0 (therefore not 0xE1, 0xE2, etc),
			   then pass it on to the rawinputkey array */
			if (!(scancode & 0xf00))
			{
				recv_key[scancode & 0x1ff] = !(rawKB.Flags & RI_KEY_BREAK);
			}
		}
		else
		{
			if (rawKB.MakeCode == 0x1D)
			{
				scancode = 0xFF;
			}
			if (!(scancode & 0xf00))
			{
				recv_key[scancode & 0x1ff] = !(rawKB.Flags & RI_KEY_BREAK);
			}
		}
	}

	free(raw);
}