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
|
#pragma once
#include "ResourceType.h"
#include "n64/Cartridge.h"
#include <cstdint>
#include <iostream>
#include <any>
#include <memory>
#include <vector>
#include <string>
#include <variant>
#include <optional>
#include <yaml-cpp/yaml.h>
#include <strhash64/StrHash64.h>
#include <binarytools/BinaryWriter.h>
#include <binarytools/BinaryReader.h>
#define REGISTER(type, c) { ExportType::type, std::make_shared<c>() },
#define SEGMENT_OFFSET(a) ((uint32_t)(a) & 0x00FFFFFF)
#define SEGMENT_NUMBER(x) (((uint32_t)(x) >> 24) & 0xFF)
// I would love to use 0x01000000, but the stupid compiler takes it as 0x01
#define IS_SEGMENTED(x) (((uint32_t)(x) > 16777216) && (SEGMENT_NUMBER(x) < 0x20))
#define ASSET_PTR(x) (IS_SEGMENTED(x) ? SEGMENT_OFFSET(x) : (x))
#define tab "\t"
#define fourSpaceTab " "
struct OffsetEntry {
uint32_t start;
uint32_t end;
};
typedef std::optional<std::variant<size_t, OffsetEntry>> ExportResult;
#define ParseResult std::optional<std::shared_ptr<IParsedData>>
enum class ExportType {
Header,
Code,
Binary,
Modding,
};
template<typename T>
std::optional<T> GetNode(YAML::Node& node, const std::string& key) {
if(!node[key]) {
return std::nullopt;
}
return std::optional<T>(node[key].as<T>());
}
template<typename T>
T GetSafeNode(YAML::Node& node, const std::string& key) {
if(!node[key]) {
throw std::runtime_error("Failed to find " + key + " in yaml");
}
return node[key].as<T>();
}
template<typename T>
T GetSafeNode(YAML::Node& node, const std::string& key, const T& def) {
if(!node[key]) {
return def;
}
return node[key].as<T>();
}
class IParsedData {};
template<typename T>
class GenericData : public IParsedData {
public:
T mData;
};
class BaseExporter {
public:
virtual ExportResult Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName, YAML::Node& node, std::string* replacement) = 0;
static void WriteHeader(LUS::BinaryWriter& write, LUS::ResourceType resType, int32_t version);
};
class BaseFactory {
public:
virtual std::optional<std::shared_ptr<IParsedData>> parse(std::vector<uint8_t>& buffer, YAML::Node& data) = 0;
virtual std::optional<std::shared_ptr<IParsedData>> parse_modding(std::vector<uint8_t>& buffer, YAML::Node& data) {
return std::nullopt;
}
std::optional<std::shared_ptr<BaseExporter>> GetExporter(ExportType type) {
auto exporters = this->GetExporters();
if (exporters.find(type) != exporters.end()) {
return exporters[type];
}
return std::nullopt;
}
virtual bool SupportModdedAssets() {
return false;
}
virtual uint32_t GetAlignment() {
return 4;
};
private:
virtual std::unordered_map<ExportType, std::shared_ptr<BaseExporter>> GetExporters() = 0;
};
|
