src/adafruit_blinka/microcontroller/generic_linux/spi.py

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