trezor-firmware/core/src/trezor/ui/loader.py

import utime
from micropython import const

from trezor import loop, res, ui

_TARGET_MS = const(1000)
_SHRINK_BY = const(2)


class Loader(ui.Widget):
    def __init__(self, style=ui.LDR_DEFAULT):
        self.target_ms = _TARGET_MS
        self.normal_style = style["normal"] or ui.LDR_DEFAULT["normal"]
        self.active_style = style["active"] or ui.LDR_DEFAULT["active"]
        self.start_ms = None
        self.stop_ms = None

    def start(self):
        self.start_ms = utime.ticks_ms()
        self.stop_ms = None

    def stop(self):
        if self.start_ms is not None and self.stop_ms is None:
            diff_ms = utime.ticks_ms() - self.start_ms
        else:
            diff_ms = 0
        self.stop_ms = utime.ticks_ms()
        return diff_ms >= self.target_ms

    def is_active(self):
        return self.start_ms is not None

    def render(self):
        target = self.target_ms
        start = self.start_ms
        stop = self.stop_ms
        now = utime.ticks_ms()
        if stop is None:
            r = min(now - start, target)
        else:
            r = max(stop - start + (stop - now) * _SHRINK_BY, 0)
            if r == 0:
                self.start_ms = None
                self.stop_ms = None
        if r == target:
            s = self.active_style
        else:
            s = self.normal_style
        if s["icon"] is None:
            ui.display.loader(r, False, -24, s["fg-color"], s["bg-color"])
        elif s["icon-fg-color"] is None:
            ui.display.loader(
                r, False, -24, s["fg-color"], s["bg-color"], res.load(s["icon"])
            )
        else:
            ui.display.loader(
                r,
                False,
                -24,
                s["fg-color"],
                s["bg-color"],
                res.load(s["icon"]),
                s["icon-fg-color"],
            )

    def __iter__(self):
        sleep = loop.sleep(1000000 // 30)  # 30 fps
        ui.display.bar(0, 32, ui.WIDTH, ui.HEIGHT - 83, ui.BG)  # clear
        while self.is_active():
            self.render()
            yield sleep
        ui.display.bar(0, 32, ui.WIDTH, ui.HEIGHT - 83, ui.BG)  # clear