/*
Copyright 2022 Terry Mathews

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "quantum.h"

#ifdef RGB_MATRIX_ENABLE
const is31fl3741_led_t PROGMEM g_is31fl3741_leds[IS31FL3741_LED_COUNT] = {
    {0, SW1_CS18, SW1_CS17, SW1_CS16}, /* RGB1 */
    {0, SW2_CS18, SW2_CS17, SW2_CS16}, /* RGB2 */
    {0, SW3_CS18, SW3_CS17, SW3_CS16}, /* RGB3 */
    {0, SW4_CS18, SW4_CS17, SW4_CS16}, /* RGB4 */
    {0, SW5_CS18, SW5_CS17, SW5_CS16}, /* RGB5 */
    {0, SW6_CS18, SW6_CS17, SW6_CS16}, /* RGB6 */
    {0, SW7_CS18, SW7_CS17, SW7_CS16}, /* RGB7 */
    {0, SW8_CS18, SW8_CS17, SW8_CS16}, /* RGB8 */
    {0, SW9_CS18, SW9_CS17, SW9_CS16}, /* RGB9 */
    {0, SW1_CS21, SW1_CS20, SW1_CS19}, /* RGB10 */
    {0, SW2_CS21, SW2_CS20, SW2_CS19}, /* RGB11 */
    {0, SW3_CS21, SW3_CS20, SW3_CS19}, /* RGB12 */
    {0, SW4_CS21, SW4_CS20, SW4_CS19}, /* RGB13 */
    {0, SW5_CS21, SW5_CS20, SW5_CS19}, /* RGB14 */
    {0, SW6_CS21, SW6_CS20, SW6_CS19}, /* RGB15 */
    {0, SW1_CS15, SW1_CS14, SW1_CS13}, /* RGB16 */
    {0, SW2_CS15, SW2_CS14, SW2_CS13}, /* RGB17 */
    {0, SW3_CS15, SW3_CS14, SW3_CS13}, /* RGB18 */
    {0, SW4_CS15, SW4_CS14, SW4_CS13}, /* RGB19 */
    {0, SW5_CS15, SW5_CS14, SW5_CS13}, /* RGB20 */
    {0, SW6_CS15, SW6_CS14, SW6_CS13}, /* RGB21 */
    {0, SW7_CS15, SW7_CS14, SW7_CS13}, /* RGB22 */
    {0, SW8_CS15, SW8_CS14, SW8_CS13}, /* RGB23 */
    {0, SW9_CS15, SW9_CS14, SW9_CS13}, /* RGB24 */
    {0, SW1_CS24, SW1_CS23, SW1_CS22}, /* RGB25 */
    {0, SW2_CS24, SW2_CS23, SW2_CS22}, /* RGB26 */
    {0, SW3_CS24, SW3_CS23, SW3_CS22}, /* RGB27 */
    {0, SW4_CS24, SW4_CS23, SW4_CS22}, /* RGB28 */
    {0, SW5_CS24, SW5_CS23, SW5_CS22}, /* RGB29 */
    {0, SW6_CS24, SW6_CS23, SW6_CS22}, /* RGB30 */
    {0, SW1_CS12, SW1_CS11, SW1_CS10}, /* RGB31 */
    {0, SW2_CS12, SW2_CS11, SW2_CS10}, /* RGB32 */
    {0, SW3_CS12, SW3_CS11, SW3_CS10}, /* RGB33 */
    {0, SW4_CS12, SW4_CS11, SW4_CS10}, /* RGB34 */
    {0, SW5_CS12, SW5_CS11, SW5_CS10}, /* RGB35 */
    {0, SW6_CS12, SW6_CS11, SW6_CS10}, /* RGB36 */
    {0, SW7_CS12, SW7_CS11, SW7_CS10}, /* RGB37 */
    {0, SW8_CS12, SW8_CS11, SW8_CS10}, /* RGB38 */
    {0, SW9_CS12, SW9_CS11, SW9_CS10}, /* RGB39 */
    {0, SW1_CS27, SW1_CS26, SW1_CS25}, /* RGB40 */
    {0, SW2_CS27, SW2_CS26, SW2_CS25}, /* RGB41 */
    {0, SW3_CS27, SW3_CS26, SW3_CS25}, /* RGB42 */
    {0, SW4_CS27, SW4_CS26, SW4_CS25}, /* RGB43 */
    {0, SW6_CS27, SW6_CS26, SW6_CS25}, /* RGB44 */
    {0, SW1_CS9, SW1_CS8, SW1_CS7}, /* RGB45 */
    {0, SW3_CS9, SW3_CS8, SW3_CS7}, /* RGB46 */
    {0, SW4_CS9, SW4_CS8, SW4_CS7}, /* RGB47 */
    {0, SW5_CS9, SW5_CS8, SW5_CS7}, /* RGB48 */
    {0, SW6_CS9, SW6_CS8, SW6_CS7}, /* RGB49 */
    {0, SW7_CS9, SW7_CS8, SW7_CS7}, /* RGB50 */
    {0, SW8_CS9, SW8_CS8, SW8_CS7}, /* RGB51 */
    {0, SW9_CS9, SW9_CS8, SW9_CS7}, /* RGB52 */
    {0, SW1_CS30, SW1_CS29, SW1_CS28}, /* RGB53 */
    {0, SW2_CS30, SW2_CS29, SW2_CS28}, /* RGB54 */
    {0, SW3_CS30, SW3_CS29, SW3_CS28}, /* RGB55 */
    {0, SW4_CS30, SW4_CS29, SW4_CS28}, /* RGB56 */
    {0, SW5_CS30, SW5_CS29, SW5_CS28}, /* RGB57 */
    {0, SW6_CS30, SW6_CS29, SW6_CS28}, /* RGB58 */
    {0, SW1_CS6, SW1_CS5, SW1_CS4}, /* RGB59 */
    {0, SW2_CS6, SW2_CS5, SW2_CS4}, /* RGB60 */
    {0, SW3_CS6, SW3_CS5, SW3_CS4}, /* RGB61 */
    {0, SW7_CS6, SW7_CS5, SW7_CS4}, /* RGB62 */
    {0, SW2_CS33, SW2_CS32, SW2_CS31}, /* RGB63 */
    {0, SW3_CS33, SW3_CS32, SW3_CS31}, /* RGB64 */
    {0, SW4_CS33, SW4_CS32, SW4_CS31}, /* RGB65 */
    {0, SW5_CS33, SW5_CS32, SW5_CS31}, /* RGB66 */
    {0, SW6_CS33, SW6_CS32, SW6_CS31}, /* RGB67 */

    {0,SW1_CS36,SW1_CS35,SW1_CS34}, /* RGB68 */
    {0,SW2_CS36,SW2_CS35,SW2_CS34}, /* RGB69 */
    {0,SW3_CS36,SW3_CS35,SW3_CS34}, /* RGB70 */
    {0,SW4_CS36,SW4_CS35,SW4_CS34}, /* RGB71 */
    {0,SW5_CS36,SW5_CS35,SW5_CS34}, /* RGB72 */
    {0,SW6_CS36,SW6_CS35,SW6_CS34}, /* RGB73 */
    {0,SW7_CS36,SW7_CS35,SW7_CS34}, /* RGB74 */
    {0,SW8_CS36,SW8_CS35,SW8_CS34}, /* RGB75 */
    {0,SW9_CS36,SW9_CS35,SW9_CS34}, /* RGB76 */
    {0,SW1_CS39,SW1_CS38,SW1_CS37}, /* RGB77 */
    {0,SW2_CS39,SW2_CS38,SW2_CS37}, /* RGB78 */
    {0,SW3_CS39,SW3_CS38,SW3_CS37}, /* RGB79 */
    {0,SW4_CS39,SW4_CS38,SW4_CS37}, /* RGB80 */
    {0,SW5_CS39,SW5_CS38,SW5_CS37}, /* RGB81 */
    {0,SW6_CS39,SW6_CS38,SW6_CS37}, /* RGB82 */
};

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, 40, 41, 42, NO_LED, 43 },
        { 44, NO_LED, 45,     46,     47,     48, 49,     50,     51,     52, 53, 54, 55,     56, 57 },
        { 58,     59, 60, NO_LED, NO_LED, NO_LED, 61, NO_LED, NO_LED, NO_LED, 62, 63, 64,     65, 66 }
    }, {
        {   0,   0 }, {  15,   0 }, {  30,   0 }, {  45,   0 }, {  60,   0 }, {  75,   0 }, {  90,   0 }, { 105,   0 }, { 120,   0 }, { 135,   0 }, { 150,   0 }, { 165,   0 }, { 180,   0 }, { 203,   0 }, { 224,   0 },
        {   4,  16 }, {  23,  16 }, {  38,  16 }, {  53,  16 }, {  68,  16 }, {  83,  16 }, {  98,  16 }, { 113,  16 }, { 128,  16 }, { 143,  16 }, { 158,  16 }, { 173,  16 }, { 188,  16 }, { 206,  16 }, { 224,  16 },
        {   6,  32 }, {  26,  32 }, {  41,  32 }, {  56,  32 }, {  71,  32 }, {  86,  32 }, { 101,  32 }, { 116,  32 }, { 131,  32 }, { 146,  32 }, { 161,  32 }, { 176,  32 },               { 201,  32 }, { 224,  32 },
        {   9,  48 },               {  34,  48 }, {  49,  48 }, {  64,  48 }, {  79,  48 }, {  94,  48 }, { 109,  48 }, { 124,  48 }, { 139,  48 }, { 154,  48 }, { 169,  48 }, { 189,  48 }, { 210,  48 }, { 224,  48 },
        {   2,  64 }, {  21,  64 }, {  39,  64 },                                           {  96,  64 },                                           { 152,  64 }, { 171,  64 }, { 195,  64 }, { 210,  64 }, { 224,  64 },
        {0,0},{180,0},{224,0},{200,64},{224,64},{224,32},{44,0},{88,0},{136,0},{0,32},{0,64},{24,64},{78,64},{112,64},{146,64}
}, {
        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, 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, 4, 4, 4, 4, 4, 4,
        4, 4, 4,          4,          4, 4, 4, 4, 4,
        2,2,2,2,2,2,2,2,2,2,2,2,2,2,2
    }
};

bool rgb_matrix_indicators_kb(void) {
    if (!rgb_matrix_indicators_user()) {
        return false;
    }
    if (host_keyboard_led_state().caps_lock) {
        for (uint8_t i = 0; i < RGB_MATRIX_LED_COUNT; i++) {
            rgb_matrix_set_color(i, 0xFF, 0xFF, 0xFF);
        }
    }
    else {
         if (rgb_matrix_get_flags() == LED_FLAG_NONE) {
            for (uint8_t i = 0; i < RGB_MATRIX_LED_COUNT; i++) {
                rgb_matrix_set_color(i, 0, 0, 0);
            }
        }
    }
    return true;
}
#endif