summaryrefslogtreecommitdiff
path: root/src/factories/sm64/AnimationFactory.cpp
blob: c35ffca6aadd64756d024c8be92940da4944b3a6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#include "AnimationFactory.h"
#include "spdlog/spdlog.h"

#include "utils/Decompressor.h"

#define ANIMINDEX_COUNT(boneCount) (((boneCount) + 1) * 6)

ExportResult SM64::AnimationBinaryExporter::Export(std::ostream& write, std::shared_ptr<IParsedData> raw,
                                                   std::string& entryName, YAML::Node& node, std::string* replacement) {
    auto writer = LUS::BinaryWriter();
    const auto anim = std::static_pointer_cast<AnimationData>(raw);

    WriteHeader(writer, Torch::ResourceType::Anim, 0);
    writer.Write(anim->mFlags);
    writer.Write(anim->mAnimYTransDivisor);
    writer.Write(anim->mStartFrame);
    writer.Write(anim->mLoopStart);
    writer.Write(anim->mLoopEnd);
    writer.Write(anim->mUnusedBoneCount);
    writer.Write(static_cast<uint64_t>(anim->mLength));

    writer.Write(static_cast<uint32_t>(anim->mIndices.size()));
    for (const auto& index : anim->mIndices) {
        writer.Write(index);
    }

    writer.Write(static_cast<uint32_t>(anim->mEntries.size()));
    for (const auto& entry : anim->mEntries) {
        writer.Write(entry);
    }

    writer.Finish(write);
    return std::nullopt;
}

std::optional<std::shared_ptr<IParsedData>> SM64::AnimationFactory::parse(std::vector<uint8_t>& buffer,
                                                                          YAML::Node& node) {
    auto offset = node["offset"];

    auto [raw, data] = Decompressor::AutoDecode(node, buffer);
    LUS::BinaryReader header = LUS::BinaryReader(data.data, data.size);
    header.SetEndianness(Torch::Endianness::Big);

    auto flags = header.ReadInt16();
    auto animYTransDivisor = header.ReadInt16();
    auto startFrame = header.ReadInt16();
    auto loopStart = header.ReadInt16();
    auto loopEnd = header.ReadInt16();
    auto unusedBoneCount = header.ReadInt16();
    auto valuesAddr = header.ReadUInt32();
    auto indexAddr = header.ReadUInt32();
    auto length = header.ReadUInt32();
    auto segmented = raw != nullptr;
    auto mainData = segmented ? raw->data : data.data;

    const auto indexLength = ANIMINDEX_COUNT(unusedBoneCount);
    const auto valuesSize = !segmented ? length * sizeof(int16_t) : indexAddr - valuesAddr;

    if (segmented) {
        valuesAddr = SEGMENT_OFFSET(valuesAddr);
        indexAddr = SEGMENT_OFFSET(indexAddr);
    }

    SPDLOG_INFO("Offset: 0x{:X}", offset.as<uint32_t>());
    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);
    SPDLOG_INFO("Loop Start: {}", loopStart);
    SPDLOG_INFO("Loop End: {}", loopEnd);
    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<uint16_t> indicesData;
    for (size_t i = 0; i < indexLength; i++) {
        indicesData.push_back(indices.ReadUInt16());
    }

    LUS::BinaryReader values = LUS::BinaryReader(mainData + valuesAddr, valuesSize);
    values.SetEndianness(Torch::Endianness::Big);
    std::vector<int16_t> valuesData;
    for (size_t i = 0; i < valuesSize / sizeof(int16_t); i++) {
        valuesData.push_back(values.ReadInt16());
    }

    return std::make_shared<AnimationData>(flags, animYTransDivisor, startFrame, loopStart, loopEnd, unusedBoneCount,
                                           length, indicesData, valuesData);
}
#ifdef BUILD_UI
#include "Companion.h"
#include "ui/Widgets.h"

void SM64::AnimationFactoryUI::DrawUI(const ParseResultData& item) {
    UI::AssetHeader(item.name, item.type);
    if (!item.data.has_value()) {
        ImGui::TextDisabled("no data");
        return;
    }
    auto anim = std::static_pointer_cast<SM64::AnimationData>(item.data.value());
    UI::KV("flags", std::to_string(anim->mFlags));
    UI::KV("start frame", std::to_string(anim->mStartFrame));
    UI::KV("loop", std::to_string(anim->mLoopStart) + " - " + std::to_string(anim->mLoopEnd));
    UI::KV("y-trans divisor", std::to_string(anim->mAnimYTransDivisor));
    UI::KV("channels", std::to_string(anim->mIndices.size() / 2));
    UI::KV("values", std::to_string(anim->mEntries.size()));
}
#endif // BUILD_UI