From 9de96f61141f29a022aa5b26a5a8e196f785abae Mon Sep 17 00:00:00 2001 From: Dusan Klinec Date: Tue, 23 Apr 2019 02:06:37 +0200 Subject: [PATCH] ui.display.loader: slice_span added for indeterminate loaders --- core/embed/bootloader/bootui.c | 14 +++++----- core/embed/extmod/modtrezorui/display.c | 4 +-- core/embed/extmod/modtrezorui/display.h | 2 +- .../extmod/modtrezorui/modtrezorui-display.h | 27 ++++++++++++------- core/mocks/generated/trezorui.py | 4 ++- 5 files changed, 31 insertions(+), 20 deletions(-) diff --git a/core/embed/bootloader/bootui.c b/core/embed/bootloader/bootui.c index e5e84b6a6c..8609d93fae 100644 --- a/core/embed/bootloader/bootui.c +++ b/core/embed/bootloader/bootui.c @@ -237,7 +237,7 @@ void ui_screen_install_confirm_newvendor(const vendor_header *const vhdr, void ui_screen_install(void) { display_bar(0, 0, DISPLAY_RESX, DISPLAY_RESY, COLOR_WHITE); display_loader(0, -20, COLOR_BL_PROCESS, COLOR_WHITE, toi_icon_install, - sizeof(toi_icon_install), COLOR_BLACK); + sizeof(toi_icon_install), COLOR_BLACK, 0); display_text_center(DISPLAY_RESX / 2, DISPLAY_RESY - 24, "Installing firmware", -1, FONT_NORMAL, COLOR_BLACK, COLOR_WHITE); @@ -245,12 +245,12 @@ void ui_screen_install(void) { void ui_screen_install_progress_erase(int pos, int len) { display_loader(250 * pos / len, -20, COLOR_BL_PROCESS, COLOR_WHITE, - toi_icon_install, sizeof(toi_icon_install), COLOR_BLACK); + toi_icon_install, sizeof(toi_icon_install), COLOR_BLACK, 0); } void ui_screen_install_progress_upload(int pos) { display_loader(pos, -20, COLOR_BL_PROCESS, COLOR_WHITE, toi_icon_install, - sizeof(toi_icon_install), COLOR_BLACK); + sizeof(toi_icon_install), COLOR_BLACK, 0); } // wipe UI @@ -275,14 +275,14 @@ void ui_screen_wipe_confirm(void) { void ui_screen_wipe(void) { display_bar(0, 0, DISPLAY_RESX, DISPLAY_RESY, COLOR_WHITE); display_loader(0, -20, COLOR_BL_PROCESS, COLOR_WHITE, toi_icon_wipe, - sizeof(toi_icon_wipe), COLOR_BLACK); + sizeof(toi_icon_wipe), COLOR_BLACK, 0); display_text_center(DISPLAY_RESX / 2, DISPLAY_RESY - 24, "Wiping device", -1, FONT_NORMAL, COLOR_BLACK, COLOR_WHITE); } void ui_screen_wipe_progress(int pos, int len) { display_loader(1000 * pos / len, -20, COLOR_BL_PROCESS, COLOR_WHITE, - toi_icon_wipe, sizeof(toi_icon_wipe), COLOR_BLACK); + toi_icon_wipe, sizeof(toi_icon_wipe), COLOR_BLACK, 0); } // done UI @@ -301,7 +301,7 @@ void ui_screen_done(int restart_seconds, secbool full_redraw) { display_bar(0, 0, DISPLAY_RESX, DISPLAY_RESY, COLOR_WHITE); } display_loader(1000, -20, COLOR_BL_DONE, COLOR_WHITE, toi_icon_done, - sizeof(toi_icon_done), COLOR_BLACK); + sizeof(toi_icon_done), COLOR_BLACK, 0); if (secfalse == full_redraw) { display_bar(0, DISPLAY_RESY - 24 - 18, 240, 23, COLOR_WHITE); } @@ -314,7 +314,7 @@ void ui_screen_done(int restart_seconds, secbool full_redraw) { void ui_screen_fail(void) { display_bar(0, 0, DISPLAY_RESX, DISPLAY_RESY, COLOR_WHITE); display_loader(1000, -20, COLOR_BL_FAIL, COLOR_WHITE, toi_icon_fail, - sizeof(toi_icon_fail), COLOR_BLACK); + sizeof(toi_icon_fail), COLOR_BLACK, 0); display_text_center(DISPLAY_RESX / 2, DISPLAY_RESY - 24, "Failed! Please, reconnect.", -1, FONT_NORMAL, COLOR_BLACK, COLOR_WHITE); diff --git a/core/embed/extmod/modtrezorui/display.c b/core/embed/extmod/modtrezorui/display.c index eba7ecfd28..f7601dd8c6 100644 --- a/core/embed/extmod/modtrezorui/display.c +++ b/core/embed/extmod/modtrezorui/display.c @@ -318,7 +318,7 @@ static void inflate_callback_loader(uint8_t byte, uint32_t pos, void display_loader(uint16_t progress, int yoffset, uint16_t fgcolor, uint16_t bgcolor, const uint8_t *icon, uint32_t iconlen, - uint16_t iconfgcolor) { + uint16_t iconfgcolor, uint16_t slice_span) { #if TREZOR_MODEL == T uint16_t colortable[16], iconcolortable[16]; set_color_table(colortable, fgcolor, bgcolor); @@ -378,7 +378,7 @@ void display_loader(uint16_t progress, int yoffset, uint16_t fgcolor, PIXELDATA(iconcolortable[c]); } else { uint8_t c; - if (progress > a) { + if (progress > a && (slice_span == 0 || progress < (a + slice_span))) { c = (img_loader[my][mx] & 0x00F0) >> 4; } else { c = img_loader[my][mx] & 0x000F; diff --git a/core/embed/extmod/modtrezorui/display.h b/core/embed/extmod/modtrezorui/display.h index 578025d49f..141aba1070 100644 --- a/core/embed/extmod/modtrezorui/display.h +++ b/core/embed/extmod/modtrezorui/display.h @@ -84,7 +84,7 @@ void display_icon(int x, int y, int w, int h, const void *data, int datalen, uint16_t fgcolor, uint16_t bgcolor); void display_loader(uint16_t progress, int yoffset, uint16_t fgcolor, uint16_t bgcolor, const uint8_t *icon, uint32_t iconlen, - uint16_t iconfgcolor); + uint16_t iconfgcolor, uint16_t slice_span); #ifndef TREZOR_PRINT_DISABLE void display_print_color(uint16_t fgcolor, uint16_t bgcolor); diff --git a/core/embed/extmod/modtrezorui/modtrezorui-display.h b/core/embed/extmod/modtrezorui/modtrezorui-display.h index 862cd11991..90c78e6c12 100644 --- a/core/embed/extmod/modtrezorui/modtrezorui-display.h +++ b/core/embed/extmod/modtrezorui/modtrezorui-display.h @@ -198,7 +198,7 @@ STATIC mp_obj_t mod_trezorui_Display_icon(size_t n_args, const mp_obj_t *args) { STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_trezorui_Display_icon_obj, 6, 6, mod_trezorui_Display_icon); /// def loader(self, progress: int, yoffset: int, fgcolor: int, bgcolor: int, -/// icon: bytes = None, iconfgcolor: int = None) -> None: +/// icon: bytes = None, iconfgcolor: int = None, slice_span: int = None) -> None: /// ''' /// Renders a rotating loader graphic. /// Progress determines its position (0-1000), fgcolor is used as foreground @@ -206,6 +206,8 @@ STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_trezorui_Display_icon_obj, 6, 6, /// icon is drawn in the middle using the color specified in iconfgcolor. /// Icon needs to be of exactly LOADER_ICON_SIZE x LOADER_ICON_SIZE pixels /// size. +/// If slice_span is defined the progress is sliced to emulate indeterminate +/// loader. /// ''' STATIC mp_obj_t mod_trezorui_Display_loader(size_t n_args, const mp_obj_t *args) { @@ -213,8 +215,11 @@ STATIC mp_obj_t mod_trezorui_Display_loader(size_t n_args, mp_int_t yoffset = mp_obj_get_int(args[2]); mp_int_t fgcolor = mp_obj_get_int(args[3]); mp_int_t bgcolor = mp_obj_get_int(args[4]); - if (n_args > 5) { // icon provided - mp_buffer_info_t icon; + mp_buffer_info_t icon = {.buf = NULL, .len=0}; + uint16_t iconfgcolor = 0; + uint16_t slice_span = 0; + + if (n_args > 5 && args[5] != mp_const_none) { // icon provided mp_get_buffer_raise(args[5], &icon, MP_BUFFER_READ); const uint8_t *data = icon.buf; if (icon.len < 8 || memcmp(data, "TOIg", 4) != 0) { @@ -229,21 +234,25 @@ STATIC mp_obj_t mod_trezorui_Display_loader(size_t n_args, if (datalen != icon.len - 12) { mp_raise_ValueError("Invalid size of data"); } - uint16_t iconfgcolor; + if (n_args > 6) { // icon color provided iconfgcolor = mp_obj_get_int(args[6]); } else { iconfgcolor = ~bgcolor; // invert } - display_loader(progress, yoffset, fgcolor, bgcolor, icon.buf, icon.len, - iconfgcolor); - } else { - display_loader(progress, yoffset, fgcolor, bgcolor, NULL, 0, 0); } + + if (n_args > 7 && args[7] != mp_const_none) { // slice span provided + slice_span = mp_obj_get_int(args[7]); + } + + display_loader(progress, yoffset, fgcolor, bgcolor, icon.buf, icon.len, + iconfgcolor, slice_span); + return mp_const_none; } STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_trezorui_Display_loader_obj, 5, - 7, mod_trezorui_Display_loader); + 8, mod_trezorui_Display_loader); /// def print(self, text: str) -> None: /// ''' diff --git a/core/mocks/generated/trezorui.py b/core/mocks/generated/trezorui.py index d42a3083c8..178902fede 100644 --- a/core/mocks/generated/trezorui.py +++ b/core/mocks/generated/trezorui.py @@ -51,12 +51,14 @@ class Display: The icon needs to be in TREZOR Optimized Image Format (TOIF) - gray-scale mode. ''' - def loader(self, progress: int, yoffset: int, fgcolor: int, bgcolor: int, icon: bytes = None, iconfgcolor: int = None) -> None: + def loader(self, progress: int, yoffset: int, fgcolor: int, bgcolor: int, icon: bytes = None, iconfgcolor: int = None, slice_span: int = None) -> None: ''' Renders a rotating loader graphic. Progress determines its position (0-1000), fgcolor is used as foreground color, bgcolor as background. When icon and iconfgcolor are provided, an icon is drawn in the middle using the color specified in iconfgcolor. Icon needs to be of exactly LOADER_ICON_SIZE x LOADER_ICON_SIZE pixels size. + If slice_span is defined the progress is sliced to emulate indeterminate + loader. ''' def print(self, text: str) -> None: