#pragma once

#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 { return x0_sources; }
    void AddEnergyDrainSource(TUniqueId, float);
    void RemoveEnergyDrainSource(TUniqueId id);
    float GetEnergyDrainIntensity() const;
    float GetEnergyDrainTime() const { return x10_energyDrainTime; }
    void ProcessEnergyDrain(const CStateManager&, float);
};
}