mirror of
https://github.com/zeldaret/mm.git
synced 2026-06-20 23:51:12 -04:00
Derek Hensley
471d86f530
* Rename outputs * Makefile target renames * Add run target * yeet z64compress * venv * baserom_uncompressed -> baserom-decompressed * input rom name to baserom.z64 * Add BUILD_DIR makefile variable * Move built roms to build dir * Move baserom to baseroms folder * Add version to map file name * Makefile cleanup * Rename ldscript to include version * Multiversion build * n64-us version name * Remove venv as dependency of setup * Readme wording * extract_baserom.py suggestion * Readd checksums * Make .venv work with windows * missed an endif * Cleaner windows venv implementation * Remove duplciate process * Build process steps * Move make_options back * Fix schedule build directory * Fix schedule includes * Makefile NON_MATCHING check -> != 0 * OOT 1704 changes * Small cleanups * Missed 1 thing * UNSET -> SYMS * Update extract_baserom.py * dmadata.py * Small cleanup * dmadata_start * Format * dmadata files * Fix makefile comment * Jenkins report fix * extracted dir * Python dependencies order in readme
80 lines
2.2 KiB
Python
Executable File
80 lines
2.2 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
|
|
import sys
|
|
|
|
import dmadata
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(
|
|
description="Extract segments from an uncompressed ROM, based on its dmadata."
|
|
)
|
|
parser.add_argument(
|
|
"rom", metavar="ROM", type=Path, help="Path to uncompressed ROM"
|
|
)
|
|
parser.add_argument(
|
|
"-o",
|
|
"--output-dir",
|
|
type=Path,
|
|
required=True,
|
|
help="Output directory for segments",
|
|
)
|
|
parser.add_argument(
|
|
"--dmadata-start",
|
|
type=lambda s: int(s, 16),
|
|
required=True,
|
|
help=(
|
|
"The dmadata location in the rom, as a hexadecimal offset (e.g. 0x12f70)"
|
|
),
|
|
)
|
|
parser.add_argument(
|
|
"--dmadata-names",
|
|
type=Path,
|
|
required=True,
|
|
help="Path to file containing segment names",
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
rom_data = memoryview(args.rom.read_bytes())
|
|
|
|
dma_names = args.dmadata_names.read_text().splitlines()
|
|
dma_entries = dmadata.read_dmadata(rom_data, args.dmadata_start)
|
|
if len(dma_names) != len(dma_entries):
|
|
print(
|
|
f"Error: expected {len(dma_names)} DMA entries but found {len(dma_entries)} in ROM",
|
|
file=sys.stderr,
|
|
)
|
|
exit(1)
|
|
|
|
args.output_dir.mkdir(parents=True, exist_ok=True)
|
|
for dma_name, dma_entry in zip(dma_names, dma_entries):
|
|
if dma_entry.is_syms():
|
|
segment_rom_start = dma_entry.vrom_start
|
|
elif not dma_entry.is_compressed():
|
|
segment_rom_start = dma_entry.rom_start
|
|
else: # Segment compressed
|
|
print(f"Error: segment {dma_name} is compressed", file=sys.stderr)
|
|
exit(1)
|
|
|
|
segment_rom_end = segment_rom_start + (
|
|
dma_entry.vrom_end - dma_entry.vrom_start
|
|
)
|
|
|
|
segment_data = rom_data[segment_rom_start:segment_rom_end]
|
|
segment_path = args.output_dir / dma_name
|
|
segment_path.write_bytes(segment_data)
|
|
|
|
print(f"Extracted {len(dma_entries)} segments to {args.output_dir}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|