archetype_graph.h

#pragma once
#include "gaia/config/config.h"
 
#include <cstdint>
 
#include "gaia/cnt/map.h"
#include "gaia/ecs/api.h"
#include "gaia/ecs/archetype_common.h"
#include "gaia/ecs/component.h"
#include "gaia/ecs/id.h"
#include "gaia/util/logging.h"
 
namespace gaia {
  namespace ecs {
    class World;
 
    using ArchetypeGraphEdge = ArchetypeIdHashPair;
 
    class ArchetypeGraph {
      using EdgeMap = cnt::map<EntityLookupKey, ArchetypeGraphEdge>;
 
      EdgeMap m_edgesAdd;
      EdgeMap m_edgesDel;
 
    private:
      void add_edge(EdgeMap& edges, Entity entity, ArchetypeId archetypeId, ArchetypeIdHash hash) {
#if GAIA_ASSERT_ENABLED
        const auto ret =
#endif
            edges.try_emplace(EntityLookupKey(entity), ArchetypeGraphEdge{archetypeId, hash});
#if GAIA_ASSERT_ENABLED
        // If the result already exists make sure the new one is the same
        if (!ret.second) {
          const auto it = edges.find(EntityLookupKey(entity));
          GAIA_ASSERT(it != edges.end());
          GAIA_ASSERT(it->second.id == archetypeId);
          GAIA_ASSERT(it->second.hash == hash);
        }
#endif
      }
 
      void del_edge(EdgeMap& edges, Entity entity) {
        edges.erase(EntityLookupKey(entity));
      }
 
      GAIA_NODISCARD ArchetypeGraphEdge find_edge(const EdgeMap& edges, Entity entity) const {
        const auto it = edges.find(EntityLookupKey(entity));
        return it != edges.end() ? it->second : ArchetypeIdHashPairBad;
      }
 
    public:
      void add_edge_right(Entity entity, ArchetypeId archetypeId, ArchetypeIdHash hash) {
        add_edge(m_edgesAdd, entity, archetypeId, hash);
      }
 
      void add_edge_left(Entity entity, ArchetypeId archetypeId, ArchetypeIdHash hash) {
        add_edge(m_edgesDel, entity, archetypeId, hash);
      }
 
      void del_edge_right(Entity entity) {
        del_edge(m_edgesAdd, entity);
      }
 
      void del_edge_left(Entity entity) {
        del_edge(m_edgesDel, entity);
      }
 
      GAIA_NODISCARD ArchetypeGraphEdge find_edge_right(Entity entity) const {
        return find_edge(m_edgesAdd, entity);
      }
 
      GAIA_NODISCARD ArchetypeGraphEdge find_edge_left(Entity entity) const {
        return find_edge(m_edgesDel, entity);
      }
 
      GAIA_NODISCARD auto& right_edges() {
        return m_edgesAdd;
      }
 
      GAIA_NODISCARD const auto& right_edges() const {
        return m_edgesAdd;
      }
 
      GAIA_NODISCARD auto& left_edges() {
        return m_edgesDel;
      }
 
      GAIA_NODISCARD const auto& left_edges() const {
        return m_edgesDel;
      }
 
      void diag(const World& world) const {
        auto diagEdge = [&](const auto& edges) {
          for (const auto& edge: edges) {
            const auto entity = edge.first.entity();
            if (entity.pair()) {
              const auto name0 = entity_name(world, entity.id());
              const auto name1 = entity_name(world, entity.gen());
              GAIA_LOG_N(
                  "    pair [%u:%u], %.*s -> %.*s, aid:%u",
                  //
                  entity.id(), entity.gen(), (int)name0.size(), name0.empty() ? "" : name0.data(), (int)name1.size(),
                  name1.empty() ? "" : name1.data(), edge.second.id);
            } else {
              const auto name = entity_name(world, entity);
              GAIA_LOG_N(
                  "    ent [%u:%u], %.*s [%s], aid:%u",
                  //
                  entity.id(), entity.gen(), (int)name.size(), name.empty() ? "" : name.data(),
                  EntityKindString[entity.kind()], edge.second.id);
            }
          }
        };
 
        // Add edges (movement towards the leafs)
        if (!m_edgesAdd.empty()) {
          GAIA_LOG_N("  Add edges - count:%u", (uint32_t)m_edgesAdd.size());
          diagEdge(m_edgesAdd);
        }
 
        // Delete edges (movement towards the root)
        if (!m_edgesDel.empty()) {
          GAIA_LOG_N("  Del edges - count:%u", (uint32_t)m_edgesDel.size());
          diagEdge(m_edgesDel);
        }
      }
    };
  } // namespace ecs
} // namespace gaia