You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
168 lines
5.1 KiB
168 lines
5.1 KiB
from __future__ import annotations
|
|
|
|
import os
|
|
from pathlib import Path
|
|
from typing import BinaryIO, TextIO
|
|
|
|
from trezorlib import toif
|
|
|
|
HERE = Path(__file__).parent
|
|
CORE_DIR = HERE.parent.parent
|
|
|
|
|
|
def process_line(infile: TextIO, outfile: BinaryIO) -> None:
|
|
line = infile.readline()
|
|
data = [x.strip().lower() for x in line.split(',')]
|
|
for c in data:
|
|
if len(c) == 4:
|
|
outfile.write(bytes((int(c, 16),)))
|
|
|
|
|
|
def header_to_toif(path: str | Path) -> str:
|
|
with open(path, "r") as infile, open('tmp.toif', "wb") as outfile:
|
|
infile.readline()
|
|
name_line = infile.readline()
|
|
name = name_line.split(" ")[3].split("[")[0]
|
|
infile.readline()
|
|
magic_line = infile.readline().split(',')[3]
|
|
|
|
outfile.write(bytes((0x54,)))
|
|
outfile.write(bytes((0x4f,)))
|
|
outfile.write(bytes((0x49,)))
|
|
if "g" in magic_line:
|
|
outfile.write(bytes((ord('g'),)))
|
|
elif "G" in magic_line:
|
|
outfile.write(bytes((ord('G'),)))
|
|
elif "f" in magic_line:
|
|
outfile.write(bytes((ord('f'),)))
|
|
elif "F" in magic_line:
|
|
outfile.write(bytes((ord('F'),)))
|
|
else:
|
|
print(magic_line)
|
|
raise Exception("Unknown format")
|
|
|
|
infile.readline()
|
|
process_line(infile, outfile)
|
|
infile.readline()
|
|
process_line(infile, outfile)
|
|
infile.readline()
|
|
process_line(infile, outfile)
|
|
infile.readline()
|
|
return name
|
|
|
|
|
|
def toif_to_header(path: str | Path, name: str) -> None:
|
|
with open('tmp_c.toif', "rb") as infile, open(path, "w") as outfile:
|
|
b = infile.read(4)
|
|
outfile.write("// clang-format off\n")
|
|
outfile.write(f'static const uint8_t {name}[] = {{\n',)
|
|
outfile.write(" // magic\n",)
|
|
if b[3] == ord('f'):
|
|
outfile.write(" 'T', 'O', 'I', 'f',\n",)
|
|
elif b[3] == ord('F'):
|
|
outfile.write(" 'T', 'O', 'I', 'F',\n",)
|
|
elif b[3] == ord('g'):
|
|
outfile.write(" 'T', 'O', 'I', 'g',\n",)
|
|
elif b[3] == ord('G'):
|
|
outfile.write(" 'T', 'O', 'I', 'G',\n",)
|
|
else:
|
|
raise Exception("Unknown format")
|
|
|
|
outfile.write(" // width (16-bit), height (16-bit)\n",)
|
|
outfile.write(" ")
|
|
for i in range(4):
|
|
hex_data = infile.read(1).hex()
|
|
outfile.write(f'0x{hex_data},')
|
|
if i != 3:
|
|
outfile.write(' ')
|
|
outfile.write("\n")
|
|
|
|
outfile.write(" // compressed data length (32-bit)\n",)
|
|
outfile.write(" ")
|
|
for i in range(4):
|
|
hex_data = infile.read(1).hex()
|
|
outfile.write(f'0x{hex_data},')
|
|
if i != 3:
|
|
outfile.write(' ')
|
|
outfile.write("\n")
|
|
|
|
outfile.write(" // compressed data\n",)
|
|
outfile.write(" ")
|
|
hex_data = infile.read(1).hex()
|
|
first = True
|
|
while hex_data:
|
|
if not first:
|
|
outfile.write(' ')
|
|
first = False
|
|
outfile.write(f'0x{hex_data},')
|
|
hex_data = infile.read(1).hex()
|
|
outfile.write("\n};\n")
|
|
|
|
_byte = infile.read(1)
|
|
|
|
|
|
def reformat_c_icon(path: str | Path) -> None:
|
|
name = header_to_toif(path)
|
|
with open("tmp.toif", "rb") as f:
|
|
toi = toif.from_bytes(f.read())
|
|
im = toi.to_image()
|
|
with open("tmp_c.toif", "wb") as f:
|
|
toi = toif.from_image(im)
|
|
f.write(toi.to_bytes())
|
|
toif_to_header(path, name)
|
|
|
|
os.remove("tmp.toif")
|
|
os.remove("tmp_c.toif")
|
|
|
|
|
|
def reformat_c_icons(p: str | Path) -> None:
|
|
files = os.listdir(p)
|
|
for file in files:
|
|
if file.startswith("icon_") and file.endswith(".h"):
|
|
reformat_c_icon(os.path.join(p, file))
|
|
|
|
|
|
def reformat_toif_icon(p: str | Path) -> None:
|
|
with open(p, "rb") as f:
|
|
toi = toif.from_bytes(f.read())
|
|
im = toi.to_image()
|
|
with open(p, "wb") as f:
|
|
toi = toif.from_image(im)
|
|
f.write(toi.to_bytes())
|
|
|
|
|
|
def reformat_toif_icons(p: str | Path) -> None:
|
|
files = os.listdir(p)
|
|
for file in files:
|
|
if file.endswith(".toif"):
|
|
reformat_toif_icon(os.path.join(p, file))
|
|
|
|
|
|
def change_icon_format():
|
|
# bootloader icons
|
|
reformat_c_icons(CORE_DIR / "embed/bootloader")
|
|
|
|
# bootloader_ci icons
|
|
reformat_c_icons(CORE_DIR / "embed/bootloader_ci")
|
|
|
|
# rust icons
|
|
reformat_toif_icons(CORE_DIR / "embed/rust/src/ui/model_tr/res")
|
|
reformat_toif_icons(CORE_DIR / "embed/rust/src/ui/model_tt/res")
|
|
|
|
# python icons
|
|
reformat_toif_icons(CORE_DIR / "src/trezor/res")
|
|
reformat_toif_icons(CORE_DIR / "src/trezor/res/header_icons")
|
|
|
|
# vendor header icons
|
|
reformat_toif_icon(CORE_DIR / "embed/vendorheader/vendor_satoshilabs.toif")
|
|
reformat_toif_icon(CORE_DIR / "embed/vendorheader/vendor_unsafe.toif")
|
|
|
|
# additional python icons
|
|
# reformat_toif_icon(CORE_DIR / "src/apps/homescreen/res/bg.toif") - unchanged - using as avatar
|
|
reformat_toif_icon(CORE_DIR / "src/apps/management/res/small-arrow.toif")
|
|
reformat_toif_icon(CORE_DIR / "src/apps/webauthn/res/icon_webauthn.toif")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
change_icon_format()
|