/*-------------------------------------------------------------------*\
|  CorsairHydroPlatinumController.cpp                                 |
|                                                                     |
|  Driver for Corsair Hydro Platinum AIO Coolers                      |
|                                                                     |
|  Kasper         28th March 2021                                     |
|                                                                     |
\*-------------------------------------------------------------------*/

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

static const uint8_t CRC_TABLE[256] =
{
    0x00, 0x07, 0x0E, 0x09, 0x1C, 0x1B, 0x12, 0x15,
    0x38, 0x3F, 0x36, 0x31, 0x24, 0x23, 0x2A, 0x2D,
    0x70, 0x77, 0x7E, 0x79, 0x6C, 0x6B, 0x62, 0x65,
    0x48, 0x4F, 0x46, 0x41, 0x54, 0x53, 0x5A, 0x5D,
    0xE0, 0xE7, 0xEE, 0xE9, 0xFC, 0xFB, 0xF2, 0xF5,
    0xD8, 0xDF, 0xD6, 0xD1, 0xC4, 0xC3, 0xCA, 0xCD,
    0x90, 0x97, 0x9E, 0x99, 0x8C, 0x8B, 0x82, 0x85,
    0xA8, 0xAF, 0xA6, 0xA1, 0xB4, 0xB3, 0xBA, 0xBD,
    0xC7, 0xC0, 0xC9, 0xCE, 0xDB, 0xDC, 0xD5, 0xD2,
    0xFF, 0xF8, 0xF1, 0xF6, 0xE3, 0xE4, 0xED, 0xEA,
    0xB7, 0xB0, 0xB9, 0xBE, 0xAB, 0xAC, 0xA5, 0xA2,
    0x8F, 0x88, 0x81, 0x86, 0x93, 0x94, 0x9D, 0x9A,
    0x27, 0x20, 0x29, 0x2E, 0x3B, 0x3C, 0x35, 0x32,
    0x1F, 0x18, 0x11, 0x16, 0x03, 0x04, 0x0D, 0x0A,
    0x57, 0x50, 0x59, 0x5E, 0x4B, 0x4C, 0x45, 0x42,
    0x6F, 0x68, 0x61, 0x66, 0x73, 0x74, 0x7D, 0x7A,
    0x89, 0x8E, 0x87, 0x80, 0x95, 0x92, 0x9B, 0x9C,
    0xB1, 0xB6, 0xBF, 0xB8, 0xAD, 0xAA, 0xA3, 0xA4,
    0xF9, 0xFE, 0xF7, 0xF0, 0xE5, 0xE2, 0xEB, 0xEC,
    0xC1, 0xC6, 0xCF, 0xC8, 0xDD, 0xDA, 0xD3, 0xD4,
    0x69, 0x6E, 0x67, 0x60, 0x75, 0x72, 0x7B, 0x7C,
    0x51, 0x56, 0x5F, 0x58, 0x4D, 0x4A, 0x43, 0x44,
    0x19, 0x1E, 0x17, 0x10, 0x05, 0x02, 0x0B, 0x0C,
    0x21, 0x26, 0x2F, 0x28, 0x3D, 0x3A, 0x33, 0x34,
    0x4E, 0x49, 0x40, 0x47, 0x52, 0x55, 0x5C, 0x5B,
    0x76, 0x71, 0x78, 0x7F, 0x6A, 0x6D, 0x64, 0x63,
    0x3E, 0x39, 0x30, 0x37, 0x22, 0x25, 0x2C, 0x2B,
    0x06, 0x01, 0x08, 0x0F, 0x1A, 0x1D, 0x14, 0x13,
    0xAE, 0xA9, 0xA0, 0xA7, 0xB2, 0xB5, 0xBC, 0xBB,
    0x96, 0x91, 0x98, 0x9F, 0x8A, 0x8D, 0x84, 0x83,
    0xDE, 0xD9, 0xD0, 0xD7, 0xC2, 0xC5, 0xCC, 0xCB,
    0xE6, 0xE1, 0xE8, 0xEF, 0xFA, 0xFD, 0xF4, 0xF3
};

static const uint8_t MAGIC_1[61] =
{
    0x01, 0x01, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x7f,
    0x7f, 0x7f, 0x7f, 0xff, 0x00, 0xff, 0xff, 0xff,
    0xff, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0xff,
    0xff, 0xff, 0xff, 0x00, 0xff, 0xff, 0xff, 0xff,
    0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0xff, 0xff,
    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
    0xff, 0xff, 0xff, 0xff, 0xff
};

static const uint8_t MAGIC_2[61] =
{
    0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
    0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,
    0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
    0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f,
    0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27,
    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
    0xff, 0xff, 0xff, 0xff, 0xff
};

static const uint8_t MAGIC_3[61] =
{
    0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f,
    0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37,
    0x38, 0x39, 0x3a, 0x3b, 0x3c, 0x3d, 0x3e, 0x3f,
    0x40, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47,
    0x48, 0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f,
    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
    0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
    0xff, 0xff, 0xff, 0xff, 0xff
};

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

    SendMagic(MAGIC_1, CORSAIR_HYDRO_PLATINUM_MAGIC_1);
    SendMagic(MAGIC_2, CORSAIR_HYDRO_PLATINUM_MAGIC_2);
    SendMagic(MAGIC_3, CORSAIR_HYDRO_PLATINUM_MAGIC_3);
}

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

std::string CorsairHydroPlatinumController::GetLocation()
{
    return location;
}

std::string CorsairHydroPlatinumController::GetFirmwareString()
{
    return firmware_version;
}

void CorsairHydroPlatinumController::SetupColors(std::vector<RGBColor> colors)
{
    unsigned int end_led = colors.size() >= 20 ? 20 : colors.size();
    SendColors(colors, 0, end_led, CORSAIR_HYDRO_PLATINUM_SET_LIGHTING_1);

    if(colors.size() > 20)
    {
        end_led = colors.size() >= 40 ? 40 : colors.size();
        SendColors(colors, 20, end_led, CORSAIR_HYDRO_PLATINUM_SET_LIGHTING_2);
    }
    if(colors.size() > 40)
    {
        end_led = colors.size() >= 48 ? 48 : colors.size();
        SendColors(colors, 40, end_led, CORSAIR_HYDRO_PLATINUM_SET_LIGHTING_3);
    }
}

void CorsairHydroPlatinumController::SendMagic(const uint8_t* magic, unsigned int command)
{
    unsigned char usb_buf[CORSAIR_HYDRO_PLATINUM_PACKET_SIZE];

    /*-----------------------------------------------------*\
    | Set up config table request packet                    |
    \*-----------------------------------------------------*/
    usb_buf[0x00] = 0x00;
    usb_buf[0x01] = 0x3F;
    usb_buf[0x02] = (GetSequenceNumber()) | command;

    /*-----------------------------------------------------*\
    | Copy the magic bytes into the buffer                  |
    \*-----------------------------------------------------*/
    memcpy(&usb_buf[3], magic, 61 * sizeof magic[0]);

    /*-----------------------------------------------------*\
    | The data sent to the PEC function should not contain  |
    | the first (report id), second (prefix) and            |
    | last (checksum) bytes                                 |
    \*-----------------------------------------------------*/
    std::vector<unsigned char> checksum_array;
    checksum_array.insert(checksum_array.begin(), std::begin(usb_buf) + 2, std::end(usb_buf) - 1);
    usb_buf[64] = ComputePEC(static_cast<void*>(checksum_array.data()), 62);

    /*-----------------------------------------------------*\
    | Send packet                                           |
    \*-----------------------------------------------------*/
    hid_write(dev, usb_buf, CORSAIR_HYDRO_PLATINUM_PACKET_SIZE);
    hid_read(dev, usb_buf, CORSAIR_HYDRO_PLATINUM_PACKET_SIZE);

    if(firmware_version.empty())
    {
        firmware_version =
            std::to_string(usb_buf[2] >> 4) + "."
            + std::to_string(usb_buf[2] & 0xf) + "."
            + std::to_string(usb_buf[3]);
    }

    /*-----------------------------------------------------*\
    | This delay prevents the AIO from soft-locking when    |
    | using an EE.                                          |
    \*-----------------------------------------------------*/
    std::this_thread::sleep_for(std::chrono::milliseconds(CORSAIR_HYDRO_PLATINUM_PACKET_DELAY));
}

void CorsairHydroPlatinumController::SendColors(std::vector<RGBColor> colors, unsigned int start, unsigned int end, unsigned int command)
{
    unsigned char usb_buf[CORSAIR_HYDRO_PLATINUM_PACKET_SIZE];

    /*-----------------------------------------------------*\
    | Zero out buffer                                       |
    \*-----------------------------------------------------*/
    memset(usb_buf, 0x00, CORSAIR_HYDRO_PLATINUM_PACKET_SIZE);

    /*-----------------------------------------------------*\
    | Set up config table request packet                    |
    \*-----------------------------------------------------*/
    usb_buf[0x00] = 0x00;
    usb_buf[0x01] = 0x3F;
    usb_buf[0x02] = (GetSequenceNumber()) | command;

    unsigned int i = 0;
    for(std::size_t color_idx = start; color_idx < end; color_idx++)
    {
        usb_buf[(i * 3) + 3] = RGBGetBValue(colors[color_idx]);
        usb_buf[(i * 3) + 4] = RGBGetGValue(colors[color_idx]);
        usb_buf[(i * 3) + 5] = RGBGetRValue(colors[color_idx]);
        i++;
    }

    /*-----------------------------------------------------*\
    | The data sent to the PEC function should not contain  |
    | the first (report id), second (prefix) and            |
    | last (checksum) bytes                                 |
    \*-----------------------------------------------------*/
    std::vector<unsigned char> checksum_array;
    checksum_array.insert(checksum_array.begin(), std::begin(usb_buf) + 2, std::end(usb_buf) - 1);
    usb_buf[64] = ComputePEC(static_cast<void*>(checksum_array.data()), 62);

    hid_write(dev, usb_buf, CORSAIR_HYDRO_PLATINUM_PACKET_SIZE);
    hid_read(dev, usb_buf, CORSAIR_HYDRO_PLATINUM_PACKET_SIZE);

    /*-----------------------------------------------------*\
    | This delay prevents the AIO from soft-locking when    |
    | using an EE.                                          |
    \*-----------------------------------------------------*/
    std::this_thread::sleep_for(std::chrono::milliseconds(CORSAIR_HYDRO_PLATINUM_PACKET_DELAY));
}

unsigned int CorsairHydroPlatinumController::GetSequenceNumber()
{
    if(sequence_number < 31)
    {
        sequence_number++;
    }
    else
    {
        sequence_number = 1;
    }

    return(sequence_number << 3);
}

uint8_t CorsairHydroPlatinumController::ComputePEC(const void * data, size_t size)
{
	uint8_t val = 0;

	uint8_t * pos = (uint8_t *) data;
	uint8_t * end = pos + size;

	while (pos < end)
    {
		val = CRC_TABLE[val ^ *pos];
		pos++;
	}

	return val;
}