OpenRGB/RGBController/RGBController_CorsairPro.cpp

/*-----------------------------------------*\
|  RGBController_CorsairPro.cpp             |
|                                           |
|  Generic RGB Interface for OpenAuraSDK    |
|  Corsair Vengeance Pro RGB driver         |
|                                           |
|  Adam Honse (CalcProgrammer1) 6/30/2019   |
\*-----------------------------------------*/

#include "RGBController_CorsairPro.h"

int RGBController_CorsairPro::GetMode()
{
    switch (corsair->GetEffect())
    {
    case CORSAIR_PRO_MODE_COLOR_SHIFT:
        return(0);
        break;
    case CORSAIR_PRO_MODE_COLOR_PULSE:
        return(1);
        break;
    case CORSAIR_PRO_MODE_RAINBOW_WAVE:
        return(2);
        break;
    case CORSAIR_PRO_MODE_COLOR_WAVE:
        return(3);
        break;
    case CORSAIR_PRO_MODE_VISOR:
        return(4);
        break;
    case CORSAIR_PRO_MODE_RAIN:
        return(5);
        break;
    case CORSAIR_PRO_MODE_MARQUEE:
        return(6);
        break;
    case CORSAIR_PRO_MODE_RAINBOW:
        return(7);
        break;
    case CORSAIR_PRO_MODE_SEQUENTIAL:
        return(8);
        break;
    case CORSAIR_PRO_MODE_STATIC:
        return(9);
        break;
    }

    return(0);
}

void RGBController_CorsairPro::SetMode(int mode)
{
    switch (mode)
    {
    case 0:
        corsair->SetEffect(CORSAIR_PRO_MODE_COLOR_SHIFT);
        break;
    case 1:
        corsair->SetEffect(CORSAIR_PRO_MODE_COLOR_PULSE);
        break;
    case 2:
        corsair->SetEffect(CORSAIR_PRO_MODE_RAINBOW_WAVE);
        break;
    case 3:
        corsair->SetEffect(CORSAIR_PRO_MODE_COLOR_WAVE);
        break;
    case 4:
        corsair->SetEffect(CORSAIR_PRO_MODE_VISOR);
        break;
    case 5:
        corsair->SetEffect(CORSAIR_PRO_MODE_RAIN);
        break;
    case 6:
        corsair->SetEffect(CORSAIR_PRO_MODE_MARQUEE);
        break;
    case 7:
        corsair->SetEffect(CORSAIR_PRO_MODE_RAINBOW);
        break;
    case 8:
        corsair->SetEffect(CORSAIR_PRO_MODE_SEQUENTIAL);
        break;
    case 9:
        corsair->SetEffect(CORSAIR_PRO_MODE_STATIC);
        break;
    }
}

void RGBController_CorsairPro::SetCustomMode()
{
    corsair->SetCustom();
}

void RGBController_CorsairPro::UpdateLEDs()
{
    for (std::size_t led = 0; led < colors.size(); led++)
    {
        RGBColor      color = colors[led];
        unsigned char red   = RGBGetRValue(color);
        unsigned char grn   = RGBGetGValue(color);
        unsigned char blu   = RGBGetBValue(color);
        corsair->SetLEDColor(led, red, grn, blu);
    }

    corsair->ApplyColors();
}

void RGBController_CorsairPro::UpdateZoneLEDs(int zone)
{
    UpdateLEDs();
}

void RGBController_CorsairPro::UpdateSingleLED(int led)
{
    RGBColor      color = colors[led];
    unsigned char red   = RGBGetRValue(color);
    unsigned char grn   = RGBGetGValue(color);
    unsigned char blu   = RGBGetBValue(color);

    corsair->SetLEDColor(led, red, grn, blu);
    corsair->ApplyColors();
}

RGBController_CorsairPro::RGBController_CorsairPro(CorsairProController* corsair_ptr)
{
    corsair = corsair_ptr;

    name = corsair->GetDeviceName();
    location = corsair->GetDeviceLocation();

    type = DEVICE_TYPE_DRAM;

    mode corsair_modes[CORSAIR_PRO_NUMBER_MODES];

    corsair_modes[0].name = "Color Shift";
    corsair_modes[1].name = "Color Pulse";
    corsair_modes[2].name = "Rainbow Wave";
    corsair_modes[3].name = "Color Wave";
    corsair_modes[4].name = "Visor";
    corsair_modes[5].name = "Rain";
    corsair_modes[6].name = "Marquee";
    corsair_modes[7].name = "Rainbow";
    corsair_modes[8].name = "Sequential";
    corsair_modes[9].name = "Static";

    for (int i = 0; i < CORSAIR_PRO_NUMBER_MODES; i++)
    {
        modes.push_back(corsair_modes[i]);
    }

    colors.resize(corsair->GetLEDCount());

    for (unsigned int i = 0; i < corsair->GetLEDCount(); i++)
    {
        led* new_led = new led();

        new_led->name = "Corsair Pro LED";

        leds.push_back(*new_led);
    }

    zone new_zone;

    new_zone.name = "Corsair Pro Zone";
    new_zone.type = ZONE_TYPE_LINEAR;

    std::vector<int> zone_row;

    for (unsigned int i = 0; i < corsair->GetLEDCount(); i++)
    {
        zone_row.push_back(i);
    }

    new_zone.map.push_back(zone_row);

    zones.push_back(new_zone);
}