1 files changed, 91 insertions, 0 deletions
diff --git a/src/factories/oot/OoTSkeletonFactory.cpp b/src/factories/oot/OoTSkeletonFactory.cpp
new file mode 100644
index 0000000..70bf132
--- /dev/null
+++ b/src/factories/oot/OoTSkeletonFactory.cpp
@@ -0,0 +1,91 @@
+#include "OoTSkeletonFactory.h"
+#include "OoTSkeletonTypes.h"
+#include "OoTSceneUtils.h"
+#include "spdlog/spdlog.h"
+#include "Companion.h"
+#include "utils/Decompressor.h"
+
+namespace OoT {
+
+std::optional<std::shared_ptr<IParsedData>> OoTSkeletonFactory::parse(std::vector<uint8_t>& buffer, YAML::Node& node) {
+    auto skelTypeStr = GetSafeNode<std::string>(node, "skel_type");
+    auto limbTypeStr = GetSafeNode<std::string>(node, "limb_type");
+    auto skelType = ParseSkeletonType(skelTypeStr);
+    auto limbType = ParseLimbType(limbTypeStr);
+
+    size_t skelSize = (skelType == OoTSkeletonType::Flex) ? 0x0C : 0x08;
+    auto [_, segment] = Decompressor::AutoDecode(node, buffer, skelSize);
+    LUS::BinaryReader reader(segment.data, segment.size);
+    reader.SetEndianness(Torch::Endianness::Big);
+
+    uint32_t limbsArrayAddr = reader.ReadUInt32();
+    limbsArrayAddr = Companion::Instance->PatchVirtualAddr(limbsArrayAddr);
+    uint8_t limbCount = reader.ReadUByte();
+    uint8_t dListCount = 0;
+
+    if (skelType == OoTSkeletonType::Flex) {
+        reader.Seek(8, LUS::SeekOffsetType::Start);
+        dListCount = reader.ReadUByte();
+    }
+
+    auto symbol = GetSafeNode<std::string>(node, "symbol");
+    std::vector<std::string> limbPaths;
+    if (limbsArrayAddr != 0 && limbCount > 0) {
+        YAML::Node limbTableNode;
+        limbTableNode["offset"] = limbsArrayAddr;
+        auto limbTableRaw = Decompressor::AutoDecode(limbTableNode, buffer, limbCount * 4);
+        LUS::BinaryReader limbTableReader(limbTableRaw.segment.data, limbTableRaw.segment.size);
+        limbTableReader.SetEndianness(Torch::Endianness::Big);
+
+        for (uint8_t i = 0; i < limbCount; i++) {
+            uint32_t limbAddr = limbTableReader.ReadUInt32();
+            limbAddr = Companion::Instance->PatchVirtualAddr(limbAddr);
+            std::string limbPath = ResolvePointer(limbAddr);
+            if (limbPath.empty() && limbAddr != 0) {
+                SPDLOG_WARN("Undeclared limb at 0x{:08X} — YAML enrichment incomplete", limbAddr);
+            }
+            limbPaths.push_back(limbPath);
+        }
+    }
+
+    if (limbsArrayAddr != 0) {
+        auto limbTablePath = ResolvePointer(limbsArrayAddr);
+        if (limbTablePath.empty()) {
+            SPDLOG_WARN("Undeclared limb table at 0x{:08X} — YAML enrichment incomplete", limbsArrayAddr);
+        }
+    }
+
+    auto skel = std::make_shared<OoTSkeletonData>();
+    skel->skelType = skelType;
+    skel->limbType = limbType;
+    skel->limbCount = limbCount;
+    skel->dListCount = dListCount;
+    skel->limbPaths = std::move(limbPaths);
+
+    return skel;
+}
+
+ExportResult OoTSkeletonBinaryExporter::Export(std::ostream& write, std::shared_ptr<IParsedData> raw,
+                                                std::string& entryName, YAML::Node& node,
+                                                std::string* replacement) {
+    auto writer = LUS::BinaryWriter();
+    auto skel = std::static_pointer_cast<OoTSkeletonData>(raw);
+
+    WriteHeader(writer, Torch::ResourceType::OoTSkeleton, 0);
+
+    writer.Write(static_cast<uint8_t>(skel->skelType));
+    writer.Write(static_cast<uint8_t>(skel->limbType));
+    writer.Write(static_cast<uint32_t>(skel->limbCount));
+    writer.Write(static_cast<uint32_t>(skel->dListCount));
+    writer.Write(static_cast<uint8_t>(skel->limbType));
+    writer.Write(static_cast<uint32_t>(skel->limbPaths.size()));
+
+    for (auto& path : skel->limbPaths) {
+        writer.Write(path);
+    }
+
+    writer.Finish(write);
+    return std::nullopt;
+}
+
+} // namespace OoT