main.c 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227
  1. /*
  2. * (c) 2015 flabberast <s3+flabbergast@sdfeu.org>
  3. *
  4. * Based on the following work:
  5. * - Guillaume Duc's raw hid example (MIT License)
  6. * https://github.com/guiduc/usb-hid-chibios-example
  7. * - PJRC Teensy examples (MIT License)
  8. * https://www.pjrc.com/teensy/usb_keyboard.html
  9. * - hasu's TMK keyboard code (GPL v2 and some code Modified BSD)
  10. * https://github.com/tmk/tmk_keyboard/
  11. * - ChibiOS demo code (Apache 2.0 License)
  12. * http://www.chibios.org
  13. *
  14. * Since some GPL'd code is used, this work is licensed under
  15. * GPL v2 or later.
  16. */
  17. #include "ch.h"
  18. #include "hal.h"
  19. #include "usb_main.h"
  20. /* TMK includes */
  21. #include "report.h"
  22. #include "host.h"
  23. #include "host_driver.h"
  24. #include "keyboard.h"
  25. #include "action.h"
  26. #include "action_util.h"
  27. #include "mousekey.h"
  28. #include "led.h"
  29. #include "sendchar.h"
  30. #include "debug.h"
  31. #include "printf.h"
  32. #include "rgblight_reconfig.h"
  33. #if (defined(RGB_MIDI) || defined(RGBLIGHT_ANIMATIONS)) && defined(RGBLIGHT_ENABLE)
  34. # include "rgblight.h"
  35. #endif
  36. #ifdef SLEEP_LED_ENABLE
  37. # include "sleep_led.h"
  38. #endif
  39. #ifdef SERIAL_LINK_ENABLE
  40. # include "serial_link/system/serial_link.h"
  41. #endif
  42. #ifdef VISUALIZER_ENABLE
  43. # include "visualizer/visualizer.h"
  44. #endif
  45. #ifdef MIDI_ENABLE
  46. # include "qmk_midi.h"
  47. #endif
  48. #ifdef STM32_EEPROM_ENABLE
  49. # include "eeprom_stm32.h"
  50. #endif
  51. #include "suspend.h"
  52. #include "wait.h"
  53. /* -------------------------
  54. * TMK host driver defs
  55. * -------------------------
  56. */
  57. /* declarations */
  58. uint8_t keyboard_leds(void);
  59. void send_keyboard(report_keyboard_t *report);
  60. void send_mouse(report_mouse_t *report);
  61. void send_system(uint16_t data);
  62. void send_consumer(uint16_t data);
  63. /* host struct */
  64. host_driver_t chibios_driver = {keyboard_leds, send_keyboard, send_mouse, send_system, send_consumer};
  65. #ifdef VIRTSER_ENABLE
  66. void virtser_task(void);
  67. #endif
  68. #ifdef RAW_ENABLE
  69. void raw_hid_task(void);
  70. #endif
  71. #ifdef CONSOLE_ENABLE
  72. void console_task(void);
  73. #endif
  74. /* TESTING
  75. * Amber LED blinker thread, times are in milliseconds.
  76. */
  77. /* set this variable to non-zero anywhere to blink once */
  78. // static THD_WORKING_AREA(waThread1, 128);
  79. // static THD_FUNCTION(Thread1, arg) {
  80. // (void)arg;
  81. // chRegSetThreadName("blinker");
  82. // while (true) {
  83. // systime_t time;
  84. // time = USB_DRIVER.state == USB_ACTIVE ? 250 : 500;
  85. // palClearLine(LINE_CAPS_LOCK);
  86. // chSysPolledDelayX(MS2RTC(STM32_HCLK, time));
  87. // palSetLine(LINE_CAPS_LOCK);
  88. // chSysPolledDelayX(MS2RTC(STM32_HCLK, time));
  89. // }
  90. // }
  91. /* Main thread
  92. */
  93. int main(void) {
  94. /* ChibiOS/RT init */
  95. halInit();
  96. chSysInit();
  97. #ifdef STM32_EEPROM_ENABLE
  98. EEPROM_Init();
  99. #endif
  100. // TESTING
  101. // chThdCreateStatic(waThread1, sizeof(waThread1), NORMALPRIO, Thread1, NULL);
  102. keyboard_setup();
  103. /* Init USB */
  104. init_usb_driver(&USB_DRIVER);
  105. /* init printf */
  106. init_printf(NULL, sendchar_pf);
  107. #ifdef MIDI_ENABLE
  108. setup_midi();
  109. #endif
  110. #ifdef SERIAL_LINK_ENABLE
  111. init_serial_link();
  112. #endif
  113. #ifdef VISUALIZER_ENABLE
  114. visualizer_init();
  115. #endif
  116. host_driver_t *driver = NULL;
  117. /* Wait until the USB or serial link is active */
  118. while (true) {
  119. #if defined(WAIT_FOR_USB) || defined(SERIAL_LINK_ENABLE)
  120. if (USB_DRIVER.state == USB_ACTIVE) {
  121. driver = &chibios_driver;
  122. break;
  123. }
  124. #else
  125. driver = &chibios_driver;
  126. break;
  127. #endif
  128. #ifdef SERIAL_LINK_ENABLE
  129. if (is_serial_link_connected()) {
  130. driver = get_serial_link_driver();
  131. break;
  132. }
  133. serial_link_update();
  134. #endif
  135. wait_ms(50);
  136. }
  137. /* Do need to wait here!
  138. * Otherwise the next print might start a transfer on console EP
  139. * before the USB is completely ready, which sometimes causes
  140. * HardFaults.
  141. */
  142. wait_ms(50);
  143. print("USB configured.\n");
  144. /* init TMK modules */
  145. keyboard_init();
  146. host_set_driver(driver);
  147. #ifdef SLEEP_LED_ENABLE
  148. sleep_led_init();
  149. #endif
  150. print("Keyboard start.\n");
  151. /* Main loop */
  152. while (true) {
  153. #if !defined(NO_USB_STARTUP_CHECK)
  154. if (USB_DRIVER.state == USB_SUSPENDED) {
  155. print("[s]");
  156. # ifdef VISUALIZER_ENABLE
  157. visualizer_suspend();
  158. # endif
  159. while (USB_DRIVER.state == USB_SUSPENDED) {
  160. /* Do this in the suspended state */
  161. # ifdef SERIAL_LINK_ENABLE
  162. serial_link_update();
  163. # endif
  164. suspend_power_down(); // on AVR this deep sleeps for 15ms
  165. /* Remote wakeup */
  166. if (suspend_wakeup_condition()) {
  167. usbWakeupHost(&USB_DRIVER);
  168. }
  169. }
  170. /* Woken up */
  171. // variables has been already cleared by the wakeup hook
  172. send_keyboard_report();
  173. # ifdef MOUSEKEY_ENABLE
  174. mousekey_send();
  175. # endif /* MOUSEKEY_ENABLE */
  176. # ifdef VISUALIZER_ENABLE
  177. visualizer_resume();
  178. # endif
  179. }
  180. #endif
  181. keyboard_task();
  182. #ifdef CONSOLE_ENABLE
  183. console_task();
  184. #endif
  185. #ifdef VIRTSER_ENABLE
  186. virtser_task();
  187. #endif
  188. #ifdef RAW_ENABLE
  189. raw_hid_task();
  190. #endif
  191. #if defined(RGBLIGHT_ANIMATIONS) && defined(RGBLIGHT_ENABLE)
  192. rgblight_task();
  193. #endif
  194. }
  195. }