/*---------------------------------------------------------*\
| RGBController_GigabyteRGBFusion2GPU.cpp                   |
|                                                           |
|   RGBController for Gigabyte Aorus RGB Fusion 2 GPU       |
|                                                           |
|   This file is part of the OpenRGB project                |
|   SPDX-License-Identifier: GPL-2.0-only                   |
\*---------------------------------------------------------*/

#include "RGBController_GigabyteRGBFusion2GPU.h"
#include "LogManager.h"

/**------------------------------------------------------------------*\
    @name Gigabyte Fusion 2 GPU
    @category GPU
    @type I2C
    @save :white_check_mark:
    @direct :white_check_mark:
    @effects :white_check_mark:
    @detectors DetectGigabyteRGBFusion2GPUControllers
    @comment
\*-------------------------------------------------------------------*/

RGBController_RGBFusion2GPU::RGBController_RGBFusion2GPU(RGBFusion2GPUController* controller_ptr)
{
    controller  = controller_ptr;

    name        = "Gigabyte GPU";
    vendor      = "Gigabyte";
    description = "Gigabyte RGB Fusion 2 GPU";
    location    = controller->GetDeviceLocation();

    type = DEVICE_TYPE_GPU;

    mode Direct;
    Direct.name                 = "Direct";
    Direct.value                = RGB_FUSION2_GPU_MODE_STATIC;
    Direct.flags                = MODE_FLAG_HAS_PER_LED_COLOR | MODE_FLAG_HAS_BRIGHTNESS | MODE_FLAG_MANUAL_SAVE;
    Direct.color_mode           = MODE_COLORS_PER_LED;
    Direct.brightness_min       = RGB_FUSION2_GPU_BRIGHTNESS_MIN;
    Direct.brightness_max       = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    Direct.brightness           = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    modes.push_back(Direct);

    mode Breathing;
    Breathing.name              = "Pulse";
    Breathing.value             = RGB_FUSION2_GPU_MODE_BREATHING;
    Breathing.flags             = MODE_FLAG_HAS_SPEED | MODE_FLAG_HAS_PER_LED_COLOR | MODE_FLAG_MANUAL_SAVE;
    Breathing.speed_min         = RGB_FUSION2_GPU_SPEED_SLOWEST;
    Breathing.speed_max         = RGB_FUSION2_GPU_SPEED_FASTEST;
    Breathing.speed             = RGB_FUSION2_GPU_SPEED_NORMAL;
    Breathing.color_mode        = MODE_COLORS_PER_LED;
    Breathing.brightness_min    = RGB_FUSION2_GPU_BRIGHTNESS_MIN;
    Breathing.brightness_max    = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    Breathing.brightness        = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    modes.push_back(Breathing);

    mode Flashing;
    Flashing.name               = "Flash";
    Flashing.value              = RGB_FUSION2_GPU_MODE_FLASHING;
    Flashing.flags              = MODE_FLAG_HAS_SPEED | MODE_FLAG_HAS_PER_LED_COLOR | MODE_FLAG_HAS_BRIGHTNESS | MODE_FLAG_MANUAL_SAVE;
    Flashing.speed_min          = RGB_FUSION2_GPU_SPEED_SLOWEST;
    Flashing.speed_max          = RGB_FUSION2_GPU_SPEED_FASTEST;
    Flashing.speed              = RGB_FUSION2_GPU_SPEED_NORMAL;
    Flashing.color_mode         = MODE_COLORS_PER_LED;
    Flashing.brightness_min     = RGB_FUSION2_GPU_BRIGHTNESS_MIN;
    Flashing.brightness_max     = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    Flashing.brightness         = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    modes.push_back(Flashing);

    mode DualFlashing;
    DualFlashing.name           = "Double Flash";
    DualFlashing.value          = RGB_FUSION2_GPU_MODE_DUAL_FLASHING;
    DualFlashing.flags          = MODE_FLAG_HAS_SPEED | MODE_FLAG_HAS_PER_LED_COLOR | MODE_FLAG_HAS_BRIGHTNESS | MODE_FLAG_MANUAL_SAVE;
    DualFlashing.speed_min      = RGB_FUSION2_GPU_SPEED_SLOWEST;
    DualFlashing.speed_max      = RGB_FUSION2_GPU_SPEED_FASTEST;
    DualFlashing.speed          = RGB_FUSION2_GPU_SPEED_NORMAL;
    DualFlashing.color_mode     = MODE_COLORS_PER_LED;
    DualFlashing.brightness_min = RGB_FUSION2_GPU_BRIGHTNESS_MIN;
    DualFlashing.brightness_max = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    DualFlashing.brightness     = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    modes.push_back(DualFlashing);

    mode SpectrumCycle;
    SpectrumCycle.name              = "Color Cycle";
    SpectrumCycle.value             = RGB_FUSION2_GPU_MODE_COLOR_CYCLE;
    SpectrumCycle.flags             = MODE_FLAG_HAS_SPEED | MODE_FLAG_HAS_BRIGHTNESS;;
    SpectrumCycle.speed_min         = RGB_FUSION2_GPU_SPEED_SLOWEST;
    SpectrumCycle.speed_max         = RGB_FUSION2_GPU_SPEED_FASTEST;
    SpectrumCycle.speed             = RGB_FUSION2_GPU_SPEED_NORMAL;
    SpectrumCycle.color_mode        = MODE_COLORS_NONE;
    SpectrumCycle.brightness_min    = RGB_FUSION2_GPU_BRIGHTNESS_MIN;
    SpectrumCycle.brightness_max    = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    SpectrumCycle.brightness        = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    modes.push_back(SpectrumCycle);

    mode Gradient;
    Gradient.name               = "Gradient";
    Gradient.value              = RGB_FUSION2_GPU_MODE_GRADIENT;
    Gradient.flags              = MODE_FLAG_HAS_SPEED | MODE_FLAG_HAS_PER_LED_COLOR | MODE_FLAG_HAS_BRIGHTNESS | MODE_FLAG_MANUAL_SAVE;
    Gradient.speed_min          = RGB_FUSION2_GPU_SPEED_SLOWEST;
    Gradient.speed_max          = RGB_FUSION2_GPU_SPEED_FASTEST;
    Gradient.speed              = RGB_FUSION2_GPU_SPEED_NORMAL;
    Gradient.color_mode         = MODE_COLORS_PER_LED;
    Gradient.brightness_min     = RGB_FUSION2_GPU_BRIGHTNESS_MIN;
    Gradient.brightness_max     = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    Gradient.brightness         = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    modes.push_back(Gradient);

    mode Wave;
    Wave.name                   = "Wave";
    Wave.value                  = RGB_FUSION2_GPU_MODE_WAVE;
    Wave.flags                  = MODE_FLAG_HAS_SPEED | MODE_FLAG_HAS_BRIGHTNESS | MODE_FLAG_MANUAL_SAVE;
    Wave.speed_min              = RGB_FUSION2_GPU_SPEED_SLOWEST;
    Wave.speed_max              = RGB_FUSION2_GPU_SPEED_FASTEST;
    Wave.speed                  = RGB_FUSION2_GPU_SPEED_NORMAL;
    Wave.color_mode             = MODE_COLORS_NONE;
    Wave.brightness_min         = RGB_FUSION2_GPU_BRIGHTNESS_MIN;
    Wave.brightness_max         = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    Wave.brightness             = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    modes.push_back(Wave);

    mode Shift;
    Shift.name                  = "Color Shift";
    Shift.value                 = RGB_FUSION2_GPU_MODE_COLOR_SHIFT;
    Shift.flags                 = MODE_FLAG_HAS_SPEED | MODE_FLAG_HAS_MODE_SPECIFIC_COLOR | MODE_FLAG_HAS_BRIGHTNESS | MODE_FLAG_MANUAL_SAVE;
    Shift.speed_min             = RGB_FUSION2_GPU_SPEED_SLOWEST;
    Shift.speed_max             = RGB_FUSION2_GPU_SPEED_FASTEST;
    Shift.speed                 = RGB_FUSION2_GPU_SPEED_NORMAL;
    Shift.color_mode            = MODE_COLORS_MODE_SPECIFIC;
    Shift.colors_min            = 1;
    Shift.colors_max            = 8;
    Shift.colors.resize(8);
    Shift.brightness_min        = RGB_FUSION2_GPU_BRIGHTNESS_MIN;
    Shift.brightness_max        = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    Shift.brightness            = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    modes.push_back(Shift);

    mode Tricolor;
    Tricolor.name               = "Tricolor";
    Tricolor.value              = RGB_FUSION2_GPU_MODE_TRICOLOR;
    Tricolor.flags              = MODE_FLAG_HAS_SPEED | MODE_FLAG_HAS_MODE_SPECIFIC_COLOR | MODE_FLAG_HAS_BRIGHTNESS | MODE_FLAG_MANUAL_SAVE;
    Tricolor.speed_min          = RGB_FUSION2_GPU_SPEED_SLOWEST;
    Tricolor.speed_max          = RGB_FUSION2_GPU_SPEED_FASTEST;
    Tricolor.speed              = RGB_FUSION2_GPU_SPEED_NORMAL;
    Tricolor.color_mode         = MODE_COLORS_MODE_SPECIFIC;
    Tricolor.colors_min         = 1;
    Tricolor.colors_max         = 3;
    Tricolor.colors.resize(3);
    Tricolor.brightness_min     = RGB_FUSION2_GPU_BRIGHTNESS_MIN;
    Tricolor.brightness_max     = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    Tricolor.brightness         = RGB_FUSION2_GPU_BRIGHTNESS_MAX;
    modes.push_back(Tricolor);

    SetupZones();
}

RGBController_RGBFusion2GPU::~RGBController_RGBFusion2GPU()
{
    delete controller;
}

void RGBController_RGBFusion2GPU::SetupZones()
{
    /*---------------------------------------------------------*\
    | This device only allows setting the entire zone for all   |
    | LED's in the zone and does not allow per LED control.     |
    \*---------------------------------------------------------*/

    for(uint8_t zone_idx = 0; zone_idx < RGB_FUSION_2_GPU_NUMBER_OF_ZONES; zone_idx++)
    {
        zone new_zone;
        led  new_led;

        new_zone.name          = "GPU zone " + std::to_string(zone_idx + 1);
        new_zone.type          = ZONE_TYPE_SINGLE;
        new_zone.leds_min      = 1;
        new_zone.leds_max      = 1;
        new_zone.leds_count    = 1;
        new_zone.matrix_map    = NULL;

        new_led.name           = new_zone.name;
        /*---------------------------------------------------------*\
        | Push the zone and LED on to device vectors                |
        \*---------------------------------------------------------*/
        leds.push_back(new_led);
        zones.push_back(new_zone);
    }

    SetupColors();
}

void RGBController_RGBFusion2GPU::ResizeZone(int /*zone*/, int /*new_size*/)
{
    /*---------------------------------------------------------*\
    | This device does not support resizing zones               |
    \*---------------------------------------------------------*/
}

void RGBController_RGBFusion2GPU::DeviceUpdateLEDs()
{
    fusion2_config zone_config;

    zone_config.brightness      = modes[active_mode].brightness;
    zone_config.speed           = modes[active_mode].speed;
    zone_config.direction       = modes[active_mode].direction;
    zone_config.numberOfColors  = (uint8_t)modes[active_mode].colors.size();

    for(uint8_t zone_idx = 0; zone_idx < RGB_FUSION_2_GPU_NUMBER_OF_ZONES; zone_idx++)
    {
        zone_config.colors[0] = colors[zone_idx];

        if (modes[active_mode].color_mode == MODE_COLORS_MODE_SPECIFIC)
        {
            for (uint8_t i = 0; i < zone_config.numberOfColors; i++)
            {
                zone_config.colors[i] = modes[active_mode].colors[i];
            }
        }

        controller->SetZone(zone_idx, modes[active_mode].value, zone_config);
    }
}

void RGBController_RGBFusion2GPU::UpdateZoneLEDs(int zone)
{
    LOG_TRACE("[%s] Update zone #%d", name.c_str(), zone);
    DeviceUpdateLEDs();
}

void RGBController_RGBFusion2GPU::UpdateSingleLED(int led)
{
    LOG_TRACE("[%s] Update single led : %d", name.c_str(), led);
    DeviceUpdateLEDs();
}

void RGBController_RGBFusion2GPU::DeviceUpdateMode()
{
    LOG_TRACE("[%s] Switching to mode %s @ brightness %d and speed %d", name.c_str(), modes[active_mode].name.c_str(), modes[active_mode].brightness, modes[active_mode].speed);
    DeviceUpdateLEDs();
}

void RGBController_RGBFusion2GPU::DeviceSaveMode()
{
    controller->SaveConfig();
}