rev2.c (4830B)
1 // Copyright 2021 bdtc123 2 // Copyright 2021 sigprof 3 // Copyright 2021 peepeetee 4 // SPDX-License-Identifier: GPL-2.0-or-later 5 #include "quantum.h" 6 7 #if defined(RGB_MATRIX_ENABLE) && defined(RGBLIGHT_ENABLE) 8 # error "RGBLight must not be enabled due to this keyboard's custom RGB Matrix driver" 9 #endif 10 11 #ifdef RGB_MATRIX_ENABLE 12 13 # include "i2c_master.h" 14 # include "is31fl3733.h" 15 # include "ws2812.h" 16 17 // clang-format off 18 const is31fl3733_led_t PROGMEM g_is31fl3733_leds[IS31FL3733_LED_COUNT] = { 19 { 0, SW2_CS1, SW1_CS1, SW3_CS1 }, 20 { 0, SW2_CS2, SW1_CS2, SW3_CS2 }, 21 { 0, SW2_CS3, SW1_CS3, SW3_CS3 }, 22 { 0, SW2_CS4, SW1_CS4, SW3_CS4 }, 23 { 0, SW2_CS5, SW1_CS5, SW3_CS5 }, 24 { 0, SW2_CS6, SW1_CS6, SW3_CS6 }, 25 { 0, SW2_CS7, SW1_CS7, SW3_CS7 }, 26 { 0, SW2_CS8, SW1_CS8, SW3_CS8 }, 27 { 0, SW2_CS9, SW1_CS9, SW3_CS9 }, 28 { 0, SW2_CS10, SW1_CS10, SW3_CS10 }, 29 { 0, SW2_CS11, SW1_CS11, SW3_CS11 }, 30 { 0, SW2_CS12, SW1_CS12, SW3_CS12 }, 31 { 0, SW2_CS13, SW1_CS13, SW3_CS13 }, 32 { 0, SW2_CS14, SW1_CS14, SW3_CS14 }, 33 34 { 0, SW5_CS1, SW4_CS1, SW6_CS1 }, 35 { 0, SW5_CS2, SW4_CS2, SW6_CS2 }, 36 { 0, SW5_CS3, SW4_CS3, SW6_CS3 }, 37 { 0, SW5_CS4, SW4_CS4, SW6_CS4 }, 38 { 0, SW5_CS5, SW4_CS5, SW6_CS5 }, 39 { 0, SW5_CS6, SW4_CS6, SW6_CS6 }, 40 { 0, SW5_CS7, SW4_CS7, SW6_CS7 }, 41 { 0, SW5_CS8, SW4_CS8, SW6_CS8 }, 42 { 0, SW5_CS9, SW4_CS9, SW6_CS9 }, 43 { 0, SW5_CS10, SW4_CS10, SW6_CS10 }, 44 { 0, SW5_CS11, SW4_CS11, SW6_CS11 }, 45 { 0, SW5_CS12, SW4_CS12, SW6_CS12 }, 46 { 0, SW5_CS13, SW4_CS13, SW6_CS13 }, 47 { 0, SW5_CS14, SW4_CS14, SW6_CS14 }, 48 49 { 0, SW8_CS1, SW7_CS1, SW9_CS1 }, 50 { 0, SW8_CS2, SW7_CS2, SW9_CS2 }, 51 { 0, SW8_CS3, SW7_CS3, SW9_CS3 }, 52 { 0, SW8_CS4, SW7_CS4, SW9_CS4 }, 53 { 0, SW8_CS5, SW7_CS5, SW9_CS5 }, 54 { 0, SW8_CS6, SW7_CS6, SW9_CS6 }, 55 { 0, SW8_CS7, SW7_CS7, SW9_CS7 }, 56 { 0, SW8_CS8, SW7_CS8, SW9_CS8 }, 57 { 0, SW8_CS9, SW7_CS9, SW9_CS9 }, 58 { 0, SW8_CS10, SW7_CS10, SW9_CS10 }, 59 { 0, SW8_CS11, SW7_CS11, SW9_CS11 }, 60 { 0, SW8_CS12, SW7_CS12, SW9_CS12 }, 61 { 0, SW8_CS13, SW7_CS13, SW9_CS13 }, 62 63 { 0, SW11_CS1, SW10_CS1, SW12_CS1 }, 64 { 0, SW11_CS11, SW10_CS11, SW12_CS11 }, 65 { 0, SW11_CS2, SW10_CS2, SW12_CS2 }, 66 { 0, SW11_CS3, SW10_CS3, SW12_CS3 }, 67 { 0, SW11_CS4, SW10_CS4, SW12_CS4 }, 68 { 0, SW11_CS5, SW10_CS5, SW12_CS5 }, 69 { 0, SW11_CS6, SW10_CS6, SW12_CS6 }, 70 { 0, SW11_CS7, SW10_CS7, SW12_CS7 }, 71 { 0, SW11_CS8, SW10_CS8, SW12_CS8 }, 72 { 0, SW11_CS9, SW10_CS9, SW12_CS9 }, 73 { 0, SW11_CS10, SW10_CS10, SW12_CS10 }, 74 { 0, SW11_CS12, SW10_CS12, SW12_CS12 }, 75 { 0, SW11_CS13, SW10_CS13, SW12_CS13 }, 76 { 0, SW11_CS14, SW10_CS14, SW12_CS14 }, 77 78 { 0, SW8_CS16, SW7_CS16, SW9_CS16 }, 79 { 0, SW8_CS14, SW7_CS14, SW9_CS14 }, 80 { 0, SW8_CS15, SW7_CS15, SW9_CS15 }, 81 { 0, SW5_CS15, SW4_CS15, SW6_CS15 }, 82 { 0, SW11_CS15, SW10_CS15, SW12_CS15 }, 83 { 0, SW11_CS16, SW10_CS16, SW12_CS16 }, 84 { 0, SW2_CS16, SW1_CS16, SW3_CS16 }, 85 { 0, SW2_CS15, SW1_CS15, SW3_CS15 }, 86 { 0, SW5_CS16, SW4_CS16, SW6_CS16 } 87 }; 88 // clang-format on 89 90 bool rgb_matrix_indicators_kb(void) { 91 if (!rgb_matrix_indicators_user()) { 92 return false; 93 } 94 if (host_keyboard_led_state().caps_lock) { 95 rgb_matrix_set_color(28, 0xFF, 0xFF, 0xFF); 96 } 97 return true; 98 } 99 100 // ========================================================================== 101 // Custom RGB Matrix driver that combines IS31FL3733 and WS2812 102 // ========================================================================== 103 104 static void rgb_matrix_driver_init(void) { 105 i2c_init(); 106 is31fl3733_init(0); 107 ws2812_init(); 108 for (uint8_t index = 0; index < IS31FL3733_LED_COUNT; index++) { 109 bool enabled = true; 110 is31fl3733_set_led_control_register(index, enabled, enabled, enabled); 111 } 112 is31fl3733_update_led_control_registers(0); 113 } 114 115 static void rgb_matrix_driver_flush(void) { 116 is31fl3733_update_pwm_buffers(0); 117 ws2812_flush(); 118 } 119 120 static void rgb_matrix_driver_set_color(int index, uint8_t red, uint8_t green, uint8_t blue) { 121 if (index < IS31FL3733_LED_COUNT) { 122 is31fl3733_set_color(index, red, green, blue); 123 } else { 124 ws2812_set_color(index - IS31FL3733_LED_COUNT, red, green, blue); 125 } 126 } 127 128 static void rgb_matrix_driver_set_color_all(uint8_t red, uint8_t green, uint8_t blue) { 129 is31fl3733_set_color_all(red, green, blue); 130 ws2812_set_color_all(red, green, blue); 131 } 132 133 // clang-format off 134 const rgb_matrix_driver_t rgb_matrix_driver = { 135 .init = rgb_matrix_driver_init, 136 .flush = rgb_matrix_driver_flush, 137 .set_color = rgb_matrix_driver_set_color, 138 .set_color_all = rgb_matrix_driver_set_color_all, 139 }; 140 // clang-format on 141 142 #endif /* RGB_MATRIX_ENABLE */