[OoT] OoT Object changes & small changes/fixes/removals

__init__.py

@@ -396,6 +396,7 @@ def upgrade_changed_props():
     """Set scene properties after a scene loads, used for migrating old properties"""
     SM64_Properties.upgrade_changed_props()
     SM64_ObjectProperties.upgrade_changed_props()
+    OOT_ObjectProperties.upgrade_changed_props()
 
 
 def upgrade_scene_props_node():

fast64_internal/oot/data/oot_actor_data.py

@@ -0,0 +1,48 @@
+from os import path
+from dataclasses import dataclass
+from .oot_getters import getXMLRoot
+from .oot_data import OoT_BaseElement
+
+
+@dataclass
+class OoT_ActorElement(OoT_BaseElement):
+    category: str
+    tiedObjects: list[str]
+
+
+class OoT_ActorData:
+    """Everything related to OoT Actors"""
+
+    def __init__(self):
+        # Path to the ``ActorList.xml`` file
+        actorXML = path.dirname(path.abspath(__file__)) + "/xml/ActorList.xml"
+        actorRoot = getXMLRoot(actorXML)
+
+        # general actor list
+        self.actorList: list[OoT_ActorElement] = []
+
+        for actor in actorRoot.iterfind("Actor"):
+            tiedObjects = []
+            objKey = actor.get("ObjectKey")
+            actorName = f"{actor.attrib['Name']} - {actor.attrib['ID'].removeprefix('ACTOR_')}"
+            if objKey is not None:  # actors don't always use an object
+                tiedObjects = objKey.split(",")
+            self.actorList.append(
+                OoT_ActorElement(
+                    actor.attrib["ID"],
+                    actor.attrib["Key"],
+                    actorName,
+                    int(actor.attrib["Index"]),
+                    actor.attrib["Category"],
+                    tiedObjects,
+                )
+            )
+        self.actorsByKey = {actor.key: actor for actor in self.actorList}
+        self.actorsByID = {actor.id: actor for actor in self.actorList}
+
+        # list of tuples used by Blender's enum properties
+        lastIndex = max(1, *(actor.index for actor in self.actorList))
+        self.ootEnumActorID = [("None", f"{i} (Deleted from the XML)", "None") for i in range(lastIndex)]
+        self.ootEnumActorID.insert(0, ("Custom", "Custom Actor", "Custom"))
+        for actor in self.actorList:
+            self.ootEnumActorID[actor.index] = (actor.id, actor.name, actor.id)

fast64_internal/oot/data/oot_data.py

@@ -0,0 +1,21 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class OoT_BaseElement:
+    id: str
+    key: str
+    name: str
+    index: int
+
+
+@dataclass
+class OoT_Data:
+    """Contains data related to OoT, like actors or objects"""
+
+    def __init__(self):
+        from .oot_object_data import OoT_ObjectData
+        from .oot_actor_data import OoT_ActorData
+
+        self.objectData = OoT_ObjectData()
+        self.actorData = OoT_ActorData()

fast64_internal/oot/data/oot_getters.py

@@ -0,0 +1,11 @@
+from xml.etree.ElementTree import parse as parseXML, Element
+
+
+def getXMLRoot(xmlPath: str) -> Element:
+    """Parse an XML file and return its root element"""
+    try:
+        return parseXML(xmlPath).getroot()
+    except:
+        from ...utility import PluginError
+
+        raise PluginError(f"ERROR: File '{xmlPath}' is missing or malformed.")

fast64_internal/oot/data/oot_object_data.py

@@ -0,0 +1,55 @@
+from dataclasses import dataclass
+from os import path
+from ...utility import PluginError
+from .oot_getters import getXMLRoot
+from .oot_data import OoT_BaseElement
+
+# Note: "object" in this context refers to an OoT Object file (like ``gameplay_keep``)
+
+
+@dataclass
+class OoT_ObjectElement(OoT_BaseElement):
+    pass
+
+
+class OoT_ObjectData:
+    """Everything related to OoT objects"""
+
+    def __init__(self):
+        # general object list
+        self.objectList: list[OoT_ObjectElement] = []
+
+        # Path to the ``ObjectList.xml`` file
+        objectXML = path.dirname(path.abspath(__file__)) + "/xml/ObjectList.xml"
+        objectRoot = getXMLRoot(objectXML)
+
+        for obj in objectRoot.iterfind("Object"):
+            objName = f"{obj.attrib['Name']} - {obj.attrib['ID'].removeprefix('OBJECT_')}"
+            self.objectList.append(
+                OoT_ObjectElement(obj.attrib["ID"], obj.attrib["Key"], objName, int(obj.attrib["Index"]))
+            )
+
+        self.objectsByID = {obj.id: obj for obj in self.objectList}
+        self.objectsByKey = {obj.key: obj for obj in self.objectList}
+
+        # list of tuples used by Blender's enum properties
+        self.deletedEntry = ("None", "(Deleted from the XML)", "None")
+        lastIndex = max(1, *(obj.index for obj in self.objectList))
+        self.ootEnumObjectKey = self.getObjectIDList(lastIndex + 1, False)
+
+        # create the legacy object list for old blends
+        self.ootEnumObjectIDLegacy = self.getObjectIDList(self.objectsByKey["obj_timeblock"].index + 1, True)
+
+        # validate the legacy list, if there's any None element then something's wrong
+        if self.deletedEntry in self.ootEnumObjectIDLegacy:
+            raise PluginError("ERROR: Legacy Object List doesn't match!")
+
+    def getObjectIDList(self, max: int, isLegacy: bool):
+        """Generates and returns the object list in the right order"""
+        objList = [self.deletedEntry] * max
+        for obj in self.objectList:
+            if obj.index < max:
+                identifier = obj.id if isLegacy else obj.key
+                objList[obj.index] = (identifier, obj.name, obj.id)
+        objList[0] = ("Custom", "Custom Object", "Custom")
+        return objList