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