component_desc.h

#pragma once
#include "gaia/config/config.h"
 
#include <cstdint>
#include <cstring>
#include <tuple>
#include <type_traits>
 
#include "gaia/core/span.h"
#include "gaia/core/utility.h"
#include "gaia/ecs/component.h"
#include "gaia/mem/data_layout_policy.h"
#include "gaia/mem/mem_utils.h"
#include "gaia/meta/reflection.h"
#include "gaia/meta/type_info.h"
#include "gaia/ser/ser_rt.h"
#include "gaia/util/str.h"
 
namespace gaia {
  namespace ecs {
    enum class RuntimeTypeKind : uint8_t {
      None,
      Primitive,
      Struct,
      Enum,
      Bitmask,
      Array,
      Vector,
      Opaque,
    };
 
#if GAIA_JSON_ENABLED
    enum class RuntimeJsonEncoding : uint8_t {
      Default,
      Utf8String
    };
#endif
 
    using RuntimeFieldFlagsType = uint16_t;
    enum RuntimeFieldFlags : RuntimeFieldFlagsType {
      RuntimeFieldFlag_None = 0,
      RuntimeFieldFlag_ReadOnly = 1u << 0,
      RuntimeFieldFlag_Hidden = 1u << 1,
      RuntimeFieldFlag_Multiline = 1u << 2,
      RuntimeFieldFlag_HasMinimum = 1u << 3,
      RuntimeFieldFlag_HasMaximum = 1u << 4,
      RuntimeFieldFlag_HasStep = 1u << 5
    };
 
    struct SymbolId final {
      uint32_t value = UINT32_MAX;
 
      GAIA_NODISCARD constexpr bool valid() const noexcept {
        return value != UINT32_MAX;
      }
 
      GAIA_NODISCARD constexpr bool operator==(SymbolId other) const noexcept {
        return value == other.value;
      }
 
      GAIA_NODISCARD constexpr bool operator!=(SymbolId other) const noexcept {
        return value != other.value;
      }
    };
 
    template <typename StringType>
    struct RuntimeFieldMeta final {
      static constexpr uint32_t MaxUnitLength = 32;
      StringType name{};
      Entity type = EntityBad;
      uint32_t offset = 0;
      uint32_t count = 0;
      Entity semantic = EntityBad;
#if GAIA_JSON_ENABLED
      RuntimeJsonEncoding jsonEncoding = RuntimeJsonEncoding::Default;
#endif
      RuntimeFieldFlagsType flags = RuntimeFieldFlag_None;
      StringType unit{};
      double minimum = 0.0;
      double maximum = 0.0;
      double step = 0.0;
    };
 
    using RuntimeFieldInit = RuntimeFieldMeta<util::str_view>;
    using RuntimeFieldDesc = RuntimeFieldMeta<SymbolId>;
 
    template <typename StringType>
    struct RuntimeConstantMeta final {
      StringType name{};
      int64_t value = 0;
    };
 
    using RuntimeConstantInit = RuntimeConstantMeta<util::str_view>;
    using RuntimeConstantDesc = RuntimeConstantMeta<SymbolId>;
 
    struct RuntimeSequenceScope final {
      Entity type = EntityBad;
      const void* data = nullptr;
      void* mutData = nullptr;
      uint32_t size = 0;
    };
 
    struct RuntimeSequenceElement final {
      Entity type = EntityBad;
      const void* data = nullptr;
      void* mutData = nullptr;
      uint32_t size = 0;
      void* token = nullptr;
    };
 
    struct RuntimeSequenceAdapter final {
      void* ctx = nullptr;
      bool (*count)(void*, const RuntimeSequenceScope&, uint32_t&) = nullptr;
      bool (*element)(void*, const RuntimeSequenceScope&, uint32_t, RuntimeSequenceElement&) = nullptr;
      bool (*resize)(void*, RuntimeSequenceScope&, uint32_t) = nullptr;
      bool (*commitElement)(void*, RuntimeSequenceScope&, RuntimeSequenceElement&) = nullptr;
    };
 
    struct RuntimeOpaqueScope final {
      Entity type = EntityBad;
      const void* data = nullptr;
      void* mutData = nullptr;
      uint32_t size = 0;
    };
 
    struct RuntimeOpaqueValue final {
      Entity type = EntityBad;
      const void* data = nullptr;
      void* mutData = nullptr;
      uint32_t size = 0;
      void* token = nullptr;
    };
 
    struct RuntimeOpaqueAdapter final {
      void* ctx = nullptr;
      bool (*project)(void*, const RuntimeOpaqueScope&, RuntimeOpaqueValue&) = nullptr;
      bool (*commit)(void*, RuntimeOpaqueScope&, RuntimeOpaqueValue&) = nullptr;
    };
 
    struct RuntimeTypeDesc final {
      RuntimeTypeKind typeKind = RuntimeTypeKind::Struct;
      Entity semantic = EntityBad;
#if GAIA_JSON_ENABLED
      RuntimeJsonEncoding jsonEncoding = RuntimeJsonEncoding::Default;
#endif
      Entity underlyingType = EntityBad;
      const RuntimeFieldInit* fields = nullptr;
      uint32_t fieldCount = 0;
      const RuntimeConstantInit* constants = nullptr;
      uint32_t constantCount = 0;
      Entity elementType = EntityBad;
      uint32_t elementCount = 0;
      Entity opaqueAsType = EntityBad;
      const RuntimeSequenceAdapter* sequenceAdapter = nullptr;
      const RuntimeOpaqueAdapter* opaqueAdapter = nullptr;
    };
 
    struct ComponentDesc final {
      using FuncCtor = void(void*, uint32_t);
      using FuncDtor = void(void*, uint32_t);
      using FuncFrom = void(void*, void*, uint32_t, uint32_t, uint32_t, uint32_t);
      using FuncCopy = void(void*, const void*, uint32_t, uint32_t, uint32_t, uint32_t);
      using FuncMove = void(void*, void*, uint32_t, uint32_t, uint32_t, uint32_t);
      using FuncSwap = void(void*, void*, uint32_t, uint32_t, uint32_t, uint32_t);
      using FuncCmp = bool(const void*, const void*);
      using FuncSave = void(ser::serializer&, const void*, uint32_t, uint32_t, uint32_t);
      using FuncLoad = void(ser::serializer&, void*, uint32_t, uint32_t, uint32_t);
 
      util::str_view name;
      uint32_t size = 0;
      uint32_t alig = 0;
      DataStorageType storageType = DataStorageType::Table;
      uint32_t soa = 0;
      const uint8_t* pSoaSizes = nullptr;
      ComponentLookupHash hashLookup{};
      RuntimeTypeDesc runtimeType{};
      FuncCtor* funcCtor = nullptr;
      FuncMove* funcMoveCtor = nullptr;
      FuncCopy* funcCopyCtor = nullptr;
      FuncDtor* funcDtor = nullptr;
      FuncCopy* funcCopy = nullptr;
      FuncMove* funcMove = nullptr;
      FuncSwap* funcSwap = nullptr;
      FuncCmp* funcCmp = nullptr;
      FuncSave* funcSave = nullptr;
      FuncLoad* funcLoad = nullptr;
    };
 
    namespace detail {
      template <typename T>
      struct ComponentDesc final {
        using CT = component_type_t<T>;
        using U = typename component_type_t<T>::Type;
        using DescU = typename CT::TypeFull;
 
        static constexpr ComponentLookupHash hash_lookup() {
          return {meta::type_info::hash<DescU>()};
        }
 
        static constexpr auto name() {
          return meta::type_info::name<DescU>();
        }
 
        static constexpr uint32_t size() {
          if constexpr (std::is_empty_v<U>)
            return 0;
          else
            return (uint32_t)sizeof(U);
        }
 
        static constexpr uint32_t alig() {
          constexpr auto alig = mem::auto_view_policy<U>::Alignment;
          static_assert(alig < Component::MaxAlignment, "Maximum supported alignment for a component is MaxAlignment");
          return alig;
        }
 
        static uint32_t soa(std::span<uint8_t, meta::StructToTupleMaxTypes> soaSizes) {
          if constexpr (mem::is_soa_layout_v<U>) {
            uint32_t i = 0;
            using TTuple = decltype(meta::struct_to_tuple(std::declval<U>()));
            // is_soa_layout_v is always false for empty types so we know there is at least one element in the tuple
            constexpr auto TTupleSize = std::tuple_size_v<TTuple>;
            static_assert(TTupleSize > 0);
            static_assert(TTupleSize <= meta::StructToTupleMaxTypes);
            core::each_tuple<TTuple>([&](auto&& item) {
              static_assert(sizeof(item) <= 255, "Each member of a SoA component can be at most 255 B long!");
              soaSizes[i] = (uint8_t)sizeof(item);
              ++i;
            });
            GAIA_ASSERT(i <= meta::StructToTupleMaxTypes);
            return i;
          } else {
            return 0U;
          }
        }
 
        static constexpr DataStorageType storage_type() {
          constexpr auto storageType = auto_storage_policy_v<U>;
          static_assert(
              storageType == DataStorageType::Table || storageType == DataStorageType::Sparse,
              "Unsupported component storage type");
          static_assert(
              storageType != DataStorageType::Sparse || CT::Kind == EntityKind::EK_Gen,
              "GAIA_STORAGE(Sparse) supports only generic components");
          static_assert(
              storageType != DataStorageType::Sparse || !std::is_empty_v<U>,
              "GAIA_STORAGE(Sparse) requires a non-empty component payload");
          static_assert(
              storageType != DataStorageType::Sparse || !mem::is_soa_layout_v<U>,
              "GAIA_STORAGE(Sparse) is not compatible with SoA layouts");
          return storageType;
        }
 
        static constexpr auto func_ctor() {
          if constexpr (!mem::is_soa_layout_v<U> && !std::is_trivially_constructible_v<U>) {
            return [](void* ptr, uint32_t cnt) {
              core::call_ctor_n((U*)ptr, cnt);
            };
          } else {
            return nullptr;
          }
        }
 
        static constexpr auto func_dtor() {
          if constexpr (!mem::is_soa_layout_v<U> && !std::is_trivially_destructible_v<U>) {
            return [](void* ptr, uint32_t cnt) {
              core::call_dtor_n((U*)ptr, cnt);
            };
          } else {
            return nullptr;
          }
        }
 
        static constexpr auto func_copy_ctor() {
          return [](void* GAIA_RESTRICT dst, const void* GAIA_RESTRICT src, uint32_t idxDst, uint32_t idxSrc,
                    uint32_t sizeDst, uint32_t sizeSrc) {
            mem::copy_ctor_element<U>((uint8_t*)dst, (const uint8_t*)src, idxDst, idxSrc, sizeDst, sizeSrc);
          };
        }
 
        static constexpr auto func_move_ctor() {
          return [](void* GAIA_RESTRICT dst, void* GAIA_RESTRICT src, uint32_t idxDst, uint32_t idxSrc,
                    uint32_t sizeDst, uint32_t sizeSrc) {
            mem::move_ctor_element<U>((uint8_t*)dst, (uint8_t*)src, idxDst, idxSrc, sizeDst, sizeSrc);
          };
        }
 
        static constexpr auto func_copy() {
          return [](void* GAIA_RESTRICT dst, const void* GAIA_RESTRICT src, uint32_t idxDst, uint32_t idxSrc,
                    uint32_t sizeDst, uint32_t sizeSrc) {
            mem::copy_element<U>((uint8_t*)dst, (const uint8_t*)src, idxDst, idxSrc, sizeDst, sizeSrc);
          };
        }
 
        static constexpr auto func_move() {
          return [](void* GAIA_RESTRICT dst, void* GAIA_RESTRICT src, uint32_t idxDst, uint32_t idxSrc,
                    uint32_t sizeDst, uint32_t sizeSrc) {
            mem::move_element<U>((uint8_t*)dst, (uint8_t*)src, idxDst, idxSrc, sizeDst, sizeSrc);
          };
        }
 
        static constexpr auto func_swap() {
          return [](void* GAIA_RESTRICT left, void* GAIA_RESTRICT right, uint32_t idxLeft, uint32_t idxRight,
                    uint32_t sizeLeft, uint32_t sizeRight) {
            mem::swap_elements<U>((uint8_t*)left, (uint8_t*)right, idxLeft, idxRight, sizeLeft, sizeRight);
          };
        }
 
        static constexpr auto func_cmp() {
          if constexpr (mem::is_soa_layout_v<U>) {
            return []([[maybe_unused]] const void* left, [[maybe_unused]] const void* right) {
              GAIA_ASSERT(false && "func_cmp for SoA not implemented yet");
              return false;
            };
          } else {
            constexpr bool hasGlobalCmp = core::has_ffunc_equals<U>::value;
            constexpr bool hasMemberCmp = core::has_func_equals<U>::value;
            if constexpr (hasGlobalCmp || hasMemberCmp) {
              return [](const void* left, const void* right) {
                const auto* l = (const U*)left;
                const auto* r = (const U*)right;
                return *l == *r;
              };
            } else {
              // fallback comparison function
              return [](const void* left, const void* right) {
                const auto* l = (const U*)left;
                const auto* r = (const U*)right;
                return memcmp(l, r, sizeof(U)) == 0;
              };
            }
          }
        }
 
        static constexpr auto func_save() {
          return [](ser::serializer& s, const void* pSrc, uint32_t from, uint32_t to, uint32_t cap) {
            const auto* pComponent = (const U*)pSrc;
 
#if GAIA_ASSERT_ENABLED
            // Check if save and load match. Testing with one item is enough.
            s.check(*pComponent);
#endif
 
            if constexpr (mem::is_soa_layout_v<U>) {
              auto view = mem::auto_view_policy_get<U>{std::span{(const uint8_t*)pSrc, cap}};
              GAIA_FOR2(from, to) {
                auto val = view[i];
                s.save(val);
              }
            } else {
              pComponent += from;
              GAIA_FOR2(from, to) {
                s.save(*pComponent);
                ++pComponent;
              }
            }
          };
        }
 
        static constexpr auto func_load() {
          return [](ser::serializer& s, void* pDst, uint32_t from, uint32_t to, uint32_t cap) {
            if constexpr (mem::is_soa_layout_v<U>) {
              auto view = mem::auto_view_policy_set<U>{std::span{(uint8_t*)pDst, cap}};
              GAIA_FOR2(from, to) {
                U val;
                s.load(val);
                view[i] = val;
              }
            } else {
              auto* pComponent = (U*)pDst + from;
              GAIA_FOR2(from, to) {
                s.load(*pComponent);
                ++pComponent;
              }
            }
          };
        }
 
#if GAIA_ECS_AUTO_COMPONENT_FIELDS
 
        template <typename Field>
        static Entity auto_field_type() noexcept {
          using FU = core::raw_t<Field>;
          if constexpr (std::is_same_v<FU, bool>)
            return Bool;
          else if constexpr (std::is_same_v<FU, char>)
            return Char8;
          else if constexpr (std::is_arithmetic_v<FU>)
            return runtime_primitive_type_entity(ser::type_id<FU>());
          else
            return EntityBad;
        }
 
        static uint32_t auto_fields(std::span<RuntimeFieldInit, meta::StructToTupleMaxTypes> fields) {
          using Raw = core::raw_t<T>;
          if constexpr (std::is_empty_v<Raw> || mem::is_soa_layout_v<Raw>) {
            return 0;
          } else if constexpr (!std::is_class_v<Raw>) {
            const auto fieldType = auto_field_type<Raw>();
            if (fieldType == EntityBad)
              return 0;
            fields[0] = {util::str_view("value", 5), fieldType, 0, 0};
            return 1;
          } else if constexpr (!std::is_aggregate_v<Raw>) {
            return 0;
          } else {
            Raw tmp{};
            const auto* pBase = reinterpret_cast<const uint8_t*>(&tmp);
            static constexpr const char* FieldNames[meta::StructToTupleMaxTypes] = {
                "f0", "f1", "f2", "f3", "f4", "f5", "f6", "f7", "f8", "f9", "f10", "f11", "f12", "f13", "f14"};
            uint32_t fieldCount = 0;
            meta::each_member(tmp, [&](auto&... members) {
              auto add_field = [&](auto& member) {
                using Field = core::raw_t<decltype(member)>;
                const auto fieldType = auto_field_type<Field>();
                if (fieldType == EntityBad)
                  return;
                GAIA_ASSERT(fieldCount < meta::StructToTupleMaxTypes);
                const auto* pField = reinterpret_cast<const uint8_t*>(&member);
                const auto offset = (uint32_t)(pField - pBase);
                const auto* fieldName = FieldNames[fieldCount];
                const auto fieldNameLen = (uint32_t)GAIA_STRLEN(fieldName, 4);
                fields[fieldCount] = {util::str_view(fieldName, fieldNameLen), fieldType, offset, 0};
                ++fieldCount;
              };
              (add_field(members), ...);
            });
            return fieldCount;
          }
        }
 
#endif
 
        static ecs::ComponentDesc
        make(util::str_view descName, std::span<uint8_t, meta::StructToTupleMaxTypes> soaSizes) {
          ecs::ComponentDesc desc{};
          desc.name = descName;
          desc.size = size();
          desc.alig = alig();
          desc.storageType = storage_type();
          desc.soa = soa(soaSizes);
          desc.pSoaSizes = soaSizes.data();
          desc.hashLookup = hash_lookup();
          desc.funcCtor = func_ctor();
          desc.funcMoveCtor = func_move_ctor();
          desc.funcCopyCtor = func_copy_ctor();
          desc.funcDtor = func_dtor();
          desc.funcCopy = func_copy();
          desc.funcMove = func_move();
          desc.funcSwap = func_swap();
          desc.funcCmp = func_cmp();
          desc.funcSave = func_save();
          desc.funcLoad = func_load();
          return desc;
        }
      };
    } // namespace detail
  } // namespace ecs
} // namespace gaia