common/command.c

   1 /*
   2 Copyright 2011 Jun Wako <wakojun@gmail.com>
   3
   4 This program is free software: you can redistribute it and/or modify
   5 it under the terms of the GNU General Public License as published by
   6 the Free Software Foundation, either version 2 of the License, or
   7 (at your option) any later version.
   8
   9 This program is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 GNU General Public License for more details.
  13
  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 <stdint.h>
  18 #include <stdbool.h>
  19 #include <util/delay.h>
  20 #include "keycode.h"
  21 #include "host.h"
  22 #include "print.h"
  23 #include "debug.h"
  24 #include "util.h"
  25 #include "timer.h"
  26 #include "keyboard.h"
  27 #include "bootloader.h"
  28 #include "command.h"
  29 #ifdef MOUSEKEY_ENABLE
  30 #include "mousekey.h"
  31 #endif
  32
  33 #ifdef HOST_PJRC
  34 #   include "usb_keyboard.h"
  35 #   ifdef EXTRAKEY_ENABLE
  36 #       include "usb_extra.h"
  37 #   endif
  38 #endif
  39
  40 #ifdef HOST_VUSB
  41 #   include "usbdrv.h"
  42 #endif
  43
  44
  45 static bool command_common(uint8_t code);
  46 static void help(void);
  47 static void switch_layer(uint8_t layer);
  48 static void clear_keyboard(void);
  49
  50 static bool last_print_enable;
  51
  52
  53 bool command_proc(uint8_t code)
  54 {
  55     if (!IS_COMMAND())
  56         return false;
  57
  58     last_print_enable = print_enable;
  59     print_enable = true;
  60     if (command_extra(code) || command_common(code)) {
  61         _delay_ms(500);
  62         return true;
  63     }
  64     print_enable = last_print_enable;
  65     return false;
  66 }
  67
  68 /* This allows to define extra commands. return 0 when not processed. */
  69 bool command_extra(uint8_t code) __attribute__ ((weak));
  70 bool command_extra(uint8_t code)
  71 {
  72     return false;
  73 }
  74
  75
  76 static bool command_common(uint8_t code)
  77 {
  78     switch (code) {
  79         case KC_H:
  80             help();
  81             break;
  82         case KC_DEL:
  83             clear_keyboard();
  84             print("jump to bootloader... ");
  85             _delay_ms(1000);
  86             bootloader_jump(); // not return
  87             print("not supported.\n");
  88             break;
  89         case KC_D:
  90             debug_enable = !debug_enable;
  91             if (debug_enable) {
  92                 last_print_enable = true;
  93                 print("debug enabled.\n");
  94                 debug_matrix = true;
  95                 debug_keyboard = true;
  96                 debug_mouse = true;
  97             } else {
  98                 print("debug disabled.\n");
  99                 last_print_enable = false;
 100                 debug_matrix = false;
 101                 debug_keyboard = false;
 102                 debug_mouse = false;
 103             }
 104             break;
 105         case KC_X: // debug matrix toggle
 106             debug_matrix = !debug_matrix;
 107             if (debug_matrix)
 108                 print("debug matrix enabled.\n");
 109             else
 110                 print("debug matrix disabled.\n");
 111             break;
 112         case KC_K: // debug keyboard toggle
 113             debug_keyboard = !debug_keyboard;
 114             if (debug_keyboard)
 115                 print("debug keyboard enabled.\n");
 116             else
 117                 print("debug keyboard disabled.\n");
 118             break;
 119         case KC_M: // debug mouse toggle
 120             debug_mouse = !debug_mouse;
 121             if (debug_mouse)
 122                 print("debug mouse enabled.\n");
 123             else
 124                 print("debug mouse disabled.\n");
 125             break;
 126         case KC_V: // print version & information
 127             print(STR(DESCRIPTION) "\n");
 128             break;
 129         case KC_T: // print timer
 130             print("timer: "); phex16(timer_count); print("\n");
 131             break;
 132         case KC_P: // print toggle
 133             if (last_print_enable) {
 134                 print("print disabled.\n");
 135                 last_print_enable = false;
 136             } else {
 137                 last_print_enable = true;
 138                 print("print enabled.\n");
 139             }
 140             break;
 141         case KC_S:
 142             print("host_keyboard_leds:"); phex(host_keyboard_leds()); print("\n");
 143 #ifdef HOST_PJRC
 144             print("UDCON: "); phex(UDCON); print("\n");
 145             print("UDIEN: "); phex(UDIEN); print("\n");
 146             print("UDINT: "); phex(UDINT); print("\n");
 147             print("usb_keyboard_leds:"); phex(usb_keyboard_leds); print("\n");
 148             print("usb_keyboard_protocol: "); phex(usb_keyboard_protocol); print("\n");
 149             print("usb_keyboard_idle_config:"); phex(usb_keyboard_idle_config); print("\n");
 150             print("usb_keyboard_idle_count:"); phex(usb_keyboard_idle_count); print("\n");
 151 #endif
 152
 153 #ifdef HOST_VUSB
 154 #   if USB_COUNT_SOF
 155             print("usbSofCount: "); phex(usbSofCount); print("\n");
 156 #   endif
 157 #endif
 158             break;
 159 #ifdef NKRO_ENABLE
 160         case KC_N:
 161             keyboard_nkro = !keyboard_nkro;
 162             if (keyboard_nkro)
 163                 print("NKRO: enabled\n");
 164             else
 165                 print("NKRO: disabled\n");
 166             break;
 167 #endif
 168 #ifdef EXTRAKEY_ENABLE
 169         case KC_ESC:
 170 #ifdef HOST_PJRC
 171             if (suspend && remote_wakeup) {
 172                 usb_remote_wakeup();
 173             } else {
 174                 host_system_send(SYSTEM_POWER_DOWN);
 175                 host_system_send(0);
 176                 _delay_ms(500);
 177             }
 178 #else
 179             host_system_send(SYSTEM_POWER_DOWN);
 180             host_system_send(0);
 181             _delay_ms(500);
 182 #endif
 183             break;
 184 #endif
 185         case KC_0:
 186         case KC_F10:
 187             switch_layer(0);
 188             break;
 189         case KC_1:
 190         case KC_F1:
 191             switch_layer(1);
 192             break;
 193         case KC_2:
 194         case KC_F2:
 195             switch_layer(2);
 196             break;
 197         case KC_3:
 198         case KC_F3:
 199             switch_layer(3);
 200             break;
 201         case KC_4:
 202         case KC_F4:
 203             switch_layer(4);
 204             break;
 205         default:
 206             return false;
 207     }
 208     return true;
 209 }
 210
 211 static void help(void)
 212 {
 213     print("d: toggle debug enable\n");
 214     print("x: toggle matrix debug\n");
 215     print("k: toggle keyboard debug\n");
 216     print("m: toggle mouse debug\n");
 217     print("p: toggle print enable\n");
 218     print("v: print version\n");
 219     print("t: print timer count\n");
 220     print("s: print status\n");
 221     print("ESC: power down/wake up\n");
 222     print("0/F10: switch to Layer0 \n");
 223     print("1/F1: switch to Layer1 \n");
 224     print("2/F2: switch to Layer2 \n");
 225     print("3/F3: switch to Layer3 \n");
 226     print("4/F4: switch to Layer4 \n");
 227 #ifdef NKRO_ENABLE
 228     print("n: toggle NKRO\n");
 229 #endif
 230     print("DEL: jump to bootloader\n");
 231 }
 232
 233 static void switch_layer(uint8_t layer)
 234 {
 235     print("current_layer: "); phex(current_layer); print("\n");
 236     print("default_layer: "); phex(default_layer); print("\n");
 237     current_layer = layer;
 238     default_layer = layer;
 239     print("switch to Layer: "); phex(layer); print("\n");
 240 }
 241
 242 static void clear_keyboard(void)
 243 {
 244     host_clear_keys();
 245     host_send_keyboard_report();
 246
 247     host_system_send(0);
 248     host_consumer_send(0);
 249
 250 #ifdef MOUSEKEY_ENABLE
 251     mousekey_clear();
 252     mousekey_send();
 253 #endif
 254 }