/*-----------------------------------------*\
|  SteelSeriesQCKControllerMat.cpp          |
|                                           |
|  Generic RGB Interface SteelSeriesQCK     |
|  Class                                    |
|                                           |
|  Edbgon 22/05/2021                        |
\*-----------------------------------------*/

#include "SteelSeriesQCKMatController.h"
#include <cstring>

SteelSeriesQCKMatController::SteelSeriesQCKMatController(hid_device* dev_handle, const char* path)
{
    dev         = dev_handle;
    location    = path;
}

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

std::string SteelSeriesQCKMatController::GetDeviceLocation()
{
    return("HID: " + location);
}

char* SteelSeriesQCKMatController::GetDeviceName()
{
    return device_name;
}

std::string SteelSeriesQCKMatController::GetSerialString()
{
    wchar_t serial_string[128];
    int ret = hid_get_serial_number_string(dev, serial_string, 128);
    
    if(ret != 0)
    {
        return("");
    }

    std::wstring return_wstring = serial_string;
    std::string return_string(return_wstring.begin(), return_wstring.end());

    return(return_string);
}

void SteelSeriesQCKMatController::SetColors(std::vector<RGBColor> colors)
{
    unsigned char buf[525];
    unsigned char cbuf[65];

    /*-----------------------------------------------------*\
    | Zero out buffer                                       |
    \*-----------------------------------------------------*/
    memset(buf, 0x00, sizeof(buf));
    memset(cbuf, 0x00, sizeof(cbuf));

    /*-----------------------------------------------------*\
    | Set up Direct packet                                  |
    \*-----------------------------------------------------*/
    buf[0x00]   = 0x00;
    buf[0x01]   = 0x0E;
    buf[0x03]   = 0x02;

    buf[0x08]   = 0xFF;
    buf[0x09]   = 0x32;
    buf[0x0A]   = 0xC8;
    buf[0x0E]   = 0x01;
    
    buf[0x14]   = 0xFF;
    buf[0x15]   = 0x32;
    buf[0x16]   = 0xC8;
    buf[0x19]   = 0x01;

    buf[0x1A]   = 0x01;
    buf[0x1C]   = 0x01;

    /*-----------------------------------------------------*\
    | Fill in color data                                    |
    \*-----------------------------------------------------*/
    buf[0x05] = RGBGetRValue(colors[0]);
    buf[0x06] = RGBGetGValue(colors[0]);
    buf[0x07] = RGBGetBValue(colors[0]);
    
    buf[0x11] = RGBGetRValue(colors[1]);
    buf[0x12] = RGBGetGValue(colors[1]);
    buf[0x13] = RGBGetBValue(colors[1]);

    /*-----------------------------------------------------*\
    | Send packet                                           |
    \*-----------------------------------------------------*/
    hid_send_feature_report(dev, buf, 525);

    cbuf[0x01] = 0x0D;
    hid_write(dev, cbuf, 65);
}