src/adafruit_blinka/microcontroller/generic_linux/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         self._port = portid
  17         self._spi = spidev.SpiDev()
  18
  19     def init(self, baudrate=100000, polarity=0, phase=0, bits=8,
  20                   firstbit=MSB, sck=None, mosi=None, miso=None):
  21         mode = 0
  22         if polarity:
  23             mode |= self.CPOL
  24         if phase:
  25             mode |= self.CPHA
  26
  27         self.clock_pin = sck
  28         self.mosi_pin = mosi
  29         self.miso_pin = miso
  30         self.baudrate = baudrate
  31         self.mode = mode
  32         self.bits = bits
  33         self.chip = detector.chip
  34
  35     def set_no_cs(self):
  36         # Linux SPI driver for AM33XX chip in BeagleBone and PocketBeagle
  37         # does not support setting SPI_NO_CS mode bit (issue #104)
  38         try:
  39             if not self.chip.AM33XX and not self.chip.IMX8MX and not self.chip.SAMA5 \
  40              and not self.chip.APQ8016 and not self.chip.T210 and not self.chip.T186 \
  41              and not self.chip.T194 and not self.chip.SUN8I:
  42                 self._spi.no_cs = True  # this doesn't work but try anyways
  43         except AttributeError:
  44             pass
  45
  46     @property
  47     def frequency(self):
  48         return self.baudrate
  49
  50     def write(self, buf, start=0, end=None):
  51         if not buf:
  52             return
  53         if end is None:
  54             end = len(buf)
  55         try:
  56             self._spi.open(self._port, 0)
  57             self.set_no_cs()
  58             self._spi.max_speed_hz = self.baudrate
  59             self._spi.mode = self.mode
  60             self._spi.bits_per_word = self.bits
  61             self._spi.writebytes2(buf[start:end])
  62             self._spi.close()
  63         except FileNotFoundError as not_found:
  64             print("Could not open SPI device - check if SPI is enabled in kernel!")
  65             raise
  66
  67     def readinto(self, buf, start=0, end=None, write_value=0):
  68         if not buf:
  69             return
  70         if end is None:
  71             end = len(buf)
  72         try:
  73             self._spi.open(self._port, 0)
  74             self.set_no_cs()
  75             self._spi.max_speed_hz = self.baudrate
  76             self._spi.mode = self.mode
  77             self._spi.bits_per_word = self.bits
  78             data = self._spi.xfer([write_value]*(end-start))
  79             for i in range(end-start):  # 'readinto' the given buffer
  80               buf[start+i] = data[i]
  81             self._spi.close()
  82         except FileNotFoundError as not_found:
  83             print("Could not open SPI device - check if SPI is enabled in kernel!")
  84             raise
  85
  86     def write_readinto(self, buffer_out, buffer_in, out_start=0,
  87                        out_end=None, in_start=0, in_end=None):
  88         if not buffer_out or not buffer_in:
  89             return
  90         if out_end is None:
  91             out_end = len(buffer_out)
  92         if in_end is None:
  93             in_end = len(buffer_in)
  94         if out_end - out_start != in_end - in_start:
  95             raise RuntimeError('Buffer slices must be of equal length.')
  96         try:
  97             self._spi.open(self._port, 0)
  98             self.set_no_cs()
  99             self._spi.max_speed_hz = self.baudrate
 100             self._spi.mode = self.mode
 101             self._spi.bits_per_word = self.bits
 102             data = self._spi.xfer(list(buffer_out[out_start:out_end+1]))
 103             for i in range((in_end - in_start)):
 104                 buffer_in[i+in_start] = data[i]
 105             self._spi.close()
 106         except FileNotFoundError as not_found:
 107             print("Could not open SPI device - check if SPI is enabled in kernel!")
 108             raise