/* Copyright 2021 DZTECH <moyi4681@live.cn>
 *
 * 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 is31fl3731_led_t PROGMEM g_is31fl3731_leds[IS31FL3731_LED_COUNT] = {

    {0, C8_8, C7_8, C6_8},     // LA17
    {0, C9_8, C7_7, C6_7},     // LA16
    {0, C9_7, C8_7, C6_6},     // LA15
    {0, C9_6, C8_6, C7_6},     // LA14
    {0, C9_5, C8_5, C7_5},     // LA13
    {0, C9_4, C8_4, C7_4},     // LA12
    {0, C9_3, C8_3, C7_3},     // LA11
    {0, C9_2, C8_2, C7_2},     // LA10
    {0, C9_1, C8_1, C7_1},     // LA9
    {0, C2_9, C3_9, C4_9},     // LB0
    {0, C1_9, C3_10, C4_10},   // LB1
    {0, C1_10, C2_10, C4_11},  // LB2
    {0, C1_11, C2_11, C3_11},  // LB3
    {0, C1_13, C2_13, C3_13},  // LB5
    {0, C1_14, C2_14, C3_14},  // LB6

    {0, C1_7, C2_7, C3_7},     // LA7
    {0, C1_6, C2_6, C3_6},     // LA6
    {0, C1_5, C2_5, C3_5},     // LA5
    {0, C1_4, C2_4, C3_4},     // LA4
    {0, C1_3, C2_3, C3_3},     // LA3
    {0, C1_2, C2_2, C4_3},     // LA2
    {0, C1_1, C3_2, C4_2},     // LA1
    {0, C2_1, C3_1, C4_1},     // LA0
    {0, C9_9, C8_9, C7_9},     // LB9
    {0, C9_10, C8_10, C7_10},  // LB10
    {0, C9_11, C8_11, C7_11},  // LB11
    {0, C9_12, C8_12, C7_12},  // LB12
    {0, C9_13, C8_13, C7_13},  // LB13
    {0, C9_14, C8_14, C7_14},  // LB14
    {0, C1_15, C2_15, C3_15},  // LB7

    {0, C1_8, C2_8, C3_8},     // LA8
    {1, C9_6, C8_6, C7_6},     // LC14
    {1, C9_5, C8_5, C7_5},     // LC13
    {1, C9_4, C8_4, C7_4},     // LC12
    {1, C9_3, C8_3, C7_3},     // LC11
    {1, C9_2, C8_2, C7_2},     // LC10
    {1, C9_1, C8_1, C7_1},     // LC9
    {1, C2_9, C3_9, C4_9},     // LD0
    {1, C1_9, C3_10, C4_10},   // LD1
    {1, C1_10, C2_10, C4_11},  // LD2
    {1, C1_11, C2_11, C3_11},  // LD3
    {1, C1_12, C2_12, C3_12},  // LD4
    {1, C1_13, C2_13, C3_13},  // LD5
    {0, C1_16, C2_16, C3_16},  // LB8

    {1, C9_8, C7_7, C6_7},     // LC16
    {1, C1_5, C2_5, C3_5},     // LC5
    {1, C1_4, C2_4, C3_4},     // LC4
    {1, C1_3, C2_3, C3_3},     // LC3
    {1, C1_2, C2_2, C4_3},     // LC2
    {1, C1_1, C3_2, C4_2},     // LC1
    {1, C9_9, C8_9, C7_9},     // LD9
    {1, C9_10, C8_10, C7_10},  // LD10
    {1, C9_11, C8_11, C7_11},  // LD11
    {1, C9_12, C8_12, C7_12},  // LD12
    {1, C1_14, C2_14, C3_14},  // LD6
    {1, C1_15, C2_15, C3_15},  // LD7
    {1, C1_16, C2_16, C3_16},  // LD8
    {0, C9_15, C8_15, C6_14},  // LB15

    {1, C8_8, C7_8, C6_8},     // LC17
    {1, C1_8, C2_8, C3_8},     // LC8
    {1, C1_7, C2_7, C3_7},     // LC7
    {1, C2_1, C3_1, C4_1},     // LC0
    {1, C9_14, C8_14, C7_14},  // LD14
    {1, C9_15, C8_15, C6_14},  // LD15
    {1, C8_16, C7_16, C6_16},  // LD17
    {0, C8_16, C7_16, C6_16},  // LB17
    {0, C9_16, C7_15, C6_15},  // LB16

};

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