mirror of
https://github.com/trezor/trezor-firmware.git
synced 2025-05-07 17:39:03 +00:00
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
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, -24, s['fg-color'], s['bg-color'])
|
|
elif s['icon-fg-color'] is None:
|
|
ui.display.loader(
|
|
r, -24, s['fg-color'], s['bg-color'], res.load(s['icon']))
|
|
else:
|
|
ui.display.loader(
|
|
r, -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
|