python/bitbangio/__init__.py

   1 from mcp import Lockable
   2 from machine import I2C as _I2C
   3 from machine import Pin
   4 import agnostic
   5
   6 class I2C(Lockable):
   7     def __init__(self, scl, sda, frequency=400000):
   8         super().__init()
   9         self.init(scl, sda, frequency)
  10
  11     def init(self, scl, sda, frequency):
  12         self.deinit()
  13         id = -1 # force bitbanging implementation - in future introspect platform if SDA/SCL matches hardware I2C
  14         self._i2c = _I2C(id, Pin(scl.id), Pin(sda.id), freq=frequency)
  15
  16     def deinit(self):
  17         try:
  18             del self._i2c
  19         except AttributeError:
  20             pass
  21
  22     def __enter__(self):
  23         return self
  24
  25     def __exit__(self, exc_type, exc_value, traceback):
  26         self.deinit()
  27
  28     def scan(self):
  29         return self._i2c.scan()
  30
  31     def readfrom_into(self, address, buffer, start=0, end=None):
  32         if start is not 0 or end is not None:
  33             if end is None:
  34                 end = len(buffer)
  35             buffer = memoryview(buffer)[start:end]
  36         stop = True # remove for efficiency later
  37         return self._i2c.readfrom_into(address, buffer, stop)
  38
  39     def writeto(self, address, buffer, start=0, end=None, stop=True, *a, **k):
  40         if start is not 0 or end is not None:
  41             if end is None:
  42                 end = len(buffer)
  43             buffer = memoryview(buffer)[start:end]
  44         return self._i2c.writeto(address, buffer, stop)
  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(baudrate=baudrate, polarity=polarity, phase=phase, bits = bits, firstbit = SPI.MSB, sck = Pin(self._pins[0]), mosi=Pin(self._pins[1]), miso=Pin(self._pins[2]))
  57         else:
  58             raise RuntimeError("First call try_lock()")
  59
  60     def write(self, buf):
  61         return self._spi.write(buf)
  62
  63     def readinto(self, buf):
  64         return self.readinto(buf)
  65
  66     def write_readinto(self, buffer_out, buffer_in)
  67         return self.write_readinto(buffer_out, buffer_in)