OpenRGB/RGBController/RGBController_RazerChromaSDK.cpp

/*-----------------------------------------*\
|  RGBController_RazerChromaSDK.h           |
|                                           |
|  Generic RGB Interface for official Razer |
|  Chroma SDK                               |
|                                           |
|  Adam Honse (CalcProgrammer1) 9/2/2019    |
\*-----------------------------------------*/

#include "RGBController_RazerChromaSDK.h"

CREATEEFFECT           RGBController_RazerChromaSDK::CreateEffect           = NULL;
CREATEKEYBOARDEFFECT   RGBController_RazerChromaSDK::CreateKeyboardEffect   = NULL;
CREATEMOUSEEFFECT      RGBController_RazerChromaSDK::CreateMouseEffect      = NULL;
CREATEMOUSEPADEFFECT   RGBController_RazerChromaSDK::CreateMousepadEffect   = NULL;
CREATEHEADSETEFFECT    RGBController_RazerChromaSDK::CreateHeadsetEffect    = NULL;
CREATECHROMALINKEFFECT RGBController_RazerChromaSDK::CreateChromaLinkEffect = NULL;

RGBController_RazerChromaSDK::RGBController_RazerChromaSDK(unsigned int device_type, HMODULE* hModule)
{
    unsigned int led_count = 0;

    for (int i = 0; i < RAZER_NUM_DEVICES; i++)
    {
        if (device_list[i]->razer_type == device_type)
        {
            name = device_list[i]->name;
            type = device_list[i]->type;
            location = "Razer Chroma SDK";

            for (int zone_id = 0; zone_id < RAZER_MAX_ZONES; zone_id++)
            {
                if (device_list[i]->zones[zone_id] != NULL)
                {
                    device = device_list[i];

                    zone* new_zone = new zone();
                    new_zone->name = device_list[i]->zones[zone_id]->name;
                    new_zone->type = device_list[i]->zones[zone_id]->type;

                    std::vector<int> new_zone_map;

                    for (int led_id = 0; led_id < device_list[i]->zones[zone_id]->leds; led_id++)
                    {
                        RGBColor new_color = 0x00000000;
                        colors.push_back(new_color);

                        led* new_led = new led();
                        new_led->name = device_list[i]->zones[zone_id]->name;
                        leds.push_back(*new_led);

                        new_zone_map.push_back(led_count);

                        led_count++;
                    }

                    new_zone->map.push_back(new_zone_map);

                    zones.push_back(*new_zone);
                }
            }
        }
    }

    if(CreateEffect           == NULL) CreateEffect           = (CREATEEFFECT)GetProcAddress(*hModule, "CreateEffect");
    if(CreateKeyboardEffect   == NULL) CreateKeyboardEffect   = (CREATEKEYBOARDEFFECT)GetProcAddress(*hModule, "CreateKeyboardEffect");
    if(CreateMouseEffect      == NULL) CreateMouseEffect      = (CREATEMOUSEEFFECT)GetProcAddress(*hModule, "CreateMouseEffect");
    if(CreateMousepadEffect   == NULL) CreateMousepadEffect   = (CREATEMOUSEPADEFFECT)GetProcAddress(*hModule, "CreateMousepadEffect");
    if(CreateHeadsetEffect    == NULL) CreateHeadsetEffect    = (CREATEHEADSETEFFECT)GetProcAddress(*hModule, "CreateHeadsetEffect");
    if(CreateChromaLinkEffect == NULL) CreateChromaLinkEffect = (CREATECHROMALINKEFFECT)GetProcAddress(*hModule, "CreateChromaLinkEffect");
}

int RGBController_RazerChromaSDK::GetMode()
{
    return 0;
}

void RGBController_RazerChromaSDK::SetMode(int mode)
{

}

void RGBController_RazerChromaSDK::SetCustomMode()
{

}

void RGBController_RazerChromaSDK::SetAllLEDs(RGBColor color)
{
    for (int i = 0; i < colors.size(); i++)
    {
        colors[i] = color;
    }

    UpdateLEDs();
}

void RGBController_RazerChromaSDK::SetAllZoneLEDs(int zone, RGBColor color)
{
    for (int x = 0; x < zones[zone].map.size(); x++)
    {
        for (int y = 0; y < zones[zone].map[x].size(); y++)
        {
            colors[zones[zone].map[x][y]] = color;
        }
    }

    UpdateLEDs();
}

void RGBController_RazerChromaSDK::SetLED(int led, RGBColor color)
{
    colors[led] = color;

    UpdateLEDs();
}

void RGBController_RazerChromaSDK::UpdateLEDs()
{
    switch (device->razer_type)
    {
    case RAZER_GENERIC_MOUSE:
        {
            ChromaSDK::Mouse::CUSTOM_EFFECT_TYPE2 MouseEffect = {};

            std::vector<int> left_zone_map = zones[0].map[0];

            for (int left = 1; left < 8; left++)
            {
                MouseEffect.Color[left][0] = colors[left_zone_map[left - 1]];
            }

            std::vector<int> right_zone_map = zones[1].map[0];

            for (int right = 1; right < 8; right++)
            {
                MouseEffect.Color[right][6] = colors[right_zone_map[right - 1]];
            }

            std::vector<int> bottom_zone_map = zones[2].map[0];

            for (int bottom = 1; bottom < 6; bottom++)
            {
                MouseEffect.Color[8][bottom] = colors[bottom_zone_map[bottom - 1]];
            }

            std::vector<int> scroll_wheel_zone_map = zones[3].map[0];

            MouseEffect.Color[2][3] = colors[scroll_wheel_zone_map[0]];

            std::vector<int> logo_zone_map = zones[4].map[0];

            MouseEffect.Color[7][3] = colors[logo_zone_map[0]];

            std::vector<int> numpad_backlight_zone_map = zones[5].map[0];

            MouseEffect.Color[4][3] = colors[numpad_backlight_zone_map[0]];

            CreateMouseEffect(ChromaSDK::Mouse::CHROMA_CUSTOM2, &MouseEffect, NULL);
        }
        break;

    case RAZER_GENERIC_MOUSEPAD:
        {
            ChromaSDK::Mousepad::CUSTOM_EFFECT_TYPE MousepadEffect = {};

            for (int x = 0; x < 15; x++)
            {
                MousepadEffect.Color[x] = colors[x];
            }

            CreateMousepadEffect(ChromaSDK::Mousepad::CHROMA_CUSTOM, &MousepadEffect, NULL);
        }
        break;

    case RAZER_CHROMA_HDK:
        {
            ChromaSDK::CUSTOM_EFFECT_TYPE ChromaBoxEffect = {};

            for (int x = 0; x < 16; x++)
            {
                for (int y = 0; y < 4; y++)
                {
                    ChromaBoxEffect.Color[y][x] = colors[(y*16) + x];
                }
            }
            CreateEffect(ChromaSDK::CHROMABOX, ChromaSDK::CHROMA_CUSTOM, &ChromaBoxEffect, NULL);
        }
        break;
    }
}