custom_identifier.cpp 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. #include <type_traits>
  2. #include <gtest/gtest.h>
  3. #include <entt/entity/entity.hpp>
  4. #include <entt/entity/registry.hpp>
  5. struct entity_id {
  6. using entity_type = typename entt::entt_traits<entt::entity>::entity_type;
  7. entity_id(entity_type value = entt::null)
  8. : entt{value}
  9. {}
  10. entity_id(const entity_id &other)
  11. : entt{other.entt}
  12. {}
  13. operator entity_type() const {
  14. return entt;
  15. }
  16. private:
  17. entity_type entt;
  18. };
  19. template<>
  20. struct entt::entt_traits<entity_id>: entt::entt_traits<entt::entity> {};
  21. TEST(Example, CustomIdentifier) {
  22. entt::basic_registry<entity_id> registry{};
  23. entity_id entity{};
  24. ASSERT_FALSE(registry.valid(entity));
  25. ASSERT_TRUE(entity == entt::null);
  26. entity = registry.create();
  27. ASSERT_TRUE(registry.valid(entity));
  28. ASSERT_TRUE(entity != entt::null);
  29. ASSERT_FALSE((registry.has<int, char>(entity)));
  30. ASSERT_EQ(registry.try_get<int>(entity), nullptr);
  31. registry.emplace<int>(entity, 42);
  32. ASSERT_TRUE((registry.any<int, char>(entity)));
  33. ASSERT_EQ(registry.get<int>(entity), 42);
  34. registry.destroy(entity);
  35. ASSERT_FALSE(registry.valid(entity));
  36. ASSERT_TRUE(entity != entt::null);
  37. entity = registry.create();
  38. ASSERT_TRUE(registry.valid(entity));
  39. ASSERT_TRUE(entity != entt::null);
  40. }