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_hhkb.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 5
  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     /* wait for debug print. no need for normal use */
  73     for (int i =0; i < 6; i++) {
  74         LED_CONFIG;
  75         LED_ON;
  76         _delay_ms(500);
  77         LED_OFF;
  78         _delay_ms(500);
  79     }
  80
  81     // Configure timer 0 to generate a timer overflow interrupt every
  82     // 256*1024 clock cycles, or approx 61 Hz when using 16 MHz clock
  83     // This demonstrates how to use interrupts to implement a simple
  84     // inactivity timeout.
  85     TCCR0A = 0x00;
  86     TCCR0B = 0x05;
  87     TIMSK0 = (1<<TOIE0);
  88
  89
  90     matrix_init();
  91
  92     bool modified = false;
  93     bool has_ghost = false;
  94     int layer = 0;
  95     int key_index = 0;
  96     uint8_t mouse_x = 0;
  97     uint8_t mouse_y = 0;
  98     uint8_t mouse_btn = 0;
  99     int8_t mouse_wheel = 0;
 100     int8_t mouse_hwheel = 0;
 101     int mouse_repeat = 0;
 102
 103     print("\nt.m.k. keyboard 1.0 for hhkb\n");
 104     while (1) {
 105         matrix_scan();
 106         modified = matrix_is_modified();
 107         has_ghost = matrix_has_ghost();
 108         layer = get_layer();
 109
 110         // print matrix state for debug
 111         if (modified) {
 112             print_matrix();
 113
 114             // LED flash for debug
 115             LED_CONFIG;
 116             LED_ON;
 117         }
 118
 119         keyboard_modifier_keys = 0;
 120         for (int i = 0; i < 3; i++) keyboard_keys[i] = KB_NO;
 121         key_index = 0;
 122         mouse_x = 0;
 123         mouse_y = 0;
 124         mouse_btn = 0;
 125         mouse_wheel = 0;
 126         mouse_hwheel = 0;
 127
 128         // convert matrix state to HID report
 129         for (int row = 0; row < MATRIX_ROWS; row++) {
 130             for (int col = 0; col < MATRIX_COLS; col++) {
 131                 if (matrix[row] & 1<<col) continue;
 132
 133                 uint8_t code = get_keycode(layer, row, col);
 134                 if (code == KB_NO) {
 135                     continue;
 136                 } else if (KB_LCTRL <= code && code <= KB_RGUI) {
 137                     // modifier keys(0xE0-0xE7)
 138                     keyboard_modifier_keys |= 1<<(code & 0x07);
 139                 } else if (code >= MS_UP) {
 140                     // mouse
 141                     if (code == MS_UP)    mouse_y -= MOUSE_MOVE_UNIT + (mouse_repeat < 50 ? mouse_repeat/5 : 10);
 142                     if (code == MS_DOWN)  mouse_y += MOUSE_MOVE_UNIT + (mouse_repeat < 50 ? mouse_repeat/5 : 10);
 143                     if (code == MS_LEFT)  mouse_x -= MOUSE_MOVE_UNIT + (mouse_repeat < 50 ? mouse_repeat/5 : 10);
 144                     if (code == MS_RIGHT) mouse_x += MOUSE_MOVE_UNIT + (mouse_repeat < 50 ? mouse_repeat/5 : 10);
 145                     if (code == MS_BTN1)  mouse_btn |= 1<<0;
 146                     if (code == MS_BTN2)  mouse_btn |= 1<<1;
 147                     if (code == MS_BTN3)  mouse_btn |= 1<<2;
 148                     if (code == MS_BTN4)  mouse_btn |= 1<<3;
 149                     if (code == MS_BTN5)  mouse_btn |= 1<<4;
 150                     if (code == MS_WH_UP)  mouse_wheel += 1;
 151                     if (code == MS_WH_DOWN)  mouse_wheel -= 1;
 152                     if (code == MS_WH_LEFT)  mouse_hwheel += 1;
 153                     if (code == MS_WH_RIGHT) mouse_hwheel -= 1;
 154                 } else {
 155                     // normal keys
 156                     if (key_index < 6)
 157                         keyboard_keys[key_index] = code;
 158                     key_index++;
 159                 }
 160             }
 161         }
 162
 163         if (!has_ghost)  {
 164             // when 4 left modifier keys down
 165             if (keyboard_modifier_keys == (MOD_LCTRL | MOD_LSHIFT | MOD_LALT | MOD_LGUI)) {
 166                 // cancel all keys
 167                 keyboard_modifier_keys = 0;
 168                 for (int i = 0; i < 6; i++) keyboard_keys[i] = KB_NO;
 169                 usb_keyboard_send();
 170
 171                 print("jump to bootloader...\n");
 172                 _delay_ms(100);
 173                 jump_bootloader(); // not return
 174             }
 175
 176             if (mouse_x || mouse_y || mouse_wheel || mouse_hwheel || mouse_btn != mouse_buttons) {
 177                 mouse_buttons = mouse_btn;
 178                 usb_mouse_move(mouse_x, mouse_y, mouse_wheel, mouse_hwheel);
 179                 key_sent = true;
 180
 181                 // acceleration
 182                 _delay_ms(MOUSE_DELAY_MS >> (mouse_repeat < MOUSE_DELAY_ACC ? mouse_repeat : MOUSE_DELAY_ACC));
 183                 mouse_repeat++;
 184             } else {
 185                 mouse_repeat = 0;
 186             }
 187
 188
 189             // send keys to host
 190             if (modified) {
 191                 if (key_index > 6) {
 192                     //Rollover
 193                 }
 194                 usb_keyboard_send();
 195                 if (keyboard_keys[0])
 196                     key_sent = true;
 197
 198                 // LED flash for debug
 199                 LED_CONFIG;
 200                 LED_OFF;
 201             }
 202         }
 203         _delay_ms(2);
 204     }
 205 }
 206
 207 static void print_matrix(void) {
 208             print("\nr/c 01234567\n");
 209             for (int row = 0; row < MATRIX_ROWS; row++) {
 210                 phex(row); print(": ");
 211                 pbin_reverse(matrix[row]);
 212                 if (matrix_has_ghost_in_row(row)) {
 213                     print(" <ghost");
 214                 }
 215                 print("\n");
 216             }
 217             print("keys: ");
 218             for (int i = 0; i < 6; i++) { phex(keyboard_keys[i]); print(" "); }
 219             print("\n");
 220             print("mod: "); phex(keyboard_modifier_keys); print("\n");
 221 }
 222
 223 // This interrupt routine is run approx 61 times per second.
 224 // A very simple inactivity timeout is implemented, where we
 225 // will send a space character and print a message to the
 226 // hid_listen debug message window.
 227 ISR(TIMER0_OVF_vect)
 228 {
 229     idle_count++;
 230 }