matrix.c (5995B)
1 /* Copyright 2022 @ Keychron (https://www.keychron.com) 2 * 3 * This program is free software: you can redistribute it and/or modify 4 * it under the terms of the GNU General Public License as published by 5 * the Free Software Foundation, either version 2 of the License, or 6 * (at your option) any later version. 7 * 8 * This program is distributed in the hope that it will be useful, 9 * but WITHOUT ANY WARRANTY; without even the implied warranty of 10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 11 * GNU General Public License for more details. 12 * 13 * You should have received a copy of the GNU General Public License 14 * along with this program. If not, see <http://www.gnu.org/licenses/>. 15 */ 16 17 #include "matrix.h" 18 #include "atomic_util.h" 19 #include <string.h> 20 21 // Pin connected to DS of 74HC595 22 #define DATA_PIN C15 23 // Pin connected to SH_CP of 74HC595 24 #define CLOCK_PIN A1 25 // Pin connected to ST_CP of 74HC595 26 #define LATCH_PIN A0 27 28 #ifdef MATRIX_ROW_PINS 29 static pin_t row_pins[MATRIX_ROWS] = MATRIX_ROW_PINS; 30 #endif // MATRIX_ROW_PINS 31 #ifdef MATRIX_COL_PINS 32 static pin_t col_pins[MATRIX_COLS] = MATRIX_COL_PINS; 33 #endif // MATRIX_COL_PINS 34 35 #define ROWS_PER_HAND (MATRIX_ROWS) 36 37 static inline void gpio_atomic_set_pin_output_low(pin_t pin) { 38 ATOMIC_BLOCK_FORCEON { 39 gpio_set_pin_output(pin); 40 gpio_write_pin_low(pin); 41 } 42 } 43 44 static inline void gpio_atomic_set_pin_output_high(pin_t pin) { 45 ATOMIC_BLOCK_FORCEON { 46 gpio_set_pin_output(pin); 47 gpio_write_pin_high(pin); 48 } 49 } 50 51 static inline void gpio_atomic_set_pin_input_high(pin_t pin) { 52 ATOMIC_BLOCK_FORCEON { 53 gpio_set_pin_input_high(pin); 54 } 55 } 56 57 static inline uint8_t readMatrixPin(pin_t pin) { 58 if (pin != NO_PIN) { 59 return gpio_read_pin(pin); 60 } else { 61 return 1; 62 } 63 } 64 65 static void shiftOut(uint8_t dataOut) { 66 for (uint8_t i = 0; i < 8; i++) { 67 if (dataOut & 0x1) { 68 gpio_atomic_set_pin_output_high(DATA_PIN); 69 } else { 70 gpio_atomic_set_pin_output_low(DATA_PIN); 71 } 72 dataOut = dataOut >> 1; 73 gpio_atomic_set_pin_output_high(CLOCK_PIN); 74 gpio_atomic_set_pin_output_low(CLOCK_PIN); 75 } 76 gpio_atomic_set_pin_output_high(LATCH_PIN); 77 gpio_atomic_set_pin_output_low(LATCH_PIN); 78 } 79 80 static void shiftout_single(uint8_t data) { 81 if (data & 0x1) { 82 gpio_atomic_set_pin_output_high(DATA_PIN); 83 } else { 84 gpio_atomic_set_pin_output_low(DATA_PIN); 85 } 86 gpio_atomic_set_pin_output_high(CLOCK_PIN); 87 gpio_atomic_set_pin_output_low(CLOCK_PIN); 88 89 gpio_atomic_set_pin_output_high(LATCH_PIN); 90 gpio_atomic_set_pin_output_low(LATCH_PIN); 91 } 92 93 static bool select_col(uint8_t col) { 94 pin_t pin = col_pins[col]; 95 96 if (pin != NO_PIN) { 97 gpio_atomic_set_pin_output_low(pin); 98 return true; 99 } else { 100 if (col == 10) { 101 shiftout_single(0x00); 102 } else { 103 shiftout_single(0x01); 104 } 105 return true; 106 } 107 return false; 108 } 109 110 static void unselect_col(uint8_t col) { 111 pin_t pin = col_pins[col]; 112 113 if (pin != NO_PIN) { 114 #ifdef MATRIX_UNSELECT_DRIVE_HIGH 115 gpio_atomic_set_pin_output_high(pin); 116 #else 117 gpio_atomic_set_pin_input_high(pin); 118 #endif 119 } else { 120 if (col == (MATRIX_COLS - 1)) 121 gpio_atomic_set_pin_output_high(CLOCK_PIN); 122 gpio_atomic_set_pin_output_low(CLOCK_PIN); 123 gpio_atomic_set_pin_output_high(LATCH_PIN); 124 gpio_atomic_set_pin_output_low(LATCH_PIN); 125 } 126 } 127 128 static void unselect_cols(void) { 129 // unselect column pins 130 for (uint8_t x = 0; x < MATRIX_COLS; x++) { 131 pin_t pin = col_pins[x]; 132 if (pin != NO_PIN) { 133 #ifdef MATRIX_UNSELECT_DRIVE_HIGH 134 gpio_atomic_set_pin_output_high(pin); 135 #else 136 gpio_atomic_set_pin_input_high(pin); 137 #endif 138 } 139 if (x == (MATRIX_COLS - 1)) 140 // unselect shift Register 141 shiftOut(0xFF); 142 } 143 } 144 145 static void matrix_init_pins(void) { 146 unselect_cols(); 147 for (uint8_t x = 0; x < MATRIX_ROWS; x++) { 148 if (row_pins[x] != NO_PIN) { 149 gpio_atomic_set_pin_input_high(row_pins[x]); 150 } 151 } 152 } 153 154 static void matrix_read_rows_on_col(matrix_row_t current_matrix[], uint8_t current_col, matrix_row_t row_shifter) { 155 bool key_pressed = false; 156 157 // Select col 158 if (!select_col(current_col)) { // select col 159 return; // skip NO_PIN col 160 } 161 162 if (current_col < 10) { 163 matrix_output_select_delay(); 164 } else { 165 for (int8_t cycle = 4; cycle > 0; cycle--) { 166 matrix_output_select_delay(); // 0.25us 167 matrix_output_select_delay(); 168 matrix_output_select_delay(); 169 matrix_output_select_delay(); 170 } 171 } 172 173 // For each row... 174 for (uint8_t row_index = 0; row_index < ROWS_PER_HAND; row_index++) { 175 // Check row pin state 176 if (readMatrixPin(row_pins[row_index]) == 0) { 177 // Pin LO, set col bit 178 current_matrix[row_index] |= row_shifter; 179 key_pressed = true; 180 } else { 181 // Pin HI, clear col bit 182 current_matrix[row_index] &= ~row_shifter; 183 } 184 185 } 186 187 // // Unselect col 188 unselect_col(current_col); 189 matrix_output_unselect_delay(current_col, key_pressed); // wait for all Row signals to go HIGH 190 } 191 192 void matrix_init_custom(void) { 193 // initialize key pins 194 matrix_init_pins(); 195 } 196 197 bool matrix_scan_custom(matrix_row_t current_matrix[]) { 198 matrix_row_t curr_matrix[MATRIX_ROWS] = {0}; 199 200 // Set col, read rows 201 matrix_row_t row_shifter = MATRIX_ROW_SHIFTER; 202 for (uint8_t current_col = 0; current_col < MATRIX_COLS; current_col++, row_shifter <<= 1) { 203 matrix_read_rows_on_col(curr_matrix, current_col, row_shifter); 204 } 205 206 bool changed = memcmp(current_matrix, curr_matrix, sizeof(curr_matrix)) != 0; 207 if (changed) memcpy(current_matrix, curr_matrix, sizeof(curr_matrix)); 208 209 return changed; 210 }