github/Shipwright
1
0
Fork 0
mirror of https://github.com/HarbourMasters/Shipwright.git synced 2025-08-22 14:23:44 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update extract_assets.py

Browse source
This commit is contained in:
MegaMech 2022-04-09 22:06:37 -07:00 • committed by GitHub
commit 4c231f2c37
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 44 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

67
OTRExporter/extract_assets.py
View file

@ -1,12 +1,20 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import argparse, json, os, signal, time, sys, shutil, glob # How to use:
# Place a rom in this directory then run the script.
# If you are using multiple roms, the script will let you choose one.
# To choose with a commandline argument:
# Python3 extract_assets.py <number>
# Invalid input results in the first rom being selected
import json, os, signal, time, sys, shutil, glob
from multiprocessing import Pool, cpu_count, Event, Manager, ProcessError from multiprocessing import Pool, cpu_count, Event, Manager, ProcessError
from enum import Enum from enum import Enum
import shutil import shutil
romVer = "..\\soh\\baserom_non_mq.z64" romVer = "..\\soh\\baserom_non_mq.z64"
roms = []; roms = [];
checksums = ["", "", ""];
class Checksums(Enum): class Checksums(Enum):
OOT_NTSC_10 = "EC7011B7" OOT_NTSC_10 = "EC7011B7"
@ -64,17 +72,12 @@ def checkChecksum(rom):
print("Compatible roms:") print("Compatible roms:")
for compat in CompatibleChecksums: for compat in CompatibleChecksums:
print(compat.name+" | 0x"+compat.value) print(compat.name+" | 0x"+compat.value)
input("Press Enter to quit the program")
sys.exit(1) sys.exit(1)
print("Wrong rom! No valid checksum found") print("Wrong rom! No valid checksum found")
input("Press Enter to quit the program")
sys.exit(1) sys.exit(1)
def main(): def main():
parser = argparse.ArgumentParser(description="baserom asset extractor")
parser.add_argument("-v", "--version", help="Sets game version.")
args = parser.parse_args()
romToUse = ""; romToUse = "";
@ -83,37 +86,53 @@ def main():
if not (roms): if not (roms):
print("Error: No roms located, place one in the OTRExporter directory", file=os.sys.stderr) print("Error: No roms located, place one in the OTRExporter directory", file=os.sys.stderr)
input("Press Enter to quit the program")
sys.exit(1) sys.exit(1)
if (len(roms) > 1): if (len(roms) > 1):
print(str(len(roms))+" roms found, please select one by pressing 1-"+str(len(roms))) # If commandline args exist
if (len(sys.argv) > 1):
for list in range(len(roms)):
print(roms[list])
while(1):
try: try:
selection = int(input()) if ((int(sys.argv[1]) - 1) < 1):
romToUse = roms[0]
elif ((int(sys.argv[1]) - 1) > len(roms)):
romToUse = roms[len(roms) - 1]
else:
romToUse = roms[int(sys.argv[1]) - 1]
except: except:
print("Bad input. Try again with the number keys.") romToUse = roms[0]
continue
if (selection < 1 or selection > len(roms)): # No commandline args, select rom using user input
print("Bad input. Try again.") else:
continue
else: break print(str(len(roms))+" roms found, please select one by pressing 1-"+str(len(roms)))
romToUse = roms[selection - 1] count = 1
for list in range(len(roms)):
print(str(count)+". "+roms[list])
count += 1
while(1):
try:
selection = int(input())
except:
print("Bad input. Try again with the number keys.")
continue
if (selection < 1 or selection > len(roms)):
print("Bad input. Try again.")
continue
else: break
romToUse = roms[selection - 1]
else: else:
romToUse = roms[0] romToUse = roms[0]
validRom = checkChecksum(romToUse) match checkChecksum(romToUse).name:
match validRom.name:
case Checksums.OOT_PAL_GC: case Checksums.OOT_PAL_GC:
xmlVer = "GC_NMQ_PAL_F" xmlVer = "GC_NMQ_PAL_F"
case Checksums.OOT_PAL_GC_DBG1: case Checksums.OOT_PAL_GC_DBG1: