79 lines
2.5 KiB
C++
79 lines
2.5 KiB
C++
|
#include <gtest/gtest.h>
|
||
|
#include <entt/core/hashed_string.hpp>
|
||
|
#include <entt/entity/registry.hpp>
|
||
|
|
||
|
enum class my_entity : entt::id_type {};
|
||
|
|
||
|
TEST(Example, EntityCopy) {
|
||
|
using namespace entt::literals;
|
||
|
|
||
|
entt::registry registry{};
|
||
|
auto &&custom = registry.storage<double>("custom"_hs);
|
||
|
|
||
|
const auto src = registry.create();
|
||
|
const auto dst = registry.create();
|
||
|
const auto other = registry.create();
|
||
|
|
||
|
custom.emplace(src, 1.);
|
||
|
registry.emplace<int>(src, 42);
|
||
|
registry.emplace<char>(src, 'c');
|
||
|
registry.emplace<double>(other, 3.);
|
||
|
|
||
|
ASSERT_TRUE(custom.contains(src));
|
||
|
ASSERT_FALSE(registry.all_of<double>(src));
|
||
|
ASSERT_TRUE((registry.all_of<int, char>(src)));
|
||
|
ASSERT_FALSE((registry.any_of<int, char, double>(dst)));
|
||
|
ASSERT_FALSE(custom.contains(dst));
|
||
|
|
||
|
for(auto [id, storage]: registry.storage()) {
|
||
|
// discard the custom storage because why not, this is just an example after all
|
||
|
if(id != "custom"_hs && storage.contains(src)) {
|
||
|
storage.emplace(dst, storage.get(src));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
ASSERT_TRUE((registry.all_of<int, char>(dst)));
|
||
|
ASSERT_FALSE((registry.all_of<double>(dst)));
|
||
|
ASSERT_FALSE(custom.contains(dst));
|
||
|
|
||
|
ASSERT_EQ(registry.get<int>(dst), 42);
|
||
|
ASSERT_EQ(registry.get<char>(dst), 'c');
|
||
|
}
|
||
|
|
||
|
TEST(Example, DifferentRegistryTypes) {
|
||
|
using namespace entt::literals;
|
||
|
|
||
|
entt::basic_registry<entt::entity> src{};
|
||
|
entt::basic_registry<my_entity> dst{};
|
||
|
|
||
|
/*
|
||
|
TODO These are currently needed to ensure that the source and
|
||
|
target registries have the proper storage initialized
|
||
|
prior to copying, as this isn't done automatically
|
||
|
when emplacing storages (as is done below).
|
||
|
|
||
|
There is an open issue about this, and these two
|
||
|
lines should be removed when a fix is properly landed.
|
||
|
https://github.com/skypjack/entt/issues/827
|
||
|
*/
|
||
|
static_cast<void>(src.storage<double>());
|
||
|
static_cast<void>(dst.storage<int>());
|
||
|
|
||
|
const auto entity = src.create();
|
||
|
const auto copy = dst.create();
|
||
|
|
||
|
src.emplace<int>(entity, 42);
|
||
|
src.emplace<char>(entity, 'c');
|
||
|
|
||
|
for(auto [id, storage]: src.storage()) {
|
||
|
if(auto *other = dst.storage(id); other && storage.contains(entity)) {
|
||
|
other->emplace(copy, storage.get(entity));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
ASSERT_TRUE((src.all_of<int, char>(entity)));
|
||
|
ASSERT_FALSE(dst.all_of<char>(copy));
|
||
|
ASSERT_TRUE(dst.all_of<int>(copy));
|
||
|
ASSERT_EQ(dst.get<int>(copy), 42);
|
||
|
}
|