2 Copyright 2017 Danny Nguyen <danny@hexwire.com>
4 This program is free software: you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation, either version 2 of the License, or
7 (at your option) any later version.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program. If not, see <http://www.gnu.org/licenses/>.
25 #include <avr/interrupt.h>
26 #include <util/delay.h>
31 #include "split_util.h"
32 #include "pro_micro.h"
45 #define ERROR_DISCONNECT_COUNT 5
47 static uint8_t debouncing = DEBOUNCE;
48 static const int ROWS_PER_HAND = MATRIX_ROWS/2;
49 static uint8_t error_count = 0;
51 static const uint8_t row_pins[MATRIX_ROWS] = MATRIX_ROW_PINS;
52 static const uint8_t col_pins[MATRIX_COLS] = MATRIX_COL_PINS;
54 /* matrix state(1:on, 0:off) */
55 static matrix_row_t matrix[MATRIX_ROWS];
56 static matrix_row_t matrix_debouncing[MATRIX_ROWS];
58 static matrix_row_t read_cols(void);
59 static void init_cols(void);
60 static void unselect_rows(void);
61 static void select_row(uint8_t row);
63 __attribute__ ((weak))
64 void matrix_init_quantum(void) {
68 __attribute__ ((weak))
69 void matrix_scan_quantum(void) {
73 __attribute__ ((weak))
74 void matrix_init_kb(void) {
78 __attribute__ ((weak))
79 void matrix_scan_kb(void) {
83 __attribute__ ((weak))
84 void matrix_init_user(void) {
87 __attribute__ ((weak))
88 void matrix_scan_user(void) {
92 uint8_t matrix_rows(void)
98 uint8_t matrix_cols(void)
103 void matrix_init(void)
108 // initialize row and col
114 // initialize matrix state: all keys off
115 for (uint8_t i=0; i < MATRIX_ROWS; i++) {
117 matrix_debouncing[i] = 0;
120 matrix_init_quantum();
123 uint8_t _matrix_scan(void)
125 // Right hand is stored after the left in the matirx so, we need to offset it
126 int offset = isLeftHand ? 0 : (ROWS_PER_HAND);
128 for (uint8_t i = 0; i < ROWS_PER_HAND; i++) {
130 _delay_us(30); // without this wait read unstable value.
131 matrix_row_t cols = read_cols();
132 if (matrix_debouncing[i+offset] != cols) {
133 matrix_debouncing[i+offset] = cols;
134 debouncing = DEBOUNCE;
143 for (uint8_t i = 0; i < ROWS_PER_HAND; i++) {
144 matrix[i+offset] = matrix_debouncing[i+offset];
154 // Get rows from other half over i2c
155 int i2c_transaction(void) {
156 int slaveOffset = (isLeftHand) ? (ROWS_PER_HAND) : 0;
158 int err = i2c_master_start(SLAVE_I2C_ADDRESS + I2C_WRITE);
159 if (err) goto i2c_error;
161 // start of matrix stored at 0x00
162 err = i2c_master_write(0x00);
163 if (err) goto i2c_error;
166 err = i2c_master_start(SLAVE_I2C_ADDRESS + I2C_READ);
167 if (err) goto i2c_error;
171 for (i = 0; i < ROWS_PER_HAND-1; ++i) {
172 matrix[slaveOffset+i] = i2c_master_read(I2C_ACK);
174 matrix[slaveOffset+i] = i2c_master_read(I2C_NACK);
177 i2c_error: // the cable is disconnceted, or something else went wrong
187 int serial_transaction(void) {
188 int slaveOffset = (isLeftHand) ? (ROWS_PER_HAND) : 0;
190 if (serial_update_buffers()) {
194 for (int i = 0; i < ROWS_PER_HAND; ++i) {
195 matrix[slaveOffset+i] = serial_slave_buffer[i];
201 uint8_t matrix_scan(void)
203 int ret = _matrix_scan();
208 if( i2c_transaction() ) {
210 if( serial_transaction() ) {
212 // turn on the indicator led when halves are disconnected
217 if (error_count > ERROR_DISCONNECT_COUNT) {
218 // reset other half if disconnected
219 int slaveOffset = (isLeftHand) ? (ROWS_PER_HAND) : 0;
220 for (int i = 0; i < ROWS_PER_HAND; ++i) {
221 matrix[slaveOffset+i] = 0;
225 // turn off the indicator led on no error
229 matrix_scan_quantum();
233 void matrix_slave_scan(void) {
236 int offset = (isLeftHand) ? 0 : (MATRIX_ROWS / 2);
239 for (int i = 0; i < ROWS_PER_HAND; ++i) {
240 /* i2c_slave_buffer[i] = matrix[offset+i]; */
241 i2c_slave_buffer[i] = matrix[offset+i];
244 for (int i = 0; i < ROWS_PER_HAND; ++i) {
245 serial_slave_buffer[i] = matrix[offset+i];
250 bool matrix_is_modified(void)
252 if (debouncing) return false;
257 bool matrix_is_on(uint8_t row, uint8_t col)
259 return (matrix[row] & ((matrix_row_t)1<<col));
263 matrix_row_t matrix_get_row(uint8_t row)
268 void matrix_print(void)
270 print("\nr/c 0123456789ABCDEF\n");
271 for (uint8_t row = 0; row < MATRIX_ROWS; row++) {
272 phex(row); print(": ");
273 pbin_reverse16(matrix_get_row(row));
278 uint8_t matrix_key_count(void)
281 for (uint8_t i = 0; i < MATRIX_ROWS; i++) {
282 count += bitpop16(matrix[i]);
287 static void init_cols(void)
289 for(int x = 0; x < MATRIX_COLS; x++) {
290 _SFR_IO8((col_pins[x] >> 4) + 1) &= ~_BV(col_pins[x] & 0xF);
291 _SFR_IO8((col_pins[x] >> 4) + 2) |= _BV(col_pins[x] & 0xF);
295 static matrix_row_t read_cols(void)
297 matrix_row_t result = 0;
298 for(int x = 0; x < MATRIX_COLS; x++) {
299 result |= (_SFR_IO8(col_pins[x] >> 4) & _BV(col_pins[x] & 0xF)) ? 0 : (1 << x);
304 static void unselect_rows(void)
306 for(int x = 0; x < ROWS_PER_HAND; x++) {
307 _SFR_IO8((row_pins[x] >> 4) + 1) &= ~_BV(row_pins[x] & 0xF);
308 _SFR_IO8((row_pins[x] >> 4) + 2) |= _BV(row_pins[x] & 0xF);
312 static void select_row(uint8_t row)
314 _SFR_IO8((row_pins[row] >> 4) + 1) |= _BV(row_pins[row] & 0xF);
315 _SFR_IO8((row_pins[row] >> 4) + 2) &= ~_BV(row_pins[row] & 0xF);