1
0

registry.cpp 48 KB

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