72 lines
2.6 KiB
C++
72 lines
2.6 KiB
C++
#include "stdafx.h"
|
|
#include "XInputManager.h"
|
|
#include <XInput.h>
|
|
|
|
XInputManager::XInputManager(uint8_t** base_controller_input_address_ptr, const float vibrationStrengthFactor)
|
|
{
|
|
this->vibrationStrengthFactor = vibrationStrengthFactor;
|
|
xinputThread = std::thread(&XInputManager::Run, this, base_controller_input_address_ptr);
|
|
}
|
|
|
|
void XInputManager::Run(uint8_t** base_controller_input_address_ptr)
|
|
{
|
|
bool hasConnected = false;
|
|
for (DWORD i = 0; i < XUSER_MAX_COUNT; i++) {
|
|
XINPUT_STATE state;
|
|
ZeroMemory(&state, sizeof(XINPUT_STATE));
|
|
|
|
const DWORD controllerState = XInputGetState(i, &state);
|
|
if (controllerState == ERROR_SUCCESS) {
|
|
controllerId = i;
|
|
hasConnected = true;
|
|
break;
|
|
}
|
|
}
|
|
if (hasConnected) {
|
|
WaitAndSetVibrationAddress(base_controller_input_address_ptr);
|
|
VibrationLoop();
|
|
}
|
|
}
|
|
|
|
void XInputManager::WaitAndSetVibrationAddress(uint8_t** base_controller_input_address_ptr)
|
|
{
|
|
do {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(4));
|
|
if (base_controller_input_address_ptr && *base_controller_input_address_ptr) {
|
|
vibration_address_low_frequency = (float*)(*base_controller_input_address_ptr + 0x40 + 0x60 - 0x4);
|
|
vibration_address_high_frequency = vibration_address_low_frequency + 1;
|
|
}
|
|
} while (vibration_address_low_frequency == NULL);
|
|
}
|
|
|
|
void XInputManager::VibrationLoop()
|
|
{
|
|
const WORD maxVibrationStrength = 65535;
|
|
bool wasVibrating = false;
|
|
SetControllerVibration(0, 0);
|
|
while (true) {
|
|
const float vibrationStrengthLowFrequency = *vibration_address_low_frequency;
|
|
const float vibrationStrengthHighFrequency = *vibration_address_high_frequency;
|
|
if (vibrationStrengthLowFrequency > 0.01f || vibrationStrengthHighFrequency > 0.01f) {
|
|
const WORD leftMotorVibration = std::min((WORD)(vibrationStrengthFactor * vibrationStrengthLowFrequency * maxVibrationStrength), maxVibrationStrength);
|
|
const WORD rightMotorVibration = std::min((WORD)(vibrationStrengthFactor * vibrationStrengthHighFrequency * maxVibrationStrength), maxVibrationStrength);
|
|
SetControllerVibration(leftMotorVibration, rightMotorVibration);
|
|
wasVibrating = true;
|
|
}
|
|
else if (wasVibrating) {
|
|
SetControllerVibration(0, 0);
|
|
wasVibrating = false;
|
|
}
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(4));
|
|
}
|
|
}
|
|
|
|
void XInputManager::SetControllerVibration(const WORD& leftMotorVibration, const WORD& rightMotorVibration)
|
|
{
|
|
XINPUT_VIBRATION vibration;
|
|
ZeroMemory(&vibration, sizeof(XINPUT_VIBRATION));
|
|
vibration.wLeftMotorSpeed = leftMotorVibration;
|
|
vibration.wRightMotorSpeed = rightMotorVibration;
|
|
XInputSetState(controllerId, &vibration);
|
|
}
|