1 files changed, 196 insertions, 0 deletions
diff --git a/keyboards/ergodox_stm32/matrix.c b/keyboards/ergodox_stm32/matrix.c
new file mode 100644
index 0000000000..5a280ee17d
--- /dev/null
+++ b/keyboards/ergodox_stm32/matrix.c
@@ -0,0 +1,196 @@
+#include <stdint.h>
+#include <stdbool.h>
+#include <string.h>
+#include <hal.h>
+#include "timer.h"
+#include "wait.h"
+#include "print.h"
+#include "matrix.h"
+#include "i2c_master.h"
+#include QMK_KEYBOARD_H
+
+#ifndef DEBOUNCE
+#define DEBOUNCE 10
+#endif
+
+//#define DEBUG_MATRIX_SCAN_RATE
+
+//#ifdef DEBUG_MATRIX_SCAN_RATE
+//uint32_t matrix_timer;
+//uint32_t matrix_scan_count;
+//#endif
+
+static uint8_t mcp23017_reset_loop = 0;
+
+volatile matrix_row_t matrix[MATRIX_ROWS];
+volatile matrix_row_t raw_matrix[MATRIX_ROWS];
+volatile uint8_t debounce_matrix[MATRIX_ROWS * MATRIX_COLS];
+
+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 init_rows(void);
+
+__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();
+}
+
+void matrix_init(void) {
+  mcp23017_status = init_mcp23017();
+  (void) mcp23017_reset_loop;
+  init_rows();
+  unselect_rows();
+  init_cols();
+
+
+  for (uint8_t i = 0; i < MATRIX_ROWS; i++) {
+    matrix[i] = 0;
+    raw_matrix[i] = 0;
+    for (uint8_t j = 0; j < MATRIX_COLS; ++j) {
+      debounce_matrix[i * MATRIX_COLS + j] = 0;
+    }
+  }
+  matrix_init_quantum();
+}
+
+void matrix_power_up(void) {
+  mcp23017_status = init_mcp23017();
+
+  init_rows();
+  unselect_rows();
+  init_cols();
+
+  for (uint8_t i = 0; i < MATRIX_ROWS; i++) {
+    matrix[i] = 0;
+  }
+}
+
+matrix_row_t debounce_mask(matrix_row_t rawcols, uint8_t row) {
+  matrix_row_t result = 0;
+  matrix_row_t change = rawcols ^raw_matrix[row];
+  raw_matrix[row] = rawcols;
+  for (uint8_t i = 0; i < MATRIX_COLS; ++i) {
+    if (debounce_matrix[row * MATRIX_COLS + i]) {
+      --debounce_matrix[row * MATRIX_COLS + i];
+    } else {
+      result |= (1 << i);
+    }
+    if (change & (1 << i)) {
+      debounce_matrix[row * MATRIX_COLS + i] = DEBOUNCE;
+    }
+  }
+  return result;
+}
+
+matrix_row_t debounce_read_cols(uint8_t row) {
+  // Read the row without debouncing filtering and store it for later usage.
+  matrix_row_t cols = read_cols(row);
+  // Get the Debounce mask.
+  matrix_row_t mask = debounce_mask(cols, row);
+  // debounce the row and return the result.
+  return (cols & mask) | (matrix[row] & ~mask);;
+}
+
+uint8_t matrix_scan(void) {
+  if (mcp23017_status) {
+    if (++mcp23017_reset_loop == 0) {
+      mcp23017_status = init_mcp23017();
+        if (!mcp23017_status) {
+            ergodox_blink_all_leds();
+        }
+    }
+  }
+  for (uint8_t i = 0; i < MATRIX_ROWS_PER_SIDE; i++) {
+    select_row(i);
+    select_row(i + MATRIX_ROWS_PER_SIDE);
+
+    matrix[i] = debounce_read_cols(i);
+    matrix[i + MATRIX_ROWS_PER_SIDE] = debounce_read_cols(i + MATRIX_ROWS_PER_SIDE);
+
+    unselect_rows();
+  }
+  matrix_scan_quantum();
+  return 0;
+}
+
+bool matrix_is_modified(void) {
+  return true;
+}
+
+inline
+bool matrix_is_on(uint8_t row, uint8_t col) {
+  return (matrix[row] & (1 << col));
+}
+
+inline
+matrix_row_t matrix_get_row(uint8_t row) {
+  return matrix[row];
+}
+
+void matrix_print(void) {
+}
+
+static matrix_row_t read_cols(uint8_t row) {
+  if (row < MATRIX_ROWS_PER_SIDE) {
+    uint8_t data = 0xFF;
+    if (!mcp23017_status) {
+      uint8_t regAddr = I2C_GPIOB;
+      mcp23017_status = i2c_readReg(I2C_ADDR, regAddr, &data, 1, 10);
+    }
+    if (mcp23017_status) {
+      return 0;
+    }
+    return (~data) & 0x3F;
+  } else {
+      uint8_t data_p = (GPIOB -> IDR);
+      uint8_t data = data_p;
+    return ((~data) & 0x3f);
+  }
+}
+
+static void init_cols(void) {
+  palSetPadMode(GPIOB, 0, PAL_MODE_INPUT_PULLUP);
+  palSetPadMode(GPIOB, 1, PAL_MODE_INPUT_PULLUP);
+  palSetPadMode(GPIOB, 2, PAL_MODE_INPUT_PULLUP);
+  palSetPadMode(GPIOB, 3, PAL_MODE_INPUT_PULLUP);
+  palSetPadMode(GPIOB, 4, PAL_MODE_INPUT_PULLUP);
+  palSetPadMode(GPIOB, 5, PAL_MODE_INPUT_PULLUP);
+}
+
+static void init_rows(void) {
+  palSetPadMode(GPIOB, 8, PAL_MODE_OUTPUT_PUSHPULL);
+  palSetPadMode(GPIOB, 9, PAL_MODE_OUTPUT_PUSHPULL);
+  palSetPadMode(GPIOB, 10, PAL_MODE_OUTPUT_PUSHPULL);
+  palSetPadMode(GPIOB, 11, PAL_MODE_OUTPUT_PUSHPULL);
+  palSetPadMode(GPIOB, 12, PAL_MODE_OUTPUT_PUSHPULL);
+  palSetPadMode(GPIOB, 13, PAL_MODE_OUTPUT_PUSHPULL);
+  palSetPadMode(GPIOB, 14, PAL_MODE_OUTPUT_PUSHPULL);
+}
+
+static void unselect_rows(void) {
+  GPIOB->BSRR = 0b1111111 << 8;
+}
+
+static void select_row(uint8_t row) {
+  if (row < MATRIX_ROWS_PER_SIDE) {
+    if (!mcp23017_status) {
+      uint8_t data = (0xFF & ~(1 << row));
+      mcp23017_status = i2c_writeReg(I2C_ADDR, I2C_GPIOA, &data, 1, 10);
+    }
+  } else {
+    GPIOB->BRR = 0x1 << (row+1);
+  }
+}