mirror of
https://github.com/zeldaret/oot.git
synced 2024-11-25 09:45:02 +00:00
7ee8c8011c
* All get item objects OK * Name z_draw functions * fix skulltula tokens draw ids * Review suggestions, merge extract_single_asset into extract_assets as an optional argument Co-authored-by: Thar0 <maximilianc64@gmail.com>
54 lines
1.9 KiB
Python
Executable file
54 lines
1.9 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import os
|
|
from shutil import copyfile
|
|
from multiprocessing import Pool
|
|
from multiprocessing import cpu_count
|
|
|
|
def Extract(xmlPath, outputPath):
|
|
ExtractFile(xmlPath, outputPath, 1, 0)
|
|
|
|
def ExtractScene(xmlPath, outputPath):
|
|
ExtractFile(xmlPath, outputPath, 1, 1)
|
|
|
|
def ExtractFile(xmlPath, outputPath, genSrcFile, incFilePrefix):
|
|
execStr = "tools/ZAPD/ZAPD.out e -eh -i %s -b baserom/ -o %s -gsf %i -ifp %i -sm tools/ZAPD/SymbolMap_OoTMqDbg.txt" % (xmlPath, outputPath, genSrcFile, incFilePrefix)
|
|
|
|
print(execStr)
|
|
os.system(execStr)
|
|
|
|
def ExtractFunc(fullPath):
|
|
outPath = ("assets/" + fullPath.split("assets/xml/")[1]).split(".xml")[0]
|
|
|
|
if (fullPath.startswith("assets/xml/scenes/")):
|
|
ExtractScene(fullPath, outPath)
|
|
else:
|
|
Extract(fullPath, outPath)
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="baserom asset extractor")
|
|
parser.add_argument("-s", "--single", help="asset path relative to assets/, e.g. objects/gameplay_keep")
|
|
args = parser.parse_args()
|
|
|
|
asset_path = args.single
|
|
if asset_path is not None:
|
|
if asset_path.endswith("/"):
|
|
asset_path = asset_path[0:-1]
|
|
Extract(f"assets/xml/{asset_path}.xml", f"assets/{asset_path}/")
|
|
else:
|
|
xmlFiles = []
|
|
for currentPath, folders, files in os.walk("assets"):
|
|
for file in files:
|
|
fullPath = os.path.join(currentPath, file)
|
|
if file.endswith(".xml") and currentPath.startswith("assets/xml/"):
|
|
outPath = ("assets/" + fullPath.split("assets/xml/")[1]).split(".xml")[0]
|
|
xmlFiles.append(fullPath)
|
|
|
|
numCores = cpu_count()
|
|
print("Extracting assets with " + str(numCores) + " CPU cores.")
|
|
p = Pool(numCores)
|
|
p.map(ExtractFunc, xmlFiles)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|