hardware/ErgoDox: Improved row selection

Improved the ATMega parts of row toggling, similar to how `ATMegaKeyboard` does
it, and separated it from selecting extender rows. This results in a tiny
increase in speed, and better clarity.

Signed-off-by: Gergely Nagy <algernon@keyboard.io>

src/kaleidoscope/hardware/ez/ErgoDox.cpp

@@ -97,13 +97,13 @@ void __attribute__((optimize(3))) ErgoDox::readMatrix() {
   scanner_.reattachExpanderOnError();
 
   for (uint8_t row = 0; row < ROWS / 2; row++) {
-    scanner_.selectRow(row);
+    scanner_.selectExtenderRow(row);
-    scanner_.selectRow(row + ROWS / 2);
+    scanner_.toggleATMegaRow(row);
 
     readMatrixRow(row);
     readMatrixRow(row + ROWS / 2);
 
-    scanner_.unselectRows();
+    scanner_.toggleATMegaRow(row);
   }
 }
 

src/kaleidoscope/hardware/ez/ErgoDox/ErgoDoxScanner.cpp

@@ -85,23 +85,24 @@ out:
   return status;
 }
 
-void
-ErgoDoxScanner::initCols() {
-  DDRF  &= ~(1 << 7 | 1 << 6 | 1 << 5 | 1 << 4 | 1 << 1 | 1 << 0);
-  PORTF |= (1 << 7 | 1 << 6 | 1 << 5 | 1 << 4 | 1 << 1 | 1 << 0);
-}
-
 void
 ErgoDoxScanner::begin() {
   expander_error_ = initExpander();
 
-  unselectRows();
+  // Init columns
-  initCols();
+  DDRF  &= ~(1 << 7 | 1 << 6 | 1 << 5 | 1 << 4 | 1 << 1 | 1 << 0);
+  PORTF |= (1 << 7 | 1 << 6 | 1 << 5 | 1 << 4 | 1 << 1 | 1 << 0);
+
+  // Init rows
+  DDRB  |= (1 << 0 | 1 << 1 | 1 << 2 | 1 << 3);
+  PORTB |= (1 << 0 | 1 << 1 | 1 << 2 | 1 << 3);
+  DDRD  |= (1 << 2 | 1 << 3);
+  PORTD |= (1 << 2 | 1 << 3);
+  DDRC  |= (1 << 6);
+  PORTC |= (1 << 6);
 }
 
-void __attribute__((optimize(3)))
+void __attribute__((optimize(3))) ErgoDoxScanner::selectExtenderRow(int row) {
-ErgoDoxScanner::selectRow(int row) {
-  if (row < 7) {
   if (!expander_error_) {
     expander_error_ = i2c_start(I2C_ADDR_WRITE);
     if (expander_error_)
@@ -115,49 +116,33 @@ ErgoDoxScanner::selectRow(int row) {
 out:
     i2c_stop();
   }
-  } else {
+}
+
+void __attribute__((optimize(3))) ErgoDoxScanner::toggleATMegaRow(int row) {
   switch (row) {
-    case 7:
+    case 0:
-      DDRB  |= (1 << 0);
+      PORTB ^= (1 << 0);
-      PORTB &= ~(1 << 0);
       break;
-    case 8:
+    case 1:
-      DDRB  |= (1 << 1);
+      PORTB ^= (1 << 1);
-      PORTB &= ~(1 << 1);
       break;
-    case 9:
+    case 2:
-      DDRB  |= (1 << 2);
+      PORTB ^= (1 << 2);
-      PORTB &= ~(1 << 2);
       break;
-    case 10:
+    case 3:
-      DDRB  |= (1 << 3);
+      PORTB ^= (1 << 3);
-      PORTB &= ~(1 << 3);
       break;
-    case 11:
+    case 4:
-      DDRD  |= (1 << 2);
+      PORTD ^= (1 << 2);
-      PORTD &= ~(1 << 3);
       break;
-    case 12:
+    case 5:
-      DDRD  |= (1 << 3);
+      PORTD ^= (1 << 3);
-      PORTD &= ~(1 << 3);
       break;
-    case 13:
+    case 6:
-      DDRC  |= (1 << 6);
+      PORTC ^= (1 << 6);
-      PORTC &= ~(1 << 6);
       break;
   }
 }
-}
-
-void __attribute__((optimize(3)))
-ErgoDoxScanner::unselectRows() {
-  DDRB  &= ~(1 << 0 | 1 << 1 | 1 << 2 | 1 << 3);
-  PORTB &= ~(1 << 0 | 1 << 1 | 1 << 2 | 1 << 3);
-  DDRD  &= ~(1 << 2 | 1 << 3);
-  PORTD &= ~(1 << 2 | 1 << 3);
-  DDRC  &= ~(1 << 6);
-  PORTC &= ~(1 << 6);
-}
 
 uint8_t __attribute__((optimize(3)))
 ErgoDoxScanner::readCols(int row) {

src/kaleidoscope/hardware/ez/ErgoDox/ErgoDoxScanner.h

@@ -38,9 +38,8 @@ class ErgoDoxScanner {
   ErgoDoxScanner() {}
 
   void begin();
-  void initCols();
+  void toggleATMegaRow(int row);
-  void selectRow(int row);
+  void selectExtenderRow(int row);
-  void unselectRows();
   uint8_t readCols(int row);
 
   void reattachExpanderOnError();