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