registry.cpp 48 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633
  1. #include <unordered_map>
  2. #include <unordered_set>
  3. #include <functional>
  4. #include <iterator>
  5. #include <memory>
  6. #include <cstdint>
  7. #include <type_traits>
  8. #include <gtest/gtest.h>
  9. #include <entt/core/type_traits.hpp>
  10. #include <entt/entity/registry.hpp>
  11. #include <entt/entity/entity.hpp>
  12. struct empty_type {};
  13. struct listener {
  14. template<typename Component>
  15. void incr(entt::entity entity, entt::registry &registry, const Component &) {
  16. ASSERT_TRUE(registry.valid(entity));
  17. ASSERT_TRUE(registry.has<Component>(entity));
  18. last = entity;
  19. ++counter;
  20. }
  21. template<typename Component>
  22. void decr(entt::entity entity, entt::registry &registry) {
  23. ASSERT_TRUE(registry.valid(entity));
  24. ASSERT_TRUE(registry.has<Component>(entity));
  25. last = entity;
  26. --counter;
  27. }
  28. entt::entity last;
  29. int counter{0};
  30. };
  31. TEST(Registry, Context) {
  32. entt::registry registry;
  33. ASSERT_EQ(registry.try_ctx<char>(), nullptr);
  34. ASSERT_EQ(registry.try_ctx<int>(), nullptr);
  35. ASSERT_EQ(registry.try_ctx<double>(), nullptr);
  36. registry.set<char>();
  37. registry.set<int>();
  38. registry.ctx_or_set<double>();
  39. ASSERT_NE(registry.try_ctx<char>(), nullptr);
  40. ASSERT_NE(registry.try_ctx<int>(), nullptr);
  41. ASSERT_NE(registry.try_ctx<double>(), nullptr);
  42. registry.unset<int>();
  43. registry.unset<double>();
  44. ASSERT_NE(registry.try_ctx<char>(), nullptr);
  45. ASSERT_EQ(registry.try_ctx<int>(), nullptr);
  46. ASSERT_EQ(registry.try_ctx<double>(), nullptr);
  47. registry.set<char>('c');
  48. registry.set<int>(0);
  49. registry.set<double>(1.);
  50. registry.set<int>(42);
  51. ASSERT_EQ(registry.ctx_or_set<char>('a'), 'c');
  52. ASSERT_NE(registry.try_ctx<char>(), nullptr);
  53. ASSERT_EQ(registry.try_ctx<char>(), &registry.ctx<char>());
  54. ASSERT_EQ(registry.ctx<char>(), std::as_const(registry).ctx<char>());
  55. ASSERT_EQ(registry.ctx<int>(), 42);
  56. ASSERT_NE(registry.try_ctx<int>(), nullptr);
  57. ASSERT_EQ(registry.try_ctx<int>(), &registry.ctx<int>());
  58. ASSERT_EQ(registry.ctx<int>(), std::as_const(registry).ctx<int>());
  59. ASSERT_EQ(registry.ctx<double>(), 1.);
  60. ASSERT_NE(registry.try_ctx<double>(), nullptr);
  61. ASSERT_EQ(registry.try_ctx<double>(), &registry.ctx<double>());
  62. ASSERT_EQ(registry.ctx<double>(), std::as_const(registry).ctx<double>());
  63. ASSERT_EQ(registry.try_ctx<float>(), nullptr);
  64. }
  65. TEST(Registry, Functionalities) {
  66. entt::registry registry;
  67. ASSERT_EQ(registry.size(), entt::registry::size_type{0});
  68. ASSERT_EQ(registry.alive(), entt::registry::size_type{0});
  69. ASSERT_NO_THROW((registry.reserve<int, char>(8)));
  70. ASSERT_NO_THROW(registry.reserve(42));
  71. ASSERT_TRUE(registry.empty());
  72. ASSERT_EQ(registry.capacity(), entt::registry::size_type{42});
  73. ASSERT_EQ(registry.capacity<int>(), entt::registry::size_type{8});
  74. ASSERT_EQ(registry.capacity<char>(), entt::registry::size_type{8});
  75. ASSERT_EQ(registry.size<int>(), entt::registry::size_type{0});
  76. ASSERT_EQ(registry.size<char>(), entt::registry::size_type{0});
  77. ASSERT_TRUE((registry.empty<int, char>()));
  78. registry.prepare<double>();
  79. const auto e0 = registry.create();
  80. const auto e1 = registry.create();
  81. registry.assign<int>(e1);
  82. registry.assign<char>(e1);
  83. ASSERT_TRUE(registry.has<>(e0));
  84. ASSERT_TRUE(registry.has<>(e1));
  85. ASSERT_EQ(registry.size<int>(), entt::registry::size_type{1});
  86. ASSERT_EQ(registry.size<char>(), entt::registry::size_type{1});
  87. ASSERT_FALSE(registry.empty<int>());
  88. ASSERT_FALSE(registry.empty<char>());
  89. ASSERT_NE(e0, e1);
  90. ASSERT_FALSE(registry.has<int>(e0));
  91. ASSERT_TRUE(registry.has<int>(e1));
  92. ASSERT_FALSE(registry.has<char>(e0));
  93. ASSERT_TRUE(registry.has<char>(e1));
  94. ASSERT_FALSE((registry.has<int, char>(e0)));
  95. ASSERT_TRUE((registry.has<int, char>(e1)));
  96. ASSERT_EQ(registry.try_get<int>(e0), nullptr);
  97. ASSERT_NE(registry.try_get<int>(e1), nullptr);
  98. ASSERT_EQ(registry.try_get<char>(e0), nullptr);
  99. ASSERT_NE(registry.try_get<char>(e1), nullptr);
  100. ASSERT_EQ(registry.try_get<double>(e0), nullptr);
  101. ASSERT_EQ(registry.try_get<double>(e1), nullptr);
  102. ASSERT_EQ(registry.assign<int>(e0, 42), 42);
  103. ASSERT_EQ(registry.assign<char>(e0, 'c'), 'c');
  104. ASSERT_NO_THROW(registry.remove<int>(e1));
  105. ASSERT_NO_THROW(registry.remove<char>(e1));
  106. ASSERT_TRUE(registry.has<int>(e0));
  107. ASSERT_FALSE(registry.has<int>(e1));
  108. ASSERT_TRUE(registry.has<char>(e0));
  109. ASSERT_FALSE(registry.has<char>(e1));
  110. ASSERT_TRUE((registry.has<int, char>(e0)));
  111. ASSERT_FALSE((registry.has<int, char>(e1)));
  112. const auto e2 = registry.create();
  113. registry.assign_or_replace<int>(e2, registry.get<int>(e0));
  114. registry.assign_or_replace<char>(e2, registry.get<char>(e0));
  115. ASSERT_TRUE(registry.has<int>(e2));
  116. ASSERT_TRUE(registry.has<char>(e2));
  117. ASSERT_EQ(registry.get<int>(e0), 42);
  118. ASSERT_EQ(registry.get<char>(e0), 'c');
  119. ASSERT_NE(registry.try_get<int>(e0), nullptr);
  120. ASSERT_NE(registry.try_get<char>(e0), nullptr);
  121. ASSERT_EQ(registry.try_get<double>(e0), nullptr);
  122. ASSERT_EQ(*registry.try_get<int>(e0), 42);
  123. ASSERT_EQ(*registry.try_get<char>(e0), 'c');
  124. ASSERT_EQ(std::get<0>(registry.get<int, char>(e0)), 42);
  125. ASSERT_EQ(*std::get<0>(registry.try_get<int, char, double>(e0)), 42);
  126. ASSERT_EQ(std::get<1>(static_cast<const entt::registry &>(registry).get<int, char>(e0)), 'c');
  127. ASSERT_EQ(*std::get<1>(static_cast<const entt::registry &>(registry).try_get<int, char, double>(e0)), 'c');
  128. ASSERT_EQ(registry.get<int>(e0), registry.get<int>(e2));
  129. ASSERT_EQ(registry.get<char>(e0), registry.get<char>(e2));
  130. ASSERT_NE(&registry.get<int>(e0), &registry.get<int>(e2));
  131. ASSERT_NE(&registry.get<char>(e0), &registry.get<char>(e2));
  132. ASSERT_NO_THROW(registry.replace<int>(e0, 0));
  133. ASSERT_EQ(registry.get<int>(e0), 0);
  134. ASSERT_NO_THROW(registry.assign_or_replace<int>(e0, 1));
  135. ASSERT_NO_THROW(registry.assign_or_replace<int>(e1, 1));
  136. ASSERT_EQ(static_cast<const entt::registry &>(registry).get<int>(e0), 1);
  137. ASSERT_EQ(static_cast<const entt::registry &>(registry).get<int>(e1), 1);
  138. ASSERT_EQ(registry.size(), entt::registry::size_type{3});
  139. ASSERT_EQ(registry.alive(), entt::registry::size_type{3});
  140. ASSERT_FALSE(registry.empty());
  141. ASSERT_EQ(registry.version(e2), entt::registry::version_type{0});
  142. ASSERT_EQ(registry.current(e2), entt::registry::version_type{0});
  143. ASSERT_NO_THROW(registry.destroy(e2));
  144. ASSERT_EQ(registry.version(e2), entt::registry::version_type{0});
  145. ASSERT_EQ(registry.current(e2), entt::registry::version_type{1});
  146. ASSERT_TRUE(registry.valid(e0));
  147. ASSERT_TRUE(registry.valid(e1));
  148. ASSERT_FALSE(registry.valid(e2));
  149. ASSERT_EQ(registry.size(), entt::registry::size_type{3});
  150. ASSERT_EQ(registry.alive(), entt::registry::size_type{2});
  151. ASSERT_FALSE(registry.empty());
  152. ASSERT_NO_THROW(registry.reset());
  153. ASSERT_EQ(registry.size(), entt::registry::size_type{3});
  154. ASSERT_EQ(registry.alive(), entt::registry::size_type{0});
  155. ASSERT_TRUE(registry.empty());
  156. const auto e3 = registry.create();
  157. ASSERT_EQ(registry.get_or_assign<int>(e3, 3), 3);
  158. ASSERT_EQ(registry.get_or_assign<char>(e3, 'c'), 'c');
  159. ASSERT_EQ(registry.size<int>(), entt::registry::size_type{1});
  160. ASSERT_EQ(registry.size<char>(), entt::registry::size_type{1});
  161. ASSERT_FALSE(registry.empty<int>());
  162. ASSERT_FALSE(registry.empty<char>());
  163. ASSERT_TRUE(registry.has<int>(e3));
  164. ASSERT_TRUE(registry.has<char>(e3));
  165. ASSERT_EQ(registry.get<int>(e3), 3);
  166. ASSERT_EQ(registry.get<char>(e3), 'c');
  167. ASSERT_NO_THROW(registry.reset<int>());
  168. ASSERT_EQ(registry.size<int>(), entt::registry::size_type{0});
  169. ASSERT_EQ(registry.size<char>(), entt::registry::size_type{1});
  170. ASSERT_TRUE(registry.empty<int>());
  171. ASSERT_FALSE(registry.empty<char>());
  172. ASSERT_NO_THROW(registry.reset());
  173. ASSERT_EQ(registry.size<int>(), entt::registry::size_type{0});
  174. ASSERT_EQ(registry.size<char>(), entt::registry::size_type{0});
  175. ASSERT_TRUE((registry.empty<int, char>()));
  176. const auto e4 = registry.create();
  177. const auto e5 = registry.create();
  178. registry.assign<int>(e4);
  179. ASSERT_NO_THROW(registry.reset<int>(e4));
  180. ASSERT_NO_THROW(registry.reset<int>(e5));
  181. ASSERT_EQ(registry.size<int>(), entt::registry::size_type{0});
  182. ASSERT_EQ(registry.size<char>(), entt::registry::size_type{0});
  183. ASSERT_TRUE(registry.empty<int>());
  184. ASSERT_EQ(registry.capacity<int>(), entt::registry::size_type{8});
  185. ASSERT_EQ(registry.capacity<char>(), entt::registry::size_type{8});
  186. registry.shrink_to_fit<int, char>();
  187. ASSERT_EQ(registry.capacity<int>(), entt::registry::size_type{});
  188. ASSERT_EQ(registry.capacity<char>(), entt::registry::size_type{});
  189. }
  190. TEST(Registry, Identifiers) {
  191. entt::registry registry;
  192. const auto pre = registry.create();
  193. ASSERT_EQ(pre, registry.entity(pre));
  194. registry.destroy(pre);
  195. const auto post = registry.create();
  196. ASSERT_NE(pre, post);
  197. ASSERT_EQ(entt::registry::entity(pre), entt::registry::entity(post));
  198. ASSERT_NE(entt::registry::version(pre), entt::registry::version(post));
  199. ASSERT_NE(registry.version(pre), registry.current(pre));
  200. ASSERT_EQ(registry.version(post), registry.current(post));
  201. }
  202. TEST(Registry, RawData) {
  203. entt::registry registry;
  204. const auto entity = registry.create();
  205. ASSERT_EQ(registry.raw<int>(), nullptr);
  206. ASSERT_EQ(std::as_const(registry).raw<int>(), nullptr);
  207. ASSERT_EQ(std::as_const(registry).data<int>(), nullptr);
  208. registry.assign<int>(entity, 42);
  209. ASSERT_NE(registry.raw<int>(), nullptr);
  210. ASSERT_NE(std::as_const(registry).raw<int>(), nullptr);
  211. ASSERT_NE(std::as_const(registry).data<int>(), nullptr);
  212. ASSERT_EQ(*registry.raw<int>(), 42);
  213. ASSERT_EQ(*std::as_const(registry).raw<int>(), 42);
  214. ASSERT_EQ(*std::as_const(registry).data<int>(), entity);
  215. }
  216. TEST(Registry, CreateDestroyCornerCase) {
  217. entt::registry registry;
  218. const auto e0 = registry.create();
  219. const auto e1 = registry.create();
  220. registry.destroy(e0);
  221. registry.destroy(e1);
  222. registry.each([](auto) { FAIL(); });
  223. ASSERT_EQ(registry.current(e0), entt::registry::version_type{1});
  224. ASSERT_EQ(registry.current(e1), entt::registry::version_type{1});
  225. }
  226. TEST(Registry, VersionOverflow) {
  227. entt::registry registry;
  228. const auto entity = registry.create();
  229. registry.destroy(entity);
  230. ASSERT_EQ(registry.version(entity), entt::registry::version_type{});
  231. for(auto i = entt::entt_traits<std::underlying_type_t<entt::entity>>::version_mask; i; --i) {
  232. ASSERT_NE(registry.current(entity), registry.version(entity));
  233. registry.destroy(registry.create());
  234. }
  235. ASSERT_EQ(registry.current(entity), registry.version(entity));
  236. }
  237. TEST(Registry, Each) {
  238. entt::registry registry;
  239. entt::registry::size_type tot;
  240. entt::registry::size_type match;
  241. registry.create();
  242. registry.create<int>();
  243. registry.create();
  244. registry.create<int>();
  245. registry.create();
  246. tot = 0u;
  247. match = 0u;
  248. registry.each([&](auto entity) {
  249. if(registry.has<int>(entity)) { ++match; }
  250. registry.create();
  251. ++tot;
  252. });
  253. ASSERT_EQ(tot, 5u);
  254. ASSERT_EQ(match, 2u);
  255. tot = 0u;
  256. match = 0u;
  257. registry.each([&](auto entity) {
  258. if(registry.has<int>(entity)) {
  259. registry.destroy(entity);
  260. ++match;
  261. }
  262. ++tot;
  263. });
  264. ASSERT_EQ(tot, 10u);
  265. ASSERT_EQ(match, 2u);
  266. tot = 0u;
  267. match = 0u;
  268. registry.each([&](auto entity) {
  269. if(registry.has<int>(entity)) { ++match; }
  270. registry.destroy(entity);
  271. ++tot;
  272. });
  273. ASSERT_EQ(tot, 8u);
  274. ASSERT_EQ(match, 0u);
  275. registry.each([&](auto) { FAIL(); });
  276. }
  277. TEST(Registry, Orphans) {
  278. entt::registry registry;
  279. entt::registry::size_type tot{};
  280. registry.create<int>();
  281. registry.create();
  282. registry.create<int>();
  283. registry.orphans([&](auto) { ++tot; });
  284. ASSERT_EQ(tot, 1u);
  285. tot = {};
  286. registry.each([&](auto entity) { registry.reset<int>(entity); });
  287. registry.orphans([&](auto) { ++tot; });
  288. ASSERT_EQ(tot, 3u);
  289. registry.reset();
  290. tot = {};
  291. registry.orphans([&](auto) { ++tot; });
  292. ASSERT_EQ(tot, 0u);
  293. }
  294. TEST(Registry, CreateDestroyEntities) {
  295. entt::registry registry;
  296. entt::entity pre{}, post{};
  297. for(int i = 0; i < 10; ++i) {
  298. const auto entity = registry.create();
  299. registry.assign<double>(entity);
  300. }
  301. registry.reset();
  302. for(int i = 0; i < 7; ++i) {
  303. const auto entity = registry.create();
  304. registry.assign<int>(entity);
  305. if(i == 3) { pre = entity; }
  306. }
  307. registry.reset();
  308. for(int i = 0; i < 5; ++i) {
  309. const auto entity = registry.create();
  310. if(i == 3) { post = entity; }
  311. }
  312. ASSERT_FALSE(registry.valid(pre));
  313. ASSERT_TRUE(registry.valid(post));
  314. ASSERT_NE(registry.version(pre), registry.version(post));
  315. ASSERT_EQ(registry.version(pre) + 1, registry.version(post));
  316. ASSERT_EQ(registry.current(pre), registry.current(post));
  317. }
  318. TEST(Registry, View) {
  319. entt::registry registry;
  320. auto mview = registry.view<int, char>();
  321. auto iview = registry.view<int>();
  322. auto cview = registry.view<char>();
  323. const auto e0 = registry.create();
  324. registry.assign<int>(e0, 0);
  325. registry.assign<char>(e0, 'c');
  326. const auto e1 = registry.create();
  327. registry.assign<int>(e1, 0);
  328. const auto e2 = registry.create();
  329. registry.assign<int>(e2, 0);
  330. registry.assign<char>(e2, 'c');
  331. ASSERT_EQ(iview.size(), 3u);
  332. ASSERT_EQ(cview.size(), 2u);
  333. std::size_t cnt{};
  334. mview.each([&cnt](auto...) { ++cnt; });
  335. ASSERT_EQ(cnt, 2u);
  336. }
  337. TEST(Registry, NonOwningGroupInitOnFirstUse) {
  338. entt::registry registry;
  339. std::get<2>(registry.create<int, char>()) = 'c';
  340. registry.create<int>();
  341. std::get<2>(registry.create<int, char>()) = 'c';
  342. std::size_t cnt{};
  343. auto group = registry.group<>(entt::get<int, char>);
  344. group.each([&cnt](auto...) { ++cnt; });
  345. ASSERT_TRUE((registry.sortable<int, char>()));
  346. ASSERT_EQ(cnt, 2u);
  347. }
  348. TEST(Registry, NonOwningGroupInitOnAssign) {
  349. entt::registry registry;
  350. auto group = registry.group<>(entt::get<int, char>);
  351. std::get<2>(registry.create<int, char>()) = 'c';
  352. registry.create<int>();
  353. std::get<2>(registry.create<int, char>()) = 'c';
  354. std::size_t cnt{};
  355. group.each([&cnt](auto...) { ++cnt; });
  356. ASSERT_TRUE((registry.sortable<int, char>()));
  357. ASSERT_EQ(cnt, 2u);
  358. }
  359. TEST(Registry, FullOwningGroupInitOnFirstUse) {
  360. entt::registry registry;
  361. std::get<2>(registry.create<int, char>()) = 'c';
  362. registry.create<int>();
  363. std::get<2>(registry.create<int, char>()) = 'c';
  364. std::size_t cnt{};
  365. auto group = registry.group<int, char>();
  366. group.each([&cnt](auto...) { ++cnt; });
  367. ASSERT_FALSE(registry.sortable<int>());
  368. ASSERT_FALSE(registry.sortable<char>());
  369. ASSERT_TRUE(registry.sortable<double>());
  370. ASSERT_EQ(cnt, 2u);
  371. }
  372. TEST(Registry, FullOwningGroupInitOnAssign) {
  373. entt::registry registry;
  374. auto group = registry.group<int, char>();
  375. std::get<2>(registry.create<int, char>()) = 'c';
  376. registry.create<int>();
  377. std::get<2>(registry.create<int, char>()) = 'c';
  378. std::size_t cnt{};
  379. group.each([&cnt](auto...) { ++cnt; });
  380. ASSERT_FALSE(registry.sortable<int>());
  381. ASSERT_FALSE(registry.sortable<char>());
  382. ASSERT_TRUE(registry.sortable<double>());
  383. ASSERT_EQ(cnt, 2u);
  384. }
  385. TEST(Registry, PartialOwningGroupInitOnFirstUse) {
  386. entt::registry registry;
  387. std::get<2>(registry.create<int, char>()) = 'c';
  388. registry.create<int>();
  389. std::get<2>(registry.create<int, char>()) = 'c';
  390. std::size_t cnt{};
  391. auto group = registry.group<int>(entt::get<char>);
  392. group.each([&cnt](auto...) { ++cnt; });
  393. ASSERT_FALSE((registry.sortable<int, char>()));
  394. ASSERT_FALSE(registry.sortable<int>());
  395. ASSERT_TRUE(registry.sortable<char>());
  396. ASSERT_EQ(cnt, 2u);
  397. }
  398. TEST(Registry, PartialOwningGroupInitOnAssign) {
  399. entt::registry registry;
  400. auto group = registry.group<int>(entt::get<char>);
  401. std::get<2>(registry.create<int, char>()) = 'c';
  402. registry.create<int>();
  403. std::get<2>(registry.create<int, char>()) = 'c';
  404. std::size_t cnt{};
  405. group.each([&cnt](auto...) { ++cnt; });
  406. ASSERT_FALSE((registry.sortable<int, char>()));
  407. ASSERT_FALSE(registry.sortable<int>());
  408. ASSERT_TRUE(registry.sortable<char>());
  409. ASSERT_EQ(cnt, 2u);
  410. }
  411. TEST(Registry, CleanViewAfterReset) {
  412. entt::registry registry;
  413. auto view = registry.view<int, char>();
  414. const auto entity = registry.create();
  415. registry.assign<int>(entity, 0);
  416. registry.assign<char>(entity, 'c');
  417. ASSERT_EQ(view.size(), entt::registry::size_type{1});
  418. registry.reset<char>(entity);
  419. ASSERT_EQ(view.size(), entt::registry::size_type{0});
  420. registry.assign<char>(entity, 'c');
  421. ASSERT_EQ(view.size(), entt::registry::size_type{1});
  422. registry.reset<int>();
  423. ASSERT_EQ(view.size(), entt::registry::size_type{0});
  424. registry.assign<int>(entity, 0);
  425. ASSERT_EQ(view.size(), entt::registry::size_type{1});
  426. registry.reset();
  427. ASSERT_EQ(view.size(), entt::registry::size_type{0});
  428. }
  429. TEST(Registry, CleanNonOwningGroupViewAfterReset) {
  430. entt::registry registry;
  431. auto group = registry.group<>(entt::get<int, char>);
  432. const auto entity = registry.create();
  433. registry.assign<int>(entity, 0);
  434. registry.assign<char>(entity, 'c');
  435. ASSERT_EQ(group.size(), entt::registry::size_type{1});
  436. registry.reset<char>(entity);
  437. ASSERT_EQ(group.size(), entt::registry::size_type{0});
  438. registry.assign<char>(entity, 'c');
  439. ASSERT_EQ(group.size(), entt::registry::size_type{1});
  440. registry.reset<int>();
  441. ASSERT_EQ(group.size(), entt::registry::size_type{0});
  442. registry.assign<int>(entity, 0);
  443. ASSERT_EQ(group.size(), entt::registry::size_type{1});
  444. registry.reset();
  445. ASSERT_EQ(group.size(), entt::registry::size_type{0});
  446. }
  447. TEST(Registry, CleanFullOwningGroupViewAfterReset) {
  448. entt::registry registry;
  449. auto group = registry.group<int, char>();
  450. const auto entity = registry.create();
  451. registry.assign<int>(entity, 0);
  452. registry.assign<char>(entity, 'c');
  453. ASSERT_EQ(group.size(), entt::registry::size_type{1});
  454. registry.reset<char>(entity);
  455. ASSERT_EQ(group.size(), entt::registry::size_type{0});
  456. registry.assign<char>(entity, 'c');
  457. ASSERT_EQ(group.size(), entt::registry::size_type{1});
  458. registry.reset<int>();
  459. ASSERT_EQ(group.size(), entt::registry::size_type{0});
  460. registry.assign<int>(entity, 0);
  461. ASSERT_EQ(group.size(), entt::registry::size_type{1});
  462. registry.reset();
  463. ASSERT_EQ(group.size(), entt::registry::size_type{0});
  464. }
  465. TEST(Registry, CleanPartialOwningGroupViewAfterReset) {
  466. entt::registry registry;
  467. auto group = registry.group<int>(entt::get<char>);
  468. const auto entity = registry.create();
  469. registry.assign<int>(entity, 0);
  470. registry.assign<char>(entity, 'c');
  471. ASSERT_EQ(group.size(), entt::registry::size_type{1});
  472. registry.reset<char>(entity);
  473. ASSERT_EQ(group.size(), entt::registry::size_type{0});
  474. registry.assign<char>(entity, 'c');
  475. ASSERT_EQ(group.size(), entt::registry::size_type{1});
  476. registry.reset<int>();
  477. ASSERT_EQ(group.size(), entt::registry::size_type{0});
  478. registry.assign<int>(entity, 0);
  479. ASSERT_EQ(group.size(), entt::registry::size_type{1});
  480. registry.reset();
  481. ASSERT_EQ(group.size(), entt::registry::size_type{0});
  482. }
  483. TEST(Registry, NestedGroups) {
  484. entt::registry registry;
  485. entt::entity entities[10];
  486. registry.create<int, char>(std::begin(entities), std::end(entities));
  487. const auto g1 = registry.group<int>(entt::get<char>, entt::exclude<double>);
  488. ASSERT_TRUE(g1.sortable());
  489. ASSERT_EQ(g1.size(), 10u);
  490. const auto g2 = registry.group<int>(entt::get<char>);
  491. ASSERT_TRUE(g1.sortable());
  492. ASSERT_FALSE(g2.sortable());
  493. ASSERT_EQ(g1.size(), 10u);
  494. ASSERT_EQ(g2.size(), 10u);
  495. for(auto i = 0u; i < 5u; ++i) {
  496. ASSERT_TRUE(g1.contains(entities[i*2+1]));
  497. ASSERT_TRUE(g1.contains(entities[i*2]));
  498. ASSERT_TRUE(g2.contains(entities[i*2+1]));
  499. ASSERT_TRUE(g2.contains(entities[i*2]));
  500. registry.assign<double>(entities[i*2]);
  501. }
  502. ASSERT_EQ(g1.size(), 5u);
  503. ASSERT_EQ(g2.size(), 10u);
  504. for(auto i = 0u; i < 5u; ++i) {
  505. ASSERT_TRUE(g1.contains(entities[i*2+1]));
  506. ASSERT_FALSE(g1.contains(entities[i*2]));
  507. ASSERT_TRUE(g2.contains(entities[i*2+1]));
  508. ASSERT_TRUE(g2.contains(entities[i*2]));
  509. registry.remove<int>(entities[i*2+1]);
  510. }
  511. ASSERT_EQ(g1.size(), 0u);
  512. ASSERT_EQ(g2.size(), 5u);
  513. const auto g3= registry.group<int, float>(entt::get<char>, entt::exclude<double>);
  514. ASSERT_FALSE(g1.sortable());
  515. ASSERT_FALSE(g2.sortable());
  516. ASSERT_TRUE(g3.sortable());
  517. ASSERT_EQ(g1.size(), 0u);
  518. ASSERT_EQ(g2.size(), 5u);
  519. ASSERT_EQ(g3.size(), 0u);
  520. for(auto i = 0u; i < 5u; ++i) {
  521. ASSERT_FALSE(g1.contains(entities[i*2+1]));
  522. ASSERT_FALSE(g1.contains(entities[i*2]));
  523. ASSERT_FALSE(g2.contains(entities[i*2+1]));
  524. ASSERT_TRUE(g2.contains(entities[i*2]));
  525. ASSERT_FALSE(g3.contains(entities[i*2+1]));
  526. ASSERT_FALSE(g3.contains(entities[i*2]));
  527. registry.assign<int>(entities[i*2+1]);
  528. }
  529. ASSERT_EQ(g1.size(), 5u);
  530. ASSERT_EQ(g2.size(), 10u);
  531. ASSERT_EQ(g3.size(), 0u);
  532. for(auto i = 0u; i < 5u; ++i) {
  533. ASSERT_TRUE(g1.contains(entities[i*2+1]));
  534. ASSERT_FALSE(g1.contains(entities[i*2]));
  535. ASSERT_TRUE(g2.contains(entities[i*2+1]));
  536. ASSERT_TRUE(g2.contains(entities[i*2]));
  537. ASSERT_FALSE(g3.contains(entities[i*2+1]));
  538. ASSERT_FALSE(g3.contains(entities[i*2]));
  539. registry.assign<float>(entities[i*2]);
  540. }
  541. ASSERT_EQ(g1.size(), 5u);
  542. ASSERT_EQ(g2.size(), 10u);
  543. ASSERT_EQ(g3.size(), 0u);
  544. for(auto i = 0u; i < 5u; ++i) {
  545. registry.remove<double>(entities[i*2]);
  546. }
  547. ASSERT_EQ(g1.size(), 10u);
  548. ASSERT_EQ(g2.size(), 10u);
  549. ASSERT_EQ(g3.size(), 5u);
  550. for(auto i = 0u; i < 5u; ++i) {
  551. ASSERT_TRUE(g1.contains(entities[i*2+1]));
  552. ASSERT_TRUE(g1.contains(entities[i*2]));
  553. ASSERT_TRUE(g2.contains(entities[i*2+1]));
  554. ASSERT_TRUE(g2.contains(entities[i*2]));
  555. ASSERT_FALSE(g3.contains(entities[i*2+1]));
  556. ASSERT_TRUE(g3.contains(entities[i*2]));
  557. registry.remove<int>(entities[i*2+1]);
  558. registry.remove<int>(entities[i*2]);
  559. }
  560. ASSERT_EQ(g1.size(), 0u);
  561. ASSERT_EQ(g2.size(), 0u);
  562. ASSERT_EQ(g3.size(), 0u);
  563. }
  564. TEST(Registry, SortSingle) {
  565. entt::registry registry;
  566. int val = 0;
  567. registry.assign<int>(registry.create(), val++);
  568. registry.assign<int>(registry.create(), val++);
  569. registry.assign<int>(registry.create(), val++);
  570. for(auto entity: registry.view<int>()) {
  571. ASSERT_EQ(registry.get<int>(entity), --val);
  572. }
  573. registry.sort<int>(std::less<int>{});
  574. for(auto entity: registry.view<int>()) {
  575. ASSERT_EQ(registry.get<int>(entity), val++);
  576. }
  577. }
  578. TEST(Registry, SortMulti) {
  579. entt::registry registry;
  580. unsigned int uval = 0u;
  581. int ival = 0;
  582. for(auto i = 0; i < 3; ++i) {
  583. const auto entity = registry.create();
  584. registry.assign<unsigned int>(entity, uval++);
  585. registry.assign<int>(entity, ival++);
  586. }
  587. for(auto entity: registry.view<unsigned int>()) {
  588. ASSERT_EQ(registry.get<unsigned int>(entity), --uval);
  589. }
  590. for(auto entity: registry.view<int>()) {
  591. ASSERT_EQ(registry.get<int>(entity), --ival);
  592. }
  593. registry.sort<unsigned int>(std::less<unsigned int>{});
  594. registry.sort<int, unsigned int>();
  595. for(auto entity: registry.view<unsigned int>()) {
  596. ASSERT_EQ(registry.get<unsigned int>(entity), uval++);
  597. }
  598. for(auto entity: registry.view<int>()) {
  599. ASSERT_EQ(registry.get<int>(entity), ival++);
  600. }
  601. }
  602. TEST(Registry, SortEmpty) {
  603. entt::registry registry;
  604. registry.assign<empty_type>(registry.create());
  605. registry.assign<empty_type>(registry.create());
  606. registry.assign<empty_type>(registry.create());
  607. ASSERT_LT(registry.data<empty_type>()[0], registry.data<empty_type>()[1]);
  608. ASSERT_LT(registry.data<empty_type>()[1], registry.data<empty_type>()[2]);
  609. registry.sort<empty_type>(std::less<entt::entity>{});
  610. ASSERT_GT(registry.data<empty_type>()[0], registry.data<empty_type>()[1]);
  611. ASSERT_GT(registry.data<empty_type>()[1], registry.data<empty_type>()[2]);
  612. }
  613. TEST(Registry, ComponentsWithTypesFromStandardTemplateLibrary) {
  614. // see #37 - the test shouldn't crash, that's all
  615. entt::registry registry;
  616. const auto entity = registry.create();
  617. registry.assign<std::unordered_set<int>>(entity).insert(42);
  618. registry.destroy(entity);
  619. }
  620. TEST(Registry, ConstructWithComponents) {
  621. // it should compile, that's all
  622. entt::registry registry;
  623. const auto value = 0;
  624. registry.assign<int>(registry.create(), value);
  625. }
  626. TEST(Registry, MergeTwoRegistries) {
  627. entt::registry src;
  628. entt::registry dst;
  629. std::unordered_map<entt::entity, entt::entity> ref;
  630. auto merge = [&ref, &dst](const auto &view) {
  631. view.each([&](auto entity, const auto &component) {
  632. if(ref.find(entity) == ref.cend()) {
  633. const auto other = dst.create();
  634. dst.template assign<std::decay_t<decltype(component)>>(other, component);
  635. ref.emplace(entity, other);
  636. } else {
  637. using component_type = std::decay_t<decltype(component)>;
  638. dst.template assign<component_type>(ref[entity], component);
  639. }
  640. });
  641. };
  642. auto e0 = src.create();
  643. src.assign<int>(e0);
  644. src.assign<float>(e0);
  645. src.assign<double>(e0);
  646. auto e1 = src.create();
  647. src.assign<char>(e1);
  648. src.assign<float>(e1);
  649. src.assign<int>(e1);
  650. auto e2 = dst.create();
  651. dst.assign<int>(e2);
  652. dst.assign<char>(e2);
  653. dst.assign<double>(e2);
  654. auto e3 = dst.create();
  655. dst.assign<float>(e3);
  656. dst.assign<int>(e3);
  657. auto eq = [](auto begin, auto end) { ASSERT_EQ(begin, end); };
  658. auto ne = [](auto begin, auto end) { ASSERT_NE(begin, end); };
  659. eq(dst.view<int, float, double>().begin(), dst.view<int, float, double>().end());
  660. eq(dst.view<char, float, int>().begin(), dst.view<char, float, int>().end());
  661. merge(src.view<int>());
  662. merge(src.view<char>());
  663. merge(src.view<double>());
  664. merge(src.view<float>());
  665. ne(dst.view<int, float, double>().begin(), dst.view<int, float, double>().end());
  666. ne(dst.view<char, float, int>().begin(), dst.view<char, float, int>().end());
  667. }
  668. TEST(Registry, Signals) {
  669. entt::registry registry;
  670. listener listener;
  671. registry.on_construct<empty_type>().connect<&listener::incr<empty_type>>(listener);
  672. registry.on_destroy<empty_type>().connect<&listener::decr<empty_type>>(listener);
  673. registry.on_construct<int>().connect<&listener::incr<int>>(listener);
  674. registry.on_destroy<int>().connect<&listener::decr<int>>(listener);
  675. auto e0 = registry.create();
  676. auto e1 = registry.create();
  677. registry.assign<empty_type>(e0);
  678. registry.assign<empty_type>(e1);
  679. ASSERT_EQ(listener.counter, 2);
  680. ASSERT_EQ(listener.last, e1);
  681. registry.assign<int>(e1);
  682. registry.assign<int>(e0);
  683. ASSERT_EQ(listener.counter, 4);
  684. ASSERT_EQ(listener.last, e0);
  685. registry.remove<empty_type>(e0);
  686. registry.remove<int>(e0);
  687. ASSERT_EQ(listener.counter, 2);
  688. ASSERT_EQ(listener.last, e0);
  689. registry.on_destroy<empty_type>().disconnect<&listener::decr<empty_type>>(listener);
  690. registry.on_destroy<int>().disconnect<&listener::decr<int>>(listener);
  691. registry.remove<empty_type>(e1);
  692. registry.remove<int>(e1);
  693. ASSERT_EQ(listener.counter, 2);
  694. ASSERT_EQ(listener.last, e0);
  695. registry.on_construct<empty_type>().disconnect<&listener::incr<empty_type>>(listener);
  696. registry.on_construct<int>().disconnect<&listener::incr<int>>(listener);
  697. registry.assign<empty_type>(e1);
  698. registry.assign<int>(e1);
  699. ASSERT_EQ(listener.counter, 2);
  700. ASSERT_EQ(listener.last, e0);
  701. registry.on_construct<int>().connect<&listener::incr<int>>(listener);
  702. registry.on_destroy<int>().connect<&listener::decr<int>>(listener);
  703. registry.assign<int>(e0);
  704. registry.reset<int>(e1);
  705. ASSERT_EQ(listener.counter, 2);
  706. ASSERT_EQ(listener.last, e1);
  707. registry.on_construct<empty_type>().connect<&listener::incr<empty_type>>(listener);
  708. registry.on_destroy<empty_type>().connect<&listener::decr<empty_type>>(listener);
  709. registry.reset<empty_type>(e1);
  710. registry.assign<empty_type>(e0);
  711. ASSERT_EQ(listener.counter, 2);
  712. ASSERT_EQ(listener.last, e0);
  713. registry.reset<empty_type>();
  714. registry.reset<int>();
  715. ASSERT_EQ(listener.counter, 0);
  716. ASSERT_EQ(listener.last, e0);
  717. registry.assign<empty_type>(e0);
  718. registry.assign<empty_type>(e1);
  719. registry.assign<int>(e0);
  720. registry.assign<int>(e1);
  721. registry.destroy(e1);
  722. ASSERT_EQ(listener.counter, 2);
  723. ASSERT_EQ(listener.last, e1);
  724. registry.remove<int>(e0);
  725. registry.remove<empty_type>(e0);
  726. registry.assign_or_replace<int>(e0);
  727. registry.assign_or_replace<empty_type>(e0);
  728. ASSERT_EQ(listener.counter, 2);
  729. ASSERT_EQ(listener.last, e0);
  730. registry.on_destroy<empty_type>().disconnect<&listener::decr<empty_type>>(listener);
  731. registry.on_destroy<int>().disconnect<&listener::decr<int>>(listener);
  732. registry.assign_or_replace<empty_type>(e0);
  733. registry.assign_or_replace<int>(e0);
  734. ASSERT_EQ(listener.counter, 2);
  735. ASSERT_EQ(listener.last, e0);
  736. registry.on_replace<empty_type>().connect<&listener::incr<empty_type>>(listener);
  737. registry.on_replace<int>().connect<&listener::incr<int>>(listener);
  738. registry.assign_or_replace<empty_type>(e0);
  739. registry.assign_or_replace<int>(e0);
  740. ASSERT_EQ(listener.counter, 4);
  741. ASSERT_EQ(listener.last, e0);
  742. registry.replace<empty_type>(e0);
  743. registry.replace<int>(e0);
  744. ASSERT_EQ(listener.counter, 6);
  745. ASSERT_EQ(listener.last, e0);
  746. }
  747. TEST(Registry, RangeDestroy) {
  748. entt::registry registry;
  749. const auto e0 = registry.create();
  750. const auto e1 = registry.create();
  751. const auto e2 = registry.create();
  752. registry.assign<int>(e0);
  753. registry.assign<char>(e0);
  754. registry.assign<double>(e0);
  755. registry.assign<int>(e1);
  756. registry.assign<char>(e1);
  757. registry.assign<int>(e2);
  758. ASSERT_TRUE(registry.valid(e0));
  759. ASSERT_TRUE(registry.valid(e1));
  760. ASSERT_TRUE(registry.valid(e2));
  761. {
  762. const auto view = registry.view<int, char>();
  763. registry.destroy(view.begin(), view.end());
  764. }
  765. ASSERT_FALSE(registry.valid(e0));
  766. ASSERT_FALSE(registry.valid(e1));
  767. ASSERT_TRUE(registry.valid(e2));
  768. {
  769. const auto view = registry.view<int>();
  770. registry.destroy(view.begin(), view.end());
  771. }
  772. ASSERT_FALSE(registry.valid(e0));
  773. ASSERT_FALSE(registry.valid(e1));
  774. ASSERT_FALSE(registry.valid(e2));
  775. }
  776. TEST(Registry, RangeAssign) {
  777. entt::registry registry;
  778. const auto e0 = registry.create();
  779. const auto e1 = registry.create();
  780. const auto e2 = registry.create();
  781. registry.assign<int>(e0);
  782. registry.assign<char>(e0);
  783. registry.assign<double>(e0);
  784. registry.assign<int>(e1);
  785. registry.assign<char>(e1);
  786. registry.assign<int>(e2);
  787. ASSERT_FALSE(registry.has<float>(e0));
  788. ASSERT_FALSE(registry.has<float>(e1));
  789. ASSERT_FALSE(registry.has<float>(e2));
  790. const auto view = registry.view<int, char>();
  791. auto it = registry.assign<float>(view.begin(), view.end());
  792. ASSERT_TRUE(registry.has<float>(e0));
  793. ASSERT_TRUE(registry.has<float>(e1));
  794. ASSERT_FALSE(registry.has<float>(e2));
  795. *it = 0.f;
  796. *(it+1) = 1.f;
  797. ASSERT_EQ(registry.get<float>(*view.begin()), 0.f);
  798. ASSERT_EQ(registry.get<float>(*(++view.begin())), 1.f);
  799. }
  800. TEST(Registry, RangeRemove) {
  801. entt::registry registry;
  802. const auto e0 = registry.create();
  803. const auto e1 = registry.create();
  804. const auto e2 = registry.create();
  805. registry.assign<int>(e0);
  806. registry.assign<char>(e0);
  807. registry.assign<double>(e0);
  808. registry.assign<int>(e1);
  809. registry.assign<char>(e1);
  810. registry.assign<int>(e2);
  811. ASSERT_TRUE(registry.has<int>(e0));
  812. ASSERT_TRUE(registry.has<int>(e1));
  813. ASSERT_TRUE(registry.has<int>(e2));
  814. const auto view = registry.view<int, char>();
  815. registry.remove<int>(view.begin(), view.end());
  816. ASSERT_FALSE(registry.has<int>(e0));
  817. ASSERT_FALSE(registry.has<int>(e1));
  818. ASSERT_TRUE(registry.has<int>(e2));
  819. }
  820. TEST(Registry, CreateManyEntitiesAtOnce) {
  821. entt::registry registry;
  822. entt::entity entities[3];
  823. const auto entity = registry.create();
  824. registry.destroy(registry.create());
  825. registry.destroy(entity);
  826. registry.destroy(registry.create());
  827. registry.create(std::begin(entities), std::end(entities));
  828. ASSERT_TRUE(registry.valid(entities[0]));
  829. ASSERT_TRUE(registry.valid(entities[1]));
  830. ASSERT_TRUE(registry.valid(entities[2]));
  831. ASSERT_EQ(registry.entity(entities[0]), entt::entity{0});
  832. ASSERT_EQ(registry.version(entities[0]), entt::registry::version_type{2});
  833. ASSERT_EQ(registry.entity(entities[1]), entt::entity{1});
  834. ASSERT_EQ(registry.version(entities[1]), entt::registry::version_type{1});
  835. ASSERT_EQ(registry.entity(entities[2]), entt::entity{2});
  836. ASSERT_EQ(registry.version(entities[2]), entt::registry::version_type{0});
  837. }
  838. TEST(Registry, CreateAnEntityWithComponents) {
  839. entt::registry registry;
  840. auto &&[entity, ivalue, cvalue, evalue] = registry.create<int, char, empty_type>();
  841. // suppress warnings
  842. (void)evalue;
  843. ASSERT_FALSE(registry.empty<int>());
  844. ASSERT_FALSE(registry.empty<char>());
  845. ASSERT_FALSE(registry.empty<empty_type>());
  846. ASSERT_EQ(registry.size<int>(), entt::registry::size_type{1});
  847. ASSERT_EQ(registry.size<char>(), entt::registry::size_type{1});
  848. ASSERT_EQ(registry.size<empty_type>(), entt::registry::size_type{1});
  849. ASSERT_TRUE((registry.has<int, char, empty_type>(entity)));
  850. ivalue = 42;
  851. cvalue = 'c';
  852. ASSERT_EQ(registry.get<int>(entity), 42);
  853. ASSERT_EQ(registry.get<char>(entity), 'c');
  854. }
  855. TEST(Registry, CreateManyEntitiesWithComponentsAtOnce) {
  856. entt::registry registry;
  857. entt::entity entities[3];
  858. const auto entity = registry.create();
  859. registry.destroy(registry.create());
  860. registry.destroy(entity);
  861. registry.destroy(registry.create());
  862. const auto [iptr, cptr, eptr] = registry.create<int, char, empty_type>(std::begin(entities), std::end(entities));
  863. static_assert(std::is_same_v<typename decltype(eptr)::reference, empty_type>);
  864. ASSERT_FALSE(registry.empty<int>());
  865. ASSERT_FALSE(registry.empty<char>());
  866. ASSERT_FALSE(registry.empty<empty_type>());
  867. ASSERT_EQ(registry.size<int>(), entt::registry::size_type{3});
  868. ASSERT_EQ(registry.size<char>(), entt::registry::size_type{3});
  869. ASSERT_EQ(registry.size<empty_type>(), entt::registry::size_type{3});
  870. ASSERT_TRUE(registry.valid(entities[0]));
  871. ASSERT_TRUE(registry.valid(entities[1]));
  872. ASSERT_TRUE(registry.valid(entities[2]));
  873. ASSERT_EQ(registry.entity(entities[0]), entt::entity{0});
  874. ASSERT_EQ(registry.version(entities[0]), entt::registry::version_type{2});
  875. ASSERT_EQ(registry.entity(entities[1]), entt::entity{1});
  876. ASSERT_EQ(registry.version(entities[1]), entt::registry::version_type{1});
  877. ASSERT_EQ(registry.entity(entities[2]), entt::entity{2});
  878. ASSERT_EQ(registry.version(entities[2]), entt::registry::version_type{0});
  879. ASSERT_TRUE((registry.has<int, char, empty_type>(entities[0])));
  880. ASSERT_TRUE((registry.has<int, char, empty_type>(entities[1])));
  881. ASSERT_TRUE((registry.has<int, char, empty_type>(entities[2])));
  882. for(auto i = 0; i < 3; ++i) {
  883. iptr[i] = i;
  884. cptr[i] = char('a'+i);
  885. }
  886. for(auto i = 0; i < 3; ++i) {
  887. ASSERT_EQ(registry.get<int>(entities[i]), i);
  888. ASSERT_EQ(registry.get<char>(entities[i]), char('a'+i));
  889. }
  890. }
  891. TEST(Registry, CreateManyEntitiesWithComponentsAtOnceWithListener) {
  892. entt::registry registry;
  893. entt::entity entities[3];
  894. listener listener;
  895. registry.on_construct<int>().connect<&listener::incr<int>>(listener);
  896. registry.create<int, char>(std::begin(entities), std::end(entities));
  897. ASSERT_EQ(listener.counter, 3);
  898. registry.on_construct<int>().disconnect<&listener::incr<int>>(listener);
  899. registry.on_construct<empty_type>().connect<&listener::incr<empty_type>>(listener);
  900. registry.create<char, empty_type>(std::begin(entities), std::end(entities));
  901. ASSERT_EQ(listener.counter, 6);
  902. }
  903. TEST(Registry, CreateWithHint) {
  904. entt::registry registry;
  905. auto e3 = registry.create(entt::entity{3});
  906. auto e2 = registry.create(entt::entity{3});
  907. ASSERT_EQ(e2, entt::entity{2});
  908. ASSERT_FALSE(registry.valid(entt::entity{1}));
  909. ASSERT_EQ(e3, entt::entity{3});
  910. registry.destroy(e2);
  911. ASSERT_EQ(registry.version(e2), 0);
  912. ASSERT_EQ(registry.current(e2), 1);
  913. e2 = registry.create();
  914. auto e1 = registry.create(entt::entity{2});
  915. ASSERT_EQ(registry.entity(e2), entt::entity{2});
  916. ASSERT_EQ(registry.version(e2), 1);
  917. ASSERT_EQ(registry.entity(e1), entt::entity{1});
  918. ASSERT_EQ(registry.version(e1), 0);
  919. registry.destroy(e1);
  920. registry.destroy(e2);
  921. auto e0 = std::get<0>(registry.create<int>(entt::entity{0}));
  922. ASSERT_EQ(e0, entt::entity{0});
  923. ASSERT_EQ(registry.version(e0), 0);
  924. ASSERT_TRUE(registry.has<int>(e0));
  925. ASSERT_FALSE(registry.has<char>(e0));
  926. }
  927. TEST(Registry, NonOwningGroupInterleaved) {
  928. entt::registry registry;
  929. typename entt::entity entity = entt::null;
  930. entity = registry.create();
  931. registry.assign<int>(entity);
  932. registry.assign<char>(entity);
  933. const auto group = registry.group<>(entt::get<int, char>);
  934. entity = registry.create();
  935. registry.assign<int>(entity);
  936. registry.assign<char>(entity);
  937. std::size_t cnt{};
  938. group.each([&cnt](auto...) { ++cnt; });
  939. ASSERT_EQ(cnt, 2u);
  940. }
  941. TEST(Registry, FullOwningGroupInterleaved) {
  942. entt::registry registry;
  943. typename entt::entity entity = entt::null;
  944. entity = registry.create();
  945. registry.assign<int>(entity);
  946. registry.assign<char>(entity);
  947. const auto group = registry.group<int, char>();
  948. entity = registry.create();
  949. registry.assign<int>(entity);
  950. registry.assign<char>(entity);
  951. std::size_t cnt{};
  952. group.each([&cnt](auto...) { ++cnt; });
  953. ASSERT_EQ(cnt, 2u);
  954. }
  955. TEST(Registry, PartialOwningGroupInterleaved) {
  956. entt::registry registry;
  957. typename entt::entity entity = entt::null;
  958. entity = registry.create();
  959. registry.assign<int>(entity);
  960. registry.assign<char>(entity);
  961. const auto group = registry.group<int>(entt::get<char>);
  962. entity = registry.create();
  963. registry.assign<int>(entity);
  964. registry.assign<char>(entity);
  965. std::size_t cnt{};
  966. group.each([&cnt](auto...) { ++cnt; });
  967. ASSERT_EQ(cnt, 2u);
  968. }
  969. TEST(Registry, NonOwningGroupSortInterleaved) {
  970. entt::registry registry;
  971. const auto group = registry.group<>(entt::get<int, char>);
  972. const auto e0 = registry.create();
  973. registry.assign<int>(e0, 0);
  974. registry.assign<char>(e0, '0');
  975. const auto e1 = registry.create();
  976. registry.assign<int>(e1, 1);
  977. registry.assign<char>(e1, '1');
  978. registry.sort<int>(std::greater{});
  979. registry.sort<char>(std::less{});
  980. const auto e2 = registry.create();
  981. registry.assign<int>(e2, 2);
  982. registry.assign<char>(e2, '2');
  983. group.each([e0, e1, e2](const auto entity, const auto &i, const auto &c) {
  984. if(entity == e0) {
  985. ASSERT_EQ(i, 0);
  986. ASSERT_EQ(c, '0');
  987. } else if(entity == e1) {
  988. ASSERT_EQ(i, 1);
  989. ASSERT_EQ(c, '1');
  990. } else if(entity == e2) {
  991. ASSERT_EQ(i, 2);
  992. ASSERT_EQ(c, '2');
  993. }
  994. });
  995. }
  996. TEST(Registry, Clone) {
  997. entt::registry registry;
  998. entt::registry other;
  999. registry.destroy(registry.create());
  1000. const auto e0 = registry.create();
  1001. registry.assign<int>(e0, 0);
  1002. registry.assign<double>(e0, 0.0);
  1003. const auto e1 = registry.create();
  1004. registry.assign<int>(e1, 1);
  1005. registry.assign<char>(e1, '1');
  1006. registry.assign<double>(e1, 1.1);
  1007. const auto e2 = registry.create();
  1008. registry.assign<int>(e2, 2);
  1009. registry.assign<char>(e2, '2');
  1010. registry.destroy(e1);
  1011. ASSERT_EQ((other.group<int, char>().size()), entt::registry::size_type{0});
  1012. other = registry.clone<int, char, float>();
  1013. ASSERT_EQ((other.group<int, char>().size()), entt::registry::size_type{1});
  1014. ASSERT_EQ(other.size(), registry.size());
  1015. ASSERT_EQ(other.alive(), registry.alive());
  1016. ASSERT_TRUE(other.valid(e0));
  1017. ASSERT_FALSE(other.valid(e1));
  1018. ASSERT_TRUE(other.valid(e2));
  1019. ASSERT_TRUE((other.has<int>(e0)));
  1020. ASSERT_FALSE((other.has<double>(e0)));
  1021. ASSERT_TRUE((other.has<int, char>(e2)));
  1022. ASSERT_EQ(other.get<int>(e0), 0);
  1023. ASSERT_EQ(other.get<int>(e2), 2);
  1024. ASSERT_EQ(other.get<char>(e2), '2');
  1025. const auto e3 = other.create();
  1026. ASSERT_NE(e1, e3);
  1027. ASSERT_EQ(registry.entity(e1), registry.entity(e3));
  1028. ASSERT_EQ(other.entity(e1), other.entity(e3));
  1029. other.assign<int>(e3, 3);
  1030. other.assign<char>(e3, '3');
  1031. ASSERT_EQ((registry.group<int, char>().size()), entt::registry::size_type{1});
  1032. ASSERT_EQ((other.group<int, char>().size()), entt::registry::size_type{2});
  1033. other = registry.clone();
  1034. ASSERT_EQ(other.size(), registry.size());
  1035. ASSERT_EQ(other.alive(), registry.alive());
  1036. ASSERT_TRUE(other.valid(e0));
  1037. ASSERT_FALSE(other.valid(e1));
  1038. ASSERT_TRUE(other.valid(e2));
  1039. ASSERT_FALSE(other.valid(e3));
  1040. ASSERT_TRUE((other.has<int, double>(e0)));
  1041. ASSERT_TRUE((other.has<int, char>(e2)));
  1042. ASSERT_EQ(other.get<int>(e0), 0);
  1043. ASSERT_EQ(other.get<double>(e0), 0.);
  1044. ASSERT_EQ(other.get<int>(e2), 2);
  1045. ASSERT_EQ(other.get<char>(e2), '2');
  1046. other = other.clone<char>();
  1047. ASSERT_EQ(other.size(), registry.size());
  1048. ASSERT_EQ(other.alive(), registry.alive());
  1049. ASSERT_TRUE(other.valid(e0));
  1050. ASSERT_FALSE(other.valid(e1));
  1051. ASSERT_TRUE(other.valid(e2));
  1052. ASSERT_FALSE(other.valid(e3));
  1053. ASSERT_FALSE((other.has<int>(e0)));
  1054. ASSERT_FALSE((other.has<double>(e0)));
  1055. ASSERT_FALSE((other.has<int>(e2)));
  1056. ASSERT_TRUE((other.has<char>(e2)));
  1057. ASSERT_TRUE(other.orphan(e0));
  1058. ASSERT_EQ(other.get<char>(e2), '2');
  1059. // the remove erased function must be available after cloning
  1060. other.reset();
  1061. }
  1062. TEST(Registry, CloneExclude) {
  1063. entt::registry registry;
  1064. entt::registry other;
  1065. const auto entity = registry.create();
  1066. registry.assign<int>(entity);
  1067. registry.assign<char>(entity);
  1068. other = registry.clone<int>();
  1069. ASSERT_TRUE(other.has(entity));
  1070. ASSERT_TRUE(other.has<int>(entity));
  1071. ASSERT_FALSE(other.has<char>(entity));
  1072. other = registry.clone(entt::exclude<int>);
  1073. ASSERT_TRUE(other.has(entity));
  1074. ASSERT_FALSE(other.has<int>(entity));
  1075. ASSERT_TRUE(other.has<char>(entity));
  1076. other = registry.clone(entt::exclude<int, char>);
  1077. ASSERT_TRUE(other.has(entity));
  1078. ASSERT_TRUE(other.orphan(entity));
  1079. }
  1080. TEST(Registry, CloneMoveOnlyComponent) {
  1081. entt::registry registry;
  1082. const auto entity = registry.create();
  1083. registry.assign<std::unique_ptr<int>>(entity);
  1084. registry.assign<char>(entity);
  1085. auto other = registry.clone();
  1086. ASSERT_TRUE(other.valid(entity));
  1087. ASSERT_TRUE(other.has<char>(entity));
  1088. ASSERT_FALSE(other.has<std::unique_ptr<int>>(entity));
  1089. }
  1090. TEST(Registry, Stamp) {
  1091. entt::registry registry;
  1092. const auto prototype = registry.create();
  1093. registry.assign<int>(prototype, 3);
  1094. registry.assign<char>(prototype, 'c');
  1095. auto entity = registry.create();
  1096. registry.stamp<int, char>(entity, registry, prototype);
  1097. ASSERT_TRUE((registry.has<int, char>(entity)));
  1098. ASSERT_EQ(registry.get<int>(entity), 3);
  1099. ASSERT_EQ(registry.get<char>(entity), 'c');
  1100. registry.replace<int>(prototype, 42);
  1101. registry.replace<char>(prototype, 'a');
  1102. registry.stamp<int>(entity, registry, prototype);
  1103. ASSERT_EQ(registry.get<int>(entity), 42);
  1104. ASSERT_EQ(registry.get<char>(entity), 'c');
  1105. }
  1106. TEST(Registry, StampExclude) {
  1107. entt::registry registry;
  1108. const auto prototype = registry.create();
  1109. registry.assign<int>(prototype, 3);
  1110. registry.assign<char>(prototype, 'c');
  1111. registry.assign<empty_type>(prototype);
  1112. const auto entity = registry.create();
  1113. registry.stamp(entity, registry, prototype, entt::exclude<char>);
  1114. ASSERT_TRUE((registry.has<int, empty_type>(entity)));
  1115. ASSERT_FALSE(registry.has<char>(entity));
  1116. ASSERT_EQ(registry.get<int>(entity), 3);
  1117. registry.replace<int>(prototype, 42);
  1118. registry.stamp(entity, registry, prototype, entt::exclude<int>);
  1119. ASSERT_TRUE((registry.has<int, char, empty_type>(entity)));
  1120. ASSERT_EQ(registry.get<int>(entity), 3);
  1121. ASSERT_EQ(registry.get<char>(entity), 'c');
  1122. registry.remove<int, char, empty_type>(entity);
  1123. registry.stamp(entity, registry, prototype, entt::exclude<int, char, empty_type>);
  1124. ASSERT_TRUE(registry.orphan(entity));
  1125. }
  1126. TEST(Registry, StampMoveOnlyComponent) {
  1127. entt::registry registry;
  1128. const auto prototype = registry.create();
  1129. registry.assign<std::unique_ptr<int>>(prototype);
  1130. registry.assign<char>(prototype);
  1131. const auto entity = registry.create();
  1132. registry.stamp(entity, registry, prototype);
  1133. ASSERT_TRUE(registry.has<char>(entity));
  1134. ASSERT_FALSE(registry.has<std::unique_ptr<int>>(entity));
  1135. }
  1136. TEST(Registry, GetOrAssign) {
  1137. entt::registry registry;
  1138. const auto entity = registry.create();
  1139. const auto value = registry.get_or_assign<int>(entity, 3);
  1140. ASSERT_TRUE(registry.has<int>(entity));
  1141. ASSERT_EQ(registry.get<int>(entity), value);
  1142. ASSERT_EQ(registry.get<int>(entity), 3);
  1143. }
  1144. TEST(Registry, Constness) {
  1145. entt::registry registry;
  1146. ASSERT_TRUE((std::is_same_v<decltype(registry.assign<int>({})), int &>));
  1147. ASSERT_TRUE((std::is_same_v<decltype(registry.assign<empty_type>({})), empty_type>));
  1148. ASSERT_TRUE((std::is_same_v<decltype(registry.get<int>({})), int &>));
  1149. ASSERT_TRUE((std::is_same_v<decltype(registry.get<int, char>({})), std::tuple<int &, char &>>));
  1150. ASSERT_TRUE((std::is_same_v<decltype(registry.try_get<int>({})), int *>));
  1151. ASSERT_TRUE((std::is_same_v<decltype(registry.try_get<int, char>({})), std::tuple<int *, char *>>));
  1152. ASSERT_TRUE((std::is_same_v<decltype(std::as_const(registry).get<int>({})), const int &>));
  1153. ASSERT_TRUE((std::is_same_v<decltype(std::as_const(registry).get<int, char>({})), std::tuple<const int &, const char &>>));
  1154. ASSERT_TRUE((std::is_same_v<decltype(std::as_const(registry).try_get<int>({})), const int *>));
  1155. ASSERT_TRUE((std::is_same_v<decltype(std::as_const(registry).try_get<int, char>({})), std::tuple<const int *, const char *>>));
  1156. }
  1157. TEST(Registry, BatchCreateAmbiguousCall) {
  1158. struct ambiguous { std::uint32_t foo; std::uint64_t bar; };
  1159. entt::registry registry;
  1160. const auto entity = registry.create();
  1161. std::uint32_t foo = 32u;
  1162. std::uint64_t bar = 64u;
  1163. // this should work, no other tests required
  1164. registry.assign<ambiguous>(entity, foo, bar);
  1165. }
  1166. TEST(Registry, MoveOnlyComponent) {
  1167. // the purpose is to ensure that move only components are always accepted
  1168. entt::registry registry;
  1169. const auto entity = registry.create();
  1170. registry.assign<std::unique_ptr<int>>(entity);
  1171. }
  1172. TEST(Registry, Dependencies) {
  1173. entt::registry registry;
  1174. const auto entity = registry.create();
  1175. // required because of an issue of VS2019
  1176. constexpr auto assign_or_replace = &entt::registry::assign_or_replace<double>;
  1177. constexpr auto remove = &entt::registry::remove<double>;
  1178. registry.on_construct<int>().connect<assign_or_replace>(registry);
  1179. registry.on_destroy<int>().connect<remove>(registry);
  1180. registry.assign<double>(entity, .3);
  1181. ASSERT_FALSE(registry.has<int>(entity));
  1182. ASSERT_EQ(registry.get<double>(entity), .3);
  1183. registry.assign<int>(entity);
  1184. ASSERT_TRUE(registry.has<int>(entity));
  1185. ASSERT_EQ(registry.get<double>(entity), .0);
  1186. registry.remove<int>(entity);
  1187. ASSERT_FALSE(registry.has<int>(entity));
  1188. ASSERT_FALSE(registry.has<double>(entity));
  1189. registry.on_construct<int>().disconnect<assign_or_replace>(registry);
  1190. registry.on_destroy<int>().disconnect<remove>(registry);
  1191. registry.assign<int>(entity);
  1192. ASSERT_TRUE(registry.has<int>(entity));
  1193. ASSERT_FALSE(registry.has<double>(entity));
  1194. }