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
|
#include "PaintingMapFactory.h"
#include "spdlog/spdlog.h"
#include "Companion.h"
#include "utils/Decompressor.h"
#include "utils/TorchUtils.h"
#include <regex>
// #define FORMAT_INT(x, w) std::dec << std::setfill(' ') << std::setw(w) << x
ExportResult SM64::PaintingMapHeaderExporter::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 std::nullopt;
}
write << "extern PaintingData " << symbol << "[];\n";
return std::nullopt;
}
ExportResult SM64::PaintingMapCodeExporter::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 paintingData = std::static_pointer_cast<SM64::PaintingData>(raw);
write << "static const PaintingData " << symbol << "[] = {\n";
write << fourSpaceTab << paintingData->mPaintingMappings.size() << ",\n";
for (auto& mapping : paintingData->mPaintingMappings) {
write << fourSpaceTab;
write << mapping.vtxId << ", " << mapping.texX << ", " << mapping.texY << ",\n";
}
write << fourSpaceTab << paintingData->mPaintingGroups.size() << ",\n";
for (auto& group : paintingData->mPaintingGroups) {
write << fourSpaceTab;
write << group.x << ", " << group.y << ", " << group.z << ",\n";
}
write << "};\n";
size_t size =
(paintingData->mPaintingMappings.size() + paintingData->mPaintingGroups.size()) * 3 * sizeof(int16_t) + 2;
return offset + size;
}
ExportResult SM64::PaintingMapBinaryExporter::Export(std::ostream& write, std::shared_ptr<IParsedData> raw,
std::string& entryName, YAML::Node& node,
std::string* replacement) {
auto writer = LUS::BinaryWriter();
auto paintingData = std::static_pointer_cast<SM64::PaintingData>(raw);
WriteHeader(writer, Torch::ResourceType::PaintingData, 0);
uint32_t mappingsSize = paintingData->mPaintingMappings.size();
uint32_t groupsSize = paintingData->mPaintingGroups.size();
writer.Write((uint32_t)((mappingsSize * 3) + (groupsSize * 3)) + 2);
writer.Write((int16_t)mappingsSize);
for (auto& mapping : paintingData->mPaintingMappings) {
writer.Write(mapping.vtxId);
writer.Write(mapping.texX);
writer.Write(mapping.texY);
}
writer.Write((int16_t)groupsSize);
for (auto& group : paintingData->mPaintingGroups) {
writer.Write(group.x);
writer.Write(group.y);
writer.Write(group.z);
}
writer.Finish(write);
return std::nullopt;
}
std::optional<std::shared_ptr<IParsedData>> SM64::PaintingMapFactory::parse(std::vector<uint8_t>& buffer,
YAML::Node& node) {
std::vector<PaintingMapping> paintingMappings;
std::vector<Vec3s> paintingGroups;
auto [_, segment] = Decompressor::AutoDecode(node, buffer);
LUS::BinaryReader reader(segment.data, segment.size);
reader.SetEndianness(Torch::Endianness::Big);
auto mappingsSize = reader.ReadInt16();
for (int16_t i = 0; i < mappingsSize; ++i) {
auto vtxId = reader.ReadInt16();
auto texX = reader.ReadInt16();
auto texY = reader.ReadInt16();
paintingMappings.emplace_back(vtxId, texX, texY);
}
auto groupSize = reader.ReadInt16();
for (int16_t i = 0; i < groupSize; ++i) {
auto x = reader.ReadInt16();
auto y = reader.ReadInt16();
auto z = reader.ReadInt16();
paintingGroups.emplace_back(x, y, z);
}
return std::make_shared<SM64::PaintingData>(paintingMappings, paintingGroups);
}
|
