sms/include/Player/ModelWaterManager.hpp

#ifndef PLAYER_MODEL_WATER_MANAGER_HPP
#define PLAYER_MODEL_WATER_MANAGER_HPP

#include <JSystem/JDrama/JDRViewObj.hpp>
#include <JSystem/JGeometry.hpp>
#include <System/ParamInst.hpp>
#include <System/Params.hpp>
#include <Strategic/HitActor.hpp>

class TBGCheckData;
class TDLTexQuad;
class JUTTexture;
class J3DModelData;

class TWaterEmitInfo : public TParams {
public:
	TWaterEmitInfo(const char*);

public:
	/* 0x8 */ TParamRT<s32> mNum;
	/* 0x1C */ TParamRT<s16> mAlive;
	/* 0x30 */ TParamRT<s16> mAttack;
	/* 0x44 */ TParamVec mDir;
	/* 0x60 */ TParamVec mPos;
	/* 0x7C */ TParamVec mV;
	/* 0x98 */ TParamRT<f32> mDirTremble;
	/* 0xAC */ TParamRT<f32> mPow;
	/* 0xC0 */ TParamRT<f32> mPowTremble;
	/* 0xD4 */ TParamRT<f32> mSize;
	/* 0xE8 */ TParamRT<f32> mSizeTremble;
	/* 0xFC */ TParamRT<f32> mHitRadius;
	/* 0x110 */ TParamRT<f32> mHitHeight;
	/* 0x124 */ TParamRT<s32> mFlag;
	/* 0x138 */ TParamRT<s16> mType;
	/* 0x14C */ TParamRT<s16> __padding;
};

class TWaterParticleType : public TParams {
public:
	TWaterParticleType(const char* path);

	f32 getAlive() const { return mAlive.get(); }

public:
	/* 0x8 */ TParamRT<f32> mExtension;
	/* 0x1C */ TParamRT<f32> mMagnify;
	/* 0x30 */ TParamRT<f32> mCleanSize;
	/* 0x44 */ TParamRT<f32> mGravity;
	/* 0x58 */ TParamRT<f32> mAlive;
};

class TWaterHitActor : public THitActor {
public:
	TWaterHitActor() { }
	TWaterHitActor(const char* name)
	    : THitActor(name)
	    , mWaterHitCounter(0)
	{
	}
	virtual BOOL receiveMessage(THitActor* sender, u32 message);
	void onWaterHitCounter();

public:
	// TODO: this is cursed, but it's what they do? Maybe ODR violation on the
	// TWaterHitActor type?
	union {
		/* 0x68 */ int unk68; // Particle index for the static water actor.
		/* 0x68 */ s16 mWaterHitCounter;
	};
};

class TModelWaterManager;

extern TModelWaterManager* gpModelWaterManager;

class TModelWaterManager : public JDrama::TViewObj {
public:
	enum WaterType {
		WATER_TYPE_NORMAL       = 0,
		WATER_TYPE_ORANGE_JUICE = 1,
		WATER_TYPE_PURPLE_JUICE = 2,
		WATER_TYPE_PINK_JUICE   = 3,
	};

	TModelWaterManager(const char* name = "<TModelWaterParticle>")
	    : JDrama::TViewObj(name)
	{
	}

	virtual void load(JSUMemoryInputStream& stream);
	virtual void loadAfter();
	f32 getWPGravity(int) const;
	void getWaterAlpha() const;
	bool askHitWaterParticleOnGround(const JGeometry::TVec3<f32>&);
	void makeEmit(const TWaterEmitInfo&);
	u8 emitRequest(const TWaterEmitInfo&);
	void splashSound(const JGeometry::TVec3<f32>&, f32) const;
	void splashGround(int);
	void touchingExec(int);
	void splashWall(int);
	void splashWallPosSize(const JGeometry::TVec3<f32>&, f32);
	f32 getPlaneFriction(const TBGCheckData*);
	f32 getPlaneFall(const TBGCheckData*);
	f32 getPlaneVanishSpeed(const TBGCheckData*);
	bool askDoWaterHitCheck();
	void wind(const JGeometry::TVec3<f32>&);
	void garbageCollect();
	void move();
	void calcWorldMinMax();
	void calcDrawVtx(MtxPtr);
	void calcVMMtxGround(MtxPtr, f32, const JGeometry::TVec3<f32>&,
	                     const JGeometry::TVec3<f32>&, MtxPtr);
	void calcVMMtxWall(MtxPtr, f32, const JGeometry::TVec3<f32>&,
	                   const JGeometry::TVec3<f32>&, MtxPtr);
	void calcVMAll(MtxPtr);
	void drawTouching();
	void drawTouchingMask();
	void drawSilhouette(MtxPtr);
	void drawWaterVolume(MtxPtr);
	void drawMirror(MtxPtr);
	void drawShineShadowVolume(MtxPtr);
	void drawRefracAndSpec() const;
	virtual void perform(u32 cue, JDrama::TGraphics* graphics);

	static TWaterHitActor mStaticHitActor;

	// fabricated
	u8 getWaterType() const { return mWaterType; }
	// fabricated
	bool isWaterType(u8 waterType) const { return mWaterType == waterType; }
	// fabricated
	void setWaterType(u8 waterType) { mWaterType = waterType; }

	// fabricated
	BOOL checkParticleFlag(TWaterHitActor* hit, u16 flag)
	{
		return mParticleFlagSOA[hit->unk68] & flag ? TRUE : FALSE;
	}
	u16 getParticleFlag(TWaterHitActor* hit)
	{
		return mParticleFlagSOA[hit->unk68];
	}
	s16 getParticleAttack(TWaterHitActor* hit)
	{
		return mParticleAttackSOA[hit->unk68];
	}
	int getFlagBottom4Bits(int i) const { return mParticleFlagSOA[i] & 0xf; }
	void setFlagBottom4Bits(int i, int flag)
	{
		mParticleFlagSOA[i] = (mParticleFlagSOA[i] & ~0xf) | flag;
	}
	bool checkFlagBottom4Bits(int i, int flag) const
	{
		return (mParticleFlagSOA[i] & 0xf) == flag ? true : false;
	}
	JGeometry::TVec3<f32> vecBetweenParticles(int i, int j) const
	{
		JGeometry::TVec3<f32> d = mParticlePositionSOA[i];
		d -= mParticlePositionSOA[j];
		return d;
	}
	const JGeometry::TVec3<f32>* getParticlePositions() const
	{
		return mParticlePositionSOA;
	}
	THitActor** getParticleUnk2514() { return unk2514; }
	u16 getParticleCount() const { return mParticleCount; }

	enum { SLOT_NUM = 256 };

	// Fabricated
	// Enum?
	const TWaterParticleType* getWaterParticleType(int type) const
	{
		return mWaterParticleTypes[type];
	}

public:
	/* 0x10 */ s16 unk10;
	/* 0x12 */ u16 mParticleCount;
	/* 0x14 */ f32 mParticleLifetimeSOA[SLOT_NUM];
	/* 0x414 */ u16 mParticleFlagSOA[SLOT_NUM];
	/* 0x614 */ s16 mParticleAttackSOA[SLOT_NUM];
	/* 0x814 */ JGeometry::TVec3<f32> mParticlePositionSOA[SLOT_NUM];
	/* 0x1414 */ JGeometry::TVec3<f32> mParticleVelocitySOA[SLOT_NUM];
	/* 0x2014 */ f32 mParticleSizeSOA[SLOT_NUM];
	/* 0x2414 */ u8 mParticleTypeSOA[SLOT_NUM];
	/* 0x2514 */ THitActor* unk2514[SLOT_NUM];
	/* 0x2914 */ const TBGCheckData* unk2914[SLOT_NUM];
	/* 0x2D14 */ Mtx unk2D14[SLOT_NUM];
	/* 0x5D14 */ f32 unk5D14;
	/* 0x5D18 */ f32 unk5D18;
	/* 0x5D1C */ f32 unk5D1C;
	/* 0x5D20 */ GXColor unk5D20;
	/* 0x5D24 */ GXColor unk5D24;
	/* 0x5D28 */ f32 unk5D28;
	/* 0x5D2C */ f32 unk5D2C;
	/* 0x5D30 */ TDLTexQuad* unk5D30;
	/* 0x5D34 */ JUTTexture* unk5D34;
	/* 0x5D38 */ JUTTexture* unk5D38;
	/* 0x5D3C */ JUTTexture* unk5D3C;
	/* 0x5D40 */ JUTTexture* unk5D40;
	/* 0x5D44 */ JUTTexture* unk5D44;
	/* 0x5D48 */ J3DModelData* unk5D48;
	/* 0x5D4C */ J3DModelData* unk5D4C;
	/* 0x5D50 */ J3DModelData* unk5D50;
	/* 0x5D54 */ J3DModelData* unk5D54;
	/* 0x5D58 */ J3DModelData* unk5D58;
	/* 0x5D5C */ u8 unk5D5C;
	/* 0x5D5D */ u8 unk5D5D;
	/* 0x5D5E */ u8 unk5D5E;
	/* 0x5D5F */ u8 mWaterType;
	/* 0x5D60 */ u16 unk5D60;
	/* 0x5D62 */ u8 unk5D62;
	/* 0x5D63 */ u8 unk5D63;
	/* 0x5D64 */ u8 unk5D64;
	/* 0x5D65 */ u8 unk5D65;
	/* 0x5D68 */ f32 unk5D68;
	/* 0x5D6C */ f32 unk5D6C;
	/* 0x5D70 */ JGeometry::TVec3<f32> unk5D70;
	/* 0x5D7C */ JGeometry::TVec3<f32> unk5D7C;
	/* 0x5D88 */ f32 unk5D88[13];
	/* 0x5DBC */ TWaterParticleType* mWaterParticleTypes[17];
	/* 0x5E00 */ int unk5E00;
	/* 0x5E04 */ u8 unk5E04;
	/* 0x5E08 */ f32 unk5E08;
	/* 0x5E0C */ f32 unk5E0C;
	/* 0x5E10 */ Mtx unk5E10;
	/* 0x5E40 */ f32 unk5E40;
	/* 0x5E44 */ u8 unk5E44;
	/* 0x5E45 */ u8 unk5E45;
	/* 0x5E46 */ u8 unk5E46;
	/* 0x5E47 */ u8 unk5E47;
	/* 0x5E48 */ u8 unk5E48;
	/* 0x5E49 */ u8 unk5E49;
};

#endif