papermario/tools/build/sprite/header.py
Marijn van der Werf 30aa49d4a8
Clean trailing whitespace (#1038)
* Add missing newlines

* Remove trailing whitepace
2023-04-29 15:28:06 +02:00

59 lines
1.9 KiB
Python
Executable File

#! /usr/bin/python3
from sys import argv, path
from pathlib import Path
path.append(str(Path(__file__).parent.parent.parent / "splat"))
path.append(str(Path(__file__).parent.parent.parent / "splat_ext"))
from pm_npc_sprites import Sprite
if __name__ == "__main__":
if len(argv) < 4:
print("usage: gen_sprite_animations_h.py [OUT] [DIR] [ID]")
exit(1)
_, outfile, sprite_dir, s = argv
with open(outfile, "w") as f:
# get sprite index
s = int(s)
assert s >= 1
sprite_dir = Path(sprite_dir)
sprite = Sprite.from_dir(sprite_dir, read_images=False)
sprite_name = sprite_dir.stem
f.write(f"#ifndef _NPC_SPRITE_{sprite_name.upper()}_H_\n")
f.write(f"#define _NPC_SPRITE_{sprite_name.upper()}_H_\n")
f.write("\n")
f.write('#include "types.h"\n')
f.write("\n")
# sprite definition
f.write(f"#define SPR_{sprite_name} 0x{s:02X}\n")
f.write("\n")
# definitions for images
for i, image_name in enumerate(sprite.image_names):
f.write(f"#define SPR_IMG_{sprite_name}_{image_name} 0x{i:X}\n")
f.write("\n")
# definitions for palettes
for p, palette_name in enumerate(sprite.palette_names):
if palette_name == "Default":
f.write(f"#define SPR_PAL_{sprite_name} 0x{p:X}\n")
else:
f.write(f"#define SPR_PAL_{sprite_name}_{palette_name} 0x{p:X}\n")
f.write("\n")
# definitions for animations
for p, palette_name in enumerate(sprite.palette_names):
for a, name in enumerate(sprite.animation_names):
if palette_name == "Default":
f.write(f"#define ANIM_{sprite_name}_{name} 0x{s:02X}{p:02X}{a:02X}\n")
else:
f.write(f"#define ANIM_{sprite_name}_{palette_name}_{name} 0x{s:02X}{p:02X}{a:02X}\n")
f.write("\n")
f.write("#endif\n")