drivers: cst816s: optimize for power saving and responsiveness

Signed-off-by: Fuji Pebri <pebri86@gmail.com>
[daniel@redfelineinja.org.uk: removed leading underscore from method
arguments, removed redundant state tracking, adopt i2c.writeto_mem]
Signed-off-by: Daniel Thompson <daniel@redfelineninja.org.uk>

4 files changed, 58 insertions, 32 deletions

diff --git a/wasp/boards/pinetime/watch.py.in b/wasp/boards/pinetime/watch.py.in
index ef7d0f4..058cdfd 100644
--- a/wasp/boards/pinetime/watch.py.in
+++ b/wasp/boards/pinetime/watch.py.in
@@ -81,7 +81,7 @@ try:
             Signal(Pin('USB_PWR', Pin.IN), invert=True))
     i2c = I2C(1, scl='I2C_SCL', sda='I2C_SDA')
     accel = BMA421(i2c)
-    touch = CST816S(i2c)
+    touch = CST816S(i2c, Pin('TP_INT', Pin.IN), Pin('TP_RST', Pin.OUT, value=0))
     vibrator = Vibrator(Pin('MOTOR', Pin.OUT, value=0), active_low=True)
 
     # Release flash from deep power-down
diff --git a/wasp/boards/simulator/watch.py b/wasp/boards/simulator/watch.py
index c6dafc3..b12ec94 100644
--- a/wasp/boards/simulator/watch.py
+++ b/wasp/boards/simulator/watch.py
@@ -13,6 +13,7 @@ def print_exception(exc, file=sys.stdout):
 sys.print_exception = print_exception
 
 import draw565
+import array
 
 from machine import I2C
 from machine import Pin
diff --git a/wasp/drivers/cst816s.py b/wasp/drivers/cst816s.py
index 182afeb..135c15f 100644
--- a/wasp/drivers/cst816s.py
+++ b/wasp/drivers/cst816s.py
@@ -6,62 +6,87 @@
 """
 
 import array
+import time
+from machine import Pin
 
 class CST816S:
     """Hynitron CST816S I2C touch controller driver.
 
     .. automethod:: __init__
     """
-    
-    def __init__(self, bus):
+
+    def __init__(self, bus, intr, rst):
         """Specify the bus used by the touch controller.
 
         :param machine.I2C bus: I2C bus for the CST816S.
         """
         self.i2c = bus
+        self.tp_int = intr
+        self.tp_rst = rst
         self.dbuf = bytearray(6)
         self.event = array.array('H', (0, 0, 0))
 
-    def get_event(self):
-        """Receive a touch event.
+        self._reset()
+        self.tp_int.irq(trigger=Pin.IRQ_FALLING, handler=self.get_touch_data)
+
+    def _reset(self):
+        self.tp_rst.off()
+        time.sleep_ms(5)
+        self.tp_rst.on()
+        time.sleep_ms(50)
+
+    def get_touch_data(self, pin_obj):
+        """Receive a touch event by interrupt.
 
         Check for a pending touch event and, if an event is pending,
         prepare it ready to go in the event queue.
-
-        :return: An event record if an event is received, None otherwise.
         """
         dbuf = self.dbuf
         event = self.event
 
-        # TODO: check the interrupt pin
-
         try:
             self.i2c.readfrom_mem_into(21, 1, dbuf)
         except OSError:
             return None
 
-        # Skip junk events
-        if dbuf[0] == 0:
-            return None
+        event[0] = dbuf[0] # event
+        event[1] = ((dbuf[2] & 0xf) << 8) + dbuf[3] # x coord
+        event[2] = ((dbuf[4] & 0xf) << 8) + dbuf[5] # y coord
 
-        x = ((dbuf[2] & 0xf) << 8) + dbuf[3]
-        y = ((dbuf[4] & 0xf) << 8) + dbuf[5]
-        swipe_start = dbuf[2] & 0x80
-        
-        # Skip identical events... when the I2C interface comes alive
-        # we can still get back stale events
-        if dbuf[0] == event[0] and x == event[1] and y == event[2] \
-                and not swipe_start:
-            return None
+    def get_event(self):
+        """Receive a touch event.
 
-        # This is a good event, lets save it
-        event[0] = dbuf[0]
-        event[1] = x
-        event[2] = y
+        Check for a pending touch event and, if an event is pending,
+        prepare it ready to go in the event queue.
 
-        # Do not forward swipe start events
-        if dbuf[2] & 0x80:
-            event[0] = 0
+        :return: An event record if an event is received, None otherwise.
+        """
+        if self.event[0] == 0:
             return None
 
-        return event
+        return self.event
+
+    def reset_touch_data(self):
+        """Reset touch data.
+
+        Reset touch data, call this function after processing an event.
+        """
+        self.event[0] = 0
+
+    def wake(self):
+        """Wake up touch controller chip.
+
+        Just reset the chip in order to wake it up
+        """
+        self._reset()
+
+    def sleep(self):
+        """Put touch controller chip on sleep mode to save power.
+        """
+        # Before we can sent the sleep command we have to reset the
+        # panel to get the I2C hardware running again...
+        self._reset()
+        self.i2c.writeto_mem(21, 0xa5, b'\x03')
+
+        # Ensure get_event() cannot return anything
+        self.event[0] = 0
diff --git a/wasp/wasp.py b/wasp/wasp.py
index b94bef5..c4a565c 100644
--- a/wasp/wasp.py
+++ b/wasp/wasp.py
@@ -232,6 +232,7 @@ class Manager():
             self.switch(self.quick_ring[0])
             self.app.sleep()
         watch.display.poweroff()
+        watch.touch.sleep()
         self._charging = watch.battery.charging()
         self.sleep_at = None
 
@@ -241,9 +242,7 @@ class Manager():
         watch.display.poweron()
         self.app.wake()
         watch.backlight.set(self._brightness)
-
-        # Discard any pending touch events
-        _ = watch.touch.get_event()
+        watch.touch.wake()
 
         self.keep_awake()
 
@@ -277,6 +276,7 @@ class Manager():
                 self.navigate(event[0])
         elif event[0] == 5 and self.event_mask & EventMask.TOUCH:
             self.app.touch(event)
+        watch.touch.reset_touch_data()
 
     def _tick(self):
         """Handle the system tick.