src/adafruit_blinka/microcontroller/tegra/t210/pin.py

   1 import sys
   2 import atexit
   3 sys.path.append("/opt/nvidia/jetson-gpio/lib/python")
   4 sys.path.append("/opt/nvidia/jetson-gpio/lib/python/Jetson/GPIO")
   5 import Jetson.GPIO as GPIO
   6 GPIO.setmode(GPIO.TEGRA_SOC)
   7 GPIO.setwarnings(False)   # shh!
   8
   9 # Pins dont exist in CPython so...lets make our own!
  10 class Pin:
  11     IN = 0
  12     OUT = 1
  13     LOW = 0
  14     HIGH = 1
  15     PULL_NONE = 0
  16     PULL_UP = 1
  17     PULL_DOWN = 2
  18
  19     id = None
  20     _value = LOW
  21     _mode = IN
  22
  23     def __init__(self, bcm_number):
  24         self.id = bcm_number
  25
  26     def __repr__(self):
  27         return str(self.id)
  28
  29     def __eq__(self, other):
  30         return self.id == other
  31
  32     def init(self, mode=IN, pull=None):
  33         if mode != None:
  34             if mode == self.IN:
  35                 self._mode = self.IN
  36                 GPIO.setup(self.id, GPIO.IN)
  37             elif mode == self.OUT:
  38                 self._mode = self.OUT
  39                 GPIO.setup(self.id, GPIO.OUT)
  40             else:
  41                 raise RuntimeError("Invalid mode for pin: %s" % self.id)
  42         if pull != None:
  43             if self._mode != self.IN:
  44                 raise RuntimeError("Cannot set pull resistor on output")
  45             if pull == self.PULL_UP:
  46                 GPIO.setup(self.id, GPIO.IN, pull_up_down=GPIO.PUD_UP)
  47             elif pull == self.PULL_DOWN:
  48                 GPIO.setup(self.id, GPIO.IN, pull_up_down=GPIO.PUD_DOWN)
  49             else:
  50                 raise RuntimeError("Invalid pull for pin: %s" % self.id)
  51
  52     def value(self, val=None):
  53         if val != None:
  54             if val == self.LOW:
  55                 self._value = val
  56                 GPIO.output(self.id, val)
  57             elif val == self.HIGH:
  58                 self._value = val
  59                 GPIO.output(self.id, val)
  60             else:
  61                 raise RuntimeError("Invalid value for pin")
  62         else:
  63             return GPIO.input(self.id)
  64
  65     @atexit.register
  66     def cleanup():
  67         print("Exiting... \nCleaning up pins")
  68         GPIO.cleanup()
  69
  70 # Cannot be used as GPIO
  71 SDA = Pin('GEN1_I2C_SDA')
  72 SCL = Pin('GEN1_I2C_SCL')
  73 SDA_1 = Pin('GEN2_I2C_SDA')
  74 SCL_1 = Pin('GEN2_I2C_SCL')
  75
  76 # These pins are native to TX1
  77 BB03 = Pin('GPIO_X1_AUD')
  78 X02 = Pin('MOTION_INT')
  79 H07 = Pin('AP_WAKE_NFC')
  80 E04 = Pin('DMIC3_CLK')
  81 U03 = Pin('UART1_CTS')
  82 U02 = Pin('UART1_RTS')
  83 B03 = Pin('DAP1_SCLK')
  84 B00 = Pin('DAP1_FS')
  85 B01 = Pin('DAP1_DIN')
  86 B02 = Pin('DAP1_DOUT')
  87 P17 = Pin('GPIO_EXP_P17')
  88 E05 = Pin('DMIC3_DAT')
  89 X00 = Pin('MODEM_WAKE_AP')
  90 P16 = Pin('GPIO_EXP_P16')
  91 X03 = Pin('ALS_PROX_INT')
  92
  93 # These pins are native to NANO
  94 S05 = Pin('CAM_AF_EN')
  95 Z00 = Pin('GPIO_PZ0')
  96 V00 = Pin('LCD_BL_PW')
  97 G03 = Pin('UART2_CTS')
  98 G02 = Pin('UART2_RTS')
  99 J07 = Pin('DAP4_SCLK')
 100 J04 = Pin('DAP4_FS')
 101 J05 = Pin('DAP4_DIN')
 102 J06 = Pin('DAP4_DOUT')
 103 Y02 = Pin('LCD_TE')
 104 DD00 = Pin('SPI2_CS1')
 105 B07 = Pin('SPI2_CS0')
 106 B05 = Pin('SPI2_MISO')
 107 B04 = Pin('SPI2_MOSI')
 108 B06 = Pin('SPI2_SCK')
 109
 110 # These pins are shared across T210
 111 BB00 = Pin('AUD_MCLK')
 112 C04 = Pin('SPI1_CS1')
 113 C03 = Pin('SPI1_CS0')
 114 C01 = Pin('SPI1_MISO')
 115 C00 = Pin('SPI1_MOSI')
 116 C02 = Pin('SPI1_SCK')
 117 E06 = Pin('GPIO_PE6')
 118
 119 i2cPorts = (
 120     (0, SCL, SDA), (1, SCL_1, SDA_1),
 121 )
 122
 123 # ordered as spiId, sckId, mosiId, misoId
 124 spiPorts = ((0, C02, C00, C01), (1, B06, B04, B05))