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
|
#!/usr/bin/env python3
import argparse
import os
from pathlib import Path
if os.name == "nt":
DEFAULT_DOLPHIN_CONFIG_PATH = Path(os.getenv('APPDATA')) / "Dolphin Emulator"
else:
DEFAULT_DOLPHIN_CONFIG_PATH = Path("~/.var/app/org.DolphinEmu.dolphin-emu/data/dolphin-emu").expanduser()
def expanded_path(path_str: str):
path = Path(path_str)
path = path.expanduser()
return path
parser = argparse.ArgumentParser()
parser.add_argument(
"vanilla_iso_path",
type=expanded_path,
help="Path to a vanilla Wind Waker ISO to use as a base.",
)
parser.add_argument(
"output_iso_path",
type=expanded_path,
help="Path to put the modified ISO.",
)
parser.add_argument(
"decomp_repo_path",
type=expanded_path,
help="Path to the root of the git repository containing the tww decompilation.",
)
parser.add_argument(
"--map",
type=expanded_path,
default=DEFAULT_DOLPHIN_CONFIG_PATH / "Maps",
help="Folder to place the symbol map for the modified ISO (defaults to Dolphin's maps directory).",
)
args = parser.parse_args()
decomp_build_path = args.decomp_repo_path / "build/GZ2E01"
import sys
import os.path
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "gclib"))
from gclib.gcm import GCM
from gclib.rarc import RARC
from gclib.yaz0_yay0 import Yaz0
from io import BytesIO
import shutil
gcm = GCM(args.vanilla_iso_path)
gcm.read_entire_disc()
rels_arc = RARC(gcm.read_file_data("files/RELS.arc"))
rels_arc.read()
for rel_name in os.listdir(decomp_build_path):
if os.path.isfile(decomp_build_path / rel_name):
continue
if not os.path.isfile(decomp_build_path / rel_name / (rel_name + ".rel")):
continue
decomp_rel_path = decomp_build_path / rel_name / (rel_name + ".rel")
with open(decomp_rel_path, "rb") as f:
decomp_rel_data = BytesIO(f.read())
rel_file_entry = rels_arc.get_file_entry(rel_name.lower() + ".rel")
if rel_file_entry:
rel_file_entry.data = Yaz0.compress(decomp_rel_data)
else:
gcm_rel_file_path = f"files/rel/Final/Release/{rel_name}.rel"
assert gcm_rel_file_path in gcm.files_by_path, f"Invalid REL path: {gcm_rel_file_path}"
gcm.changed_files[gcm_rel_file_path] = decomp_rel_data
rels_arc.save_changes()
gcm.changed_files["files/RELS.arc"] = rels_arc.data
with open(decomp_build_path / "framework.dol", "rb") as f:
gcm.changed_files["sys/main.dol"] = BytesIO(f.read())
with open(decomp_build_path / "framework.elf.MAP", "rb") as f:
gcm.changed_files["files/map/Final/Release/frameworkF.map"] = BytesIO(f.read())
shutil.copy(decomp_build_path / "framework.elf.MAP", args.map / "GZ2E01.map")
for _ in gcm.export_disc_to_iso_with_changed_files(args.output_iso_path): pass
print("Done")
|
