#ifndef __URDE_CHUDENERGYINTERFACE_HPP__
#define __URDE_CHUDENERGYINTERFACE_HPP__

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

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

class CHudEnergyInterface
{
    EHudType x0_hudType;
    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, EHudType hudType);
    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__