summaryrefslogtreecommitdiff
path: root/tools/disasm/file_addresses.py
blob: 61761478d8b73fe84650b0ea2485f5f04d994a5b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
# SPDX-FileCopyrightText: © 2024 ZeldaRET
# SPDX-License-Identifier: CC0-1.0

import csv
from dataclasses import dataclass
from pathlib import Path
from typing import List, Optional


@dataclass
class DmaFile:
    name: str
    vrom_start: int
    vrom_end: int
    rom_start: int
    rom_end: int
    vram_start: int
    overlay_dir: Optional[str]


def parse_file_addresses(path: Path) -> List[DmaFile]:
    result = []
    with open(path) as f:
        reader = csv.DictReader(f)
        for row in reader:
            result.append(
                DmaFile(
                    name=row["name"],
                    vrom_start=int(row["vrom_start"], 16),
                    vrom_end=int(row["vrom_end"], 16),
                    rom_start=int(row["rom_start"], 16),
                    rom_end=int(row["rom_end"], 16),
                    vram_start=int(row["vram_start"], 16),
                    overlay_dir=row["overlay_dir"] if row["overlay_dir"] else None,
                )
            )
    return result


def get_z_name_for_overlay(filename: str) -> str:
    if filename == "ovl_player_actor":
        return "z_player"
    elif filename.startswith("ovl_Effect_"):
        return "z_eff_" + filename[len("ovl_Effect_") :].lower()
    else:
        return "z_" + filename[len("ovl_") :].lower()