arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-26 09:28:13 +00:00
Code Issues Releases Wiki Activity

tests/examples: update 2048 code

Browse Source
This commit is contained in:
Pavol Rusnak 2018-12-21 23:51:22 +01:00
parent a1f4e1a7b6
commit fb495a6afc
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
1 changed files with 40 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
tests/examples/2048/main.py
View File

@ -4,37 +4,35 @@
# inspired by https://github.com/gabrielecirulli/2048 # inspired by https://github.com/gabrielecirulli/2048
# #
from trezor import ui from trezor import ui
from trezor import loop from trezor import loop
from trezor import workflow from trezor import workflow
from trezor.crypto import random from trezor.crypto import random
from trezor.ui.swipe import Swipe, SWIPE_DOWN, SWIPE_UP, SWIPE_LEFT from trezor.ui.swipe import Swipe, SWIPE_DOWN, SWIPE_UP, SWIPE_LEFT
color_bg = ui.rgb(0xbb, 0xad, 0xa0) color_bg = ui.rgb(0xBB, 0xAD, 0xA0)
color_empty = ui.rgb(0xcc, 0xc0, 0xb3) color_empty = ui.rgb(0xCC, 0xC0, 0xB3)
color_fg1 = ui.rgb(0x77, 0x6e, 0x65) color_fg1 = ui.rgb(0x77, 0x6E, 0x65)
color_fg2 = ui.rgb(0xf9, 0xf6, 0xf2) color_fg2 = ui.rgb(0xF9, 0xF6, 0xF2)
color_lose_fg = ui.rgb(0xff, 0xff, 0xff) color_lose_fg = ui.rgb(0xFF, 0xFF, 0xFF)
color_lose_bg = ui.rgb(0xff, 0x00, 0x00) color_lose_bg = ui.rgb(0xFF, 0x00, 0x00)
color_win_fg = ui.rgb(0xff, 0xff, 0xff) color_win_fg = ui.rgb(0xFF, 0xFF, 0xFF)
color_win_bg = ui.rgb(0x00, 0xff, 0x00) color_win_bg = ui.rgb(0x00, 0xFF, 0x00)
color = {} color = {}
color[2] = ui.rgb(0xee, 0xe4, 0xda), color_fg1 color[2] = ui.rgb(0xEE, 0xE4, 0xDA), color_fg1
color[4] = ui.rgb(0xed, 0xe0, 0xc8), color_fg1 color[4] = ui.rgb(0xED, 0xE0, 0xC8), color_fg1
color[8] = ui.rgb(0xf2, 0xb1, 0x79), color_fg2 color[8] = ui.rgb(0xF2, 0xB1, 0x79), color_fg2
color[16] = ui.rgb(0xf5, 0x95, 0x63), color_fg2 color[16] = ui.rgb(0xF5, 0x95, 0x63), color_fg2
color[32] = ui.rgb(0xf6, 0x7c, 0x5f), color_fg2 color[32] = ui.rgb(0xF6, 0x7C, 0x5F), color_fg2
color[64] = ui.rgb(0xf6, 0x5e, 0x3b), color_fg2 color[64] = ui.rgb(0xF6, 0x5E, 0x3B), color_fg2
color[128] = ui.rgb(0xed, 0xcf, 0x72), color_fg2 color[128] = ui.rgb(0xED, 0xCF, 0x72), color_fg2
color[256] = ui.rgb(0xed, 0xcc, 0x61), color_fg2 color[256] = ui.rgb(0xED, 0xCC, 0x61), color_fg2
color[512] = ui.rgb(0xed, 0xc8, 0x50), color_fg2 color[512] = ui.rgb(0xED, 0xC8, 0x50), color_fg2
color[1024] = ui.rgb(0xed, 0xc5, 0x3f), color_fg2 color[1024] = ui.rgb(0xED, 0xC5, 0x3F), color_fg2
color[2048] = ui.rgb(0xed, 0xc2, 0x2e), color_fg2 color[2048] = ui.rgb(0xED, 0xC2, 0x2E), color_fg2
class State(): class State:
def __init__(self): def __init__(self):
self.m = [[0 for _ in range(4)] for _ in range(4)] self.m = [[0 for _ in range(4)] for _ in range(4)]
self.add_new() self.add_new()
@ -47,32 +45,39 @@ class State():
self.m[i][j] = 2 self.m[i][j] = 2
break break
def contains(self, val):
return val in [i for l in self.m for i in l]
def render(self): def render(self):
if 0 not in self.m[0] and 0 not in self.m[1] and 0 not in self.m[2] and 0 not in self.m[3]: if not self.contains(0):
d.bar(0, 0, 240, 240, color_lose_bg) d.bar(0, 0, ui.WIDTH, ui.HEIGHT, color_lose_bg)
d.text_center(120, 128, "YOU LOSE!", ui.BOLD, color_lose_fg, color_lose_bg) d.text_center(120, 128, "YOU LOSE!", ui.BOLD, color_lose_fg, color_lose_bg)
elif 2048 in self.m[0] or 2048 in self.m[1] or 2048 in self.m[2] or 2048 in self.m[3]: elif self.contains(2048):
d.bar(0, 0, 240, 240, color_win_bg) d.bar(0, 0, ui.WIDTH, ui.HEIGHT, color_win_bg)
d.text_center(120, 128, "YOU WIN!", ui.BOLD, color_win_fg, color_win_bg) d.text_center(120, 128, "YOU WIN!", ui.BOLD, color_win_fg, color_win_bg)
else: else:
for i in range(4): for i in range(4):
for j in range(4): for j in range(4):
if not self.m[i][j]: if not self.m[i][j]:
d.bar_radius(8 + i * 58, 8 + j * 58, 50, 50, color_empty, color_bg, 2) d.bar_radius(
8 + i * 58, 8 + j * 58, 50, 50, color_empty, color_bg, 2
)
else: else:
v = self.m[i][j] v = self.m[i][j]
cb, cf = color[v] cb, cf = color[v]
d.bar_radius(8 + i * 58, 8 + j * 58, 50, 50, cb, color_bg, 2) d.bar_radius(8 + i * 58, 8 + j * 58, 50, 50, cb, color_bg, 2)
d.text_center(8 + i * 58 + 25, 8 + j * 58 + 33, str(v), ui.BOLD, cf, cb) d.text_center(
8 + i * 58 + 25, 8 + j * 58 + 33, str(v), ui.BOLD, cf, cb
)
d.backlight(ui.BACKLIGHT_NORMAL) d.backlight(ui.BACKLIGHT_NORMAL)
d.refresh() d.refresh()
def update(self, dir): def update(self, d):
for _ in range(4): for _ in range(4):
self.compact(dir) self.compact(d)
def compact(self, dir): def compact(self, d):
if dir == SWIPE_DOWN: if d == SWIPE_DOWN:
for i in range(4): for i in range(4):
for j in [2, 1, 0]: for j in [2, 1, 0]:
if self.m[i][j] == self.m[i][j + 1]: if self.m[i][j] == self.m[i][j + 1]:
@ -81,7 +86,7 @@ class State():
elif 0 == self.m[i][j + 1]: elif 0 == self.m[i][j + 1]:
self.m[i][j + 1] = self.m[i][j] self.m[i][j + 1] = self.m[i][j]
self.m[i][j] = 0 self.m[i][j] = 0
elif dir == SWIPE_UP: elif d == SWIPE_UP:
for i in range(4): for i in range(4):
for j in [1, 2, 3]: for j in [1, 2, 3]:
if self.m[i][j] == self.m[i][j - 1]: if self.m[i][j] == self.m[i][j - 1]:
@ -90,7 +95,7 @@ class State():
elif 0 == self.m[i][j - 1]: elif 0 == self.m[i][j - 1]:
self.m[i][j - 1] = self.m[i][j] self.m[i][j - 1] = self.m[i][j]
self.m[i][j] = 0 self.m[i][j] = 0
elif dir == SWIPE_LEFT: elif d == SWIPE_LEFT:
for j in range(4): for j in range(4):
for i in [1, 2, 3]: for i in [1, 2, 3]:
if self.m[i][j] == self.m[i - 1][j]: if self.m[i][j] == self.m[i - 1][j]:
@ -112,7 +117,7 @@ class State():
d = ui.Display() d = ui.Display()
d.backlight(ui.BACKLIGHT_NORMAL) d.backlight(ui.BACKLIGHT_NORMAL)
d.bar(0, 0, 240, 240, color_bg) d.bar(0, 0, ui.WIDTH, ui.HEIGHT, color_bg)
s = State() s = State()