process_rgb.c 9.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210
  1. /* Copyright 2019
  2. *
  3. * This program is free software: you can redistribute it and/or modify
  4. * it under the terms of the GNU General Public License as published by
  5. * the Free Software Foundation, either version 2 of the License, or
  6. * (at your option) any later version.
  7. *
  8. * This program is distributed in the hope that it will be useful,
  9. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11. * GNU General Public License for more details.
  12. *
  13. * You should have received a copy of the GNU General Public License
  14. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  15. */
  16. #include "process_rgb.h"
  17. #include "rgb.h"
  18. typedef void (*rgb_func_pointer)(void);
  19. /**
  20. * Wrapper for inc/dec rgb keycode
  21. *
  22. * noinline to optimise for firmware size not speed (not in hot path)
  23. */
  24. static void __attribute__((noinline)) handleKeycodeRGB(const uint8_t is_shifted, const rgb_func_pointer inc_func, const rgb_func_pointer dec_func) {
  25. if (is_shifted) {
  26. dec_func();
  27. } else {
  28. inc_func();
  29. }
  30. }
  31. /**
  32. * Wrapper for animation mode
  33. * - if not in animation family -> jump to that animation
  34. * - otherwise -> wrap round animation speed
  35. *
  36. * noinline to optimise for firmware size not speed (not in hot path)
  37. */
  38. static void __attribute__((noinline, unused)) handleKeycodeRGBMode(const uint8_t start, const uint8_t end) {
  39. if ((start <= rgblight_get_mode()) && (rgblight_get_mode() < end)) {
  40. rgblight_step();
  41. } else {
  42. rgblight_mode(start);
  43. }
  44. }
  45. /**
  46. * Handle keycodes for both rgblight and rgbmatrix
  47. */
  48. bool process_rgb(const uint16_t keycode, const keyrecord_t *record) {
  49. #ifndef SPLIT_KEYBOARD
  50. if (record->event.pressed) {
  51. #else
  52. // Split keyboards need to trigger on key-up for edge-case issue
  53. if (!record->event.pressed) {
  54. #endif
  55. uint8_t shifted = get_mods() & MOD_MASK_SHIFT;
  56. switch (keycode) {
  57. case RGB_TOG:
  58. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES)
  59. rgblight_toggle();
  60. #endif
  61. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES)
  62. rgb_matrix_toggle();
  63. #endif
  64. return false;
  65. case RGB_MODE_FORWARD:
  66. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES)
  67. handleKeycodeRGB(shifted, rgblight_step, rgblight_step_reverse);
  68. #endif
  69. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES)
  70. handleKeycodeRGB(shifted, rgb_matrix_step, rgb_matrix_step_reverse);
  71. #endif
  72. return false;
  73. case RGB_MODE_REVERSE:
  74. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES)
  75. handleKeycodeRGB(shifted, rgblight_step_reverse, rgblight_step);
  76. #endif
  77. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES)
  78. handleKeycodeRGB(shifted, rgb_matrix_step_reverse, rgb_matrix_step);
  79. #endif
  80. return false;
  81. case RGB_HUI:
  82. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES)
  83. handleKeycodeRGB(shifted, rgblight_increase_hue, rgblight_decrease_hue);
  84. #endif
  85. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES)
  86. handleKeycodeRGB(shifted, rgb_matrix_increase_hue, rgb_matrix_decrease_hue);
  87. #endif
  88. return false;
  89. case RGB_HUD:
  90. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES)
  91. handleKeycodeRGB(shifted, rgblight_decrease_hue, rgblight_increase_hue);
  92. #endif
  93. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES)
  94. handleKeycodeRGB(shifted, rgb_matrix_decrease_hue, rgb_matrix_increase_hue);
  95. #endif
  96. return false;
  97. case RGB_SAI:
  98. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES)
  99. handleKeycodeRGB(shifted, rgblight_increase_sat, rgblight_decrease_sat);
  100. #endif
  101. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES)
  102. handleKeycodeRGB(shifted, rgb_matrix_increase_sat, rgb_matrix_decrease_sat);
  103. #endif
  104. return false;
  105. case RGB_SAD:
  106. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES)
  107. handleKeycodeRGB(shifted, rgblight_decrease_sat, rgblight_increase_sat);
  108. #endif
  109. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES)
  110. handleKeycodeRGB(shifted, rgb_matrix_decrease_sat, rgb_matrix_increase_sat);
  111. #endif
  112. return false;
  113. case RGB_VAI:
  114. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES)
  115. handleKeycodeRGB(shifted, rgblight_increase_val, rgblight_decrease_val);
  116. #endif
  117. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES)
  118. handleKeycodeRGB(shifted, rgb_matrix_increase_val, rgb_matrix_decrease_val);
  119. #endif
  120. return false;
  121. case RGB_VAD:
  122. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES)
  123. handleKeycodeRGB(shifted, rgblight_decrease_val, rgblight_increase_val);
  124. #endif
  125. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES)
  126. handleKeycodeRGB(shifted, rgb_matrix_decrease_val, rgb_matrix_increase_val);
  127. #endif
  128. return false;
  129. case RGB_SPI:
  130. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES)
  131. handleKeycodeRGB(shifted, rgblight_increase_speed, rgblight_decrease_speed);
  132. #endif
  133. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES)
  134. handleKeycodeRGB(shifted, rgb_matrix_increase_speed, rgb_matrix_decrease_speed);
  135. #endif
  136. return false;
  137. case RGB_SPD:
  138. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES)
  139. handleKeycodeRGB(shifted, rgblight_decrease_speed, rgblight_increase_speed);
  140. #endif
  141. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES)
  142. handleKeycodeRGB(shifted, rgb_matrix_decrease_speed, rgb_matrix_increase_speed);
  143. #endif
  144. return false;
  145. case RGB_MODE_PLAIN:
  146. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES)
  147. rgblight_mode(RGBLIGHT_MODE_STATIC_LIGHT);
  148. #endif
  149. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES)
  150. rgb_matrix_mode(RGB_MATRIX_SOLID_COLOR);
  151. #endif
  152. return false;
  153. case RGB_MODE_BREATHE:
  154. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES) && defined(RGBLIGHT_EFFECT_BREATHING)
  155. handleKeycodeRGBMode(RGBLIGHT_MODE_BREATHING, RGBLIGHT_MODE_BREATHING_end);
  156. #endif
  157. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES) && !defined(DISABLE_RGB_MATRIX_BREATHING)
  158. rgb_matrix_mode(RGB_MATRIX_BREATHING);
  159. #endif
  160. return false;
  161. case RGB_MODE_RAINBOW:
  162. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES) && defined( RGBLIGHT_EFFECT_RAINBOW_MOOD)
  163. handleKeycodeRGBMode(RGBLIGHT_MODE_RAINBOW_MOOD, RGBLIGHT_MODE_RAINBOW_MOOD_end);
  164. #endif
  165. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES) && !defined(DISABLE_RGB_MATRIX_CYCLE_LEFT_RIGHT)
  166. rgb_matrix_mode(RGB_MATRIX_CYCLE_LEFT_RIGHT);
  167. #endif
  168. return false;
  169. case RGB_MODE_SWIRL:
  170. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES) && defined( RGBLIGHT_EFFECT_RAINBOW_SWIRL)
  171. handleKeycodeRGBMode(RGBLIGHT_MODE_RAINBOW_SWIRL, RGBLIGHT_MODE_RAINBOW_SWIRL_end);
  172. #endif
  173. #if defined(RGB_MATRIX_ENABLE) && !defined(RGB_MATRIX_DISABLE_KEYCODES) && !defined(DISABLE_RGB_MATRIX_CYCLE_PINWHEEL)
  174. rgb_matrix_mode(RGB_MATRIX_CYCLE_PINWHEEL);
  175. #endif
  176. return false;
  177. case RGB_MODE_SNAKE:
  178. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES) && defined( RGBLIGHT_EFFECT_SNAKE)
  179. handleKeycodeRGBMode(RGBLIGHT_MODE_SNAKE, RGBLIGHT_MODE_SNAKE_end);
  180. #endif
  181. return false;
  182. case RGB_MODE_KNIGHT:
  183. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES) && defined( RGBLIGHT_EFFECT_KNIGHT)
  184. handleKeycodeRGBMode(RGBLIGHT_MODE_KNIGHT, RGBLIGHT_MODE_KNIGHT_end);
  185. #endif
  186. return false;
  187. case RGB_MODE_XMAS:
  188. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES) && defined( RGBLIGHT_EFFECT_CHRISTMAS)
  189. rgblight_mode(RGBLIGHT_MODE_CHRISTMAS);
  190. #endif
  191. return false;
  192. case RGB_MODE_GRADIENT:
  193. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES) && defined( RGBLIGHT_EFFECT_STATIC_GRADIENT)
  194. handleKeycodeRGBMode(RGBLIGHT_MODE_STATIC_GRADIENT, RGBLIGHT_MODE_STATIC_GRADIENT_end);
  195. #endif
  196. return false;
  197. case RGB_MODE_RGBTEST:
  198. #if defined(RGBLIGHT_ENABLE) && !defined(RGBLIGHT_DISABLE_KEYCODES) && defined( RGBLIGHT_EFFECT_RGB_TEST)
  199. rgblight_mode(RGBLIGHT_MODE_RGB_TEST);
  200. #endif
  201. return false;
  202. }
  203. }
  204. return true;
  205. }