GeoLayoutFactory.h

#pragma once
 
#include <factories/BaseFactory.h>
#include "factories/bk64/BKHeaderExporter.h"
 
namespace BK64 {
 
typedef std::variant<uint8_t, int8_t, uint16_t, int16_t, uint32_t, int32_t, float> GeoLayoutArg;
 
enum class GeoLayoutArgType { U8, S8, U16, S16, U32, S32, FLOAT };
 
enum class GeoLayoutOpCode {
    UnknownCmd0,
    Sort,
    Bone,
    LoadDL,
    NOP,
    Skinning,
    Branch,
    UnknownCmd7,
    LOD,
    NOP2,
    ReferencePoint,
    NOP3,
    Selector,
    DrawDistance,
    UnknownCmdE,
    UnknownCmdF,
    UnknownCmd10,
};
 
class GeoLayoutCommand {
  public:
    GeoLayoutOpCode opCode;
    uint32_t cmdLength;
    std::vector<GeoLayoutArg> args;
    uint32_t originalOffset; // where this command sat in the original N64 binary
 
    GeoLayoutCommand(GeoLayoutOpCode opCode, uint32_t cmdLength, std::vector<GeoLayoutArg> args,
                     uint32_t originalOffset = 0)
        : opCode(opCode), cmdLength(cmdLength), args(std::move(args)), originalOffset(originalOffset) {
    }
};
 
class GeoLayoutData : public IParsedData {
  public:
    std::vector<GeoLayoutCommand> mCmds;
 
    GeoLayoutData(std::vector<GeoLayoutCommand> cmds) : mCmds(std::move(cmds)) {
    }
};
 
class GeoLayoutBinaryExporter : public BaseExporter {
    ExportResult Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName,
                        YAML::Node& node, std::string* replacement) override;
};
 
class GeoLayoutCodeExporter : public BaseExporter {
    ExportResult Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName,
                        YAML::Node& node, std::string* replacement) override;
};
 
class GeoLayoutModdingExporter : public BaseExporter {
    ExportResult Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName,
                        YAML::Node& node, std::string* replacement) override;
};
 
class GeoLayoutFactory : 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;
    inline std::unordered_map<ExportType, std::shared_ptr<BaseExporter>> GetExporters() override {
        return { REGISTER(Code, GeoLayoutCodeExporter) REGISTER(Header, BKHeaderExporter)
                     REGISTER(Binary, GeoLayoutBinaryExporter) REGISTER(Modding, GeoLayoutModdingExporter) };
    }
    bool SupportModdedAssets() override {
        return true;
    }
};
} // namespace BK64