summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/Companion.cpp2
-rw-r--r--src/factories/ResourceType.h3
-rw-r--r--src/factories/sf64/HitboxFactory.cpp112
-rw-r--r--src/factories/sf64/HitboxFactory.h42
4 files changed, 159 insertions, 0 deletions
diff --git a/src/Companion.cpp b/src/Companion.cpp
index 9fda03e..e099c5e 100644
--- a/src/Companion.cpp
+++ b/src/Companion.cpp
@@ -33,6 +33,7 @@
#include "factories/sf64/SkeletonFactory.h"
#include "factories/sf64/AnimFactory.h"
#include "factories/sf64/ScriptFactory.h"
+#include "factories/sf64/HitboxFactory.h"
using namespace std::chrono;
namespace fs = std::filesystem;
@@ -75,6 +76,7 @@ void Companion::Init(const ExportType type) {
this->RegisterFactory("SF64:MESSAGE", std::make_shared<SF64::MessageFactory>());
this->RegisterFactory("SF64:MSG_TABLE", std::make_shared<SF64::MessageLookupFactory>());
this->RegisterFactory("SF64:SCRIPT", std::make_shared<SF64::ScriptFactory>());
+ this->RegisterFactory("SF64:HITBOX", std::make_shared<SF64::HitboxFactory>());
this->Process();
diff --git a/src/factories/ResourceType.h b/src/factories/ResourceType.h
index 27e7f82..c8d9257 100644
--- a/src/factories/ResourceType.h
+++ b/src/factories/ResourceType.h
@@ -36,5 +36,8 @@ enum class ResourceType {
AnimData = 0x414E494D, // ANIM
Message = 0x4D53474C, // MSG
MessageTable = 0x4D534754, // MSGT
+ Skeleton = 0x534B454C, // SKEL
+ Script = 0x53435250, // SCRP
+ Hitbox = 0x48544258, // HTBX
};
} // namespace LUS
diff --git a/src/factories/sf64/HitboxFactory.cpp b/src/factories/sf64/HitboxFactory.cpp
new file mode 100644
index 0000000..2266b86
--- /dev/null
+++ b/src/factories/sf64/HitboxFactory.cpp
@@ -0,0 +1,112 @@
+#include "HitboxFactory.h"
+#include "spdlog/spdlog.h"
+
+#include "Companion.h"
+#include "utils/Decompressor.h"
+#include "utils/TorchUtils.h"
+
+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 off = offset;
+ if(IS_SEGMENTED(off)) {
+ off = SEGMENT_OFFSET(off);
+ }
+
+ 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++) {
+ write << hitbox->mData[index++] << ", ";
+ }
+ write << fourSpaceTab;
+ } else {
+ write << " /* HITBOX_TYPE_1 */ ";
+ }
+ for(int j = 0; j < 6; j++) {
+ write << hitbox->mData[index++] << ", ";
+ }
+ 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);
+}
diff --git a/src/factories/sf64/HitboxFactory.h b/src/factories/sf64/HitboxFactory.h
new file mode 100644
index 0000000..8c54c7e
--- /dev/null
+++ b/src/factories/sf64/HitboxFactory.h
@@ -0,0 +1,42 @@
+#pragma once
+
+#include "../BaseFactory.h"
+
+namespace SF64 {
+
+class HitboxData : public IParsedData {
+public:
+ std::vector<float> mData;
+ std::vector<int> mTypes;
+
+ HitboxData(std::vector<float> data, std::vector<int> types);
+};
+
+class HitboxHeaderExporter : public BaseExporter {
+ void Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName, YAML::Node& node, std::string* replacement) override;
+};
+
+class HitboxBinaryExporter : public BaseExporter {
+ void Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName, YAML::Node& node, std::string* replacement) override;
+};
+
+class HitboxCodeExporter : public BaseExporter {
+ void Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName, YAML::Node& node, std::string* replacement) override;
+};
+
+class HitboxFactory : public BaseFactory {
+public:
+ std::optional<std::shared_ptr<IParsedData>> parse(std::vector<uint8_t>& buffer, YAML::Node& data) override;
+ std::optional<std::shared_ptr<IParsedData>> parse_modding(std::vector<uint8_t>& buffer, YAML::Node& data) override {
+ return std::nullopt;
+ }
+ inline std::unordered_map<ExportType, std::shared_ptr<BaseExporter>> GetExporters() override {
+ return {
+ REGISTER(Code, HitboxCodeExporter)
+ REGISTER(Header, HitboxHeaderExporter)
+ REGISTER(Binary, HitboxBinaryExporter)
+ };
+ }
+ bool SupportModdedAssets() override { return false; }
+};
+}