#pragma once

#include "RetroTypes.hpp"
#include "CHudInterface.hpp"
#include "zeus/CTransform.hpp"

namespace urde
{
class CStateManager;
class CGuiFrame;
class CGuiWidget;
class CGuiTextPane;
class CAuiEnergyBarT01;
class CGuiModel;

class CHudMissileInterface
{
    enum class EInventoryStatus
    {
        Normal,
        Warning,
        Depleted
    };

    EHudType x0_hudType;
    int x4_missileCapacity;
    int x8_numMissles;
    float xc_arrowTimer = 0.f;
    zeus::CTransform x10_missleIconXf;
    float x40_missileWarningAlpha = 0.f;
    EInventoryStatus x44_latestStatus = EInventoryStatus::Normal;
    float x48_missileWarningPulse = 0.f;
    float x4c_chargeBeamFactor;
    float x50_missileIconAltDeplete = 0.f;
    float x54_missileIconIncrement = 0.f;
    union
    {
        struct
        {
            bool x58_24_missilesActive : 1;
            bool x58_25_visibleDebug : 1;
            bool x58_26_visibleGame : 1;
            bool x58_27_hasArrows : 1;
            bool x58_28_notXRay : 1;
        };
        u16 _dummy = 0;
    };
    CGuiWidget* x5c_basewidget_missileicon;
    CGuiTextPane* x60_textpane_missiledigits;
    CAuiEnergyBarT01* x64_energybart01_missilebar;
    CGuiTextPane* x68_textpane_missilewarning;
    CGuiModel* x6c_model_missilearrowup;
    CGuiModel* x70_model_missilearrowdown;
    CGuiWidget* x74_basewidget_missileicon;
    void UpdateVisibility(const CStateManager& mgr);
public:
    CHudMissileInterface(CGuiFrame& selHud, int missileCapacity, int numMissiles,
                         float chargeFactor, bool missilesActive, EHudType hudType,
                         const CStateManager& mgr);
    void Update(float dt, const CStateManager& mgr);
    void SetIsVisibleGame(bool v, const CStateManager& mgr);
    void SetIsVisibleDebug(bool v, const CStateManager& mgr);
    void SetIsMissilesActive(bool active);
    void SetChargeBeamFactor(float t);
    void SetNumMissiles(int numMissiles, const CStateManager& mgr);
    void SetMissileCapacity(int missileCapacity);
    EInventoryStatus GetMissileInventoryStatus();
    static std::pair<zeus::CVector3f, zeus::CVector3f> CombatMissileBarCoordFunc(float t);
    static std::pair<zeus::CVector3f, zeus::CVector3f> XRayMissileBarCoordFunc(float t);
    static std::pair<zeus::CVector3f, zeus::CVector3f> ThermalMissileBarCoordFunc(float t);
};

}