WeaponSystem.hpp

Go to the documentation of this file.

/*
** EPITECH PROJECT, 2025
** RTYPE
** File description:
** WeaponSystem
*/
 
#pragma once
 
#include <cstdint>
#include <functional>
#include <memory>
#include "../../Components/Buff.hpp"
#include "../../Components/Projectile.hpp"
#include "../../Components/Transform.hpp"
#include "../../Components/Velocity.hpp"
#include "../../Components/Weapon.hpp"
#include "../ISystem.hpp"
 
namespace ecs {
    class WeaponSystem : public ISystem {
       public:
        using ProjectileCreatedCallback =
            std::function<void(uint32_t, uint32_t, float, float, float, float, float, int, bool)>;
 
        WeaponSystem() = default;
 
        explicit WeaponSystem(ProjectileCreatedCallback callback) : _projectileCreatedCallback(callback) {}
 
        ~WeaponSystem() override = default;
 
        void setProjectileCreatedCallback(ProjectileCreatedCallback callback) {
            _projectileCreatedCallback = callback;
        }
 
        void update(Registry &registry, float deltaTime) override;
 
        std::uint32_t fireWeapon(Registry &registry, std::uint32_t ownerId, bool isFriendly);
 
        std::uint32_t fireChargedShot(Registry &registry, std::uint32_t ownerId, float chargeLevel,
                                      bool isFriendly);
 
        ComponentMask getComponentMask() const override;
 
       private:
        Velocity calculateProjectileVelocity(float baseSpeed, bool isFriendly);
 
        Transform calculateProjectileTransform(Registry &registry, std::uint32_t ownerId, bool isFriendly);
 
        std::uint32_t createProjectile(Registry &registry, std::uint32_t ownerId, const Transform &transform,
                                       const Velocity &velocity, float damage, bool isFriendly,
                                       bool isCharged);
 
        void fireMultipleShots(Registry &registry, std::uint32_t ownerId, float damage, float speed,
                               bool isFriendly, bool isCharged, int shotCount);
 
        ProjectileCreatedCallback _projectileCreatedCallback;
    };
}  // namespace ecs