#ifndef JPABASESHAPE_H
#define JPABASESHAPE_H

#include "dolphin/gx/GX.h"
#include "dolphin/types.h"

class JPAEmitterWorkData;
class JKRHeap;

struct JPABaseShapeData {
    // Common header.
    /* 0x00 */ u8 mMagic[4];
    /* 0x04 */ u32 mSize;

    /* 0x08 */ u32 mFlags;
    /* 0x0C */ s16 mClrPrmAnmOffset;
    /* 0x0E */ s16 mClrEnvAnmOffset;
    /* 0x10 */ f32 mBaseSizeX;
    /* 0x14 */ f32 mBaseSizeY;
    /* 0x18 */ u16 mBlendModeCfg;
    /* 0x1A */ u8 mAlphaCompareCfg;
    /* 0x1B */ u8 mAlphaRef0;
    /* 0x1C */ u8 mAlphaRef1;
    /* 0x1D */ u8 mZModeCfg;
    /* 0x1E */ u8 mTexFlg;
    /* 0x1F */ u8 mTexAnmNum;
    /* 0x20 */ u8 mTexIdx;
    /* 0x21 */ u8 mClrFlg;
    /* 0x22 */ u8 mClrPrmKeyNum;
    /* 0x23 */ u8 mClrEnvKeyNum;
    /* 0x24 */ s16 mClrAnmFrmMax;
    /* 0x26 */ GXColor mClrPrm;
    /* 0x2A */ GXColor mClrEnv;
    /* 0x2E */ u8 mAnmRndm;
    /* 0x2F */ u8 mClrAnmRndmMask;
    /* 0x30 */ u8 mTexAnmRndmMask;
};

struct JPABaseShape {
public:
    /* 8027A6DC */ JPABaseShape(u8 const*, JKRHeap*);
    /* 8027A7E8 */ void setGX(JPAEmitterWorkData*) const;

    static GXBlendMode st_bm[3];
    static GXBlendFactor st_bf[10];
    static GXLogicOp st_lo[16];
    static GXCompare st_c[8];
    static GXAlphaOp st_ao[4];
    static GXTevColorArg st_ca[6][4];
    static GXTevAlphaArg st_aa[2][4];

    GXBlendMode getBlendMode() const { return st_bm[mpData->mBlendModeCfg & 0x03]; }
    GXBlendFactor getBlendSrc() const { return st_bf[(mpData->mBlendModeCfg >> 2) & 0x0F]; }
    GXBlendFactor getBlendDst() const { return st_bf[(mpData->mBlendModeCfg >> 6) & 0x0F]; }
    GXLogicOp getLogicOp() const { return st_lo[(mpData->mBlendModeCfg >> 10) & 0x0F]; }
    GXBool getZCompLoc() const { return (GXBool)((mpData->mZModeCfg >> 5) & 0x01); }

    GXBool getZEnable() const { return (GXBool)(mpData->mZModeCfg & 0x01); }
    GXCompare getZCmp() const { return st_c[(mpData->mZModeCfg >> 1) & 0x07]; }
    GXBool getZUpd() const { return (GXBool)((mpData->mZModeCfg >> 4) & 0x01); }

    GXCompare getAlphaCmp0() const { return st_c[mpData->mAlphaCompareCfg & 0x07]; }
    u8 getAlphaRef0() const { return mpData->mAlphaRef0; }
    GXAlphaOp getAlphaOp() const { return st_ao[(mpData->mAlphaCompareCfg >> 3) & 0x03]; }
    GXCompare getAlphaCmp1() const { return st_c[(mpData->mAlphaCompareCfg >> 5) & 0x07]; }
    u8 getAlphaRef1() const { return mpData->mAlphaRef1; }

    const GXTevColorArg* getTevColorArg() const { return st_ca[(mpData->mFlags >> 0x0F) & 0x07]; }
    const GXTevAlphaArg* getTevAlphaArg() const { return st_aa[(mpData->mFlags >> 0x12) & 0x01]; }

    u32 getType() const { return (mpData->mFlags >> 0) & 0x0F; }
    u32 getDirType() const { return (mpData->mFlags >> 4) & 0x07; }
    u32 getRotType() const { return (mpData->mFlags >> 7) & 0x07; }
    u32 getBasePlaneType() const { return (mpData->mFlags >> 10) & 0x07; }
    u32 getTilingS() const { return (mpData->mFlags >> 25) & 0x01; }
    u32 getTilingT() const { return (mpData->mFlags >> 26) & 0x01; }
    bool isGlblClrAnm() const { return !!(mpData->mFlags & 0x00001000); }
    bool isGlblTexAnm() const { return !!(mpData->mFlags & 0x00004000); }
    bool isPrjTex() const { return !!(mpData->mFlags & 0x00100000); }
    bool isDrawFwdAhead() const { return !!(mpData->mFlags & 0x00200000); }
    bool isDrawPrntAhead() const { return !!(mpData->mFlags & 0x00400000); }
    bool isClipOn() const { return !!(mpData->mFlags & 0x00800000); }
    bool isTexCrdAnm() const { return !!(mpData->mFlags & 0x01000000); }
    bool isNoDrawParent() const { return !!(mpData->mFlags & 0x08000000); }
    bool isNoDrawChild() const { return !!(mpData->mFlags & 0x10000000); }

    bool isPrmAnm() const { return !!(mpData->mClrFlg & 0x02); }
    bool isEnvAnm() const { return !!(mpData->mClrFlg & 0x08); }
    u8 getClrAnmType() const { return (mpData->mClrFlg >> 4) & 0x07; }
    s16 getClrAnmMaxFrm() const { return mpData->mClrAnmFrmMax; }
    void getPrmClr(s16 idx, GXColor* dst) { *dst = mpPrmClrAnmTbl[idx]; }
    void getEnvClr(s16 idx, GXColor* dst) { *dst = mpEnvClrAnmTbl[idx]; }

    bool isTexAnm() const { return !!(mpData->mTexFlg & 0x01); }
    u8 getTexAnmType() const { return (mpData->mTexFlg >> 2) & 0x07; }
    u32 getTexIdx() const { return mpData->mTexIdx; }

    f32 getBaseSizeX() const { return mpData->mBaseSizeX; }
    f32 getBaseSizeY() const { return mpData->mBaseSizeY; }

public:
    /* 0x00 */ const JPABaseShapeData* mpData;
    /* 0x04 */ const void* mpTexCrdMtxAnmTbl;
    /* 0x08 */ const u8* mpTexIdxAnimTbl;
    /* 0x0C */ GXColor* mpPrmClrAnmTbl;
    /* 0x10 */ GXColor* mpEnvClrAnmTbl;
};

#endif /* JPABASESHAPE_H */