#ifndef J3DJOINTTREE_H
#define J3DJOINTTREE_H

#include "JSystem/J3DGraphAnimator/J3DJoint.h"

class JUTNameTab;

/**
 * @ingroup jsystem-j3d
 * 
 */
struct J3DModelHierarchy {
    /* 0x0 */ u16 mType;
    /* 0x2 */ u16 mValue;
};

class J3DMaterialTable;

/**
 * @ingroup jsystem-j3d
 * 
 */
struct J3DDrawMtxData {
    /* 803115E0 */ J3DDrawMtxData();
    /* 803115F4 */ ~J3DDrawMtxData();

    /* 0x0 */ u16 mEntryNum;
    /* 0x2 */ u16 mDrawFullWgtMtxNum;
    /* 0x4 */ u8* mDrawMtxFlag;
    /* 0x8 */ u16* mDrawMtxIndex;
};  // Size: 0xC

class J3DShapeTable;

/**
 * @ingroup jsystem-j3d
 * 
 */
class J3DJointTree {
public:
    /* 80325A18 */ J3DJointTree();
    /* 80325A9C */ void makeHierarchy(J3DJoint*, J3DModelHierarchy const**, J3DMaterialTable*,
                                      J3DShapeTable*);
    /* 80325C00 */ void findImportantMtxIndex();

    /* 80325CAC */ virtual void calc(J3DMtxBuffer*, Vec const&, f32 const (&)[3][4]);
    /* 80325D24 */ virtual ~J3DJointTree() {}

    J3DModelHierarchy const* getHierarchy() { return mHierarchy; }
    void setHierarchy(J3DModelHierarchy* hierarchy) { mHierarchy = hierarchy; }
    void setBasicMtxCalc(J3DMtxCalc* calc) { mBasicMtxCalc = calc; }
    u16 getWEvlpMtxNum() const { return mWEvlpMtxNum; }
    u16* getWEvlpMixIndex() const { return mWEvlpMixMtxIndex; }
    u8 getWEvlpMixMtxNum(u16 idx) const { return mWEvlpMixMtxNum[idx]; }
    u16 * getWEvlpMixMtxIndex() const { return mWEvlpMixMtxIndex; }
    f32 * getWEvlpMixWeight() const { return mWEvlpMixWeight; }
    u16 * getWEvlpImportantMtxIndex() const { return mWEvlpImportantMtxIdx; }
    u16 getDrawFullWgtMtxNum() const { return mDrawMtxData.mDrawFullWgtMtxNum; }
    u16 getJointNum() const { return mJointNum; }
    u16 getDrawMtxNum() const { return mDrawMtxData.mEntryNum; }
    u8 getDrawMtxFlag(u16 idx) const { return mDrawMtxData.mDrawMtxFlag[idx]; }
    u16 getDrawMtxIndex(u16 idx) const { return mDrawMtxData.mDrawMtxIndex[idx]; }
    J3DDrawMtxData* getDrawMtxData() { return &mDrawMtxData; }
    JUTNameTab* getJointName() const { return mJointName; }
    J3DJoint* getRootNode() { return mRootNode; }
    J3DJoint* getJointNodePointer(u16 idx) const {
        J3D_ASSERT(139, idx < mJointNum, "Error : range over.");
        return mJointNodePointer[idx];
    }
    J3DMtxCalc* getBasicMtxCalc() const { return mBasicMtxCalc; }
    Mtx& getInvJointMtx(int idx) { return mInvJointMtx[idx]; }
    u32 getModelDataType() const { return mModelDataType; }
    void setModelDataType(u32 type) { mModelDataType = type; }
    bool checkFlag(u32 flag) { return mFlags & flag; }
    void setFlag(u32 flag) { mFlags = flag; }

private:
    friend class J3DModelLoader;
    
    /* 0x04 */ J3DModelHierarchy* mHierarchy;
    /* 0x08 */ u32 mFlags;
    /* 0x0C */ u32 mModelDataType;
    /* 0x10 */ J3DJoint* mRootNode;
    /* 0x14 */ J3DMtxCalc* mBasicMtxCalc;
    /* 0x18 */ J3DJoint** mJointNodePointer;
    /* 0x1C */ u16 mJointNum;
    /* 0x1E */ u16 mWEvlpMtxNum;
    /* 0x20 */ u8* mWEvlpMixMtxNum;
    /* 0x24 */ u16* mWEvlpMixMtxIndex;
    /* 0x28 */ f32* mWEvlpMixWeight;
    /* 0x2C */ Mtx* mInvJointMtx;
    /* 0x30 */ u16* mWEvlpImportantMtxIdx;
    /* 0x34 */ J3DDrawMtxData mDrawMtxData;
    /* 0x40 */ u32 field_0x40;
    /* 0x44 */ JUTNameTab* mJointName;
};  // Size: 0x48

#endif /* J3DJOINTTREE_H */