#ifndef __URDE_CPARTICLEGENINFO_HPP__
#define __URDE_CPARTICLEGENINFO_HPP__

#include "RetroTypes.hpp"
#include "CParticleData.hpp"
#include "zeus/CVector3f.hpp"
#include "zeus/CAABox.hpp"

namespace urde
{
struct SObjectTag;
class CParticleGen;
class CStateManager;

enum class EParticleGenState
{
    NotStarted,
    Started
};

class CParticleGenInfo
{
    SObjectTag x4_part;
    float xc_seconds;
    std::string x10_boneName;
    float x20_curTime = 0.f;
    bool x24_active = false;
    CParticleData::EParentedMode x28_parentMode;
    s32 x2c_flags;
    zeus::CVector3f x30_particleScale;
    float x3c_finishTime = 0.f;
    bool x40_grabInitialData = false;
    zeus::CTransform x44_transform;
    zeus::CVector3f x74_offset;
    EParticleGenState x80_state;

public:
    CParticleGenInfo(const SObjectTag& part, int frameCount, std::string_view boneName,
                     const zeus::CVector3f& scale, CParticleData::EParentedMode parentMode,
                     int flags, EParticleGenState state);

    virtual ~CParticleGenInfo() = default;
    virtual void AddToRenderer() = 0;
    virtual void Render() = 0;
    virtual void Update(float dt, CStateManager& stateMgr) = 0;
    virtual void SetOrientation(const zeus::CTransform& xf, CStateManager& stateMgr) = 0;
    virtual void SetTranslation(const zeus::CVector3f& trans, CStateManager& stateMgr) = 0;
    virtual void SetGlobalOrientation(const zeus::CTransform& xf, CStateManager& stateMgr) = 0;
    virtual void SetGlobalTranslation(const zeus::CVector3f& trans, CStateManager& stateMgr) = 0;
    virtual void SetGlobalScale(const zeus::CVector3f& scale) = 0;
    virtual void SetParticleEmission(bool, CStateManager& stateMgr) = 0;
    virtual bool IsSystemDeletable() const = 0;
    virtual rstl::optional_object<zeus::CAABox> GetBounds() const = 0;
    virtual bool HasActiveParticles() const = 0;
    virtual void DestroyParticles() = 0;
    virtual bool HasLight() const = 0;
    virtual TUniqueId GetLightId() const = 0;
    virtual void DeleteLight(CStateManager& stateMgr) = 0;
    virtual void SetModulationColor(const zeus::CColor& color) = 0;

    void SetFlags(s32 f) { x2c_flags = f; }
    s32 GetFlags() const { return x2c_flags; }
    void SetIsGrabInitialData(bool g) { x40_grabInitialData = g; }
    bool GetIsGrabInitialData() const { return x40_grabInitialData; }
    bool GetIsActive() const { return x24_active; }
    void SetIsActive(bool a) { x24_active = a; }
    void OffsetTime(float dt) { x20_curTime += dt; }
    const zeus::CVector3f& GetCurOffset() const { return x74_offset; }
    void SetCurOffset(const zeus::CVector3f& offset) { x74_offset = offset; }
    const zeus::CTransform& GetCurTransform() const { return x44_transform; }
    void SetCurTransform(const zeus::CTransform& xf) { x44_transform = xf; }
    const zeus::CVector3f& GetCurScale() const { return x30_particleScale; }
    void SetCurScale(const zeus::CVector3f& scale) { x30_particleScale = scale; }
    void SetInactiveStartTime(float s) { xc_seconds = s; }
    float GetInactiveStartTime() const { return xc_seconds; }
    void MarkFinishTime() { x3c_finishTime = x20_curTime; }
    float GetFinishTime() const { return x3c_finishTime; }
    float GetCurrentTime() const { return x20_curTime; }
    void SetCurrentTime(float t) { x20_curTime = t; }
    EParticleGenState GetState() const { return x80_state; }
    void SetState(EParticleGenState s) { x80_state = s; }

    CParticleData::EParentedMode GetParentedMode() const { return x28_parentMode; }
    std::string_view GetLocatorName() const { return x10_boneName; }
};

class CParticleGenInfoGeneric : public CParticleGenInfo
{
    std::shared_ptr<CParticleGen> x84_system;
    TUniqueId x88_lightId;

public:
    CParticleGenInfoGeneric(const SObjectTag& part, const std::weak_ptr<CParticleGen>& system,
                            int frames, std::string_view boneName, const zeus::CVector3f& scale,
                            CParticleData::EParentedMode parentMode, int flags, CStateManager& stateMgr, TAreaId,
                            int lightId, EParticleGenState state);

    void AddToRenderer();
    void Render();
    void Update(float dt, CStateManager& stateMgr);
    void SetOrientation(const zeus::CTransform& xf, CStateManager& stateMgr);
    void SetTranslation(const zeus::CVector3f& trans, CStateManager& stateMgr);
    void SetGlobalOrientation(const zeus::CTransform& xf, CStateManager& stateMgr);
    void SetGlobalTranslation(const zeus::CVector3f& trans, CStateManager& stateMgr);
    void SetGlobalScale(const zeus::CVector3f& scale);
    void SetParticleEmission(bool, CStateManager& stateMgr);
    bool IsSystemDeletable() const;
    rstl::optional_object<zeus::CAABox> GetBounds() const;
    bool HasActiveParticles() const;
    void DestroyParticles();
    bool HasLight() const;
    TUniqueId GetLightId() const;
    void DeleteLight(CStateManager& mgr);
    void SetModulationColor(const zeus::CColor& color);
    const std::shared_ptr<CParticleGen> GetParticleSystem() const { return x84_system; }
};
}

#endif // __URDE_CPARTICLEGENINFO_HPP__