Add functionality to set eGPIO pin in/out mode

emfcamp · Jun 5, 2024 · bf55c5a · bf55c5a
1 parent 6e1ddc4
commit bf55c5a
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/modules/tildagon/pins.py b/modules/tildagon/pins.py
@@ -41,6 +41,12 @@ def __init__(self, pin_name, mode=-1):
         self.name = pin_name
         self.pin = HEXPANSION_GPIOS[pin_name]
         self.mode = mode
+        if self.mode in (machine.Pin.IN, -1):
+            tildagonos.set_pin_mode(self.pin, machine.Pin.IN)
+        elif self.mode == machine.Pin.OUT:
+            tildagonos.set_pin_mode(self.pin, machine.Pin.OUT)
+        else:
+            raise ValueError("Invalid pin mode")
 
     def on(self):
         self.value(1)

diff --git a/modules/tildagonos.py b/modules/tildagonos.py
@@ -64,6 +64,19 @@ def init_gpio(self):
         # chip C - switch mode to push-pull
         self.system_i2c.writeto_mem(0x5A, 0x11, bytes([0x10]))
 
+    def set_pin_mode(self, pin, mode=Pin.IN):
+        portstates = list(map(int, self.system_i2c.readfrom_mem(pin[0], 0x04, 2)))
+        if mode == Pin.IN:
+            self.system_i2c.writeto_mem(
+                pin[0], 0x04 + pin[1], bytes([portstates[pin[1]] | pin[2]])
+            )
+        elif mode == Pin.OUT:
+            self.system_i2c.writeto_mem(
+                pin[0], 0x04 + pin[1], bytes([portstates[pin[1]] & (pin[2] ^ 0xFF)])
+            )
+        else:
+            raise ValueError("Invalid pin mode")
+
     def set_egpio_pin(self, pin, state):
         portstates = list(map(int, self.system_i2c.readfrom_mem(pin[0], 0x02, 2)))
         if state: