mykey.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
  33 #include "usb.h"
  34 #include "usb_keyboard.h"
  35 #include "usb_mouse.h"
  36 #include "print.h"
  37 #include "matrix.h"
  38 #include "keymap.h"
  39 #include "jump_bootloader.h"
  40
  41 #define LED_CONFIG    (DDRD |= (1<<6))
  42 #define LED_ON        (PORTD &= ~(1<<6))
  43 #define LED_OFF        (PORTD |= (1<<6))
  44 #define CPU_PRESCALE(n)    (CLKPR = 0x80, CLKPR = (n))
  45
  46 static void print_matrix(void);
  47
  48
  49 uint16_t idle_count=0;
  50
  51
  52
  53 int main(void)
  54 {
  55     // set for 16 MHz clock
  56     CPU_PRESCALE(0);
  57
  58     // Initialize the USB, and then wait for the host to set configuration.
  59     // If the Teensy is powered without a PC connected to the USB port,
  60     // this will wait forever.
  61     usb_init();
  62     while (!usb_configured()) /* wait */ ;
  63
  64     // Wait an extra second for the PC's operating system to load drivers
  65     // and do whatever it does to actually be ready for input
  66     _delay_ms(1000);
  67
  68     // Configure timer 0 to generate a timer overflow interrupt every
  69     // 256*1024 clock cycles, or approx 61 Hz when using 16 MHz clock
  70     // This demonstrates how to use interrupts to implement a simple
  71     // inactivity timeout.
  72     TCCR0A = 0x00;
  73     TCCR0B = 0x05;
  74     TIMSK0 = (1<<TOIE0);
  75
  76
  77     matrix_init();
  78     print("firmware 0.3 for t.m.k.\n");
  79
  80     bool modified = false;
  81     bool has_ghost = false;
  82     int layer = 0;
  83     int key_index = 0;
  84     uint8_t mouse_x = 0;
  85     uint8_t mouse_y = 0;
  86     uint8_t mouse_btn = 0;
  87     int8_t mouse_wheel = 0;
  88     int8_t mouse_hwheel = 0;
  89     while (1) {
  90         matrix_scan();
  91         modified = matrix_is_modified();
  92         has_ghost = matrix_has_ghost();
  93         layer = get_layer();
  94
  95         // print matrix state for debug
  96         if (modified) {
  97             print_matrix();
  98
  99             // LED flush
 100             DDRD |= 1<<PD6;
 101             PORTD |= 1<<PD6;
 102         }
 103
 104         keyboard_modifier_keys = 0;
 105         for (int i = 0; i < 6; i++) keyboard_keys[i] = KB_NO;
 106         key_index = 0;
 107         mouse_x = 0;
 108         mouse_y = 0;
 109         mouse_btn = 0;
 110         mouse_wheel = 0;
 111         mouse_hwheel = 0;
 112
 113         // convert matrix state to HID report
 114         for (int row = 0; row < MATRIX_ROWS; row++) {
 115             for (int col = 0; col < MATRIX_COLS; col++) {
 116                 if (matrix[row] & 1<<col) continue;
 117
 118                 uint8_t code = get_keycode(layer, row, col);
 119                 if (code == KB_NO) {
 120                     continue;
 121                 } else if (KB_LCTRL <= code && code <= KB_RGUI) {
 122                     // modifier keycode: 0xE0-0xE7
 123                     keyboard_modifier_keys |= 1<<(code & 0x07);
 124                 } else if (code >= MS_UP) {
 125                     // mouse
 126                     if (code == MS_UP)    mouse_y -= 15;
 127                     if (code == MS_DOWN)  mouse_y += 15;
 128                     if (code == MS_LEFT)  mouse_x -= 15;
 129                     if (code == MS_RIGHT) mouse_x += 15;
 130                     if (code == MS_BTN1)  mouse_btn |= 1<<0;
 131                     if (code == MS_BTN2)  mouse_btn |= 1<<1;
 132                     if (code == MS_BTN3)  mouse_btn |= 1<<2;
 133                     if (code == MS_BTN4)  mouse_btn |= 1<<3;
 134                     if (code == MS_BTN5)  mouse_btn |= 1<<4;
 135                     if (code == MS_WH_UP)  mouse_wheel -= 1;
 136                     if (code == MS_WH_DOWN)  mouse_wheel += 1;
 137                     if (code == MS_WH_LEFT)  mouse_hwheel -= 1;
 138                     if (code == MS_WH_RIGHT) mouse_hwheel += 1;
 139                 } else {
 140                     if (key_index < 6)
 141                         keyboard_keys[key_index] = code;
 142                     key_index++;
 143                 }
 144             }
 145         }
 146
 147         if (!has_ghost)  {
 148             // when 4 left modifier keys down
 149             if (keyboard_modifier_keys == (MOD_LCTRL | MOD_LSHIFT | MOD_LALT | MOD_LGUI)) {
 150                 // cancel all keys
 151                 keyboard_modifier_keys = 0;
 152                 for (int i = 0; i < 6; i++) keyboard_keys[i] = KB_NO;
 153                 usb_keyboard_send();
 154
 155                 print("jump to bootloader...\n");
 156                 _delay_ms(1000);
 157                 jump_bootloader(); // not return
 158             }
 159
 160             if (mouse_x || mouse_y || mouse_wheel || mouse_hwheel || mouse_btn != mouse_buttons) {
 161                 mouse_buttons = mouse_btn;
 162                 print("mouse_buttons: 0b"); pbin(mouse_buttons); print("\n");
 163                 print("mouse_wheel: 0x"); phex(mouse_wheel); print("\n");
 164                 usb_mouse_move(mouse_x, mouse_y, mouse_wheel, mouse_hwheel);
 165                 _delay_ms(100);
 166             }
 167
 168
 169             // send keys to host
 170             if (modified) {
 171                 if (key_index > 6) {
 172                     //Rollover
 173                 }
 174                 usb_keyboard_send();
 175             }
 176         }
 177         _delay_ms(2);
 178     }
 179 }
 180
 181 static void print_matrix(void) {
 182             print("\nr/c 01234567\n");
 183             for (int row = 0; row < MATRIX_ROWS; row++) {
 184                 phex(row); print(": ");
 185                 pbin_reverse(matrix[row]);
 186                 if (matrix_has_ghost_in_row(row)) {
 187                     print(" <ghost");
 188                 }
 189                 print("\n");
 190             }
 191             print("keys: ");
 192             for (int i = 0; i < 6; i++) { phex(keyboard_keys[i]); print(" "); }
 193             print("\n");
 194             print("mod: "); phex(keyboard_modifier_keys); print("\n");
 195 }
 196
 197 // This interrupt routine is run approx 61 times per second.
 198 // A very simple inactivity timeout is implemented, where we
 199 // will send a space character and print a message to the
 200 // hid_listen debug message window.
 201 ISR(TIMER0_OVF_vect)
 202 {
 203     idle_count++;
 204 }