sotn-decomp/diff_settings.py at master · Xeeynamo/sotn-decomp · GitHub

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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#!/usr/bin/env python3

import os
import re


def add_custom_arguments(parser):
    parser.add_argument(
        "--version",
        default=os.getenv("VERSION") or "us",
        dest="version",
        help="Decide what version of the game to use (us, jp12, pspeu, etc.)",
    )
    parser.add_argument(
        "--overlay",
        default="dra",
        dest="overlay",
        help="Defines which overlay to use for the diff (main, dra, mad, etc.)",
    )


def apply(config, args):
    version = args.version
    name = re.sub(
        r"^(?:st|bos?s?)/?(?=[a-z])",
        "",
        f"{args.overlay.lower() if version != 'us' else args.overlay.upper()}",
        flags=re.IGNORECASE,
    )

    baseimgRoot = {
        "us": "disks/us",
        "hd": "disks/pspeu/PSP_GAME/USRDIR/res/ps/hdbin",
        "pspeu": "disks/pspeu/PSP_GAME/USRDIR/res/ps/PSPBIN",
        "saturn": "disks/saturn",
    }
    binaryExt = {"us": "BIN", "hd": "bin", "pspeu": "bin", "saturn": "PRG"}
    path_prefix = ""
    name_prefix = ""

    if name.lower().startswith("w_"):
        name = name.lower().replace("w_", "w0_")
        path_prefix = "BIN" if version == "us" else ""
        name_prefix = "weapon/"
        config["myimg"] = os.path.join(f"build/{version}", f"{name_prefix}{name}.bin")
        config["source_directories"] = (
            [f"src/saturn/"]
            if version == "saturn"
            else [
                "include",
                "src/weapon",
                f"asm/{version}/weapon/data",
            ]
        )
    else:
        if name.lower() in os.listdir("src/st"):
            name_prefix = "st"
            path_prefix = f"ST/{name}" if version == "us" else ""
        elif name.lower() in os.listdir("src/boss"):
            name_prefix = "bo"
            path_prefix = f"BOSS/{name}" if version == "us" else ""
        elif version == "us" and name.lower() in os.listdir("src/servant"):
            path_prefix = "SERVANT"
        elif version == "us" and f"{name}.BIN" in os.listdir(f"disks/us/BIN"):
            path_prefix = "BIN"

        config["myimg"] = os.path.join(
            f"build/{version}", f"{name}.{binaryExt[version]}"
        )
        if version == "saturn":
            config["source_directories"] = [f"src/saturn/"]
        else:
            config["source_directories"] = [
                "include",
                os.path.join(
                    "src",
                    name_prefix.replace("bo", "boss"),
                    name.lower().replace("tt_", "servant/tt_"),
                ),
                os.path.join(
                    f"asm/{version}",
                    name_prefix.replace("bo", "boss").replace("tt_", "servant/tt_"),
                    name.lower().replace("tt_", "servant/tt_"),
                ),
            ]
            if version == "pspeu" and not name.startswith("tt_"):
                config["source_directories"].append(
                    os.path.join(
                        "src", name_prefix.replace("bo", "boss"), f"{name}_psp"
                    )
                )

    config["mapfile"] = os.path.join(
        f"build/{version}", f"{name_prefix}{name.lower()}.map"
    )

    if version == "us" or version == "saturn":
        name = re.sub(r"w0_\d{3}", "WEAPON0", name)

    config["baseimg"] = os.path.join(
        baseimgRoot[version],
        path_prefix,
        f"{name.lower() if version == 'hd' else name}.{binaryExt[version]}",
    )

    config["objdump_executable"] = (
        "sh-elf-objdump" if version == "saturn" else "mipsel-linux-gnu-objdump"
    )
    config["arch"] = {
        "us": "mipsel",
        "hd": "mipsel",
        "pspeu": "mipsel:4000",
        "saturn": "sh2",
    }[version]