#include "quantum.h"

#ifdef RGB_MATRIX_ENABLE
const is31fl3733_led_t PROGMEM g_is31fl3733_leds[IS31FL3733_LED_COUNT] = {
    {0, SW2_CS1,    SW1_CS1,    SW3_CS1},
    {0, SW2_CS2,    SW1_CS2,    SW3_CS2},
    {0, SW2_CS3,    SW1_CS3,    SW3_CS3},
    {0, SW2_CS4,    SW1_CS4,    SW3_CS4},
    {0, SW2_CS5,    SW1_CS5,    SW3_CS5},
    {0, SW2_CS6,    SW1_CS6,    SW3_CS6},
    {0, SW2_CS7,    SW1_CS7,    SW3_CS7},
    {0, SW2_CS8,    SW1_CS8,    SW3_CS8},
    {0, SW2_CS9,    SW1_CS9,    SW3_CS9},
    {0, SW2_CS10,   SW1_CS10,   SW3_CS10},
    {0, SW2_CS11,   SW1_CS11,   SW3_CS11},
    {0, SW2_CS12,   SW1_CS12,   SW3_CS12},
    {0, SW2_CS13,   SW1_CS13,   SW3_CS13},
    {0, SW2_CS14,   SW1_CS14,   SW3_CS14},
    {0, SW5_CS1,    SW4_CS1,    SW6_CS1},
    {0, SW5_CS2,    SW4_CS2,    SW6_CS2},
    {0, SW5_CS3,    SW4_CS3,    SW6_CS3},
    {0, SW5_CS4,    SW4_CS4,    SW6_CS4},
    {0, SW5_CS5,    SW4_CS5,    SW6_CS5},
    {0, SW5_CS6,    SW4_CS6,    SW6_CS6},
    {0, SW5_CS7,    SW4_CS7,    SW6_CS7},
    {0, SW5_CS8,    SW4_CS8,    SW6_CS8},
    {0, SW5_CS9,    SW4_CS9,    SW6_CS9},
    {0, SW5_CS10,   SW4_CS10,   SW6_CS10},
    {0, SW5_CS11,   SW4_CS11,   SW6_CS11},
    {0, SW5_CS12,   SW4_CS12,   SW6_CS12},
    {0, SW5_CS13,   SW4_CS13,   SW6_CS13},
    {0, SW5_CS14,   SW4_CS14,   SW6_CS14},
    {0, SW8_CS1,    SW7_CS1,    SW9_CS1},
    {0, SW8_CS2,    SW7_CS2,    SW9_CS2},
    {0, SW8_CS3,    SW7_CS3,    SW9_CS3},
    {0, SW8_CS4,    SW7_CS4,    SW9_CS4},
    {0, SW8_CS5,    SW7_CS5,    SW9_CS5},
    {0, SW8_CS6,    SW7_CS6,    SW9_CS6},
    {0, SW8_CS7,    SW7_CS7,    SW9_CS7},
    {0, SW8_CS8,    SW7_CS8,    SW9_CS8},
    {0, SW8_CS9,    SW7_CS9,    SW9_CS9},
    {0, SW8_CS10,   SW7_CS10,   SW9_CS10},
    {0, SW8_CS11,   SW7_CS11,   SW9_CS11},
    {0, SW8_CS12,   SW7_CS12,   SW9_CS12},
    {0, SW8_CS13,   SW7_CS13,   SW9_CS13},
    {0, SW11_CS1,    SW10_CS1,    SW12_CS1},
    {0, SW11_CS2,    SW10_CS2,    SW12_CS2},
    {0, SW11_CS3,    SW10_CS3,    SW12_CS3},
    {0, SW11_CS4,    SW10_CS4,    SW12_CS4},
    {0, SW11_CS5,    SW10_CS5,    SW12_CS5},
    {0, SW11_CS6,    SW10_CS6,    SW12_CS6},
    {0, SW11_CS7,    SW10_CS7,    SW12_CS7},
    {0, SW11_CS8,    SW10_CS8,    SW12_CS8},
    {0, SW11_CS9,    SW10_CS9,    SW12_CS9},
    {0, SW11_CS10,   SW10_CS10,   SW12_CS10},
    {0, SW11_CS11,   SW10_CS11,   SW12_CS11},
    {0, SW11_CS14,   SW10_CS14,   SW12_CS14},
    {0, SW2_CS16,   SW1_CS16,   SW3_CS16},
    {0, SW5_CS16,   SW4_CS16,   SW6_CS16},
    {0, SW8_CS16,   SW7_CS16,   SW9_CS16},
    {0, SW11_CS16,   SW10_CS16,   SW12_CS16},
    {0, SW11_CS15,   SW10_CS15,   SW12_CS15},
    {0, SW11_CS12,   SW10_CS12,   SW12_CS12},
    {0, SW11_CS13,   SW10_CS13,   SW12_CS13},
    {0, SW8_CS14,   SW7_CS14,   SW9_CS14},
};

led_config_t g_led_config = { {
    {  0,   1,   2,   3,      4,       5,   6,      7,      8,      9,   10,  11,    12,     13 },
    {  14,  15,  16,  17,     18,      19,  20,     21,     22,     23,  24,  25,    26,     27 },
    {  28,  29,  30,  31,     32,      33,  34,     35,     36,     37,  38,  39,    NO_LED, 40 },
    {  41,  42,  43,  44,     45,      46,  47,     48,     49,     50,  51,  NO_LED,NO_LED, 52 },
    {  53,  54,  55,  NO_LED, NO_LED,  56,  NO_LED, NO_LED, NO_LED, 57,  58,  NO_LED,59,     60 }
}, {
    {   0,   0 }, {  16,   0 }, {  32,   0 }, {  48,   0 }, {  64,   0 }, {  80,   0 }, {  96,   0 }, { 112,   0 },  { 128,   0 },  { 144,   0 }, { 160,   0 }, { 176,   0 }, { 192,   0 }, { 216,   0 },
    {   4,  16 }, {  24,  16 }, {  40,  16 }, {  56,  16 }, {  72,  16 }, {  88,  16 }, { 104,  16 }, { 120,  16 },  { 136,  16 },  { 152,  16 }, { 168,  16 }, { 184,  16 }, { 200,  16 }, { 220,  16 },
    {   6,  32 }, {  28,  32 }, {  44,  32 }, {  60,  32 }, {  76,  32 }, {  92,  32 }, { 108,  32 }, { 124,  32 },  { 140,  32 },  { 156,  32 }, { 172,  32 }, { 188,  32 }, { 204,  32 },
    {  10,  48 }, {  20,  48 }, {  52,  48 }, {  68,  48 }, {  84,  48 }, { 100,  48 }, { 116,  48 }, { 132,  48 },  { 148,  48 },  { 164,  48 }, { 180,  48 }, { 210,  48 },
    {   2,  64 }, {  22,  64 }, {  42,  64 }, { 102,  64 }, { 162,  64 }, { 182,  64 }, { 202,  64 }, { 222,  64 },
}, {
    1, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 1,
    4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
    4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 1,
    1, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 1,
    1, 1, 1, 4, 1, 1, 1, 1
} };
#endif

bool rgb_matrix_indicators_kb(void) {
    if (!rgb_matrix_indicators_user()) {
        return false;
    }
    if (host_keyboard_led_state().caps_lock) {
        rgb_matrix_set_color(28, 0xFF, 0xFF, 0xFF);
    }
    return true;
}