group.cpp 30 KB

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