1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-27 01:48:17 +00:00
trezor-firmware/tools/codegen/gen_loader.py

39 lines
1.4 KiB
Python
Executable File

#!/usr/bin/python3
# script used to generate /embed/extmod/modtrezorui/loader.h
import math
import sys
outer = 64
inner = 42
with open('loader.h', 'wt') as f:
f.write('static const int img_loader_size = %d;\n' % outer)
f.write('static const uint16_t img_loader[%d][%d] = {\n' % (outer, outer))
for y in range(outer):
f.write(' {')
for x in range(outer):
d = math.sqrt((outer - 1 - x) ** 2 + (outer - 1 - y) ** 2)
c = {}
for i in [5, 15]:
if (inner - 0.5 <= d) and (d <= inner + 0.5):
c[i] = 15 * (d - inner + 0.5)
elif (inner + 0.5 <= d) and (d <= inner + 1.5):
c[i] = 15
elif (inner + 1.5 <= d) and (d <= inner + 2.5):
c[i] = 15 if i == 15 else 15 - (15 - i) * (d - inner - 1.5)
elif (outer - 1.5 <= d) and (d <= outer - 0.5):
c[i] = i - i * (d - outer + 1.5)
elif (inner + 2.5 < d) and (d < outer - 1.5):
c[i] = i
else:
c[i] = 0
# clamp (should not be needed)
c[i] = max(0, min(int(c[i]), 15))
a = int(math.atan2((outer - 1 - x), (outer - 1 - y)) * 2 * 249 / math.pi)
v = (a << 8) | (c[15] << 4) | c[5]
f.write('%d,' % v)
f.write('},\n')
f.write('};\n')