ergodox_ez.c 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451
  1. /*
  2. Copyright 2012 Jun Wako <wakojun@gmail.com>
  3. Copyright 2013 Oleg Kostyuk <cub.uanic@gmail.com>
  4. Copyright 2015 ZSA Technology Labs Inc (@zsa)
  5. Copyright 2020 Christopher Courtney <drashna@live.com> (@drashna)
  6. This program is free software: you can redistribute it and/or modify
  7. it under the terms of the GNU General Public License as published by
  8. the Free Software Foundation, either version 2 of the License, or
  9. (at your option) any later version.
  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. You should have received a copy of the GNU General Public License
  15. along with this program. If not, see <http://www.gnu.org/licenses/>.
  16. */
  17. #include "ergodox_ez.h"
  18. extern inline void ergodox_board_led_on(void);
  19. extern inline void ergodox_right_led_1_on(void);
  20. extern inline void ergodox_right_led_2_on(void);
  21. extern inline void ergodox_right_led_3_on(void);
  22. extern inline void ergodox_right_led_on(uint8_t led);
  23. extern inline void ergodox_board_led_off(void);
  24. extern inline void ergodox_right_led_1_off(void);
  25. extern inline void ergodox_right_led_2_off(void);
  26. extern inline void ergodox_right_led_3_off(void);
  27. extern inline void ergodox_right_led_off(uint8_t led);
  28. extern inline void ergodox_led_all_on(void);
  29. extern inline void ergodox_led_all_off(void);
  30. extern inline void ergodox_right_led_1_set(uint8_t n);
  31. extern inline void ergodox_right_led_2_set(uint8_t n);
  32. extern inline void ergodox_right_led_3_set(uint8_t n);
  33. extern inline void ergodox_right_led_set(uint8_t led, uint8_t n);
  34. extern inline void ergodox_led_all_set(uint8_t n);
  35. keyboard_config_t keyboard_config;
  36. bool i2c_initialized = 0;
  37. i2c_status_t mcp23018_status = 0x20;
  38. void matrix_init_kb(void) {
  39. // keyboard LEDs (see "PWM on ports OC1(A|B|C)" in "teensy-2-0.md")
  40. TCCR1A = 0b10101001; // set and configure fast PWM
  41. TCCR1B = 0b00001001; // set and configure fast PWM
  42. // (tied to Vcc for hardware convenience)
  43. DDRB &= ~(1 << 4); // set B(4) as input
  44. PORTB &= ~(1 << 4); // set B(4) internal pull-up disabled
  45. // unused pins - C7, D4, D5, D7, E6
  46. // set as input with internal pull-up enabled
  47. DDRC &= ~(1 << 7);
  48. DDRD &= ~(1 << 5 | 1 << 4);
  49. DDRE &= ~(1 << 6);
  50. PORTC |= (1 << 7);
  51. PORTD |= (1 << 5 | 1 << 4);
  52. PORTE |= (1 << 6);
  53. keyboard_config.raw = eeconfig_read_kb();
  54. ergodox_led_all_set((uint8_t)keyboard_config.led_level * 255 / 4);
  55. #ifdef RGB_MATRIX_ENABLE
  56. if (keyboard_config.rgb_matrix_enable) {
  57. rgb_matrix_set_flags(LED_FLAG_ALL);
  58. } else {
  59. rgb_matrix_set_flags(LED_FLAG_NONE);
  60. }
  61. #endif
  62. ergodox_blink_all_leds();
  63. matrix_init_user();
  64. }
  65. void ergodox_blink_all_leds(void) {
  66. ergodox_led_all_off();
  67. ergodox_led_all_set(LED_BRIGHTNESS_DEFAULT);
  68. ergodox_right_led_1_on();
  69. _delay_ms(50);
  70. ergodox_right_led_2_on();
  71. _delay_ms(50);
  72. ergodox_right_led_3_on();
  73. _delay_ms(50);
  74. #ifdef LEFT_LEDS
  75. ergodox_left_led_1_on();
  76. _delay_ms(50);
  77. if (!mcp23018_status) {
  78. mcp23018_status = ergodox_left_leds_update();
  79. }
  80. ergodox_left_led_2_on();
  81. _delay_ms(50);
  82. if (!mcp23018_status) {
  83. mcp23018_status = ergodox_left_leds_update();
  84. }
  85. ergodox_left_led_3_on();
  86. _delay_ms(50);
  87. if (!mcp23018_status) {
  88. mcp23018_status = ergodox_left_leds_update();
  89. }
  90. #endif
  91. ergodox_right_led_1_off();
  92. _delay_ms(50);
  93. ergodox_right_led_2_off();
  94. _delay_ms(50);
  95. ergodox_right_led_3_off();
  96. #ifdef LEFT_LEDS
  97. _delay_ms(50);
  98. ergodox_left_led_1_off();
  99. if (!mcp23018_status) {
  100. mcp23018_status = ergodox_left_leds_update();
  101. }
  102. _delay_ms(50);
  103. ergodox_left_led_2_off();
  104. if (!mcp23018_status) {
  105. mcp23018_status = ergodox_left_leds_update();
  106. }
  107. _delay_ms(50);
  108. ergodox_left_led_3_off();
  109. if (!mcp23018_status) {
  110. mcp23018_status = ergodox_left_leds_update();
  111. }
  112. #endif
  113. // ergodox_led_all_on();
  114. //_delay_ms(333);
  115. ergodox_led_all_off();
  116. }
  117. uint8_t init_mcp23018(void) {
  118. mcp23018_status = 0x20;
  119. // I2C subsystem
  120. // uint8_t sreg_prev;
  121. // sreg_prev=SREG;
  122. // cli();
  123. if (i2c_initialized == 0) {
  124. i2c_init(); // on pins D(1,0)
  125. i2c_initialized = true;
  126. _delay_ms(1000);
  127. }
  128. // i2c_init(); // on pins D(1,0)
  129. // _delay_ms(1000);
  130. // set pin direction
  131. // - unused : input : 1
  132. // - input : input : 1
  133. // - driving : output : 0
  134. mcp23018_status = i2c_start(I2C_ADDR_WRITE, ERGODOX_EZ_I2C_TIMEOUT);
  135. if (mcp23018_status) goto out;
  136. mcp23018_status = i2c_write(IODIRA, ERGODOX_EZ_I2C_TIMEOUT);
  137. if (mcp23018_status) goto out;
  138. mcp23018_status = i2c_write(0b00000000, ERGODOX_EZ_I2C_TIMEOUT);
  139. if (mcp23018_status) goto out;
  140. mcp23018_status = i2c_write(0b00111111, ERGODOX_EZ_I2C_TIMEOUT);
  141. if (mcp23018_status) goto out;
  142. i2c_stop();
  143. // set pull-up
  144. // - unused : on : 1
  145. // - input : on : 1
  146. // - driving : off : 0
  147. mcp23018_status = i2c_start(I2C_ADDR_WRITE, ERGODOX_EZ_I2C_TIMEOUT);
  148. if (mcp23018_status) goto out;
  149. mcp23018_status = i2c_write(GPPUA, ERGODOX_EZ_I2C_TIMEOUT);
  150. if (mcp23018_status) goto out;
  151. mcp23018_status = i2c_write(0b00000000, ERGODOX_EZ_I2C_TIMEOUT);
  152. if (mcp23018_status) goto out;
  153. mcp23018_status = i2c_write(0b00111111, ERGODOX_EZ_I2C_TIMEOUT);
  154. if (mcp23018_status) goto out;
  155. out:
  156. i2c_stop();
  157. #ifdef LEFT_LEDS
  158. if (!mcp23018_status) mcp23018_status = ergodox_left_leds_update();
  159. #endif // LEFT_LEDS
  160. // SREG=sreg_prev;
  161. return mcp23018_status;
  162. }
  163. #ifdef LEFT_LEDS
  164. uint8_t ergodox_left_leds_update(void) {
  165. if (mcp23018_status) { // if there was an error
  166. return mcp23018_status;
  167. }
  168. # define LEFT_LED_1_SHIFT 7 // in MCP23018 port B
  169. # define LEFT_LED_2_SHIFT 6 // in MCP23018 port B
  170. # define LEFT_LED_3_SHIFT 7 // in MCP23018 port A
  171. // set logical value (doesn't matter on inputs)
  172. // - unused : hi-Z : 1
  173. // - input : hi-Z : 1
  174. // - driving : hi-Z : 1
  175. mcp23018_status = i2c_start(I2C_ADDR_WRITE, ERGODOX_EZ_I2C_TIMEOUT);
  176. if (mcp23018_status) goto out;
  177. mcp23018_status = i2c_write(OLATA, ERGODOX_EZ_I2C_TIMEOUT);
  178. if (mcp23018_status) goto out;
  179. mcp23018_status = i2c_write(0b11111111 & ~(ergodox_left_led_3 << LEFT_LED_3_SHIFT), ERGODOX_EZ_I2C_TIMEOUT);
  180. if (mcp23018_status) goto out;
  181. mcp23018_status = i2c_write(0b11111111 & ~(ergodox_left_led_2 << LEFT_LED_2_SHIFT) & ~(ergodox_left_led_1 << LEFT_LED_1_SHIFT), ERGODOX_EZ_I2C_TIMEOUT);
  182. if (mcp23018_status) goto out;
  183. out:
  184. i2c_stop();
  185. return mcp23018_status;
  186. }
  187. #endif
  188. #ifdef SWAP_HANDS_ENABLE
  189. __attribute__((weak))
  190. // swap-hands action needs a matrix to define the swap
  191. const keypos_t PROGMEM hand_swap_config[MATRIX_ROWS][MATRIX_COLS] = {
  192. /* Left hand, matrix positions */
  193. {{0, 13}, {1, 13}, {2, 13}, {3, 13}, {4, 13}, {5, 13}},
  194. {{0, 12}, {1, 12}, {2, 12}, {3, 12}, {4, 12}, {5, 12}},
  195. {{0, 11}, {1, 11}, {2, 11}, {3, 11}, {4, 11}, {5, 11}},
  196. {{0, 10}, {1, 10}, {2, 10}, {3, 10}, {4, 10}, {5, 10}},
  197. {{0, 9}, {1, 9}, {2, 9}, {3, 9}, {4, 9}, {5, 9}},
  198. {{0, 8}, {1, 8}, {2, 8}, {3, 8}, {4, 8}, {5, 8}},
  199. {{0, 7}, {1, 7}, {2, 7}, {3, 7}, {4, 7}, {5, 7}},
  200. /* Right hand, matrix positions */
  201. {{0, 6}, {1, 6}, {2, 6}, {3, 6}, {4, 6}, {5, 6}},
  202. {{0, 5}, {1, 5}, {2, 5}, {3, 5}, {4, 5}, {5, 5}},
  203. {{0, 4}, {1, 4}, {2, 4}, {3, 4}, {4, 4}, {5, 4}},
  204. {{0, 3}, {1, 3}, {2, 3}, {3, 3}, {4, 3}, {5, 3}},
  205. {{0, 2}, {1, 2}, {2, 2}, {3, 2}, {4, 2}, {5, 2}},
  206. {{0, 1}, {1, 1}, {2, 1}, {3, 1}, {4, 1}, {5, 1}},
  207. {{0, 0}, {1, 0}, {2, 0}, {3, 0}, {4, 0}, {5, 0}},
  208. };
  209. #endif
  210. #ifdef RGB_MATRIX_ENABLE
  211. // clang-format off
  212. const is31_led PROGMEM g_is31_leds[RGB_MATRIX_LED_COUNT] = {
  213. /* driver
  214. * | R location
  215. * | | G location
  216. * | | | B location
  217. * | | | | */
  218. {0, C3_1, C2_1, C4_1}, // LED1 on right
  219. {0, C6_1, C5_1, C7_1}, // LED2
  220. {0, C4_2, C3_2, C5_2}, // LED3
  221. {0, C7_2, C6_2, C8_2}, // LED4
  222. {0, C2_3, C1_3, C3_3}, // LED5
  223. {0, C5_3, C4_3, C6_3}, // LED6
  224. {0, C8_3, C7_3, C9_3}, // LED7
  225. {0, C2_4, C1_4, C3_4}, // LED8
  226. {0, C6_4, C5_4, C7_4}, // LED9
  227. {0, C2_5, C1_5, C3_5}, // LED10
  228. {0, C7_5, C6_5, C8_5}, // LED11
  229. {0, C2_6, C1_6, C3_6}, // LED12
  230. {0, C5_6, C4_6, C6_6}, // LED13
  231. {0, C8_6, C7_6, C9_6}, // LED14
  232. {0, C2_7, C1_7, C3_7}, // LED15
  233. {0, C5_7, C4_7, C6_7}, // LED16
  234. {0, C2_8, C1_8, C3_8}, // LED17
  235. {0, C5_8, C4_8, C6_8}, // LED18
  236. {0, C3_9, C2_9, C4_9}, // LED19
  237. {0, C6_9, C5_9, C7_9}, // LED20
  238. {0, C4_10, C3_10, C5_10}, // LED21
  239. {0, C7_10, C6_10, C8_10}, // LED22
  240. {0, C2_11, C1_11, C3_11}, // LED23
  241. {0, C5_11, C4_11, C6_11}, // LED24
  242. {1, C3_1, C2_1, C4_1}, // LED1 on left
  243. {1, C6_1, C5_1, C7_1}, // LED2
  244. {1, C4_2, C3_2, C5_2}, // LED3
  245. {1, C7_2, C6_2, C8_2}, // LED4
  246. {1, C2_3, C1_3, C3_3}, // LED5
  247. {1, C5_3, C4_3, C6_3}, // LED6
  248. {1, C8_3, C7_3, C9_3}, // LED7
  249. {1, C2_4, C1_4, C3_4}, // LED8
  250. {1, C6_4, C5_4, C7_4}, // LED9
  251. {1, C2_5, C1_5, C3_5}, // LED10
  252. {1, C7_5, C6_5, C8_5}, // LED11
  253. {1, C2_6, C1_6, C3_6}, // LED12
  254. {1, C5_6, C4_6, C6_6}, // LED13
  255. {1, C8_6, C7_6, C9_6}, // LED14
  256. {1, C2_7, C1_7, C3_7}, // LED15
  257. {1, C5_7, C4_7, C6_7}, // LED16
  258. {1, C2_8, C1_8, C3_8}, // LED17
  259. {1, C5_8, C4_8, C6_8}, // LED18
  260. {1, C3_9, C2_9, C4_9}, // LED19
  261. {1, C6_9, C5_9, C7_9}, // LED20
  262. {1, C4_10, C3_10, C5_10}, // LED21
  263. {1, C7_10, C6_10, C8_10}, // LED22
  264. {1, C2_11, C1_11, C3_11}, // LED23
  265. {1, C5_11, C4_11, C6_11} // LED24
  266. };
  267. led_config_t g_led_config = { {
  268. { NO_LED, NO_LED, NO_LED, NO_LED, NO_LED, NO_LED },
  269. { 28, 33, 38, 43, 47, NO_LED },
  270. { 27, 32, 37, 42, 46, NO_LED },
  271. { 26, 31, 36, 41, 45, NO_LED },
  272. { 25, 30, 35, 40, 44, NO_LED },
  273. { 24, 29, 34, 39, NO_LED, NO_LED },
  274. { NO_LED, NO_LED, NO_LED, NO_LED, NO_LED, NO_LED },
  275. { NO_LED, NO_LED, NO_LED, NO_LED, NO_LED, NO_LED },
  276. { 0, 5, 10, 15, NO_LED, NO_LED },
  277. { 1, 6, 11, 16, 20, NO_LED },
  278. { 2, 7, 12, 17, 21, NO_LED },
  279. { 3, 8, 13, 18, 22, NO_LED },
  280. { 4, 9, 14, 19, 23, NO_LED },
  281. { NO_LED, NO_LED, NO_LED, NO_LED, NO_LED, NO_LED }
  282. }, {
  283. { 137, 0 }, { 154, 0 }, { 172, 0 }, { 189, 0 }, { 206, 0 }, { 137, 12 },
  284. { 154, 12 }, { 172, 12 }, { 189, 12 }, { 206, 12 }, { 137, 25 }, { 154, 25 },
  285. { 172, 25 }, { 189, 25 }, { 206, 25 }, { 137, 38 }, { 154, 38 }, { 172, 38 },
  286. { 189, 38 }, { 206, 38 }, { 154, 51 }, { 172, 51 }, { 189, 51 }, { 206, 51 },
  287. { 86, 0 }, { 68, 0 }, { 51, 0 }, { 34, 0 }, { 17, 0 }, { 86, 12 },
  288. { 68, 12 }, { 51, 12 }, { 34, 12 }, { 17, 12 }, { 86, 25 }, { 68, 25 },
  289. { 51, 25 }, { 34, 25 }, { 17, 25 }, { 86, 38 }, { 68, 38 }, { 51, 38 },
  290. { 34, 38 }, { 17, 38 }, { 68, 51 }, { 51, 51 }, { 34, 51 }, { 17, 51 }
  291. }, {
  292. 4, 4, 4, 4, 4, 4,
  293. 4, 4, 4, 4, 4, 4,
  294. 4, 4, 4, 4, 4, 4,
  295. 4, 4, 1, 1, 1, 1,
  296. 4, 4, 4, 4, 4, 4,
  297. 4, 4, 4, 4, 4, 4,
  298. 4, 4, 4, 4, 4, 4,
  299. 4, 4, 1, 1, 1, 1
  300. } };
  301. // clang-format on
  302. # ifdef ORYX_CONFIGURATOR
  303. void keyboard_post_init_kb(void) {
  304. rgb_matrix_enable_noeeprom();
  305. keyboard_post_init_user();
  306. }
  307. # endif
  308. #endif
  309. #ifdef ORYX_CONFIGURATOR
  310. bool process_record_kb(uint16_t keycode, keyrecord_t *record) {
  311. switch (keycode) {
  312. case LED_LEVEL:
  313. if (record->event.pressed) {
  314. keyboard_config.led_level++;
  315. if (keyboard_config.led_level > 4) {
  316. keyboard_config.led_level = 0;
  317. }
  318. ergodox_led_all_set((uint8_t)keyboard_config.led_level * 255 / 4);
  319. eeconfig_update_kb(keyboard_config.raw);
  320. layer_state_set_kb(layer_state);
  321. }
  322. break;
  323. # ifdef RGB_MATRIX_ENABLE
  324. case TOGGLE_LAYER_COLOR:
  325. if (record->event.pressed) {
  326. keyboard_config.disable_layer_led ^= 1;
  327. if (keyboard_config.disable_layer_led) rgb_matrix_set_color_all(0, 0, 0);
  328. eeconfig_update_kb(keyboard_config.raw);
  329. }
  330. break;
  331. case RGB_TOG:
  332. if (record->event.pressed) {
  333. switch (rgb_matrix_get_flags()) {
  334. case LED_FLAG_ALL: {
  335. rgb_matrix_set_flags(LED_FLAG_NONE);
  336. keyboard_config.rgb_matrix_enable = false;
  337. rgb_matrix_set_color_all(0, 0, 0);
  338. } break;
  339. default: {
  340. rgb_matrix_set_flags(LED_FLAG_ALL);
  341. keyboard_config.rgb_matrix_enable = true;
  342. } break;
  343. }
  344. eeconfig_update_kb(keyboard_config.raw);
  345. }
  346. return false;
  347. # endif
  348. }
  349. return process_record_user(keycode, record);
  350. }
  351. #endif
  352. void eeconfig_init_kb(void) { // EEPROM is getting reset!
  353. keyboard_config.raw = 0;
  354. keyboard_config.led_level = 4;
  355. keyboard_config.rgb_matrix_enable = true;
  356. eeconfig_update_kb(keyboard_config.raw);
  357. eeconfig_init_user();
  358. }
  359. #ifdef ORYX_ENABLE
  360. static uint16_t loops = 0;
  361. static bool is_on = false;
  362. #endif
  363. #ifdef DYNAMIC_MACRO_ENABLE
  364. static bool is_dynamic_recording = false;
  365. static uint16_t dynamic_loop_timer;
  366. void dynamic_macro_record_start_user(void) {
  367. is_dynamic_recording = true;
  368. dynamic_loop_timer = timer_read();
  369. ergodox_right_led_1_on();
  370. }
  371. void dynamic_macro_record_end_user(int8_t direction) {
  372. is_dynamic_recording = false;
  373. layer_state_set_user(layer_state);
  374. }
  375. #endif
  376. void matrix_scan_kb(void) {
  377. #ifdef DYNAMIC_MACRO_ENABLE
  378. if (is_dynamic_recording) {
  379. ergodox_right_led_1_off();
  380. // if (timer_elapsed(dynamic_loop_timer) > 5)
  381. {
  382. static uint8_t counter;
  383. counter++;
  384. if (counter > 100) ergodox_right_led_1_on();
  385. dynamic_loop_timer = timer_read();
  386. }
  387. }
  388. #endif
  389. #ifdef CAPS_LOCK_STATUS
  390. led_t led_state = host_keyboard_led_state();
  391. if(led_state.caps_lock) {
  392. ergodox_right_led_3_on();
  393. }
  394. else {
  395. uint8_t layer = get_highest_layer(layer_state);
  396. if(layer != 1) {
  397. ergodox_right_led_3_off();
  398. }
  399. }
  400. #endif
  401. matrix_scan_user();
  402. }