PRO X Wireless Headset

src/devices/logitech_gpro.c

@@ -2,6 +2,9 @@
 #include "../utility.h"
 #include "logitech.h"
 
+#include <unistd.h>
+
+#include <math.h>
 #include <string.h>
 
 static struct device device_gpro;
@@ -17,6 +20,8 @@ static const uint16_t PRODUCT_IDS[] = {
 };
 
 static int gpro_send_sidetone(hid_device* device_handle, uint8_t num);
+static int gpro_request_battery(hid_device* device_handle);
+static int gpro_send_inactive_time(hid_device* device_handle, uint8_t num);
 
 void gpro_init(struct device** device)
 {
@@ -26,8 +31,14 @@ void gpro_init(struct device** device)
 
     strncpy(device_gpro.device_name, "Logitech G PRO Series", sizeof(device_gpro.device_name));
 
-    device_gpro.capabilities  = B(CAP_SIDETONE);
-    device_gpro.send_sidetone = &gpro_send_sidetone;
+    device_gpro.capabilities = B(CAP_SIDETONE | B(CAP_BATTERY_STATUS) | B(CAP_INACTIVE_TIME));
+
+    device_gpro.capability_details[CAP_BATTERY_STATUS] = (struct capability_detail) { .usagepage = 0xff43, .usageid = 0x0202 };
+    device_gpro.capability_details[CAP_INACTIVE_TIME]  = (struct capability_detail) { .usagepage = 0xff43, .usageid = 0x0202 };
+
+    device_gpro.send_sidetone      = &gpro_send_sidetone;
+    device_gpro.request_battery    = &gpro_request_battery;
+    device_gpro.send_inactive_time = &gpro_send_inactive_time;
 
     *device = &device_gpro;
 }
@@ -40,3 +51,98 @@ static int gpro_send_sidetone(hid_device* device_handle, uint8_t num)
 
     return hid_write(device_handle, sidetone_data, sizeof(sidetone_data) / sizeof(sidetone_data[0]));
 }
+
+/**
+ * @brief This function calculates the estimate batttery level in percent.
+ *
+ * Battery stats:
+ *   - Voltage capacity:  4200 (100%)
+ *   - Charging voltage:  4700
+ *   - Power off voltage: 3300 (  0%)
+ *
+ * @param voltage readings
+ * @return battery level in percent
+ */
+static float estimate_battery_level(uint16_t voltage)
+{
+    double terms[] = {
+        -1.7790085824253613e+006,
+        2.3692153307344888e+003,
+        -1.2580905041506840e+000,
+        3.3295201187388395e-004,
+        -4.3913981733597497e-008,
+        2.3093121045539907e-012
+    };
+
+    size_t csz = sizeof terms / sizeof *terms;
+
+    double t       = 1;
+    double percent = 0;
+    for (int i = 0; i < csz; i++) {
+        percent += terms[i] * t;
+        t *= voltage;
+    }
+
+    if (percent > 100)
+        percent = 100;
+    if (percent < 0)
+        percent = 0;
+    return percent;
+}
+
+static int gpro_request_battery(hid_device* device_handle)
+{
+    /*
+        CREDIT GOES TO https://github.com/ashkitten/ for the project
+        https://github.com/ashkitten/g933-utils/
+        I've simply ported that implementation to this project!
+    */
+
+    int r = 0;
+    // request battery voltage
+    uint8_t buf[HIDPP_LONG_MESSAGE_LENGTH] = { HIDPP_LONG_MESSAGE, HIDPP_DEVICE_RECEIVER, 0x06, 0x0d, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
+
+    r = hid_write(device_handle, buf, sizeof(buf) / sizeof(buf[0]));
+    if (r < 0)
+        return r;
+
+    r = hid_read_timeout(device_handle, buf, HIDPP_LONG_MESSAGE_LENGTH, hsc_device_timeout);
+    if (r < 0)
+        return r;
+
+    if (r == 0)
+        return HSC_READ_TIMEOUT;
+
+    if (buf[6] == 0x03)
+        return BATTERY_CHARGING;
+
+    // Headset offline
+    if (buf[2] == 0xff)
+        return HSC_ERROR;
+
+    const uint16_t voltage = (buf[4] << 8) | buf[5];
+    return (int)(roundf(estimate_battery_level(voltage)));
+}
+
+static int gpro_send_inactive_time(hid_device* device_handle, uint8_t num)
+{
+    int r = 0;
+    // request new inactivity timeout
+    uint8_t buf[HIDPP_LONG_MESSAGE_LENGTH] = { HIDPP_LONG_MESSAGE, HIDPP_DEVICE_RECEIVER, 0x06, 0x2d, num, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
+
+    r = hid_write(device_handle, buf, sizeof(buf) / sizeof(buf[0]));
+    if (r < 0)
+        return r;
+
+    r = hid_read_timeout(device_handle, buf, HIDPP_LONG_MESSAGE_LENGTH, hsc_device_timeout);
+    if (r < 0)
+        return r;
+
+    if (r == 0)
+        return HSC_READ_TIMEOUT;
+
+    if (buf[2] == 0xff)
+        return HSC_ERROR;
+
+    return r;
+}