src/adafruit_blinka/microcontroller/generic_linux/spi.py

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