ModelFactory.h

#pragma once
 
#include <factories/BaseFactory.h>
 
namespace BK64 {

typedef struct BoneData {
    float pos[3];
    uint16_t id;
    uint16_t parentId;
} BoneData;

typedef struct GeoCube {
    uint16_t startTri;
    uint16_t triCount;
} GeoCube;

typedef struct CollisionTri {
    uint16_t vtxIds[3];
    uint16_t unk6;
    uint32_t flags;
} CollisionTri;

typedef struct Effect {
    uint16_t dataInfo;                // packed effect type + params
    std::vector<uint16_t> vtxIndices; // the vertices this effect acts on
} Effect;

typedef struct AnimTexture {
    uint16_t frameSize;
    uint16_t frameCount;
    float frameRate;
} AnimTexture;

typedef struct CollisionHeader {
    int16_t minIndex[3];
    int16_t maxIndex[3];
    uint16_t yStride;
    uint16_t zStride;
    uint16_t geoCubeScale;
} CollisionHeader;

typedef struct AnimationHeader {
    float scalingFactor;
} AnimationHeader;

typedef struct VtxHeader {
    int16_t minCoord[3];
    int16_t maxCoord[3];
    int16_t centerCoord[3];
    int16_t localNorm;
    uint16_t count;
    int16_t globalNorm;
} VtxHeader;

typedef struct TexInfo {
    uint16_t type;
    uint8_t width;
    uint8_t height;
    uint16_t tlutColors;
    uint32_t textureDataOffset; // ROM offset, relative to the texture data base
} TexInfo;

typedef struct Unk14_0 {
    int16_t scale1[3];
    int16_t scale2[3];
    int16_t pos[3];
    uint8_t rot[3];
    uint8_t unk15;
    int8_t animIndex;
    uint8_t pad;
} Unk14_0; // 24 bytes
 
typedef struct Unk14_1 {
    int16_t unk0;
    int16_t unk2;
    int16_t pos[3];
    uint8_t rot[3];
    uint8_t unkD;
    int8_t animIndex;
    uint8_t pad;
} Unk14_1; // 16 bytes
 
typedef struct Unk14_2 {
    int16_t unk0;
    int16_t unk2[3];
    uint8_t unk8;
    int8_t unk9;
    uint8_t pad[2];
} Unk14_2; // 12 bytes

typedef struct Unk20_0 {
    int16_t unk0[3];
    int16_t unk6[3];
    uint8_t unkC;
    uint8_t pad;
} Unk20_0; // 14 bytes raw, pad to 16

typedef struct Unk28_0 {
    int16_t coord[3];
    int8_t animIndex;
    std::vector<int16_t> vtxList;
} Unk28_0;
 
class ModelData : public IParsedData {
  public:
    uint16_t mGeoType;
    uint16_t mTriCount;
    uint16_t mVertCount;
 
    // ── GEO layout ────────────────────────────────────────────────────────────
    bool mHasGeo = false;
 
    // ── Vertex list ───────────────────────────────────────────────────────────
    bool mHasVtx = false;
    VtxHeader mVtxHeader{};
 
    // ── Display lists ─────────────────────────────────────────────────────────
    bool mHasDL = false;
    uint32_t mDLCount = 0;             // total GFX command words over all sub-lists
    uint32_t mDLUnkInfo = 0;           // checksum, or whatever — from the GFX header
    uint32_t mGfxSubListCount = 0;     // how many _GFX_* sub-assets we made
    std::vector<uint32_t> mRawDLWords; // raw N64 DL words, as w0/w1 pairs
 
    // ── Texture list ──────────────────────────────────────────────────────────
    std::vector<TexInfo> mTexInfos;
    uint32_t mTexDataSize = 0;        // size of the raw texture area, for segment 2 allocation
    std::vector<uint8_t> mRawTexData; // the whole raw texture blob, animated frames and all
 
    // ── Animation data ────────────────────────────────────────────────────────
    bool mHasAnimation = false;
    AnimationHeader mAnimHeader{};
    std::vector<BoneData> mBones;
 
    // ── Collision data ────────────────────────────────────────────────────────
    bool mHasCollision = false;
    CollisionHeader mCollisionHeader{};
    std::vector<GeoCube> mGeoCubes;
    std::vector<CollisionTri> mCollisionTris;
 
    // ── Unk14 (hitbox) ────────────────────────────────────────────────────────
    bool mHasUnk14 = false;
    int16_t mUnk14Unk6 = 0;
    std::vector<Unk14_0> mUnk14Entries0;
    std::vector<Unk14_1> mUnk14Entries1;
    std::vector<Unk14_2> mUnk14Entries2;
 
    // ── Unk20 ─────────────────────────────────────────────────────────────────
    bool mHasUnk20 = false;
    std::vector<Unk20_0> mUnk20Entries;
 
    // ── Effects ───────────────────────────────────────────────────────────────
    std::vector<Effect> mEffects;
 
    // ── Unk28 ─────────────────────────────────────────────────────────────────
    bool mHasUnk28 = false;
    std::vector<Unk28_0> mUnk28Entries;
 
    // ── Animated textures ─────────────────────────────────────────────────────
    std::vector<AnimTexture> mAnimTextures;
 
    ModelData(uint16_t geoType, uint16_t triCount, uint16_t vertCount)
        : mGeoType(geoType), mTriCount(triCount), mVertCount(vertCount) {
    }
};
 
class ModelHeaderExporter : public BaseExporter {
    ExportResult Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName,
                        YAML::Node& node, std::string* replacement) override;
};
 
class ModelBinaryExporter : public BaseExporter {
    ExportResult Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName,
                        YAML::Node& node, std::string* replacement) override;
};
 
class ModelCodeExporter : public BaseExporter {
    ExportResult Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName,
                        YAML::Node& node, std::string* replacement) override;
};
 
class ModelFactory : public BaseFactory {
  public:
    std::optional<std::shared_ptr<IParsedData>> parse(std::vector<uint8_t>& buffer, YAML::Node& data) override;
    inline std::unordered_map<ExportType, std::shared_ptr<BaseExporter>> GetExporters() override {
        return { REGISTER(Code, ModelCodeExporter) REGISTER(Header, ModelHeaderExporter)
                     REGISTER(Binary, ModelBinaryExporter) };
    }
 
    bool HasModdedDependencies() override {
        return true;
    }
};
} // namespace BK64