mirror of
https://github.com/zeldaret/af.git
synced 2026-09-26 05:31:51 -04:00
Anghelo Carvajal
c7a746e71e
* find object_table and make tool to extract it * migrate object table * update the assets yaml a bit * comment * name gameplay_keep * change gameplay_keep to segment 4 * rename exclusive_ram_id to fit the segment used * review
95 lines
2.5 KiB
Python
Executable File
95 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# SPDX-FileCopyrightText: © 2023 ZeldaRET
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import dataclasses
|
|
from pathlib import Path
|
|
import struct
|
|
|
|
|
|
ENTRY_SIZE = 0x8
|
|
|
|
@dataclasses.dataclass
|
|
class ObjectOverlay:
|
|
vromStart: int # uintptr_t
|
|
vromEnd: int # uintptr_t
|
|
|
|
def rawstr(self) -> str:
|
|
result = f"0x{self.vromStart:08X}, "
|
|
result += f"0x{self.vromEnd:08X}"
|
|
|
|
return result
|
|
|
|
def isUnset(self) -> bool:
|
|
return self.vromStart == 0 and self.vromEnd == 0
|
|
|
|
def _macroEntry_Unset(self, index: int) -> str:
|
|
return f"DEFINE_OBJECT_UNSET(OBJECT_UNSET_{index})"
|
|
|
|
def _macroEntry_Normal(self, index: int) -> str:
|
|
return f"DEFINE_OBJECT(object_{self.vromStart:08X}, OBJECT_{index})"
|
|
|
|
def macroEntry(self, index: int) -> str:
|
|
if self.isUnset():
|
|
return self._macroEntry_Unset(index)
|
|
return self._macroEntry_Normal(index)
|
|
|
|
def renames(self) -> str:
|
|
result = f""
|
|
|
|
if self.vromStart != 0:
|
|
result += f'"segment_{self.vromStart:08X}": "object_{self.vromStart:08X}",\n'
|
|
|
|
return result
|
|
|
|
|
|
def extractEntry(rom: bytes, offset: int, index: int) -> ObjectOverlay:
|
|
entry = struct.unpack_from(">II", rom, offset + ENTRY_SIZE * index)
|
|
|
|
return ObjectOverlay(*entry)
|
|
|
|
|
|
def ExtractActorTableMain():
|
|
parser = argparse.ArgumentParser(description="Object table tool")
|
|
parser.add_argument("rom_path", help="Path to uncompressed rom")
|
|
parser.add_argument("offset", help="Raw offset to the object table. Either dec or hex")
|
|
parser.add_argument("length", help="Amount of entries on the actor table. Either dec or hex")
|
|
|
|
parser.add_argument("--renames", help="Print a list of strings to rename instead of the object table", action="store_true")
|
|
|
|
args = parser.parse_args()
|
|
|
|
rom_path = Path(args.rom_path)
|
|
offset = int(args.offset, 0)
|
|
length = int(args.length, 0)
|
|
renames = args.renames
|
|
|
|
with rom_path.open("rb") as f:
|
|
rom = f.read()
|
|
|
|
for i in range(length):
|
|
entry = extractEntry(rom, offset, i)
|
|
|
|
# print(entry.rawstr())
|
|
# if not entry.isUnset():
|
|
# if entry.vromStart == entry.vromEnd:
|
|
# print(entry.rawstr())
|
|
|
|
if renames:
|
|
print(entry.renames(), end="")
|
|
pass
|
|
|
|
else:
|
|
macroEntry = entry.macroEntry(i)
|
|
print(f"/* 0x{i:03X} */ {macroEntry}")
|
|
pass
|
|
|
|
|
|
if __name__ == '__main__':
|
|
ExtractActorTableMain()
|
|
|