#ifndef __URDE_CPLAYERENERGYDRAIN_HPP__
#define __URDE_CPLAYERENERGYDRAIN_HPP__

#include "RetroTypes.hpp"
#include "CEnergyDrainSource.hpp"

namespace urde
{
class CStateManager;
class CPlayerEnergyDrain
{
    std::vector<CEnergyDrainSource> x0_sources;
    float x10_energyDrainTime;

public:
    CPlayerEnergyDrain(u32);
    const std::vector<CEnergyDrainSource>& GetEnergyDrainSources() const;
    void AddEnergyDrainSource(TUniqueId, float);
    void RemoveEnergyDrainSource(TUniqueId id);
    float GetEnergyDrainIntensity() const;
    float GetEnergyDrainTime() const;
    void ProcessEnergyDrain(const CStateManager&, float);
};
}
#endif // __URDE_CPLAYERENERGYDRAIN_HPP__