src/adafruit_blinka/microcontroller/rp2040_u2if/spi.py

   1 """SPI Classes for RP2040s with u2if firmware"""
   2 from .rp2040_u2if import rp2040_u2if
   3
   4 # pylint: disable=protected-access, no-self-use
   5 class SPI:
   6     """SPI Base Class for RP2040 u2if"""
   7
   8     MSB = 0
   9
  10     def __init__(self, index, *, baudrate=100000):
  11         self._index = index
  12         self._frequency = baudrate
  13         rp2040_u2if.spi_set_port(self._index)
  14         rp2040_u2if.spi_configure(self._frequency)
  15
  16     # pylint: disable=too-many-arguments,unused-argument
  17     def init(
  18         self,
  19         baudrate=1000000,
  20         polarity=0,
  21         phase=0,
  22         bits=8,
  23         firstbit=MSB,
  24         sck=None,
  25         mosi=None,
  26         miso=None,
  27     ):
  28         """Initialize the Port"""
  29         self._frequency = baudrate
  30         rp2040_u2if.spi_set_port(self._index)
  31         rp2040_u2if.spi_configure(self._frequency)
  32
  33     # pylint: enable=too-many-arguments
  34
  35     @property
  36     def frequency(self):
  37         """Return the current frequency"""
  38         return self._frequency
  39
  40     def write(self, buf, start=0, end=None):
  41         """Write data from the buffer to SPI"""
  42         rp2040_u2if.spi_write(buf, start=start, end=end)
  43
  44     def readinto(self, buf, start=0, end=None, write_value=0):
  45         """Read data from SPI and into the buffer"""
  46         rp2040_u2if.spi_readinto(buf, start=start, end=end, write_value=write_value)
  47
  48     # pylint: disable=too-many-arguments
  49     def write_readinto(
  50         self, buffer_out, buffer_in, out_start=0, out_end=None, in_start=0, in_end=None
  51     ):
  52         """Perform a half-duplex write from buffer_out and then
  53         read data into buffer_in
  54         """
  55         rp2040_u2if.spi_write_readinto(
  56             buffer_out,
  57             buffer_in,
  58             out_start=out_start,
  59             out_end=out_end,
  60             in_start=in_start,
  61             in_end=in_end,
  62         )
  63
  64     # pylint: enable=too-many-arguments
  65
  66 class SPI_Pico(SPI):
  67
  68     def __init__(self, clock, *, baudrate=100000):
  69         index = None
  70         if clock.id == 18:
  71             index = 0
  72         if clock.id == 10:
  73             index = 1
  74         if index is None:
  75             raise ValueError("No SPI port on specified pin.")
  76         super().__init__(index, baudrate=baudrate)
  77
  78 class SPI_Feather(SPI):
  79
  80     def __init__(self, clock, *, baudrate=100000):
  81         index = None
  82         if clock.id == 18:
  83             index = 0
  84         if index is None:
  85             raise ValueError("No SPI port on specified pin.")
  86         super().__init__(index, baudrate=baudrate)