python/bitbangio/__init__.py

   1 from machine import I2C as _I2C
   2 from machine import Pin
   3
   4 class I2C:
   5     def __init__(self, scl, sda, frequency=400000):
   6         self._locked = False
   7         self.init(scl, sda, frequency)
   8
   9     def init(self, scl, sda, frequency):
  10         self.deinit()
  11         id = -1 # force bitbanging implementation - in future introspect platform if SDA/SCL matches hardware I2C
  12         self._i2c = _I2C(id, Pin(scl.id), Pin(sda.id), freq=frequency)
  13
  14     def deinit(self):
  15         try:
  16             del self._i2c
  17         except AttributeError:
  18             pass
  19
  20     def __enter__(self):
  21         return self
  22
  23     def __exit__(self, exc_type, exc_value, traceback):
  24         self.deinit()
  25
  26     def scan(self):
  27         return self._i2c.scan()
  28
  29     def try_lock(self):
  30         if self._locked:
  31             return False
  32         else:
  33             self._locked=True
  34             return True
  35
  36     def unlock(self):
  37         if self._locked:
  38             self._locked = False
  39         else:
  40             raise ValueError("Not locked")
  41
  42     def readfrom_into(self, address, buffer, start=0, end=None):
  43         if start is not 0 or end is not None:
  44             if end is None:
  45                 end = len(buffer)
  46             buffer = memoryview(buffer)[start:end]
  47         stop = True # remove for efficiency later
  48         return self._i2c.readfrom_into(address, buffer, stop)
  49
  50     def writeto(self, address, buffer, start=0, end=None, stop=True, *a, **k):
  51         if start is not 0 or end is not None:
  52             if end is None:
  53                 end = len(buffer)
  54             buffer = memoryview(buffer)[start:end]
  55         return self._i2c.writeto(address, buffer, stop)