tmk.c

   1 /* 2010/08/23 noname
   2  * keyboard firmware based on PJRC USB keyboard example
   3  */
   4 /* Keyboard example with debug channel, for Teensy USB Development Board
   5  * http://www.pjrc.com/teensy/usb_keyboard.html
   6  * Copyright (c) 2008 PJRC.COM, LLC
   7  *
   8  * Permission is hereby granted, free of charge, to any person obtaining a copy
   9  * of this software and associated documentation files (the "Software"), to deal
  10  * in the Software without restriction, including without limitation the rights
  11  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  12  * copies of the Software, and to permit persons to whom the Software is
  13  * furnished to do so, subject to the following conditions:
  14  *
  15  * The above copyright notice and this permission notice shall be included in
  16  * all copies or substantial portions of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  21  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  23  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  24  * THE SOFTWARE.
  25  */
  26
  27 #include <stdbool.h>
  28 #include <avr/io.h>
  29 #include <avr/pgmspace.h>
  30 #include <avr/interrupt.h>
  31 #include <util/delay.h>
  32 #include "usb.h"
  33 #include "usb_keyboard.h"
  34 #include "usb_mouse.h"
  35 #include "print.h"
  36 #include "matrix.h"
  37 #include "keymap.h"
  38 #include "jump_bootloader.h"
  39
  40
  41 // for teensy 2.0
  42 #define LED_CONFIG    (DDRD |= (1<<6))
  43 #define LED_ON        (PORTD |= (1<<6))
  44 #define LED_OFF       (PORTD &= ~(1<<6))
  45 #define CPU_PRESCALE(n)    (CLKPR = 0x80, CLKPR = (n))
  46
  47 #define MOUSE_MOVE_UNIT 10
  48 #define MOUSE_DELAY_MS 200
  49 #define MOUSE_DELAY_ACC 4
  50
  51
  52 static void print_matrix(void);
  53
  54
  55 uint16_t idle_count=0;
  56
  57
  58 int main(void)
  59 {
  60     // set for 16 MHz clock
  61     CPU_PRESCALE(0);
  62
  63     // Initialize the USB, and then wait for the host to set configuration.
  64     // If the Teensy is powered without a PC connected to the USB port,
  65     // this will wait forever.
  66     usb_init();
  67     while (!usb_configured()) /* wait */ ;
  68
  69     // Wait an extra second for the PC's operating system to load drivers
  70     // and do whatever it does to actually be ready for input
  71     // needs such long time in my PC.
  72     for (int i =0; i < 6; i++) {
  73         LED_CONFIG;
  74         LED_ON;
  75         _delay_ms(500);
  76         LED_OFF;
  77         _delay_ms(500);
  78     }
  79
  80     // Configure timer 0 to generate a timer overflow interrupt every
  81     // 256*1024 clock cycles, or approx 61 Hz when using 16 MHz clock
  82     // This demonstrates how to use interrupts to implement a simple
  83     // inactivity timeout.
  84     TCCR0A = 0x00;
  85     TCCR0B = 0x05;
  86     TIMSK0 = (1<<TOIE0);
  87
  88
  89     matrix_init();
  90
  91     bool modified = false;
  92     bool has_ghost = false;
  93     int layer = 0;
  94     int key_index = 0;
  95     uint8_t mouse_x = 0;
  96     uint8_t mouse_y = 0;
  97     uint8_t mouse_btn = 0;
  98     int8_t mouse_wheel = 0;
  99     int8_t mouse_hwheel = 0;
 100     int mouse_repeat = 0;
 101
 102     print("\nt.m.k. keyboard 1.0\n");
 103     while (1) {
 104         matrix_scan();
 105         modified = matrix_is_modified();
 106         has_ghost = matrix_has_ghost();
 107         layer = get_layer();
 108
 109         // print matrix state for debug
 110         if (modified) {
 111             print_matrix();
 112
 113             LED_CONFIG;
 114             LED_ON;
 115         }
 116
 117         keyboard_modifier_keys = 0;
 118         for (int i = 0; i < 6; i++) keyboard_keys[i] = KB_NO;
 119         key_index = 0;
 120         mouse_x = 0;
 121         mouse_y = 0;
 122         mouse_btn = 0;
 123         mouse_wheel = 0;
 124         mouse_hwheel = 0;
 125
 126         // convert matrix state to HID report
 127         for (int row = 0; row < MATRIX_ROWS; row++) {
 128             for (int col = 0; col < MATRIX_COLS; col++) {
 129                 if (matrix[row] & 1<<col) continue;
 130
 131                 uint8_t code = get_keycode(layer, row, col);
 132                 if (code == KB_NO) {
 133                     continue;
 134                 } else if (KB_LCTRL <= code && code <= KB_RGUI) {
 135                     // modifier keys(0xE0-0xE7)
 136                     keyboard_modifier_keys |= 1<<(code & 0x07);
 137                 } else if (code >= MS_UP) {
 138                     // mouse
 139                     if (code == MS_UP)    mouse_y -= MOUSE_MOVE_UNIT;
 140                     if (code == MS_DOWN)  mouse_y += MOUSE_MOVE_UNIT;
 141                     if (code == MS_LEFT)  mouse_x -= MOUSE_MOVE_UNIT;
 142                     if (code == MS_RIGHT) mouse_x += MOUSE_MOVE_UNIT;
 143                     if (code == MS_BTN1)  mouse_btn |= 1<<0;
 144                     if (code == MS_BTN2)  mouse_btn |= 1<<1;
 145                     if (code == MS_BTN3)  mouse_btn |= 1<<2;
 146                     if (code == MS_BTN4)  mouse_btn |= 1<<3;
 147                     if (code == MS_BTN5)  mouse_btn |= 1<<4;
 148                     if (code == MS_WH_UP)  mouse_wheel -= 1;
 149                     if (code == MS_WH_DOWN)  mouse_wheel += 1;
 150                     if (code == MS_WH_LEFT)  mouse_hwheel -= 1;
 151                     if (code == MS_WH_RIGHT) mouse_hwheel += 1;
 152                 } else {
 153                     // normal keys
 154                     if (key_index < 6)
 155                         keyboard_keys[key_index] = code;
 156                     key_index++;
 157                 }
 158             }
 159         }
 160
 161         if (!has_ghost)  {
 162             // when 4 left modifier keys down
 163             if (keyboard_modifier_keys == (MOD_LCTRL | MOD_LSHIFT | MOD_LALT | MOD_LGUI)) {
 164                 // cancel all keys
 165                 keyboard_modifier_keys = 0;
 166                 for (int i = 0; i < 6; i++) keyboard_keys[i] = KB_NO;
 167                 usb_keyboard_send();
 168
 169                 print("jump to bootloader...\n");
 170                 _delay_ms(100);
 171                 jump_bootloader(); // not return
 172             }
 173
 174             if (mouse_x || mouse_y || mouse_wheel || mouse_hwheel || mouse_btn != mouse_buttons) {
 175                 mouse_buttons = mouse_btn;
 176                 usb_mouse_move(mouse_x, mouse_y, mouse_wheel, mouse_hwheel);
 177
 178                 // acceleration
 179                 _delay_ms(MOUSE_DELAY_MS >> (mouse_repeat < MOUSE_DELAY_ACC ? mouse_repeat : MOUSE_DELAY_ACC));
 180                 mouse_repeat++;
 181             } else {
 182                 mouse_repeat = 0;
 183             }
 184
 185
 186             // send keys to host
 187             if (modified) {
 188                 if (key_index > 6) {
 189                     //Rollover
 190                 }
 191                 usb_keyboard_send();
 192             }
 193         }
 194         _delay_ms(2);
 195     }
 196 }
 197
 198 static void print_matrix(void) {
 199             print("\nr/c 01234567\n");
 200             for (int row = 0; row < MATRIX_ROWS; row++) {
 201                 phex(row); print(": ");
 202                 pbin_reverse(matrix[row]);
 203                 if (matrix_has_ghost_in_row(row)) {
 204                     print(" <ghost");
 205                 }
 206                 print("\n");
 207             }
 208             print("keys: ");
 209             for (int i = 0; i < 6; i++) { phex(keyboard_keys[i]); print(" "); }
 210             print("\n");
 211             print("mod: "); phex(keyboard_modifier_keys); print("\n");
 212 }
 213
 214 // This interrupt routine is run approx 61 times per second.
 215 // A very simple inactivity timeout is implemented, where we
 216 // will send a space character and print a message to the
 217 // hid_listen debug message window.
 218 ISR(TIMER0_OVF_vect)
 219 {
 220     idle_count++;
 221 }