diff options
Diffstat (limited to 'ps2_usb/matrix.c')
-rw-r--r-- | ps2_usb/matrix.c | 456 |
1 files changed, 456 insertions, 0 deletions
diff --git a/ps2_usb/matrix.c b/ps2_usb/matrix.c new file mode 100644 index 000000000..366568d68 --- /dev/null +++ b/ps2_usb/matrix.c @@ -0,0 +1,456 @@ +/* + * scan matrix + */ +#include <stdint.h> +#include <stdbool.h> +#include <avr/io.h> +#include <util/delay.h> +#include "print.h" +#include "util.h" +#include "debug.h" +#include "ps2.h" +#include "usb_keyboard.h" +#include "matrix_skel.h" + + +#if (MATRIX_COLS > 16) +# error "MATRIX_COLS must not exceed 16" +#endif +#if (MATRIX_ROWS > 255) +# error "MATRIX_ROWS must not exceed 255" +#endif + + +/* + * Matrix usage: + * "PS/2 Scan Codes Set 2" is assigned to 256(32x8)cells matrix. + * Hmm, It is very sparse and not efficient :( + * + * 8bit + * --------- + * 0| | + * :| XX | 00-7F for normal codes + * f|_________| + * 10| | + * :| E0 XX | 80-FF for E0-prefix codes(use (XX|0x80) as code) + * 1f| | + * --------- + * exceptions: + * 0x83: F8(normal code placed beyond 0x7F) + * 0xFE: PrintScreen + * 0xFF: Puause/Break + */ +#define _PRINT_SCREEN (0xFE) +#define _PAUSE_BREAK (0xFF) +#define _ROW(code) (code>>3) +#define _COL(code) (code&0x07) + +static bool _matrix_is_modified = false; + +// matrix state buffer(1:on, 0:off) +#if (MATRIX_COLS <= 8) +static uint8_t *matrix; +static uint8_t _matrix0[MATRIX_ROWS]; +#else +static uint16_t *matrix; +static uint16_t _matrix0[MATRIX_ROWS]; +#endif + +#ifdef MATRIX_HAS_GHOST +static bool matrix_has_ghost_in_row(uint8_t row); +#endif +static void _matrix_make(uint8_t code); +static void _matrix_break(uint8_t code); +static void _ps2_reset(void); +static void _ps2_set_leds(uint8_t leds); + + +inline +uint8_t matrix_rows(void) +{ + return MATRIX_ROWS; +} + +inline +uint8_t matrix_cols(void) +{ + return MATRIX_COLS; +} + +void matrix_init(void) +{ + print_enable = true; + ps2_host_init(); + + _ps2_reset(); + + // flush LEDs + _ps2_set_leds(1<<PS2_LED_NUM_LOCK); + _delay_ms(100); + _ps2_set_leds(1<<PS2_LED_NUM_LOCK|1<<PS2_LED_CAPS_LOCK); + _delay_ms(100); + _ps2_set_leds(1<<PS2_LED_NUM_LOCK|1<<PS2_LED_CAPS_LOCK|1<<PS2_LED_SCROLL_LOCK); + _delay_ms(300); + _ps2_set_leds(0x00); + + // initialize matrix state: all keys off + for (uint8_t i=0; i < MATRIX_ROWS; i++) _matrix0[i] = 0x00; + matrix = _matrix0; + + return; +} + +uint8_t matrix_scan(void) +{ + + static enum { + INIT, + BREAK, + E0, + E0_F0, + // states for PrintScreen + E0_12, + E0_12_E0, + E0_F0_7C, + E0_F0_7C_E0, + E0_F0_7C_E0_F0, + // states for Pause/Break + E1, + E1_14, + E1_14_77, + E1_14_77_E1, + E1_14_77_E1_F0, + E1_14_77_E1_F0_14, + E1_14_77_E1_F0_14_F0, + } state = INIT; + + + _matrix_is_modified = false; + + // Pause/Break off(PS/2 has no break for this key) + if (matrix_is_on(_ROW(_PAUSE_BREAK), _COL(_PAUSE_BREAK))) { + _matrix_break(_PAUSE_BREAK); + } + + uint8_t code; + while ((code = ps2_host_recv())) { + switch (state) { + case INIT: + switch (code) { + case 0xE0: // 2byte make + state = E0; + break; + case 0xF0: // break code + state = BREAK; + break; + case 0xE1: // Pause/Break + state = E1; + break; + default: // normal key make + if (code < 0x80) { + _matrix_make(code); + } else { + debug("ps/2 unknow code: "); debug_hex(code); debug("\n"); + } + state = INIT; + } + break; + case E0: + switch (code) { + case 0x12: // PrintScreen(make) + state = E0_12; + break; + case 0x7C: // PrintScreen(typematic) + // ignore + state = INIT; + break; + case 0xF0: // E0 break + state = E0_F0; + break; + default: // E0 make + if (code < 0x80) { + _matrix_make(code|0x80); + } else { + debug("ps/2 unknow code: "); debug_hex(code); debug("\n"); + } + state = INIT; + } + break; + case BREAK: + if (code < 0x80) { + _matrix_break(code); + } else { + debug("ps/2 unknow code: "); debug_hex(code); debug("\n"); + } + state = INIT; + break; + case E0_F0: // E0 break + switch (code) { + case 0x7C: + state = E0_F0_7C; + break; + default: + if (code < 0x80) { + _matrix_break(code|0x80); + } else { + debug("ps/2 unknow code: "); debug_hex(code); debug("\n"); + } + state = INIT; + } + break; + /* PrintScreen(make) */ + case E0_12: + switch (code) { + case 0xE0: + state = E0_12_E0; + break; + default: + state = INIT; + } + break; + case E0_12_E0: + switch (code) { + case 0x7C: + _matrix_make(_PRINT_SCREEN); + state = INIT; + break; + default: + state = INIT; + } + break; + /* PrintScreen(break) */ + case E0_F0_7C: + switch (code) { + case 0xE0: + state = E0_F0_7C_E0; + break; + default: + state = INIT; + } + break; + case E0_F0_7C_E0: + switch (code) { + case 0xF0: + state = E0_F0_7C_E0_F0; + break; + default: + state = INIT; + } + break; + case E0_F0_7C_E0_F0: + switch (code) { + case 0x12: + _matrix_break(_PRINT_SCREEN); + state = INIT; + break; + default: + state = INIT; + } + break; + /* Pause/Break */ + case E1: + switch (code) { + case 0x14: + state = E1_14; + break; + default: + state = INIT; + } + break; + case E1_14: + switch (code) { + case 0x77: + state = E1_14_77; + break; + default: + state = INIT; + } + break; + case E1_14_77: + switch (code) { + case 0xE1: + state = E1_14_77_E1; + break; + default: + state = INIT; + } + break; + case E1_14_77_E1: + switch (code) { + case 0xF0: + state = E1_14_77_E1_F0; + break; + default: + state = INIT; + } + break; + case E1_14_77_E1_F0: + switch (code) { + case 0x14: + state = E1_14_77_E1_F0_14; + break; + default: + state = INIT; + } + break; + case E1_14_77_E1_F0_14: + switch (code) { + case 0xF0: + state = E1_14_77_E1_F0_14_F0; + break; + default: + state = INIT; + } + break; + case E1_14_77_E1_F0_14_F0: + switch (code) { + case 0x77: + _matrix_make(_PAUSE_BREAK); + state = INIT; + break; + default: + state = INIT; + } + break; + default: + state = INIT; + } + } + + static uint8_t prev_leds = 0; + if (prev_leds != usb_keyboard_leds) { + uint8_t leds = 0; + if (usb_keyboard_leds&(1<<USB_LED_SCROLL_LOCK)) + leds |= (1<<PS2_LED_SCROLL_LOCK); + if (usb_keyboard_leds&(1<<USB_LED_NUM_LOCK)) + leds |= (1<<PS2_LED_NUM_LOCK); + if (usb_keyboard_leds&(1<<USB_LED_CAPS_LOCK)) + leds |= (1<<PS2_LED_CAPS_LOCK); + + _ps2_set_leds(leds); + prev_leds = usb_keyboard_leds; + } + + return 1; +} + +bool matrix_is_modified(void) +{ + return _matrix_is_modified; +} + +inline +bool matrix_has_ghost(void) +{ +#ifdef MATRIX_HAS_GHOST + for (uint8_t i = 0; i < MATRIX_ROWS; i++) { + if (matrix_has_ghost_in_row(i)) + return true; + } +#endif + return false; +} + +inline +bool matrix_is_on(uint8_t row, uint8_t col) +{ + return (matrix[row] & (1<<col)); +} + +inline +#if (MATRIX_COLS <= 8) +uint8_t matrix_get_row(uint8_t row) +#else +uint16_t matrix_get_row(uint8_t row) +#endif +{ + return matrix[row]; +} + +void matrix_print(void) +{ +#if (MATRIX_COLS <= 8) + print("\nr/c 01234567\n"); +#else + print("\nr/c 0123456789ABCDEF\n"); +#endif + for (uint8_t row = 0; row < matrix_rows(); row++) { + phex(row); print(": "); +#if (MATRIX_COLS <= 8) + pbin_reverse(matrix_get_row(row)); +#else + pbin_reverse16(matrix_get_row(row)); +#endif +#ifdef MATRIX_HAS_GHOST + if (matrix_has_ghost_in_row(row)) { + print(" <ghost"); + } +#endif + print("\n"); + } +} + +uint8_t matrix_key_count(void) +{ + uint8_t count = 0; + for (uint8_t i = 0; i < MATRIX_ROWS; i++) { +#if (MATRIX_COLS <= 8) + count += bitpop(matrix[i]); +#else + count += bitpop16(matrix[i]); +#endif + } + return count; +} + +#ifdef MATRIX_HAS_GHOST +inline +static bool matrix_has_ghost_in_row(uint8_t row) +{ + // no ghost exists in case less than 2 keys on + if (((matrix[row] - 1) & matrix[row]) == 0) + return false; + + // ghost exists in case same state as other row + for (uint8_t i=0; i < MATRIX_ROWS; i++) { + if (i != row && (matrix[i] & matrix[row]) == matrix[row]) + return true; + } + return false; +} +#endif + + +inline +static void _matrix_make(uint8_t code) +{ + if (!matrix_is_on(_ROW(code), _COL(code))) { + matrix[_ROW(code)] |= 1<<_COL(code); + _matrix_is_modified = true; + } +} + +inline +static void _matrix_break(uint8_t code) +{ + if (matrix_is_on(_ROW(code), _COL(code))) { + matrix[_ROW(code)] &= ~(1<<_COL(code)); + _matrix_is_modified = true; + } +} + +static void _ps2_reset(void) +{ + ps2_host_send(0xFF); + ps2_host_recv(); // 0xFA + ps2_host_recv(); // 0xAA + _delay_ms(1000); +} + +static void _ps2_set_leds(uint8_t leds) +{ + ps2_host_send(0xED); + ps2_host_recv(); // 0xFA + ps2_host_send(leds); + ps2_host_recv(); // 0xFA +} |