Momentum-Apps/scripts/flipper/assets/icon.py

import logging
import argparse
import subprocess
import io
import os
import sys

ICONS_SUPPORTED_FORMATS = ["png"]


class Image:
    def __init__(self, width: int, height: int, data: bytes):
        self.width = width
        self.height = height
        self.data = data

    def write(self, filename):
        with open(filename, "wb") as file:
            file.write(self.data)

    def data_as_carray(self):
        return (
            "{" + "".join("0x{:02x},".format(img_byte) for img_byte in self.data) + "}"
        )


def is_file_an_icon(filename):
    extension = filename.lower().split(".")[-1]
    return extension in ICONS_SUPPORTED_FORMATS


class ImageTools:
    __pil_unavailable = False
    __hs2_unavailable = False

    @staticmethod
    def is_processing_slow():
        try:
            from PIL import Image, ImageOps
            import heatshrink2

            return False
        except ImportError as e:
            return True

    def __init__(self):
        self.logger = logging.getLogger()

    def png2xbm(self, file):
        if self.__pil_unavailable:
            return subprocess.check_output(["convert", file, "xbm:-"])

        try:
            from PIL import Image, ImageOps
        except ImportError as e:
            self.__pil_unavailable = True
            self.logger.info("pillow module is missing, using convert cli util")
            return self.png2xbm(file)

        with Image.open(file) as im:
            with io.BytesIO() as output:
                bw = im.convert("1")
                bw = ImageOps.invert(bw)
                bw.save(output, format="XBM")
                return output.getvalue()

    def xbm2hs(self, data):
        if self.__hs2_unavailable:
            return subprocess.check_output(
                ["heatshrink", "-e", "-w8", "-l4"], input=data
            )

        try:
            import heatshrink2
        except ImportError as e:
            self.__hs2_unavailable = True
            self.logger.info("heatshrink2 module is missing, using heatshrink cli util")
            return self.xbm2hs(data)

        return heatshrink2.compress(data, window_sz2=8, lookahead_sz2=4)


__tools = ImageTools()


def file2image(file):
    output = __tools.png2xbm(file)
    assert output

    # Extract data from text
    f = io.StringIO(output.decode().strip())
    width = int(f.readline().strip().split(" ")[2])
    height = int(f.readline().strip().split(" ")[2])
    data = f.read().strip().replace("\n", "").replace(" ", "").split("=")[1][:-1]
    data_str = data[1:-1].replace(",", " ").replace("0x", "")

    data_bin = bytearray.fromhex(data_str)

    # Encode icon data with LZSS
    data_encoded_str = __tools.xbm2hs(data_bin)

    assert data_encoded_str

    data_enc = bytearray(data_encoded_str)
    data_enc = bytearray([len(data_enc) & 0xFF, len(data_enc) >> 8]) + data_enc

    # Use encoded data only if its length less than original, including header
    if len(data_enc) < len(data_bin) + 1:
        data = b"\x01\x00" + data_enc
    else:
        data = b"\x00" + data_bin

    return Image(width, height, data)