add support for jetson thor

[Adafruit_Blinka-hackapet.git] / src / adafruit_blinka / microcontroller / tegra / t264 / pin.py

diff --git a/src/adafruit_blinka/microcontroller/tegra/t264/pin.py b/src/adafruit_blinka/microcontroller/tegra/t264/pin.py
new file mode 100644 (file)
index 0000000..63b72fc
--- /dev/null
+++ b/src/adafruit_blinka/microcontroller/tegra/t264/pin.py
@@ -0,0 +1,118 @@
+# SPDX-FileCopyrightText: 2025 Gautham Srinivasan for NVIDIA
+#
+# SPDX-License-Identifier: MIT
+
+"""Tegra T264 pin names"""
+import atexit
+from Jetson import GPIO
+
+GPIO.setmode(GPIO.TEGRA_SOC)
+GPIO.setwarnings(False)  # shh!
+
+
+class Pin:
+    """Pins dont exist in CPython so...lets make our own!"""
+
+    IN = 0
+    OUT = 1
+    LOW = 0
+    HIGH = 1
+    PULL_NONE = 0
+    PULL_UP = 1
+    PULL_DOWN = 2
+
+    id = None
+    _value = LOW
+    _mode = IN
+
+    def __init__(self, bcm_number):
+        self.id = bcm_number
+
+    def __repr__(self):
+        return str(self.id)
+
+    def __eq__(self, other):
+        return self.id == other
+
+    def init(self, mode=IN, pull=None):
+        """Initialize the Pin"""
+        if mode is not None:
+            if mode == self.IN:
+                self._mode = self.IN
+                GPIO.setup(self.id, GPIO.IN)
+            elif mode == self.OUT:
+                self._mode = self.OUT
+                GPIO.setup(self.id, GPIO.OUT)
+            else:
+                raise RuntimeError("Invalid mode for pin: %s" % self.id)
+        if pull is not None:
+            if self._mode != self.IN:
+                raise RuntimeError("Cannot set pull resistor on output")
+            if pull == self.PULL_UP:
+                GPIO.setup(self.id, GPIO.IN, pull_up_down=GPIO.PUD_UP)
+            elif pull == self.PULL_DOWN:
+                GPIO.setup(self.id, GPIO.IN, pull_up_down=GPIO.PUD_DOWN)
+            else:
+                raise RuntimeError("Invalid pull for pin: %s" % self.id)
+
+    def value(self, val=None):
+        """Set or return the Pin Value"""
+        if val is not None:
+            if val == self.LOW:
+                self._value = val
+                GPIO.output(self.id, val)
+                return None
+            if val == self.HIGH:
+                self._value = val
+                GPIO.output(self.id, val)
+                return None
+            raise RuntimeError("Invalid value for pin")
+        return GPIO.input(self.id)
+
+    # pylint: disable=no-method-argument
+    @atexit.register
+    def cleanup():
+        """Clean up pins"""
+        print("Exiting... \nCleaning up pins")
+        GPIO.cleanup()
+
+    # pylint: enable=no-method-argument
+
+
+# Cannot be used as GPIO
+SDA = Pin("GP16_I2C8_DAT")  # I2C4
+SCL = Pin("GP81_I2C9_CLK")
+SDA_1 = Pin("GP14_I2C2_DAT")  # I2C2
+SCL_1 = Pin("GP13_I2C2_CLK")
+
+# Jetson Thor
+L06 = Pin("GP130")
+M04 = Pin("GP136_UART9_RTS_N")
+V06 = Pin("GP184_DAP2_CLK")
+M00 = Pin("GP132_PWM9")
+F07 = Pin("GP257_PWM2")
+DD03 = Pin("GP21")
+U07 = Pin("GP177")
+K01 = Pin("GP117_SPI1_MOSI")
+K00 = Pin("GP116_SPI1_MISO")
+U00 = Pin("GP170")
+J07 = Pin("GP115_SPI1_CLK")
+K02 = Pin("GP118_SPI1_CS0_N")
+K03 = Pin("GP119_SPI1_CS1_N")
+AD01 = Pin("GP211_CAN2_DIN")
+AD00 = Pin("GP210_CAN2_DOUT")
+DD04 = Pin("GGP22_SOCKET_ID_STRA")
+AE00 = Pin("GP215_CAN3_DOUT")
+W01 = Pin("GP187_DAP2_FS")
+M05 = Pin("GP137_UART9_CTS_N")
+AE01 = Pin("GP216_CAN3_DIN")
+W00 = Pin("GP186_DAP2_DIN")
+V07 = Pin("GP185_DAP2_DOUT")
+
+i2cPorts = (
+    (7, SCL, SDA),
+    (1, SCL_1, SDA_1),
+)
+
+# ordered as spiId, sckId, mosiId, misoId
+spiPorts = ((K02, J07, K01, K00),)