registry.cpp 42 KB

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