src/adafruit_blinka/microcontroller/ft232h/spi.py

   1 from adafruit_blinka.microcontroller.ft232h.pin import Pin
   2
   3 class SPI:
   4     MSB = 0
   5
   6     def __init__(self):
   7         from pyftdi.spi import SpiController
   8         self._spi = SpiController(cs_count=1)
   9         self._spi.configure('ftdi:///1')
  10         self._port = self._spi.get_port(0)
  11         self._port.set_frequency(100000)
  12         self._port._cpol = 0
  13         self._port._cpha = 0
  14         # Change GPIO controller to SPI
  15         Pin.ft232h_gpio = self._spi.get_gpio()
  16
  17     def init(self, baudrate=100000, polarity=0, phase=0, bits=8,
  18                   firstbit=MSB, sck=None, mosi=None, miso=None):
  19         self._port.set_frequency(baudrate)
  20         # FTDI device can only support mode 0 and mode 2
  21         # due to the limitation of MPSSE engine.
  22         # This means CPHA must = 0
  23         self._port._cpol = polarity
  24         if phase != 0:
  25             raise ValueError("Only SPI phase 0 is supported by FT232H.")
  26         self._port._cpha = phase
  27
  28     @property
  29     def frequency(self):
  30         return self._port.frequency
  31
  32     def write(self, buf, start=0, end=None):
  33         end = end if end else len(buf)
  34         chunks, rest = divmod(end - start, self._spi.PAYLOAD_MAX_LENGTH)
  35         for i in range(chunks):
  36             chunk_start = start + i * self._spi.PAYLOAD_MAX_LENGTH
  37             chunk_end = chunk_start + self._spi.PAYLOAD_MAX_LENGTH
  38             self._port.write(buf[chunk_start:chunk_end])
  39         if rest:
  40             self._port.write(buf[-1*rest:])
  41
  42     def readinto(self, buf, start=0, end=None, write_value=0):
  43         end = end if end else len(buf)
  44         result = self._port.read(end-start)
  45         for i, b in enumerate(result):
  46             buf[start+i] = b
  47
  48     def write_readinto(self, buffer_out, buffer_in,  out_start=0, out_end=None, in_start=0, in_end=None):
  49         out_end = out_end if out_end else len(buffer_out)
  50         in_end = in_end if in_end else len(buffer_in)
  51         result = self._port.exchange(buffer_out[out_start:out_end],
  52                                      in_end-in_start, duplex=True)
  53         for i, b in enumerate(result):
  54             buffer_in[in_start+i] = b