group.cpp 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122
  1. #include <utility>
  2. #include <iterator>
  3. #include <algorithm>
  4. #include <type_traits>
  5. #include <gtest/gtest.h>
  6. #include <entt/entity/helper.hpp>
  7. #include <entt/entity/registry.hpp>
  8. #include <entt/entity/group.hpp>
  9. struct empty_type {};
  10. struct boxed_int { int value; };
  11. bool operator==(const boxed_int &lhs, const boxed_int &rhs) {
  12. return lhs.value == rhs.value;
  13. }
  14. TEST(NonOwningGroup, Functionalities) {
  15. entt::registry registry;
  16. auto group = registry.group(entt::get<int, char>);
  17. auto cgroup = std::as_const(registry).group(entt::get<const int, const char>);
  18. ASSERT_TRUE(group.empty());
  19. ASSERT_TRUE((group.empty<int, char>()));
  20. ASSERT_TRUE((cgroup.empty<const int, const char>()));
  21. const auto e0 = registry.create();
  22. registry.assign<char>(e0);
  23. const auto e1 = registry.create();
  24. registry.assign<int>(e1);
  25. registry.assign<char>(e1);
  26. ASSERT_FALSE(group.empty());
  27. ASSERT_FALSE((group.empty<int>()));
  28. ASSERT_FALSE((cgroup.empty<const char>()));
  29. ASSERT_NO_THROW((group.begin()++));
  30. ASSERT_NO_THROW((++cgroup.begin()));
  31. ASSERT_NE(group.begin(), group.end());
  32. ASSERT_NE(cgroup.begin(), cgroup.end());
  33. ASSERT_EQ(group.size(), typename decltype(group)::size_type{1});
  34. ASSERT_EQ(group.size<int>(), typename decltype(group)::size_type{1});
  35. ASSERT_EQ(cgroup.size<const char>(), typename decltype(group)::size_type{2});
  36. registry.assign<int>(e0);
  37. ASSERT_EQ(group.size(), typename decltype(group)::size_type{2});
  38. ASSERT_EQ(group.size<int>(), typename decltype(group)::size_type{2});
  39. ASSERT_EQ(cgroup.size<const char>(), typename decltype(group)::size_type{2});
  40. registry.remove<int>(e0);
  41. ASSERT_EQ(group.size(), typename decltype(group)::size_type{1});
  42. ASSERT_EQ(group.size<int>(), typename decltype(group)::size_type{1});
  43. ASSERT_EQ(cgroup.size<const char>(), typename decltype(group)::size_type{2});
  44. registry.get<char>(e0) = '1';
  45. registry.get<char>(e1) = '2';
  46. registry.get<int>(e1) = 42;
  47. for(auto entity: group) {
  48. ASSERT_EQ(std::get<0>(cgroup.get<const int, const char>(entity)), 42);
  49. ASSERT_EQ(std::get<1>(group.get<int, char>(entity)), '2');
  50. ASSERT_EQ(cgroup.get<const char>(entity), '2');
  51. }
  52. ASSERT_EQ(*(group.data() + 0), e1);
  53. ASSERT_EQ(*(group.data<int>() + 0), e1);
  54. ASSERT_EQ(*(group.data<char>() + 0), e0);
  55. ASSERT_EQ(*(cgroup.data<const char>() + 1), e1);
  56. ASSERT_EQ(*(group.raw<int>() + 0), 42);
  57. ASSERT_EQ(*(group.raw<char>() + 0), '1');
  58. ASSERT_EQ(*(cgroup.raw<const char>() + 1), '2');
  59. registry.remove<char>(e0);
  60. registry.remove<char>(e1);
  61. ASSERT_EQ(group.begin(), group.end());
  62. ASSERT_EQ(cgroup.begin(), cgroup.end());
  63. ASSERT_TRUE(group.empty());
  64. ASSERT_TRUE(group.capacity());
  65. group.shrink_to_fit();
  66. ASSERT_FALSE(group.capacity());
  67. }
  68. TEST(NonOwningGroup, ElementAccess) {
  69. entt::registry registry;
  70. auto group = registry.group(entt::get<int, char>);
  71. auto cgroup = std::as_const(registry).group(entt::get<const int, const char>);
  72. const auto e0 = registry.create();
  73. registry.assign<int>(e0);
  74. registry.assign<char>(e0);
  75. const auto e1 = registry.create();
  76. registry.assign<int>(e1);
  77. registry.assign<char>(e1);
  78. for(typename decltype(group)::size_type i{}; i < group.size(); ++i) {
  79. ASSERT_EQ(group[i], i ? e0 : e1);
  80. ASSERT_EQ(cgroup[i], i ? e0 : e1);
  81. }
  82. }
  83. TEST(NonOwningGroup, Contains) {
  84. entt::registry registry;
  85. auto group = registry.group(entt::get<int, char>);
  86. const auto e0 = registry.create();
  87. registry.assign<int>(e0);
  88. registry.assign<char>(e0);
  89. const auto e1 = registry.create();
  90. registry.assign<int>(e1);
  91. registry.assign<char>(e1);
  92. registry.destroy(e0);
  93. ASSERT_FALSE(group.contains(e0));
  94. ASSERT_TRUE(group.contains(e1));
  95. }
  96. TEST(NonOwningGroup, Empty) {
  97. entt::registry registry;
  98. const auto e0 = registry.create();
  99. registry.assign<double>(e0);
  100. registry.assign<int>(e0);
  101. registry.assign<float>(e0);
  102. const auto e1 = registry.create();
  103. registry.assign<char>(e1);
  104. registry.assign<float>(e1);
  105. ASSERT_TRUE(registry.group(entt::get<char, int, float>).empty());
  106. ASSERT_TRUE(registry.group(entt::get<double, char, int, float>).empty());
  107. }
  108. TEST(NonOwningGroup, Each) {
  109. entt::registry registry;
  110. auto group = registry.group(entt::get<int, char>);
  111. const auto e0 = registry.create();
  112. registry.assign<int>(e0);
  113. registry.assign<char>(e0);
  114. const auto e1 = registry.create();
  115. registry.assign<int>(e1);
  116. registry.assign<char>(e1);
  117. auto cgroup = std::as_const(registry).group(entt::get<const int, const char>);
  118. std::size_t cnt = 0;
  119. group.each([&cnt](auto, int &, char &) { ++cnt; });
  120. group.each([&cnt](int &, char &) { ++cnt; });
  121. ASSERT_EQ(cnt, std::size_t{4});
  122. cgroup.each([&cnt](auto, const int &, const char &) { --cnt; });
  123. cgroup.each([&cnt](const int &, const char &) { --cnt; });
  124. ASSERT_EQ(cnt, std::size_t{0});
  125. }
  126. TEST(NonOwningGroup, Sort) {
  127. entt::registry registry;
  128. auto group = registry.group(entt::get<const int, unsigned int>);
  129. const auto e0 = registry.create();
  130. const auto e1 = registry.create();
  131. const auto e2 = registry.create();
  132. const auto e3 = registry.create();
  133. registry.assign<unsigned int>(e0, 0u);
  134. registry.assign<unsigned int>(e1, 1u);
  135. registry.assign<unsigned int>(e2, 2u);
  136. registry.assign<unsigned int>(e3, 3u);
  137. registry.assign<int>(e0, 0);
  138. registry.assign<int>(e1, 1);
  139. registry.assign<int>(e2, 2);
  140. ASSERT_EQ(*(group.raw<unsigned int>() + 0u), 0u);
  141. ASSERT_EQ(*(group.raw<unsigned int>() + 1u), 1u);
  142. ASSERT_EQ(*(group.raw<unsigned int>() + 2u), 2u);
  143. ASSERT_EQ(*(group.raw<const int>() + 0u), 0);
  144. ASSERT_EQ(*(group.raw<const int>() + 1u), 1);
  145. ASSERT_EQ(*(group.raw<const int>() + 2u), 2);
  146. ASSERT_EQ(*(group.data() + 0u), e0);
  147. ASSERT_EQ(*(group.data() + 1u), e1);
  148. ASSERT_EQ(*(group.data() + 2u), e2);
  149. group.sort([](const entt::entity lhs, const entt::entity rhs) {
  150. return std::underlying_type_t<entt::entity>(lhs) < std::underlying_type_t<entt::entity>(rhs);
  151. });
  152. ASSERT_EQ(*(group.raw<unsigned int>() + 0u), 0u);
  153. ASSERT_EQ(*(group.raw<unsigned int>() + 1u), 1u);
  154. ASSERT_EQ(*(group.raw<unsigned int>() + 2u), 2u);
  155. ASSERT_EQ(*(group.raw<const int>() + 0u), 0);
  156. ASSERT_EQ(*(group.raw<const int>() + 1u), 1);
  157. ASSERT_EQ(*(group.raw<const int>() + 2u), 2);
  158. ASSERT_EQ(*(group.data() + 0u), e2);
  159. ASSERT_EQ(*(group.data() + 1u), e1);
  160. ASSERT_EQ(*(group.data() + 2u), e0);
  161. ASSERT_EQ((group.get<const int, unsigned int>(e0)), (std::make_tuple(0, 0u)));
  162. ASSERT_EQ((group.get<const int, unsigned int>(e1)), (std::make_tuple(1, 1u)));
  163. ASSERT_EQ((group.get<const int, unsigned int>(e2)), (std::make_tuple(2, 2u)));
  164. ASSERT_FALSE(group.contains(e3));
  165. group.sort<const int>([](const int lhs, const int rhs) {
  166. return lhs > rhs;
  167. });
  168. ASSERT_EQ(*(group.data() + 0u), e0);
  169. ASSERT_EQ(*(group.data() + 1u), e1);
  170. ASSERT_EQ(*(group.data() + 2u), e2);
  171. ASSERT_EQ((group.get<const int, unsigned int>(e0)), (std::make_tuple(0, 0u)));
  172. ASSERT_EQ((group.get<const int, unsigned int>(e1)), (std::make_tuple(1, 1u)));
  173. ASSERT_EQ((group.get<const int, unsigned int>(e2)), (std::make_tuple(2, 2u)));
  174. ASSERT_FALSE(group.contains(e3));
  175. }
  176. TEST(NonOwningGroup, SortAsAPool) {
  177. entt::registry registry;
  178. auto group = registry.group(entt::get<const int, unsigned int>);
  179. const auto e0 = registry.create();
  180. const auto e1 = registry.create();
  181. const auto e2 = registry.create();
  182. const auto e3 = registry.create();
  183. auto uval = 0u;
  184. auto ival = 0;
  185. registry.assign<unsigned int>(e0, uval++);
  186. registry.assign<unsigned int>(e1, uval++);
  187. registry.assign<unsigned int>(e2, uval++);
  188. registry.assign<unsigned int>(e3, uval+1);
  189. registry.assign<int>(e0, ival++);
  190. registry.assign<int>(e1, ival++);
  191. registry.assign<int>(e2, ival++);
  192. for(auto entity: group) {
  193. ASSERT_EQ(group.get<unsigned int>(entity), --uval);
  194. ASSERT_EQ(group.get<const int>(entity), --ival);
  195. }
  196. registry.sort<unsigned int>(std::less<unsigned int>{});
  197. group.sort<unsigned int>();
  198. ASSERT_EQ((group.get<const int, unsigned int>(e0)), (std::make_tuple(0, 0u)));
  199. ASSERT_EQ((group.get<const int, unsigned int>(e1)), (std::make_tuple(1, 1u)));
  200. ASSERT_EQ((group.get<const int, unsigned int>(e2)), (std::make_tuple(2, 2u)));
  201. ASSERT_FALSE(group.contains(e3));
  202. for(auto entity: group) {
  203. ASSERT_EQ(group.get<unsigned int>(entity), uval++);
  204. ASSERT_EQ(group.get<const int>(entity), ival++);
  205. }
  206. }
  207. TEST(NonOwningGroup, IndexRebuiltOnDestroy) {
  208. entt::registry registry;
  209. auto group = registry.group(entt::get<int, unsigned int>);
  210. const auto e0 = registry.create();
  211. const auto e1 = registry.create();
  212. registry.assign<unsigned int>(e0, 0u);
  213. registry.assign<unsigned int>(e1, 1u);
  214. registry.assign<int>(e0, 0);
  215. registry.assign<int>(e1, 1);
  216. registry.destroy(e0);
  217. registry.assign<int>(registry.create(), 42);
  218. ASSERT_EQ(group.size(), typename decltype(group)::size_type{1});
  219. ASSERT_EQ(group[{}], e1);
  220. ASSERT_EQ(group.get<int>(e1), 1);
  221. ASSERT_EQ(group.get<unsigned int>(e1), 1u);
  222. group.each([e1](auto entity, auto ivalue, auto uivalue) {
  223. ASSERT_EQ(entity, e1);
  224. ASSERT_EQ(ivalue, 1);
  225. ASSERT_EQ(uivalue, 1u);
  226. });
  227. }
  228. TEST(NonOwningGroup, ConstNonConstAndAllInBetween) {
  229. entt::registry registry;
  230. auto group = registry.group(entt::get<int, const char, std::true_type>);
  231. ASSERT_EQ(group.size(), decltype(group.size()){0});
  232. const auto entity = registry.create();
  233. registry.assign<int>(entity, 0);
  234. registry.assign<char>(entity, 'c');
  235. registry.assign<std::true_type>(entity);
  236. ASSERT_EQ(group.size(), decltype(group.size()){1});
  237. ASSERT_TRUE((std::is_same_v<decltype(group.get<int>({})), int &>));
  238. ASSERT_TRUE((std::is_same_v<decltype(group.get<const char>({})), const char &>));
  239. ASSERT_TRUE((std::is_same_v<decltype(group.get<std::true_type>({})), std::true_type>));
  240. ASSERT_TRUE((std::is_same_v<decltype(group.get<int, const char, std::true_type>({})), std::tuple<int &, const char &, std::true_type>>));
  241. ASSERT_TRUE((std::is_same_v<decltype(group.raw<const char>()), const char *>));
  242. ASSERT_TRUE((std::is_same_v<decltype(group.raw<int>()), int *>));
  243. group.each([](auto &&i, auto &&c, auto &&e) {
  244. ASSERT_TRUE((std::is_same_v<decltype(i), int &>));
  245. ASSERT_TRUE((std::is_same_v<decltype(c), const char &>));
  246. ASSERT_TRUE((std::is_same_v<decltype(e), std::true_type &&>));
  247. });
  248. }
  249. TEST(NonOwningGroup, Find) {
  250. entt::registry registry;
  251. auto group = registry.group(entt::get<int, const char>);
  252. const auto e0 = registry.create();
  253. registry.assign<int>(e0);
  254. registry.assign<char>(e0);
  255. const auto e1 = registry.create();
  256. registry.assign<int>(e1);
  257. registry.assign<char>(e1);
  258. const auto e2 = registry.create();
  259. registry.assign<int>(e2);
  260. registry.assign<char>(e2);
  261. const auto e3 = registry.create();
  262. registry.assign<int>(e3);
  263. registry.assign<char>(e3);
  264. registry.remove<int>(e1);
  265. ASSERT_NE(group.find(e0), group.end());
  266. ASSERT_EQ(group.find(e1), group.end());
  267. ASSERT_NE(group.find(e2), group.end());
  268. ASSERT_NE(group.find(e3), group.end());
  269. auto it = group.find(e2);
  270. ASSERT_EQ(*it, e2);
  271. ASSERT_EQ(*(++it), e3);
  272. ASSERT_EQ(*(++it), e0);
  273. ASSERT_EQ(++it, group.end());
  274. ASSERT_EQ(++group.find(e0), group.end());
  275. const auto e4 = registry.create();
  276. registry.destroy(e4);
  277. const auto e5 = registry.create();
  278. registry.assign<int>(e5);
  279. registry.assign<char>(e5);
  280. ASSERT_NE(group.find(e5), group.end());
  281. ASSERT_EQ(group.find(e4), group.end());
  282. }
  283. TEST(NonOwningGroup, ExcludedComponents) {
  284. entt::registry registry;
  285. const auto e0 = registry.create();
  286. registry.assign<int>(e0, 0);
  287. const auto e1 = registry.create();
  288. registry.assign<int>(e1, 1);
  289. registry.assign<char>(e1);
  290. const auto group = registry.group(entt::get<int>, entt::exclude<char>);
  291. const auto e2 = registry.create();
  292. registry.assign<int>(e2, 2);
  293. const auto e3 = registry.create();
  294. registry.assign<int>(e3, 3);
  295. registry.assign<char>(e3);
  296. for(const auto entity: group) {
  297. ASSERT_TRUE(entity == e0 || entity == e2);
  298. if(entity == e0) {
  299. ASSERT_EQ(group.get<int>(e0), 0);
  300. } else if(entity == e2) {
  301. ASSERT_EQ(group.get<int>(e2), 2);
  302. }
  303. }
  304. registry.assign<char>(e0);
  305. registry.assign<char>(e2);
  306. ASSERT_TRUE(group.empty());
  307. registry.remove<char>(e1);
  308. registry.remove<char>(e3);
  309. for(const auto entity: group) {
  310. ASSERT_TRUE(entity == e1 || entity == e3);
  311. if(entity == e1) {
  312. ASSERT_EQ(group.get<int>(e1), 1);
  313. } else if(entity == e3) {
  314. ASSERT_EQ(group.get<int>(e3), 3);
  315. }
  316. }
  317. }
  318. TEST(NonOwningGroup, EmptyAndNonEmptyTypes) {
  319. entt::registry registry;
  320. const auto group = registry.group(entt::get<int, empty_type>);
  321. const auto e0 = registry.create();
  322. registry.assign<empty_type>(e0);
  323. registry.assign<int>(e0);
  324. const auto e1 = registry.create();
  325. registry.assign<empty_type>(e1);
  326. registry.assign<int>(e1);
  327. registry.assign<int>(registry.create());
  328. for(const auto entity: group) {
  329. ASSERT_TRUE(entity == e0 || entity == e1);
  330. }
  331. group.each([e0, e1](const auto entity, const int &, empty_type) {
  332. ASSERT_TRUE(entity == e0 || entity == e1);
  333. });
  334. ASSERT_EQ(group.size(), typename decltype(group)::size_type{2});
  335. }
  336. TEST(NonOwningGroup, TrackEntitiesOnComponentDestruction) {
  337. entt::registry registry;
  338. const auto group = registry.group(entt::get<int>, entt::exclude<char>);
  339. const auto cgroup = std::as_const(registry).group(entt::get<const int>, entt::exclude<char>);
  340. const auto entity = registry.create();
  341. registry.assign<int>(entity);
  342. registry.assign<char>(entity);
  343. ASSERT_TRUE(group.empty());
  344. ASSERT_TRUE(cgroup.empty());
  345. registry.remove<char>(entity);
  346. ASSERT_FALSE(group.empty());
  347. ASSERT_FALSE(cgroup.empty());
  348. }
  349. TEST(NonOwningGroup, Less) {
  350. entt::registry registry;
  351. const auto entity = registry.create();
  352. registry.assign<int>(entity);
  353. registry.assign<char>(entity);
  354. registry.assign<entt::tag<"empty"_hs>>(entity);
  355. registry.group(entt::get<int, char, entt::tag<"empty"_hs>>).less([entity](const auto entt, int, char) {
  356. ASSERT_EQ(entity, entt);
  357. });
  358. registry.group(entt::get<int, entt::tag<"empty"_hs>, char>).less([check = true](int, char) mutable {
  359. ASSERT_TRUE(check);
  360. check = false;
  361. });
  362. registry.group(entt::get<entt::tag<"empty"_hs>, int, char>).less([entity](const auto entt, int, char) {
  363. ASSERT_EQ(entity, entt);
  364. });
  365. registry.group(entt::get<int, char, double>).less([](const auto, int, char, double) { FAIL(); });
  366. }
  367. TEST(NonOwningGroup, FrontBack) {
  368. entt::registry registry;
  369. auto group = registry.group<>(entt::get<const int, const char>);
  370. ASSERT_EQ(group.front(), static_cast<entt::entity>(entt::null));
  371. ASSERT_EQ(group.back(), static_cast<entt::entity>(entt::null));
  372. const auto e0 = registry.create();
  373. registry.assign<int>(e0);
  374. registry.assign<char>(e0);
  375. const auto e1 = registry.create();
  376. registry.assign<int>(e1);
  377. registry.assign<char>(e1);
  378. const auto entity = registry.create();
  379. registry.assign<char>(entity);
  380. ASSERT_EQ(group.front(), e1);
  381. ASSERT_EQ(group.back(), e0);
  382. }
  383. TEST(NonOwningGroup, SignalRace) {
  384. entt::registry registry;
  385. registry.on_construct<double>().connect<&entt::registry::assign_or_replace<int>>();
  386. registry.group(entt::get<int, double>);
  387. auto entity = registry.create();
  388. registry.assign<double>(entity);
  389. ASSERT_EQ(registry.group(entt::get<int, double>).size(), 1u);
  390. }
  391. TEST(OwningGroup, Functionalities) {
  392. entt::registry registry;
  393. auto group = registry.group<int>(entt::get<char>);
  394. auto cgroup = std::as_const(registry).group<const int>(entt::get<const char>);
  395. ASSERT_TRUE(group.empty());
  396. ASSERT_TRUE((group.empty<int, char>()));
  397. ASSERT_TRUE((cgroup.empty<const int, const char>()));
  398. const auto e0 = registry.create();
  399. registry.assign<char>(e0);
  400. const auto e1 = registry.create();
  401. registry.assign<int>(e1);
  402. registry.assign<char>(e1);
  403. ASSERT_FALSE(group.empty());
  404. ASSERT_FALSE((group.empty<int>()));
  405. ASSERT_FALSE((cgroup.empty<const char>()));
  406. ASSERT_NO_THROW((group.begin()++));
  407. ASSERT_NO_THROW((++cgroup.begin()));
  408. ASSERT_NE(group.begin(), group.end());
  409. ASSERT_NE(cgroup.begin(), cgroup.end());
  410. ASSERT_EQ(group.size(), typename decltype(group)::size_type{1});
  411. ASSERT_EQ(group.size<int>(), typename decltype(group)::size_type{1});
  412. ASSERT_EQ(cgroup.size<const char>(), typename decltype(group)::size_type{2});
  413. registry.assign<int>(e0);
  414. ASSERT_EQ(group.size(), typename decltype(group)::size_type{2});
  415. ASSERT_EQ(group.size<int>(), typename decltype(group)::size_type{2});
  416. ASSERT_EQ(cgroup.size<const char>(), typename decltype(group)::size_type{2});
  417. registry.remove<int>(e0);
  418. ASSERT_EQ(group.size(), typename decltype(group)::size_type{1});
  419. ASSERT_EQ(group.size<int>(), typename decltype(group)::size_type{1});
  420. ASSERT_EQ(cgroup.size<const char>(), typename decltype(group)::size_type{2});
  421. registry.get<char>(e0) = '1';
  422. registry.get<char>(e1) = '2';
  423. registry.get<int>(e1) = 42;
  424. ASSERT_EQ(*(cgroup.raw<const int>() + 0), 42);
  425. ASSERT_EQ(*(group.raw<int>() + 0), 42);
  426. for(auto entity: group) {
  427. ASSERT_EQ(std::get<0>(cgroup.get<const int, const char>(entity)), 42);
  428. ASSERT_EQ(std::get<1>(group.get<int, char>(entity)), '2');
  429. ASSERT_EQ(cgroup.get<const char>(entity), '2');
  430. }
  431. ASSERT_EQ(*(group.data() + 0), e1);
  432. ASSERT_EQ(*(group.data<int>() + 0), e1);
  433. ASSERT_EQ(*(group.data<char>() + 0), e0);
  434. ASSERT_EQ(*(cgroup.data<const char>() + 1), e1);
  435. ASSERT_EQ(*(group.raw<int>() + 0), 42);
  436. ASSERT_EQ(*(group.raw<char>() + 0), '1');
  437. ASSERT_EQ(*(cgroup.raw<const char>() + 1), '2');
  438. registry.remove<char>(e0);
  439. registry.remove<char>(e1);
  440. ASSERT_EQ(group.begin(), group.end());
  441. ASSERT_EQ(cgroup.begin(), cgroup.end());
  442. ASSERT_TRUE(group.empty());
  443. }
  444. TEST(OwningGroup, ElementAccess) {
  445. entt::registry registry;
  446. auto group = registry.group<int>(entt::get<char>);
  447. auto cgroup = std::as_const(registry).group<const int>(entt::get<const char>);
  448. const auto e0 = registry.create();
  449. registry.assign<int>(e0);
  450. registry.assign<char>(e0);
  451. const auto e1 = registry.create();
  452. registry.assign<int>(e1);
  453. registry.assign<char>(e1);
  454. for(typename decltype(group)::size_type i{}; i < group.size(); ++i) {
  455. ASSERT_EQ(group[i], i ? e0 : e1);
  456. ASSERT_EQ(cgroup[i], i ? e0 : e1);
  457. }
  458. }
  459. TEST(OwningGroup, Contains) {
  460. entt::registry registry;
  461. auto group = registry.group<int>(entt::get<char>);
  462. const auto e0 = registry.create();
  463. registry.assign<int>(e0);
  464. registry.assign<char>(e0);
  465. const auto e1 = registry.create();
  466. registry.assign<int>(e1);
  467. registry.assign<char>(e1);
  468. registry.destroy(e0);
  469. ASSERT_FALSE(group.contains(e0));
  470. ASSERT_TRUE(group.contains(e1));
  471. }
  472. TEST(OwningGroup, Empty) {
  473. entt::registry registry;
  474. const auto e0 = registry.create();
  475. registry.assign<double>(e0);
  476. registry.assign<int>(e0);
  477. registry.assign<float>(e0);
  478. const auto e1 = registry.create();
  479. registry.assign<char>(e1);
  480. registry.assign<float>(e1);
  481. ASSERT_TRUE((registry.group<char, int>(entt::get<float>).empty()));
  482. ASSERT_TRUE((registry.group<double, float>(entt::get<char, int>).empty()));
  483. }
  484. TEST(OwningGroup, Each) {
  485. entt::registry registry;
  486. auto group = registry.group<int>(entt::get<char>);
  487. const auto e0 = registry.create();
  488. registry.assign<int>(e0);
  489. registry.assign<char>(e0);
  490. const auto e1 = registry.create();
  491. registry.assign<int>(e1);
  492. registry.assign<char>(e1);
  493. auto cgroup = std::as_const(registry).group<const int>(entt::get<const char>);
  494. std::size_t cnt = 0;
  495. group.each([&cnt](auto, int &, char &) { ++cnt; });
  496. group.each([&cnt](int &, char &) { ++cnt; });
  497. ASSERT_EQ(cnt, std::size_t{4});
  498. cgroup.each([&cnt](auto, const int &, const char &) { --cnt; });
  499. cgroup.each([&cnt](const int &, const char &) { --cnt; });
  500. ASSERT_EQ(cnt, std::size_t{0});
  501. }
  502. TEST(OwningGroup, SortOrdered) {
  503. entt::registry registry;
  504. auto group = registry.group<boxed_int, char>();
  505. entt::entity entities[5]{};
  506. registry.create(std::begin(entities), std::end(entities));
  507. registry.assign<boxed_int>(entities[0], 12);
  508. registry.assign<char>(entities[0], 'a');
  509. registry.assign<boxed_int>(entities[1], 9);
  510. registry.assign<char>(entities[1], 'b');
  511. registry.assign<boxed_int>(entities[2], 6);
  512. registry.assign<char>(entities[2], 'c');
  513. registry.assign<boxed_int>(entities[3], 1);
  514. registry.assign<boxed_int>(entities[4], 2);
  515. group.sort([&group](const entt::entity lhs, const entt::entity rhs) {
  516. return group.get<boxed_int>(lhs).value < group.get<boxed_int>(rhs).value;
  517. });
  518. ASSERT_EQ(*(group.data() + 0u), entities[0]);
  519. ASSERT_EQ(*(group.data() + 1u), entities[1]);
  520. ASSERT_EQ(*(group.data() + 2u), entities[2]);
  521. ASSERT_EQ(*(group.data() + 3u), entities[3]);
  522. ASSERT_EQ(*(group.data() + 4u), entities[4]);
  523. ASSERT_EQ((group.raw<boxed_int>() + 0u)->value, 12);
  524. ASSERT_EQ((group.raw<boxed_int>() + 1u)->value, 9);
  525. ASSERT_EQ((group.raw<boxed_int>() + 2u)->value, 6);
  526. ASSERT_EQ((group.raw<boxed_int>() + 3u)->value, 1);
  527. ASSERT_EQ((group.raw<boxed_int>() + 4u)->value, 2);
  528. ASSERT_EQ(*(group.raw<char>() + 0u), 'a');
  529. ASSERT_EQ(*(group.raw<char>() + 1u), 'b');
  530. ASSERT_EQ(*(group.raw<char>() + 2u), 'c');
  531. ASSERT_EQ((group.get<boxed_int, char>(entities[0])), (std::make_tuple(boxed_int{12}, 'a')));
  532. ASSERT_EQ((group.get<boxed_int, char>(entities[1])), (std::make_tuple(boxed_int{9}, 'b')));
  533. ASSERT_EQ((group.get<boxed_int, char>(entities[2])), (std::make_tuple(boxed_int{6}, 'c')));
  534. ASSERT_FALSE(group.contains(entities[3]));
  535. ASSERT_FALSE(group.contains(entities[4]));
  536. }
  537. TEST(OwningGroup, SortReverse) {
  538. entt::registry registry;
  539. auto group = registry.group<boxed_int, char>();
  540. entt::entity entities[5]{};
  541. registry.create(std::begin(entities), std::end(entities));
  542. registry.assign<boxed_int>(entities[0], 6);
  543. registry.assign<char>(entities[0], 'a');
  544. registry.assign<boxed_int>(entities[1], 9);
  545. registry.assign<char>(entities[1], 'b');
  546. registry.assign<boxed_int>(entities[2], 12);
  547. registry.assign<char>(entities[2], 'c');
  548. registry.assign<boxed_int>(entities[3], 1);
  549. registry.assign<boxed_int>(entities[4], 2);
  550. group.sort<boxed_int>([](const auto &lhs, const auto &rhs) {
  551. return lhs.value < rhs.value;
  552. });
  553. ASSERT_EQ(*(group.data() + 0u), entities[2]);
  554. ASSERT_EQ(*(group.data() + 1u), entities[1]);
  555. ASSERT_EQ(*(group.data() + 2u), entities[0]);
  556. ASSERT_EQ(*(group.data() + 3u), entities[3]);
  557. ASSERT_EQ(*(group.data() + 4u), entities[4]);
  558. ASSERT_EQ((group.raw<boxed_int>() + 0u)->value, 12);
  559. ASSERT_EQ((group.raw<boxed_int>() + 1u)->value, 9);
  560. ASSERT_EQ((group.raw<boxed_int>() + 2u)->value, 6);
  561. ASSERT_EQ((group.raw<boxed_int>() + 3u)->value, 1);
  562. ASSERT_EQ((group.raw<boxed_int>() + 4u)->value, 2);
  563. ASSERT_EQ(*(group.raw<char>() + 0u), 'c');
  564. ASSERT_EQ(*(group.raw<char>() + 1u), 'b');
  565. ASSERT_EQ(*(group.raw<char>() + 2u), 'a');
  566. ASSERT_EQ((group.get<boxed_int, char>(entities[0])), (std::make_tuple(boxed_int{6}, 'a')));
  567. ASSERT_EQ((group.get<boxed_int, char>(entities[1])), (std::make_tuple(boxed_int{9}, 'b')));
  568. ASSERT_EQ((group.get<boxed_int, char>(entities[2])), (std::make_tuple(boxed_int{12}, 'c')));
  569. ASSERT_FALSE(group.contains(entities[3]));
  570. ASSERT_FALSE(group.contains(entities[4]));
  571. }
  572. TEST(OwningGroup, SortUnordered) {
  573. entt::registry registry;
  574. auto group = registry.group<boxed_int>(entt::get<char>);
  575. entt::entity entities[7]{};
  576. registry.create(std::begin(entities), std::end(entities));
  577. registry.assign<boxed_int>(entities[0], 6);
  578. registry.assign<char>(entities[0], 'c');
  579. registry.assign<boxed_int>(entities[1], 3);
  580. registry.assign<char>(entities[1], 'b');
  581. registry.assign<boxed_int>(entities[2], 1);
  582. registry.assign<char>(entities[2], 'a');
  583. registry.assign<boxed_int>(entities[3], 9);
  584. registry.assign<char>(entities[3], 'd');
  585. registry.assign<boxed_int>(entities[4], 12);
  586. registry.assign<char>(entities[4], 'e');
  587. registry.assign<boxed_int>(entities[5], 4);
  588. registry.assign<boxed_int>(entities[6], 5);
  589. group.sort<char>([](const auto lhs, const auto rhs) {
  590. return lhs < rhs;
  591. });
  592. ASSERT_EQ(*(group.data() + 0u), entities[4]);
  593. ASSERT_EQ(*(group.data() + 1u), entities[3]);
  594. ASSERT_EQ(*(group.data() + 2u), entities[0]);
  595. ASSERT_EQ(*(group.data() + 3u), entities[1]);
  596. ASSERT_EQ(*(group.data() + 4u), entities[2]);
  597. ASSERT_EQ(*(group.data() + 5u), entities[5]);
  598. ASSERT_EQ(*(group.data() + 6u), entities[6]);
  599. ASSERT_EQ((group.raw<boxed_int>() + 0u)->value, 12);
  600. ASSERT_EQ((group.raw<boxed_int>() + 1u)->value, 9);
  601. ASSERT_EQ((group.raw<boxed_int>() + 2u)->value, 6);
  602. ASSERT_EQ((group.raw<boxed_int>() + 3u)->value, 3);
  603. ASSERT_EQ((group.raw<boxed_int>() + 4u)->value, 1);
  604. ASSERT_EQ((group.raw<boxed_int>() + 5u)->value, 4);
  605. ASSERT_EQ((group.raw<boxed_int>() + 6u)->value, 5);
  606. ASSERT_EQ(*(group.raw<char>() + 0u), 'c');
  607. ASSERT_EQ(*(group.raw<char>() + 1u), 'b');
  608. ASSERT_EQ(*(group.raw<char>() + 2u), 'a');
  609. ASSERT_EQ(*(group.raw<char>() + 3u), 'd');
  610. ASSERT_EQ(*(group.raw<char>() + 4u), 'e');
  611. ASSERT_EQ((group.get<boxed_int, char>(entities[0])), (std::make_tuple(boxed_int{6}, 'c')));
  612. ASSERT_EQ((group.get<boxed_int, char>(entities[1])), (std::make_tuple(boxed_int{3}, 'b')));
  613. ASSERT_EQ((group.get<boxed_int, char>(entities[2])), (std::make_tuple(boxed_int{1}, 'a')));
  614. ASSERT_EQ((group.get<boxed_int, char>(entities[3])), (std::make_tuple(boxed_int{9}, 'd')));
  615. ASSERT_EQ((group.get<boxed_int, char>(entities[4])), (std::make_tuple(boxed_int{12}, 'e')));
  616. ASSERT_FALSE(group.contains(entities[5]));
  617. ASSERT_FALSE(group.contains(entities[6]));
  618. }
  619. TEST(OwningGroup, SortWithExclusionList) {
  620. entt::registry registry;
  621. auto group = registry.group<boxed_int>(entt::exclude<char>);
  622. entt::entity entities[5]{};
  623. registry.create(std::begin(entities), std::end(entities));
  624. registry.assign<boxed_int>(entities[0], 0);
  625. registry.assign<boxed_int>(entities[1], 1);
  626. registry.assign<boxed_int>(entities[2], 2);
  627. registry.assign<boxed_int>(entities[3], 3);
  628. registry.assign<boxed_int>(entities[4], 4);
  629. registry.assign<char>(entities[2]);
  630. group.sort([](const entt::entity lhs, const entt::entity rhs) {
  631. return lhs < rhs;
  632. });
  633. ASSERT_EQ(*(group.data() + 0u), entities[4]);
  634. ASSERT_EQ(*(group.data() + 1u), entities[3]);
  635. ASSERT_EQ(*(group.data() + 2u), entities[1]);
  636. ASSERT_EQ(*(group.data() + 3u), entities[0]);
  637. ASSERT_EQ((group.raw<boxed_int>() + 0u)->value, 4);
  638. ASSERT_EQ((group.raw<boxed_int>() + 1u)->value, 3);
  639. ASSERT_EQ((group.raw<boxed_int>() + 2u)->value, 1);
  640. ASSERT_EQ((group.raw<boxed_int>() + 3u)->value, 0);
  641. ASSERT_EQ(group.get<boxed_int>(entities[0]).value, 0);
  642. ASSERT_EQ(group.get<boxed_int>(entities[1]).value, 1);
  643. ASSERT_EQ(group.get<boxed_int>(entities[3]).value, 3);
  644. ASSERT_EQ(group.get<boxed_int>(entities[4]).value, 4);
  645. ASSERT_FALSE(group.contains(entities[2]));
  646. }
  647. TEST(OwningGroup, IndexRebuiltOnDestroy) {
  648. entt::registry registry;
  649. auto group = registry.group<int>(entt::get<unsigned int>);
  650. const auto e0 = registry.create();
  651. const auto e1 = registry.create();
  652. registry.assign<unsigned int>(e0, 0u);
  653. registry.assign<unsigned int>(e1, 1u);
  654. registry.assign<int>(e0, 0);
  655. registry.assign<int>(e1, 1);
  656. registry.destroy(e0);
  657. registry.assign<int>(registry.create(), 42);
  658. ASSERT_EQ(group.size(), typename decltype(group)::size_type{1});
  659. ASSERT_EQ(group[{}], e1);
  660. ASSERT_EQ(group.get<int>(e1), 1);
  661. ASSERT_EQ(group.get<unsigned int>(e1), 1u);
  662. group.each([e1](auto entity, auto ivalue, auto uivalue) {
  663. ASSERT_EQ(entity, e1);
  664. ASSERT_EQ(ivalue, 1);
  665. ASSERT_EQ(uivalue, 1u);
  666. });
  667. }
  668. TEST(OwningGroup, ConstNonConstAndAllInBetween) {
  669. entt::registry registry;
  670. auto group = registry.group<int, const char>(entt::get<double, const float, std::true_type>);
  671. ASSERT_EQ(group.size(), decltype(group.size()){0});
  672. const auto entity = registry.create();
  673. registry.assign<int>(entity, 0);
  674. registry.assign<char>(entity, 'c');
  675. registry.assign<double>(entity, 0.);
  676. registry.assign<float>(entity, 0.f);
  677. registry.assign<std::true_type>(entity);
  678. ASSERT_EQ(group.size(), decltype(group.size()){1});
  679. ASSERT_TRUE((std::is_same_v<decltype(group.get<int>({})), int &>));
  680. ASSERT_TRUE((std::is_same_v<decltype(group.get<const char>({})), const char &>));
  681. ASSERT_TRUE((std::is_same_v<decltype(group.get<double>({})), double &>));
  682. ASSERT_TRUE((std::is_same_v<decltype(group.get<const float>({})), const float &>));
  683. ASSERT_TRUE((std::is_same_v<decltype(group.get<std::true_type>({})), std::true_type>));
  684. ASSERT_TRUE((std::is_same_v<decltype(group.get<int, const char, double, const float, std::true_type>({})), std::tuple<int &, const char &, double &, const float &, std::true_type>>));
  685. ASSERT_TRUE((std::is_same_v<decltype(group.raw<const float>()), const float *>));
  686. ASSERT_TRUE((std::is_same_v<decltype(group.raw<double>()), double *>));
  687. ASSERT_TRUE((std::is_same_v<decltype(group.raw<const char>()), const char *>));
  688. ASSERT_TRUE((std::is_same_v<decltype(group.raw<int>()), int *>));
  689. group.each([](auto &&i, auto &&c, auto &&d, auto &&f, auto &&e) {
  690. ASSERT_TRUE((std::is_same_v<decltype(i), int &>));
  691. ASSERT_TRUE((std::is_same_v<decltype(c), const char &>));
  692. ASSERT_TRUE((std::is_same_v<decltype(d), double &>));
  693. ASSERT_TRUE((std::is_same_v<decltype(f), const float &>));
  694. ASSERT_TRUE((std::is_same_v<decltype(e), std::true_type &&>));
  695. });
  696. }
  697. TEST(OwningGroup, Find) {
  698. entt::registry registry;
  699. auto group = registry.group<int>(entt::get<const char>);
  700. const auto e0 = registry.create();
  701. registry.assign<int>(e0);
  702. registry.assign<char>(e0);
  703. const auto e1 = registry.create();
  704. registry.assign<int>(e1);
  705. registry.assign<char>(e1);
  706. const auto e2 = registry.create();
  707. registry.assign<int>(e2);
  708. registry.assign<char>(e2);
  709. const auto e3 = registry.create();
  710. registry.assign<int>(e3);
  711. registry.assign<char>(e3);
  712. registry.remove<int>(e1);
  713. ASSERT_NE(group.find(e0), group.end());
  714. ASSERT_EQ(group.find(e1), group.end());
  715. ASSERT_NE(group.find(e2), group.end());
  716. ASSERT_NE(group.find(e3), group.end());
  717. auto it = group.find(e2);
  718. ASSERT_EQ(*it, e2);
  719. ASSERT_EQ(*(++it), e3);
  720. ASSERT_EQ(*(++it), e0);
  721. ASSERT_EQ(++it, group.end());
  722. ASSERT_EQ(++group.find(e0), group.end());
  723. const auto e4 = registry.create();
  724. registry.destroy(e4);
  725. const auto e5 = registry.create();
  726. registry.assign<int>(e5);
  727. registry.assign<char>(e5);
  728. ASSERT_NE(group.find(e5), group.end());
  729. ASSERT_EQ(group.find(e4), group.end());
  730. }
  731. TEST(OwningGroup, ExcludedComponents) {
  732. entt::registry registry;
  733. const auto e0 = registry.create();
  734. registry.assign<int>(e0, 0);
  735. const auto e1 = registry.create();
  736. registry.assign<int>(e1, 1);
  737. registry.assign<char>(e1);
  738. const auto group = registry.group<int>(entt::exclude<char, double>);
  739. const auto e2 = registry.create();
  740. registry.assign<int>(e2, 2);
  741. const auto e3 = registry.create();
  742. registry.assign<int>(e3, 3);
  743. registry.assign<double>(e3);
  744. for(const auto entity: group) {
  745. ASSERT_TRUE(entity == e0 || entity == e2);
  746. if(entity == e0) {
  747. ASSERT_EQ(group.get<int>(e0), 0);
  748. } else if(entity == e2) {
  749. ASSERT_EQ(group.get<int>(e2), 2);
  750. }
  751. }
  752. registry.assign<char>(e0);
  753. registry.assign<double>(e2);
  754. ASSERT_TRUE(group.empty());
  755. registry.remove<char>(e1);
  756. registry.remove<double>(e3);
  757. for(const auto entity: group) {
  758. ASSERT_TRUE(entity == e1 || entity == e3);
  759. if(entity == e1) {
  760. ASSERT_EQ(group.get<int>(e1), 1);
  761. } else if(entity == e3) {
  762. ASSERT_EQ(group.get<int>(e3), 3);
  763. }
  764. }
  765. }
  766. TEST(OwningGroup, EmptyAndNonEmptyTypes) {
  767. entt::registry registry;
  768. const auto group = registry.group<empty_type>(entt::get<int>);
  769. const auto e0 = registry.create();
  770. registry.assign<empty_type>(e0);
  771. registry.assign<int>(e0);
  772. const auto e1 = registry.create();
  773. registry.assign<empty_type>(e1);
  774. registry.assign<int>(e1);
  775. registry.assign<int>(registry.create());
  776. for(const auto entity: group) {
  777. ASSERT_TRUE(entity == e0 || entity == e1);
  778. }
  779. group.each([e0, e1](const auto entity, empty_type, const int &) {
  780. ASSERT_TRUE(entity == e0 || entity == e1);
  781. });
  782. ASSERT_EQ(group.size(), typename decltype(group)::size_type{2});
  783. }
  784. TEST(OwningGroup, TrackEntitiesOnComponentDestruction) {
  785. entt::registry registry;
  786. const auto group = registry.group<int>(entt::exclude<char>);
  787. const auto cgroup = std::as_const(registry).group<const int>(entt::exclude<char>);
  788. const auto entity = registry.create();
  789. registry.assign<int>(entity);
  790. registry.assign<char>(entity);
  791. ASSERT_TRUE(group.empty());
  792. ASSERT_TRUE(cgroup.empty());
  793. registry.remove<char>(entity);
  794. ASSERT_FALSE(group.empty());
  795. ASSERT_FALSE(cgroup.empty());
  796. }
  797. TEST(OwningGroup, Less) {
  798. entt::registry registry;
  799. const auto entity = registry.create();
  800. registry.assign<int>(entity);
  801. registry.assign<char>(entity);
  802. registry.assign<entt::tag<"empty"_hs>>(entity);
  803. registry.group<int>(entt::get<char, entt::tag<"empty"_hs>>).less([entity](const auto entt, int, char) {
  804. ASSERT_EQ(entity, entt);
  805. });
  806. registry.group<char>(entt::get<entt::tag<"empty"_hs>, int>).less([check = true](int, char) mutable {
  807. ASSERT_TRUE(check);
  808. check = false;
  809. });
  810. registry.group<entt::tag<"empty"_hs>>(entt::get<int, char>).less([entity](const auto entt, int, char) {
  811. ASSERT_EQ(entity, entt);
  812. });
  813. registry.group<double>(entt::get<int, char>).less([](const auto, double, int, char) { FAIL(); });
  814. }
  815. TEST(OwningGroup, FrontBack) {
  816. entt::registry registry;
  817. auto group = registry.group<const char>(entt::get<const int>);
  818. ASSERT_EQ(group.front(), static_cast<entt::entity>(entt::null));
  819. ASSERT_EQ(group.back(), static_cast<entt::entity>(entt::null));
  820. const auto e0 = registry.create();
  821. registry.assign<int>(e0);
  822. registry.assign<char>(e0);
  823. const auto e1 = registry.create();
  824. registry.assign<int>(e1);
  825. registry.assign<char>(e1);
  826. const auto entity = registry.create();
  827. registry.assign<char>(entity);
  828. ASSERT_EQ(group.front(), e1);
  829. ASSERT_EQ(group.back(), e0);
  830. }
  831. TEST(OwningGroup, SignalRace) {
  832. entt::registry registry;
  833. registry.on_construct<double>().connect<&entt::registry::assign_or_replace<int>>();
  834. registry.group<int>(entt::get<double>);
  835. auto entity = registry.create();
  836. registry.assign<double>(entity);
  837. ASSERT_EQ(registry.group<int>(entt::get<double>).size(), 1u);
  838. }