X-Git-Url: https://git.ayoreis.com/hackapet/Adafruit_Blinka.git/blobdiff_plain/8601be951a6c95853058fefcd818bc041b8edca1..777a446080c6654846c6f19944e3c28180ca6c19:/src/adafruit_blinka/microcontroller/raspi_23/i2c.py

diff --git a/src/adafruit_blinka/microcontroller/raspi_23/i2c.py b/src/adafruit_blinka/microcontroller/raspi_23/i2c.py
index 5273fdd..6e9f31c 100644
--- a/src/adafruit_blinka/microcontroller/raspi_23/i2c.py
+++ b/src/adafruit_blinka/microcontroller/raspi_23/i2c.py
@@ -1,9 +1,41 @@
-import smbus
+import Adafruit_PureIO.smbus as smbus
 import time
 
-
 class I2C:
-
+    MASTER = 0
+    SLAVE = 1
+    _baudrate = None
+    _mode = None
     _i2c_bus = None
-    def __init__(self, bus_num=0):
-        i2c_bus = smbus.SMBus(bus_num)
+
+    def __init__(self, bus_num, mode=MASTER, baudrate=None):
+        if mode != self.MASTER:
+            raise NotImplementedError("Only I2C Master supported!")
+        _mode = self.MASTER
+
+        if baudrate != None:
+            print("I2C frequency is not settable in python, ignoring!")
+        
+        try:
+            self._i2c_bus = smbus.SMBus(bus_num)
+        except FileNotFoundError:
+            raise RuntimeError("I2C Bus #%d not found, check if enabled in config!" % bus_num)
+
+    def scan(self):
+        """Try to read a byte from each address, if you get an OSError it means the device isnt there"""
+        found = []
+        for addr in range(0,0x80):
+            try:
+                self._i2c_bus.read_byte(addr)
+            except OSError:
+                continue
+            found.append(addr)
+        return found
+
+    def writeto(self, address, buffer, stop=True):
+        self._i2c_bus.write_i2c_block_data(address, buffer[0], buffer[1:])
+
+    def readfrom_into(self, address, buffer, stop=True):
+        readin = self._i2c_bus.read_bytes(address, len(buffer))
+        for i in range(len(buffer)):
+            buffer[i] = readin[i]