src/bitbangio.py

   1 """
   2 `bitbangio` - Bitbanged bus protocols
   3 ==============================================================
   4
   5 See `CircuitPython:bitbangio` in CircuitPython for more details.
   6
   7 * Author(s): cefn
   8 """
   9
  10 from adafruit_blinka import Lockable, agnostic
  11
  12
  13 class I2C(Lockable):
  14     def __init__(self, scl, sda, frequency=400000):
  15         if agnostic.microcontroller == "stm32":
  16             raise NotImplementedError("No software I2C on {}".format(agnostic.board))
  17         self.init(scl, sda, frequency)
  18
  19     def init(self, scl, sda, frequency):
  20         from machine import Pin
  21         from machine import I2C as _I2C
  22         self.deinit()
  23         id = -1  # force bitbanging implementation - in future introspect platform if SDA/SCL matches hardware I2C
  24         self._i2c = _I2C(id, Pin(scl.id), Pin(sda.id), freq=frequency)
  25
  26     def deinit(self):
  27         try:
  28             del self._i2c
  29         except AttributeError:
  30             pass
  31
  32     def __enter__(self):
  33         return self
  34
  35     def __exit__(self, exc_type, exc_value, traceback):
  36         self.deinit()
  37
  38     def scan(self):
  39         return self._i2c.scan()
  40
  41     def readfrom_into(self, address, buffer, start=0, end=None):
  42         if start is not 0 or end is not None:
  43             if end is None:
  44                 end = len(buffer)
  45             buffer = memoryview(buffer)[start:end]
  46         stop = True  # remove for efficiency later
  47         return self._i2c.readfrom_into(address, buffer, stop)
  48
  49     def writeto(self, address, buffer, start=0, end=None, stop=True):
  50         if start is not 0 or end is not None:
  51             if end is None:
  52                 return self._i2c.writeto(address, memoryview(buffer)[start:], stop)
  53             else:
  54                 return self._i2c.writeto(address, memoryview(buffer)[start:end], stop)
  55         return self._i2c.writeto(address, buffer, stop)
  56
  57
  58 # TODO untested, as actually busio.SPI was on tasklist https://github.com/adafruit/Adafruit_Micropython_Blinka/issues/2 :(
  59 class SPI(Lockable):
  60     def __init__(self, clock, MOSI=None, MISO=None):
  61         from machine import SPI
  62         self._spi = SPI(-1)
  63         self._pins = (clock, MOSI, MISO)
  64
  65     def configure(self, baudrate=100000, polarity=0, phase=0, bits=8):
  66         from machine import SPI,Pin
  67         if self._locked:
  68             # TODO verify if _spi obj 'caches' sck, mosi, miso to avoid storing in _attributeIds (duplicated in busio)
  69             # i.e. #init ignores MOSI=None rather than unsetting
  70             self._spi.init(
  71                 baudrate=baudrate,
  72                 polarity=polarity,
  73                 phase=phase,
  74                 bits=bits,
  75                 firstbit=SPI.MSB,
  76                 sck=Pin(self._pins[0].id),
  77                 mosi=Pin(self._pins[1].id),
  78                 miso=Pin(self._pins[2].id))
  79         else:
  80             raise RuntimeError("First call try_lock()")
  81
  82     def write(self, buf):
  83         return self._spi.write(buf)
  84
  85     def readinto(self, buf):
  86         return self.readinto(buf)
  87
  88     def write_readinto(self, buffer_out, buffer_in):
  89         return self.write_readinto(buffer_out, buffer_in)