diff --git a/src/factories/sm64/AnimationFactory.cpp b/src/factories/sm64/AnimationFactory.cpp index dec28a2..3d9edf6 100644 --- a/src/factories/sm64/AnimationFactory.cpp +++ b/src/factories/sm64/AnimationFactory.cpp @@ -3,7 +3,7 @@ #include "utils/Decompressor.h" -#define ANIMINDEX_COUNT(boneCount) (boneCount + 1) * 6 * sizeof(uint16_t) +#define ANIMINDEX_COUNT(boneCount) (((boneCount) + 1) * 6) ExportResult SM64::AnimationBinaryExporter::Export(std::ostream &write, std::shared_ptr raw, std::string& entryName, YAML::Node &node, std::string* replacement ) { auto writer = LUS::BinaryWriter(); @@ -35,7 +35,7 @@ ExportResult SM64::AnimationBinaryExporter::Export(std::ostream &write, std::sha std::optional> SM64::AnimationFactory::parse(std::vector& buffer, YAML::Node& node) { auto offset = node["offset"]; - auto [_, data] = Decompressor::AutoDecode(node, buffer); + auto [raw, data] = Decompressor::AutoDecode(node, buffer); LUS::BinaryReader header = LUS::BinaryReader(data.data, data.size); header.SetEndianness(Torch::Endianness::Big); @@ -48,24 +48,22 @@ std::optional> SM64::AnimationFactory::parse(std::v auto valuesAddr = header.ReadUInt32(); auto indexAddr = header.ReadUInt32(); auto length = header.ReadUInt32(); + auto segmented = raw != nullptr; + auto mainData = segmented ? raw->data : data.data; - size_t indexLength = ANIMINDEX_COUNT(unusedBoneCount); + const auto indexLength = ANIMINDEX_COUNT(unusedBoneCount); + const auto valuesSize = !segmented ? length * sizeof(int16_t) : indexAddr - valuesAddr; - LUS::BinaryReader indices = LUS::BinaryReader(data.data + indexAddr, indexLength * sizeof(uint16_t)); - indices.SetEndianness(Torch::Endianness::Big); - std::vector indicesData; - for (size_t i = 0; i < indexLength; i++) { - indicesData.push_back(indices.ReadUInt16()); - } - - size_t valuesSize = length * sizeof(int16_t); - LUS::BinaryReader values = LUS::BinaryReader(data.data + valuesAddr, valuesSize); - values.SetEndianness(Torch::Endianness::Big); - std::vector valuesData; - for (size_t i = 0; i < valuesSize / sizeof(int16_t); i++) { - valuesData.push_back(values.ReadInt16()); + if(segmented) { + valuesAddr = SEGMENT_OFFSET(valuesAddr); + indexAddr = SEGMENT_OFFSET(indexAddr); } + SPDLOG_INFO("Offset: 0x{:X}", offset.as()); + SPDLOG_INFO("Index Addr: 0x{:X}", indexAddr); + SPDLOG_INFO("Values Addr: 0x{:X}", valuesAddr); + SPDLOG_INFO("Values Size: {}", valuesSize / sizeof(int16_t)); + SPDLOG_INFO("Segmented: {}", segmented); SPDLOG_INFO("Flags: {}", flags); SPDLOG_INFO("Anim-Y Trans Divisor: {}", animYTransDivisor); SPDLOG_INFO("Start Frame: {}", startFrame); @@ -74,5 +72,21 @@ std::optional> SM64::AnimationFactory::parse(std::v SPDLOG_INFO("Unused Bone Count: {}", unusedBoneCount); SPDLOG_INFO("Length: {}", length); + LUS::BinaryReader indices = LUS::BinaryReader(mainData + indexAddr, indexLength * sizeof(uint16_t)); + indices.SetEndianness(Torch::Endianness::Big); + std::vector indicesData; + for (size_t i = 0; i < indexLength; i++) { + indicesData.push_back(indices.ReadUInt16()); + SPDLOG_INFO("Index: {}", indicesData.back()); + } + + LUS::BinaryReader values = LUS::BinaryReader(mainData + valuesAddr, valuesSize); + values.SetEndianness(Torch::Endianness::Big); + std::vector valuesData; + for (size_t i = 0; i < valuesSize / sizeof(int16_t); i++) { + valuesData.push_back(values.ReadInt16()); + SPDLOG_INFO("Value: {}", valuesData.back()); + } + return std::make_shared(flags, animYTransDivisor, startFrame, loopStart, loopEnd, unusedBoneCount, length, indicesData, valuesData); } \ No newline at end of file