trezor.ui.button: add disabled state

This commit is contained in:
Jan Pochyla 2017-02-08 16:20:38 +01:00
parent e4af95e11b
commit d939cabcb5
1 changed files with 34 additions and 13 deletions

View File

@ -1,6 +1,6 @@
from micropython import const
from trezor import ui, loop
from . import display, in_area, rotate_coords
from . import display, in_area, rotate_coords, Widget
DEFAULT_BUTTON = {
@ -15,6 +15,12 @@ DEFAULT_BUTTON_ACTIVE = {
'text-style': ui.BOLD,
'border-color': ui.GREY,
}
DEFAULT_BUTTON_DISABLED = {
'bg-color': ui.BLACK,
'fg-color': ui.GREY,
'text-style': ui.NORMAL,
'border-color': ui.BLACK,
}
CANCEL_BUTTON = {
'bg-color': ui.RED,
@ -60,23 +66,44 @@ BTN_CLICKED = const(1)
BTN_STARTED = const(1)
BTN_ACTIVE = const(2)
BTN_DIRTY = const(4)
BTN_DISABLED = const(8)
class Button():
class Button(Widget):
def __init__(self, area, content, normal_style=None, active_style=None, absolute=False):
def __init__(self, area, content,
normal_style=None,
active_style=None,
disabled_style=None,
absolute=False):
self.area = area
self.content = content
self.normal_style = normal_style or DEFAULT_BUTTON
self.active_style = active_style or DEFAULT_BUTTON_ACTIVE
self.disabled_style = disabled_style or DEFAULT_BUTTON_DISABLED
self.absolute = absolute
self.state = BTN_DIRTY
def enable(self):
self.state &= ~BTN_DISABLED
self.state |= BTN_DIRTY
def disable(self):
self.state |= BTN_DISABLED | BTN_DIRTY
def taint(self):
self.state |= BTN_DIRTY
def render(self):
if not self.state & BTN_DIRTY:
return
state = self.state & ~BTN_DIRTY
style = self.active_style if state & BTN_ACTIVE else self.normal_style
if state & BTN_DISABLED:
style = self.disabled_style
elif state & BTN_ACTIVE:
style = self.active_style
else:
style = self.normal_style
ax, ay, aw, ah = self.area
tx = ax + aw // 2
ty = ay + ah // 2 + 8
@ -96,7 +123,9 @@ class Button():
self.state = state
def send(self, event, pos):
def touch(self, event, pos):
if self.state & BTN_DISABLED:
return
if not self.absolute:
pos = rotate_coords(pos)
if event == loop.TOUCH_START:
@ -113,11 +142,3 @@ class Button():
self.state = BTN_DIRTY
if in_area(pos, self.area):
return BTN_CLICKED
def __iter__(self):
while True:
self.render()
event, *pos = yield loop.Select(loop.TOUCH)
result = self.send(event, pos)
if result is not None:
return result