src/adafruit_blinka/microcontroller/starfive/JH7110/pin.py

   1 # SPDX-FileCopyrightText: 2024 Vladimir Shtarev
   2 #
   3 # SPDX-License-Identifier: MIT
   4 """A Pin class for use with StarFive JH7110."""
   5
   6 import VisionFive.gpio as GPIO
   7
   8 GPIO.setmode(GPIO.BOARD)
   9
  10
  11 class Pin:
  12     """Pins don't exist in CPython so...lets make our own!"""
  13
  14     IN = 0
  15     OUT = 1
  16     LOW = 0
  17     HIGH = 1
  18     PULL_NONE = 0
  19     PULL_UP = 1
  20     PULL_DOWN = 2
  21
  22     id = None
  23     _value = LOW
  24     _mode = IN
  25
  26     def __init__(self, bcm_number):
  27         self.id = bcm_number
  28
  29     def __repr__(self):
  30         return str(self.id)
  31
  32     def __eq__(self, other):
  33         return self.id == other
  34
  35     def init(self, mode=IN, pull=None):
  36         """Initialize the Pin"""
  37         if mode is not None:
  38             if mode == self.IN:
  39                 self._mode = self.IN
  40                 GPIO.setup(self.id, GPIO.IN)
  41             elif mode == self.OUT:
  42                 self._mode = self.OUT
  43                 GPIO.setup(self.id, GPIO.OUT)
  44             else:
  45                 raise RuntimeError("Invalid mode for pin: %s" % self.id)
  46         if pull is not None:
  47             if self._mode != self.IN:
  48                 raise RuntimeError("Cannot set pull resistor on output")
  49             if pull == self.PULL_UP:
  50                 GPIO.setup(self.id, GPIO.IN, pull_up_down=GPIO.PUD_UP)
  51             elif pull == self.PULL_DOWN:
  52                 GPIO.setup(self.id, GPIO.IN, pull_up_down=GPIO.PUD_DOWN)
  53             else:
  54                 raise RuntimeError("Invalid pull for pin: %s" % self.id)
  55
  56     def value(self, val=None):
  57         """Set or return the Pin Value"""
  58         if val is not None:
  59             if val == self.LOW:
  60                 self._value = val
  61                 GPIO.output(self.id, val)
  62             elif val == self.HIGH:
  63                 self._value = val
  64                 GPIO.output(self.id, val)
  65             else:
  66                 raise RuntimeError("Invalid value for pin")
  67             return None
  68         return GPIO.input(self.id)
  69
  70
  71 D3 = Pin(3)
  72 D5 = Pin(5)
  73 D7 = Pin(7)
  74 D8 = Pin(9)
  75 D10 = Pin(10)
  76 D11 = Pin(11)
  77 D12 = Pin(12)
  78 D13 = Pin(13)
  79 D15 = Pin(15)
  80 D16 = Pin(16)
  81 D18 = Pin(18)
  82 D19 = Pin(19)
  83 D21 = Pin(21)
  84 D22 = Pin(22)
  85 D23 = Pin(23)
  86 D24 = Pin(24)
  87 D26 = Pin(26)
  88 D27 = Pin(27)
  89 D28 = Pin(28)
  90 D29 = Pin(29)
  91 D31 = Pin(31)
  92 D32 = Pin(32)
  93 D33 = Pin(33)
  94 D35 = Pin(35)
  95 D36 = Pin(36)
  96 D37 = Pin(37)
  97 D38 = Pin(38)
  98 D40 = Pin(40)
  99 # I2C
 100 SDA = D3
 101 SCL = D5
 102
 103 # SPI
 104 MISO = D21
 105 MOSI = D19
 106 SCLK = D23
 107
 108 # UART
 109 UART_TX = D8
 110 UART_RX = D10
 111
 112 # ordered as i2cId, SCL, SDA
 113 i2cPorts = ((0, SCL, SDA),)
 114
 115 # ordered as spiId, sckId, mosiId, misoId
 116 spiPorts = ((0, SCLK, MOSI, MISO),)
 117
 118 # ordered as uartId, txId, rxId
 119 uartPorts = ((0, UART_TX, UART_RX),)