From de4241861633949f2e80566cefd789defbc162c5 Mon Sep 17 00:00:00 2001 From: KiritoDv Date: Sat, 17 Oct 2026 01:09:18 -0600 Subject: Added gamecube support --- src/factories/ac/StructArrayFactory.cpp | 107 ++++++++++++++++++++++++++++++++ 1 file changed, 107 insertions(+) create mode 100644 src/factories/ac/StructArrayFactory.cpp (limited to 'src/factories/ac/StructArrayFactory.cpp') diff --git a/src/factories/ac/StructArrayFactory.cpp b/src/factories/ac/StructArrayFactory.cpp new file mode 100644 index 0000000..e71f364 --- /dev/null +++ b/src/factories/ac/StructArrayFactory.cpp @@ -0,0 +1,107 @@ +#include "StructArrayFactory.h" +#include "ACUtils.h" +#include "utils/TorchUtils.h" + +#include + +namespace AC { + +static StructArrayField ParseField(const std::string& spec) { + static const std::map> types = { + { "u8", { 1, StructArrayType::U8 } }, { "s8", { 1, StructArrayType::S8 } }, + { "pad", { 1, StructArrayType::U8 } }, { "u16", { 2, StructArrayType::U16 } }, + { "s16", { 2, StructArrayType::S16 } }, { "u32", { 4, StructArrayType::U32 } }, + { "s32", { 4, StructArrayType::S32 } }, { "f32", { 4, StructArrayType::F32 } }, + }; + const auto star = spec.find('*'); + const auto base = spec.substr(0, star); + const uint32_t repeat = star == std::string::npos ? 1 : (uint32_t)std::stoul(spec.substr(star + 1), nullptr, 0); + const auto it = types.find(base); + if (it == types.end() || repeat == 0) { + throw std::runtime_error("AC:STRUCT_ARRAY: bad field '" + spec + "'"); + } + return { it->second.first, repeat, it->second.second }; +} + +ExportResult StructArrayBinaryExporter::Export(std::ostream& write, std::shared_ptr raw, + std::string& entryName, YAML::Node& node, std::string* replacement) { + auto data = std::static_pointer_cast(raw); + auto writer = LUS::BinaryWriter(); + WriteHeader(writer, Torch::ResourceType::GenericArray, 0); + writer.Write((uint32_t)data->mType); + writer.Write(data->mCount); + const uint8_t* p = data->mBytes.data(); + for (uint32_t i = 0; i < data->mCount; i++) { + switch (data->mType) { + case StructArrayType::U8: + case StructArrayType::S8: + writer.Write(p[i]); + break; + case StructArrayType::U16: + case StructArrayType::S16: + writer.Write(Be16(p + i * 2)); + break; + case StructArrayType::U32: + case StructArrayType::S32: + case StructArrayType::F32: + writer.Write(Be32(p + i * 4)); + break; + default: + throw std::runtime_error("AC:STRUCT_ARRAY: unsupported element type"); + } + } + writer.Finish(write); + return std::nullopt; +} + +std::optional> StructArrayFactory::parse(std::vector& buffer, YAML::Node& node) { + const auto name = NameOf(node); + const auto offset = GetSafeNode(node, "offset"); + std::vector specs = mFixedLayout; + if (specs.empty()) { + if (!node["fields"] || !node["fields"].IsSequence()) { + throw std::runtime_error(name + ": AC:STRUCT_ARRAY needs `fields`"); + } + for (const auto& f : node["fields"]) { + specs.push_back(f.as()); + } + } + std::vector fields; + uint32_t stride = 0; + for (const auto& spec : specs) { + fields.push_back(ParseField(spec)); + stride += fields.back().size * fields.back().repeat; + } + + uint32_t count; + if (node["count"]) { + count = GetSafeNode(node, "count"); + } else { + const auto size = GetSafeNode(node, "size"); + if (size % stride != 0) { + throw std::runtime_error(name + ": size 0x" + Torch::to_hex(size, false) + + " is not a multiple of the record size 0x" + Torch::to_hex(stride, false)); + } + count = size / stride; + } + auto raw = ReadBytes(offset, count * stride, name); + + if (fields.size() == 1) { + return std::make_shared(fields[0].type, count * fields[0].repeat, std::move(raw)); + } + + std::vector bytes(raw.size()); + size_t pos = 0; + for (uint32_t r = 0; r < count; r++) { + for (const auto& f : fields) { + for (uint32_t k = 0; k < f.repeat; k++, pos += f.size) { + for (uint32_t b = 0; b < f.size; b++) { + bytes[pos + b] = raw[pos + f.size - 1 - b]; + } + } + } + } + return std::make_shared(StructArrayType::U8, count * stride, std::move(bytes)); +} + +} // namespace AC -- cgit v1.2.3