WeaponState.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <array>
#include <cstddef>
#include <cstdint>

enum class WeaponType : uint8_t
{
    Rifle,     
    Rocket,    
    RailGun,   
    EnergyGun, 
    HEGrenade, 
    Molotov,   
    Impulse,   
};

enum class WeaponSlot : uint8_t
{
    PRIMARY,
    SECONDARY,
    Count, 
};

struct GunInstance
{
    WeaponType type = WeaponType::Rifle;
    int totalAmmo = 0;
    int currentMagAmmo = 0;
    float fireCooldown = 0.f;
    float chargeTime = 0.f; 
};

struct WeaponState
{
    std::array<GunInstance, static_cast<std::size_t>(WeaponSlot::Count)> slots{};
    WeaponSlot current = WeaponSlot::PRIMARY; 
};

inline GunInstance& getSlot(WeaponState& w, WeaponSlot s)
{
    return w.slots[static_cast<std::size_t>(s)];
}
inline const GunInstance& getSlot(const WeaponState& w, WeaponSlot s)
{
    return w.slots[static_cast<std::size_t>(s)];
}

inline GunInstance& getEquippedGun(WeaponState& w)
{
    return getSlot(w, w.current);
}
inline const GunInstance& getEquippedGun(const WeaponState& w)
{
    return getSlot(w, w.current);
}