OpenRGB/Controllers/SteelSeriesController/SteelSeriesAerox9Controller.cpp

/*----------------------------------------------------------------------*\
|  SteelSeriesAerox9Controller.cpp                                       |
|                                                                        |
|  OpenRGB controller driver for the Steelseries Aerox9 Wireless (Wired) |
|                                                                        |
|  rom4ster         11th October 2022                                    |
\*---------------------------------------------------------------------*/

#include "SteelSeriesAerox9Controller.h"
#include "LogManager.h"

SteelSeriesAerox9Controller::SteelSeriesAerox9Controller(hid_device* dev_handle, steelseries_type proto_type, const char* path)
    : SteelSeriesMouseController(dev_handle, proto_type, path)
{
    SendInit();
}

SteelSeriesAerox9Controller::~SteelSeriesAerox9Controller()
{
    hid_close(dev);
}

void SteelSeriesAerox9Controller::SendInit()
{
    /*-----------------------------------------------------------------*\
    | This sets sensitivity and allows software mode?? max 5 uint8      |
    |   buffer[2]        = Count eg. 0 thru 5                           |
    |   buffer[4] to [8] = dpi / 50 range = 0x04 - 0xC7 eg. 400 = 0x08  |
    \*-----------------------------------------------------------------*/
    uint8_t buffer[STEELSERIES_AEORX9_PACKET_SIZE] = { 0x00, 0x2D };

    hid_send_feature_report(dev, buffer, STEELSERIES_AEORX9_PACKET_SIZE);
}

std::string SteelSeriesAerox9Controller::GetFirmwareVersion()
{
    uint8_t result      = 0;
    const uint8_t CMD   = 0x90;
    const uint8_t sz    = 16;
    char version[sz];

    uint8_t buffer[STEELSERIES_AEORX9_PACKET_SIZE] = { 0x00, CMD, 0x00 };

    hid_send_feature_report(dev, buffer, STEELSERIES_AEORX9_PACKET_SIZE);
    do
    {
        result = hid_read_timeout(dev, buffer, STEELSERIES_AEORX9_PACKET_SIZE, STEELSERIES_AEORX9_TIMEOUT);
        LOG_DEBUG("[%s] Reading version buffer: Bytes Read %d Buffer %02X %02X %02X %02X %02X %02X %02X %02X %02X %02X %02X", device_name, result,
                  buffer[0], buffer[1], buffer[2], buffer[3], buffer[4], buffer[5], buffer[6], buffer[7], buffer[8], buffer[9], buffer[10]);
    } while(result > 0 && buffer[0] != CMD);

    if(buffer[0] == CMD)
    {
        /*-----------------------------------------------------------------*\
        | Read the version from the second character                        |
        \*-----------------------------------------------------------------*/
        memcpy(version, &buffer[1], sz);
        std::string tmp = std::string(version);
        LOG_DEBUG("[%s] Version: %s as string %s", device_name, version, tmp.c_str());

        return tmp;
    }
    else
    {
        LOG_DEBUG("[%s] Unable to get version: giving up!", device_name);
        return "";
    }
}

steelseries_mouse SteelSeriesAerox9Controller::GetMouse()
{
    return aerox_9;
}

void SteelSeriesAerox9Controller::SetLightEffectAll(uint8_t /*effect*/)
{
    /*-----------------------------------------------------------------*\
    | Not used by this device                                           |
    \*-----------------------------------------------------------------*/
}

void SteelSeriesAerox9Controller::SetColor
    (
    unsigned char       zone_id,
    unsigned char       red,
    unsigned char       green,
    unsigned char       blue,
    unsigned char       brightness
    )
{

    uint8_t buffer[STEELSERIES_AEORX9_PACKET_SIZE] =
    {
        0x00,
        0x21
    };

    buffer[0x02]        = 0x1;
    uint8_t offset      = 0x3;

    if (zone_id == 3)
    {
        offset          = 0x3;
        buffer[0x01]    = 0x26;
        buffer[offset]  = 0x00;
    } else {
        buffer[offset]      = zone_id;
    }



    buffer[offset + 1]  = red;
    buffer[offset + 2]  = green;
    buffer[offset + 3]  = blue;

    hid_write(dev, buffer, STEELSERIES_AEORX9_PACKET_SIZE);

    if(brightness != current_brightness)
    {
        SetBrightness(brightness);
        current_brightness = brightness;
    }
    Save();
}

void SteelSeriesAerox9Controller::SetBrightness(uint8_t brightness)
{
    uint8_t buffer[STEELSERIES_AEORX9_PACKET_SIZE]  = { 0x00, 0x23, brightness };

    hid_write(dev, buffer, STEELSERIES_AEORX9_PACKET_SIZE);
}

void SteelSeriesAerox9Controller::Save()
{
    /*---------------------------------------------------------------------------------*\
    | Save packet was not confirmed as working but packet is verified as correct.       |
    |   https://github.com/flozz/rivalcfg/blob/master/rivalcfg/devices/aerox3.py#L141   |
    \*---------------------------------------------------------------------------------*/
    uint8_t buffer2[STEELSERIES_AEORX9_PACKET_SIZE]  = { 0x00, 0x11, 0x00 };

    hid_write(dev, buffer2, STEELSERIES_AEORX9_PACKET_SIZE);
}