You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 

140 lines
3.6 KiB

import argparse
import pathlib
import subprocess
import sys
import tempfile
from typing import NoReturn
from PIL import Image
GREEN = (0, 255, 0)
RED = (255, 0, 0)
def abort(msg: str) -> NoReturn:
print(msg, file=sys.stderr)
sys.exit(1)
def generate_coll_map(
in_path: pathlib.Path, width: int, height: int, compress: bool = False
) -> str:
png = Image.open(in_path).convert("RGB")
if png.width % 8 != 0 or png.height % 8 != 0:
abort(f"file '{in_path}' has invalid dimensions (should be multiple of 8)")
if png.width // 8 != width or png.height // 8 != height:
abort(f"file '{in_path}' has different size from map")
out_bytes = []
bits = []
for y in range(png.height // 8):
for x in range(png.width // 8):
pixel = png.getpixel((x * 8, y * 8))
bit = None
if pixel == RED:
bit = 0
elif pixel == GREEN:
bit = 1
else:
abort(f"unsupported pixel in collision map: {pixel}")
if compress:
bits.append(bit)
if len(bits) == 8:
byte = sum([bit << i for i, bit in enumerate(bits)])
out_bytes.append(byte)
bits = []
else:
out_bytes.append(bit)
png.close()
return format_bytes(out_bytes, width=width)
def format_bytes(data: bytes, width: int = 16) -> str:
lines = []
for line_no in range(0, len(data), width):
line = data[line_no : line_no + width]
lines.append(" DB " + ", ".join(["$%02X" % b for b in line]))
return "\n".join(lines)
def generate_map(pngfile: str, compress: bool = False) -> None:
pngpath = pathlib.Path(pngfile).resolve()
incpath = pngpath.parent / pngpath.name.replace(".png", ".inc")
spath = pngpath.parent / pngpath.name.replace(".png", ".s")
png = Image.open(pngpath)
if png.width % 8 != 0 or png.height % 8 != 0:
abort(f"file '{pngfile}' has invalid dimensions (should be multiple of 8)")
width = png.width // 8
height = png.height // 8
png.close()
with tempfile.NamedTemporaryFile() as tilef, tempfile.NamedTemporaryFile() as mapf:
subprocess.run(
[
"rgbgfx",
"-u",
"-t",
tilef.name,
"-o",
mapf.name,
pngfile,
]
)
map_data = format_bytes(tilef.read(), width=width)
tile_data = format_bytes(mapf.read())
section = pngpath.name.replace(".png", "")
collpath = pngpath.parent / pngpath.name.replace(".png", "_coll.png")
coll_map = generate_coll_map(collpath, width, height, compress=compress)
with open(incpath, "w") as outf:
outf.write(
f"""DEF {section}_WIDTH EQU {width}
DEF {section}_HEIGHT EQU {height}
DEF {section}_NUM_TILES EQUS "({section}_TILES_end - {section}_TILES)"
DEF {section}_MAP_SIZE EQUS "({section}_MAP_end - {section}_MAP)"
ASSERT {section}_MAP_SIZE == {section}_WIDTH * {section}_HEIGHT
"""
)
with open(spath, "w") as outf:
outf.write(
f"""SECTION "MAP - {section}", ROMX
{section}_MAP::
{map_data}
{section}_MAP_end::
{section}_TILES::
{tile_data}
{section}_TILES_end::
{section}_COLLISION::
{coll_map}
{section}_COLLISION_end::
"""
)
def main() -> None:
parser = argparse.ArgumentParser("generate_map")
parser.add_argument("-c", "--compress", default=False)
parser.add_argument("png")
args = parser.parse_args()
generate_map(args.png, compress=(not not args.compress))
if __name__ == "__main__":
main()