metaforce/Runtime/Weapon/CWeaponMgr.hpp

#ifndef __URDE_CWEAPONMGR_HPP__
#define __URDE_CWEAPONMGR_HPP__

#include "RetroTypes.hpp"

namespace urde
{

enum class EWeaponType
{
    None = -1,
    Power = 0,
    Ice,
    Wave,
    Plasma,
    Bomb,
    PowerBomb,
    Missile,
    BoostBall,
    Phazon,
    AI,
    PoisonWater,
    Lava,
    Hot,
    Unused1,
    Unused2
};

class CWeaponMgr
{
    std::map<TUniqueId, rstl::reserved_vector<s32, 10>> x0_weapons;
public:
    void Add(TUniqueId, EWeaponType);
    void Remove(TUniqueId);
    void IncrCount(TUniqueId, EWeaponType);
    void DecrCount(TUniqueId, EWeaponType);
    s32 GetNumActive(TUniqueId, EWeaponType) const;
    s32 GetIndex(TUniqueId) const;
};

}

#endif // __URDE_CWEAPONMGR_HPP__