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     baudrate = 100000
   7     mode = 0
   8     bits = 8
   9
  10     def __init__(self):
  11         from pyftdi.spi import SpiController
  12         self._spi = SpiController(cs_count=1)
  13         self._spi.configure('ftdi:///1')
  14         Pin.ft232h_gpio = self._spi.get_gpio()
  15
  16     def init(self, baudrate=100000, polarity=0, phase=0, bits=8,
  17                   firstbit=MSB, sck=None, mosi=None, miso=None):
  18         self.cs = 0
  19         self.freq = baudrate
  20         if polarity == 0 and phase == 0:
  21             self.mode = 0
  22         elif polarity == 0 and phase == 1:
  23             self.mode = 1
  24         elif polarity == 1 and phase == 0:
  25             raise ValueError("SPI mode 2 is not supported.")
  26         elif polarity == 1 and phase == 1:
  27             self.mode = 3
  28         else:
  29             raise ValueError("Unknown SPI mode.")
  30
  31     def write(self, buf, start=0, end=None):
  32         end = end if end else len(buf)
  33         port = self._spi.get_port(self.cs, self.freq, self.mode)
  34         port.write(buf[start:end])
  35
  36     def readinto(self, buf, start=0, end=None, write_value=0):
  37         end = end if end else len(buf)
  38         port = self._spi.get_port(self.cs, self.freq, self.mode)
  39         result = port.read(end-start)
  40         for i, b in enumerate(result):
  41             buf[start+i] = b
  42
  43     def write_readinto(self, buffer_out, buffer_in,  out_start=0, out_end=None, in_start=0, in_end=None):
  44         out_end = out_end if out_end else len(buffer_out)
  45         in_end = in_end if in_end else len(buffer_in)
  46         port = self._spi.get_port(self.cs, self.freq, self.mode)
  47         result = port.exchange(buffer_out[out_start:out_end],
  48                                in_end-in_start)
  49         for i, b in enumerate(result):
  50             buffer_in[in_start+i] = b