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 "usb_keycodes.h"
  21 #include "host.h"
  22 #include "print.h"
  23 #include "debug.h"
  24 #include "util.h"
  25 #include "timer.h"
  26 #include "layer.h"
  27 #include "matrix.h"
  28 #include "bootloader.h"
  29 #include "command.h"
  30
  31 #ifdef HOST_PJRC
  32 #   include "usb_keyboard.h"
  33 #   ifdef EXTRAKEY_ENABLE
  34 #       include "usb_extra.h"
  35 #   endif
  36 #endif
  37
  38 #ifdef HOST_VUSB
  39 #   include "usbdrv.h"
  40 #endif
  41
  42
  43 static uint8_t command_common(void);
  44 static void help(void);
  45 static void switch_layer(uint8_t layer);
  46
  47 static bool last_print_enable;
  48
  49 uint8_t command_proc(void)
  50 {
  51     uint8_t processed = 0;
  52     last_print_enable = print_enable;
  53
  54     if (!IS_COMMAND())
  55         return 0;
  56
  57     print_enable = true;
  58     if (command_extra() || command_common()) {
  59         processed = 1;
  60         _delay_ms(500);
  61     }
  62     print_enable = last_print_enable;
  63     return processed;
  64 }
  65
  66 /* This allows to define extra commands. return 0 when not processed. */
  67 uint8_t command_extra(void) __attribute__ ((weak));
  68 uint8_t command_extra(void)
  69 {
  70     return 0;
  71 }
  72
  73
  74 static uint8_t command_common(void)
  75 {
  76     switch (host_get_first_key()) {
  77         case KB_H:
  78             help();
  79             break;
  80         case KB_B:
  81             host_clear_keyboard_report();
  82             host_send_keyboard_report();
  83             print("jump to bootloader... ");
  84             _delay_ms(1000);
  85             bootloader_jump(); // not return
  86             print("not supported.\n");
  87             break;
  88         case KB_D:
  89             debug_enable = !debug_enable;
  90             if (debug_enable) {
  91                 last_print_enable = true;
  92                 print("debug enabled.\n");
  93                 debug_matrix = true;
  94                 debug_keyboard = true;
  95                 debug_mouse = true;
  96             } else {
  97                 print("debug disabled.\n");
  98                 last_print_enable = false;
  99                 debug_matrix = false;
 100                 debug_keyboard = false;
 101                 debug_mouse = false;
 102             }
 103             break;
 104         case KB_X: // debug matrix toggle
 105             debug_matrix = !debug_matrix;
 106             if (debug_matrix)
 107                 print("debug matrix enabled.\n");
 108             else
 109                 print("debug matrix disabled.\n");
 110             break;
 111         case KB_K: // debug keyboard toggle
 112             debug_keyboard = !debug_keyboard;
 113             if (debug_keyboard)
 114                 print("debug keyboard enabled.\n");
 115             else
 116                 print("debug keyboard disabled.\n");
 117             break;
 118         case KB_M: // debug mouse toggle
 119             debug_mouse = !debug_mouse;
 120             if (debug_mouse)
 121                 print("debug mouse enabled.\n");
 122             else
 123                 print("debug mouse disabled.\n");
 124             break;
 125         case KB_V: // print version & information
 126             print(STR(DESCRIPTION) "\n");
 127             break;
 128         case KB_T: // print timer
 129             print("timer: "); phex16(timer_count); print("\n");
 130             break;
 131         case KB_P: // print toggle
 132             if (last_print_enable) {
 133                 print("print disabled.\n");
 134                 last_print_enable = false;
 135             } else {
 136                 last_print_enable = true;
 137                 print("print enabled.\n");
 138             }
 139             break;
 140         case KB_S:
 141             print("host_keyboard_leds:"); phex(host_keyboard_leds()); print("\n");
 142 #ifdef HOST_PJRC
 143             print("UDCON: "); phex(UDCON); print("\n");
 144             print("UDIEN: "); phex(UDIEN); print("\n");
 145             print("UDINT: "); phex(UDINT); print("\n");
 146             print("usb_keyboard_leds:"); phex(usb_keyboard_leds); print("\n");
 147             print("usb_keyboard_protocol: "); phex(usb_keyboard_protocol); print("\n");
 148             print("usb_keyboard_idle_config:"); phex(usb_keyboard_idle_config); print("\n");
 149             print("usb_keyboard_idle_count:"); phex(usb_keyboard_idle_count); print("\n");
 150 #endif
 151
 152 #ifdef HOST_VUSB
 153 #   if USB_COUNT_SOF
 154             print("usbSofCount: "); phex(usbSofCount); print("\n");
 155 #   endif
 156 #endif
 157             break;
 158 #ifdef NKRO_ENABLE
 159         case KB_N:
 160             // send empty report before change
 161             host_clear_keyboard_report();
 162             host_send_keyboard_report();
 163             keyboard_nkro = !keyboard_nkro;
 164             if (keyboard_nkro)
 165                 print("NKRO: enabled\n");
 166             else
 167                 print("NKRO: disabled\n");
 168             break;
 169 #endif
 170 #ifdef EXTRAKEY_ENABLE
 171         case KB_ESC:
 172             host_clear_keyboard_report();
 173             host_send_keyboard_report();
 174 #ifdef HOST_PJRC
 175             if (suspend && remote_wakeup) {
 176                 usb_remote_wakeup();
 177             } else {
 178                 host_system_send(SYSTEM_POWER_DOWN);
 179                 host_system_send(0);
 180                 _delay_ms(500);
 181             }
 182 #else
 183             host_system_send(SYSTEM_POWER_DOWN);
 184             host_system_send(0);
 185             _delay_ms(500);
 186 #endif
 187             break;
 188 #endif
 189         case KB_BSPC:
 190             matrix_init();
 191             print("clear matrix\n");
 192             break;
 193         case KB_0:
 194             switch_layer(0);
 195             break;
 196         case KB_1:
 197             switch_layer(1);
 198             break;
 199         case KB_2:
 200             switch_layer(2);
 201             break;
 202         case KB_3:
 203             switch_layer(3);
 204             break;
 205         case KB_4:
 206             switch_layer(4);
 207             break;
 208         default:
 209             return 0;
 210     }
 211     return 1;
 212 }
 213
 214 static void help(void)
 215 {
 216     print("b: jump to bootloader\n");
 217     print("d: toggle debug enable\n");
 218     print("x: toggle matrix debug\n");
 219     print("k: toggle keyboard debug\n");
 220     print("m: toggle mouse debug\n");
 221     print("p: toggle print enable\n");
 222     print("v: print version\n");
 223     print("t: print timer count\n");
 224     print("s: print status\n");
 225 #ifdef NKRO_ENABLE
 226     print("n: toggle NKRO\n");
 227 #endif
 228     print("Backspace: clear matrix\n");
 229     print("ESC: power down/wake up\n");
 230     print("0: switch to Layer0 \n");
 231     print("1: switch to Layer1 \n");
 232     print("2: switch to Layer2 \n");
 233     print("3: switch to Layer3 \n");
 234     print("4: switch to Layer4 \n");
 235 }
 236
 237 static void switch_layer(uint8_t layer)
 238 {
 239     print("current_layer: "); phex(current_layer); print("\n");
 240     print("default_layer: "); phex(default_layer); print("\n");
 241     current_layer = layer;
 242     default_layer = layer;
 243     print("switch to Layer: "); phex(layer); print("\n");
 244 }