logo

qmk_firmware

custom branch of QMK firmware git clone https://anongit.hacktivis.me/git/qmk_firmware.git

rev2.c (7745B)

    

  1. /* Copyright 2021 bdtc123
  2.  * Copyright 2021 sigprof
  3.  * Copyright 2021 peepeetee
  4.  *
  5.  * This program is free software: you can redistribute it and/or modify
  6.  * it under the terms of the GNU General Public License as published by
  7.  * the Free Software Foundation, either version 2 of the License, or
  8.  * (at your option) any later version.
  9.  *
  10.  * This program is distributed in the hope that it will be useful,
  11.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13.  * GNU General Public License for more details.
  14.  *
  15.  * You should have received a copy of the GNU General Public License
  16.  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17.  */
  18. #include "quantum.h"
  20. #ifdef RGB_MATRIX_ENABLE
  22. #    include "i2c_master.h"
  23. #    include "is31fl3733.h"
  24. #    include "ws2812.h"
  27. const is31fl3733_led_t PROGMEM g_is31fl3733_leds[IS31FL3733_LED_COUNT] = {
  28.     { 0, SW2_CS1, SW1_CS1, SW3_CS1 },
  29.     { 0, SW2_CS2, SW1_CS2, SW3_CS2 },
  30.     { 0, SW2_CS3, SW1_CS3, SW3_CS3 },
  31.     { 0, SW2_CS4, SW1_CS4, SW3_CS4 },
  32.     { 0, SW2_CS5, SW1_CS5, SW3_CS5 },
  33.     { 0, SW2_CS6, SW1_CS6, SW3_CS6 },
  34.     { 0, SW2_CS7, SW1_CS7, SW3_CS7 },
  35.     { 0, SW2_CS8, SW1_CS8, SW3_CS8 },
  36.     { 0, SW2_CS9, SW1_CS9, SW3_CS9 },
  37.     { 0, SW2_CS10, SW1_CS10, SW3_CS10 },
  38.     { 0, SW2_CS11, SW1_CS11, SW3_CS11 },
  39.     { 0, SW2_CS12, SW1_CS12, SW3_CS12 },
  40.     { 0, SW2_CS13, SW1_CS13, SW3_CS13 },
  41.     { 0, SW2_CS14, SW1_CS14, SW3_CS14 },
  43.     { 0, SW5_CS1,  SW4_CS1,  SW6_CS1 },
  44.     { 0, SW5_CS2,  SW4_CS2,  SW6_CS2 },
  45.     { 0, SW5_CS3,  SW4_CS3,  SW6_CS3 },
  46.     { 0, SW5_CS4,  SW4_CS4,  SW6_CS4 },
  47.     { 0, SW5_CS5,  SW4_CS5,  SW6_CS5 },
  48.     { 0, SW5_CS6,  SW4_CS6,  SW6_CS6 },
  49.     { 0, SW5_CS7,  SW4_CS7,  SW6_CS7 },
  50.     { 0, SW5_CS8,  SW4_CS8,  SW6_CS8 },
  51.     { 0, SW5_CS9,  SW4_CS9,  SW6_CS9 },
  52.     { 0, SW5_CS10, SW4_CS10, SW6_CS10 },
  53.     { 0, SW5_CS11, SW4_CS11, SW6_CS11 },
  54.     { 0, SW5_CS12, SW4_CS12, SW6_CS12 },
  55.     { 0, SW5_CS13, SW4_CS13, SW6_CS13 },
  56.     { 0, SW5_CS14, SW4_CS14, SW6_CS14 },
  58.     { 0, SW8_CS1,  SW7_CS1,  SW9_CS1 },
  59.     { 0, SW8_CS2,  SW7_CS2,  SW9_CS2 },
  60.     { 0, SW8_CS3,  SW7_CS3,  SW9_CS3 },
  61.     { 0, SW8_CS4,  SW7_CS4,  SW9_CS4 },
  62.     { 0, SW8_CS5,  SW7_CS5,  SW9_CS5 },
  63.     { 0, SW8_CS6,  SW7_CS6,  SW9_CS6 },
  64.     { 0, SW8_CS7,  SW7_CS7,  SW9_CS7 },
  65.     { 0, SW8_CS8,  SW7_CS8,  SW9_CS8 },
  66.     { 0, SW8_CS9,  SW7_CS9,  SW9_CS9 },
  67.     { 0, SW8_CS10, SW7_CS10, SW9_CS10 },
  68.     { 0, SW8_CS11, SW7_CS11, SW9_CS11 },
  69.     { 0, SW8_CS12, SW7_CS12, SW9_CS12 },
  70.     { 0, SW8_CS13, SW7_CS13, SW9_CS13 },
  72.     { 0, SW11_CS1,  SW10_CS1,  SW12_CS1 },
  73.     { 0, SW11_CS11, SW10_CS11, SW12_CS11 },
  74.     { 0, SW11_CS2,  SW10_CS2,  SW12_CS2 },
  75.     { 0, SW11_CS3,  SW10_CS3,  SW12_CS3 },
  76.     { 0, SW11_CS4,  SW10_CS4,  SW12_CS4 },
  77.     { 0, SW11_CS5,  SW10_CS5,  SW12_CS5 },
  78.     { 0, SW11_CS6,  SW10_CS6,  SW12_CS6 },
  79.     { 0, SW11_CS7,  SW10_CS7,  SW12_CS7 },
  80.     { 0, SW11_CS8,  SW10_CS8,  SW12_CS8 },
  81.     { 0, SW11_CS9,  SW10_CS9,  SW12_CS9 },
  82.     { 0, SW11_CS10, SW10_CS10, SW12_CS10 },
  83.     { 0, SW11_CS12, SW10_CS12, SW12_CS12 },
  84.     { 0, SW11_CS13, SW10_CS13, SW12_CS13 },
  85.     { 0, SW11_CS14, SW10_CS14, SW12_CS14 },
  87.     { 0, SW8_CS16, SW7_CS16, SW9_CS16 },
  88.     { 0, SW8_CS14, SW7_CS14, SW9_CS14 },
  89.     { 0, SW8_CS15, SW7_CS15, SW9_CS15 },
  90.     { 0, SW5_CS15, SW4_CS15, SW6_CS15 },
  91.     { 0, SW11_CS15, SW10_CS15, SW12_CS15 },
  92.     { 0, SW11_CS16, SW10_CS16, SW12_CS16 },
  93.     { 0, SW2_CS16, SW1_CS16, SW3_CS16 },
  94.     { 0, SW2_CS15, SW1_CS15, SW3_CS15 },
  95.     { 0, SW5_CS16, SW4_CS16, SW6_CS16 }
  96. };
  98. led_config_t g_led_config = { {
  99.   {   0,   1,   2,     3,     4,     5,     6,     7,     8,   9,  10,  11,  12,  13 },
  100.   {  14,  15,  16,    17,    18,    19,    20,    21,    22,  23,  24,  25,  26,  27 },
  101.   {  28,  29,  30,    31,    32,    33,    34,    35,    36,  37,  38,  39,  40,NO_LED},
  102.   {  41,  58,  42,    43,    44,    45,    46,    47,    48,  49,  51,  52,  53,  54 },
  103.   {  55,  56,  57,NO_LED,NO_LED,    58,NO_LED,NO_LED,NO_LED,  59,  60,  61,  62,  63 }
  104. }, {
  105.     // Esc, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, -, =, Backspace
  106.     {   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 }, { 202,   0 },
  107.     // Tab, Q, W, E, R, T, Y, U, I, O, P, [, ],  ,
  108.     {   7,  16 }, {  22,  16 }, {  37,  16 }, {  52,  16 }, {  67,  16 }, {  82,  16 }, {  97,  16 }, { 112,  16 }, { 127,  16 }, { 142,  16 }, { 157,  16 }, { 172,  16 }, { 187,  16 }, { 206,  16 },
  109.     // Capslock, A, S, D, F, G, H, J, K, L, ;, ', Enter
  110.     {  11,  32 }, {  26,  32 }, {  41,  32 }, {  56,  32 }, {  71,  32 }, {  86,  32 }, { 101,  32 }, { 116,  32 }, { 131,  32 }, { 146,  32 }, { 161,  32 }, { 176,  32 }, { 198,  32 },
  111.     // LShift, <, Z, X, C, V, B, N, M, ,, ., /, Shift, Up,
  112.     {   3,  48 }, {  22,  48 }, {  33,  48 }, {  48,  48 }, {  63,  48 }, {  78,  48 }, {  93,  48 }, { 108,  48 }, { 123,  48 }, { 138,  48 }, { 153,  48 }, { 168,  48 }, { 194,  48 }, { 213,  48 },
  113.     // Ctrl, GUI, Alt, Space, RAlt, FN, Left, Down, Right
  114.     {   3,  64 }, {  22,  64 }, {  33,  64 }, { 101,  64 }, { 135,  64 }, { 153,  64 }, { 195,  64 }, { 210,  64 }, { 225,  64 }
  115. #    if WS2812_LED_COUNT > 0
  116.         ,{ 28, 40}, { 62, 40}, { 96, 40}, {130, 40}, {164, 40}, {198, 40}
  117. #    endif
  118. }, {
  119.     // Esc, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, -, =, Backspace
  120.     4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 1,
  121.     // Tab, Q, W, E, R, T, Y, U, I, O, P, [, ], Enter
  122.     1, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 1,
  123.     // Capslock, A, S, D, F, G, H, J, K, L, ;, ',#
  124.     9, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  125.     // LShift,backslash, Z, X, C, V, B, N, M, ,, ., Shift, Up, /
  126.     1, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 1, 1, 4,
  127.     // Ctrl, GUI, Alt, Space, RAlt, FN, Left, Down, Right
  128.     1, 1, 1, 4, 1, 1, 1, 1, 1
  129. #    if WS2812_LED_COUNT > 0
  130.         ,2, 2, 2, 2, 2, 2
  131. #    endif
  132. } };
  134. bool rgb_matrix_indicators_kb(void) {
  135.     if (!rgb_matrix_indicators_user()) {
  136.         return false;
  137.     }
  138.     if (host_keyboard_led_state().caps_lock) {
  139.         rgb_matrix_set_color(28, 0xFF, 0xFF, 0xFF);
  140.     }
  141.     return true;
  142. }
  144. // clang-format on
  146. // ==========================================================================
  147. // Custom RGB Matrix driver that combines IS31FL3733 and WS2812
  148. // ==========================================================================
  150. static void rgb_matrix_driver_init(void) {
  151.     i2c_init();
  152.     is31fl3733_init(0);
  153.     ws2812_init();
  154.     for (uint8_t index = 0; index < IS31FL3733_LED_COUNT; index++) {
  155.         bool enabled = true;
  156.         is31fl3733_set_led_control_register(index, enabled, enabled, enabled);
  157.     }
  158.     is31fl3733_update_led_control_registers(0);
  159. }
  161. static void rgb_matrix_driver_flush(void) {
  162.     is31fl3733_update_pwm_buffers(0);
  163. #    if WS2812_LED_COUNT > 0
  164.     ws2812_flush();
  165. #    endif
  166. }
  168. static void rgb_matrix_driver_set_color(int index, uint8_t red, uint8_t green, uint8_t blue) {
  169.     if (index < IS31FL3733_LED_COUNT) {
  170.         is31fl3733_set_color(index, red, green, blue);
  171. #    if WS2812_LED_COUNT > 0
  172.     } else {
  173.         ws2812_set_color(index - IS31FL3733_LED_COUNT, red, green, blue);
  174. #    endif
  175.     }
  176. }
  178. static void rgb_matrix_driver_set_color_all(uint8_t red, uint8_t green, uint8_t blue) {
  179.     is31fl3733_set_color_all(red, green, blue);
  180. #    if WS2812_LED_COUNT > 0
  181.     ws2812_set_color_all(red, green, blue);
  182. #    endif
  183. }
  185. // clang-format off
  186. const rgb_matrix_driver_t rgb_matrix_driver = {
  187.     .init          = rgb_matrix_driver_init,
  188.     .flush         = rgb_matrix_driver_flush,
  189.     .set_color     = rgb_matrix_driver_set_color,
  190.     .set_color_all = rgb_matrix_driver_set_color_all,
  191. };
  192. // clang-format on
  194. #endif /* RGB_MATRIX_ENABLE */