102 lines
2.6 KiB
Python
102 lines
2.6 KiB
Python
from typing import Tuple
|
|
from PIL import Image
|
|
import sys
|
|
import shared
|
|
|
|
TEMPLATE = """
|
|
// generated code! be nice!
|
|
#include "sys/sys.h"
|
|
sys_sprite spr_{{spritesheet_name}}_data[{{n_sprites}}] = {
|
|
{% for sprite in sprites -%}
|
|
{ .pixels={
|
|
{% for row in sprite -%}
|
|
{ {{ row|join(",") }} }{% if not loop.last %},{% endif %}
|
|
{% endfor %}
|
|
} }{% if not loop.last %},{% endif %}
|
|
{%- endfor %}
|
|
};
|
|
sys_spritesheet spr_{{spritesheet_name}} = {
|
|
.sprites=spr_{{spritesheet_name}}_data,
|
|
.width={{width}},
|
|
.height={{height}},
|
|
};
|
|
""".lstrip()
|
|
|
|
|
|
def main(spritesheet_name, n_sprites, key_color, fname_png, fname_c):
|
|
sprites, width, height = load_sprites(fname_png, key_color)
|
|
assert(len(sprites) == n_sprites), f"must be exactly {n_sprites} sprites, not {len(sprites)}"
|
|
|
|
with open(fname_c, "wt") as output:
|
|
output.write(
|
|
shared.templates.from_string(TEMPLATE).render(
|
|
spritesheet_name=spritesheet_name,
|
|
n_sprites=n_sprites,
|
|
sprites=sprites,
|
|
width=width,
|
|
height=height,
|
|
)
|
|
)
|
|
|
|
|
|
def load_sprites(fname_png: str, key_color: int):
|
|
width, height, data = shared.load_image(fname_png)
|
|
|
|
sprites = []
|
|
for gy in range(0, height, 8):
|
|
for gx in range(0, width, 8):
|
|
pixels = []
|
|
for py in range(0, 8):
|
|
row = []
|
|
for px in range(0, 8):
|
|
x = gx + px
|
|
y = gy + py
|
|
pal = pixel_to_palette(data[y * width + x], key_color, x, y)
|
|
row.append(pal)
|
|
pixels.append(row)
|
|
sprites.append(pixels)
|
|
|
|
return sprites, width//8, height//8
|
|
|
|
|
|
palette_colors = {
|
|
(0, 0, 0): 0,
|
|
(29, 43, 83): 1,
|
|
(126, 37, 83): 2,
|
|
(0, 135, 81): 3,
|
|
(171, 82, 54): 4,
|
|
(95, 87, 79): 5,
|
|
(194, 195, 199): 6,
|
|
(255, 241, 232): 7,
|
|
(255, 0, 77): 8,
|
|
(255, 163, 0): 9,
|
|
(255, 236, 39): 10,
|
|
(0, 228, 54): 11,
|
|
(41, 173, 255): 12,
|
|
(131, 118, 156): 13,
|
|
(255, 119, 168): 14,
|
|
(255, 204, 170): 15,
|
|
}
|
|
|
|
|
|
def pixel_to_palette(rgba, key_color, x, y):
|
|
(r, g, b, a) = rgba
|
|
|
|
if a < 128:
|
|
return 255
|
|
|
|
rgb = (r, g, b)
|
|
ix = palette_colors.get(rgb)
|
|
if ix is None:
|
|
raise ValueError(f"unrecognized color at ({x}, {y}): {rgb}")
|
|
|
|
if ix == key_color:
|
|
return 255
|
|
|
|
return ix
|
|
|
|
|
|
if __name__ == "__main__":
|
|
assert len(sys.argv) == 6, \
|
|
"there must be five args (spritesheet name, n sprites, key color, src png, out c)"
|
|
main(sys.argv[1], int(sys.argv[2]), int(sys.argv[3]), sys.argv[4], sys.argv[5]) |