123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329 |
- #include <stdbool.h>
- #include <avr/io.h>
- #include <avr/interrupt.h>
- #include <util/delay.h>
- #include "print.h"
- #include "debug.h"
- #include "timer.h"
- #include "util.h"
- #include "jump_bootloader.h"
- #include "usb_keyboard.h"
- #include "usb_keycodes.h"
- #include "usb.h"
- #include "layer.h"
- #include "matrix_skel.h"
- #include "keymap_skel.h"
- #include "key_process.h"
- #ifdef MOUSEKEY_ENABLE
- # include "mousekey.h"
- #endif
- #ifdef PS2_MOUSE_ENABLE
- # include "ps2_mouse.h"
- #endif
- #ifdef USB_EXTRA_ENABLE
- # include "usb_extra.h"
- #endif
- #ifdef USB_MOUSE_ENABLE
- # include "usb_mouse.h"
- #endif
-
-
- // TODO: refactoring
- void proc_matrix(void) {
- bool modified = false;
- uint8_t fn_bits = 0;
-
- matrix_scan();
- modified = matrix_is_modified();
-
- if (modified) {
- if (debug_matrix) matrix_print();
- #ifdef DEBUG_LED
- // LED flash for debug
- DEBUG_LED_CONFIG;
- DEBUG_LED_ON;
- #endif
- }
-
- if (matrix_has_ghost()) {
- // should send error?
- debug("matrix has ghost!!\n");
- return;
- }
-
- usb_keyboard_swap_report();
- usb_keyboard_clear_report();
- for (int row = 0; row < matrix_rows(); row++) {
- for (int col = 0; col < matrix_cols(); col++) {
- if (!matrix_is_on(row, col)) continue;
-
- // TODO: clean code
- uint8_t code = layer_get_keycode(row, col);
- if (code == KB_NO) {
- // do nothing
- } else if (IS_MOD(code)) {
- usb_keyboard_add_mod(code);
- } else if (IS_FN(code)) {
- fn_bits |= FN_BIT(code);
- } else if (IS_MOUSE(code)) {
- #ifdef MOUSEKEY_ENABLE
- mousekey_decode(code);
- #endif
- }
-
- #ifdef USB_EXTRA_ENABLE
- // audio control & system control
- else if (code == KB_MUTE) {
- usb_extra_audio_send(AUDIO_MUTE);
- usb_extra_audio_send(0);
- _delay_ms(500);
- } else if (code == KB_VOLU) {
- usb_extra_audio_send(AUDIO_VOL_UP);
- usb_extra_audio_send(0);
- _delay_ms(100);
- } else if (code == KB_VOLD) {
- usb_extra_audio_send(AUDIO_VOL_DOWN);
- usb_extra_audio_send(0);
- _delay_ms(100);
- } else if (code == KB_PWR) {
- if (suspend && remote_wakeup) {
- usb_remote_wakeup();
- } else {
- usb_extra_system_send(SYSTEM_POWER_DOWN);
- }
- _delay_ms(1000);
- }
- #endif
-
- // normal keys
- else {
- usb_keyboard_add_key(code);
- }
- }
- }
-
- if (modified) {
- #ifdef DEBUG_LED
- // LED flash for debug
- DEBUG_LED_CONFIG;
- DEBUG_LED_OFF;
- #endif
- }
-
- layer_switching(fn_bits);
-
- // TODO: clean code
- // special mode for control, develop and debug
- if (keymap_is_special_mode(fn_bits)) {
- switch (usb_keyboard_get_key()) {
- case KB_H: // help
- print_enable = true;
- print("b: jump to bootloader\n");
- print("d: toggle debug enable\n");
- print("x: toggle matrix debug\n");
- print("k: toggle keyboard debug\n");
- print("m: toggle mouse debug\n");
- print("p: toggle print enable\n");
- print("v: print version\n");
- print("t: print timer count\n");
- print("s: print status\n");
- print("`: toggle protcol(boot/report)\n");
- #ifdef USB_NKRO_ENABLE
- print("n: toggle USB_NKRO\n");
- #endif
- print("ESC: power down/wake up\n");
- #ifdef PS2_MOUSE_ENABLE
- print("1: ps2_mouse_init \n");
- print("2: ps2_mouse_read \n");
- print("3: ps2_mouse: on/off toggle \n");
- #endif
- _delay_ms(500);
- print_enable = false;
- break;
- #ifdef PS2_MOUSE_ENABLE
- case KB_1:
- usb_keyboard_clear_report();
- usb_keyboard_send();
- print_enable = true;
- print("ps2_mouse_init...\n");
- _delay_ms(500);
- ps2_mouse_init();
- break;
- case KB_2:
- usb_keyboard_clear_report();
- usb_keyboard_send();
- print_enable = true;
- print("ps2_mouse_read[btn x y]: ");
- _delay_ms(100);
- ps2_mouse_read();
- phex(ps2_mouse_btn); print(" ");
- phex(ps2_mouse_x); print(" ");
- phex(ps2_mouse_y); print("\n");
- print("ps2_mouse_error_count: "); phex(ps2_mouse_error_count); print("\n");
- break;
- case KB_3:
- ps2_mouse_enable = !ps2_mouse_enable;
- print("ps2_mouse: ");
- if (ps2_mouse_enable)
- print("on");
- else
- print("off");
- print("\n");
- _delay_ms(500);
- break;
- #endif
- case KB_B: // bootloader
- usb_keyboard_clear_report();
- usb_keyboard_send();
- print_enable = true;
- print("jump to bootloader...\n");
- _delay_ms(1000);
- jump_bootloader(); // not return
- break;
- case KB_D: // debug all toggle
- usb_keyboard_clear_report();
- usb_keyboard_send();
- debug_enable = !debug_enable;
- if (debug_enable) {
- print_enable = true;
- print("debug enabled.\n");
- debug_matrix = true;
- debug_keyboard = true;
- debug_mouse = true;
- } else {
- print("debug disabled.\n");
- print_enable = false;
- debug_matrix = false;
- debug_keyboard = false;
- debug_mouse = false;
- }
- _delay_ms(1000);
- break;
- case KB_X: // debug matrix toggle
- usb_keyboard_clear_report();
- usb_keyboard_send();
- debug_matrix = !debug_matrix;
- if (debug_matrix)
- print("debug matrix enabled.\n");
- else
- print("debug matrix disabled.\n");
- _delay_ms(1000);
- break;
- case KB_K: // debug keyboard toggle
- usb_keyboard_clear_report();
- usb_keyboard_send();
- debug_keyboard = !debug_keyboard;
- if (debug_keyboard)
- print("debug keyboard enabled.\n");
- else
- print("debug keyboard disabled.\n");
- _delay_ms(1000);
- break;
- case KB_M: // debug mouse toggle
- usb_keyboard_clear_report();
- usb_keyboard_send();
- debug_mouse = !debug_mouse;
- if (debug_mouse)
- print("debug mouse enabled.\n");
- else
- print("debug mouse disabled.\n");
- _delay_ms(1000);
- break;
- case KB_V: // print version & information
- usb_keyboard_clear_report();
- usb_keyboard_send();
- print_enable = true;
- print(STR(DESCRIPTION) "\n");
- _delay_ms(1000);
- break;
- case KB_T: // print timer
- usb_keyboard_clear_report();
- usb_keyboard_send();
- print_enable = true;
- print("timer: "); phex16(timer_count); print("\n");
- _delay_ms(500);
- break;
- case KB_P: // print toggle
- usb_keyboard_clear_report();
- usb_keyboard_send();
- if (print_enable) {
- print("print disabled.\n");
- print_enable = false;
- } else {
- print_enable = true;
- print("print enabled.\n");
- }
- _delay_ms(1000);
- break;
- case KB_S:
- usb_keyboard_clear_report();
- usb_keyboard_send();
- print("UDCON: "); phex(UDCON); print("\n");
- print("UDIEN: "); phex(UDIEN); print("\n");
- print("UDINT: "); phex(UDINT); print("\n");
- print("usb_keyboard_leds:"); phex(usb_keyboard_leds); print("\n");
- print("usb_keyboard_protocol:"); phex(usb_keyboard_protocol); print("\n");
- print("usb_keyboard_idle_config:"); phex(usb_keyboard_idle_config); print("\n");
- print("usb_keyboard_idle_count:"); phex(usb_keyboard_idle_count); print("\n");
- #ifdef USB_MOUSE_ENABLE
- print("usb_mouse_protocol:"); phex(usb_mouse_protocol); print("\n");
- #endif
- if (usb_keyboard_nkro) print("USB_NKRO: enabled\n"); else print("USB_NKRO: disabled\n");
- _delay_ms(500);
- break;
- case KB_GRV:
- usb_keyboard_clear_report();
- usb_keyboard_send();
- usb_keyboard_protocol = !usb_keyboard_protocol;
- print("keyboard protcol: ");
- if (usb_keyboard_protocol) print("report"); else print("boot");
- print("\n");
-
- #ifdef USB_MOUSE_ENABLE
- usb_mouse_protocol = !usb_mouse_protocol;
- print("mouse protcol: ");
- if (usb_mouse_protocol) print("report"); else print("boot");
- print("\n");
- #endif
- _delay_ms(1000);
- break;
- #ifdef USB_NKRO_ENABLE
- case KB_N:
- usb_keyboard_clear_report();
- usb_keyboard_send();
- usb_keyboard_nkro = !usb_keyboard_nkro;
- if (usb_keyboard_nkro) print("USB_NKRO: enabled\n"); else print("USB_NKRO: disabled\n");
- _delay_ms(1000);
- break;
- #endif
- #ifdef USB_EXTRA_ENABLE
- case KB_ESC:
- usb_keyboard_clear_report();
- usb_keyboard_send();
- if (suspend && remote_wakeup) {
- usb_remote_wakeup();
- } else {
- usb_extra_system_send(SYSTEM_POWER_DOWN);
- }
- _delay_ms(1000);
- break;
- #endif
- }
- }
-
-
- if (modified) {
- usb_keyboard_send();
- }
-
- #ifdef MOUSEKEY_ENABLE
- // mouse keys
- mousekey_usb_send();
- #endif
-
- #ifdef PS2_MOUSE_ENABLE
- // ps2 mouse
- if (ps2_mouse_read() == 0)
- ps2_mouse_usb_send();
- #endif
- }
|