mirror of
https://github.com/zeldaret/oot.git
synced 2024-11-25 09:45:02 +00:00
Add actorfixer.py ~~and graphovl.py~~; and a few improvements to extract_assets.py (#783)
* Only process touched files and abort the extraction if one file wasn't extracted properly Signed-off-by: angie <angheloalf95@gmail.com> * add actorfixer and graphovl Signed-off-by: angie <angheloalf95@gmail.com> * Re set execution permisions to diff.py Signed-off-by: angie <angheloalf95@gmail.com> * Add graphs/ to gitignore Signed-off-by: angie <angheloalf95@gmail.com> * Add looners flag to graphovl Signed-off-by: angie <angheloalf95@gmail.com> * Parse macros Signed-off-by: angie <angheloalf95@gmail.com> * cleanup Signed-off-by: angie <angheloalf95@gmail.com> * Add enum parser to graphovl Signed-off-by: angie <angheloalf95@gmail.com> * Remove redundant code Signed-off-by: angie <angheloalf95@gmail.com> * cleanup Signed-off-by: angie <angheloalf95@gmail.com> * Custom colors to graphovl! * Select multiples styles for graphovl * Add solarized light style * Add renames of #796 * Add unaccounted warning flag * remove graphovl * git subrepo clone https://github.com/AngheloAlf/graphovl.git tools/graphovl subrepo: subdir: "tools/graphovl" merged: "577e71592" upstream: origin: "https://github.com/AngheloAlf/graphovl.git" branch: "master" commit: "577e71592" git-subrepo: version: "0.4.3" origin: "???" commit: "???" * remove graphovl * Add `graphovl/` to gitignore
This commit is contained in:
Anghelo Carvajal
GitHub
parent
5c4fdb706b
commit
5062f785fc
4 changed files with 210 additions and 25 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
|
@ -36,6 +36,7 @@ tools/asmsplitter/asm/*
|
|||
tools/asmsplitter/c/*
|
||||
ctx.c
|
||||
tools/*dSYM/
|
||||
graphs/
|
||||
|
||||
# Assets
|
||||
*.png
|
||||
|
|
|
|||
|
|
@ -1,55 +1,100 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
import argparse
|
||||
from multiprocessing import Pool, cpu_count, Event
|
||||
import os
|
||||
from shutil import copyfile
|
||||
from multiprocessing import Pool
|
||||
from multiprocessing import cpu_count
|
||||
|
||||
# Returns True if outFile doesn't exists
|
||||
# or if inFile has been modified after the last modification of outFile
|
||||
def checkTouchedFile(inFile: str, outFile: str) -> bool:
|
||||
if not os.path.exists(outFile):
|
||||
return True
|
||||
return os.path.getmtime(inFile) > os.path.getmtime(outFile)
|
||||
|
||||
def Extract(xmlPath, outputPath, outputSourcePath):
|
||||
ExtractFile(xmlPath, outputPath, outputSourcePath, 1, 0)
|
||||
ExtractFile(xmlPath, outputPath, outputSourcePath, 1, 0)
|
||||
|
||||
def ExtractScene(xmlPath, outputPath, outputSourcePath):
|
||||
ExtractFile(xmlPath, outputPath, outputSourcePath, 1, 1)
|
||||
ExtractFile(xmlPath, outputPath, outputSourcePath, 1, 1)
|
||||
|
||||
def ExtractFile(xmlPath, outputPath, outputSourcePath, genSrcFile, incFilePrefix):
|
||||
execStr = "tools/ZAPD/ZAPD.out e -eh -i %s -b baserom/ -o %s -osf %s -gsf %i -ifp %i -rconf tools/ZAPDConfigs/MqDbg/Config.xml" % (xmlPath, outputPath, outputSourcePath, genSrcFile, incFilePrefix)
|
||||
if globalAbort.is_set():
|
||||
# Don't extract if another file wasn't extracted properly.
|
||||
return
|
||||
|
||||
print(execStr)
|
||||
os.system(execStr)
|
||||
execStr = "tools/ZAPD/ZAPD.out e -eh -i %s -b baserom/ -o %s -osf %s -gsf %i -ifp %i -rconf tools/ZAPDConfigs/MqDbg/Config.xml" % (xmlPath, outputPath, outputSourcePath, genSrcFile, incFilePrefix)
|
||||
if globalUnaccounted:
|
||||
execStr += " -wu"
|
||||
|
||||
print(execStr)
|
||||
exitValue = os.system(execStr)
|
||||
if exitValue != 0:
|
||||
globalAbort.set()
|
||||
print("\n")
|
||||
print("Error when extracting from file " + xmlPath, file=os.sys.stderr)
|
||||
print("Aborting...", file=os.sys.stderr)
|
||||
print("\n")
|
||||
|
||||
def ExtractFunc(fullPath):
|
||||
outPath = ("assets/" + fullPath.split("assets/xml/")[1]).split(".xml")[0]
|
||||
outSourcePath = ("assets/" + fullPath.split("assets/xml/")[1]).split(".xml")[0]
|
||||
*pathList, xmlName = fullPath.split(os.sep)
|
||||
objectName = os.path.splitext(xmlName)[0]
|
||||
|
||||
if (fullPath.startswith("assets/xml/scenes/")):
|
||||
ExtractScene(fullPath, outPath, outSourcePath)
|
||||
else:
|
||||
Extract(fullPath, outPath, outSourcePath)
|
||||
outPath = os.path.join("assets", *pathList[2:], objectName)
|
||||
outSourcePath = outPath
|
||||
|
||||
isScene = fullPath.startswith("assets/xml/scenes/")
|
||||
if isScene:
|
||||
objectName += "_scene"
|
||||
|
||||
if not globalForce:
|
||||
cFile = os.path.join(outPath, objectName + ".c")
|
||||
hFile = os.path.join(outPath, objectName + ".h")
|
||||
if not checkTouchedFile(fullPath, cFile) and not checkTouchedFile(fullPath, hFile):
|
||||
return
|
||||
|
||||
if isScene:
|
||||
ExtractScene(fullPath, outPath, outSourcePath)
|
||||
else:
|
||||
Extract(fullPath, outPath, outSourcePath)
|
||||
|
||||
def initializeWorker(force: bool, abort, unaccounted: bool):
|
||||
global globalForce
|
||||
global globalAbort
|
||||
global globalUnaccounted
|
||||
globalForce = force
|
||||
globalAbort = abort
|
||||
globalUnaccounted = unaccounted
|
||||
|
||||
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")
|
||||
parser.add_argument("-f", "--force", help="Force the extraction of every xml instead of checking the touched ones.", action="store_true")
|
||||
parser.add_argument("-u", "--unaccounted", help="Enables ZAPD unaccounted detector warning system.", action="store_true")
|
||||
args = parser.parse_args()
|
||||
|
||||
abort = Event()
|
||||
|
||||
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}/", f"assets/{asset_path}/")
|
||||
# Always force if -s is used.
|
||||
initializeWorker(True, abort, args.unaccounted)
|
||||
fullPath = os.path.join("assets", "xml", asset_path + ".xml")
|
||||
ExtractFunc(fullPath)
|
||||
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)
|
||||
for currentPath, folders, files in os.walk(os.path.join("assets", "xml")):
|
||||
for file in files:
|
||||
fullPath = os.path.join(currentPath, file)
|
||||
if file.endswith(".xml"):
|
||||
xmlFiles.append(fullPath)
|
||||
|
||||
numCores = cpu_count()
|
||||
print("Extracting assets with " + str(numCores) + " CPU cores.")
|
||||
p = Pool(numCores)
|
||||
p.map(ExtractFunc, xmlFiles)
|
||||
with Pool(numCores, initializer=initializeWorker, initargs=(args.force, abort, args.unaccounted)) as p:
|
||||
p.map(ExtractFunc, xmlFiles)
|
||||
|
||||
if abort.is_set():
|
||||
exit(1)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
|
|
|||
2
tools/.gitignore
vendored
2
tools/.gitignore
vendored
|
|
@ -5,3 +5,5 @@ makeromfs
|
|||
elf2rom
|
||||
mkldscript
|
||||
vtxdis
|
||||
|
||||
graphovl/
|
||||
|
|
|
|||
137
tools/actorfixer.py
Executable file
137
tools/actorfixer.py
Executable file
|
|
@ -0,0 +1,137 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
import os
|
||||
import argparse
|
||||
|
||||
animdict ={
|
||||
"Actor_SetHeight":"Actor_SetFocus",
|
||||
"func_8002E4B4":"Actor_UpdateBgCheckInfo",
|
||||
"func_8002BDB0":"Actor_SetFeetPos",
|
||||
"func_8002DA78":"Actor_WorldYawTowardActor",
|
||||
"func_8002DAC0":"Actor_WorldYawTowardPoint",
|
||||
"func_8002DB48":"Actor_WorldDistXYZToActor",
|
||||
"func_8002DB6C":"Actor_WorldDistXYZToPoint",
|
||||
"func_8002DAE0":"Actor_WorldPitchTowardActor",
|
||||
"func_8002DB28":"Actor_WorldPitchTowardPoint",
|
||||
"func_8002DB8C":"Actor_WorldDistXZToActor",
|
||||
"func_8002DBB0":"Actor_WorldDistXZToPoint",
|
||||
"func_8002EEE4":"Actor_GetFocus",
|
||||
"func_8002EF14":"Actor_GetWorld",
|
||||
"func_8002EF44":"Actor_GetWorldPosShapeRot",
|
||||
"actor.unk_1F":"actor.targetMode",
|
||||
"ICHAIN_U8(unk_1F":"ICHAIN_U8(targetMode",
|
||||
"actor.initPosRot":"actor.home",
|
||||
"actor.posRot.":"actor.world.",
|
||||
"actor.posRot2":"actor.focus",
|
||||
"actor.unk_4C":"actor.targetArrowOffset",
|
||||
"ICHAIN_F32(unk_4C":"ICHAIN_F32(targetArrowOffset",
|
||||
"actor.groundY":"actor.floorHeight",
|
||||
"actor.wallPolySource":"actor.wallBgId",
|
||||
"actor.floorPolySource":"actor.floorBgId",
|
||||
"actor.wallPolyRot":"actor.wallYaw",
|
||||
"DistToLink":"DistToPlayer",
|
||||
"yawTowardsLink":"yawTowardsPlayer",
|
||||
"colChkInfo.unk_10":"colChkInfo.cylRadius",
|
||||
"colChkInfo.unk_12":"colChkInfo.cylHeight",
|
||||
"colChkInfo.unk_14":"colChkInfo.cylYShift",
|
||||
"shape.unk_08":"shape.yOffset",
|
||||
"shape.shadowDrawFunc":"shape.shadowDraw",
|
||||
"shape.unk_10":"shape.shadowScale",
|
||||
"shape.unk_14":"shape.shadowAlpha",
|
||||
"actor.unk_CC":"actor.shape.feetPos",
|
||||
"actor.pos4":"actor.prevPos",
|
||||
"actor.unk_10C":"actor.isTargeted",
|
||||
"actor.unk_10D":"actor.targetPriority",
|
||||
"actor.dmgEffectTimer":"actor.colorFilterTimer",
|
||||
"actor.dmgEffectParams":"actor.colorFilterParams",
|
||||
"actor.unk_116":"actor.dropFlag",
|
||||
"actorCtx.actorList[":"actorCtx.actorLists[",
|
||||
"Actor_ChangeType":"Actor_ChangeCategory",
|
||||
"ActorShadow_DrawFunc_Squiggly":"ActorShadow_DrawHorse",
|
||||
"ActorShadow_DrawFunc_":"ActorShadow_Draw",
|
||||
"ACTORTYPE":"ACTORCAT",
|
||||
"actor.type":"actor.category",
|
||||
".body.":".info.",
|
||||
"HitItem":"HitInfo",
|
||||
"bumper.unk_06":"bumper.hitPos",
|
||||
"base.list":"base.elements",
|
||||
"toucher.flags":"toucher.dmgFlags",
|
||||
"bumper.flags":"bumper.dmgFlags",
|
||||
"maskA ":"ocFlags1 ",
|
||||
"maskB ":"ocFlags2 ",
|
||||
"base.type":"base.colType",
|
||||
"COLTYPE_UNK11":"COLTYPE_HARD",
|
||||
"COLTYPE_UNK12":"COLTYPE_WOOD",
|
||||
"COLTYPE_UNK13":"COLTYPE_TREE",
|
||||
"COLTYPE_METAL_SHIELD":"COLTYPE_METAL",
|
||||
"COLTYPE_UNK10":"COLTYPE_NONE",
|
||||
"COLTYPE_UNK":"COLTYPE_HIT",
|
||||
"info.flags":"info.elemtype",
|
||||
"ColliderBody":"ColliderInfo",
|
||||
"ColliderJntSphItem":"ColliderJntSphElement",
|
||||
"ColliderTrisItem":"ColliderTrisElement",
|
||||
"Collider_CylinderUpdate":"Collider_UpdateCylinder",
|
||||
"func_800628A4":"Collider_UpdateSpheres",
|
||||
"func_800627A0":"Collider_SetTrisVertices",
|
||||
"func_80062734":"Collider_SetQuadVertices",
|
||||
"func_80061ED4":"CollisionCheck_SetInfo",
|
||||
"func_80061EFC":"CollisionCheck_SetInfo2",
|
||||
"func_80061EB0":"CollisionCheck_SetInfoNoDamageTable",
|
||||
# "func_8002E084": "Actor_YawInRangeWithPlayer",
|
||||
# "func_8002E0D0": "Actor_YawInRangeWithActor",
|
||||
# "func_8002E12C": "Actor_YawAndDistInRangeWithPlayer",
|
||||
# "func_8002E1A8": "Actor_YawAndDistInRangeWithActor",
|
||||
# "func_80033A84": "Actor_IsTargeted",
|
||||
# "func_80033AB8": "Actor_OtherIsTargeted",
|
||||
# "func_80035650": "Actor_SetDropFlag",
|
||||
# "func_8003573C": "Actor_SetDropFlagJntSph",
|
||||
# "func_80033780": "Actor_GetProjectileActor",
|
||||
# "func_80033260": "Actor_SpawnFloorDust",
|
||||
# "func_80032C7C": "Actor_PlayDeathFx",
|
||||
# "actorCtx.unk_00": "actorCtx.freezeFlashTimer",
|
||||
}
|
||||
|
||||
def replace_anim(file):
|
||||
with open(file,'r',encoding = 'utf-8') as infile:
|
||||
srcdata = infile.read()
|
||||
|
||||
funcs = list(animdict.keys())
|
||||
fixes = 0
|
||||
for func in funcs:
|
||||
newfunc = animdict.get(func)
|
||||
if(newfunc is None):
|
||||
print("How did this happen?")
|
||||
return -1
|
||||
if(func in srcdata):
|
||||
fixes += 1
|
||||
print(func)
|
||||
srcdata = srcdata.replace(func, newfunc)
|
||||
|
||||
if(fixes > 0):
|
||||
print('Changed', fixes,'entr' + ('y' if fixes == 1 else 'ies') + ' in',file)
|
||||
with open(file, 'w', encoding = 'utf-8', newline = '\n') as outfile:
|
||||
outfile.write(srcdata)
|
||||
return 1
|
||||
|
||||
def replace_anim_all(repo):
|
||||
for subdir, dirs, files in os.walk(repo + os.sep + 'src'):
|
||||
for filename in files:
|
||||
if(filename.endswith('.c')):
|
||||
file = subdir + os.sep + filename
|
||||
replace_anim(file)
|
||||
for subdir, dirs, files in os.walk(repo + os.sep + 'asm' + os.sep + 'non_matchings'):
|
||||
for filename in files:
|
||||
if(filename.endswith('.s')):
|
||||
file = subdir + os.sep + filename
|
||||
replace_anim(file)
|
||||
return 1
|
||||
|
||||
parser = argparse.ArgumentParser(description='Update to the new animation names')
|
||||
parser.add_argument('file', help="source file to be processed. use . to process the whole repo", default = None)
|
||||
|
||||
if __name__ == "__main__":
|
||||
args = parser.parse_args()
|
||||
if(args.file == '.'):
|
||||
replace_anim_all(os.curdir)
|
||||
else:
|
||||
replace_anim(args.file)
|
||||
Loading…
Reference in a new issue