From 1b5cf118e7a4747b3e64aa20f0abaece6cbc9dd5 Mon Sep 17 00:00:00 2001
From: Megamouse <studienricky89@googlemail.com>
Date: Sun, 7 Mar 2021 01:53:27 +0100
Subject: [PATCH] Input/Qt: implement XInput battery level

---
 rpcs3/Input/xinput_pad_handler.cpp | 60 ++++++++++++++++++++++++------
 rpcs3/Input/xinput_pad_handler.h   |  1 +
 2 files changed, 49 insertions(+), 12 deletions(-)

diff --git a/rpcs3/Input/xinput_pad_handler.cpp b/rpcs3/Input/xinput_pad_handler.cpp
index bec0aa5ca9..f661f12165 100644
--- a/rpcs3/Input/xinput_pad_handler.cpp
+++ b/rpcs3/Input/xinput_pad_handler.cpp
@@ -59,6 +59,7 @@ xinput_pad_handler::xinput_pad_handler() : PadHandlerBase(pad_handler::xinput)
 	b_has_config = true;
 	b_has_rumble = true;
 	b_has_deadzones = true;
+	b_has_battery = true;
 
 	m_name_string = "XInput Pad #";
 	m_max_devices = XUSER_MAX_COUNT;
@@ -129,7 +130,7 @@ void xinput_pad_handler::init_config(pad_config* cfg, const std::string& name)
 
 void xinput_pad_handler::SetPadData(const std::string& padId, u32 largeMotor, u32 smallMotor, s32/* r*/, s32/* g*/, s32/* b*/, bool /*battery_led*/, u32 /*battery_led_brightness*/)
 {
-	int device_number = GetDeviceNumber(padId);
+	const int device_number = GetDeviceNumber(padId);
 	if (device_number < 0)
 		return;
 
@@ -143,12 +144,47 @@ void xinput_pad_handler::SetPadData(const std::string& padId, u32 largeMotor, u3
 	(*xinputSetState)(static_cast<u32>(device_number), &vibrate);
 }
 
+u32 xinput_pad_handler::get_battery_level(const std::string& padId)
+{
+	const int device_number = GetDeviceNumber(padId);
+	if (device_number < 0)
+		return 0;
+
+	// Receive Battery Info. If device is not on cable, get battery level, else assume full.
+	XINPUT_BATTERY_INFORMATION battery_info;
+	(*xinputGetBatteryInformation)(device_number, BATTERY_DEVTYPE_GAMEPAD, &battery_info);
+
+	switch (battery_info.BatteryType)
+	{
+	case BATTERY_TYPE_DISCONNECTED:
+		return 0;
+	case BATTERY_TYPE_WIRED:
+		return 100;
+	default:
+		break;
+	}
+
+	switch (battery_info.BatteryLevel)
+	{
+	case BATTERY_LEVEL_EMPTY:
+		return 0;
+	case BATTERY_LEVEL_LOW:
+		return 33;
+	case BATTERY_LEVEL_MEDIUM:
+		return 66;
+	case BATTERY_LEVEL_FULL:
+		return 100;
+	default:
+		return 0;
+	}
+}
+
 int xinput_pad_handler::GetDeviceNumber(const std::string& padId)
 {
 	if (!Init())
 		return -1;
 
-	usz pos = padId.find(m_name_string);
+	const usz pos = padId.find(m_name_string);
 	if (pos == umax)
 		return -1;
 
@@ -184,10 +220,10 @@ xinput_pad_handler::PadButtonValues xinput_pad_handler::get_button_values_base(c
 	values[XInputKeyCodes::RT] = state.Gamepad.bRightTrigger;
 
 	// Sticks
-	int lx = state.Gamepad.sThumbLX;
-	int ly = state.Gamepad.sThumbLY;
-	int rx = state.Gamepad.sThumbRX;
-	int ry = state.Gamepad.sThumbRY;
+	const int lx = state.Gamepad.sThumbLX;
+	const int ly = state.Gamepad.sThumbLY;
+	const int rx = state.Gamepad.sThumbRX;
+	const int ry = state.Gamepad.sThumbRY;
 
 	// Left Stick X Axis
 	values[XInputKeyCodes::LSXNeg] = lx < 0 ? abs(lx) - 1 : 0;
@@ -243,10 +279,10 @@ xinput_pad_handler::PadButtonValues xinput_pad_handler::get_button_values_scp(co
 	values[xinput_pad_handler::XInputKeyCodes::RT] = static_cast<u16>(state.SCP_R2 * 255.0f);
 
 	// Sticks
-	float lx = state.SCP_LX;
-	float ly = state.SCP_LY;
-	float rx = state.SCP_RX;
-	float ry = state.SCP_RY;
+	const float lx = state.SCP_LX;
+	const float ly = state.SCP_LY;
+	const float rx = state.SCP_RX;
+	const float ry = state.SCP_RY;
 
 	// Left Stick X Axis
 	values[xinput_pad_handler::XInputKeyCodes::LSXNeg] = lx < 0.0f ? static_cast<u16>(lx * -32768.0f) : 0;
@@ -376,7 +412,7 @@ std::vector<std::string> xinput_pad_handler::ListDevices()
 std::shared_ptr<PadDevice> xinput_pad_handler::get_device(const std::string& device)
 {
 	// Convert device string to u32 representing xinput device number
-	int device_number = GetDeviceNumber(device);
+	const int device_number = GetDeviceNumber(device);
 	if (device_number < 0)
 		return nullptr;
 
@@ -455,7 +491,7 @@ void xinput_pad_handler::get_extended_info(const std::shared_ptr<PadDevice>& dev
 	if (!dev || !pad)
 		return;
 
-	auto padnum = dev->deviceNumber;
+	const auto padnum = dev->deviceNumber;
 
 	// Receive Battery Info. If device is not on cable, get battery level, else assume full
 	XINPUT_BATTERY_INFORMATION battery_info;
diff --git a/rpcs3/Input/xinput_pad_handler.h b/rpcs3/Input/xinput_pad_handler.h
index 8ca0c82680..d5ee07076c 100644
--- a/rpcs3/Input/xinput_pad_handler.h
+++ b/rpcs3/Input/xinput_pad_handler.h
@@ -112,6 +112,7 @@ public:
 
 	std::vector<std::string> ListDevices() override;
 	void SetPadData(const std::string& padId, u32 largeMotor, u32 smallMotor, s32 r, s32 g, s32 b, bool battery_led, u32 battery_led_brightness) override;
+	u32 get_battery_level(const std::string& padId) override;
 	void init_config(pad_config* cfg, const std::string& name) override;
 
 private: