entity_container.h

#pragma once
#include "gaia/config/config.h"
 
#include <cstdint>
#include <type_traits>
 
#include "gaia/cnt/ilist.h"
#include "gaia/cnt/map.h"
#include "gaia/cnt/paged_storage.h"
#include "gaia/ecs/api.h"
#include "gaia/ecs/id.h"
#include "gaia/mem/smallblock_allocator.h"
 
namespace gaia {
  namespace ecs {
    struct EntityContainer;
  }
 
  namespace cnt {
    template <>
    struct to_page_storage_id<ecs::EntityContainer> {
      static page_storage_id get(const ecs::EntityContainer& item) noexcept;
    };
  } // namespace cnt
 
  namespace ecs {
    class Chunk;
    class Archetype;
    class World;
#if GAIA_USE_WEAK_ENTITY
    struct WeakEntityTracker;
#endif
 
    struct EntityContainerCtx {
      bool isEntity;
      bool isPair;
      EntityKind kind;
    };
 
    using EntityContainerFlagsType = uint16_t;
    enum EntityContainerFlags : EntityContainerFlagsType {
      OnDelete_Remove = 1 << 0,
      OnDelete_Delete = 1 << 1,
      OnDelete_Error = 1 << 2,
      OnDeleteTarget_Remove = 1 << 3,
      OnDeleteTarget_Delete = 1 << 4,
      OnDeleteTarget_Error = 1 << 5,
      HasAcyclic = 1 << 6,
      HasCantCombine = 1 << 7,
      IsExclusive = 1 << 8,
      HasAliasOf = 1 << 9,
      IsSingleton = 1 << 10,
      DeleteRequested = 1 << 11,
      RefDecreased = 1 << 12, // GAIA_USE_SAFE_ENTITY
      Load = 1 << 13, // EntityContainer is being loaded from a file
      IsObserved = 1 << 14, // At least one observer is registered for this term.
      IsDontFragment = 1 << 15,
    };
 
    struct EntityContainer {
      uint32_t idx;
 
      // Bits in this section need to be 1:1 with Entity internal data
 
      struct EntityData {
        uint32_t gen : 28;
        uint32_t ent : 1;
        uint32_t pair : 1;
        uint32_t kind : 1;
        uint32_t dis : 1;
      };
 
      union {
        EntityData data;
        uint32_t dataRaw;
      };
 
 
      uint16_t row;
      uint16_t flags = 0;
 
#if GAIA_USE_SAFE_ENTITY
      uint32_t refCnt = 1;
#else
      uint32_t unused = 0;
#endif
 
#if GAIA_USE_WEAK_ENTITY
      WeakEntityTracker* pWeakTracker = nullptr;
#endif
 
      Archetype* pArchetype = nullptr;
      Chunk* pChunk = nullptr;
      const Entity* pEntity = nullptr;
      // uint8_t depthDependsOn = 0;
 
      EntityContainer() = default;
 
      GAIA_NODISCARD static EntityContainer create(uint32_t index, uint32_t generation, void* pCtx) {
        auto* ctx = (EntityContainerCtx*)pCtx;
 
        EntityContainer ec{};
        ec.idx = index;
        ec.data.gen = generation;
        ec.data.ent = (uint32_t)ctx->isEntity;
        ec.data.pair = (uint32_t)ctx->isPair;
        ec.data.kind = (uint32_t)ctx->kind;
        return ec;
      }
 
      GAIA_NODISCARD static Entity handle(const EntityContainer& ec) {
        return Entity(ec.idx, ec.data.gen, (bool)ec.data.ent, (bool)ec.data.pair, (EntityKind)ec.data.kind);
      }
 
      void req_del() {
        flags |= DeleteRequested;
      }
    };
 
    struct PairRecords {
    private:
      cnt::map<EntityLookupKey, EntityContainer> m_records;
 
    public:
      void clear() {
        m_records = {};
      }
 
      GAIA_NODISCARD bool contains(Entity entity) const {
        GAIA_ASSERT(entity.pair());
        return m_records.contains(EntityLookupKey(entity));
      }
 
      GAIA_NODISCARD EntityContainer* find(Entity entity) {
        GAIA_ASSERT(entity.pair());
 
        const auto it = m_records.find(EntityLookupKey(entity));
        if (it == m_records.end())
          return nullptr;
 
        return &it->second;
      }
 
      GAIA_NODISCARD const EntityContainer* find(Entity entity) const {
        GAIA_ASSERT(entity.pair());
 
        const auto it = m_records.find(EntityLookupKey(entity));
        if (it == m_records.end())
          return nullptr;
 
        return &it->second;
      }
 
      void add(Entity entity, EntityContainer&& ec) {
        GAIA_ASSERT(entity.pair());
        m_records.emplace(EntityLookupKey(entity), GAIA_MOV(ec));
      }
 
      GAIA_NODISCARD bool try_add(Entity entity, EntityContainer&& ec) {
        GAIA_ASSERT(entity.pair());
 
        if (contains(entity))
          return false;
 
        add(entity, GAIA_MOV(ec));
        return true;
      }
 
      void remove(Entity entity) {
        GAIA_ASSERT(entity.pair());
        m_records.erase(EntityLookupKey(entity));
      }
 
      GAIA_NODISCARD bool remove(Entity entity, EntityContainer& ec) {
        GAIA_ASSERT(entity.pair());
 
        const auto it = m_records.find(EntityLookupKey(entity));
        if (it == m_records.end())
          return false;
 
        ec = GAIA_MOV(it->second);
        m_records.erase(it);
        return true;
      }
 
      GAIA_NODISCARD auto begin() {
        return m_records.begin();
      }
 
      GAIA_NODISCARD auto begin() const {
        return m_records.begin();
      }
 
      GAIA_NODISCARD auto end() {
        return m_records.end();
      }
 
      GAIA_NODISCARD auto end() const {
        return m_records.end();
      }
    };
 
    struct EntityContainers {
    private:
      PairRecords m_pairRecords;
 
    public:
      cnt::paged_ilist<EntityContainer, Entity> entities;
 
      void pair_records_clear() {
        m_pairRecords.clear();
      }
 
      GAIA_NODISCARD bool pair_record_contains(Entity entity) const {
        return m_pairRecords.contains(entity);
      }
 
      GAIA_NODISCARD EntityContainer* pair_record_find(Entity entity) {
        return m_pairRecords.find(entity);
      }
 
      GAIA_NODISCARD const EntityContainer* pair_record_find(Entity entity) const {
        return m_pairRecords.find(entity);
      }
 
      void pair_record_add(Entity entity, EntityContainer&& ec) {
        m_pairRecords.add(entity, GAIA_MOV(ec));
      }
 
      GAIA_NODISCARD bool pair_record_try_add(Entity entity, EntityContainer&& ec) {
        return m_pairRecords.try_add(entity, GAIA_MOV(ec));
      }
 
      void pair_record_remove(Entity entity) {
        m_pairRecords.remove(entity);
      }
 
      GAIA_NODISCARD bool pair_record_remove(Entity entity, EntityContainer& ec) {
        return m_pairRecords.remove(entity, ec);
      }
 
      GAIA_NODISCARD auto pair_record_begin() {
        return m_pairRecords.begin();
      }
 
      GAIA_NODISCARD auto pair_record_begin() const {
        return m_pairRecords.begin();
      }
 
      GAIA_NODISCARD auto pair_record_end() {
        return m_pairRecords.end();
      }
 
      GAIA_NODISCARD auto pair_record_end() const {
        return m_pairRecords.end();
      }
 
      EntityContainer& operator[](Entity entity) {
        if (!entity.pair())
          return entities[entity.id()];
 
        auto* pPair = m_pairRecords.find(entity);
        GAIA_ASSERT(pPair != nullptr);
        return *pPair;
      }
 
      const EntityContainer& operator[](Entity entity) const {
        if (!entity.pair())
          return entities[entity.id()];
 
        const auto* pPair = m_pairRecords.find(entity);
        GAIA_ASSERT(pPair != nullptr);
        return *pPair;
      }
    };
 
  } // namespace ecs
 
  namespace cnt {
    template <>
    struct ilist_handle_traits<ecs::Entity> {
      static ecs::Entity make(uint32_t id, uint32_t gen, const ecs::Entity& prev) {
        return ecs::Entity((ecs::EntityId)id, gen, prev.entity(), prev.pair(), prev.kind());
      }
    };
  } // namespace cnt
 
  namespace ecs {
 
#if GAIA_USE_SAFE_ENTITY
    class SafeEntity {
      World* m_w = nullptr;
      Entity m_entity;
 
    public:
      SafeEntity() = default;
      SafeEntity(World& w, Entity entity): m_w(&w), m_entity(entity) {
        auto& ec = fetch_mut(w, entity);
        ++ec.refCnt;
      }
 
      ~SafeEntity() {
        // EntityContainer can be null only from moved-from SharedEntities.
        // This is not a common occurrence.
        if GAIA_UNLIKELY (m_w == nullptr)
          return;
 
        auto& ec = fetch_mut(*m_w, m_entity);
        GAIA_ASSERT(ec.refCnt > 0);
        --ec.refCnt;
        if (ec.refCnt == 0)
          del(*m_w, m_entity);
      }
 
      SafeEntity(const SafeEntity& other): m_w(other.m_w), m_entity(other.m_entity) {
        auto& ec = fetch_mut(*m_w, m_entity);
        ++ec.refCnt;
      }
      SafeEntity& operator=(const SafeEntity& other) {
        GAIA_ASSERT(core::addressof(other) != this);
 
        m_w = other.m_w;
        m_entity = other.m_entity;
 
        auto& ec = fetch_mut(*m_w, m_entity);
        ++ec.refCnt;
        return *this;
      }
 
      SafeEntity(SafeEntity&& other) noexcept: m_w(other.m_w), m_entity(other.m_entity) {
        other.m_w = nullptr;
        other.m_entity = EntityBad;
      }
      SafeEntity& operator=(SafeEntity&& other) noexcept {
        GAIA_ASSERT(core::addressof(other) != this);
 
        m_w = other.m_w;
        m_entity = other.m_entity;
 
        other.m_w = nullptr;
        other.m_entity = EntityBad;
        return *this;
      }
 
      template <typename Serializer>
      void save(Serializer& s) const {
        s.save(m_entity);
      }
      template <typename Serializer>
      void load(Serializer& s) {
        s.load(m_entity);
      }
 
      GAIA_NODISCARD Entity entity() const noexcept {
        return m_entity;
      }
      GAIA_NODISCARD operator Entity() const noexcept {
        return m_entity;
      }
 
      bool operator==(const SafeEntity& other) const noexcept {
        return m_entity == other.entity();
      }
    };
 
    inline bool operator==(const SafeEntity& e1, Entity e2) noexcept {
      return e1.entity() == e2;
    }
    inline bool operator==(Entity e1, const SafeEntity& e2) noexcept {
      return e1 == e2.entity();
    }
 
    inline bool operator!=(const SafeEntity& e1, Entity e2) noexcept {
      return e1.entity() != e2;
    }
    inline bool operator!=(Entity e1, const SafeEntity& e2) noexcept {
      return e1 != e2.entity();
    }
#endif
 
#if GAIA_USE_WEAK_ENTITY
    class WeakEntity;
 
    struct WeakEntityTracker {
      GAIA_USE_SMALLBLOCK(WeakEntityTracker)
 
      WeakEntityTracker* next = nullptr;
      WeakEntityTracker* prev = nullptr;
      WeakEntity* pWeakEntity = nullptr;
    };
 
    class WeakEntity {
      friend class World;
      friend struct WeakEntityTracker;
 
      World* m_w = nullptr;
      WeakEntityTracker* m_pTracker = nullptr;
      Entity m_entity;
 
    public:
      WeakEntity() = default;
      WeakEntity(World& w, Entity entity): m_w(&w), m_pTracker(new WeakEntityTracker()), m_entity(entity) {
        set_tracker();
      }
      ~WeakEntity() {
        del_tracker();
      }
 
      WeakEntity(const WeakEntity& other): m_w(other.m_w), m_entity(other.m_entity) {
        if (other.m_pTracker == nullptr)
          return;
 
        m_pTracker = new WeakEntityTracker();
        set_tracker();
      }
      WeakEntity& operator=(const WeakEntity& other) {
        GAIA_ASSERT(core::addressof(other) != this);
 
        del_tracker();
 
        m_w = other.m_w;
        m_entity = other.m_entity;
 
        if (other.m_pTracker != nullptr) {
          m_pTracker = new WeakEntityTracker();
          set_tracker();
        }
 
        return *this;
      }
 
      WeakEntity(WeakEntity&& other) noexcept: m_w(other.m_w), m_pTracker(other.m_pTracker), m_entity(other.m_entity) {
        other.m_w = nullptr;
        if (other.m_pTracker != nullptr)
          other.m_pTracker->pWeakEntity = this;
        other.m_pTracker = nullptr;
        other.m_entity = EntityBad;
      }
      WeakEntity& operator=(WeakEntity&& other) noexcept {
        GAIA_ASSERT(core::addressof(other) != this);
 
        del_tracker();
 
        m_w = other.m_w;
        m_pTracker = other.m_pTracker;
        m_entity = other.m_entity;
 
        other.m_w = nullptr;
        if (other.m_pTracker != nullptr)
          other.m_pTracker->pWeakEntity = this;
        other.m_pTracker = nullptr;
        other.m_entity = EntityBad;
        return *this;
      }
 
      void set_tracker() {
        GAIA_ASSERT(m_pTracker != nullptr);
        GAIA_ASSERT(m_w != nullptr);
        GAIA_ASSERT(m_entity != EntityBad);
 
        m_pTracker->pWeakEntity = this;
        m_pTracker->prev = nullptr;
 
        auto& ec = fetch_mut(*m_w, m_entity);
        m_pTracker->next = ec.pWeakTracker;
        if (ec.pWeakTracker != nullptr)
          ec.pWeakTracker->prev = m_pTracker;
        ec.pWeakTracker = m_pTracker;
      }
 
      void del_tracker() {
        if (m_pTracker == nullptr)
          return;
 
        if (m_pTracker->next != nullptr)
          m_pTracker->next->prev = m_pTracker->prev;
        if (m_pTracker->prev != nullptr)
          m_pTracker->prev->next = m_pTracker->next;
 
        if (m_w != nullptr && valid(*m_w, m_entity)) {
          auto& ec = fetch_mut(*m_w, m_entity);
          if (ec.pWeakTracker == m_pTracker)
            ec.pWeakTracker = m_pTracker->next;
        }
 
        delete m_pTracker;
        m_pTracker = nullptr;
      }
 
      template <typename Serializer>
      void save(Serializer& s) const {
        s.save(m_entity.val);
      }
      template <typename Serializer>
      void load(Serializer& s) {
        del_tracker();
        Identifier id{};
        s.load(id);
        m_entity = Entity(id);
        if (m_w != nullptr && m_entity != EntityBad) {
          m_pTracker = new WeakEntityTracker();
          set_tracker();
        }
      }
 
      GAIA_NODISCARD Entity entity() const noexcept {
        return m_entity;
      }
      GAIA_NODISCARD operator Entity() const noexcept {
        return entity();
      }
 
      bool operator==(const WeakEntity& other) const noexcept {
        return m_entity == other.entity();
      }
    };
 
    inline bool operator==(const WeakEntity& e1, Entity e2) noexcept {
      return e1.entity() == e2;
    }
    inline bool operator==(Entity e1, const WeakEntity& e2) noexcept {
      return e1 == e2.entity();
    }
 
    inline bool operator!=(const WeakEntity& e1, Entity e2) noexcept {
      return e1.entity() != e2;
    }
    inline bool operator!=(Entity e1, const WeakEntity& e2) noexcept {
      return e1 != e2.entity();
    }
#endif
  } // namespace ecs
 
  namespace cnt {
    inline page_storage_id to_page_storage_id<ecs::EntityContainer>::get(const ecs::EntityContainer& item) noexcept {
      return item.idx;
    }
  } // namespace cnt
} // namespace gaia