You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

620 lines
22 KiB

  1. /* Copyright 2019 Jason Williams (Wilba)
  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. #ifndef RAW_ENABLE
  17. # error "RAW_ENABLE is not enabled"
  18. #endif
  19. #ifndef DYNAMIC_KEYMAP_ENABLE
  20. # error "DYNAMIC_KEYMAP_ENABLE is not enabled"
  21. #endif
  22. // If VIA_CUSTOM_LIGHTING_ENABLE is not defined, then VIA_QMK_BACKLIGHT_ENABLE is set
  23. // if BACKLIGHT_ENABLE is set, so handling of QMK Backlight values happens here by default.
  24. // if VIA_CUSTOM_LIGHTING_ENABLE is defined, then VIA_QMK_BACKLIGHT_ENABLE must be explicitly
  25. // set in keyboard-level config.h, so handling of QMK Backlight values happens here
  26. #if defined(BACKLIGHT_ENABLE) && !defined(VIA_CUSTOM_LIGHTING_ENABLE)
  27. # define VIA_QMK_BACKLIGHT_ENABLE
  28. #endif
  29. // If VIA_CUSTOM_LIGHTING_ENABLE is not defined, then VIA_QMK_RGBLIGHT_ENABLE is set
  30. // if RGBLIGHT_ENABLE is set, so handling of QMK RGBLIGHT values happens here by default.
  31. // If VIA_CUSTOM_LIGHTING_ENABLE is defined, then VIA_QMK_RGBLIGHT_ENABLE must be explicitly
  32. // set in keyboard-level config.h, so handling of QMK RGBLIGHT values happens here
  33. #if defined(RGBLIGHT_ENABLE) && !defined(VIA_CUSTOM_LIGHTING_ENABLE)
  34. # define VIA_QMK_RGBLIGHT_ENABLE
  35. #endif
  36. #if defined(RGB_MATRIX_ENABLE) && !defined(VIA_QMK_RGBLIGHT_ENABLE) && !defined(VIA_CUSTOM_LIGHTING_ENABLE)
  37. # define VIA_QMK_RGB_MATRIX_ENABLE
  38. #endif
  39. #include "quantum.h"
  40. #include "via.h"
  41. #include "raw_hid.h"
  42. #include "dynamic_keymap.h"
  43. #include "eeprom.h"
  44. #include "version.h" // for QMK_BUILDDATE used in EEPROM magic
  45. #include "via_ensure_keycode.h"
  46. // Forward declare some helpers.
  47. #if defined(VIA_QMK_BACKLIGHT_ENABLE)
  48. void via_qmk_backlight_set_value(uint8_t *data);
  49. void via_qmk_backlight_get_value(uint8_t *data);
  50. #endif
  51. #if defined(VIA_QMK_RGBLIGHT_ENABLE)
  52. void via_qmk_rgblight_set_value(uint8_t *data);
  53. void via_qmk_rgblight_get_value(uint8_t *data);
  54. #endif
  55. #if defined(VIA_QMK_RGB_MATRIX_ENABLE)
  56. # include <lib/lib8tion/lib8tion.h>
  57. void via_qmk_rgb_matrix_set_value(uint8_t *data);
  58. void via_qmk_rgb_matrix_get_value(uint8_t *data);
  59. void eeconfig_update_rgb_matrix(void);
  60. #endif
  61. // Can be called in an overriding via_init_kb() to test if keyboard level code usage of
  62. // EEPROM is invalid and use/save defaults.
  63. bool via_eeprom_is_valid(void) {
  64. char * p = QMK_BUILDDATE; // e.g. "2019-11-05-11:29:54"
  65. uint8_t magic0 = ((p[2] & 0x0F) << 4) | (p[3] & 0x0F);
  66. uint8_t magic1 = ((p[5] & 0x0F) << 4) | (p[6] & 0x0F);
  67. uint8_t magic2 = ((p[8] & 0x0F) << 4) | (p[9] & 0x0F);
  68. return (eeprom_read_byte((void *)VIA_EEPROM_MAGIC_ADDR + 0) == magic0 && eeprom_read_byte((void *)VIA_EEPROM_MAGIC_ADDR + 1) == magic1 && eeprom_read_byte((void *)VIA_EEPROM_MAGIC_ADDR + 2) == magic2);
  69. }
  70. // Sets VIA/keyboard level usage of EEPROM to valid/invalid
  71. // Keyboard level code (eg. via_init_kb()) should not call this
  72. void via_eeprom_set_valid(bool valid) {
  73. char * p = QMK_BUILDDATE; // e.g. "2019-11-05-11:29:54"
  74. uint8_t magic0 = ((p[2] & 0x0F) << 4) | (p[3] & 0x0F);
  75. uint8_t magic1 = ((p[5] & 0x0F) << 4) | (p[6] & 0x0F);
  76. uint8_t magic2 = ((p[8] & 0x0F) << 4) | (p[9] & 0x0F);
  77. eeprom_update_byte((void *)VIA_EEPROM_MAGIC_ADDR + 0, valid ? magic0 : 0xFF);
  78. eeprom_update_byte((void *)VIA_EEPROM_MAGIC_ADDR + 1, valid ? magic1 : 0xFF);
  79. eeprom_update_byte((void *)VIA_EEPROM_MAGIC_ADDR + 2, valid ? magic2 : 0xFF);
  80. }
  81. // Override this at the keyboard code level to check
  82. // VIA's EEPROM valid state and reset to defaults as needed.
  83. // Used by keyboards that store their own state in EEPROM,
  84. // for backlight, rotary encoders, etc.
  85. // The override should not set via_eeprom_set_valid(true) as
  86. // the caller also needs to check the valid state.
  87. __attribute__((weak)) void via_init_kb(void) {}
  88. // Called by QMK core to initialize dynamic keymaps etc.
  89. void via_init(void) {
  90. // Let keyboard level test EEPROM valid state,
  91. // but not set it valid, it is done here.
  92. via_init_kb();
  93. via_set_layout_options_kb(via_get_layout_options());
  94. // If the EEPROM has the magic, the data is good.
  95. // OK to load from EEPROM.
  96. if (!via_eeprom_is_valid()) {
  97. eeconfig_init_via();
  98. }
  99. }
  100. void eeconfig_init_via(void) {
  101. // set the magic number to false, in case this gets interrupted
  102. via_eeprom_set_valid(false);
  103. // This resets the layout options
  104. via_set_layout_options(VIA_EEPROM_LAYOUT_OPTIONS_DEFAULT);
  105. // This resets the keymaps in EEPROM to what is in flash.
  106. dynamic_keymap_reset();
  107. // This resets the macros in EEPROM to nothing.
  108. dynamic_keymap_macro_reset();
  109. // Save the magic number last, in case saving was interrupted
  110. via_eeprom_set_valid(true);
  111. }
  112. // This is generalized so the layout options EEPROM usage can be
  113. // variable, between 1 and 4 bytes.
  114. uint32_t via_get_layout_options(void) {
  115. uint32_t value = 0;
  116. // Start at the most significant byte
  117. void *source = (void *)(VIA_EEPROM_LAYOUT_OPTIONS_ADDR);
  118. for (uint8_t i = 0; i < VIA_EEPROM_LAYOUT_OPTIONS_SIZE; i++) {
  119. value = value << 8;
  120. value |= eeprom_read_byte(source);
  121. source++;
  122. }
  123. return value;
  124. }
  125. __attribute__((weak)) void via_set_layout_options_kb(uint32_t value) {}
  126. void via_set_layout_options(uint32_t value) {
  127. via_set_layout_options_kb(value);
  128. // Start at the least significant byte
  129. void *target = (void *)(VIA_EEPROM_LAYOUT_OPTIONS_ADDR + VIA_EEPROM_LAYOUT_OPTIONS_SIZE - 1);
  130. for (uint8_t i = 0; i < VIA_EEPROM_LAYOUT_OPTIONS_SIZE; i++) {
  131. eeprom_update_byte(target, value & 0xFF);
  132. value = value >> 8;
  133. target--;
  134. }
  135. }
  136. // Called by QMK core to process VIA-specific keycodes.
  137. bool process_record_via(uint16_t keycode, keyrecord_t *record) {
  138. // Handle macros
  139. if (record->event.pressed) {
  140. if (keycode >= MACRO00 && keycode <= MACRO15) {
  141. uint8_t id = keycode - MACRO00;
  142. dynamic_keymap_macro_send(id);
  143. return false;
  144. }
  145. }
  146. // TODO: ideally this would be generalized and refactored into
  147. // QMK core as advanced keycodes, until then, the simple case
  148. // can be available here to keyboards using VIA
  149. switch (keycode) {
  150. case FN_MO13:
  151. if (record->event.pressed) {
  152. layer_on(1);
  153. update_tri_layer(1, 2, 3);
  154. } else {
  155. layer_off(1);
  156. update_tri_layer(1, 2, 3);
  157. }
  158. return false;
  159. break;
  160. case FN_MO23:
  161. if (record->event.pressed) {
  162. layer_on(2);
  163. update_tri_layer(1, 2, 3);
  164. } else {
  165. layer_off(2);
  166. update_tri_layer(1, 2, 3);
  167. }
  168. return false;
  169. break;
  170. }
  171. return true;
  172. }
  173. // Keyboard level code can override this to handle custom messages from VIA.
  174. // See raw_hid_receive() implementation.
  175. // DO NOT call raw_hid_send() in the override function.
  176. __attribute__((weak)) void raw_hid_receive_kb(uint8_t *data, uint8_t length) {
  177. uint8_t *command_id = &(data[0]);
  178. *command_id = id_unhandled;
  179. }
  180. // VIA handles received HID messages first, and will route to
  181. // raw_hid_receive_kb() for command IDs that are not handled here.
  182. // This gives the keyboard code level the ability to handle the command
  183. // specifically.
  184. //
  185. // raw_hid_send() is called at the end, with the same buffer, which was
  186. // possibly modified with returned values.
  187. void raw_hid_receive(uint8_t *data, uint8_t length) {
  188. uint8_t *command_id = &(data[0]);
  189. uint8_t *command_data = &(data[1]);
  190. switch (*command_id) {
  191. case id_get_protocol_version: {
  192. command_data[0] = VIA_PROTOCOL_VERSION >> 8;
  193. command_data[1] = VIA_PROTOCOL_VERSION & 0xFF;
  194. break;
  195. }
  196. case id_get_keyboard_value: {
  197. switch (command_data[0]) {
  198. case id_uptime: {
  199. uint32_t value = timer_read32();
  200. command_data[1] = (value >> 24) & 0xFF;
  201. command_data[2] = (value >> 16) & 0xFF;
  202. command_data[3] = (value >> 8) & 0xFF;
  203. command_data[4] = value & 0xFF;
  204. break;
  205. }
  206. case id_layout_options: {
  207. uint32_t value = via_get_layout_options();
  208. command_data[1] = (value >> 24) & 0xFF;
  209. command_data[2] = (value >> 16) & 0xFF;
  210. command_data[3] = (value >> 8) & 0xFF;
  211. command_data[4] = value & 0xFF;
  212. break;
  213. }
  214. case id_switch_matrix_state: {
  215. #if ((MATRIX_COLS / 8 + 1) * MATRIX_ROWS <= 28)
  216. uint8_t i = 1;
  217. for (uint8_t row = 0; row < MATRIX_ROWS; row++) {
  218. matrix_row_t value = matrix_get_row(row);
  219. # if (MATRIX_COLS > 24)
  220. command_data[i++] = (value >> 24) & 0xFF;
  221. # endif
  222. # if (MATRIX_COLS > 16)
  223. command_data[i++] = (value >> 16) & 0xFF;
  224. # endif
  225. # if (MATRIX_COLS > 8)
  226. command_data[i++] = (value >> 8) & 0xFF;
  227. # endif
  228. command_data[i++] = value & 0xFF;
  229. }
  230. #endif
  231. break;
  232. }
  233. default: {
  234. raw_hid_receive_kb(data, length);
  235. break;
  236. }
  237. }
  238. break;
  239. }
  240. case id_set_keyboard_value: {
  241. switch (command_data[0]) {
  242. case id_layout_options: {
  243. uint32_t value = ((uint32_t)command_data[1] << 24) | ((uint32_t)command_data[2] << 16) | ((uint32_t)command_data[3] << 8) | (uint32_t)command_data[4];
  244. via_set_layout_options(value);
  245. break;
  246. }
  247. default: {
  248. raw_hid_receive_kb(data, length);
  249. break;
  250. }
  251. }
  252. break;
  253. }
  254. case id_dynamic_keymap_get_keycode: {
  255. uint16_t keycode = dynamic_keymap_get_keycode(command_data[0], command_data[1], command_data[2]);
  256. command_data[3] = keycode >> 8;
  257. command_data[4] = keycode & 0xFF;
  258. break;
  259. }
  260. case id_dynamic_keymap_set_keycode: {
  261. dynamic_keymap_set_keycode(command_data[0], command_data[1], command_data[2], (command_data[3] << 8) | command_data[4]);
  262. break;
  263. }
  264. case id_dynamic_keymap_reset: {
  265. dynamic_keymap_reset();
  266. break;
  267. }
  268. case id_lighting_set_value: {
  269. #if defined(VIA_QMK_BACKLIGHT_ENABLE)
  270. via_qmk_backlight_set_value(command_data);
  271. #endif
  272. #if defined(VIA_QMK_RGBLIGHT_ENABLE)
  273. via_qmk_rgblight_set_value(command_data);
  274. #endif
  275. #if defined(VIA_QMK_RGB_MATRIX_ENABLE)
  276. via_qmk_rgb_matrix_set_value(command_data);
  277. #endif
  278. #if defined(VIA_CUSTOM_LIGHTING_ENABLE)
  279. raw_hid_receive_kb(data, length);
  280. #endif
  281. #if !defined(VIA_QMK_BACKLIGHT_ENABLE) && !defined(VIA_QMK_RGBLIGHT_ENABLE) && !defined(VIA_CUSTOM_LIGHTING_ENABLE) && !defined(VIA_QMK_RGB_MATRIX_ENABLE)
  282. // Return the unhandled state
  283. *command_id = id_unhandled;
  284. #endif
  285. break;
  286. }
  287. case id_lighting_get_value: {
  288. #if defined(VIA_QMK_BACKLIGHT_ENABLE)
  289. via_qmk_backlight_get_value(command_data);
  290. #endif
  291. #if defined(VIA_QMK_RGBLIGHT_ENABLE)
  292. via_qmk_rgblight_get_value(command_data);
  293. #endif
  294. #if defined(VIA_QMK_RGB_MATRIX_ENABLE)
  295. via_qmk_rgb_matrix_get_value(command_data);
  296. #endif
  297. #if defined(VIA_CUSTOM_LIGHTING_ENABLE)
  298. raw_hid_receive_kb(data, length);
  299. #endif
  300. #if !defined(VIA_QMK_BACKLIGHT_ENABLE) && !defined(VIA_QMK_RGBLIGHT_ENABLE) && !defined(VIA_CUSTOM_LIGHTING_ENABLE) && !defined(VIA_QMK_RGB_MATRIX_ENABLE)
  301. // Return the unhandled state
  302. *command_id = id_unhandled;
  303. #endif
  304. break;
  305. }
  306. case id_lighting_save: {
  307. #if defined(VIA_QMK_BACKLIGHT_ENABLE)
  308. eeconfig_update_backlight_current();
  309. #endif
  310. #if defined(VIA_QMK_RGBLIGHT_ENABLE)
  311. eeconfig_update_rgblight_current();
  312. #endif
  313. #if defined(VIA_QMK_RGB_MATRIX_ENABLE)
  314. eeconfig_update_rgb_matrix();
  315. #endif
  316. #if defined(VIA_CUSTOM_LIGHTING_ENABLE)
  317. raw_hid_receive_kb(data, length);
  318. #endif
  319. #if !defined(VIA_QMK_BACKLIGHT_ENABLE) && !defined(VIA_QMK_RGBLIGHT_ENABLE) && !defined(VIA_CUSTOM_LIGHTING_ENABLE) && !defined(VIA_QMK_RGB_MATRIX_ENABLE)
  320. // Return the unhandled state
  321. *command_id = id_unhandled;
  322. #endif
  323. break;
  324. }
  325. #ifdef VIA_EEPROM_ALLOW_RESET
  326. case id_eeprom_reset: {
  327. via_eeprom_set_valid(false);
  328. eeconfig_init_via();
  329. break;
  330. }
  331. #endif
  332. case id_dynamic_keymap_macro_get_count: {
  333. command_data[0] = dynamic_keymap_macro_get_count();
  334. break;
  335. }
  336. case id_dynamic_keymap_macro_get_buffer_size: {
  337. uint16_t size = dynamic_keymap_macro_get_buffer_size();
  338. command_data[0] = size >> 8;
  339. command_data[1] = size & 0xFF;
  340. break;
  341. }
  342. case id_dynamic_keymap_macro_get_buffer: {
  343. uint16_t offset = (command_data[0] << 8) | command_data[1];
  344. uint16_t size = command_data[2]; // size <= 28
  345. dynamic_keymap_macro_get_buffer(offset, size, &command_data[3]);
  346. break;
  347. }
  348. case id_dynamic_keymap_macro_set_buffer: {
  349. uint16_t offset = (command_data[0] << 8) | command_data[1];
  350. uint16_t size = command_data[2]; // size <= 28
  351. dynamic_keymap_macro_set_buffer(offset, size, &command_data[3]);
  352. break;
  353. }
  354. case id_dynamic_keymap_macro_reset: {
  355. dynamic_keymap_macro_reset();
  356. break;
  357. }
  358. case id_dynamic_keymap_get_layer_count: {
  359. command_data[0] = dynamic_keymap_get_layer_count();
  360. break;
  361. }
  362. case id_dynamic_keymap_get_buffer: {
  363. uint16_t offset = (command_data[0] << 8) | command_data[1];
  364. uint16_t size = command_data[2]; // size <= 28
  365. dynamic_keymap_get_buffer(offset, size, &command_data[3]);
  366. break;
  367. }
  368. case id_dynamic_keymap_set_buffer: {
  369. uint16_t offset = (command_data[0] << 8) | command_data[1];
  370. uint16_t size = command_data[2]; // size <= 28
  371. dynamic_keymap_set_buffer(offset, size, &command_data[3]);
  372. break;
  373. }
  374. #ifdef ENCODER_MAP_ENABLE
  375. case id_dynamic_keymap_get_encoder: {
  376. uint16_t keycode = dynamic_keymap_get_encoder(command_data[0], command_data[1], command_data[2] != 0);
  377. command_data[3] = keycode >> 8;
  378. command_data[4] = keycode & 0xFF;
  379. break;
  380. }
  381. case id_dynamic_keymap_set_encoder: {
  382. dynamic_keymap_set_encoder(command_data[0], command_data[1], command_data[2] != 0, (command_data[3] << 8) | command_data[4]);
  383. break;
  384. }
  385. #endif
  386. default: {
  387. // The command ID is not known
  388. // Return the unhandled state
  389. *command_id = id_unhandled;
  390. break;
  391. }
  392. }
  393. // Return the same buffer, optionally with values changed
  394. // (i.e. returning state to the host, or the unhandled state).
  395. raw_hid_send(data, length);
  396. }
  397. #if defined(VIA_QMK_BACKLIGHT_ENABLE)
  398. # if BACKLIGHT_LEVELS == 0
  399. # error BACKLIGHT_LEVELS == 0
  400. # endif
  401. void via_qmk_backlight_get_value(uint8_t *data) {
  402. uint8_t *value_id = &(data[0]);
  403. uint8_t *value_data = &(data[1]);
  404. switch (*value_id) {
  405. case id_qmk_backlight_brightness: {
  406. // level / BACKLIGHT_LEVELS * 255
  407. value_data[0] = ((uint16_t)get_backlight_level() * UINT8_MAX) / BACKLIGHT_LEVELS;
  408. break;
  409. }
  410. case id_qmk_backlight_effect: {
  411. # ifdef BACKLIGHT_BREATHING
  412. value_data[0] = is_backlight_breathing() ? 1 : 0;
  413. # else
  414. value_data[0] = 0;
  415. # endif
  416. break;
  417. }
  418. }
  419. }
  420. void via_qmk_backlight_set_value(uint8_t *data) {
  421. uint8_t *value_id = &(data[0]);
  422. uint8_t *value_data = &(data[1]);
  423. switch (*value_id) {
  424. case id_qmk_backlight_brightness: {
  425. // level / 255 * BACKLIGHT_LEVELS
  426. backlight_level_noeeprom(((uint16_t)value_data[0] * BACKLIGHT_LEVELS) / UINT8_MAX);
  427. break;
  428. }
  429. case id_qmk_backlight_effect: {
  430. # ifdef BACKLIGHT_BREATHING
  431. if (value_data[0] == 0) {
  432. backlight_disable_breathing();
  433. } else {
  434. backlight_enable_breathing();
  435. }
  436. # endif
  437. break;
  438. }
  439. }
  440. }
  441. #endif // #if defined(VIA_QMK_BACKLIGHT_ENABLE)
  442. #if defined(VIA_QMK_RGBLIGHT_ENABLE)
  443. # ifndef RGBLIGHT_LIMIT_VAL
  444. # define RGBLIGHT_LIMIT_VAL 255
  445. # endif
  446. void via_qmk_rgblight_get_value(uint8_t *data) {
  447. uint8_t *value_id = &(data[0]);
  448. uint8_t *value_data = &(data[1]);
  449. switch (*value_id) {
  450. case id_qmk_rgblight_brightness: {
  451. value_data[0] = ((uint16_t)rgblight_get_val() * UINT8_MAX) / RGBLIGHT_LIMIT_VAL;
  452. break;
  453. }
  454. case id_qmk_rgblight_effect: {
  455. value_data[0] = rgblight_get_mode();
  456. break;
  457. }
  458. case id_qmk_rgblight_effect_speed: {
  459. value_data[0] = rgblight_get_speed();
  460. break;
  461. }
  462. case id_qmk_rgblight_color: {
  463. value_data[0] = rgblight_get_hue();
  464. value_data[1] = rgblight_get_sat();
  465. break;
  466. }
  467. }
  468. }
  469. void via_qmk_rgblight_set_value(uint8_t *data) {
  470. uint8_t *value_id = &(data[0]);
  471. uint8_t *value_data = &(data[1]);
  472. switch (*value_id) {
  473. case id_qmk_rgblight_brightness: {
  474. rgblight_sethsv_noeeprom(rgblight_get_hue(), rgblight_get_sat(), ((uint16_t)value_data[0] * RGBLIGHT_LIMIT_VAL) / UINT8_MAX);
  475. break;
  476. }
  477. case id_qmk_rgblight_effect: {
  478. rgblight_mode_noeeprom(value_data[0]);
  479. if (value_data[0] == 0) {
  480. rgblight_disable_noeeprom();
  481. } else {
  482. rgblight_enable_noeeprom();
  483. }
  484. break;
  485. }
  486. case id_qmk_rgblight_effect_speed: {
  487. rgblight_set_speed_noeeprom(value_data[0]);
  488. break;
  489. }
  490. case id_qmk_rgblight_color: {
  491. rgblight_sethsv_noeeprom(value_data[0], value_data[1], rgblight_get_val());
  492. break;
  493. }
  494. }
  495. }
  496. #endif // #if defined(VIA_QMK_RGBLIGHT_ENABLE)
  497. #if defined(VIA_QMK_RGB_MATRIX_ENABLE)
  498. # if !defined(RGB_MATRIX_MAXIMUM_BRIGHTNESS) || RGB_MATRIX_MAXIMUM_BRIGHTNESS > UINT8_MAX
  499. # undef RGB_MATRIX_MAXIMUM_BRIGHTNESS
  500. # define RGB_MATRIX_MAXIMUM_BRIGHTNESS UINT8_MAX
  501. # endif
  502. // VIA supports only 4 discrete values for effect speed; map these to some
  503. // useful speed values for RGB Matrix.
  504. enum speed_values {
  505. RGBLIGHT_SPEED_0 = UINT8_MAX / 16, // not 0 to avoid really slow effects
  506. RGBLIGHT_SPEED_1 = UINT8_MAX / 4,
  507. RGBLIGHT_SPEED_2 = UINT8_MAX / 2, // matches the default value
  508. RGBLIGHT_SPEED_3 = UINT8_MAX / 4 * 3, // UINT8_MAX is really fast
  509. };
  510. static uint8_t speed_from_rgblight(uint8_t rgblight_speed) {
  511. switch (rgblight_speed) {
  512. case 0:
  513. return RGBLIGHT_SPEED_0;
  514. case 1:
  515. return RGBLIGHT_SPEED_1;
  516. case 2:
  517. default:
  518. return RGBLIGHT_SPEED_2;
  519. case 3:
  520. return RGBLIGHT_SPEED_3;
  521. }
  522. }
  523. static uint8_t speed_to_rgblight(uint8_t rgb_matrix_speed) {
  524. if (rgb_matrix_speed < ((RGBLIGHT_SPEED_0 + RGBLIGHT_SPEED_1) / 2)) {
  525. return 0;
  526. } else if (rgb_matrix_speed < ((RGBLIGHT_SPEED_1 + RGBLIGHT_SPEED_2) / 2)) {
  527. return 1;
  528. } else if (rgb_matrix_speed < ((RGBLIGHT_SPEED_2 + RGBLIGHT_SPEED_3) / 2)) {
  529. return 2;
  530. } else {
  531. return 3;
  532. }
  533. }
  534. void via_qmk_rgb_matrix_get_value(uint8_t *data) {
  535. uint8_t *value_id = &(data[0]);
  536. uint8_t *value_data = &(data[1]);
  537. switch (*value_id) {
  538. case id_qmk_rgblight_brightness:
  539. value_data[0] = ((uint16_t)rgb_matrix_get_val() * UINT8_MAX) / RGB_MATRIX_MAXIMUM_BRIGHTNESS;
  540. break;
  541. case id_qmk_rgblight_effect:
  542. value_data[0] = rgb_matrix_get_mode();
  543. break;
  544. case id_qmk_rgblight_effect_speed:
  545. value_data[0] = speed_to_rgblight(rgb_matrix_get_speed());
  546. break;
  547. case id_qmk_rgblight_color:
  548. value_data[0] = rgb_matrix_get_hue();
  549. value_data[1] = rgb_matrix_get_sat();
  550. break;
  551. }
  552. }
  553. void via_qmk_rgb_matrix_set_value(uint8_t *data) {
  554. uint8_t *value_id = &(data[0]);
  555. uint8_t *value_data = &(data[1]);
  556. switch (*value_id) {
  557. case id_qmk_rgblight_brightness:
  558. rgb_matrix_sethsv_noeeprom(rgb_matrix_get_hue(), rgb_matrix_get_sat(), scale8(value_data[0], RGB_MATRIX_MAXIMUM_BRIGHTNESS));
  559. break;
  560. case id_qmk_rgblight_effect:
  561. rgb_matrix_mode_noeeprom(value_data[0]);
  562. if (value_data[0] == 0) {
  563. rgb_matrix_disable_noeeprom();
  564. } else {
  565. rgb_matrix_enable_noeeprom();
  566. }
  567. break;
  568. case id_qmk_rgblight_effect_speed:
  569. rgb_matrix_set_speed_noeeprom(speed_from_rgblight(value_data[0]));
  570. break;
  571. case id_qmk_rgblight_color:
  572. rgb_matrix_sethsv_noeeprom(value_data[0], value_data[1], rgb_matrix_get_val());
  573. break;
  574. }
  575. }
  576. #endif // #if defined(VIA_QMK_RGB_MATRIX_ENABLE)