175 lines
5.6 KiB
C
175 lines
5.6 KiB
C
// TODO: clean unused headers
|
|
#include <stdbool.h>
|
|
#include <avr/io.h>
|
|
#include <avr/pgmspace.h>
|
|
#include <avr/interrupt.h>
|
|
#include <util/delay.h>
|
|
#include "usb.h"
|
|
#include "usb_keyboard.h"
|
|
#include "usb_mouse.h"
|
|
#include "print.h"
|
|
#include "matrix.h"
|
|
#include "keymap.h"
|
|
#include "jump_bootloader.h"
|
|
|
|
#include "key_process.h"
|
|
|
|
|
|
// for Teensy/Teensy++ 2.0
|
|
#define LED_CONFIG (DDRD |= (1<<6))
|
|
#define LED_ON (PORTD |= (1<<6))
|
|
#define LED_OFF (PORTD &= ~(1<<6))
|
|
|
|
#define MOUSE_MOVE_UNIT 10
|
|
#define MOUSE_DELAY_MS 200
|
|
#define MOUSE_DELAY_ACC 5
|
|
|
|
|
|
|
|
static void print_matrix(void);
|
|
static void print_keys(void);
|
|
static void print_mouse(int8_t mouse_x, int8_t mouse_y, int8_t wheel_v, int8_t wheel_h);
|
|
|
|
void proc_matrix(void) {
|
|
static int mouse_repeat = 0;
|
|
|
|
bool modified = false;
|
|
bool has_ghost = false;
|
|
int layer = 0;
|
|
int key_index = 0;
|
|
uint8_t mouse_btn = 0;
|
|
int8_t mouse_x = 0;
|
|
int8_t mouse_y = 0;
|
|
int8_t mouse_wheel = 0;
|
|
int8_t mouse_hwheel = 0;
|
|
|
|
matrix_scan();
|
|
modified = matrix_is_modified();
|
|
has_ghost = matrix_has_ghost();
|
|
layer = get_layer();
|
|
|
|
// print matrix state for debug
|
|
if (modified) {
|
|
print_matrix();
|
|
|
|
// LED flash for debug
|
|
LED_CONFIG;
|
|
LED_ON;
|
|
}
|
|
|
|
keyboard_modifier_keys = 0;
|
|
for (int i = 0; i < 6; i++) keyboard_keys[i] = KB_NO;
|
|
key_index = 0;
|
|
mouse_btn = 0;
|
|
mouse_x = 0;
|
|
mouse_y = 0;
|
|
mouse_wheel = 0;
|
|
mouse_hwheel = 0;
|
|
|
|
// convert matrix state to HID report
|
|
for (int row = 0; row < MATRIX_ROWS; row++) {
|
|
for (int col = 0; col < MATRIX_COLS; col++) {
|
|
if (matrix[row] & 1<<col) continue;
|
|
|
|
uint8_t code = get_keycode(layer, row, col);
|
|
if (code == KB_NO) {
|
|
continue;
|
|
} else if (KB_LCTRL <= code && code <= KB_RGUI) {
|
|
// modifier keys(0xE0-0xE7)
|
|
keyboard_modifier_keys |= 1<<(code & 0x07);
|
|
} else if (code >= MS_UP) {
|
|
// mouse
|
|
if (code == MS_UP) mouse_y -= MOUSE_MOVE_UNIT + (mouse_repeat < 50 ? mouse_repeat/5 : 10);
|
|
if (code == MS_DOWN) mouse_y += MOUSE_MOVE_UNIT + (mouse_repeat < 50 ? mouse_repeat/5 : 10);
|
|
if (code == MS_LEFT) mouse_x -= MOUSE_MOVE_UNIT + (mouse_repeat < 50 ? mouse_repeat/5 : 10);
|
|
if (code == MS_RIGHT) mouse_x += MOUSE_MOVE_UNIT + (mouse_repeat < 50 ? mouse_repeat/5 : 10);
|
|
if (code == MS_BTN1) mouse_btn |= 1<<0;
|
|
if (code == MS_BTN2) mouse_btn |= 1<<1;
|
|
if (code == MS_BTN3) mouse_btn |= 1<<2;
|
|
if (code == MS_BTN4) mouse_btn |= 1<<3;
|
|
if (code == MS_BTN5) mouse_btn |= 1<<4;
|
|
if (code == MS_WH_UP) mouse_wheel += 1;
|
|
if (code == MS_WH_DOWN) mouse_wheel -= 1;
|
|
if (code == MS_WH_LEFT) mouse_hwheel -= 1;
|
|
if (code == MS_WH_RIGHT) mouse_hwheel += 1;
|
|
} else {
|
|
// normal keys
|
|
if (key_index < 6)
|
|
keyboard_keys[key_index] = code;
|
|
key_index++;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!has_ghost) {
|
|
// when 4 left modifier keys down
|
|
if (keyboard_modifier_keys == (MOD_LCTRL | MOD_LSHIFT | MOD_LALT | MOD_LGUI)) {
|
|
// cancel all keys
|
|
keyboard_modifier_keys = 0;
|
|
for (int i = 0; i < 6; i++) keyboard_keys[i] = KB_NO;
|
|
usb_keyboard_send();
|
|
|
|
print("jump to bootloader...\n");
|
|
_delay_ms(100);
|
|
jump_bootloader(); // not return
|
|
}
|
|
|
|
if (mouse_x || mouse_y || mouse_wheel || mouse_hwheel || mouse_btn != mouse_buttons) {
|
|
mouse_buttons = mouse_btn;
|
|
usb_mouse_move(mouse_x, mouse_y, mouse_wheel, mouse_hwheel);
|
|
print_mouse(mouse_x, mouse_y, mouse_wheel, mouse_hwheel);
|
|
key_sent = true;
|
|
|
|
// acceleration
|
|
_delay_ms(MOUSE_DELAY_MS >> (mouse_repeat < MOUSE_DELAY_ACC ? mouse_repeat : MOUSE_DELAY_ACC));
|
|
mouse_repeat++;
|
|
} else {
|
|
mouse_repeat = 0;
|
|
}
|
|
|
|
|
|
// send keys to host
|
|
if (modified) {
|
|
if (key_index > 6) {
|
|
//Rollover
|
|
}
|
|
usb_keyboard_send();
|
|
if (keyboard_keys[0])
|
|
key_sent = true;
|
|
|
|
print_keys();
|
|
// LED flash for debug
|
|
LED_CONFIG;
|
|
LED_OFF;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void print_matrix(void) {
|
|
print("\nr/c 01234567\n");
|
|
for (int row = 0; row < MATRIX_ROWS; row++) {
|
|
phex(row); print(": ");
|
|
pbin_reverse(matrix[row]);
|
|
if (matrix_has_ghost_in_row(row)) {
|
|
print(" <ghost");
|
|
}
|
|
print("\n");
|
|
}
|
|
}
|
|
|
|
static void print_keys(void) {
|
|
print("\nkeys: ");
|
|
for (int i = 0; i < 6; i++) { phex(keyboard_keys[i]); print(" "); }
|
|
print("\n");
|
|
print("mods: "); phex(keyboard_modifier_keys); print("\n");
|
|
}
|
|
|
|
static void print_mouse(int8_t mouse_x, int8_t mouse_y, int8_t wheel_v, int8_t wheel_h) {
|
|
print("\nmouse_x y v h: ");
|
|
phex(mouse_x); print(" ");
|
|
phex(mouse_y); print(" ");
|
|
phex(wheel_v); print(" ");
|
|
phex(wheel_h); print("\n");
|
|
print("buttons: "); phex(mouse_buttons); print("\n");
|
|
}
|