src/adafruit_blinka/microcontroller/generic_agnostic_board/pin.py

   1 # SPDX-FileCopyrightText: 2024 Melissa LeBlanc-Williams for Adafruit Industries
   2 #
   3 # SPDX-License-Identifier: MIT
   4 """generic_agnostic_board pin interface"""
   5 import random
   6
   7
   8 class Pin:
   9     """A basic Pin class for use with generic_agnostic_board"""
  10
  11     # pin modes
  12     OUT = 0
  13     IN = 1
  14     ADC = 2
  15     DAC = 3
  16     # pin values
  17     LOW = 0
  18     HIGH = 1
  19     # pin pulls
  20     PULL_NONE = 0
  21     PULL_UP = 1
  22     PULL_DOWN = 2
  23
  24     def return_toggle(self):
  25         """Returns the pin's expected value, toggling between True and False"""
  26         toggle_state = not self.previous_value
  27         return toggle_state
  28
  29     def return_false(self):
  30         """Returns the pin's expected value, False"""
  31         return False
  32
  33     def return_true(self):
  34         """Returns the pin's expected value, True"""
  35         return True
  36
  37     def return_random_int(self):
  38         """Returns a random integer"""
  39         return random.randint(0, 65535)
  40
  41     def return_fixed_int_pi(self):
  42         """Returns the first five digits of Pi, 31415"""
  43         return 31415
  44
  45     def __init__(self, pin_id=None):
  46         self.id = pin_id
  47         self._mode = None
  48         self._pull = None
  49         self.previous_value = False
  50         self.current_value = None
  51
  52         # mapping of pin definition names to expected behavior
  53         self.pin_behavior = {
  54             0: self.return_true,  # Dx_INPUT_TRUE
  55             1: self.return_false,  # Dx_INPUT_FALSE
  56             2: self.return_true,  # Dx_INPUT_TRUE_PULL_UP
  57             3: self.return_true,  # Dx_INPUT_TRUE_PULL_DOWN
  58             4: self.return_true,  # Dx_OUTPUT
  59             6: self.return_true,  # NEOPIXEL
  60             7: self.return_random_int,  # Ax_INPUT_RAND_INT
  61             8: self.return_fixed_int_pi,  # Ax_INPUT_FIXED_INT_PI
  62             9: self.return_true,  # Ax_OUTPUT_WAVE_SINE
  63             10: self.return_true,  # Ax_OUTPUT_WAVE_SAWTOOTH
  64             11: self.return_toggle,  # Dx_INPUT_TOGGLE
  65             # Add other mappings as needed
  66         }
  67
  68     def init(self, mode=IN, pull=None):
  69         """Initialize the Pin"""
  70         if self.id is None:
  71             raise RuntimeError("Can not init a None type pin.")
  72         pull = Pin.PULL_NONE if pull is None else pull
  73         self._pull = pull
  74         self._mode = mode
  75
  76     def write(self, new_value):
  77         """Saves the new_value to the pin for subsequent calls to .value"""
  78         self.previous_value = self.current_value
  79         self.current_value = new_value
  80
  81     def read(self):
  82         """Returns the pin's expected value."""
  83         self.previous_value = self.current_value
  84         # perform a lookup on the pin_behavior dict to get the value
  85         self.current_value = self.pin_behavior.get(self.id)()
  86
  87         # is pin a pull up and pin is LOW?
  88         if self._pull == Pin.PULL_UP and self.current_value == False:
  89             self.current_value = False
  90         # is pin a pull down and pin is HIGH?
  91         if self._pull == Pin.PULL_DOWN and self.current_value == True:
  92             self.current_value = False
  93
  94         return self.current_value
  95
  96     def value(self, val=None):
  97         """Set or return the Pin Value"""
  98         # Digital In / Out
  99         if self._mode in (Pin.IN, Pin.OUT):
 100             # digital read
 101             if val is None:
 102                 return self.read()
 103             # digital write
 104             if val in (Pin.LOW, Pin.HIGH):
 105                 return self.write(val)
 106             # nope
 107             raise ValueError("Invalid value for pin.")
 108         # Analog In
 109         if self._mode == Pin.ADC:
 110             if val is None:
 111                 return self.read()
 112             # read only
 113             raise AttributeError("'AnalogIn' object has no attribute 'value'")
 114         # Analog Out
 115         if self._mode == Pin.DAC:
 116             if val is None:
 117                 # write only
 118                 raise AttributeError("unreadable attribute")
 119             self.write(val)
 120             return None
 121         raise RuntimeError(
 122             "No action for mode {} with value {}".format(self._mode, val)
 123         )
 124
 125
 126 # create pin instances for each pin
 127 D0 = Pin(0)
 128 D1 = Pin(1)
 129 D2 = Pin(2)
 130 D3 = Pin(3)
 131 D4 = Pin(4)
 132 # Special "digital" pins
 133 D6 = Pin(6)
 134 # Analog pins
 135 A0 = Pin(7)
 136 A1 = Pin(8)
 137 A2 = Pin(9)
 138 A3 = Pin(10)
 139
 140 D7 = Pin(11)
 141
 142 # I2C pins
 143 SDA = Pin()
 144 SCL = Pin()
 145
 146 # SPI pins
 147 SCLK = Pin()
 148 SCK = Pin()
 149 MOSI = Pin()
 150 MISO = Pin()
 151 CS = Pin()
 152
 153 # UART pins
 154 UART_TX = Pin()
 155 UART_RX = Pin()