/*---------------------------------------------------------*\
|  Processing Code for Corsair Lighting Node Pro            |
|                                                           |
|  Adam Honse (calcprogrammer1@gmail.com), 1/12/2020        |
\*---------------------------------------------------------*/

#include "CorsairNodeProController.h"

#include <fstream>
#include <iostream>
#include <string>

CorsairNodeProController::CorsairNodeProController(libusb_device_handle* dev_handle)
{
    dev = dev_handle;
    channel_leds[0] = 60;
    channel_leds[1] = 60;
}

CorsairNodeProController::~CorsairNodeProController()
{
}

void CorsairNodeProController::SetChannelLEDs(unsigned int channel, std::vector<RGBColor> colors)
{
    unsigned char usb_start[] =
    {
        0x38, 0x00, 0x02, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00
    };

    unsigned char usb_data[] =
    {
        0x32, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00
    };

    unsigned char usb_apply[] =
    {
        0x33, 0xFF, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x80, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00
    };
    int             actual;

    /*-----------------------------------------------------*\
    | Set channel in USB packets                            |
    \*-----------------------------------------------------*/
    usb_data[0x01]   = channel;
    usb_start[0x01]  = channel;

    /*-----------------------------------------------------*\
    | Send start packet                                     |
    \*-----------------------------------------------------*/
    libusb_interrupt_transfer(dev, 0x01, usb_start, 64, &actual, 0);

    /*-----------------------------------------------------*\
    | Send red channel packet 1                             |
    \*-----------------------------------------------------*/
    std::size_t pkt_max = 50;

    if(pkt_max > colors.size())
    {
        pkt_max = colors.size();
    }

    usb_data[0x02]  = 0;
    usb_data[0x03]  = pkt_max;
    usb_data[0x04]  = 0x00; //red

    for (std::size_t idx = 0; idx < pkt_max; idx++)
    {
        RGBColor color = colors[idx];
        usb_data[idx + 5] = RGBGetRValue(color);
    }

    libusb_interrupt_transfer(dev, 0x01, usb_data, 64, &actual, 0);

    /*-----------------------------------------------------*\
    | Send red channel packet 2 if necessary                |
    \*-----------------------------------------------------*/
    for(int idx = 4; idx < 64; idx++)
    {
        usb_data[idx] = 0;
    }

    pkt_max         = 0;

    if (colors.size() > 50)
    {
        pkt_max = colors.size() - 50;
    }

    if(pkt_max > 0)
    {
        usb_data[0x02]  = 50;
        usb_data[0x03]  = pkt_max;
        usb_data[0x04]  = 0x00; //red

        for (std::size_t idx = 0; idx < pkt_max; idx++)
        {
            RGBColor color = colors[idx + 50];
            usb_data[idx + 5] = RGBGetRValue(color);
        }

        libusb_interrupt_transfer(dev, 0x01, usb_data, 64, &actual, 0);
    }
    
    /*-----------------------------------------------------*\
    | Send green channel packet 1                           |
    \*-----------------------------------------------------*/
    pkt_max = 50;

    if(pkt_max > colors.size())
    {
        pkt_max = colors.size();
    }

    usb_data[0x02]  = 0;
    usb_data[0x03]  = pkt_max;
    usb_data[0x04]  = 0x01; //green

    for (std::size_t idx = 0; idx < pkt_max; idx++)
    {
        RGBColor color = colors[idx];
        usb_data[idx + 5] = RGBGetGValue(color);
    }

    libusb_interrupt_transfer(dev, 0x01, usb_data, 64, &actual, 0);

    /*-----------------------------------------------------*\
    | Send green channel packet 2 if necessary              |
    \*-----------------------------------------------------*/
    for(int idx = 4; idx < 64; idx++)
    {
        usb_data[idx] = 0;
    }

    pkt_max         = 0;

    if (colors.size() > 50)
    {
        pkt_max = colors.size() - 50;
    }

    if(pkt_max > 0)
    {
        usb_data[0x02]  = 50;
        usb_data[0x03]  = pkt_max;
        usb_data[0x04]  = 0x01; //green

        for (std::size_t idx = 0; idx < pkt_max; idx++)
        {
            RGBColor color = colors[idx + 50];
            usb_data[idx + 5] = RGBGetGValue(color);
        }

        libusb_interrupt_transfer(dev, 0x01, usb_data, 64, &actual, 0);
    }

    /*-----------------------------------------------------*\
    | Send blue channel packet 1                            |
    \*-----------------------------------------------------*/
    pkt_max = 50;

    if(pkt_max > colors.size())
    {
        pkt_max = colors.size();
    }

    usb_data[0x02]  = 0;
    usb_data[0x03]  = pkt_max;
    usb_data[0x04]  = 0x02; //blue

    for (std::size_t idx = 0; idx < pkt_max; idx++)
    {
        RGBColor color = colors[idx];
        usb_data[idx + 5] = RGBGetBValue(color);
    }

    libusb_interrupt_transfer(dev, 0x01, usb_data, 64, &actual, 0);

    /*-----------------------------------------------------*\
    | Send blue channel packet 2 if necessary               |
    \*-----------------------------------------------------*/
    for(int idx = 4; idx < 64; idx++)
    {
        usb_data[idx] = 0;
    }

    pkt_max         = 0;

    if (colors.size() > 50)
    {
        pkt_max = colors.size() - 50;
    }

    if(pkt_max > 0)
    {
        usb_data[0x02]  = 50;
        usb_data[0x03]  = pkt_max;
        usb_data[0x04]  = 0x02; //blue

        for (std::size_t idx = 0; idx < pkt_max; idx++)
        {
            RGBColor color = colors[idx + 50];
            usb_data[idx + 5] = RGBGetBValue(color);
        }

        libusb_interrupt_transfer(dev, 0x01, usb_data, 64, &actual, 0);
    }

    /*-----------------------------------------------------*\
    | Send apply packet                                     |
    \*-----------------------------------------------------*/
    libusb_interrupt_transfer(dev, 0x01, usb_apply, 64, &actual, 0);
}