query_cache.h

#pragma once
#include "gaia/config/config.h"
 
#include "gaia/cnt/darray.h"
#include "gaia/cnt/map.h"
#include "gaia/cnt/paged_storage.h"
#include "gaia/core/utility.h"
#include "gaia/ecs/component.h"
#include "gaia/ecs/id.h"
#include "gaia/ecs/query_common.h"
#include "gaia/ecs/query_info.h"
 
namespace gaia {
  namespace cnt {
    template <>
    struct to_page_storage_id<ecs::QueryInfo> {
      static page_storage_id get(const ecs::QueryInfo& item) noexcept {
        return item.idx;
      }
    };
  } // namespace cnt
 
  namespace ecs {
    class QueryLookupKey {
      QueryLookupHash m_hash;
      const QueryCtx* m_pCtx;
 
    public:
      static constexpr bool IsDirectHashKey = true;
 
      QueryLookupKey(): m_hash({0}), m_pCtx(nullptr) {}
      explicit QueryLookupKey(QueryLookupHash hash, const QueryCtx* pCtx): m_hash(hash), m_pCtx(pCtx) {}
 
      size_t hash() const {
        return (size_t)m_hash.hash;
      }
 
      bool operator==(const QueryLookupKey& other) const {
        // Hash doesn't match we don't have a match.
        // Hash collisions are expected to be very unlikely so optimize for this case.
        if GAIA_LIKELY (m_hash != other.m_hash)
          return false;
 
        if (m_pCtx == other.m_pCtx)
          return true;
 
        const auto lhsReal = m_pCtx->q.handle.id() != QueryIdBad;
        const auto rhsReal = other.m_pCtx->q.handle.id() != QueryIdBad;
 
        // Two persisted cached-query keys with different stable context pointers cannot represent
        // the same query.
        if (lhsReal && rhsReal)
          return false;
 
        // At least one side is a temporary lookup key. Fall back to structural comparison so cached
        // queries deduplicate correctly regardless of comparison direction.
        return QueryCtx::equals_no_handle_assumption(*m_pCtx, *other.m_pCtx);
      }
    };
 
    class QueryCache {
    public:
      enum class ChangeKind : uint8_t {
        // Query membership may have changed due to structural world changes.
        Structural,
        // Only dynamic/source-driven results may have changed.
        DynamicResult,
        // Full query cache invalidation.
        All,
      };
 
    private:
      QueryInfo& register_query_info(QueryHandle handle, QueryInfo& info) {
        // Add the entity->query pair
        add_entity_to_query_pairs(info.ctx().data.ids_view(), handle);
        add_rel_to_query_pairs(info.ctx(), handle);
        add_sort_to_query_pairs(info.ctx(), handle);
        add_sorted_query(info.ctx(), handle);
        add_create_to_query_pairs(info.ctx(), handle);
 
        return info;
      }
 
      struct CreateQueryCandidate {
        QueryHandle handle;
        Entity matchedSelector = EntityBad;
      };
 
      enum class CreateSelectorKind : uint8_t {
        Other,
        ExactPair,
        RelWildcardPair,
        TgtWildcardPair,
        AnyPairWildcard,
        Count
      };
 
      struct TrackedArchetypes {
        cnt::darray<const Archetype*> archetypes;
        uint32_t syncedRevision = 0;
      };
 
      cnt::map<QueryLookupKey, QueryInfo*> m_pCache;
      cnt::paged_ilist<QueryInfo, QueryHandle> m_queryArr;
      uint64_t m_epoch = 0;
 
      cnt::map<EntityLookupKey, cnt::darray<QueryHandle>> m_entityToQuery;
      cnt::map<EntityLookupKey, cnt::darray<QueryHandle>> m_relationToQuery;
      cnt::map<EntityLookupKey, cnt::darray<QueryHandle>> m_sortEntityToQuery;
      cnt::darray<QueryHandle> m_sortedQueries;
      cnt::map<EntityLookupKey, cnt::darray<QueryHandle>> m_entityToCreateQuery;
      cnt::map<ArchetypeIdLookupKey, cnt::darray<QueryHandle>> m_archetypeToQuery;
      cnt::map<QueryHandleLookupKey, TrackedArchetypes> m_queryToArchetype;
      cnt::darray<CreateQueryCandidate> m_createQueryHandleScratch;
      cnt::darray<uint32_t> m_createQueryHandleStampById;
      uint32_t m_createQueryHandleStamp = 1;
      uint32_t m_createQuerySelectorCnt[(size_t)CreateSelectorKind::Count] = {};
 
    public:
      QueryCache() {
        m_queryArr.reserve(256);
      }
 
      ~QueryCache() = default;
 
      QueryCache(QueryCache&&) = delete;
      QueryCache(const QueryCache&) = delete;
      QueryCache& operator=(QueryCache&&) = delete;
      QueryCache& operator=(const QueryCache&) = delete;
 
      GAIA_NODISCARD bool valid(QueryHandle handle) const {
        if (handle.id() == QueryIdBad)
          return false;
 
        if (!m_queryArr.has(handle.id()))
          return false;
 
        const auto& h = m_queryArr[handle.id()];
        return h.idx == handle.id() && h.gen == handle.gen();
      }
 
      void clear() {
        ++m_epoch;
        m_pCache.clear();
        m_queryArr.clear();
        m_entityToQuery.clear();
        m_relationToQuery.clear();
        m_sortEntityToQuery.clear();
        m_sortedQueries.clear();
        m_entityToCreateQuery.clear();
        m_archetypeToQuery.clear();
        m_queryToArchetype.clear();
        m_createQueryHandleScratch.clear();
        m_createQueryHandleStampById.clear();
        m_createQueryHandleStamp = 1;
        for (auto& cnt: m_createQuerySelectorCnt)
          cnt = 0;
      }
 
      GAIA_NODISCARD uint64_t epoch() const {
        return m_epoch;
      }
 
      void clear_archetype_tracking() {
        m_archetypeToQuery.clear();
        m_queryToArchetype.clear();
      }
 
      QueryInfo* try_get(QueryHandle handle) {
        if (!valid(handle))
          return nullptr;
 
        auto& info = m_queryArr[handle.id()];
        GAIA_ASSERT(info.idx == handle.id());
        GAIA_ASSERT(info.gen == handle.gen());
        return &info;
      }
 
      const QueryInfo* try_get(QueryHandle handle) const {
        if (!valid(handle))
          return nullptr;
 
        const auto& info = m_queryArr[handle.id()];
        GAIA_ASSERT(info.idx == handle.id());
        GAIA_ASSERT(info.gen == handle.gen());
        return &info;
      }
 
#if GAIA_ECS_TEST_HOOKS
      GAIA_NODISCARD bool verify_archetype_tracking() const {
        for (const auto& pair: m_queryToArchetype) {
          const auto handle = pair.first.handle();
          const auto* pInfo = try_get(handle);
          if (pInfo == nullptr || pInfo->refs() == 0)
            return false;
 
          const auto& tracked = pair.second.archetypes;
          for (uint32_t i = 0; i < tracked.size(); ++i) {
            const auto* pArchetype = tracked[i];
            if (pArchetype == nullptr)
              return false;
 
            for (uint32_t j = i + 1; j < tracked.size(); ++j) {
              if (tracked[j] == pArchetype)
                return false;
            }
 
            if (!archetype_span_contains(pInfo->cache_archetype_view(), pArchetype))
              return false;
 
            const auto archetypeKey = ArchetypeIdLookupKey(pArchetype->id(), pArchetype->id_hash());
            const auto reverseIt = m_archetypeToQuery.find(archetypeKey);
            if (reverseIt == m_archetypeToQuery.end() || !core::has(reverseIt->second, handle))
              return false;
          }
        }
 
        for (const auto& pair: m_archetypeToQuery) {
          const auto& handles = pair.second;
          for (uint32_t i = 0; i < handles.size(); ++i) {
            const auto handle = handles[i];
            const auto* pInfo = try_get(handle);
            if (pInfo == nullptr || pInfo->refs() == 0)
              return false;
 
            for (uint32_t j = i + 1; j < handles.size(); ++j) {
              if (handles[j] == handle)
                return false;
            }
 
            const auto trackedIt = m_queryToArchetype.find(QueryHandleLookupKey(handle));
            if (trackedIt == m_queryToArchetype.end())
              return false;
 
            if (!tracked_archetypes_contain(trackedIt->second.archetypes, pair.first))
              return false;
          }
        }
 
        return true;
      }
 
      GAIA_NODISCARD uint32_t test_query_count() const {
        return (uint32_t)m_queryArr.item_count();
      }
#endif
 
      QueryInfo& get(QueryHandle handle) {
        GAIA_ASSERT(valid(handle));
 
        auto& info = m_queryArr[handle.id()];
        GAIA_ASSERT(info.idx == handle.id());
        GAIA_ASSERT(info.gen == handle.gen());
        return info;
      }
 
      QueryInfo&
      add(QueryCtx&& ctx, //
          const EntityToArchetypeMap& entityToArchetypeMap, //
          std::span<const Archetype*> allArchetypes) {
        GAIA_ASSERT(ctx.hashLookup.hash != 0);
 
        // First check if the query cache record exists
        auto ret = m_pCache.try_emplace(QueryLookupKey(ctx.hashLookup, &ctx), nullptr);
        if (!ret.second) {
          auto* pInfo = ret.first->second;
          GAIA_ASSERT(pInfo != nullptr);
          pInfo->add_ref();
          return *pInfo;
        }
 
        // No record exists, let us create a new one
        QueryInfoCreationCtx creationCtx{};
        creationCtx.pQueryCtx = &ctx;
        creationCtx.pEntityToArchetypeMap = &entityToArchetypeMap;
        creationCtx.allArchetypes = allArchetypes;
        auto handle = m_queryArr.alloc(&creationCtx);
 
        // We are moving the rvalue to "ctx". As a result, the pointer stored in m_pCache.emplace above is no longer
        // going to be valid. Therefore we swap the map key with a one with a valid pointer.
        auto& info = get(handle);
        info.add_ref();
        ret.first->second = &info;
        auto new_p = robin_hood::pair(std::make_pair(QueryLookupKey(ctx.hashLookup, &info.ctx()), &info));
        ret.first->swap(new_p);
 
        return register_query_info(handle, info);
      }
 
      QueryInfo& add_local(
          QueryCtx&& ctx, //
          const EntityToArchetypeMap& entityToArchetypeMap, //
          std::span<const Archetype*> allArchetypes) {
        QueryInfoCreationCtx creationCtx{};
        creationCtx.pQueryCtx = &ctx;
        creationCtx.pEntityToArchetypeMap = &entityToArchetypeMap;
        creationCtx.allArchetypes = allArchetypes;
        auto handle = m_queryArr.alloc(&creationCtx);
 
        auto& info = get(handle);
        info.add_ref();
        return register_query_info(handle, info);
      }
 
      bool del(QueryHandle handle) {
        auto* pInfo = try_get(handle);
        if (pInfo == nullptr)
          return false;
 
        pInfo->dec_ref();
        if (pInfo->refs() != 0)
          return false;
 
        unregister_query_archetypes(handle);
 
        // If this was the last reference to the query, we can safely remove it
        auto it = m_pCache.find(QueryLookupKey(pInfo->ctx().hashLookup, &pInfo->ctx()));
        if (it != m_pCache.end())
          m_pCache.erase(it);
 
        // Remove the entity->query pair
        del_entity_to_query_pairs(pInfo->ctx().data.ids_view(), handle);
        del_rel_to_query_pairs(pInfo->ctx(), handle);
        del_sort_to_query_pairs(pInfo->ctx(), handle);
        del_sorted_query(pInfo->ctx(), handle);
        del_create_to_query_pairs(pInfo->ctx(), handle);
        m_queryArr.free(handle);
 
        return true;
      }
 
      auto begin() {
        return m_queryArr.begin();
      }
 
      auto end() {
        return m_queryArr.end();
      }
 
      GAIA_NODISCARD bool has_relation_query_dependencies() const {
        return !m_relationToQuery.empty();
      }
 
      void invalidate_queries_for_entity(EntityLookupKey entityKey, ChangeKind changeKind) {
        auto it = m_entityToQuery.find(entityKey);
        if (it == m_entityToQuery.end())
          return;
 
        const auto& handles = it->second;
        for (const auto& handle: handles) {
          auto& info = get(handle);
          // World mutations invalidate cached results, but they do not change query shape.
          // Recomputing QueryCtx metadata here only adds overhead to the invalidation path.
          info.invalidate(select_invalidation_kind(info, changeKind));
        }
      }
 
      void invalidate_queries_for_rel(Entity relation, ChangeKind changeKind) {
        auto it = m_relationToQuery.find(EntityLookupKey(relation));
        if (it == m_relationToQuery.end())
          return;
 
        for (const auto handle: it->second) {
          auto& info = get(handle);
          // Relation changes affect dynamic freshness, not the query definition itself.
          info.invalidate(select_invalidation_kind(info, changeKind));
        }
      }
 
      GAIA_NODISCARD bool has_sorted_queries() const {
        return !m_sortedQueries.empty();
      }
 
      GAIA_NODISCARD bool has_sorted_queries_for_entity(Entity entity) const {
        const auto it = m_sortEntityToQuery.find(EntityLookupKey(entity));
        return it != m_sortEntityToQuery.end() && !it->second.empty();
      }
 
      void invalidate_sorted_queries_for_entity(Entity entity) {
        auto it = m_sortEntityToQuery.find(EntityLookupKey(entity));
        if (it == m_sortEntityToQuery.end())
          return;
 
        for (const auto handle: it->second) {
          auto* pInfo = try_get(handle);
          if (pInfo == nullptr || pInfo->refs() == 0)
            continue;
 
          pInfo->invalidate_sort();
        }
      }
 
      void invalidate_sorted_queries() {
        for (const auto handle: m_sortedQueries) {
          auto* pInfo = try_get(handle);
          if (pInfo == nullptr || pInfo->refs() == 0)
            continue;
 
          pInfo->invalidate_sort();
        }
      }
 
      void sync_archetype_cache(QueryInfo& queryInfo) {
        const auto handle = QueryInfo::handle(queryInfo);
        if (!valid(handle))
          return;
 
        const auto archetypes = queryInfo.cache_archetype_view();
        const auto key = QueryHandleLookupKey(handle);
        auto it = m_queryToArchetype.find(key);
        if (it != m_queryToArchetype.end() && it->second.syncedRevision == queryInfo.result_cache_rev())
          return;
 
        unregister_query_archetypes(handle);
 
        if (archetypes.empty())
          return;
 
        auto [trackedIt, inserted] = m_queryToArchetype.try_emplace(key);
        auto& tracked = trackedIt->second.archetypes;
        if (!inserted)
          tracked.clear();
 
        tracked.reserve((uint32_t)archetypes.size());
        for (const auto* pArchetype: archetypes) {
          tracked.push_back(pArchetype);
          add_archetype_query_pair(pArchetype, handle);
        }
        trackedIt->second.syncedRevision = queryInfo.result_cache_rev();
      }
 
      void remove_archetype_from_queries(Archetype* pArchetype) {
        const auto archetypeKey = ArchetypeIdLookupKey(pArchetype->id(), pArchetype->id_hash());
        auto it = m_archetypeToQuery.find(archetypeKey);
        if (it == m_archetypeToQuery.end())
          return;
 
        const auto handles = it->second;
        for (const auto handle: handles) {
          auto* pInfo = try_get(handle);
          if (pInfo != nullptr && pInfo->refs() != 0)
            pInfo->remove(pArchetype);
 
          auto trackedIt = m_queryToArchetype.find(QueryHandleLookupKey(handle));
          if (trackedIt == m_queryToArchetype.end())
            continue;
 
          auto& tracked = trackedIt->second.archetypes;
          core::swap_erase(tracked, core::get_index(tracked, pArchetype));
          if (tracked.empty())
            m_queryToArchetype.erase(trackedIt);
        }
 
        m_archetypeToQuery.erase(it);
      }
 
      void register_archetype_with_queries(const Archetype* pArchetype) {
        if (m_entityToCreateQuery.empty()) {
          (void)pArchetype;
          return;
        }
 
        auto& handles = prepare_create_query_handles();
        const bool needsExactPairSelectors = has_create_selector_kind(CreateSelectorKind::ExactPair);
        const bool needsRelWildcardSelectors = has_create_selector_kind(CreateSelectorKind::RelWildcardPair);
        const bool needsTgtWildcardSelectors = has_create_selector_kind(CreateSelectorKind::TgtWildcardPair);
        const bool needsAnyPairWildcardSelectors = has_create_selector_kind(CreateSelectorKind::AnyPairWildcard);
        bool hasAnyPair = false;
        cnt::darray_ext<Entity, 16> pairWildcardRelations;
        for (const auto entity: pArchetype->ids_view()) {
          if (!entity.pair()) {
            add_create_query_handles(entity, handles);
            continue;
          }
 
          hasAnyPair = true;
          if (needsExactPairSelectors)
            add_create_query_handles(entity, handles);
 
          // Pair ids retain the relation and target ids. That is enough to
          // rebuild wildcard pair lookup keys without touching the world record storage.
          const auto rel = Entity((EntityId)entity.id(), 0, false, false);
          const auto tgt = Entity((EntityId)entity.gen(), 0, false, false);
          if (needsTgtWildcardSelectors)
            add_create_query_handles(Pair(All, tgt), handles);
          if (needsRelWildcardSelectors && !core::has(pairWildcardRelations, rel)) {
            pairWildcardRelations.push_back(rel);
            add_create_query_handles(Pair(rel, All), handles);
          }
        }
 
        if (hasAnyPair && needsAnyPairWildcardSelectors)
          add_create_query_handles(Pair(All, All), handles);
 
        for (const auto& candidate: handles) {
          auto* pInfo = try_get(candidate.handle);
          if (pInfo == nullptr || pInfo->refs() == 0)
            continue;
 
          if (!pInfo->register_archetype(*pArchetype, candidate.matchedSelector, true))
            continue;
 
          register_query_archetype(candidate.handle, pArchetype, pInfo->result_cache_rev());
        }
      }
 
    private:
#if GAIA_ECS_TEST_HOOKS
      GAIA_NODISCARD static bool
      archetype_span_contains(std::span<const Archetype*> archetypes, const Archetype* pArchetype) {
        for (const auto* pCachedArchetype: archetypes) {
          if (pCachedArchetype == pArchetype)
            return true;
        }
        return false;
      }
 
      GAIA_NODISCARD static bool tracked_archetypes_contain(
          const cnt::darray<const Archetype*>& archetypes, const ArchetypeIdLookupKey& archetypeKey) {
        for (const auto* pArchetype: archetypes) {
          if (pArchetype == nullptr)
            return false;
 
          if (ArchetypeIdLookupKey(pArchetype->id(), pArchetype->id_hash()) == archetypeKey)
            return true;
        }
        return false;
      }
#endif
 
      static CreateSelectorKind classify_create_selector(Entity entity) {
        if (!entity.pair())
          return CreateSelectorKind::Other;
        if (is_wildcard(entity.id()))
          return is_wildcard(entity.gen()) ? CreateSelectorKind::AnyPairWildcard : CreateSelectorKind::TgtWildcardPair;
        if (is_wildcard(entity.gen()))
          return CreateSelectorKind::RelWildcardPair;
        return CreateSelectorKind::ExactPair;
      }
 
      GAIA_NODISCARD static constexpr uint32_t selector_kind_idx(CreateSelectorKind kind) {
        return (uint32_t)kind;
      }
 
      GAIA_NODISCARD bool has_create_selector_kind(CreateSelectorKind kind) const {
        return m_createQuerySelectorCnt[selector_kind_idx(kind)] != 0;
      }
 
      void track_create_selector(Entity entity) {
        const auto kind = classify_create_selector(entity);
        ++m_createQuerySelectorCnt[selector_kind_idx(kind)];
      }
 
      void untrack_create_selector(Entity entity) {
        const auto kind = classify_create_selector(entity);
        auto& cnt = m_createQuerySelectorCnt[selector_kind_idx(kind)];
        GAIA_ASSERT(cnt != 0);
        --cnt;
      }
 
      static QueryInfo::InvalidationKind select_invalidation_kind(const QueryInfo& info, ChangeKind changeKind) {
        switch (changeKind) {
          case ChangeKind::DynamicResult:
            return QueryInfo::InvalidationKind::Result;
          case ChangeKind::All:
            return QueryInfo::InvalidationKind::All;
          case ChangeKind::Structural:
            // Structural changes invalidate seed caches for structural queries.
            // Dynamic queries reuse structural compilation state and only need their
            // final result refreshed on the next read.
            return (info.ctx().data.deps.has_dep_flag(QueryCtx::DependencyHasSourceTerms) ||
                    info.ctx().data.deps.has_dep_flag(QueryCtx::DependencyHasVariableTerms))
                       ? QueryInfo::InvalidationKind::Result
                       : QueryInfo::InvalidationKind::Seed;
        }
 
        GAIA_ASSERT(false);
        return QueryInfo::InvalidationKind::All;
      }
 
      void add_entity_query_pair(Entity entity, QueryHandle handle) {
        EntityLookupKey entityKey(entity);
        const auto it = m_entityToQuery.find(entityKey);
        if (it == m_entityToQuery.end()) {
          m_entityToQuery.try_emplace(entityKey, cnt::darray<QueryHandle>{handle});
          return;
        }
 
        auto& handles = it->second;
        if (!core::has(handles, handle))
          handles.push_back(handle);
      }
 
      void del_entity_query_pair(Entity entity, QueryHandle handle) {
        auto it = m_entityToQuery.find(EntityLookupKey(entity));
        if (it == m_entityToQuery.end())
          return;
 
        auto& handles = it->second;
        const auto idx = core::get_index_unsafe(handles, handle);
        core::swap_erase_unsafe(handles, idx);
 
        // Remove the mapping if there are no more matches
        if (handles.empty())
          m_entityToQuery.erase(it);
      }
 
      void add_entity_to_query_pairs(EntitySpan entities, QueryHandle handle) {
        for (auto entity: entities) {
          add_entity_query_pair(entity, handle);
        }
      }
 
      void del_entity_to_query_pairs(EntitySpan entities, QueryHandle handle) {
        for (auto entity: entities) {
          del_entity_query_pair(entity, handle);
        }
      }
 
      void add_create_to_query_pair(Entity entity, QueryHandle handle) {
        EntityLookupKey entityKey(entity);
        const auto it = m_entityToCreateQuery.find(entityKey);
        if (it == m_entityToCreateQuery.end()) {
          m_entityToCreateQuery.try_emplace(entityKey, cnt::darray<QueryHandle>{handle});
          track_create_selector(entity);
          return;
        }
 
        auto& handles = it->second;
        if (!core::has(handles, handle)) {
          handles.push_back(handle);
          track_create_selector(entity);
        }
      }
 
      void add_sort_to_query_pair(Entity entity, QueryHandle handle) {
        auto it = m_sortEntityToQuery.find(EntityLookupKey(entity));
        if (it == m_sortEntityToQuery.end()) {
          m_sortEntityToQuery.try_emplace(EntityLookupKey(entity), cnt::darray<QueryHandle>{handle});
          return;
        }
 
        auto& handles = it->second;
        if (!core::has(handles, handle))
          handles.push_back(handle);
      }
 
      void del_sort_to_query_pair(Entity entity, QueryHandle handle) {
        auto it = m_sortEntityToQuery.find(EntityLookupKey(entity));
        if (it == m_sortEntityToQuery.end())
          return;
 
        auto& handles = it->second;
        core::swap_erase(handles, core::get_index(handles, handle));
        if (handles.empty())
          m_sortEntityToQuery.erase(it);
      }
 
      void add_sort_to_query_pairs(const QueryCtx& ctx, QueryHandle handle) {
        if (ctx.data.sortByFunc == nullptr || ctx.data.sortBy == EntityBad)
          return;
 
        add_sort_to_query_pair(ctx.data.sortBy, handle);
      }
 
      void del_sort_to_query_pairs(const QueryCtx& ctx, QueryHandle handle) {
        if (ctx.data.sortByFunc == nullptr || ctx.data.sortBy == EntityBad)
          return;
 
        del_sort_to_query_pair(ctx.data.sortBy, handle);
      }
 
      void add_sorted_query(const QueryCtx& ctx, QueryHandle handle) {
        if (ctx.data.sortByFunc == nullptr)
          return;
 
        m_sortedQueries.push_back(handle);
      }
 
      void del_sorted_query(const QueryCtx& ctx, QueryHandle handle) {
        if (ctx.data.sortByFunc == nullptr)
          return;
 
        const auto idx = core::get_index(m_sortedQueries, handle);
        GAIA_ASSERT(idx != BadIndex);
        if (idx != BadIndex)
          core::swap_erase(m_sortedQueries, idx);
      }
 
      void del_create_to_query_pair(Entity entity, QueryHandle handle) {
        auto it = m_entityToCreateQuery.find(EntityLookupKey(entity));
        if (it == m_entityToCreateQuery.end())
          return;
 
        auto& handles = it->second;
        core::swap_erase(handles, core::get_index(handles, handle));
        untrack_create_selector(entity);
        if (handles.empty())
          m_entityToCreateQuery.erase(it);
      }
 
      void add_create_to_query_pairs(const QueryCtx& ctx, QueryHandle handle) {
        if (ctx.data.cachePolicy != QueryCtx::CachePolicy::Immediate)
          return;
 
        // Only structural queries with positive selector dependencies are tracked here.
        // Dependency metadata is refreshed together with cache policy classification so
        // create-time propagation can consume it without re-deriving query shape here.
        for (const auto entity: ctx.data.deps.create_selectors_view())
          add_create_to_query_pair(entity, handle);
      }
 
      void del_create_to_query_pairs(const QueryCtx& ctx, QueryHandle handle) {
        if (ctx.data.cachePolicy != QueryCtx::CachePolicy::Immediate)
          return;
 
        for (const auto entity: ctx.data.deps.create_selectors_view())
          del_create_to_query_pair(entity, handle);
      }
 
      void add_create_query_handles(Entity selector, cnt::darray<CreateQueryCandidate>& handles) {
        const auto it = m_entityToCreateQuery.find(EntityLookupKey(selector));
        if (it == m_entityToCreateQuery.end())
          return;
 
        for (const auto handle: it->second) {
          if (mark_create_query_handle(handle))
            handles.push_back(CreateQueryCandidate{handle, selector});
        }
      }
 
      GAIA_NODISCARD cnt::darray<CreateQueryCandidate>& prepare_create_query_handles() {
        m_createQueryHandleScratch.clear();
 
        // Archetype creation can fan out through many positive selector ids. Use a monotonic stamp table
        // keyed by query-handle id so duplicate candidates do not devolve into repeated linear scans.
        ++m_createQueryHandleStamp;
        if (m_createQueryHandleStamp == 0) {
          m_createQueryHandleStampById = {};
          m_createQueryHandleStamp = 1;
        }
 
        return m_createQueryHandleScratch;
      }
 
      GAIA_NODISCARD bool mark_create_query_handle(QueryHandle handle) {
        const auto handleId = (uint32_t)handle.id();
        // darray::resize() leaves trivial types uninitialized. Zero-fill new slots explicitly,
        // otherwise recycled heap memory may already hold the current stamp and the query gets skipped.
        if (handleId >= m_createQueryHandleStampById.size())
          m_createQueryHandleStampById.resize(handleId + 1, 0U, cnt::ResizePolicy::Scale);
 
        auto& stamp = m_createQueryHandleStampById[handleId];
        if (stamp == m_createQueryHandleStamp)
          return false;
 
        stamp = m_createQueryHandleStamp;
        return true;
      }
 
      void add_archetype_query_pair(const Archetype* pArchetype, QueryHandle handle) {
        const auto archetypeKey = ArchetypeIdLookupKey(pArchetype->id(), pArchetype->id_hash());
        const auto it = m_archetypeToQuery.find(archetypeKey);
        if (it == m_archetypeToQuery.end()) {
          m_archetypeToQuery.try_emplace(archetypeKey, cnt::darray<QueryHandle>{handle});
          return;
        }
 
        auto& handles = it->second;
        // Callers only register a <query, archetype> edge after they proved that edge is new.
        GAIA_ASSERT(!core::has(handles, handle));
        handles.push_back(handle);
      }
 
      void del_archetype_query_pair(const Archetype* pArchetype, QueryHandle handle) {
        auto it = m_archetypeToQuery.find(ArchetypeIdLookupKey(pArchetype->id(), pArchetype->id_hash()));
        if (it == m_archetypeToQuery.end())
          return;
 
        auto& handles = it->second;
        const auto idx = core::get_index(handles, handle);
        GAIA_ASSERT(idx != BadIndex);
        core::swap_erase(handles, idx);
        if (handles.empty())
          m_archetypeToQuery.erase(it);
      }
 
      void unregister_query_archetypes(QueryHandle handle) {
        auto it = m_queryToArchetype.find(QueryHandleLookupKey(handle));
        if (it == m_queryToArchetype.end())
          return;
 
        const auto& tracked = it->second.archetypes;
        for (const auto* pArchetype: tracked)
          del_archetype_query_pair(pArchetype, handle);
 
        m_queryToArchetype.erase(it);
      }
 
      void register_query_archetype(QueryHandle handle, const Archetype* pArchetype, uint32_t syncedRevision) {
        auto [trackedIt, inserted] = m_queryToArchetype.try_emplace(QueryHandleLookupKey(handle));
        auto& tracked = trackedIt->second.archetypes;
 
        // Newly-created archetypes and sync_archetype_cache() both route through a deduplicated edge set,
        // so reverse-index registration can append directly instead of re-scanning tracked archetypes.
        GAIA_ASSERT(inserted || !core::has(tracked, pArchetype));
        tracked.push_back(pArchetype);
        trackedIt->second.syncedRevision = syncedRevision;
        add_archetype_query_pair(pArchetype, handle);
      }
 
      void add_rel_query_pair(Entity relation, QueryHandle handle) {
        const auto key = EntityLookupKey(relation);
        const auto it = m_relationToQuery.find(key);
        if (it == m_relationToQuery.end()) {
          m_relationToQuery.try_emplace(key, cnt::darray<QueryHandle>{handle});
          return;
        }
 
        auto& handles = it->second;
        if (!core::has(handles, handle))
          handles.push_back(handle);
      }
 
      void del_rel_query_pair(Entity relation, QueryHandle handle) {
        auto it = m_relationToQuery.find(EntityLookupKey(relation));
        if (it == m_relationToQuery.end())
          return;
 
        auto& handles = it->second;
        core::swap_erase(handles, core::get_index(handles, handle));
        if (handles.empty())
          m_relationToQuery.erase(it);
      }
 
      void add_rel_to_query_pairs(const QueryCtx& ctx, QueryHandle handle) {
        for (const auto relation: ctx.data.deps.relations_view())
          add_rel_query_pair(relation, handle);
      }
 
      void del_rel_to_query_pairs(const QueryCtx& ctx, QueryHandle handle) {
        for (const auto relation: ctx.data.deps.relations_view())
          del_rel_query_pair(relation, handle);
      }
    };
  } // namespace ecs
} // namespace gaia