]> git.donarmstrong.com Git - tmk_firmware.git/blob - adb/matrix.c
ADB keyboard LEDs support
[tmk_firmware.git] / adb / matrix.c
1 /*
2  * scan matrix
3  */
4 #include <stdint.h>
5 #include <stdbool.h>
6 #include <avr/io.h>
7 #include <util/delay.h>
8 #include "print.h"
9 #include "util.h"
10 #include "debug.h"
11 #include "adb.h"
12 #include "usb_keyboard.h"
13 #include "matrix_skel.h"
14
15
16 #if (MATRIX_COLS > 16)
17 #   error "MATRIX_COLS must not exceed 16"
18 #endif
19 #if (MATRIX_ROWS > 255)
20 #   error "MATRIX_ROWS must not exceed 255"
21 #endif
22
23
24 static bool _matrix_is_modified = false;
25
26 // matrix state buffer(1:on, 0:off)
27 #if (MATRIX_COLS <= 8)
28 static uint8_t *matrix;
29 static uint8_t _matrix0[MATRIX_ROWS];
30 #else
31 static uint16_t *matrix;
32 static uint16_t _matrix0[MATRIX_ROWS];
33 #endif
34
35 #ifdef MATRIX_HAS_GHOST
36 static bool matrix_has_ghost_in_row(uint8_t row);
37 #endif
38 static void _register_key(uint8_t key);
39
40
41 inline
42 uint8_t matrix_rows(void)
43 {
44     return MATRIX_ROWS;
45 }
46
47 inline
48 uint8_t matrix_cols(void)
49 {
50     return MATRIX_COLS;
51 }
52
53 void matrix_init(void)
54 {
55     adb_host_init();
56
57     // initialize matrix state: all keys off
58     for (uint8_t i=0; i < MATRIX_ROWS; i++) _matrix0[i] = 0x00;
59     matrix = _matrix0;
60
61     print_enable = true;
62     debug_enable = true;
63     debug_matrix = true;
64     debug_keyboard = true;
65     debug_mouse = true;
66     print("debug enabled.\n");
67     return;
68 }
69
70 uint8_t matrix_scan(void)
71 {
72     uint16_t codes;
73     uint8_t key0, key1;
74
75     static uint8_t prev_led = 0;
76     if (prev_led != usb_keyboard_leds) {
77         adb_host_kbd_led(~usb_keyboard_leds);
78         prev_led = usb_keyboard_leds;
79     }
80
81     _matrix_is_modified = false;
82     codes = adb_host_kbd_recv();
83     key0 = codes>>8;
84     key1 = codes&0xFF;
85
86     if (codes == 0) {                           // no keys
87         return 0;
88     } else if (key0 == 0xFF && key1 != 0xFF) {  // error
89         return codes&0xFF;
90     } else {
91         _matrix_is_modified = true;
92         _register_key(key0);
93         if (key1 != 0xFF)       // key1 is 0xFF when no second key.
94             _register_key(key1);
95     }
96
97     if (debug_matrix && matrix_is_modified()) {
98         print("adb_host_kbd_recv: "); phex16(codes); print("\n");
99     }
100     return 1;
101 }
102
103 bool matrix_is_modified(void)
104 {
105     return _matrix_is_modified;
106 }
107
108 inline
109 bool matrix_has_ghost(void)
110 {
111 #ifdef MATRIX_HAS_GHOST
112     for (uint8_t i = 0; i < MATRIX_ROWS; i++) {
113         if (matrix_has_ghost_in_row(i))
114             return true;
115     }
116 #endif
117     return false;
118 }
119
120 inline
121 bool matrix_is_on(uint8_t row, uint8_t col)
122 {
123     return (matrix[row] & (1<<col));
124 }
125
126 inline
127 #if (MATRIX_COLS <= 8)
128 uint8_t matrix_get_row(uint8_t row)
129 #else
130 uint16_t matrix_get_row(uint8_t row)
131 #endif
132 {
133     return matrix[row];
134 }
135
136 void matrix_print(void)
137 {
138 #if (MATRIX_COLS <= 8)
139     print("\nr/c 01234567\n");
140 #else
141     print("\nr/c 0123456789ABCDEF\n");
142 #endif
143     for (uint8_t row = 0; row < matrix_rows(); row++) {
144         phex(row); print(": ");
145 #if (MATRIX_COLS <= 8)
146         pbin_reverse(matrix_get_row(row));
147 #else
148         pbin_reverse16(matrix_get_row(row));
149 #endif
150 #ifdef MATRIX_HAS_GHOST
151         if (matrix_has_ghost_in_row(row)) {
152             print(" <ghost");
153         }
154 #endif
155         print("\n");
156     }
157 }
158
159 uint8_t matrix_key_count(void)
160 {
161     uint8_t count = 0;
162     for (uint8_t i = 0; i < MATRIX_ROWS; i++) {
163 #if (MATRIX_COLS <= 8)
164         count += bitpop(matrix[i]);
165 #else
166         count += bitpop16(matrix[i]);
167 #endif
168     }
169     return count;
170 }
171
172 #ifdef MATRIX_HAS_GHOST
173 inline
174 static bool matrix_has_ghost_in_row(uint8_t row)
175 {
176     // no ghost exists in case less than 2 keys on
177     if (((matrix[row] - 1) & matrix[row]) == 0)
178         return false;
179
180     // ghost exists in case same state as other row
181     for (uint8_t i=0; i < MATRIX_ROWS; i++) {
182         if (i != row && (matrix[i] & matrix[row]) == matrix[row])
183             return true;
184     }
185     return false;
186 }
187 #endif
188
189 inline
190 static void _register_key(uint8_t key)
191 {
192     uint8_t col, row;
193     col = key&0x07;
194     row = (key>>3)&0x0F;
195     if (key&0x80) {
196         matrix[row] &= ~(1<<col);
197     } else {
198         matrix[row] |=  (1<<col);
199     }
200 }