summaryrefslogtreecommitdiff
path: root/src/factories/sf64/HitboxFactory.cpp
blob: 69d5ef171a5fea45f78108745919dbd5527d5d27 (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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#include "HitboxFactory.h"
#include "spdlog/spdlog.h"

#include "Companion.h"
#include "utils/Decompressor.h"
#include "utils/TorchUtils.h"

#define FLOAT(x, w) std::dec << std::setfill(' ') << (((int) x == x) ? "" : "  ") << std::setw(w - 2) << x << (((int) x == x) ? ".0f" : "f")

SF64::HitboxData::HitboxData(std::vector<float> data, std::vector<int> types): mData(data), mTypes(types) {
    
}

void SF64::HitboxHeaderExporter::Export(std::ostream &write, std::shared_ptr<IParsedData> raw, std::string& entryName, YAML::Node &node, std::string* replacement) {
    const auto symbol = GetSafeNode(node, "symbol", entryName);

    if(Companion::Instance->IsOTRMode()){
        write << "static const char " << symbol << "[] = \"__OTR__" << (*replacement) << "\";\n\n";
        return;
    }

    write << "extern f32 " << symbol << "[];\n";
}

void SF64::HitboxCodeExporter::Export(std::ostream &write, std::shared_ptr<IParsedData> raw, std::string& entryName, YAML::Node &node, std::string* replacement ) {
    const auto symbol = GetSafeNode(node, "symbol", entryName);
    const auto offset = GetSafeNode<uint32_t>(node, "offset");
    auto hitbox = std::static_pointer_cast<SF64::HitboxData>(raw);
    auto index = 0;
    auto count = hitbox->mData[index++];
    auto hasType2 = false;
    auto off = offset;
    
    if(IS_SEGMENTED(off)) {
        off = SEGMENT_OFFSET(off);
    } 
    for(int type : hitbox->mTypes) {
        if(type == 2) {
            hasType2 = true;
            break;
        }
    }
    if (Companion::Instance->IsDebug()) {
        write << "// 0x" << std::uppercase << std::hex << off << "\n";
    }
    write << "f32 " << symbol << "[] = {\n";
    write << fourSpaceTab << count << ",\n";
    
    for(int i = 0; i < (int)count; i++) {
        if(hitbox->mTypes[i]  == 4) {
            write << fourSpaceTab << "HITBOX_TYPE_4,   ";
            index++;
        } else if (hitbox->mTypes[i] == 3) {
            write << fourSpaceTab << "HITBOX_TYPE_3,   ";
            index++;
        } else if (hitbox->mTypes[i] == 2) {
            write << fourSpaceTab << "HITBOX_TYPE_2,    ";
            index++;
            for(int j = 0; j < 3; j++) {
                auto tempf = hitbox->mData[index++];
                write << FLOAT(tempf, 6) << ", ";
            }
            write << fourSpaceTab;
        } else {
            write << " /* HITBOX_TYPE_1 */ ";
        }
        if(hasType2 && hitbox->mTypes[i] != 2) {
            for(int j = 0; j < 8; j++) {
                write << fourSpaceTab;
            }
        }
        for(int j = 0; j < 6; j++) {
            auto tempf = hitbox->mData[index++];
            write << FLOAT(tempf, 7) << ", ";
        }
        write << "\n";
    }
    write << "};\n";
    if (Companion::Instance->IsDebug()) {
        write << "// 0x" << std::uppercase << std::hex << off + sizeof(float) * index << "\n";
    }
    write << "\n";
}

void SF64::HitboxBinaryExporter::Export(std::ostream &write, std::shared_ptr<IParsedData> raw, std::string& entryName, YAML::Node &node, std::string* replacement ) {
    auto hitbox = std::static_pointer_cast<SF64::HitboxData>(raw);
    auto writer = LUS::BinaryWriter();

    WriteHeader(writer, LUS::ResourceType::Hitbox, 0);
    for (float value : hitbox->mData) {
        writer.Write(value);
    }
    writer.Finish(write);
}

std::optional<std::shared_ptr<IParsedData>> SF64::HitboxFactory::parse(std::vector<uint8_t>& buffer, YAML::Node& node) {
    std::vector<float> data;
    int count;
    std::vector<int> types;
    auto [_, segment] = Decompressor::AutoDecode(node, buffer, 0x10000);
    LUS::BinaryReader reader(segment.data, segment.size);
    reader.SetEndianness(LUS::Endianness::Big);

    data.push_back(reader.ReadFloat());
    count = data[0];

    while(count > 0) {
        auto typecode = reader.ReadFloat();
        auto readCount = 0;
        if(typecode == 300000.0f || typecode == 400000.0f) {
            readCount = 6;
            types.push_back((typecode == 300000.0f) ? 3 : 4);
        } else if (typecode == 200000.0f) {
            readCount = 9;
            types.push_back(2);
        } else {
            readCount = 5;
            types.push_back(1);
        }
        data.push_back(typecode);
        for(int i = 0; i < readCount; i++) {
            data.push_back(reader.ReadFloat());
        }
        count--;
    }

    return std::make_shared<SF64::HitboxData>(data, types);
}