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     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 for hhkb\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 flash for debug
 114             LED_CONFIG;
 115             LED_ON;
 116         }
 117
 118         keyboard_modifier_keys = 0;
 119         for (int i = 0; i < 3; i++) keyboard_keys[i] = KB_NO;
 120         key_index = 0;
 121         mouse_x = 0;
 122         mouse_y = 0;
 123         mouse_btn = 0;
 124         mouse_wheel = 0;
 125         mouse_hwheel = 0;
 126
 127         // convert matrix state to HID report
 128         for (int row = 0; row < MATRIX_ROWS; row++) {
 129             for (int col = 0; col < MATRIX_COLS; col++) {
 130                 if (matrix[row] & 1<<col) continue;
 131
 132                 uint8_t code = get_keycode(layer, row, col);
 133                 if (code == KB_NO) {
 134                     continue;
 135                 } else if (KB_LCTRL <= code && code <= KB_RGUI) {
 136                     // modifier keys(0xE0-0xE7)
 137                     keyboard_modifier_keys |= 1<<(code & 0x07);
 138                 } else if (code >= MS_UP) {
 139                     // mouse
 140                     if (code == MS_UP)    mouse_y -= MOUSE_MOVE_UNIT + (mouse_repeat < 50 ? mouse_repeat/5 : 10);
 141                     if (code == MS_DOWN)  mouse_y += MOUSE_MOVE_UNIT + (mouse_repeat < 50 ? mouse_repeat/5 : 10);
 142                     if (code == MS_LEFT)  mouse_x -= MOUSE_MOVE_UNIT + (mouse_repeat < 50 ? mouse_repeat/5 : 10);
 143                     if (code == MS_RIGHT) mouse_x += MOUSE_MOVE_UNIT + (mouse_repeat < 50 ? mouse_repeat/5 : 10);
 144                     if (code == MS_BTN1)  mouse_btn |= 1<<0;
 145                     if (code == MS_BTN2)  mouse_btn |= 1<<1;
 146                     if (code == MS_BTN3)  mouse_btn |= 1<<2;
 147                     if (code == MS_BTN4)  mouse_btn |= 1<<3;
 148                     if (code == MS_BTN5)  mouse_btn |= 1<<4;
 149                     if (code == MS_WH_UP)  mouse_wheel -= 1;
 150                     if (code == MS_WH_DOWN)  mouse_wheel += 1;
 151                     if (code == MS_WH_LEFT)  mouse_hwheel -= 1;
 152                     if (code == MS_WH_RIGHT) mouse_hwheel += 1;
 153                 } else {
 154                     // normal keys
 155                     if (key_index < 6)
 156                         keyboard_keys[key_index] = code;
 157                     key_index++;
 158                 }
 159             }
 160         }
 161
 162         if (!has_ghost)  {
 163             // when 4 left modifier keys down
 164             if (keyboard_modifier_keys == (MOD_LCTRL | MOD_LSHIFT | MOD_LALT | MOD_LGUI)) {
 165                 // cancel all keys
 166                 keyboard_modifier_keys = 0;
 167                 for (int i = 0; i < 6; i++) keyboard_keys[i] = KB_NO;
 168                 usb_keyboard_send();
 169
 170                 print("jump to bootloader...\n");
 171                 _delay_ms(100);
 172                 jump_bootloader(); // not return
 173             }
 174
 175             if (mouse_x || mouse_y || mouse_wheel || mouse_hwheel || mouse_btn != mouse_buttons) {
 176                 mouse_buttons = mouse_btn;
 177                 usb_mouse_move(mouse_x, mouse_y, mouse_wheel, mouse_hwheel);
 178
 179                 // acceleration
 180                 _delay_ms(MOUSE_DELAY_MS >> (mouse_repeat < MOUSE_DELAY_ACC ? mouse_repeat : MOUSE_DELAY_ACC));
 181                 mouse_repeat++;
 182             } else {
 183                 mouse_repeat = 0;
 184             }
 185
 186
 187             // send keys to host
 188             if (modified) {
 189                 if (key_index > 6) {
 190                     //Rollover
 191                 }
 192                 usb_keyboard_send();
 193
 194                 // LED flash for debug
 195                 LED_CONFIG;
 196                 LED_OFF;
 197             }
 198         }
 199         _delay_ms(2);
 200     }
 201 }
 202
 203 static void print_matrix(void) {
 204             print("\nr/c 01234567\n");
 205             for (int row = 0; row < MATRIX_ROWS; row++) {
 206                 phex(row); print(": ");
 207                 pbin_reverse(matrix[row]);
 208                 if (matrix_has_ghost_in_row(row)) {
 209                     print(" <ghost");
 210                 }
 211                 print("\n");
 212             }
 213             print("keys: ");
 214             for (int i = 0; i < 6; i++) { phex(keyboard_keys[i]); print(" "); }
 215             print("\n");
 216             print("mod: "); phex(keyboard_modifier_keys); print("\n");
 217 }
 218
 219 // This interrupt routine is run approx 61 times per second.
 220 // A very simple inactivity timeout is implemented, where we
 221 // will send a space character and print a message to the
 222 // hid_listen debug message window.
 223 ISR(TIMER0_OVF_vect)
 224 {
 225     idle_count++;
 226 }