782 lines
26 KiB
C++
Executable File
782 lines
26 KiB
C++
Executable File
//RawInput driver
|
|
//author: byuu
|
|
|
|
//this driver utilizes RawInput (WM_INPUT) to capture keyboard and mouse input.
|
|
//although this requires WinXP or newer, it is the only way to uniquely identify
|
|
//and independently map multiple keyboards and mice. DirectInput merges all
|
|
//keyboards and mice into one device per.
|
|
//
|
|
//as WM_INPUT lacks specific RAWINPUT structures for gamepads, giving only raw
|
|
//data, and because DirectInput supports up to 16 joypads, DirectInput is used
|
|
//for joypad mapping.
|
|
//
|
|
//further, Xbox 360 controllers are explicitly detected and supported through
|
|
//XInput. this is because under DirectInput, the LT / RT (trigger) buttons are
|
|
//merged into a single Z-axis -- making it impossible to detect both buttons
|
|
//being pressed at the same time. with XInput, the state of both trigger
|
|
//buttons can be read independently.
|
|
//
|
|
//so in essence, this is actually more of a hybrid driver.
|
|
|
|
#define DIRECTINPUT_VERSION 0x0800
|
|
#include <dinput.h>
|
|
#include <xinput.h>
|
|
|
|
namespace ruby {
|
|
|
|
#include "rawinput.hpp"
|
|
|
|
DWORD WINAPI RawInputThreadProc(void*);
|
|
LRESULT CALLBACK RawInputWindowProc(HWND, UINT, WPARAM, LPARAM);
|
|
|
|
class RawInput {
|
|
public:
|
|
HANDLE mutex;
|
|
HWND hwnd;
|
|
bool initialized;
|
|
bool ready;
|
|
|
|
struct Device {
|
|
HANDLE handle;
|
|
};
|
|
|
|
struct Keyboard : Device {
|
|
bool state[keyboard<>::length];
|
|
|
|
void update(RAWINPUT *input) {
|
|
unsigned code = input->data.keyboard.MakeCode;
|
|
unsigned flags = input->data.keyboard.Flags;
|
|
|
|
#define map(id, flag, name) if(code == id) state[name] = (bool)(flags == flag);
|
|
map(0x0001, 0, keyboard<>::escape)
|
|
map(0x003b, 0, keyboard<>::f1)
|
|
map(0x003c, 0, keyboard<>::f2)
|
|
map(0x003d, 0, keyboard<>::f3)
|
|
map(0x003e, 0, keyboard<>::f4)
|
|
map(0x003f, 0, keyboard<>::f5)
|
|
map(0x0040, 0, keyboard<>::f6)
|
|
map(0x0041, 0, keyboard<>::f7)
|
|
map(0x0042, 0, keyboard<>::f8)
|
|
map(0x0043, 0, keyboard<>::f9)
|
|
map(0x0044, 0, keyboard<>::f10)
|
|
map(0x0057, 0, keyboard<>::f11)
|
|
map(0x0058, 0, keyboard<>::f12)
|
|
|
|
map(0x0037, 2, keyboard<>::print_screen)
|
|
map(0x0046, 0, keyboard<>::scroll_lock)
|
|
map(0x001d, 4, keyboard<>::pause)
|
|
map(0x0029, 0, keyboard<>::tilde)
|
|
|
|
map(0x0002, 0, keyboard<>::num_1)
|
|
map(0x0003, 0, keyboard<>::num_2)
|
|
map(0x0004, 0, keyboard<>::num_3)
|
|
map(0x0005, 0, keyboard<>::num_4)
|
|
map(0x0006, 0, keyboard<>::num_5)
|
|
map(0x0007, 0, keyboard<>::num_6)
|
|
map(0x0008, 0, keyboard<>::num_7)
|
|
map(0x0009, 0, keyboard<>::num_8)
|
|
map(0x000a, 0, keyboard<>::num_9)
|
|
map(0x000b, 0, keyboard<>::num_0)
|
|
|
|
map(0x000c, 0, keyboard<>::dash)
|
|
map(0x000d, 0, keyboard<>::equal)
|
|
map(0x000e, 0, keyboard<>::backspace)
|
|
|
|
map(0x0052, 2, keyboard<>::insert)
|
|
map(0x0053, 2, keyboard<>::delete_)
|
|
map(0x0047, 2, keyboard<>::home)
|
|
map(0x004f, 2, keyboard<>::end)
|
|
map(0x0049, 2, keyboard<>::page_up)
|
|
map(0x0051, 2, keyboard<>::page_down)
|
|
|
|
map(0x001e, 0, keyboard<>::a)
|
|
map(0x0030, 0, keyboard<>::b)
|
|
map(0x002e, 0, keyboard<>::c)
|
|
map(0x0020, 0, keyboard<>::d)
|
|
map(0x0012, 0, keyboard<>::e)
|
|
map(0x0021, 0, keyboard<>::f)
|
|
map(0x0022, 0, keyboard<>::g)
|
|
map(0x0023, 0, keyboard<>::h)
|
|
map(0x0017, 0, keyboard<>::i)
|
|
map(0x0024, 0, keyboard<>::j)
|
|
map(0x0025, 0, keyboard<>::k)
|
|
map(0x0026, 0, keyboard<>::l)
|
|
map(0x0032, 0, keyboard<>::m)
|
|
map(0x0031, 0, keyboard<>::n)
|
|
map(0x0018, 0, keyboard<>::o)
|
|
map(0x0019, 0, keyboard<>::p)
|
|
map(0x0010, 0, keyboard<>::q)
|
|
map(0x0013, 0, keyboard<>::r)
|
|
map(0x001f, 0, keyboard<>::s)
|
|
map(0x0014, 0, keyboard<>::t)
|
|
map(0x0016, 0, keyboard<>::u)
|
|
map(0x002f, 0, keyboard<>::v)
|
|
map(0x0011, 0, keyboard<>::w)
|
|
map(0x002d, 0, keyboard<>::x)
|
|
map(0x0015, 0, keyboard<>::y)
|
|
map(0x002c, 0, keyboard<>::z)
|
|
|
|
map(0x001a, 0, keyboard<>::lbracket)
|
|
map(0x001b, 0, keyboard<>::rbracket)
|
|
map(0x002b, 0, keyboard<>::backslash)
|
|
map(0x0027, 0, keyboard<>::semicolon)
|
|
map(0x0028, 0, keyboard<>::apostrophe)
|
|
map(0x0033, 0, keyboard<>::comma)
|
|
map(0x0034, 0, keyboard<>::period)
|
|
map(0x0035, 0, keyboard<>::slash)
|
|
|
|
map(0x004f, 0, keyboard<>::pad_1)
|
|
map(0x0050, 0, keyboard<>::pad_2)
|
|
map(0x0051, 0, keyboard<>::pad_3)
|
|
map(0x004b, 0, keyboard<>::pad_4)
|
|
map(0x004c, 0, keyboard<>::pad_5)
|
|
map(0x004d, 0, keyboard<>::pad_6)
|
|
map(0x0047, 0, keyboard<>::pad_7)
|
|
map(0x0048, 0, keyboard<>::pad_8)
|
|
map(0x0049, 0, keyboard<>::pad_9)
|
|
map(0x0052, 0, keyboard<>::pad_0)
|
|
|
|
map(0x0053, 0, keyboard<>::point)
|
|
map(0x001c, 2, keyboard<>::enter)
|
|
map(0x004e, 0, keyboard<>::add)
|
|
map(0x004a, 0, keyboard<>::subtract)
|
|
map(0x0037, 0, keyboard<>::multiply)
|
|
map(0x0035, 2, keyboard<>::divide)
|
|
|
|
map(0x0045, 0, keyboard<>::num_lock)
|
|
map(0x003a, 0, keyboard<>::caps_lock)
|
|
|
|
//pause signals 0x1d:4 + 0x45:0, whereas num_lock signals only 0x45:0.
|
|
//this makes it impractical to detect both pause+num_lock independently.
|
|
//workaround: always detect pause; detect num_lock only when pause is released.
|
|
if(state[keyboard<>::pause]) state[keyboard<>::num_lock] = false;
|
|
|
|
map(0x0048, 2, keyboard<>::up)
|
|
map(0x0050, 2, keyboard<>::down)
|
|
map(0x004b, 2, keyboard<>::left)
|
|
map(0x004d, 2, keyboard<>::right)
|
|
|
|
map(0x000f, 0, keyboard<>::tab)
|
|
map(0x001c, 0, keyboard<>::return_)
|
|
map(0x0039, 0, keyboard<>::spacebar)
|
|
|
|
map(0x001d, 0, keyboard<>::lctrl)
|
|
map(0x001d, 2, keyboard<>::rctrl)
|
|
map(0x0038, 0, keyboard<>::lalt)
|
|
map(0x0038, 2, keyboard<>::ralt)
|
|
map(0x002a, 0, keyboard<>::lshift)
|
|
map(0x0036, 0, keyboard<>::rshift)
|
|
map(0x005b, 2, keyboard<>::lsuper)
|
|
map(0x005c, 2, keyboard<>::rsuper)
|
|
map(0x005d, 2, keyboard<>::menu)
|
|
#undef map
|
|
}
|
|
|
|
Keyboard() {
|
|
for(unsigned i = 0; i < keyboard<>::length; i++) state[i] = false;
|
|
}
|
|
};
|
|
|
|
struct Mouse : Device {
|
|
signed xDistance;
|
|
signed yDistance;
|
|
signed zDistance;
|
|
unsigned buttonState;
|
|
|
|
void sync() {
|
|
xDistance = 0;
|
|
yDistance = 0;
|
|
zDistance = 0;
|
|
}
|
|
|
|
void update(RAWINPUT *input) {
|
|
if((input->data.mouse.usFlags & 1) == MOUSE_MOVE_RELATIVE) {
|
|
xDistance += input->data.mouse.lLastX;
|
|
yDistance += input->data.mouse.lLastY;
|
|
}
|
|
|
|
if(input->data.mouse.usButtonFlags & RI_MOUSE_BUTTON_1_DOWN) buttonState |= 1 << 0;
|
|
if(input->data.mouse.usButtonFlags & RI_MOUSE_BUTTON_1_UP ) buttonState &=~ 1 << 0;
|
|
if(input->data.mouse.usButtonFlags & RI_MOUSE_BUTTON_2_DOWN) buttonState |= 1 << 2; //swap middle and right buttons,
|
|
if(input->data.mouse.usButtonFlags & RI_MOUSE_BUTTON_2_UP ) buttonState &=~ 1 << 2; //for consistency with Linux:
|
|
if(input->data.mouse.usButtonFlags & RI_MOUSE_BUTTON_3_DOWN) buttonState |= 1 << 1; //left = 0, middle = 1, right = 2
|
|
if(input->data.mouse.usButtonFlags & RI_MOUSE_BUTTON_3_UP ) buttonState &=~ 1 << 1;
|
|
if(input->data.mouse.usButtonFlags & RI_MOUSE_BUTTON_4_DOWN) buttonState |= 1 << 3;
|
|
if(input->data.mouse.usButtonFlags & RI_MOUSE_BUTTON_4_UP ) buttonState &=~ 1 << 3;
|
|
if(input->data.mouse.usButtonFlags & RI_MOUSE_BUTTON_5_DOWN) buttonState |= 1 << 4;
|
|
if(input->data.mouse.usButtonFlags & RI_MOUSE_BUTTON_5_UP ) buttonState &=~ 1 << 4;
|
|
|
|
if(input->data.mouse.usButtonFlags & RI_MOUSE_WHEEL) {
|
|
zDistance += (int16_t)input->data.mouse.usButtonData;
|
|
}
|
|
}
|
|
|
|
Mouse() {
|
|
xDistance = yDistance = zDistance = 0;
|
|
buttonState = 0;
|
|
}
|
|
};
|
|
|
|
//keep track of gamepads for the sole purpose of distinguishing XInput devices
|
|
//from all other devices. this is necessary, as DirectInput does not provide
|
|
//a way to retrieve the necessary RIDI_DEVICENAME string.
|
|
struct Gamepad : Device {
|
|
bool isXInputDevice;
|
|
uint16_t vendorId;
|
|
uint16_t productId;
|
|
};
|
|
|
|
vector<Keyboard> lkeyboard;
|
|
vector<Mouse> lmouse;
|
|
vector<Gamepad> lgamepad;
|
|
|
|
LRESULT window_proc(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam) {
|
|
if(msg == WM_INPUT) {
|
|
unsigned size = 0;
|
|
GetRawInputData((HRAWINPUT)lparam, RID_INPUT, NULL, &size, sizeof(RAWINPUTHEADER));
|
|
RAWINPUT *input = new RAWINPUT[size];
|
|
GetRawInputData((HRAWINPUT)lparam, RID_INPUT, input, &size, sizeof(RAWINPUTHEADER));
|
|
WaitForSingleObject(mutex, INFINITE);
|
|
|
|
if(input->header.dwType == RIM_TYPEKEYBOARD) {
|
|
for(unsigned i = 0; i < lkeyboard.size(); i++) {
|
|
if(input->header.hDevice == lkeyboard[i].handle) {
|
|
lkeyboard[i].update(input);
|
|
break;
|
|
}
|
|
}
|
|
} else if(input->header.dwType == RIM_TYPEMOUSE) {
|
|
for(unsigned i = 0; i < lmouse.size(); i++) {
|
|
if(input->header.hDevice == lmouse[i].handle) {
|
|
lmouse[i].update(input);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
ReleaseMutex(mutex);
|
|
//allow propogation of WM_INPUT message
|
|
LRESULT result = DefRawInputProc(&input, size, sizeof(RAWINPUTHEADER));
|
|
delete[] input;
|
|
return result;
|
|
}
|
|
|
|
return DefWindowProc(hwnd, msg, wparam, lparam);
|
|
}
|
|
|
|
//this is used to sort device IDs
|
|
struct DevicePool {
|
|
HANDLE handle;
|
|
char name[4096];
|
|
bool operator<(const DevicePool &pool) const { return strcmp(name, pool.name) < 0; }
|
|
};
|
|
|
|
int main() {
|
|
//create an invisible window to act as a sink, capturing all WM_INPUT messages
|
|
WNDCLASS wc;
|
|
wc.cbClsExtra = 0;
|
|
wc.cbWndExtra = 0;
|
|
wc.hbrBackground = (HBRUSH)COLOR_WINDOW;
|
|
wc.hCursor = LoadCursor(0, IDC_ARROW);
|
|
wc.hIcon = LoadIcon(0, IDI_APPLICATION);
|
|
wc.hInstance = GetModuleHandle(0);
|
|
wc.lpfnWndProc = RawInputWindowProc;
|
|
wc.lpszClassName = "RawInputClass";
|
|
wc.lpszMenuName = 0;
|
|
wc.style = CS_VREDRAW | CS_HREDRAW;
|
|
RegisterClass(&wc);
|
|
|
|
hwnd = CreateWindow("RawInputClass", "RawInputClass", WS_POPUP,
|
|
0, 0, 64, 64, 0, 0, GetModuleHandle(0), 0);
|
|
|
|
//enumerate all HID devices
|
|
unsigned devices = 0;
|
|
GetRawInputDeviceList(NULL, &devices, sizeof(RAWINPUTDEVICELIST));
|
|
RAWINPUTDEVICELIST *list = new RAWINPUTDEVICELIST[devices];
|
|
GetRawInputDeviceList(list, &devices, sizeof(RAWINPUTDEVICELIST));
|
|
|
|
//sort all devices by name. this has two important properties:
|
|
//1) it consistently orders peripherals, so mapped IDs remain constant
|
|
//2) it sorts the virtual keyboard and mouse to the bottom of the list
|
|
// (real devices start with \\?\HID#, virtual with \\?\Root#)
|
|
DevicePool pool[devices];
|
|
for(unsigned i = 0; i < devices; i++) {
|
|
pool[i].handle = list[i].hDevice;
|
|
unsigned size = sizeof(pool[i].name) - 1;
|
|
GetRawInputDeviceInfo(list[i].hDevice, RIDI_DEVICENAME, &pool[i].name, &size);
|
|
}
|
|
nall::sort(pool, devices);
|
|
delete[] list;
|
|
|
|
for(unsigned i = 0; i < devices; i++) {
|
|
RID_DEVICE_INFO info;
|
|
info.cbSize = sizeof(RID_DEVICE_INFO);
|
|
|
|
unsigned size = info.cbSize;
|
|
GetRawInputDeviceInfo(pool[i].handle, RIDI_DEVICEINFO, &info, &size);
|
|
|
|
if(info.dwType == RIM_TYPEKEYBOARD) {
|
|
unsigned n = lkeyboard.size();
|
|
lkeyboard[n].handle = pool[i].handle;
|
|
} else if(info.dwType == RIM_TYPEMOUSE) {
|
|
unsigned n = lmouse.size();
|
|
lmouse[n].handle = pool[i].handle;
|
|
} else if(info.dwType == RIM_TYPEHID) {
|
|
//if this is a gamepad or joystick device ...
|
|
if(info.hid.usUsagePage == 1 && (info.hid.usUsage == 4 || info.hid.usUsage == 5)) {
|
|
//... then cache device information for later use
|
|
unsigned n = lgamepad.size();
|
|
lgamepad[n].handle = pool[i].handle;
|
|
lgamepad[n].vendorId = (uint16_t)info.hid.dwVendorId;
|
|
lgamepad[n].productId = (uint16_t)info.hid.dwProductId;
|
|
|
|
//per MSDN: XInput devices have "IG_" in their device strings,
|
|
//which is how they should be identified.
|
|
const char *p = strstr(pool[i].name, "IG_");
|
|
lgamepad[n].isXInputDevice = (bool)p;
|
|
}
|
|
}
|
|
}
|
|
|
|
RAWINPUTDEVICE device[2];
|
|
//capture all keyboard input
|
|
device[0].usUsagePage = 1;
|
|
device[0].usUsage = 6;
|
|
device[0].dwFlags = RIDEV_INPUTSINK;
|
|
device[0].hwndTarget = hwnd;
|
|
//capture all mouse input
|
|
device[1].usUsagePage = 1;
|
|
device[1].usUsage = 2;
|
|
device[1].dwFlags = RIDEV_INPUTSINK;
|
|
device[1].hwndTarget = hwnd;
|
|
RegisterRawInputDevices(device, 2, sizeof(RAWINPUTDEVICE));
|
|
|
|
WaitForSingleObject(mutex, INFINITE);
|
|
ready = true;
|
|
ReleaseMutex(mutex);
|
|
|
|
while(true) {
|
|
MSG msg;
|
|
GetMessage(&msg, hwnd, 0, 0);
|
|
TranslateMessage(&msg);
|
|
DispatchMessage(&msg);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
RawInput() : initialized(false), ready(false) {
|
|
}
|
|
};
|
|
|
|
static RawInput rawinput;
|
|
|
|
DWORD WINAPI RawInputThreadProc(void*) {
|
|
return rawinput.main();
|
|
}
|
|
|
|
LRESULT CALLBACK RawInputWindowProc(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam) {
|
|
return rawinput.window_proc(hwnd, msg, wparam, lparam);
|
|
}
|
|
|
|
class XInput {
|
|
public:
|
|
struct Gamepad {
|
|
unsigned id;
|
|
|
|
int16_t hat;
|
|
int16_t axis[6];
|
|
bool button[10];
|
|
|
|
void poll(XINPUT_STATE &state) {
|
|
hat = joypad<>::hat_center;
|
|
if(state.Gamepad.wButtons & XINPUT_GAMEPAD_DPAD_UP ) hat |= joypad<>::hat_up;
|
|
if(state.Gamepad.wButtons & XINPUT_GAMEPAD_DPAD_RIGHT) hat |= joypad<>::hat_right;
|
|
if(state.Gamepad.wButtons & XINPUT_GAMEPAD_DPAD_DOWN ) hat |= joypad<>::hat_down;
|
|
if(state.Gamepad.wButtons & XINPUT_GAMEPAD_DPAD_LEFT ) hat |= joypad<>::hat_left;
|
|
|
|
axis[0] = (int16_t)state.Gamepad.sThumbLX;
|
|
axis[1] = (int16_t)state.Gamepad.sThumbLY;
|
|
axis[2] = (int16_t)state.Gamepad.sThumbRX;
|
|
axis[3] = (int16_t)state.Gamepad.sThumbRY;
|
|
|
|
//transform left and right trigger ranges:
|
|
//from: 0 (low, eg released) to 255 (high, eg pressed all the way down)
|
|
//to: +32767 (low) to -32768 (high)
|
|
uint16_t triggerX = state.Gamepad.bLeftTrigger;
|
|
uint16_t triggerY = state.Gamepad.bRightTrigger;
|
|
|
|
triggerX = (triggerX << 8) | triggerX;
|
|
triggerY = (triggerY << 8) | triggerY;
|
|
|
|
axis[4] = (~triggerX) - 32768;
|
|
axis[5] = (~triggerY) - 32768;
|
|
|
|
button[0] = (bool)(state.Gamepad.wButtons & XINPUT_GAMEPAD_A);
|
|
button[1] = (bool)(state.Gamepad.wButtons & XINPUT_GAMEPAD_B);
|
|
button[2] = (bool)(state.Gamepad.wButtons & XINPUT_GAMEPAD_X);
|
|
button[3] = (bool)(state.Gamepad.wButtons & XINPUT_GAMEPAD_Y);
|
|
button[4] = (bool)(state.Gamepad.wButtons & XINPUT_GAMEPAD_BACK);
|
|
button[5] = (bool)(state.Gamepad.wButtons & XINPUT_GAMEPAD_START);
|
|
button[6] = (bool)(state.Gamepad.wButtons & XINPUT_GAMEPAD_LEFT_SHOULDER);
|
|
button[7] = (bool)(state.Gamepad.wButtons & XINPUT_GAMEPAD_RIGHT_SHOULDER);
|
|
button[8] = (bool)(state.Gamepad.wButtons & XINPUT_GAMEPAD_LEFT_THUMB);
|
|
button[9] = (bool)(state.Gamepad.wButtons & XINPUT_GAMEPAD_RIGHT_THUMB);
|
|
}
|
|
|
|
Gamepad() {
|
|
hat = joypad<>::hat_center;
|
|
for(unsigned n = 0; n < 6; n++) axis[n] = 0;
|
|
for(unsigned n = 0; n < 10; n++) button[n] = false;
|
|
}
|
|
};
|
|
|
|
vector<Gamepad> lgamepad;
|
|
|
|
void poll() {
|
|
for(unsigned i = 0; i < lgamepad.size(); i++) {
|
|
XINPUT_STATE state;
|
|
DWORD result = XInputGetState(lgamepad[i].id, &state);
|
|
if(result == ERROR_SUCCESS) lgamepad[i].poll(state);
|
|
}
|
|
}
|
|
|
|
void init() {
|
|
//XInput only supports up to four controllers
|
|
for(unsigned i = 0; i <= 3; i++) {
|
|
XINPUT_STATE state;
|
|
DWORD result = XInputGetState(i, &state);
|
|
if(result == ERROR_SUCCESS) {
|
|
//valid controller detected, add to gamepad list
|
|
unsigned n = lgamepad.size();
|
|
lgamepad[n].id = i;
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
static BOOL CALLBACK DirectInput_EnumJoypadsCallback(const DIDEVICEINSTANCE*, void*);
|
|
static BOOL CALLBACK DirectInput_EnumJoypadAxesCallback(const DIDEVICEOBJECTINSTANCE*, void*);
|
|
|
|
class DirectInput {
|
|
public:
|
|
HWND handle;
|
|
LPDIRECTINPUT8 context;
|
|
struct Gamepad {
|
|
LPDIRECTINPUTDEVICE8 handle;
|
|
|
|
int16_t hat[4];
|
|
int16_t axis[6];
|
|
bool button[128];
|
|
|
|
void poll(DIJOYSTATE2 &state) {
|
|
//POV hats
|
|
for(unsigned n = 0; n < 4; n++) {
|
|
hat[n] = joypad<>::hat_center;
|
|
|
|
//POV value is in clockwise-hundredth degree units
|
|
unsigned pov = state.rgdwPOV[n];
|
|
|
|
//some drivers report a centered POV hat as -1U, others as 65535U.
|
|
//>= 36000 will match both, as well as invalid ranges.
|
|
if(pov >= 36000) continue;
|
|
|
|
if(pov >= 31500 || pov <= 4500) hat[n] |= joypad<>::hat_up;
|
|
if(pov >= 4500 && pov <= 13500) hat[n] |= joypad<>::hat_right;
|
|
if(pov >= 13500 && pov <= 22500) hat[n] |= joypad<>::hat_down;
|
|
if(pov >= 22500 && pov <= 31500) hat[n] |= joypad<>::hat_left;
|
|
}
|
|
|
|
//axes
|
|
axis[0] = state.lX;
|
|
axis[1] = state.lY;
|
|
axis[2] = state.lZ;
|
|
axis[3] = state.lRx;
|
|
axis[4] = state.lRy;
|
|
axis[5] = state.lRz;
|
|
|
|
//buttons
|
|
for(unsigned n = 0; n < 128; n++) {
|
|
button[n] = (bool)state.rgbButtons[n];
|
|
}
|
|
}
|
|
|
|
Gamepad() {
|
|
handle = 0;
|
|
for(unsigned n = 0; n < 4; n++) hat[n] = joypad<>::hat_center;
|
|
for(unsigned n = 0; n < 6; n++) axis[n] = 0;
|
|
for(unsigned n = 0; n < 128; n++) button[n] = false;
|
|
}
|
|
};
|
|
vector<Gamepad> lgamepad;
|
|
|
|
void poll() {
|
|
for(unsigned i = 0; i < lgamepad.size(); i++) {
|
|
if(FAILED(lgamepad[i].handle->Poll())) {
|
|
lgamepad[i].handle->Acquire();
|
|
continue;
|
|
}
|
|
|
|
DIJOYSTATE2 state;
|
|
lgamepad[i].handle->GetDeviceState(sizeof(DIJOYSTATE2), &state);
|
|
lgamepad[i].poll(state);
|
|
}
|
|
}
|
|
|
|
bool init_joypad(const DIDEVICEINSTANCE *instance) {
|
|
//if this is an XInput device, do not acquire it via DirectInput ...
|
|
//the XInput driver above will handle said device.
|
|
for(unsigned i = 0; i < rawinput.lgamepad.size(); i++) {
|
|
uint32_t guid = MAKELONG(rawinput.lgamepad[i].vendorId, rawinput.lgamepad[i].productId);
|
|
if(guid == instance->guidProduct.Data1) {
|
|
if(rawinput.lgamepad[i].isXInputDevice == true) {
|
|
return DIENUM_CONTINUE;
|
|
}
|
|
}
|
|
}
|
|
|
|
if(FAILED(context->CreateDevice(instance->guidInstance, &device, 0))) {
|
|
return DIENUM_CONTINUE;
|
|
}
|
|
|
|
device->SetDataFormat(&c_dfDIJoystick2);
|
|
device->SetCooperativeLevel(handle, DISCL_NONEXCLUSIVE | DISCL_BACKGROUND);
|
|
device->EnumObjects(DirectInput_EnumJoypadAxesCallback, (void*)this, DIDFT_ABSAXIS);
|
|
unsigned n = lgamepad.size();
|
|
lgamepad[n].handle = device;
|
|
return DIENUM_CONTINUE;
|
|
}
|
|
|
|
bool init_axis(const DIDEVICEOBJECTINSTANCE *instance) {
|
|
DIPROPRANGE range;
|
|
range.diph.dwSize = sizeof(DIPROPRANGE);
|
|
range.diph.dwHeaderSize = sizeof(DIPROPHEADER);
|
|
range.diph.dwHow = DIPH_BYID;
|
|
range.diph.dwObj = instance->dwType;
|
|
range.lMin = -32768;
|
|
range.lMax = +32767;
|
|
device->SetProperty(DIPROP_RANGE, &range.diph);
|
|
return DIENUM_CONTINUE;
|
|
}
|
|
|
|
void init(HWND handle_) {
|
|
handle = handle_;
|
|
DirectInput8Create(GetModuleHandle(0), DIRECTINPUT_VERSION, IID_IDirectInput8, (void**)&context, 0);
|
|
context->EnumDevices(DI8DEVCLASS_GAMECTRL, DirectInput_EnumJoypadsCallback, (void*)this, DIEDFL_ATTACHEDONLY);
|
|
}
|
|
|
|
void term() {
|
|
for(unsigned i = 0; i < lgamepad.size(); i++) {
|
|
lgamepad[i].handle->Unacquire();
|
|
lgamepad[i].handle->Release();
|
|
}
|
|
lgamepad.reset();
|
|
|
|
if(context) {
|
|
context->Release();
|
|
context = 0;
|
|
}
|
|
}
|
|
|
|
private:
|
|
LPDIRECTINPUTDEVICE8 device;
|
|
};
|
|
|
|
BOOL CALLBACK DirectInput_EnumJoypadsCallback(const DIDEVICEINSTANCE *instance, void *p) {
|
|
return ((DirectInput*)p)->init_joypad(instance);
|
|
}
|
|
|
|
BOOL CALLBACK DirectInput_EnumJoypadAxesCallback(const DIDEVICEOBJECTINSTANCE *instance, void *p) {
|
|
return ((DirectInput*)p)->init_axis(instance);
|
|
}
|
|
|
|
class pInputRaw {
|
|
public:
|
|
InputRaw &self;
|
|
XInput xinput;
|
|
DirectInput dinput;
|
|
|
|
bool acquire_mouse;
|
|
bool cursor_visible;
|
|
|
|
struct {
|
|
HWND handle;
|
|
} settings;
|
|
|
|
bool cap(Input::Setting setting) {
|
|
if(setting == Input::Handle) return true;
|
|
if(setting == Input::KeyboardSupport) return true;
|
|
if(setting == Input::MouseSupport) return true;
|
|
if(setting == Input::JoypadSupport) return true;
|
|
return false;
|
|
}
|
|
|
|
uintptr_t get(Input::Setting setting) {
|
|
if(setting == Input::Handle) return (uintptr_t)settings.handle;
|
|
return false;
|
|
}
|
|
|
|
bool set(Input::Setting setting, uintptr_t param) {
|
|
if(setting == Input::Handle) {
|
|
settings.handle = (HWND)param;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool acquire() {
|
|
acquire_mouse = true;
|
|
if(cursor_visible == true) {
|
|
ShowCursor(cursor_visible = false);
|
|
}
|
|
return acquired();
|
|
}
|
|
|
|
bool unacquire() {
|
|
acquire_mouse = false;
|
|
ReleaseCapture();
|
|
ClipCursor(NULL);
|
|
if(cursor_visible == false) {
|
|
ShowCursor(cursor_visible = true);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool acquired() {
|
|
if(acquire_mouse == true) {
|
|
SetFocus(settings.handle);
|
|
SetCapture(settings.handle);
|
|
RECT rc;
|
|
GetWindowRect(settings.handle, &rc);
|
|
ClipCursor(&rc);
|
|
}
|
|
return GetCapture() == settings.handle;
|
|
}
|
|
|
|
bool poll(int16_t *table) {
|
|
memset(table, 0, nall::input_limit * sizeof(int16_t));
|
|
|
|
WaitForSingleObject(rawinput.mutex, INFINITE);
|
|
|
|
//=========
|
|
//Keyboards
|
|
//=========
|
|
for(unsigned i = 0; i < min(rawinput.lkeyboard.size(), (unsigned)keyboard<>::count); i++) {
|
|
unsigned index = keyboard<>::index(i, keyboard<>::none);
|
|
|
|
for(unsigned n = 0; n < keyboard<>::length; n++) {
|
|
table[index + n] = rawinput.lkeyboard[i].state[n];
|
|
}
|
|
}
|
|
|
|
//====
|
|
//Mice
|
|
//====
|
|
for(unsigned i = 0; i < min(rawinput.lmouse.size(), (unsigned)mouse<>::count); i++) {
|
|
unsigned index = mouse<>::index(i, mouse<>::none);
|
|
|
|
table[index + mouse<>::x] = rawinput.lmouse[i].xDistance;
|
|
table[index + mouse<>::y] = rawinput.lmouse[i].yDistance;
|
|
table[index + mouse<>::z] = rawinput.lmouse[i].zDistance;
|
|
|
|
for(unsigned n = 0; n < min(5U, (unsigned)mouse<>::buttons); n++) {
|
|
table[index + mouse<>::button + n] = (bool)(rawinput.lmouse[i].buttonState & (1 << n));
|
|
}
|
|
|
|
rawinput.lmouse[i].sync();
|
|
}
|
|
|
|
ReleaseMutex(rawinput.mutex);
|
|
|
|
unsigned joy = 0;
|
|
|
|
//==================
|
|
//XInput controllers
|
|
//==================
|
|
xinput.poll();
|
|
for(unsigned i = 0; i < xinput.lgamepad.size(); i++) {
|
|
if(joy >= joypad<>::count) break;
|
|
unsigned index = joypad<>::index(joy++, joypad<>::none);
|
|
|
|
table[index + joypad<>::hat + 0] = xinput.lgamepad[i].hat;
|
|
|
|
for(unsigned axis = 0; axis < min(6U, (unsigned)joypad<>::axes); axis++) {
|
|
table[index + joypad<>::axis + axis] = xinput.lgamepad[i].axis[axis];
|
|
}
|
|
|
|
for(unsigned button = 0; button < min(10U, (unsigned)joypad<>::buttons); button++) {
|
|
table[index + joypad<>::button + button] = xinput.lgamepad[i].button[button];
|
|
}
|
|
}
|
|
|
|
//=======================
|
|
//DirectInput controllers
|
|
//=======================
|
|
dinput.poll();
|
|
for(unsigned i = 0; i < dinput.lgamepad.size(); i++) {
|
|
if(joy >= joypad<>::count) break;
|
|
unsigned index = joypad<>::index(joy++, joypad<>::none);
|
|
|
|
for(unsigned hat = 0; hat < min(4U, (unsigned)joypad<>::hats); hat++) {
|
|
table[index + joypad<>::hat + hat] = dinput.lgamepad[i].hat[hat];
|
|
}
|
|
|
|
for(unsigned axis = 0; axis < min(6U, (unsigned)joypad<>::axes); axis++) {
|
|
table[index + joypad<>::axis + axis] = dinput.lgamepad[i].axis[axis];
|
|
}
|
|
|
|
for(unsigned button = 0; button < min(128U, (unsigned)joypad<>::buttons); button++) {
|
|
table[index + joypad<>::button + button] = dinput.lgamepad[i].button[button];
|
|
}
|
|
}
|
|
}
|
|
|
|
bool init() {
|
|
//only spawn RawInput processing thread one time
|
|
if(rawinput.initialized == false) {
|
|
rawinput.initialized = true;
|
|
rawinput.mutex = CreateMutex(NULL, FALSE, NULL);
|
|
CreateThread(NULL, 0, RawInputThreadProc, 0, 0, NULL);
|
|
|
|
//RawInput device calibration needs to finish before initializing DirectInput;
|
|
//as it needs device GUIDs to distinguish XInput devices from ordinary joypads.
|
|
bool ready = false;
|
|
do {
|
|
Sleep(10);
|
|
WaitForSingleObject(rawinput.mutex, INFINITE);
|
|
ready = rawinput.ready;
|
|
ReleaseMutex(rawinput.mutex);
|
|
} while(ready == false);
|
|
}
|
|
|
|
xinput.init();
|
|
dinput.init(settings.handle);
|
|
|
|
acquire_mouse = false;
|
|
cursor_visible = true;
|
|
return true;
|
|
}
|
|
|
|
void term() {
|
|
unacquire();
|
|
dinput.term();
|
|
}
|
|
|
|
pInputRaw(InputRaw &self_) : self(self_) {
|
|
}
|
|
};
|
|
|
|
bool InputRaw::cap(Setting setting) { return p.cap(setting); }
|
|
uintptr_t InputRaw::get(Setting setting) { return p.get(setting); }
|
|
bool InputRaw::set(Setting setting, uintptr_t param) { return p.set(setting, param); }
|
|
bool InputRaw::acquire() { return p.acquire(); }
|
|
bool InputRaw::unacquire() { return p.unacquire(); }
|
|
bool InputRaw::acquired() { return p.acquired(); }
|
|
bool InputRaw::poll(int16_t *table) { return p.poll(table); }
|
|
bool InputRaw::init() { return p.init(); }
|
|
void InputRaw::term() { p.term(); }
|
|
InputRaw::InputRaw() : p(*new pInputRaw(*this)) {}
|
|
InputRaw::~InputRaw() { delete &p; }
|
|
|
|
}
|