/* Copyright 2013 Oleg Kostyuk <cub.uanic@gmail.com> This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 2 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see <http://www.gnu.org/licenses/>. */ #include "matrix.h" #include <stdint.h> #include <stdbool.h> #include <avr/io.h> #include "wait.h" #include "action_layer.h" #include "print.h" #include "debug.h" #include "util.h" #include "debounce.h" #include QMK_KEYBOARD_H #ifdef BALLER #include <avr/interrupt.h> #include "pointing_device.h" #endif #ifndef DEBOUNCE # define DEBOUNCE 5 #endif // MCP Pin Defs #define RROW1 (1u<<3) #define RROW2 (1u<<2) #define RROW3 (1u<<1) #define RROW4 (1u<<0) #define COL0 (1u<<0) #define COL1 (1u<<1) #define COL2 (1u<<2) #define COL3 (1u<<3) #define COL4 (1u<<4) #define COL5 (1u<<5) #define COL6 (1u<<6) // ATmega pin defs #define ROW1 (1u<<6) #define ROW2 (1u<<5) #define ROW3 (1u<<4) #define ROW4 (1u<<1) #define COL7 (1u<<0) #define COL8 (1u<<1) #define COL9 (1u<<2) #define COL10 (1u<<3) #define COL11 (1u<<2) #define COL12 (1u<<3) #define COL13 (1u<<6) //Trackball pin defs #define TRKUP (1u<<4) #define TRKDN (1u<<5) #define TRKLT (1u<<6) #define TRKRT (1u<<7) #define TRKBTN (1u<<6) // Multiple for mouse moves #ifndef TRKSTEP #define TRKSTEP 20 #endif // multiple for mouse scroll #ifndef SCROLLSTEP #define SCROLLSTEP 5 #endif // bit masks #define BMASK (COL7 | COL8 | COL9 | COL10) #define CMASK (COL13) #define DMASK (COL11 | COL12) #define FMASK (ROW1 | ROW2 | ROW3 | ROW4) #define RROWMASK (RROW1 | RROW2 | RROW3 | RROW4) #define MCPMASK (COL0 | COL1 | COL2 | COL3 | COL4 | COL5 | COL6) #define TRKMASK (TRKUP | TRKDN | TRKRT | TRKLT) // Trackball interrupts accumulate over here. Processed on scan // Stores prev state of mouse, high bits store direction uint8_t trkState = 0; uint8_t trkBtnState = 0; volatile uint8_t tbUpCnt = 0; volatile uint8_t tbDnCnt = 0; volatile uint8_t tbLtCnt = 0; volatile uint8_t tbRtCnt = 0; /* matrix state(1:on, 0:off) */ static matrix_row_t matrix[MATRIX_ROWS]; /* * matrix state(1:on, 0:off) * contains the raw values without debounce filtering of the last read cycle. */ static matrix_row_t raw_matrix[MATRIX_ROWS]; // Debouncing: store for each key the number of scans until it's eligible to // change. When scanning the matrix, ignore any changes in keys that have // already changed in the last DEBOUNCE scans. static matrix_row_t read_cols(uint8_t row); static void init_cols(void); static void unselect_rows(void); static void select_row(uint8_t row); static void enableInterrupts(void); static uint8_t mcp23018_reset_loop; // static uint16_t mcp23018_reset_loop; __attribute__ ((weak)) void matrix_init_user(void) {} __attribute__ ((weak)) void matrix_scan_user(void) {} __attribute__ ((weak)) void matrix_init_kb(void) { matrix_init_user(); } __attribute__ ((weak)) void matrix_scan_kb(void) { matrix_scan_user(); } inline uint8_t matrix_rows(void) { return MATRIX_ROWS; } inline uint8_t matrix_cols(void) { return MATRIX_COLS; } void matrix_init(void) { // initialize row and col mcp23018_status = init_mcp23018(); unselect_rows(); init_cols(); // initialize matrix state: all keys off for (uint8_t i = 0; i < MATRIX_ROWS; i++) { matrix[i] = 0; raw_matrix[i] = 0; } debounce_init(MATRIX_ROWS); matrix_init_quantum(); } void matrix_power_up(void) { mcp23018_status = init_mcp23018(); unselect_rows(); init_cols(); // initialize matrix state: all keys off for (uint8_t i=0; i < MATRIX_ROWS; i++) { matrix[i] = 0; } } // Reads and stores a row, returning // whether a change occurred. static inline bool store_raw_matrix_row(uint8_t index) { matrix_row_t temp = read_cols(index); if (raw_matrix[index] != temp) { raw_matrix[index] = temp; return true; } return false; } uint8_t matrix_scan(void) { // TODO: Find what is trashing interrupts enableInterrupts(); // First we handle the mouse inputs #ifdef BALLER uint8_t pBtn = PINE & TRKBTN; #ifdef DEBUG_BALLER // Compare to previous, mod report if (tbUpCnt + tbDnCnt + tbLtCnt + tbRtCnt != 0) xprintf("U: %d D: %d L: %d R: %d B: %d\n", tbUpCnt, tbDnCnt, tbLtCnt, tbRtCnt, (trkBtnState >> 6)); #endif // Modify the report report_mouse_t pRprt = pointing_device_get_report(); // Scroll by default, move on layer if (layer_state == 0) { pRprt.h += tbLtCnt * SCROLLSTEP; tbLtCnt = 0; pRprt.h -= tbRtCnt * SCROLLSTEP; tbRtCnt = 0; pRprt.v -= tbUpCnt * SCROLLSTEP; tbUpCnt = 0; pRprt.v += tbDnCnt * SCROLLSTEP; tbDnCnt = 0; } else { pRprt.x -= tbLtCnt * TRKSTEP * (layer_state - 1); tbLtCnt = 0; pRprt.x += tbRtCnt * TRKSTEP * (layer_state - 1); tbRtCnt = 0; pRprt.y -= tbUpCnt * TRKSTEP * (layer_state - 1); tbUpCnt = 0; pRprt.y += tbDnCnt * TRKSTEP * (layer_state - 1); tbDnCnt = 0; } #ifdef DEBUG_BALLER if (pRprt.x != 0 || pRprt.y != 0) xprintf("X: %d Y: %d\n", pRprt.x, pRprt.y); #endif if ((pBtn != trkBtnState) && ((pBtn >> 6) == 0)) pRprt.buttons |= MOUSE_BTN1; if ((pBtn != trkBtnState) && ((pBtn >> 6) == 1)) pRprt.buttons &= ~MOUSE_BTN1; // Save state, push update if (pRprt.x != 0 || pRprt.y != 0 || pRprt.h != 0 || pRprt.v != 0 || (trkBtnState != pBtn)) pointing_device_set_report(pRprt); trkBtnState = pBtn; #endif // Then the keyboard if (mcp23018_status) { // if there was an error if (++mcp23018_reset_loop == 0) { // if (++mcp23018_reset_loop >= 1300) { // since mcp23018_reset_loop is 8 bit - we'll try to reset once in 255 matrix scans // this will be approx bit more frequent than once per second print("trying to reset mcp23018\n"); mcp23018_status = init_mcp23018(); if (mcp23018_status) { print("left side not responding\n"); } else { print("left side attached\n"); } } } bool changed = false; for (uint8_t i = 0; i < MATRIX_ROWS_PER_SIDE; i++) { // select rows from left and right hands uint8_t left_index = i; uint8_t right_index = i + MATRIX_ROWS_PER_SIDE; select_row(left_index); select_row(right_index); // we don't need a 30us delay anymore, because selecting a // left-hand row requires more than 30us for i2c. changed |= store_raw_matrix_row(left_index); changed |= store_raw_matrix_row(right_index); unselect_rows(); } debounce(raw_matrix, matrix, MATRIX_ROWS, changed); matrix_scan_quantum(); enableInterrupts(); #ifdef DEBUG_MATRIX for (uint8_t c = 0; c < MATRIX_COLS; c++) for (uint8_t r = 0; r < MATRIX_ROWS; r++) if (matrix_is_on(r, c)) xprintf("r:%d c:%d \n", r, c); #endif return 1; } bool matrix_is_modified(void) // deprecated and evidently not called. { return true; } inline bool matrix_is_on(uint8_t row, uint8_t col) { return (matrix[row] & ((matrix_row_t)1 << col)); } inline matrix_row_t matrix_get_row(uint8_t row) { return matrix[row]; } void matrix_print(void) { print("\nr/c 0123456789ABCDEF\n"); for (uint8_t row = 0; row < MATRIX_ROWS; row++) { print_hex8(row); print(": "); print_bin_reverse16(matrix_get_row(row)); print("\n"); } } uint8_t matrix_key_count(void) { uint8_t count = 0; for (uint8_t i = 0; i < MATRIX_ROWS; i++) { count += bitpop16(matrix[i]); } return count; } // Remember this means ROWS static void init_cols(void) { // init on mcp23018 // not needed, already done as part of init_mcp23018() // Input with pull-up(DDR:0, PORT:1) DDRF &= ~FMASK; PORTF |= FMASK; } static matrix_row_t read_cols(uint8_t row) { if (row < 7) { if (mcp23018_status) { // if there was an error return 0; } else { uint8_t data = 0; mcp23018_status = i2c_start(I2C_ADDR_WRITE, I2C_TIMEOUT); if (mcp23018_status) goto out; mcp23018_status = i2c_write(GPIOB, I2C_TIMEOUT); if (mcp23018_status) goto out; mcp23018_status = i2c_start(I2C_ADDR_READ, I2C_TIMEOUT); if (mcp23018_status) goto out; mcp23018_status = i2c_read_nack(I2C_TIMEOUT); if (mcp23018_status < 0) goto out; data = ~((uint8_t)mcp23018_status); mcp23018_status = I2C_STATUS_SUCCESS; out: i2c_stop(); #ifdef DEBUG_MATRIX if (data != 0x00) xprintf("I2C: %d\n", data); #endif return data; } } else { /* read from teensy * bitmask is 0b0111001, but we want the lower four * we'll return 1s for the top two, but that's harmless. */ // So I need to confuckulate all this //return ~(((PIND & DMASK) >> 1 | ((PINC & CMASK) >> 6) | (PIN))); //return ~((PINF & 0x03) | ((PINF & 0xF0) >> 2)); return ~( (((PINF & ROW4) >> 1) | ((PINF & (ROW1 | ROW2 | ROW3)) >> 3)) & 0xF); } } // Row pin configuration static void unselect_rows(void) { // no need to unselect on mcp23018, because the select step sets all // the other row bits high, and it's not changing to a different // direction // Hi-Z(DDR:0, PORT:0) to unselect DDRB &= ~(BMASK | TRKMASK); PORTB &= ~(BMASK); DDRC &= ~CMASK; PORTC &= ~CMASK; DDRD &= ~DMASK; PORTD &= ~DMASK; // Fix trashing of DDRB for TB PORTB |= TRKMASK; } static void select_row(uint8_t row) { if (row < 7) { // select on mcp23018 if (mcp23018_status) { // do nothing on error } else { // set active row low : 0 // set other rows hi-Z : 1 mcp23018_status = i2c_start(I2C_ADDR_WRITE, I2C_TIMEOUT); if (mcp23018_status) goto out; mcp23018_status = i2c_write(GPIOA, I2C_TIMEOUT); if (mcp23018_status) goto out; mcp23018_status = i2c_write(0xFF & ~(1<<row), I2C_TIMEOUT); if (mcp23018_status) goto out; out: i2c_stop(); } } else { // Output low(DDR:1, PORT:0) to select switch (row) { case 7: DDRB |= COL7; PORTB &= ~COL7; break; case 8: DDRB |= COL8; PORTB &= ~COL8; break; case 9: DDRB |= COL9; PORTB &= ~COL9; break; case 10: DDRB |= COL10; PORTB &= ~COL10; break; case 11: DDRD |= COL11; PORTD &= ~COL11; break; case 12: DDRD |= COL12; PORTD &= ~COL12; break; case 13: DDRC |= COL13; PORTC &= ~COL13; break; } } } // Trackball Interrupts static void enableInterrupts(void) { #ifdef BALLER // Set interrupt mask // Set port defs DDRB &= ~TRKMASK; PORTB |= TRKMASK; DDRE &= ~TRKBTN; PORTE |= TRKBTN; // Interrupt shenanigans //EIMSK |= (1 << PCIE0); PCMSK0 |= TRKMASK; PCICR |= (1 << PCIE0); sei(); #endif return; } #ifdef BALLER ISR (PCINT0_vect) { // Don't get fancy, we're in a interrupt here // PCINT reports a interrupt for a change on the bus // We hand the button at scantime for debounce volatile uint8_t pState = PINB & TRKMASK; if ((pState & TRKUP) != (trkState & TRKUP)) tbUpCnt++; if ((pState & TRKDN) != (trkState & TRKDN)) tbDnCnt++; if ((pState & TRKLT) != (trkState & TRKLT)) tbLtCnt++; if ((pState & TRKRT) != (trkState & TRKRT)) tbRtCnt++; trkState = pState; } #endif