src/bitbangio.py

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