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

feat(tools): add toif conversion from greyscale with transparency (alpha channel)

Browse Source 
[no changelog]
This commit is contained in:
tychovrahe 2022-09-05 12:41:23 +02:00 committed by TychoVrahe
parent a41cfadbf1
commit 661ae37506
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
python/src/trezorlib/toif.py
View File

@ -73,6 +73,17 @@ def _from_pil_grayscale(pixels: Sequence[int]) -> bytes:
return bytes(data)
def _from_pil_grayscale_alpha(pixels: Sequence[Tuple[int, int]]) -> bytes:
data = bytearray()
for i in range(0, len(pixels), 2):
left_w_alpha, right_w_alpha = pixels[i], pixels[i + 1]
left = int((left_w_alpha[0] * left_w_alpha[1]) / 255)
right = int((right_w_alpha[0] * right_w_alpha[1]) / 255)
c = (left & 0xF0) | ((right & 0xF0) >> 4)
data += struct.pack(">B", c)
return bytes(data)
def _to_grayscale(data: bytes) -> bytes:
res = bytearray()
for pixel in data:
@ -161,6 +172,11 @@ def from_image(
if image.size[0] % 2 != 0:
raise ValueError("Only even-width grayscale images are supported")
toif_data = _from_pil_grayscale(image.getdata())
elif image.mode == "LA":
toif_mode = firmware.ToifMode.grayscale
if image.size[0] % 2 != 0:
raise ValueError("Only even-width grayscale images are supported")
toif_data = _from_pil_grayscale_alpha(image.getdata())
elif image.mode == "RGB":
toif_mode = firmware.ToifMode.full_color
toif_data = _from_pil_rgb(image.getdata())