mirror of
https://github.com/Next-Flip/Momentum-Firmware.git
synced 2025-10-11 07:32:34 +02:00
83 lines
2.0 KiB
Python
83 lines
2.0 KiB
Python
import heatshrink2
|
|
import argparse
|
|
import os
|
|
|
|
|
|
def padded_hex(i, l):
|
|
given_int = i
|
|
given_len = l
|
|
|
|
hex_result = hex(given_int)[2:] # remove '0x' from beginning of str
|
|
num_hex_chars = len(hex_result)
|
|
extra_zeros = "0" * (given_len - num_hex_chars) # may not get used..
|
|
|
|
return (
|
|
"0x" + hex_result
|
|
if num_hex_chars == given_len
|
|
else "?" * given_len
|
|
if num_hex_chars > given_len
|
|
else "0x" + extra_zeros + hex_result
|
|
if num_hex_chars < given_len
|
|
else None
|
|
)
|
|
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description="Turn cooked Flipper .bm files back into .xbm"
|
|
)
|
|
|
|
parser.add_argument("infile", metavar="i", help="Input file")
|
|
parser.add_argument("outfile", metavar="o", help="File to write to")
|
|
parser.add_argument(
|
|
"Width",
|
|
metavar="W",
|
|
type=int,
|
|
nargs="?",
|
|
default="128",
|
|
help="Width of the image. Find from meta.txt or directory name",
|
|
)
|
|
parser.add_argument(
|
|
"Height",
|
|
metavar="H",
|
|
type=int,
|
|
nargs="?",
|
|
default="64",
|
|
help="Height of the image. Find from meta.txt or directory name",
|
|
)
|
|
|
|
args = vars(parser.parse_args())
|
|
|
|
with open(args["infile"], "rb") as f:
|
|
fileStream = f.read()
|
|
filename = os.path.splitext(os.path.basename(args["outfile"]))[0]
|
|
|
|
|
|
imageWidth = args["Width"]
|
|
imageHeight = args["Height"]
|
|
|
|
|
|
# remove headers and padding
|
|
if fileStream[0:2] == bytes([0x01, 0x00]):
|
|
unpad = fileStream[4:]
|
|
else:
|
|
if fileStream[0:1] == bytes([0x00]):
|
|
unpad = fileStream[2:]
|
|
|
|
|
|
# lzss decompress
|
|
data_decoded_str = heatshrink2.decompress(unpad, window_sz2=8, lookahead_sz2=4)
|
|
|
|
# turn it back into xbm
|
|
|
|
b = list(data_decoded_str)
|
|
c = ", ".join(padded_hex(my_int, 2) for my_int in b)
|
|
|
|
width_out = "#define " + filename + "_width " + str(imageWidth) + "\n"
|
|
height_out = "#define " + filename + "_height " + str(imageHeight) + "\n"
|
|
bytes_out = "static unsigned char " + filename + "_bits[] = {" + str(c) + "};"
|
|
|
|
data = width_out + height_out + bytes_out
|
|
|
|
with open(args["outfile"], "w") as f:
|
|
f.write(data)
|