From bdb3f94391b658192fb8afe4be8b7802c0ed6afb Mon Sep 17 00:00:00 2001 From: inspectredc Date: Thu, 25 Apr 2024 01:49:59 +0100 Subject: [PATCH] Asset Array Factory --- src/Companion.cpp | 2 + src/factories/AssetArrayFactory.cpp | 98 +++++++++++++++++++++++++++++ src/factories/AssetArrayFactory.h | 35 +++++++++++ src/factories/ResourceType.h | 1 + 4 files changed, 136 insertions(+) create mode 100644 src/factories/AssetArrayFactory.cpp create mode 100644 src/factories/AssetArrayFactory.h diff --git a/src/Companion.cpp b/src/Companion.cpp index e413ff7..bc218b5 100644 --- a/src/Companion.cpp +++ b/src/Companion.cpp @@ -27,6 +27,7 @@ #include "factories/LightsFactory.h" #include "factories/Vec3fFactory.h" #include "factories/Vec3sFactory.h" +#include "factories/AssetArrayFactory.h" #include "factories/sm64/AnimationFactory.h" #include "factories/sm64/CollisionFactory.h" @@ -89,6 +90,7 @@ void Companion::Init(const ExportType type) { this->RegisterFactory("VEC3F", std::make_shared()); this->RegisterFactory("VEC3S", std::make_shared()); this->RegisterFactory("ARRAY", std::make_shared()); + this->RegisterFactory("ASSET_ARRAY", std::make_shared()); // SM64 specific this->RegisterFactory("SM64:DIALOG", std::make_shared()); diff --git a/src/factories/AssetArrayFactory.cpp b/src/factories/AssetArrayFactory.cpp new file mode 100644 index 0000000..0e91fec --- /dev/null +++ b/src/factories/AssetArrayFactory.cpp @@ -0,0 +1,98 @@ +#include "AssetArrayFactory.h" +#include "spdlog/spdlog.h" + +#include "Companion.h" +#include "utils/Decompressor.h" +#include "utils/TorchUtils.h" +#include + +#define FORMAT_HEX(ptr) (ptr) + +ExportResult AssetArrayHeaderExporter::Export(std::ostream &write, std::shared_ptr raw, std::string& entryName, YAML::Node &node, std::string* replacement) { + const auto symbol = GetSafeNode(node, "symbol", entryName); + auto data = std::static_pointer_cast(raw); + + if(Companion::Instance->IsOTRMode()){ + write << "static const char " << symbol << "[] = \"__OTR__" << (*replacement) << "\";\n\n"; + return std::nullopt; + } + + write << "extern " << data->mType << "* " << symbol << "[];\n"; + return std::nullopt; +} + +ExportResult AssetArrayCodeExporter::Export(std::ostream &write, std::shared_ptr raw, std::string& entryName, YAML::Node &node, std::string* replacement ) { + const auto symbol = GetSafeNode(node, "symbol", entryName); + const auto offset = GetSafeNode(node, "offset"); + + auto data = std::static_pointer_cast(raw); + + write << "static " << data->mType << "* " << symbol << "[] = {\n"; + for (auto ptr : data->mPtrs) { + write << fourSpaceTab; + auto dec = Companion::Instance->GetNodeByAddr(ptr); + if (dec.has_value()) { + auto node = std::get<1>(dec.value()); + auto assetSymbol = GetSafeNode(node, "symbol"); + write << "&" << assetSymbol << ",\n"; + } else { + write << FORMAT_HEX(ptr) << ",\n"; + } + } + write << "};\n"; + + size_t size = (data->mPtrs.size()) * sizeof(uint32_t); + + return offset + size; +} + +ExportResult AssetArrayBinaryExporter::Export(std::ostream &write, std::shared_ptr raw, std::string& entryName, YAML::Node &node, std::string* replacement ) { + auto writer = LUS::BinaryWriter(); + auto data = std::static_pointer_cast(raw); + + WriteHeader(writer, LUS::ResourceType::AssetArray, 0); + + writer.Write((uint32_t)data->mPtrs.size()); + + for (auto ptr : data->mPtrs) { + if (ptr == 0) { + writer.Write((uint64_t)0); + continue; + } + + auto dec = Companion::Instance->GetNodeByAddr(ptr); + if (dec.has_value()) { + uint64_t hash = CRC64(std::get<0>(dec.value()).c_str()); + SPDLOG_INFO("Found Asset: 0x{:X} Hash: 0x{:X} Path: {}", ptr, hash, std::get<0>(dec.value())); + writer.Write(hash); + } else { + SPDLOG_WARN("Could not find Asset at 0x{:X}", ptr); + } + } + + writer.Finish(write); + return std::nullopt; +} + +std::optional> AssetArrayFactory::parse(std::vector& buffer, YAML::Node& node) { + std::vector ptrs; + auto assetType = GetSafeNode(node, "assetType"); + auto factoryType = GetSafeNode(node, "factoryType"); + const auto count = GetSafeNode(node, "count"); + auto [_, segment] = Decompressor::AutoDecode(node, buffer); + LUS::BinaryReader reader(segment.data, segment.size); + reader.SetEndianness(LUS::Endianness::Big); + + for (uint32_t i = 0; i < count; ++i) { + auto ptr = reader.ReadUInt32(); + + YAML::Node assetNode; + assetNode["type"] = factoryType; + assetNode["offset"] = ptr; + Companion::Instance->AddAsset(assetNode); + + ptrs.emplace_back(ptr); + } + + return std::make_shared(ptrs, assetType); +} diff --git a/src/factories/AssetArrayFactory.h b/src/factories/AssetArrayFactory.h new file mode 100644 index 0000000..dabd96d --- /dev/null +++ b/src/factories/AssetArrayFactory.h @@ -0,0 +1,35 @@ +#pragma once + +#include "factories/BaseFactory.h" + +class AssetArrayData : public IParsedData { +public: + std::vector mPtrs; + std::string mType; + + AssetArrayData(std::vector ptrs, std::string type) : mPtrs(std::move(ptrs)), mType(type) {} +}; + +class AssetArrayHeaderExporter : public BaseExporter { + ExportResult Export(std::ostream& write, std::shared_ptr data, std::string& entryName, YAML::Node& node, std::string* replacement) override; +}; + +class AssetArrayBinaryExporter : public BaseExporter { + ExportResult Export(std::ostream& write, std::shared_ptr data, std::string& entryName, YAML::Node& node, std::string* replacement) override; +}; + +class AssetArrayCodeExporter : public BaseExporter { + ExportResult Export(std::ostream& write, std::shared_ptr data, std::string& entryName, YAML::Node& node, std::string* replacement) override; +}; + +class AssetArrayFactory : public BaseFactory { +public: + std::optional> parse(std::vector& buffer, YAML::Node& data) override; + inline std::unordered_map> GetExporters() override { + return { + REGISTER(Code, AssetArrayCodeExporter) + REGISTER(Header, AssetArrayHeaderExporter) + REGISTER(Binary, AssetArrayBinaryExporter) + }; + } +}; diff --git a/src/factories/ResourceType.h b/src/factories/ResourceType.h index 218bd92..98c8cf5 100644 --- a/src/factories/ResourceType.h +++ b/src/factories/ResourceType.h @@ -22,6 +22,7 @@ enum class ResourceType { Vec3f = 0x56433346, // VC3F Vec3s = 0x56433353, // VC3S GenericArray = 0x47415252, // GARR + AssetArray = 0x41415252, // AARR // SM64 Anim = 0x414E494D, // ANIM