env (#38)

3 files changed, 153 insertions, 0 deletions

diff --git a/src/Companion.cpp b/src/Companion.cpp
index b59adb1..396322d 100644
--- a/src/Companion.cpp
+++ b/src/Companion.cpp
@@ -34,6 +34,7 @@
 #include "factories/sf64/AnimFactory.h"
 #include "factories/sf64/ScriptFactory.h"
 #include "factories/sf64/HitboxFactory.h"
+#include "factories/sf64/EnvSettingsFactory.h"
 #include "factories/sf64/ObjInitFactory.h"
 #include <regex>
 
@@ -79,6 +80,7 @@ void Companion::Init(const ExportType type) {
     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->RegisterFactory("SF64:ENVSETTINGS", std::make_shared<SF64::EnvSettingsFactory>());
     this->RegisterFactory("SF64:OBJECT_INIT", std::make_shared<SF64::ObjInitFactory>());
 
     this->Process();
diff --git a/src/factories/sf64/EnvSettingsFactory.cpp b/src/factories/sf64/EnvSettingsFactory.cpp
new file mode 100644
index 0000000..7fd6613
--- /dev/null
+++ b/src/factories/sf64/EnvSettingsFactory.cpp
@@ -0,0 +1,93 @@
+#include "EnvSettingsFactory.h"
+#include "spdlog/spdlog.h"
+
+#include "Companion.h"
+#include "utils/Decompressor.h"
+#include "utils/TorchUtils.h"
+
+SF64::EnvSettingsData::EnvSettingsData(int32_t type, int32_t unk_04, uint16_t bgColor, uint16_t seqId, int32_t fogR, int32_t fogG, int32_t fogB, int32_t fogN, int32_t fogF, float unk_20x, float unk_20y, float unk_20z, int32_t lightR, int32_t lightG, int32_t lightB, int32_t ambR, int32_t ambG, int32_t ambB): mType(type), mUnk_04(unk_04), mBgColor(bgColor), mSeqId(seqId), mFogR(fogR), mFogG(fogG), mFogB(fogB), mFogN(fogN), mFogF(fogF), mUnk_20x(unk_20x), mUnk_20y(unk_20y), mUnk_20z(unk_20z), mLightR(lightR), mLightG(lightG), mLightB(lightB), mAmbR(ambR), mAmbG(ambG), mAmbB(ambB) {
+    
+}
+
+void SF64::EnvSettingsHeaderExporter::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 EnvSettings " << symbol << ";\n";
+}
+
+void SF64::EnvSettingsCodeExporter::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 env = std::static_pointer_cast<SF64::EnvSettingsData>(raw);
+    auto off = offset;
+    if(IS_SEGMENTED(off)) {
+        off = SEGMENT_OFFSET(off);
+    } 
+    if (Companion::Instance->IsDebug()) {
+        write << "// 0x" << std::uppercase << std::hex << off << "\n";
+    }
+    write << "EnvSettings " << symbol << " = {\n";
+    write << fourSpaceTab;
+    write << std::dec << env->mType << ", ";
+    write << env->mUnk_04 << ", ";
+    write << env->mBgColor << ", ";
+    if(env->mSeqId < 0x8000) {
+        write << env->mSeqId << ", ";
+    } else {
+        write << env->mSeqId - 0x8000 << " | 0x8000, ";
+    }
+    write << env->mFogR << ", ";
+    write << env->mFogG << ", ";
+    write << env->mFogB << ", ";
+    write << env->mFogN << ", ";
+    write << env->mFogF << ", ";
+    write << "{" << env->mUnk_20x << ", ";
+    write << env->mUnk_20y << ", ";
+    write << env->mUnk_20z << "}, ";
+    write << env->mLightR << ", ";
+    write << env->mLightG << ", ";
+    write << env->mLightB << ", ";
+    write << env->mAmbR << ", ";
+    write << env->mAmbG << ", ";
+    write << env->mAmbB << ",\n";
+    write << "};\n";
+    if (Companion::Instance->IsDebug()) {
+        write << "// 0x" << std::uppercase << std::hex << off + sizeof(EnvSettingsData) << "\n";
+    }
+    write << "\n";
+}
+
+void SF64::EnvSettingsBinaryExporter::Export(std::ostream &write, std::shared_ptr<IParsedData> raw, std::string& entryName, YAML::Node &node, std::string* replacement ) {
+
+}
+
+std::optional<std::shared_ptr<IParsedData>> SF64::EnvSettingsFactory::parse(std::vector<uint8_t>& buffer, YAML::Node& node) {
+    auto [_, segment] = Decompressor::AutoDecode(node, buffer, sizeof(SF64::EnvSettingsData));
+    LUS::BinaryReader reader(segment.data, segment.size);
+    reader.SetEndianness(LUS::Endianness::Big);
+    int32_t  type = reader.ReadInt32();
+    int32_t  unk_04 = reader.ReadInt32();
+    uint16_t bgColor = reader.ReadUInt16();
+    uint16_t seqId = reader.ReadUInt16();
+    int32_t  fogR = reader.ReadInt32();
+    int32_t  fogG = reader.ReadInt32();
+    int32_t  fogB = reader.ReadInt32();
+    int32_t  fogN = reader.ReadInt32();
+    int32_t  fogF = reader.ReadInt32();
+    float    unk_20x = reader.ReadFloat();
+    float    unk_20y = reader.ReadFloat();
+    float    unk_20z = reader.ReadFloat();
+    int32_t  lightR = reader.ReadInt32();
+    int32_t  lightG = reader.ReadInt32();
+    int32_t  lightB = reader.ReadInt32();
+    int32_t  ambR = reader.ReadInt32();
+    int32_t  ambG = reader.ReadInt32();
+    int32_t  ambB = reader.ReadInt32();
+
+    return std::make_shared<SF64::EnvSettingsData>(type, unk_04, bgColor, seqId, fogR, fogG, fogB, fogN, fogF, unk_20x, unk_20y, unk_20z, lightR, lightG, lightB, ambR, ambG, ambB);
+}
diff --git a/src/factories/sf64/EnvSettingsFactory.h b/src/factories/sf64/EnvSettingsFactory.h
new file mode 100644
index 0000000..985a391
--- /dev/null
+++ b/src/factories/sf64/EnvSettingsFactory.h
@@ -0,0 +1,58 @@
+#pragma once
+
+#include "../BaseFactory.h"
+
+namespace SF64 {
+
+class EnvSettingsData : public IParsedData {
+public:
+    int32_t  mType;
+    int32_t  mUnk_04;
+    uint16_t mBgColor;
+    uint16_t mSeqId;
+    int32_t  mFogR;
+    int32_t  mFogG;
+    int32_t  mFogB;
+    int32_t  mFogN;
+    int32_t  mFogF;
+    float    mUnk_20x;
+    float    mUnk_20y;
+    float    mUnk_20z;
+    int32_t  mLightR;
+    int32_t  mLightG;
+    int32_t  mLightB;
+    int32_t  mAmbR;
+    int32_t  mAmbG;
+    int32_t  mAmbB;
+
+    EnvSettingsData(int32_t type, int32_t unk_04, uint16_t bgColor, uint16_t seqId, int32_t fogR, int32_t fogG, int32_t fogB, int32_t fogN, int32_t fogF, float unk_20x, float unk_20y, float unk_20z, int32_t lightR, int32_t lightG, int32_t lightB, int32_t ambR, int32_t ambG, int32_t ambB);
+};
+
+class EnvSettingsHeaderExporter : public BaseExporter {
+    void Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName, YAML::Node& node, std::string* replacement) override;
+};
+
+class EnvSettingsBinaryExporter : public BaseExporter {
+    void Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName, YAML::Node& node, std::string* replacement) override;
+};
+
+class EnvSettingsCodeExporter : public BaseExporter {
+    void Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName, YAML::Node& node, std::string* replacement) override;
+};
+
+class EnvSettingsFactory : 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, EnvSettingsCodeExporter)
+            REGISTER(Header, EnvSettingsHeaderExporter)
+            REGISTER(Binary, EnvSettingsBinaryExporter)
+        };
+    }
+    bool SupportModdedAssets() override { return false; }
+};
+}