Reworked wait into class

src/apps/homescreen/layout_homescreen.py

@@ -7,17 +7,17 @@ def layout_homescreen():
 
     # ui.display.bar(0, 0, 240, 240, ui.WHITE)
 
-    # f = open('apps/homescreen/trezor.toig', 'r')
+    f = open('apps/homescreen/trezor.toig', 'r')
 
-    # def func(foreground):
-    #     f.seek(0)
-    #     ui.display.icon(0, 0, f.read(), foreground, ui.BLACK)
+    def func(foreground):
+        f.seek(0)
+        ui.display.icon(0, 0, f.read(), foreground, ui.BLACK)
 
-    # animation = ui.animate_pulse(func, ui.WHITE, ui.GREY, speed=400000)
+    animation = ui.animate_pulse(func, ui.WHITE, ui.GREY, speed=400000)
 
-    timeout = loop.sleep(1000 * 1000)
+    timeout = loop.sleep(5000 * 1000)
 
-    yield loop.wait([timeout])
+    yield loop.Wait([animation, timeout])
     print('back to layout')
 
     # try:
@@ -25,5 +25,5 @@ def layout_homescreen():
     # except:
     #     pass
 
-    # from apps import playground
-    # return playground.layout_tap_to_confirm('1BitkeyP2nDd5oa64x7AjvBbbwST54W5Zmx2', 110.126967, 'BTC')
+    from apps import playground
+    return playground.layout_tap_to_confirm('1BitkeyP2nDd5oa64x7AjvBbbwST54W5Zmx2', 110.126967, 'BTC')

src/trezor/loop.py

@@ -36,49 +36,38 @@ def __call_at(time, gen):
         time = utime.ticks_us()
     heappush(time_queue, (time, gen))
 
+class Wait():
+    def __init__(self, gens, wait_for=1, exit_others=True):
+        self.wait_for = wait_for
+        self.exit_others = exit_others
+        self.received = 0
+        self.callback = None
+        self.gens = gens
+
+        for g in gens:
+            __call_at(None, self._wait(g))
+
+    def _wait(self, gen):
+        if isinstance(gen, type_gen):
+            ret = yield from gen
+        else:
+            ret = yield gen
 
-def __wait_for_gen(gen, cb):
-    if isinstance(gen, type_gen):
-        ret = yield from gen
-    else:
-        ret = yield gen
-    try:
-        cb.throw(StopIteration)
-    except Exception as e:
-        log.info(__name__, '__wait_gen throw raised %s', e)
-
-    log.info(__name__, '__wait_gen returning %s', ret)
-
-
-class __Wait():
-    pass
-
-
-def __wait_callback(call_after):
-    # TODO: rewrite as instance of __Wait instead of generator
-    delegate = yield
-    received = 0
-    while received < call_after:
-        try:
-            yield
-        except StopIteration:
-            received += 1
-    __call_at(None, delegate)
-
-
-def wait_for_first(gens):
-    cb = __wait_callback(1)
-    for g in gens:
-        __call_at(None, __wait_for_gen(g, cb))
-    return cb
+        self.finish(gen, ret)
 
+    def finish(self, gen, result):
+        self.received += 1
 
-def wait_for_all(gens):
-    cb = __wait_callback(len(gens))
-    for g in gens:
-        __call_at(None, __wait_for_gen(g, cb))
-    return cb
+        if self.received >= self.wait_for:
+            __call_at(None, self.callback)
+            self.callback = None
 
+            if self.exit_others:
+                for g in self.gens:
+                    try:
+                        g.throw(StopIteration())
+                    except:
+                        pass
 
 def sleep(us):
     return utime.ticks_us() + us
@@ -138,10 +127,9 @@ def run_forever(start_gens):
                 # wait for event
                 raise NotImplementedError()
 
-        elif isinstance(ret, type_gen):
+        elif isinstance(ret, Wait):
             log.info(__name__, 'Scheduling %s -> %s', gen, ret)
-            ret.send(None)
-            ret.send(gen)
+            ret.callback = gen
 
         elif ret is None:
             # just call us asap