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, 
    Shotgun,   
    HEGrenade, 
    Molotov,   
    Sticky,    
    None,      
};
 
inline constexpr std::size_t kRenderableWeaponTypeCount = static_cast<std::size_t>(WeaponType::Shotgun) + 1;

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

struct GunInstance
{
    WeaponType type = WeaponType::None;
    int totalAmmo = 0;
    int currentMagAmmo = 0;
    float fireCooldown = 0.f;
    float chargeTime = 0.f;              
    bool isReloading = false;            
    float reloadTime = 0.f;              
    float recoilHeat = 0.f;              
    float recoilIdleTime = 0.f;          
    bool firedSinceTriggerPress = false; 
};

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);
}
 
inline GunInstance* findSlotWithType(WeaponState& w, WeaponType type)
{
    for (auto& slot : w.slots) {
        if (slot.type == type) {
            return &slot;
        }
    }
    return nullptr;
}