MapStorage.hpp

Go to the documentation of this file.

 
 
#ifndef ECSTASY_STORAGE_MAPSTORAGE_HPP_
#define ECSTASY_STORAGE_MAPSTORAGE_HPP_
 
#include <unordered_map>
 
#include "AStorage.hpp"
 
namespace ecstasy
{
    template <typename C>
    class MapStorage : public AStorage<C> {
      public:
        using Component = typename AStorage<C>::Component;
        MapStorage() = default;
 
        MapStorage(const MapStorage &other) = delete;
 
        template <typename... Args>
        Component &emplace(Entity::Index index, Args &&...args)
        {
            _mask.resize(std::max(_mask.size(), index + 1));
            _mask[index] = true;
            return _components.emplace(std::make_pair(index, Component(std::forward<Args>(args)...))).first->second;
        }
 
        Component &insert(Entity::Index index, Component &&c) override final
        {
            if constexpr (!std::movable<Component>)
                throw std::runtime_error("MapStorage: Component is not movable");
            else {
                _mask.resize(std::max(_mask.size(), index + 1));
                _mask[index] = true;
                return _components.emplace(std::make_pair(index, std::move(c))).first->second;
            }
        }
 
        bool erase(Entity::Index index) override final
        {
            auto it = _components.find(index);
 
            if (it != _components.end()) [[likely]] {
                _components.erase(index);
                _mask[index] = false;
                return true;
            }
            return false;
        }
 
        [[nodiscard]] Component &operator[](Entity::Index index) override final
        {
            return _components.at(index);
        }
 
        [[nodiscard]] const Component &operator[](Entity::Index index) const override final
        {
            return _components.at(index);
        }
 
        [[nodiscard]] size_t size() const noexcept
        {
            return _components.size();
        }
 
        [[nodiscard]] constexpr const util::BitSet &getMask() const noexcept override final
        {
            return _mask;
        }
 
      private:
        std::unordered_map<Entity::Index, Component> _components;
        util::BitSet _mask;
    };
} // namespace ecstasy
 
#endif /* !ECSTASY_STORAGE_MAPSTORAGE_HPP_ */