#ifndef __URDE_CHUDENERGYINTERFACE_HPP__
#define __URDE_CHUDENERGYINTERFACE_HPP__

#include "RetroTypes.hpp"

namespace urde
{
class CGuiFrame;
class CGuiWidget;
class CGuiTextPane;
class CAuiMeter;
class CAuiEnergyBarT01;

class CHudEnergyInterface
{
public:
    enum class EBarType
    {
        Combat,
        Scan,
        XRay,
        Thermal,
        Ball
    };
private:
    EBarType x0_barType;
    float x4_energyLowFader = 0.f;
    float x8_flashMag = 0.f;
    float xc_tankEnergy;
    int x10_totalEnergyTanks;
    int x14_numTanksFilled;
    float x18_cachedBarEnergy = 0.f;
    union
    {
        struct
        {
            bool x1c_24_ : 1;
            bool x1c_25_ : 1;
            bool x1c_26_barDirty : 1;
            bool x1c_27_energyLow : 1;
        };
        u16 _dummy = 0;
    };
    CGuiTextPane* x20_textpane_energydigits;
    CAuiMeter* x24_meter_energytanks;
    CGuiTextPane* x28_textpane_energywarning;
    CAuiEnergyBarT01* x2c_energybart01_energybar;
public:
    CHudEnergyInterface(CGuiFrame& selHud, float tankEnergy, int totalEnergyTanks,
                        int numTanksFilled, bool energyLow, EBarType barType);
    void Update(float dt, float energyLowPulse);
    void SetEnergyLow(bool energyLow);
    void SetFlashMagnitude(float mag);
    void SetNumFilledEnergyTanks(int numTanksFilled);
    void SetNumTotalEnergyTanks(int totalEnergyTanks);
    void SetCurrEnergy(float tankEnergy, bool wrapped);
    static std::pair<zeus::CVector3f, zeus::CVector3f> CombatEnergyCoordFunc(float t);
    static std::pair<zeus::CVector3f, zeus::CVector3f> XRayEnergyCoordFunc(float t);
    static std::pair<zeus::CVector3f, zeus::CVector3f> ThermalEnergyCoordFunc(float t);
    static std::pair<zeus::CVector3f, zeus::CVector3f> BallEnergyCoordFunc(float t);
};

}

#endif // __URDE_CHUDENERGYINTERFACE_HPP__