244 lines
		
	
	
		
			6.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			244 lines
		
	
	
		
			6.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
Copyright 2011 Jun Wako <wakojun@gmail.com>
 | 
						|
 | 
						|
This program is free software: you can redistribute it and/or modify
 | 
						|
it under the terms of the GNU General Public License as published by
 | 
						|
the Free Software Foundation, either version 2 of the License, or
 | 
						|
(at your option) any later version.
 | 
						|
 | 
						|
This program is distributed in the hope that it will be useful,
 | 
						|
but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
						|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
						|
GNU General Public License for more details.
 | 
						|
 | 
						|
You should have received a copy of the GNU General Public License
 | 
						|
along with this program.  If not, see <http://www.gnu.org/licenses/>.
 | 
						|
*/
 | 
						|
#include <stdint.h>
 | 
						|
#include <stdbool.h>
 | 
						|
#include <util/delay.h>
 | 
						|
#include "usb_keycodes.h"
 | 
						|
#include "host.h"
 | 
						|
#include "print.h"
 | 
						|
#include "debug.h"
 | 
						|
#include "util.h"
 | 
						|
#include "timer.h"
 | 
						|
#include "layer.h"
 | 
						|
#include "matrix.h"
 | 
						|
#include "bootloader.h"
 | 
						|
#include "command.h"
 | 
						|
 | 
						|
#ifdef HOST_PJRC
 | 
						|
#   include "usb_keyboard.h"
 | 
						|
#   ifdef EXTRAKEY_ENABLE
 | 
						|
#       include "usb_extra.h"
 | 
						|
#   endif
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HOST_VUSB
 | 
						|
#   include "usbdrv.h"
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
static uint8_t command_common(void);
 | 
						|
static void help(void);
 | 
						|
static void switch_layer(uint8_t layer);
 | 
						|
 | 
						|
static bool last_print_enable;
 | 
						|
 | 
						|
uint8_t command_proc(void)
 | 
						|
{
 | 
						|
    uint8_t processed = 0;
 | 
						|
    last_print_enable = print_enable;
 | 
						|
 | 
						|
    if (!IS_COMMAND())
 | 
						|
        return 0;
 | 
						|
 | 
						|
    print_enable = true;
 | 
						|
    if (command_extra() || command_common()) {
 | 
						|
        processed = 1;
 | 
						|
        _delay_ms(500);
 | 
						|
    }
 | 
						|
    print_enable = last_print_enable;
 | 
						|
    return processed;
 | 
						|
}
 | 
						|
 | 
						|
/* This allows to define extra commands. return 0 when not processed. */
 | 
						|
uint8_t command_extra(void) __attribute__ ((weak));
 | 
						|
uint8_t command_extra(void)
 | 
						|
{
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static uint8_t command_common(void)
 | 
						|
{
 | 
						|
    switch (host_get_first_key()) {
 | 
						|
        case KB_H:
 | 
						|
            help();
 | 
						|
            break;
 | 
						|
        case KB_B:
 | 
						|
            host_clear_keyboard_report();
 | 
						|
            host_send_keyboard_report();
 | 
						|
            print("jump to bootloader... ");
 | 
						|
            _delay_ms(1000);
 | 
						|
            bootloader_jump(); // not return
 | 
						|
            print("not supported.\n");
 | 
						|
            break;
 | 
						|
        case KB_D:
 | 
						|
            debug_enable = !debug_enable;
 | 
						|
            if (debug_enable) {
 | 
						|
                last_print_enable = true;
 | 
						|
                print("debug enabled.\n");
 | 
						|
                debug_matrix = true;
 | 
						|
                debug_keyboard = true;
 | 
						|
                debug_mouse = true;
 | 
						|
            } else {
 | 
						|
                print("debug disabled.\n");
 | 
						|
                last_print_enable = false;
 | 
						|
                debug_matrix = false;
 | 
						|
                debug_keyboard = false;
 | 
						|
                debug_mouse = false;
 | 
						|
            }
 | 
						|
            break;
 | 
						|
        case KB_X: // debug matrix toggle
 | 
						|
            debug_matrix = !debug_matrix;
 | 
						|
            if (debug_matrix)
 | 
						|
                print("debug matrix enabled.\n");
 | 
						|
            else
 | 
						|
                print("debug matrix disabled.\n");
 | 
						|
            break;
 | 
						|
        case KB_K: // debug keyboard toggle
 | 
						|
            debug_keyboard = !debug_keyboard;
 | 
						|
            if (debug_keyboard)
 | 
						|
                print("debug keyboard enabled.\n");
 | 
						|
            else
 | 
						|
                print("debug keyboard disabled.\n");
 | 
						|
            break;
 | 
						|
        case KB_M: // debug mouse toggle
 | 
						|
            debug_mouse = !debug_mouse;
 | 
						|
            if (debug_mouse)
 | 
						|
                print("debug mouse enabled.\n");
 | 
						|
            else
 | 
						|
                print("debug mouse disabled.\n");
 | 
						|
            break;
 | 
						|
        case KB_V: // print version & information
 | 
						|
            print(STR(DESCRIPTION) "\n");
 | 
						|
            break;
 | 
						|
        case KB_T: // print timer
 | 
						|
            print("timer: "); phex16(timer_count); print("\n");
 | 
						|
            break;
 | 
						|
        case KB_P: // print toggle
 | 
						|
            if (last_print_enable) {
 | 
						|
                print("print disabled.\n");
 | 
						|
                last_print_enable = false;
 | 
						|
            } else {
 | 
						|
                last_print_enable = true;
 | 
						|
                print("print enabled.\n");
 | 
						|
            }
 | 
						|
            break;
 | 
						|
        case KB_S:
 | 
						|
#ifdef HOST_PJRC
 | 
						|
            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");
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HOST_VUSB
 | 
						|
#   if USB_COUNT_SOF
 | 
						|
            print("usbSofCount: "); phex(usbSofCount); print("\n");
 | 
						|
#   endif
 | 
						|
#endif
 | 
						|
            break;
 | 
						|
#ifdef NKRO_ENABLE
 | 
						|
        case KB_N:
 | 
						|
            // send empty report before change
 | 
						|
            host_clear_keyboard_report();
 | 
						|
            host_send_keyboard_report();
 | 
						|
            keyboard_nkro = !keyboard_nkro;
 | 
						|
            if (keyboard_nkro)
 | 
						|
                print("NKRO: enabled\n");
 | 
						|
            else
 | 
						|
                print("NKRO: disabled\n");
 | 
						|
            break;
 | 
						|
#endif
 | 
						|
#ifdef EXTRAKEY_ENABLE
 | 
						|
        case KB_ESC:
 | 
						|
            host_clear_keyboard_report();
 | 
						|
            host_send_keyboard_report();
 | 
						|
#ifdef HOST_PJRC
 | 
						|
            if (suspend && remote_wakeup) {
 | 
						|
                usb_remote_wakeup();
 | 
						|
            } else {
 | 
						|
                host_system_send(SYSTEM_POWER_DOWN);
 | 
						|
                host_system_send(0);
 | 
						|
                _delay_ms(500);
 | 
						|
            }
 | 
						|
#else
 | 
						|
            host_system_send(SYSTEM_POWER_DOWN);
 | 
						|
            host_system_send(0);
 | 
						|
            _delay_ms(500);
 | 
						|
#endif
 | 
						|
            break;
 | 
						|
#endif
 | 
						|
        case KB_BSPC:
 | 
						|
            matrix_init();
 | 
						|
            print("clear matrix\n");
 | 
						|
            break;
 | 
						|
        case KB_0:
 | 
						|
            switch_layer(0);
 | 
						|
            break;
 | 
						|
        case KB_1:
 | 
						|
            switch_layer(1);
 | 
						|
            break;
 | 
						|
        case KB_2:
 | 
						|
            switch_layer(2);
 | 
						|
            break;
 | 
						|
        case KB_3:
 | 
						|
            switch_layer(3);
 | 
						|
            break;
 | 
						|
        case KB_4:
 | 
						|
            switch_layer(4);
 | 
						|
            break;
 | 
						|
        default:
 | 
						|
            return 0;
 | 
						|
    }
 | 
						|
    return 1;
 | 
						|
}
 | 
						|
 | 
						|
static void help(void)
 | 
						|
{
 | 
						|
    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");
 | 
						|
#ifdef NKRO_ENABLE
 | 
						|
    print("n: toggle NKRO\n");
 | 
						|
#endif
 | 
						|
    print("Backspace: clear matrix\n");
 | 
						|
    print("ESC: power down/wake up\n");
 | 
						|
    print("0: switch to Layer0 \n");
 | 
						|
    print("1: switch to Layer1 \n");
 | 
						|
    print("2: switch to Layer2 \n");
 | 
						|
    print("3: switch to Layer3 \n");
 | 
						|
    print("4: switch to Layer4 \n");
 | 
						|
}
 | 
						|
 | 
						|
static void switch_layer(uint8_t layer)
 | 
						|
{
 | 
						|
    print("current_layer: "); phex(current_layer); print("\n");
 | 
						|
    print("default_layer: "); phex(default_layer); print("\n");
 | 
						|
    current_layer = layer;
 | 
						|
    default_layer = layer;
 | 
						|
    print("switch to Layer: "); phex(layer); print("\n");
 | 
						|
}
 |