bilbo/model_generator
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

color_convert: importable

Browse Source
This commit is contained in:
Zack Buhman 2024-09-13 07:20:00 -05:00
parent 9b7f0e8bf4
commit 5097e0e5ac
1 changed files with 48 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
color_convert.py
View File

@ -31,41 +31,25 @@ class color_format:
b5 = (b >> 3) & 31 b5 = (b >> 3) & 31
return (r5 << 11) | (g5 << 5) | (b5 << 0) return (r5 << 11) | (g5 << 5) | (b5 << 0)
def axxx4444(r, g, b, a):
a4 = (a >> 4) & 15
return (a4 << 12)
def from_string(s): def from_string(s):
return dict([ return dict([
("gbgr1555", color_format.gbgr1555), ("gbgr1555", color_format.gbgr1555),
("argb4444", color_format.argb4444), ("argb4444", color_format.argb4444),
("argb1555", color_format.argb1555), ("argb1555", color_format.argb1555),
("rgb565", color_format.rgb565), ("rgb565", color_format.rgb565),
("axxx4444", color_format.axxx4444),
])[s] ])[s]
in_file = sys.argv[1] def convert_colors(f, convert, colors):
format = sys.argv[2]
out_file = sys.argv[3]
palette = None
with Image.open(in_file) as im:
width, height = im.size
if not im.palette:
pixels = list(im.convert("RGBA").getdata())
else:
pixels = list(im.convert("P").getdata())
palette = list(im.palette.colors)
has_alpha = False
convert = color_format.from_string(format)
def convert_colors(f, colors):
for color in colors: for color in colors:
value = convert(*color) value = convert(*color)
f.write(struct.pack("<H", value)) f.write(struct.pack("<H", value))
if palette is None: def convert_indices(f, palette, pixels):
with open(out_file, 'wb') as f:
convert_colors(f, pixels)
else:
with open(out_file, 'wb') as f:
if len(palette) <= 4: if len(palette) <= 4:
for i in range(len(pixels) // 4): for i in range(len(pixels) // 4):
a = pixels[i * 4 + 0] a = pixels[i * 4 + 0]
@ -88,5 +72,23 @@ else:
f.write(struct.pack("<B", pixel)) f.write(struct.pack("<B", pixel))
else: else:
assert False, len(palette) assert False, len(palette)
if __name__ == "__main__":
in_file = sys.argv[1]
format = sys.argv[2]
out_file = sys.argv[3]
convert = color_format.from_string(format)
with Image.open(in_file) as im:
if not im.palette:
pixels = list(im.convert("RGBA").getdata())
with open(out_file, 'wb') as f:
convert_colors(f, convert, pixels)
else:
pixels = list(im.convert("P").getdata())
palette = list(im.palette.colors)
with open(out_file, 'wb') as f:
convert_indices(f, palette, pixels)
with open(out_file + '.pal', 'wb') as f: with open(out_file + '.pal', 'wb') as f:
convert_colors(f, [(*c, 255) for c in palette]) convert_colors(f, convert, [(*c, 255) for c in palette])