registry.cpp 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735
  1. #include <unordered_map>
  2. #include <unordered_set>
  3. #include <functional>
  4. #include <type_traits>
  5. #include <gtest/gtest.h>
  6. #include <entt/entity/entt_traits.hpp>
  7. #include <entt/entity/registry.hpp>
  8. struct Listener {
  9. template<typename Component>
  10. void incrComponent(entt::DefaultRegistry &registry, entt::DefaultRegistry::entity_type entity) {
  11. ASSERT_TRUE(registry.valid(entity));
  12. ASSERT_TRUE(registry.has<Component>(entity));
  13. last = entity;
  14. ++counter;
  15. }
  16. template<typename Tag>
  17. void incrTag(entt::DefaultRegistry &registry, entt::DefaultRegistry::entity_type entity) {
  18. ASSERT_TRUE(registry.valid(entity));
  19. ASSERT_TRUE(registry.has<Tag>());
  20. ASSERT_EQ(registry.attachee<Tag>(), entity);
  21. last = entity;
  22. ++counter;
  23. }
  24. template<typename Component>
  25. void decrComponent(entt::DefaultRegistry &registry, entt::DefaultRegistry::entity_type entity) {
  26. ASSERT_TRUE(registry.valid(entity));
  27. ASSERT_TRUE(registry.has<Component>(entity));
  28. last = entity;
  29. --counter;
  30. }
  31. template<typename Tag>
  32. void decrTag(entt::DefaultRegistry &registry, entt::DefaultRegistry::entity_type entity) {
  33. ASSERT_TRUE(registry.valid(entity));
  34. ASSERT_TRUE(registry.has<Tag>());
  35. ASSERT_EQ(registry.attachee<Tag>(), entity);
  36. last = entity;
  37. --counter;
  38. }
  39. entt::DefaultRegistry::entity_type last;
  40. int counter{0};
  41. };
  42. TEST(DefaultRegistry, Types) {
  43. entt::DefaultRegistry registry;
  44. ASSERT_EQ(registry.type<int>(entt::tag_t{}), registry.type<int>(entt::tag_t{}));
  45. ASSERT_EQ(registry.type<int>(), registry.type<int>());
  46. ASSERT_NE(registry.type<int>(entt::tag_t{}), registry.type<double>(entt::tag_t{}));
  47. ASSERT_NE(registry.type<int>(), registry.type<double>(entt::tag_t{}));
  48. }
  49. TEST(DefaultRegistry, Functionalities) {
  50. entt::DefaultRegistry registry;
  51. ASSERT_EQ(registry.size(), entt::DefaultRegistry::size_type{0});
  52. ASSERT_NO_THROW(registry.reserve(42));
  53. ASSERT_NO_THROW(registry.reserve<int>(8));
  54. ASSERT_NO_THROW(registry.reserve<char>(8));
  55. ASSERT_TRUE(registry.empty());
  56. ASSERT_EQ(registry.capacity(), entt::DefaultRegistry::size_type{0});
  57. ASSERT_EQ(registry.size<int>(), entt::DefaultRegistry::size_type{0});
  58. ASSERT_EQ(registry.size<char>(), entt::DefaultRegistry::size_type{0});
  59. ASSERT_TRUE(registry.empty<int>());
  60. ASSERT_TRUE(registry.empty<char>());
  61. const auto e0 = registry.create();
  62. const auto e1 = registry.create();
  63. registry.assign<int>(e1);
  64. registry.assign<char>(e1);
  65. ASSERT_TRUE(registry.has<>(e0));
  66. ASSERT_TRUE(registry.has<>(e1));
  67. ASSERT_EQ(registry.capacity(), entt::DefaultRegistry::size_type{2});
  68. ASSERT_EQ(registry.size<int>(), entt::DefaultRegistry::size_type{1});
  69. ASSERT_EQ(registry.size<char>(), entt::DefaultRegistry::size_type{1});
  70. ASSERT_FALSE(registry.empty<int>());
  71. ASSERT_FALSE(registry.empty<char>());
  72. ASSERT_NE(e0, e1);
  73. ASSERT_FALSE(registry.has<int>(e0));
  74. ASSERT_TRUE(registry.has<int>(e1));
  75. ASSERT_FALSE(registry.has<char>(e0));
  76. ASSERT_TRUE(registry.has<char>(e1));
  77. ASSERT_FALSE((registry.has<int, char>(e0)));
  78. ASSERT_TRUE((registry.has<int, char>(e1)));
  79. ASSERT_EQ(registry.assign<int>(e0, 42), 42);
  80. ASSERT_EQ(registry.assign<char>(e0, 'c'), 'c');
  81. ASSERT_NO_THROW(registry.remove<int>(e1));
  82. ASSERT_NO_THROW(registry.remove<char>(e1));
  83. ASSERT_TRUE(registry.has<int>(e0));
  84. ASSERT_FALSE(registry.has<int>(e1));
  85. ASSERT_TRUE(registry.has<char>(e0));
  86. ASSERT_FALSE(registry.has<char>(e1));
  87. ASSERT_TRUE((registry.has<int, char>(e0)));
  88. ASSERT_FALSE((registry.has<int, char>(e1)));
  89. const auto e2 = registry.create();
  90. registry.accommodate<int>(e2, registry.get<int>(e0));
  91. registry.accommodate<char>(e2, registry.get<char>(e0));
  92. ASSERT_TRUE(registry.has<int>(e2));
  93. ASSERT_TRUE(registry.has<char>(e2));
  94. ASSERT_EQ(registry.get<int>(e0), 42);
  95. ASSERT_EQ(registry.get<char>(e0), 'c');
  96. ASSERT_EQ(std::get<0>(registry.get<int, char>(e0)), 42);
  97. ASSERT_EQ(std::get<1>(static_cast<const entt::DefaultRegistry &>(registry).get<int, char>(e0)), 'c');
  98. ASSERT_EQ(registry.get<int>(e0), registry.get<int>(e2));
  99. ASSERT_EQ(registry.get<char>(e0), registry.get<char>(e2));
  100. ASSERT_NE(&registry.get<int>(e0), &registry.get<int>(e2));
  101. ASSERT_NE(&registry.get<char>(e0), &registry.get<char>(e2));
  102. ASSERT_NO_THROW(registry.replace<int>(e0, 0));
  103. ASSERT_EQ(registry.get<int>(e0), 0);
  104. ASSERT_NO_THROW(registry.accommodate<int>(e0, 1));
  105. ASSERT_NO_THROW(registry.accommodate<int>(e1, 1));
  106. ASSERT_EQ(static_cast<const entt::DefaultRegistry &>(registry).get<int>(e0), 1);
  107. ASSERT_EQ(static_cast<const entt::DefaultRegistry &>(registry).get<int>(e1), 1);
  108. ASSERT_EQ(registry.size(), entt::DefaultRegistry::size_type{3});
  109. ASSERT_FALSE(registry.empty());
  110. ASSERT_EQ(registry.version(e2), entt::DefaultRegistry::version_type{0});
  111. ASSERT_EQ(registry.current(e2), entt::DefaultRegistry::version_type{0});
  112. ASSERT_EQ(registry.capacity(), entt::DefaultRegistry::size_type{3});
  113. ASSERT_NO_THROW(registry.destroy(e2));
  114. ASSERT_EQ(registry.capacity(), entt::DefaultRegistry::size_type{3});
  115. ASSERT_EQ(registry.version(e2), entt::DefaultRegistry::version_type{0});
  116. ASSERT_EQ(registry.current(e2), entt::DefaultRegistry::version_type{1});
  117. ASSERT_TRUE(registry.valid(e0));
  118. ASSERT_TRUE(registry.fast(e0));
  119. ASSERT_TRUE(registry.valid(e1));
  120. ASSERT_TRUE(registry.fast(e1));
  121. ASSERT_FALSE(registry.valid(e2));
  122. ASSERT_FALSE(registry.fast(e2));
  123. ASSERT_EQ(registry.size(), entt::DefaultRegistry::size_type{2});
  124. ASSERT_FALSE(registry.empty());
  125. ASSERT_NO_THROW(registry.reset());
  126. ASSERT_EQ(registry.size(), entt::DefaultRegistry::size_type{0});
  127. ASSERT_TRUE(registry.empty());
  128. const auto e3 = registry.create();
  129. registry.assign<int>(e3);
  130. registry.assign<char>(e3);
  131. ASSERT_EQ(registry.size<int>(), entt::DefaultRegistry::size_type{1});
  132. ASSERT_EQ(registry.size<char>(), entt::DefaultRegistry::size_type{1});
  133. ASSERT_FALSE(registry.empty<int>());
  134. ASSERT_FALSE(registry.empty<char>());
  135. ASSERT_NO_THROW(registry.reset<int>());
  136. ASSERT_EQ(registry.size<int>(), entt::DefaultRegistry::size_type{0});
  137. ASSERT_EQ(registry.size<char>(), entt::DefaultRegistry::size_type{1});
  138. ASSERT_TRUE(registry.empty<int>());
  139. ASSERT_FALSE(registry.empty<char>());
  140. ASSERT_NO_THROW(registry.reset());
  141. ASSERT_EQ(registry.size<int>(), entt::DefaultRegistry::size_type{0});
  142. ASSERT_EQ(registry.size<char>(), entt::DefaultRegistry::size_type{0});
  143. ASSERT_TRUE(registry.empty<int>());
  144. ASSERT_TRUE(registry.empty<char>());
  145. const auto e4 = registry.create();
  146. const auto e5 = registry.create();
  147. registry.assign<int>(e4);
  148. ASSERT_NO_THROW(registry.reset<int>(e4));
  149. ASSERT_NO_THROW(registry.reset<int>(e5));
  150. ASSERT_EQ(registry.size<int>(), entt::DefaultRegistry::size_type{0});
  151. ASSERT_EQ(registry.size<char>(), entt::DefaultRegistry::size_type{0});
  152. ASSERT_TRUE(registry.empty<int>());
  153. }
  154. TEST(DefaultRegistry, RawData) {
  155. entt::DefaultRegistry registry;
  156. const entt::DefaultRegistry &cregistry = registry;
  157. const auto entity = registry.create();
  158. ASSERT_EQ(registry.raw<int>(), nullptr);
  159. ASSERT_EQ(cregistry.raw<int>(), nullptr);
  160. ASSERT_EQ(cregistry.data<int>(), nullptr);
  161. registry.assign<int>(entity, 42);
  162. ASSERT_NE(registry.raw<int>(), nullptr);
  163. ASSERT_NE(cregistry.raw<int>(), nullptr);
  164. ASSERT_NE(cregistry.data<int>(), nullptr);
  165. ASSERT_EQ(*registry.raw<int>(), 42);
  166. ASSERT_EQ(*cregistry.raw<int>(), 42);
  167. ASSERT_EQ(*cregistry.data<int>(), entity);
  168. }
  169. TEST(DefaultRegistry, CreateDestroyCornerCase) {
  170. entt::DefaultRegistry registry;
  171. const auto e0 = registry.create();
  172. const auto e1 = registry.create();
  173. registry.destroy(e0);
  174. registry.destroy(e1);
  175. registry.each([](auto) { FAIL(); });
  176. ASSERT_EQ(registry.current(e0), entt::DefaultRegistry::version_type{1});
  177. ASSERT_EQ(registry.current(e1), entt::DefaultRegistry::version_type{1});
  178. }
  179. TEST(DefaultRegistry, VersionOverflow) {
  180. entt::DefaultRegistry registry;
  181. const auto entity = registry.create();
  182. registry.destroy(entity);
  183. ASSERT_EQ(registry.version(entity), entt::DefaultRegistry::version_type{});
  184. for(auto i = entt::entt_traits<entt::DefaultRegistry::entity_type>::version_mask; i; --i) {
  185. ASSERT_NE(registry.current(entity), registry.version(entity));
  186. registry.destroy(registry.create());
  187. }
  188. ASSERT_EQ(registry.current(entity), registry.version(entity));
  189. }
  190. TEST(DefaultRegistry, Each) {
  191. entt::DefaultRegistry registry;
  192. entt::DefaultRegistry::size_type tot;
  193. entt::DefaultRegistry::size_type match;
  194. registry.create();
  195. registry.assign<int>(registry.create());
  196. registry.create();
  197. registry.assign<int>(registry.create());
  198. registry.create();
  199. tot = 0u;
  200. match = 0u;
  201. registry.each([&](auto entity) {
  202. if(registry.has<int>(entity)) { ++match; }
  203. registry.create();
  204. ++tot;
  205. });
  206. ASSERT_EQ(tot, 5u);
  207. ASSERT_EQ(match, 2u);
  208. tot = 0u;
  209. match = 0u;
  210. registry.each([&](auto entity) {
  211. if(registry.has<int>(entity)) {
  212. registry.destroy(entity);
  213. ++match;
  214. }
  215. ++tot;
  216. });
  217. ASSERT_EQ(tot, 10u);
  218. ASSERT_EQ(match, 2u);
  219. tot = 0u;
  220. match = 0u;
  221. registry.each([&](auto entity) {
  222. if(registry.has<int>(entity)) { ++match; }
  223. registry.destroy(entity);
  224. ++tot;
  225. });
  226. ASSERT_EQ(tot, 8u);
  227. ASSERT_EQ(match, 0u);
  228. registry.each([&](auto) { FAIL(); });
  229. }
  230. TEST(DefaultRegistry, Orphans) {
  231. entt::DefaultRegistry registry;
  232. entt::DefaultRegistry::size_type tot{};
  233. registry.assign<int>(registry.create());
  234. registry.create();
  235. registry.assign<int>(registry.create());
  236. registry.create();
  237. registry.assign<double>(entt::tag_t{}, registry.create());
  238. registry.orphans([&](auto) { ++tot; });
  239. ASSERT_EQ(tot, 2u);
  240. tot = 0u;
  241. registry.each([&](auto entity) { registry.reset<int>(entity); });
  242. registry.orphans([&](auto) { ++tot; });
  243. ASSERT_EQ(tot, 4u);
  244. registry.reset();
  245. tot = 0u;
  246. registry.orphans([&](auto) { ++tot; });
  247. ASSERT_EQ(tot, 0u);
  248. }
  249. TEST(DefaultRegistry, CreateDestroyEntities) {
  250. entt::DefaultRegistry registry;
  251. entt::DefaultRegistry::entity_type pre{}, post{};
  252. for(int i = 0; i < 10; ++i) {
  253. const auto entity = registry.create();
  254. registry.assign<double>(entity);
  255. }
  256. registry.reset();
  257. for(int i = 0; i < 7; ++i) {
  258. const auto entity = registry.create();
  259. registry.assign<int>(entity);
  260. if(i == 3) { pre = entity; }
  261. }
  262. registry.reset();
  263. for(int i = 0; i < 5; ++i) {
  264. const auto entity = registry.create();
  265. if(i == 3) { post = entity; }
  266. }
  267. ASSERT_FALSE(registry.valid(pre));
  268. ASSERT_TRUE(registry.valid(post));
  269. ASSERT_NE(registry.version(pre), registry.version(post));
  270. ASSERT_EQ(registry.version(pre) + 1, registry.version(post));
  271. ASSERT_EQ(registry.current(pre), registry.current(post));
  272. }
  273. TEST(DefaultRegistry, AttachSetRemoveTags) {
  274. entt::DefaultRegistry registry;
  275. const auto &cregistry = registry;
  276. ASSERT_FALSE(registry.has<int>());
  277. const auto entity = registry.create();
  278. registry.assign<int>(entt::tag_t{}, entity, 42);
  279. ASSERT_TRUE(registry.has<int>());
  280. ASSERT_EQ(registry.get<int>(), 42);
  281. ASSERT_EQ(cregistry.get<int>(), 42);
  282. ASSERT_EQ(registry.attachee<int>(), entity);
  283. registry.replace<int>(entt::tag_t{}, 3);
  284. ASSERT_TRUE(registry.has<int>());
  285. ASSERT_EQ(registry.get<int>(), 3);
  286. ASSERT_EQ(cregistry.get<int>(), 3);
  287. ASSERT_EQ(registry.attachee<int>(), entity);
  288. const auto other = registry.create();
  289. registry.move<int>(other);
  290. ASSERT_TRUE(registry.has<int>());
  291. ASSERT_EQ(registry.get<int>(), 3);
  292. ASSERT_EQ(cregistry.get<int>(), 3);
  293. ASSERT_EQ(registry.attachee<int>(), other);
  294. registry.remove<int>();
  295. ASSERT_FALSE(registry.has<int>());
  296. registry.assign<int>(entt::tag_t{}, entity, 42);
  297. registry.destroy(entity);
  298. ASSERT_FALSE(registry.has<int>());
  299. }
  300. TEST(DefaultRegistry, StandardViews) {
  301. entt::DefaultRegistry registry;
  302. auto mview = registry.view<int, char>();
  303. auto iview = registry.view<int>();
  304. auto cview = registry.view<char>();
  305. const auto e0 = registry.create();
  306. registry.assign<int>(e0, 0);
  307. registry.assign<char>(e0, 'c');
  308. const auto e1 = registry.create();
  309. registry.assign<int>(e1, 0);
  310. const auto e2 = registry.create();
  311. registry.assign<int>(e2, 0);
  312. registry.assign<char>(e2, 'c');
  313. ASSERT_EQ(iview.size(), decltype(iview)::size_type{3});
  314. ASSERT_EQ(cview.size(), decltype(cview)::size_type{2});
  315. decltype(mview)::size_type cnt{0};
  316. mview.each([&cnt](auto...) { ++cnt; });
  317. ASSERT_EQ(cnt, decltype(mview)::size_type{2});
  318. }
  319. TEST(DefaultRegistry, PersistentViews) {
  320. entt::DefaultRegistry registry;
  321. auto view = registry.view<int, char>(entt::persistent_t{});
  322. ASSERT_TRUE((registry.contains<int, char>()));
  323. ASSERT_FALSE((registry.contains<int, double>()));
  324. registry.prepare<int, double>();
  325. ASSERT_TRUE((registry.contains<int, double>()));
  326. registry.discard<int, double>();
  327. ASSERT_FALSE((registry.contains<int, double>()));
  328. const auto e0 = registry.create();
  329. registry.assign<int>(e0, 0);
  330. registry.assign<char>(e0, 'c');
  331. const auto e1 = registry.create();
  332. registry.assign<int>(e1, 0);
  333. const auto e2 = registry.create();
  334. registry.assign<int>(e2, 0);
  335. registry.assign<char>(e2, 'c');
  336. decltype(view)::size_type cnt{0};
  337. view.each([&cnt](auto...) { ++cnt; });
  338. ASSERT_EQ(cnt, decltype(view)::size_type{2});
  339. }
  340. TEST(DefaultRegistry, RawViews) {
  341. entt::DefaultRegistry registry;
  342. auto view = registry.view<int>(entt::raw_t{});
  343. const auto e0 = registry.create();
  344. registry.assign<int>(e0, 0);
  345. registry.assign<char>(e0, 'c');
  346. const auto e1 = registry.create();
  347. registry.assign<int>(e1, 0);
  348. registry.assign<char>(e1, 'c');
  349. decltype(view)::size_type cnt{0};
  350. view.each([&cnt](auto &...) { ++cnt; });
  351. ASSERT_EQ(cnt, decltype(view)::size_type{2});
  352. }
  353. TEST(DefaultRegistry, CleanStandardViewsAfterReset) {
  354. entt::DefaultRegistry registry;
  355. auto view = registry.view<int>();
  356. registry.assign<int>(registry.create(), 0);
  357. ASSERT_EQ(view.size(), entt::DefaultRegistry::size_type{1});
  358. registry.reset();
  359. ASSERT_EQ(view.size(), entt::DefaultRegistry::size_type{0});
  360. }
  361. TEST(DefaultRegistry, CleanPersistentViewsAfterReset) {
  362. entt::DefaultRegistry registry;
  363. auto view = registry.view<int, char>(entt::persistent_t{});
  364. const auto entity = registry.create();
  365. registry.assign<int>(entity, 0);
  366. registry.assign<char>(entity, 'c');
  367. ASSERT_EQ(view.size(), entt::DefaultRegistry::size_type{1});
  368. registry.reset();
  369. ASSERT_EQ(view.size(), entt::DefaultRegistry::size_type{0});
  370. }
  371. TEST(DefaultRegistry, CleanRawViewsAfterReset) {
  372. entt::DefaultRegistry registry;
  373. auto view = registry.view<int>(entt::raw_t{});
  374. registry.assign<int>(registry.create(), 0);
  375. ASSERT_EQ(view.size(), entt::DefaultRegistry::size_type{1});
  376. registry.reset();
  377. ASSERT_EQ(view.size(), entt::DefaultRegistry::size_type{0});
  378. }
  379. TEST(DefaultRegistry, CleanTagsAfterReset) {
  380. entt::DefaultRegistry registry;
  381. const auto entity = registry.create();
  382. registry.assign<int>(entt::tag_t{}, entity);
  383. ASSERT_TRUE(registry.has<int>());
  384. registry.reset();
  385. ASSERT_FALSE(registry.has<int>());
  386. }
  387. TEST(DefaultRegistry, SortSingle) {
  388. entt::DefaultRegistry registry;
  389. int val = 0;
  390. registry.assign<int>(registry.create(), val++);
  391. registry.assign<int>(registry.create(), val++);
  392. registry.assign<int>(registry.create(), val++);
  393. for(auto entity: registry.view<int>()) {
  394. ASSERT_EQ(registry.get<int>(entity), --val);
  395. }
  396. registry.sort<int>(std::less<int>{});
  397. for(auto entity: registry.view<int>()) {
  398. ASSERT_EQ(registry.get<int>(entity), val++);
  399. }
  400. }
  401. TEST(DefaultRegistry, SortMulti) {
  402. entt::DefaultRegistry registry;
  403. unsigned int uval = 0u;
  404. int ival = 0;
  405. for(auto i = 0; i < 3; ++i) {
  406. const auto entity = registry.create();
  407. registry.assign<unsigned int>(entity, uval++);
  408. registry.assign<int>(entity, ival++);
  409. }
  410. for(auto entity: registry.view<unsigned int>()) {
  411. ASSERT_EQ(registry.get<unsigned int>(entity), --uval);
  412. }
  413. for(auto entity: registry.view<int>()) {
  414. ASSERT_EQ(registry.get<int>(entity), --ival);
  415. }
  416. registry.sort<unsigned int>(std::less<unsigned int>{});
  417. registry.sort<int, unsigned int>();
  418. for(auto entity: registry.view<unsigned int>()) {
  419. ASSERT_EQ(registry.get<unsigned int>(entity), uval++);
  420. }
  421. for(auto entity: registry.view<int>()) {
  422. ASSERT_EQ(registry.get<int>(entity), ival++);
  423. }
  424. }
  425. TEST(DefaultRegistry, ComponentsWithTypesFromStandardTemplateLibrary) {
  426. // see #37 - the test shouldn't crash, that's all
  427. entt::DefaultRegistry registry;
  428. const auto entity = registry.create();
  429. registry.assign<std::unordered_set<int>>(entity).insert(42);
  430. registry.destroy(entity);
  431. }
  432. TEST(DefaultRegistry, ConstructWithComponents) {
  433. // it should compile, that's all
  434. entt::DefaultRegistry registry;
  435. const auto value = 0;
  436. registry.assign<int>(registry.create(), value);
  437. }
  438. TEST(DefaultRegistry, MergeTwoRegistries) {
  439. using entity_type = entt::DefaultRegistry::entity_type;
  440. entt::DefaultRegistry src;
  441. entt::DefaultRegistry dst;
  442. std::unordered_map<entity_type, entity_type> ref;
  443. auto merge = [&ref](const auto &view, auto &dst) {
  444. view.each([&](auto entity, const auto &component) {
  445. if(ref.find(entity) == ref.cend()) {
  446. const auto other = dst.create();
  447. dst.template assign<std::decay_t<decltype(component)>>(other, component);
  448. ref.emplace(entity, other);
  449. } else {
  450. using component_type = std::decay_t<decltype(component)>;
  451. dst.template assign<component_type>(ref[entity], component);
  452. }
  453. });
  454. };
  455. auto e0 = src.create();
  456. src.assign<int>(e0);
  457. src.assign<float>(e0);
  458. src.assign<double>(e0);
  459. auto e1 = src.create();
  460. src.assign<char>(e1);
  461. src.assign<float>(e1);
  462. src.assign<int>(e1);
  463. auto e2 = dst.create();
  464. dst.assign<int>(e2);
  465. dst.assign<char>(e2);
  466. dst.assign<double>(e2);
  467. auto e3 = dst.create();
  468. dst.assign<float>(e3);
  469. dst.assign<int>(e3);
  470. auto eq = [](auto begin, auto end) { ASSERT_EQ(begin, end); };
  471. auto ne = [](auto begin, auto end) { ASSERT_NE(begin, end); };
  472. eq(dst.view<int, float, double>().begin(), dst.view<int, float, double>().end());
  473. eq(dst.view<char, float, int>().begin(), dst.view<char, float, int>().end());
  474. merge(src.view<int>(), dst);
  475. merge(src.view<char>(), dst);
  476. merge(src.view<double>(), dst);
  477. merge(src.view<float>(), dst);
  478. ne(dst.view<int, float, double>().begin(), dst.view<int, float, double>().end());
  479. ne(dst.view<char, float, int>().begin(), dst.view<char, float, int>().end());
  480. }
  481. TEST(DefaultRegistry, ComponentSignals) {
  482. entt::DefaultRegistry registry;
  483. Listener listener;
  484. registry.construction<int>().connect<Listener, &Listener::incrComponent<int>>(&listener);
  485. registry.destruction<int>().connect<Listener, &Listener::decrComponent<int>>(&listener);
  486. auto e0 = registry.create();
  487. auto e1 = registry.create();
  488. registry.assign<int>(e0);
  489. registry.assign<int>(e1);
  490. ASSERT_EQ(listener.counter, 2);
  491. ASSERT_EQ(listener.last, e1);
  492. registry.remove<int>(e0);
  493. ASSERT_EQ(listener.counter, 1);
  494. ASSERT_EQ(listener.last, e0);
  495. registry.destruction<int>().disconnect<Listener, &Listener::decrComponent<int>>(&listener);
  496. registry.remove<int>(e1);
  497. ASSERT_EQ(listener.counter, 1);
  498. ASSERT_EQ(listener.last, e0);
  499. registry.construction<int>().disconnect<Listener, &Listener::incrComponent<int>>(&listener);
  500. registry.assign<int>(e1);
  501. ASSERT_EQ(listener.counter, 1);
  502. ASSERT_EQ(listener.last, e0);
  503. registry.construction<int>().connect<Listener, &Listener::incrComponent<int>>(&listener);
  504. registry.destruction<int>().connect<Listener, &Listener::decrComponent<int>>(&listener);
  505. registry.assign<int>(e0);
  506. registry.reset<int>(e1);
  507. ASSERT_EQ(listener.counter, 1);
  508. ASSERT_EQ(listener.last, e1);
  509. registry.reset<int>();
  510. ASSERT_EQ(listener.counter, 0);
  511. ASSERT_EQ(listener.last, e0);
  512. registry.assign<int>(e0);
  513. registry.assign<int>(e1);
  514. registry.destroy(e1);
  515. ASSERT_EQ(listener.counter, 1);
  516. ASSERT_EQ(listener.last, e1);
  517. }
  518. TEST(DefaultRegistry, TagSignals) {
  519. entt::DefaultRegistry registry;
  520. Listener listener;
  521. registry.construction<int>(entt::tag_t{}).connect<Listener, &Listener::incrTag<int>>(&listener);
  522. registry.destruction<int>(entt::tag_t{}).connect<Listener, &Listener::decrTag<int>>(&listener);
  523. auto e0 = registry.create();
  524. registry.assign<int>(entt::tag_t{}, e0);
  525. ASSERT_EQ(listener.counter, 1);
  526. ASSERT_EQ(listener.last, e0);
  527. auto e1 = registry.create();
  528. registry.move<int>(e1);
  529. registry.remove<int>();
  530. ASSERT_EQ(listener.counter, 0);
  531. ASSERT_EQ(listener.last, e1);
  532. registry.construction<int>(entt::tag_t{}).disconnect<Listener, &Listener::incrTag<int>>(&listener);
  533. registry.destruction<int>(entt::tag_t{}).disconnect<Listener, &Listener::decrTag<int>>(&listener);
  534. registry.assign<int>(entt::tag_t{}, e0);
  535. registry.remove<int>();
  536. ASSERT_EQ(listener.counter, 0);
  537. ASSERT_EQ(listener.last, e1);
  538. registry.construction<int>(entt::tag_t{}).connect<Listener, &Listener::incrTag<int>>(&listener);
  539. registry.destruction<int>(entt::tag_t{}).connect<Listener, &Listener::decrTag<int>>(&listener);
  540. registry.assign<int>(entt::tag_t{}, e0);
  541. registry.destroy(e0);
  542. ASSERT_EQ(listener.counter, 0);
  543. ASSERT_EQ(listener.last, e0);
  544. }