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.

225 lines
8.3 KiB

  1. #include QMK_KEYBOARD_H
  2. #include "bootloader.h"
  3. #ifdef PROTOCOL_LUFA
  4. #include "lufa.h"
  5. #include "split_util.h"
  6. #endif
  7. #ifdef SSD1306OLED
  8. #include "ssd1306.h"
  9. #endif
  10. #ifdef RGBLIGHT_ENABLE
  11. //Following line allows macro to read current RGB settings
  12. extern rgblight_config_t rgblight_config;
  13. #endif
  14. extern uint8_t is_master;
  15. enum layer_names {
  16. _QWERTY,
  17. _LOWER,
  18. _RAISE,
  19. _ADJUST
  20. };
  21. enum custom_keycodes {
  22. QWERTY = SAFE_RANGE,
  23. LOWER,
  24. RAISE,
  25. RGBRST,
  26. MBTN1,
  27. SCRL
  28. };
  29. #define KC_LOWER LOWER
  30. #define KC_RAISE RAISE
  31. #define KC_RST RESET
  32. #define KC_LRST RGBRST
  33. #define KC_LTOG RGB_TOG
  34. #define KC_LHUI RGB_HUI
  35. #define KC_LHUD RGB_HUD
  36. #define KC_LSAI RGB_SAI
  37. #define KC_LSAD RGB_SAD
  38. #define KC_LVAI RGB_VAI
  39. #define KC_LVAD RGB_VAD
  40. #define KC_LMOD RGB_MOD
  41. #define KC_CTLA CTL_T(KC_A)
  42. #define KC_CTLSC CTL_T(KC_SCLN)
  43. #define KC_SFTZ SFT_T(KC_Z)
  44. #define KC_SFTSL SFT_T(KC_SLSH)
  45. #define KC_WINX LWIN_T(KC_X)
  46. #define KC_WINDO RWIN_T(KC_DOT)
  47. #define KC_MBTN1 MBTN1
  48. #define KC_SCRL SCRL
  49. const uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = {
  50. [_QWERTY] = LAYOUT(
  51. //,-----------------------------------------. ,-----------------------------------------.
  52. KC_ESC, KC_Q, KC_W, KC_E, KC_R, KC_T, KC_Y, KC_U, KC_I, KC_O, KC_P, KC_BSPC,
  53. //|------+------+------+------+------+------| |------+------+------+------+------+------|
  54. KC_TAB, KC_CTLA, KC_S, KC_D, KC_F, KC_G, KC_H, KC_J, KC_K, KC_L, KC_CTLSC, KC_QUOT,
  55. //|------+------+------+------+------+------| |------+------+------+------+------+------|
  56. KC_GRAVE, KC_SFTZ, KC_WINX, KC_C, KC_V, KC_B, KC_N, KC_M, KC_COMM, KC_WINDO, KC_SFTSL,KC_BSLASH,
  57. //|------+------+------+------+------+------+------| |------+------+------+------+------+------+------|
  58. KC_LOWER, KC_SPC, KC_SCRL, KC_MBTN1, KC_ENT, KC_RAISE
  59. //`--------------------' `--------------------'
  60. ),
  61. [_LOWER] = LAYOUT(
  62. //,-----------------------------------------. ,-----------------------------------------.
  63. KC_ESC, XXXXXXX, KC_PGDN, KC_PSCR, KC_PGUP, KC_LBRC, KC_RBRC, KC_7, KC_8, KC_9, XXXXXXX, XXXXXXX,
  64. //|------+------+------+------+------+------| |------+------+------+------+------+------|
  65. XXXXXXX, KC_LCTRL, KC_PLUS, KC_MINS, KC_EQL, KC_LPRN, KC_RPRN, KC_4, KC_5, KC_6, KC_RCTRL, XXXXXXX,
  66. //|------+------+------+------+------+------| |------+------+------+------+------+------|
  67. XXXXXXX, KC_LSFT, KC_HOME, XXXXXXX, KC_END, KC_LCBR, KC_RCBR, KC_1, KC_2, KC_3, KC_RSFT, XXXXXXX,
  68. //|------+------+------+------+------+------+------| |------+------+------+------+------+------+------|
  69. KC_LOWER, KC_SPC, KC_SCRL, KC_MBTN1, KC_ENT, KC_0
  70. //`--------------------' `--------------------'
  71. ),
  72. [_RAISE] = LAYOUT(
  73. //,-----------------------------------------. ,-----------------------------------------.
  74. KC_ESC, XXXXXXX, KC_F7, KC_F8, KC_F9, KC_F10, KC_BTN2, KC_BTN2, KC_MNXT, KC_MPRV, KC_MPLY, KC_MSTP,
  75. //|------+------+------+------+------+------| |------+------+------+------+------+------|
  76. XXXXXXX, KC_LCTRL, KC_F4, KC_F5, KC_F6, KC_F11, KC_LEFT, KC_DOWN, KC_UP, KC_RIGHT, KC_RCTRL, XXXXXXX,
  77. //|------+------+------+------+------+------| |------+------+------+------+------+------|
  78. XXXXXXX, KC_LSFT, KC_F1, KC_F2, KC_F3, KC_F12, XXXXXXX, XXXXXXX, KC_VOLU, KC_VOLD, KC_MUTE, KC_RSFT,
  79. //|------+------+------+------+------+------+------| |------+------+------+------+------+------+------|
  80. KC_LOWER, KC_SPC, KC_SCRL, KC_MBTN1, KC_ENT, KC_RAISE
  81. //`--------------------' `--------------------'
  82. ),
  83. [_ADJUST] = LAYOUT(
  84. //,-----------------------------------------. ,-----------------------------------------.
  85. KC_RST, KC_LRST, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, KC_RST,
  86. //|------+------+------+------+------+------| |------+------+------+------+------+------|
  87. KC_LTOG, KC_LHUI, KC_LSAI, KC_LVAI, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX,
  88. //|------+------+------+------+------+------| |------+------+------+------+------+------|
  89. KC_LMOD, KC_LHUD, KC_LSAD, KC_LVAD, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX,
  90. //|------+------+------+------+------+------+------| |------+------+------+------+------+------+------|
  91. KC_LOWER, KC_SPC, KC_SCRL, KC_MBTN1, KC_ENT, KC_RAISE
  92. //`--------------------' `--------------------'
  93. )
  94. };
  95. int RGB_current_mode;
  96. void persistent_default_layer_set(uint16_t default_layer) {
  97. eeconfig_update_default_layer(default_layer);
  98. default_layer_set(default_layer);
  99. }
  100. // Setting ADJUST layer RGB back to default
  101. void update_tri_layer_RGB(uint8_t layer1, uint8_t layer2, uint8_t layer3) {
  102. if (IS_LAYER_ON(layer1) && IS_LAYER_ON(layer2)) {
  103. layer_on(layer3);
  104. } else {
  105. layer_off(layer3);
  106. }
  107. }
  108. void matrix_init_user(void) {
  109. #ifdef RGBLIGHT_ENABLE
  110. RGB_current_mode = rgblight_config.mode;
  111. #endif
  112. //SSD1306 OLED init, make sure to add #define SSD1306OLED in config.h
  113. #ifdef SSD1306OLED
  114. iota_gfx_init(!has_usb()); // turns on the display
  115. #endif
  116. }
  117. //SSD1306 OLED update loop, make sure to add #define SSD1306OLED in config.h
  118. #ifdef SSD1306OLED
  119. // When add source files to SRC in rules.mk, you can use functions.
  120. const char *read_layer_state(void);
  121. const char *read_logo(void);
  122. void set_keylog(uint16_t keycode, keyrecord_t *record);
  123. const char *read_keylog(void);
  124. const char *read_keylogs(void);
  125. void matrix_scan_user(void) {
  126. iota_gfx_task();
  127. }
  128. void matrix_render_user(struct CharacterMatrix *matrix) {
  129. if (is_master) {
  130. // If you want to change the display of OLED, you need to change here
  131. matrix_write_ln(matrix, read_layer_state());
  132. matrix_write_ln(matrix, read_keylog());
  133. matrix_write_ln(matrix, read_keylogs());
  134. } else {
  135. matrix_write(matrix, read_logo());
  136. }
  137. }
  138. void matrix_update(struct CharacterMatrix *dest, const struct CharacterMatrix *source) {
  139. if (memcmp(dest->display, source->display, sizeof(dest->display))) {
  140. memcpy(dest->display, source->display, sizeof(dest->display));
  141. dest->dirty = true;
  142. }
  143. }
  144. void iota_gfx_task_user(void) {
  145. struct CharacterMatrix matrix;
  146. matrix_clear(&matrix);
  147. matrix_render_user(&matrix);
  148. matrix_update(&display, &matrix);
  149. }
  150. #endif//SSD1306OLED
  151. bool process_record_user(uint16_t keycode, keyrecord_t *record) {
  152. if (record->event.pressed) {
  153. #ifdef SSD1306OLED
  154. set_keylog(keycode, record);
  155. #endif
  156. }
  157. switch (keycode) {
  158. case QWERTY:
  159. if (record->event.pressed) {
  160. persistent_default_layer_set(1UL<<_QWERTY);
  161. }
  162. return false;
  163. case LOWER:
  164. if (record->event.pressed) {
  165. layer_on(_LOWER);
  166. update_tri_layer_RGB(_LOWER, _RAISE, _ADJUST);
  167. } else {
  168. layer_off(_LOWER);
  169. update_tri_layer_RGB(_LOWER, _RAISE, _ADJUST);
  170. }
  171. return false;
  172. case RAISE:
  173. if (record->event.pressed) {
  174. layer_on(_RAISE);
  175. update_tri_layer_RGB(_LOWER, _RAISE, _ADJUST);
  176. } else {
  177. layer_off(_RAISE);
  178. update_tri_layer_RGB(_LOWER, _RAISE, _ADJUST);
  179. }
  180. return false;
  181. case RGB_MOD:
  182. #ifdef RGBLIGHT_ENABLE
  183. if (record->event.pressed) {
  184. rgblight_mode(RGB_current_mode);
  185. rgblight_step();
  186. RGB_current_mode = rgblight_config.mode;
  187. }
  188. #endif
  189. return false;
  190. case RGBRST:
  191. #ifdef RGBLIGHT_ENABLE
  192. if (record->event.pressed) {
  193. eeconfig_update_rgblight_default();
  194. rgblight_enable();
  195. RGB_current_mode = rgblight_config.mode;
  196. }
  197. #endif
  198. break;
  199. }
  200. return true;
  201. }