mirror of
https://github.com/zeldaret/oot.git
synced 2025-08-24 16:01:26 +00:00
8411c34b38
Builds gc-eu-mq-dbg OK from clean after 1) make setup 2) python3 -m tools.assets.extract -j 3) replace 0x80A8E610 with sShadowTex in extracted/gc-eu-mq-dbg/assets/overlays/ovl_En_Jsjutan/sShadowMaterialDL.inc.c 4) make various symbols in extracted data like sTex static
109 lines
3.9 KiB
Python
Executable file
109 lines
3.9 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
VERBOSE = False
|
|
|
|
from n64 import G_IM_FMT, G_IM_SIZ
|
|
import n64yatc
|
|
from png2raw import png2raw
|
|
|
|
|
|
def main():
|
|
png_path = Path(sys.argv[1])
|
|
out_bin_path = Path(sys.argv[2])
|
|
|
|
suffixes = png_path.suffixes
|
|
assert len(suffixes) >= 2
|
|
assert suffixes[-1] == ".png"
|
|
suffixes.pop()
|
|
if suffixes[-1] in {".u64", ".u32"}:
|
|
suffixes.pop()
|
|
assert len(suffixes) > 0
|
|
if suffixes[-1].startswith(".tlut_"):
|
|
tlut_info = suffixes.pop().removeprefix(".")
|
|
else:
|
|
tlut_info = None
|
|
assert len(suffixes) > 0
|
|
fmtsiz_str = suffixes[-1].removeprefix(".")
|
|
|
|
fmt, siz = None, None
|
|
for candidate_fmt in G_IM_FMT:
|
|
for candidate_siz in G_IM_SIZ:
|
|
candidate_fmtsiz_str = f"{candidate_fmt.name.lower()}{candidate_siz.bpp}"
|
|
if candidate_fmtsiz_str == fmtsiz_str:
|
|
fmt = candidate_fmt
|
|
siz = candidate_siz
|
|
|
|
assert fmt is not None and siz is not None, fmtsiz_str
|
|
|
|
if fmt != G_IM_FMT.CI:
|
|
with png2raw.Instance(png_path) as png:
|
|
data_rgba32 = png.read_to_rgba32()
|
|
tex_bin = n64yatc.convert(data_rgba32, G_IM_FMT.RGBA, G_IM_SIZ._32b, fmt, siz)
|
|
# print(len(tex_bin), tex_bin[:0x10], tex_bin[-0x10:], file=sys.stderr)
|
|
# sys.stdout.buffer.write(tex_bin) # for some reason the *string* "None." is also written to stdout???
|
|
out_bin_path.write_bytes(tex_bin)
|
|
else:
|
|
# TODO probably move tlut_info and overall tex file suffix construction/parsing to its own library
|
|
|
|
if tlut_info is None:
|
|
tlut_elem_type = "u64"
|
|
tlut_out_bin_path_base_str = str(out_bin_path)
|
|
tlut_out_bin_path_base_str = tlut_out_bin_path_base_str.removesuffix(".bin")
|
|
if tlut_out_bin_path_base_str.endswith(".u64"):
|
|
tlut_out_bin_path_base_str = tlut_out_bin_path_base_str.removesuffix(
|
|
".u64"
|
|
)
|
|
all_pngs_using_tlut = [png_path]
|
|
else:
|
|
tlut_elem_type = "u64"
|
|
if tlut_info.endswith("_u64"):
|
|
tlut_elem_type = "u64"
|
|
tlut_name = tlut_info.removeprefix("tlut_").removesuffix("_u64")
|
|
elif tlut_info.endswith("_u32"):
|
|
tlut_elem_type = "u32"
|
|
tlut_name = tlut_info.removeprefix("tlut_").removesuffix("_u32")
|
|
else:
|
|
tlut_name = tlut_info.removeprefix("tlut_")
|
|
tlut_out_bin_path_base_str = str(out_bin_path.parent / tlut_name)
|
|
|
|
# TODO this is far from perfect.
|
|
# what if a tlut_name is included in another
|
|
# what if not in the same folder (just don't support that)
|
|
# does the same png get built several times
|
|
all_pngs_using_tlut = list(png_path.parent.glob(f"*.tlut_{tlut_name}*.png"))
|
|
assert png_path in all_pngs_using_tlut
|
|
tlut_out_bin_path = Path(
|
|
f"{tlut_out_bin_path_base_str}.tlut.rgba16.{tlut_elem_type}.bin"
|
|
)
|
|
|
|
if VERBOSE:
|
|
print(all_pngs_using_tlut)
|
|
|
|
with png2raw.Instance(png_path) as png:
|
|
palette_rgba32 = png.get_palette_rgba32()
|
|
data_ci8 = png.read_palette_indices()
|
|
tex_bin = n64yatc.convert(data_ci8, G_IM_FMT.CI, G_IM_SIZ._8b, fmt, siz)
|
|
tlut_bin = n64yatc.convert(
|
|
palette_rgba32, G_IM_FMT.RGBA, G_IM_SIZ._32b, G_IM_FMT.RGBA, G_IM_SIZ._16b
|
|
)
|
|
out_bin_path.write_bytes(tex_bin)
|
|
tlut_out_bin_path.write_bytes(tlut_bin)
|
|
|
|
import subprocess
|
|
|
|
# HACK since the makefile doesn't know the tlut file should be built (bin2c'd), build it here
|
|
cmd = [
|
|
"tools/assets/bin2c/bin2c",
|
|
tlut_elem_type,
|
|
]
|
|
with tlut_out_bin_path.open("rb") as fin:
|
|
with tlut_out_bin_path.with_suffix(".inc.c").open("w") as fout:
|
|
subprocess.check_call(cmd, stdin=fin, stdout=fout)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|