src/adafruit_blinka/microcontroller/mips24kec/spi.py

   1 import spidev
   2 import time
   3 from adafruit_blinka.agnostic import detector
   4
   5 class SPI:
   6     MSB = 0
   7     LSB = 1
   8     CPHA = 1
   9     CPOL = 2
  10
  11     baudrate = 100000
  12     mode = 0
  13     bits = 8
  14
  15     def __init__(self, portid):
  16         if isinstance(portid, tuple):
  17             self._bus_id = portid[0]
  18             self._device_id = portid[1]
  19         else:
  20             self._bus_id = portid
  21             self._device_id = 0
  22         self._spi = spidev.SpiDev()
  23
  24     def init(self, baudrate=100000, polarity=0, phase=0, bits=8,
  25                   firstbit=MSB, sck=None, mosi=None, miso=None):
  26         mode = 0
  27         if polarity:
  28             mode |= self.CPOL
  29         if phase:
  30             mode |= self.CPHA
  31
  32         self.clock_pin = sck
  33         self.mosi_pin = mosi
  34         self.miso_pin = miso
  35         self.baudrate = baudrate
  36         self.mode = mode
  37         self.bits = bits
  38         self.chip = detector.chip
  39
  40     def set_no_cs(self):
  41         # No kernel seems to support this, so we're just going to pass
  42         pass
  43
  44     @property
  45     def frequency(self):
  46         return self.baudrate
  47
  48     def write(self, buf, start=0, end=None):
  49         if not buf:
  50             return
  51         if end is None:
  52             end = len(buf)
  53         try:
  54             self._spi.open(self._bus_id, self._device_id)
  55             self.set_no_cs()
  56             self._spi.max_speed_hz = self.baudrate
  57             self._spi.mode = self.mode
  58             self._spi.bits_per_word = self.bits
  59             self._spi.writebytes(buf[start:end])
  60             self._spi.close()
  61         except FileNotFoundError as not_found:
  62             print("Could not open SPI device - check if SPI is enabled in kernel!")
  63             raise
  64
  65     def readinto(self, buf, start=0, end=None, write_value=0):
  66         if not buf:
  67             return
  68         if end is None:
  69             end = len(buf)
  70         try:
  71             self._spi.open(self._bus_id, self._device_id)
  72             self.set_no_cs()
  73             self._spi.max_speed_hz = self.baudrate
  74             self._spi.mode = self.mode
  75             self._spi.bits_per_word = self.bits
  76             data = self._spi.xfer3([write_value],(end-start))
  77             for i in range(end-start):  # 'readinto' the given buffer
  78               buf[start+i] = data[i]
  79             self._spi.close()
  80         except FileNotFoundError as not_found:
  81             print("Could not open SPI device - check if SPI is enabled in kernel!")
  82             raise
  83
  84     def write_readinto(self, buffer_out, buffer_in, out_start=0,
  85                        out_end=None, in_start=0, in_end=None):
  86         if not buffer_out or not buffer_in:
  87             return
  88         if out_end is None:
  89             out_end = len(buffer_out)
  90         if in_end is None:
  91             in_end = len(buffer_in)
  92         if out_end - out_start != in_end - in_start:
  93             raise RuntimeError('Buffer slices must be of equal length.')
  94         try:
  95             self._spi.open(self._bus_id, self._device_id)
  96             self.set_no_cs()
  97             self._spi.max_speed_hz = self.baudrate
  98             self._spi.mode = self.mode
  99             self._spi.bits_per_word = self.bits
 100             data = self._spi.xfer(list(buffer_out[out_start:out_end+1]))
 101             for i in range((in_end - in_start)):
 102                 buffer_in[i+in_start] = data[i]
 103             self._spi.close()
 104         except FileNotFoundError as not_found:
 105             print("Could not open SPI device - check if SPI is enabled in kernel!")
 106             raise