wasp: simulator: Use Tab to replicate the physical button

1 files changed, 7 insertions, 2 deletions

diff --git a/wasp/boards/simulator/machine.py b/wasp/boards/simulator/machine.py
index 2fc9754..1a5cf46 100644
--- a/wasp/boards/simulator/machine.py
+++ b/wasp/boards/simulator/machine.py
@@ -20,11 +20,16 @@ class Pin(object):
     IN = 'IN'
     OUT = 'OUT'
 
+    pins = {}
+
     def __init__(self, id, direction, value=1, quiet=False):
         self._id = id
         self._value = 0
         self._quiet = quiet
 
+        # Update the pin registry
+        self.pins[id] = self
+
     def init(self, d, value):
         self.value(value)
 
@@ -81,7 +86,7 @@ class I2C():
 
     def readfrom_mem_into(self, addr, reg, dbuf):
         if self.sim:
-            self.sim.readfrom_mem_into(addr, reg, dbuf)
+            self.sim.readfrom_mem_into(addr, reg, dbuf, Pin.pins)
         else:
             raise OSError
 
@@ -107,7 +112,7 @@ class Timer():
         self.time()
 
 def lightsleep(ms=10):
-    display.tick()
+    display.tick(Pin.pins)
     time.sleep(ms / 1000)
 
 def deepsleep(ms=10):