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     """SPI Class for Pico u2if"""
  68
  69     def __init__(self, clock, *, baudrate=100000):
  70         index = None
  71         if clock.id == 18:
  72             index = 0
  73         if clock.id == 10:
  74             index = 1
  75         if index is None:
  76             raise ValueError("No SPI port on specified pin.")
  77         super().__init__(index, baudrate=baudrate)
  78
  79 class SPI_Feather(SPI):
  80     """SPI Class for Feather u2if"""
  81
  82     def __init__(self, clock, *, baudrate=100000):
  83         index = None
  84         if clock.id == 18:
  85             index = 0
  86         if index is None:
  87             raise ValueError("No SPI port on specified pin.")
  88         super().__init__(index, baudrate=baudrate)
  89
  90 class SPI_QTPY(SPI):
  91     """SPI Class for QT Py u2if"""
  92
  93     def __init__(self, clock, *, baudrate=100000):
  94         index = None
  95         if clock.id == 6:
  96             index = 0
  97         if index is None:
  98             raise ValueError("No SPI port on specified pin.")
  99         super().__init__(index, baudrate=baudrate)
 100
 101 class SPI_ItsyBitsy(SPI):
 102     """SPI Class for ItsyBitsy u2if"""
 103
 104     def __init__(self, clock, *, baudrate=100000):
 105         index = None
 106         if clock.id == 18:
 107             index = 0
 108         if index is None:
 109             raise ValueError("No SPI port on specified pin.")
 110         super().__init__(index, baudrate=baudrate)