mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-25 17:09:44 +00:00
48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
|
"""
|
||
|
|
Creates a header file containing image data.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import click
|
||
|
|
import pathlib
|
||
|
|
|
||
|
|
h_file_template = """\
|
||
|
|
// clang-format off
|
||
|
|
unsigned char {name}_jpg[] = {content};
|
||
|
|
unsigned int {name}_jpg_len = {length};
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
@click.command()
|
||
|
|
@click.argument("infile", type=click.File("rb"))
|
||
|
|
def convert(infile):
|
||
|
|
|
||
|
|
path = pathlib.Path(infile.name)
|
||
|
|
|
||
|
|
img_name = path.stem
|
||
|
|
|
||
|
|
h_file_name = path.with_suffix(".h")
|
||
|
|
|
||
|
|
image_data = infile.read()
|
||
|
|
|
||
|
|
column_count = 12
|
||
|
|
|
||
|
|
content = "{{\n{image_bytes}\n}}"
|
||
|
|
image_bytes = " " # begin with indent
|
||
|
|
for index, byte in enumerate(image_data, start=1):
|
||
|
|
image_bytes += f"0x{byte:02x},"
|
||
|
|
# If at the end of line, include a newline with indent, otherwise just space
|
||
|
|
if index % column_count == 0:
|
||
|
|
image_bytes += "\n "
|
||
|
|
else:
|
||
|
|
image_bytes += " "
|
||
|
|
|
||
|
|
# Get rid of trailing coma
|
||
|
|
image_bytes = image_bytes.rstrip(", \n")
|
||
|
|
|
||
|
|
with open(h_file_name, "w") as f:
|
||
|
|
f.write(h_file_template.format(name=img_name, content=content.format(image_bytes=image_bytes), length=len(image_data)))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
convert()
|