mirror of https://github.com/zeldaret/mm.git
53 lines
1.3 KiB
Python
Executable File
53 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# SPDX-FileCopyrightText: © 2024 ZeldaRET
|
|
# SPDX-License-Identifier: CC0-1.0
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
from pathlib import Path
|
|
|
|
from version import version_config
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(
|
|
description="Extract incbin pieces from an uncompressed ROM."
|
|
)
|
|
parser.add_argument(
|
|
"baserom_segments_dir",
|
|
type=Path,
|
|
help="Directory of uncompressed ROM segments",
|
|
)
|
|
parser.add_argument(
|
|
"output_dir",
|
|
type=Path,
|
|
help="Output directory for incbin pieces",
|
|
)
|
|
parser.add_argument(
|
|
"-v",
|
|
"--version",
|
|
required=True,
|
|
help="MM version",
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
config = version_config.load_version_config(args.version)
|
|
|
|
args.output_dir.mkdir(parents=True, exist_ok=True)
|
|
for incbin in config.incbins:
|
|
incbin_path = args.output_dir / incbin.name
|
|
with open(args.baserom_segments_dir / incbin.segment, "rb") as f:
|
|
offset = incbin.vram - config.dmadata_segments[incbin.segment].vram
|
|
f.seek(offset)
|
|
incbin_data = f.read(incbin.size)
|
|
incbin_path.write_bytes(incbin_data)
|
|
|
|
print(f"Extracted {len(config.incbins)} incbin pieces to {args.output_dir}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|