Registry.hpp

Go to the documentation of this file.

/*
** EPITECH PROJECT, 2025
** rtype
** File description:
** Registry
*/
 
#pragma once
 
#include <any>
#include <bitset>
#include <mutex>
#include <queue>
#include <shared_mutex>
#include <unordered_map>
#include <vector>
 
#include "Components/IComponent.hpp"
 
#define N_MAX_COMPONENTS 64
namespace ecs {
 
    typedef std::bitset<N_MAX_COMPONENTS> Signature;
 
    typedef uint32_t Address;
 
    class Registry {
 
       private:
        mutable std::shared_mutex _mutex;
        std::unordered_map<Address, Signature> _signatures;
 
        Address _nextAddress;
 
        Address _generateAddress();
 
        Signature _registerComponent(ComponentType componentType);
 
        std::priority_queue<Address, std::vector<Address>, std::greater<Address>> _freeAddresses = {};
        std::unordered_map<ComponentType, Signature> _componentMap = {};
        std::unordered_map<ComponentType, std::unordered_map<Address, std::any>> _componentStorage = {};
 
       public:
        Registry();
 
        ~Registry();
 
        Address newEntity();
 
        template <typename T>
        void setComponent(Address address, const T &component);
 
        template <typename T>
        T &getComponent(Address address);
 
        template <typename T>
        bool hasComponent(Address address);
 
        template <typename T>
        void removeComponent(Address address);
 
        template <typename T>
        void addEntityProp(Address address);
 
        void destroyEntity(Address address);
 
        Signature getSignature(Address address);
 
        template <typename... Components>
        std::vector<Address> view();
 
        std::vector<Address> getEntitiesWithMask(Signature requiredMask);
    };
}  // namespace ecs
 
#include "Registry.tpp"