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         chunks, rest = divmod(end - start, self._spi.PAYLOAD_MAX_LENGTH)
  36         for i in range(chunks):
  37             chunk_start = start + i * self._spi.PAYLOAD_MAX_LENGTH
  38             chunk_end = chunk_start + self._spi.PAYLOAD_MAX_LENGTH
  39             port.write(buf[chunk_start:chunk_end])
  40         if rest:
  41             port.write(buf[-1*rest:])
  42
  43     def readinto(self, buf, start=0, end=None, write_value=0):
  44         end = end if end else len(buf)
  45         port = self._spi.get_port(self.cs, self.freq, self.mode)
  46         result = port.read(end-start)
  47         for i, b in enumerate(result):
  48             buf[start+i] = b
  49
  50     def write_readinto(self, buffer_out, buffer_in,  out_start=0, out_end=None, in_start=0, in_end=None):
  51         out_end = out_end if out_end else len(buffer_out)
  52         in_end = in_end if in_end else len(buffer_in)
  53         port = self._spi.get_port(self.cs, self.freq, self.mode)
  54         result = port.exchange(buffer_out[out_start:out_end],
  55                                in_end-in_start, duplex=True)
  56         for i, b in enumerate(result):
  57             buffer_in[in_start+i] = b