1
0

storage.cpp 45 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382
  1. #include <memory>
  2. #include <utility>
  3. #include <iterator>
  4. #include <exception>
  5. #include <type_traits>
  6. #include <unordered_set>
  7. #include <gtest/gtest.h>
  8. #include <entt/entity/component.hpp>
  9. #include <entt/entity/storage.hpp>
  10. #include "throwing_allocator.hpp"
  11. #include "throwing_component.hpp"
  12. struct empty_type {};
  13. struct boxed_int { int value; };
  14. struct stable_type { int value; };
  15. struct non_default_constructible {
  16. non_default_constructible() = delete;
  17. non_default_constructible(int v): value{v} {}
  18. int value;
  19. };
  20. struct update_from_destructor {
  21. update_from_destructor(entt::storage<update_from_destructor> &ref, entt::entity other)
  22. : storage{&ref},
  23. target{other}
  24. {}
  25. update_from_destructor(update_from_destructor &&other) ENTT_NOEXCEPT
  26. : storage{std::exchange(other.storage, nullptr)},
  27. target{std::exchange(other.target, entt::null)}
  28. {}
  29. update_from_destructor & operator=(update_from_destructor &&other) ENTT_NOEXCEPT {
  30. storage = std::exchange(other.storage, nullptr);
  31. target = std::exchange(other.target, entt::null);
  32. return *this;
  33. }
  34. ~update_from_destructor() {
  35. if(target != entt::null && storage->contains(target)) {
  36. storage->erase(target);
  37. }
  38. }
  39. private:
  40. entt::storage<update_from_destructor> *storage{};
  41. entt::entity target{entt::null};
  42. };
  43. template<>
  44. struct entt::component_traits<stable_type>: basic_component_traits {
  45. using in_place_delete = std::true_type;
  46. };
  47. bool operator==(const boxed_int &lhs, const boxed_int &rhs) {
  48. return lhs.value == rhs.value;
  49. }
  50. TEST(Storage, Functionalities) {
  51. entt::storage<int> pool;
  52. ASSERT_NO_THROW([[maybe_unused]] auto alloc = pool.get_allocator());
  53. pool.reserve(42);
  54. ASSERT_EQ(pool.capacity(), ENTT_PACKED_PAGE);
  55. ASSERT_TRUE(pool.empty());
  56. ASSERT_EQ(pool.size(), 0u);
  57. ASSERT_EQ(std::as_const(pool).begin(), std::as_const(pool).end());
  58. ASSERT_EQ(pool.begin(), pool.end());
  59. ASSERT_FALSE(pool.contains(entt::entity{0}));
  60. ASSERT_FALSE(pool.contains(entt::entity{41}));
  61. pool.reserve(0);
  62. ASSERT_EQ(pool.capacity(), ENTT_PACKED_PAGE);
  63. ASSERT_TRUE(pool.empty());
  64. pool.emplace(entt::entity{41}, 3);
  65. ASSERT_FALSE(pool.empty());
  66. ASSERT_EQ(pool.size(), 1u);
  67. ASSERT_NE(std::as_const(pool).begin(), std::as_const(pool).end());
  68. ASSERT_NE(pool.begin(), pool.end());
  69. ASSERT_FALSE(pool.contains(entt::entity{0}));
  70. ASSERT_TRUE(pool.contains(entt::entity{41}));
  71. ASSERT_EQ(pool.get(entt::entity{41}), 3);
  72. ASSERT_EQ(std::as_const(pool).get(entt::entity{41}), 3);
  73. ASSERT_EQ(pool.get_as_tuple(entt::entity{41}), std::make_tuple(3));
  74. ASSERT_EQ(std::as_const(pool).get_as_tuple(entt::entity{41}), std::make_tuple(3));
  75. pool.erase(entt::entity{41});
  76. ASSERT_TRUE(pool.empty());
  77. ASSERT_EQ(pool.size(), 0u);
  78. ASSERT_EQ(std::as_const(pool).begin(), std::as_const(pool).end());
  79. ASSERT_EQ(pool.begin(), pool.end());
  80. ASSERT_FALSE(pool.contains(entt::entity{0}));
  81. ASSERT_FALSE(pool.contains(entt::entity{41}));
  82. pool.emplace(entt::entity{41}, 12);
  83. ASSERT_EQ(pool.get(entt::entity{41}), 12);
  84. ASSERT_EQ(std::as_const(pool).get(entt::entity{41}), 12);
  85. ASSERT_EQ(pool.get_as_tuple(entt::entity{41}), std::make_tuple(12));
  86. ASSERT_EQ(std::as_const(pool).get_as_tuple(entt::entity{41}), std::make_tuple(12));
  87. pool.clear();
  88. ASSERT_TRUE(pool.empty());
  89. ASSERT_EQ(pool.size(), 0u);
  90. ASSERT_EQ(std::as_const(pool).begin(), std::as_const(pool).end());
  91. ASSERT_EQ(pool.begin(), pool.end());
  92. ASSERT_FALSE(pool.contains(entt::entity{0}));
  93. ASSERT_FALSE(pool.contains(entt::entity{41}));
  94. ASSERT_EQ(pool.capacity(), ENTT_PACKED_PAGE);
  95. pool.shrink_to_fit();
  96. ASSERT_EQ(pool.capacity(), 0u);
  97. }
  98. TEST(Storage, Move) {
  99. entt::storage<int> pool;
  100. pool.emplace(entt::entity{3}, 3);
  101. ASSERT_TRUE(std::is_move_constructible_v<decltype(pool)>);
  102. ASSERT_TRUE(std::is_move_assignable_v<decltype(pool)>);
  103. entt::storage<int> other{std::move(pool)};
  104. ASSERT_TRUE(pool.empty());
  105. ASSERT_FALSE(other.empty());
  106. ASSERT_EQ(pool.at(0u), static_cast<entt::entity>(entt::null));
  107. ASSERT_EQ(other.at(0u), entt::entity{3});
  108. ASSERT_EQ(other.get(entt::entity{3}), 3);
  109. pool = std::move(other);
  110. ASSERT_FALSE(pool.empty());
  111. ASSERT_TRUE(other.empty());
  112. ASSERT_EQ(pool.at(0u), entt::entity{3});
  113. ASSERT_EQ(pool.get(entt::entity{3}), 3);
  114. ASSERT_EQ(other.at(0u), static_cast<entt::entity>(entt::null));
  115. other = entt::storage<int>{};
  116. other.emplace(entt::entity{42}, 42);
  117. other = std::move(pool);
  118. ASSERT_TRUE(pool.empty());
  119. ASSERT_FALSE(other.empty());
  120. ASSERT_EQ(pool.at(0u), static_cast<entt::entity>(entt::null));
  121. ASSERT_EQ(other.at(0u), entt::entity{3});
  122. ASSERT_EQ(other.get(entt::entity{3}), 3);
  123. }
  124. TEST(Storage, Swap) {
  125. entt::storage<int> pool;
  126. entt::storage<int> other;
  127. pool.emplace(entt::entity{42}, 41);
  128. other.emplace(entt::entity{9}, 8);
  129. other.emplace(entt::entity{3}, 2);
  130. other.erase(entt::entity{9});
  131. ASSERT_EQ(pool.size(), 1u);
  132. ASSERT_EQ(other.size(), 1u);
  133. pool.swap(other);
  134. ASSERT_EQ(pool.size(), 1u);
  135. ASSERT_EQ(other.size(), 1u);
  136. ASSERT_EQ(pool.at(0u), entt::entity{3});
  137. ASSERT_EQ(pool.get(entt::entity{3}), 2);
  138. ASSERT_EQ(other.at(0u), entt::entity{42});
  139. ASSERT_EQ(other.get(entt::entity{42}), 41);
  140. }
  141. TEST(Storage, StableSwap) {
  142. entt::storage<stable_type> pool;
  143. entt::storage<stable_type> other;
  144. pool.emplace(entt::entity{42}, 41);
  145. other.emplace(entt::entity{9}, 8);
  146. other.emplace(entt::entity{3}, 2);
  147. other.erase(entt::entity{9});
  148. ASSERT_EQ(pool.size(), 1u);
  149. ASSERT_EQ(other.size(), 2u);
  150. pool.swap(other);
  151. ASSERT_EQ(pool.size(), 2u);
  152. ASSERT_EQ(other.size(), 1u);
  153. ASSERT_EQ(pool.at(1u), entt::entity{3});
  154. ASSERT_EQ(pool.get(entt::entity{3}).value, 2);
  155. ASSERT_EQ(other.at(0u), entt::entity{42});
  156. ASSERT_EQ(other.get(entt::entity{42}).value, 41);
  157. }
  158. TEST(Storage, EmptyType) {
  159. entt::storage<empty_type> pool;
  160. pool.emplace(entt::entity{99});
  161. ASSERT_NO_THROW([[maybe_unused]] auto alloc = pool.get_allocator());
  162. ASSERT_TRUE(pool.contains(entt::entity{99}));
  163. ASSERT_DEATH(pool.get(entt::entity{}), "");
  164. }
  165. TEST(Storage, Insert) {
  166. entt::storage<stable_type> pool;
  167. entt::entity entities[2u]{entt::entity{3}, entt::entity{42}};
  168. pool.insert(std::begin(entities), std::end(entities), stable_type{99});
  169. ASSERT_TRUE(pool.contains(entities[0u]));
  170. ASSERT_TRUE(pool.contains(entities[1u]));
  171. ASSERT_FALSE(pool.empty());
  172. ASSERT_EQ(pool.size(), 2u);
  173. ASSERT_EQ(pool.get(entities[0u]).value, 99);
  174. ASSERT_EQ(pool.get(entities[1u]).value, 99);
  175. pool.erase(std::begin(entities), std::end(entities));
  176. const stable_type values[2u] = { stable_type{42}, stable_type{3} };
  177. pool.insert(std::rbegin(entities), std::rend(entities), std::begin(values));
  178. ASSERT_EQ(pool.size(), 2u);
  179. ASSERT_EQ(pool.at(0u), entities[0u]);
  180. ASSERT_EQ(pool.at(1u), entities[1u]);
  181. ASSERT_EQ(pool.index(entities[0u]), 0u);
  182. ASSERT_EQ(pool.index(entities[1u]), 1u);
  183. ASSERT_EQ(pool.get(entities[0u]).value, 3);
  184. ASSERT_EQ(pool.get(entities[1u]).value, 42);
  185. }
  186. TEST(Storage, InsertEmptyType) {
  187. entt::storage<empty_type> pool;
  188. entt::entity entities[2u]{entt::entity{3}, entt::entity{42}};
  189. pool.insert(std::begin(entities), std::end(entities));
  190. ASSERT_TRUE(pool.contains(entities[0u]));
  191. ASSERT_TRUE(pool.contains(entities[1u]));
  192. ASSERT_FALSE(pool.empty());
  193. ASSERT_EQ(pool.size(), 2u);
  194. pool.erase(std::begin(entities), std::end(entities));
  195. const empty_type values[2u]{};
  196. pool.insert(std::rbegin(entities), std::rend(entities), std::begin(values));
  197. ASSERT_EQ(pool.size(), 2u);
  198. ASSERT_EQ(pool.at(0u), entities[1u]);
  199. ASSERT_EQ(pool.at(1u), entities[0u]);
  200. ASSERT_EQ(pool.index(entities[0u]), 1u);
  201. ASSERT_EQ(pool.index(entities[1u]), 0u);
  202. }
  203. TEST(Storage, Erase) {
  204. entt::storage<int> pool;
  205. entt::entity entities[3u]{entt::entity{3}, entt::entity{42}, entt::entity{9}};
  206. pool.emplace(entities[0u]);
  207. pool.emplace(entities[1u]);
  208. pool.emplace(entities[2u]);
  209. pool.erase(std::begin(entities), std::end(entities));
  210. ASSERT_DEATH(pool.erase(std::begin(entities), std::end(entities)), "");
  211. ASSERT_TRUE(pool.empty());
  212. pool.emplace(entities[0u], 0);
  213. pool.emplace(entities[1u], 1);
  214. pool.emplace(entities[2u], 2);
  215. pool.erase(entities, entities + 2u);
  216. ASSERT_FALSE(pool.empty());
  217. ASSERT_EQ(*pool.begin(), 2);
  218. pool.erase(entities[2u]);
  219. ASSERT_DEATH(pool.erase(entities[2u]), "");
  220. ASSERT_TRUE(pool.empty());
  221. pool.emplace(entities[0u], 0);
  222. pool.emplace(entities[1u], 1);
  223. pool.emplace(entities[2u], 2);
  224. std::swap(entities[1u], entities[2u]);
  225. pool.erase(entities, entities + 2u);
  226. ASSERT_FALSE(pool.empty());
  227. ASSERT_EQ(*pool.begin(), 1);
  228. }
  229. TEST(Storage, StableErase) {
  230. entt::storage<stable_type> pool;
  231. entt::entity entities[3u]{entt::entity{3}, entt::entity{42}, entt::entity{9}};
  232. ASSERT_DEATH([[maybe_unused]] auto &&value = pool.get(entt::tombstone), "");
  233. ASSERT_DEATH([[maybe_unused]] auto &&value = pool.get(entt::null), "");
  234. pool.emplace(entities[0u], stable_type{0});
  235. pool.emplace(entities[1u], stable_type{1});
  236. pool.emplace(entities[2u], stable_type{2});
  237. pool.erase(std::begin(entities), std::end(entities));
  238. ASSERT_DEATH(pool.erase(std::begin(entities), std::end(entities)), "");
  239. ASSERT_FALSE(pool.empty());
  240. ASSERT_EQ(pool.size(), 3u);
  241. ASSERT_TRUE(pool.at(2u) == entt::tombstone);
  242. ASSERT_DEATH([[maybe_unused]] auto &&value = pool.get(entt::tombstone), "");
  243. ASSERT_DEATH([[maybe_unused]] auto &&value = pool.get(entt::null), "");
  244. ASSERT_DEATH([[maybe_unused]] auto &&value = pool.get(entities[1u]), "");
  245. pool.emplace(entities[2u], stable_type{2});
  246. pool.emplace(entities[0u], stable_type{0});
  247. pool.emplace(entities[1u], stable_type{1});
  248. ASSERT_EQ(pool.get(entities[0u]).value, 0);
  249. ASSERT_EQ(pool.get(entities[1u]).value, 1);
  250. ASSERT_EQ(pool.get(entities[2u]).value, 2);
  251. ASSERT_EQ(pool.begin()->value, 2);
  252. ASSERT_EQ(pool.index(entities[0u]), 1u);
  253. ASSERT_EQ(pool.index(entities[1u]), 0u);
  254. ASSERT_EQ(pool.index(entities[2u]), 2u);
  255. pool.erase(entities, entities + 2u);
  256. ASSERT_FALSE(pool.empty());
  257. ASSERT_EQ(pool.size(), 3u);
  258. ASSERT_EQ(pool.begin()->value, 2);
  259. ASSERT_EQ(pool.index(entities[2u]), 2u);
  260. pool.erase(entities[2u]);
  261. ASSERT_DEATH(pool.erase(entities[2u]), "");
  262. ASSERT_FALSE(pool.empty());
  263. ASSERT_EQ(pool.size(), 3u);
  264. ASSERT_FALSE(pool.contains(entities[0u]));
  265. ASSERT_FALSE(pool.contains(entities[1u]));
  266. ASSERT_FALSE(pool.contains(entities[2u]));
  267. pool.emplace(entities[0u], stable_type{0});
  268. pool.emplace(entities[1u], stable_type{1});
  269. pool.emplace(entities[2u], stable_type{2});
  270. std::swap(entities[1u], entities[2u]);
  271. pool.erase(entities, entities + 2u);
  272. ASSERT_FALSE(pool.empty());
  273. ASSERT_EQ(pool.size(), 3u);
  274. ASSERT_TRUE(pool.contains(entities[2u]));
  275. ASSERT_EQ(pool.index(entities[2u]), 0u);
  276. ASSERT_EQ(pool.get(entities[2u]).value, 1);
  277. pool.compact();
  278. ASSERT_FALSE(pool.empty());
  279. ASSERT_EQ(pool.size(), 1u);
  280. ASSERT_EQ(pool.begin()->value, 1);
  281. pool.clear();
  282. ASSERT_EQ(pool.size(), 1u);
  283. pool.compact();
  284. ASSERT_EQ(pool.size(), 0u);
  285. pool.emplace(entities[0u], stable_type{0});
  286. pool.emplace(entities[1u], stable_type{2});
  287. pool.emplace(entities[2u], stable_type{1});
  288. pool.erase(entities[2u]);
  289. ASSERT_DEATH(pool.erase(entities[2u]), "");
  290. pool.erase(entities[0u]);
  291. pool.erase(entities[1u]);
  292. ASSERT_DEATH(pool.erase(entities, entities + 2u), "");
  293. ASSERT_EQ(pool.size(), 3u);
  294. ASSERT_TRUE(pool.at(2u) == entt::tombstone);
  295. pool.emplace(entities[0u], stable_type{99});
  296. ASSERT_EQ((++pool.begin())->value, 99);
  297. pool.emplace(entities[1u], stable_type{2});
  298. pool.emplace(entities[2u], stable_type{1});
  299. pool.emplace(entt::entity{0}, stable_type{7});
  300. ASSERT_EQ(pool.size(), 4u);
  301. ASSERT_EQ(pool.begin()->value, 7);
  302. ASSERT_EQ(pool.at(0u), entities[1u]);
  303. ASSERT_EQ(pool.at(1u), entities[0u]);
  304. ASSERT_EQ(pool.at(2u), entities[2u]);
  305. ASSERT_EQ(pool.get(entities[0u]).value, 99);
  306. ASSERT_EQ(pool.get(entities[1u]).value, 2);
  307. ASSERT_EQ(pool.get(entities[2u]).value, 1);
  308. }
  309. TEST(Storage, Remove) {
  310. entt::storage<int> pool;
  311. entt::entity entities[3u]{entt::entity{3}, entt::entity{42}, entt::entity{9}};
  312. pool.emplace(entities[0u]);
  313. pool.emplace(entities[1u]);
  314. pool.emplace(entities[2u]);
  315. ASSERT_EQ(pool.remove(std::begin(entities), std::end(entities)), 3u);
  316. ASSERT_EQ(pool.remove(std::begin(entities), std::end(entities)), 0u);
  317. ASSERT_TRUE(pool.empty());
  318. pool.emplace(entities[0u], 0);
  319. pool.emplace(entities[1u], 1);
  320. pool.emplace(entities[2u], 2);
  321. ASSERT_EQ(pool.remove(entities, entities + 2u), 2u);
  322. ASSERT_FALSE(pool.empty());
  323. ASSERT_EQ(*pool.begin(), 2);
  324. ASSERT_EQ(pool.remove(entities[2u]), 1u);
  325. ASSERT_EQ(pool.remove(entities[2u]), 0u);
  326. ASSERT_TRUE(pool.empty());
  327. pool.emplace(entities[0u], 0);
  328. pool.emplace(entities[1u], 1);
  329. pool.emplace(entities[2u], 2);
  330. std::swap(entities[1u], entities[2u]);
  331. ASSERT_EQ(pool.remove(entities, entities + 2u), 2u);
  332. ASSERT_FALSE(pool.empty());
  333. ASSERT_EQ(*pool.begin(), 1);
  334. }
  335. TEST(Storage, StableRemove) {
  336. entt::storage<stable_type> pool;
  337. entt::entity entities[3u]{entt::entity{3}, entt::entity{42}, entt::entity{9}};
  338. pool.emplace(entities[0u], stable_type{0});
  339. pool.emplace(entities[1u], stable_type{1});
  340. pool.emplace(entities[2u], stable_type{2});
  341. ASSERT_EQ(pool.remove(std::begin(entities), std::end(entities)), 3u);
  342. ASSERT_EQ(pool.remove(std::begin(entities), std::end(entities)), 0u);
  343. ASSERT_FALSE(pool.empty());
  344. ASSERT_EQ(pool.size(), 3u);
  345. ASSERT_TRUE(pool.at(2u) == entt::tombstone);
  346. ASSERT_DEATH([[maybe_unused]] auto &&value = pool.get(entt::tombstone), "");
  347. ASSERT_DEATH([[maybe_unused]] auto &&value = pool.get(entt::null), "");
  348. ASSERT_DEATH([[maybe_unused]] auto &&value = pool.get(entities[1u]), "");
  349. pool.emplace(entities[2u], stable_type{2});
  350. pool.emplace(entities[0u], stable_type{0});
  351. pool.emplace(entities[1u], stable_type{1});
  352. ASSERT_EQ(pool.get(entities[0u]).value, 0);
  353. ASSERT_EQ(pool.get(entities[1u]).value, 1);
  354. ASSERT_EQ(pool.get(entities[2u]).value, 2);
  355. ASSERT_EQ(pool.begin()->value, 2);
  356. ASSERT_EQ(pool.index(entities[0u]), 1u);
  357. ASSERT_EQ(pool.index(entities[1u]), 0u);
  358. ASSERT_EQ(pool.index(entities[2u]), 2u);
  359. ASSERT_EQ(pool.remove(entities, entities + 2u), 2u);
  360. ASSERT_FALSE(pool.empty());
  361. ASSERT_EQ(pool.size(), 3u);
  362. ASSERT_EQ(pool.begin()->value, 2);
  363. ASSERT_EQ(pool.index(entities[2u]), 2u);
  364. ASSERT_EQ(pool.remove(entities[2u]), 1u);
  365. ASSERT_EQ(pool.remove(entities[2u]), 0u);
  366. ASSERT_EQ(pool.remove(entities[2u]), 0u);
  367. ASSERT_FALSE(pool.empty());
  368. ASSERT_EQ(pool.size(), 3u);
  369. ASSERT_FALSE(pool.contains(entities[0u]));
  370. ASSERT_FALSE(pool.contains(entities[1u]));
  371. ASSERT_FALSE(pool.contains(entities[2u]));
  372. pool.emplace(entities[0u], stable_type{0});
  373. pool.emplace(entities[1u], stable_type{1});
  374. pool.emplace(entities[2u], stable_type{2});
  375. std::swap(entities[1u], entities[2u]);
  376. ASSERT_EQ(pool.remove(entities, entities + 2u), 2u);
  377. ASSERT_FALSE(pool.empty());
  378. ASSERT_EQ(pool.size(), 3u);
  379. ASSERT_TRUE(pool.contains(entities[2u]));
  380. ASSERT_EQ(pool.index(entities[2u]), 0u);
  381. ASSERT_EQ(pool.get(entities[2u]).value, 1);
  382. pool.compact();
  383. ASSERT_FALSE(pool.empty());
  384. ASSERT_EQ(pool.size(), 1u);
  385. ASSERT_EQ(pool.begin()->value, 1);
  386. pool.clear();
  387. ASSERT_EQ(pool.size(), 1u);
  388. pool.compact();
  389. ASSERT_EQ(pool.size(), 0u);
  390. pool.emplace(entities[0u], stable_type{0});
  391. pool.emplace(entities[1u], stable_type{2});
  392. pool.emplace(entities[2u], stable_type{1});
  393. ASSERT_EQ(pool.remove(entities[2u]), 1u);
  394. ASSERT_EQ(pool.remove(entities[2u]), 0u);
  395. ASSERT_EQ(pool.remove(entities[0u]), 1u);
  396. ASSERT_EQ(pool.remove(entities[1u]), 1u);
  397. ASSERT_EQ(pool.remove(entities, entities + 2u), 0u);
  398. ASSERT_EQ(pool.size(), 3u);
  399. ASSERT_TRUE(pool.at(2u) == entt::tombstone);
  400. pool.emplace(entities[0u], stable_type{99});
  401. ASSERT_EQ((++pool.begin())->value, 99);
  402. pool.emplace(entities[1u], stable_type{2});
  403. pool.emplace(entities[2u], stable_type{1});
  404. pool.emplace(entt::entity{0}, stable_type{7});
  405. ASSERT_EQ(pool.size(), 4u);
  406. ASSERT_EQ(pool.begin()->value, 7);
  407. ASSERT_EQ(pool.at(0u), entities[1u]);
  408. ASSERT_EQ(pool.at(1u), entities[0u]);
  409. ASSERT_EQ(pool.at(2u), entities[2u]);
  410. ASSERT_EQ(pool.get(entities[0u]).value, 99);
  411. ASSERT_EQ(pool.get(entities[1u]).value, 2);
  412. ASSERT_EQ(pool.get(entities[2u]).value, 1);
  413. }
  414. TEST(Storage, TypeFromBase) {
  415. entt::storage<int> pool;
  416. entt::sparse_set &base = pool;
  417. entt::entity entities[2u]{entt::entity{3}, entt::entity{42}};
  418. ASSERT_FALSE(pool.contains(entities[0u]));
  419. ASSERT_FALSE(pool.contains(entities[1u]));
  420. base.emplace(entities[0u]);
  421. ASSERT_TRUE(pool.contains(entities[0u]));
  422. ASSERT_FALSE(pool.contains(entities[1u]));
  423. ASSERT_EQ(pool.get(entities[0u]), 0);
  424. base.erase(entities[0u]);
  425. base.insert(std::begin(entities), std::end(entities));
  426. ASSERT_TRUE(pool.contains(entities[0u]));
  427. ASSERT_TRUE(pool.contains(entities[1u]));
  428. ASSERT_EQ(pool.get(entities[1u]), 0);
  429. base.erase(std::begin(entities), std::end(entities));
  430. ASSERT_TRUE(pool.empty());
  431. }
  432. TEST(Storage, EmptyTypeFromBase) {
  433. entt::storage<empty_type> pool;
  434. entt::sparse_set &base = pool;
  435. entt::entity entities[2u]{entt::entity{3}, entt::entity{42}};
  436. ASSERT_FALSE(pool.contains(entities[0u]));
  437. ASSERT_FALSE(pool.contains(entities[1u]));
  438. base.emplace(entities[0u]);
  439. ASSERT_TRUE(pool.contains(entities[0u]));
  440. ASSERT_FALSE(pool.contains(entities[1u]));
  441. base.erase(entities[0u]);
  442. base.insert(std::begin(entities), std::end(entities));
  443. ASSERT_TRUE(pool.contains(entities[0u]));
  444. ASSERT_TRUE(pool.contains(entities[1u]));
  445. base.erase(std::begin(entities), std::end(entities));
  446. ASSERT_TRUE(pool.empty());
  447. }
  448. TEST(Storage, NonDefaultConstructibleTypeFromBase) {
  449. entt::storage<non_default_constructible> pool;
  450. entt::sparse_set &base = pool;
  451. entt::entity entities[2u]{entt::entity{3}, entt::entity{42}};
  452. ASSERT_FALSE(pool.contains(entities[0u]));
  453. ASSERT_FALSE(pool.contains(entities[1u]));
  454. ASSERT_DEATH(base.emplace(entities[0u]), "");
  455. ASSERT_FALSE(pool.contains(entities[0u]));
  456. ASSERT_FALSE(pool.contains(entities[1u]));
  457. ASSERT_EQ(base.find(entities[0u]), base.end());
  458. ASSERT_TRUE(pool.empty());
  459. pool.emplace(entities[0u], 3);
  460. ASSERT_TRUE(pool.contains(entities[0u]));
  461. ASSERT_FALSE(pool.contains(entities[1u]));
  462. base.erase(entities[0u]);
  463. ASSERT_TRUE(pool.empty());
  464. ASSERT_FALSE(pool.contains(entities[0u]));
  465. ASSERT_DEATH(base.insert(std::begin(entities), std::end(entities)), "");
  466. ASSERT_FALSE(pool.contains(entities[0u]));
  467. ASSERT_FALSE(pool.contains(entities[1u]));
  468. ASSERT_EQ(base.find(entities[0u]), base.end());
  469. ASSERT_EQ(base.find(entities[1u]), base.end());
  470. ASSERT_TRUE(pool.empty());
  471. }
  472. TEST(Storage, Compact) {
  473. entt::storage<stable_type> pool;
  474. ASSERT_TRUE(pool.empty());
  475. ASSERT_EQ(pool.size(), 0u);
  476. pool.compact();
  477. ASSERT_TRUE(pool.empty());
  478. ASSERT_EQ(pool.size(), 0u);
  479. pool.emplace(entt::entity{0}, stable_type{0});
  480. pool.compact();
  481. ASSERT_FALSE(pool.empty());
  482. ASSERT_EQ(pool.size(), 1u);
  483. pool.emplace(entt::entity{42}, stable_type{42});
  484. pool.erase(entt::entity{0});
  485. ASSERT_EQ(pool.size(), 2u);
  486. ASSERT_EQ(pool.index(entt::entity{42}), 1u);
  487. ASSERT_EQ(pool.get(entt::entity{42}).value, 42);
  488. pool.compact();
  489. ASSERT_EQ(pool.size(), 1u);
  490. ASSERT_EQ(pool.index(entt::entity{42}), 0u);
  491. ASSERT_EQ(pool.get(entt::entity{42}).value, 42);
  492. pool.emplace(entt::entity{0}, stable_type{0});
  493. pool.compact();
  494. ASSERT_EQ(pool.size(), 2u);
  495. ASSERT_EQ(pool.index(entt::entity{42}), 0u);
  496. ASSERT_EQ(pool.index(entt::entity{0}), 1u);
  497. ASSERT_EQ(pool.get(entt::entity{42}).value, 42);
  498. ASSERT_EQ(pool.get(entt::entity{0}).value, 0);
  499. pool.erase(entt::entity{0});
  500. pool.erase(entt::entity{42});
  501. pool.compact();
  502. ASSERT_TRUE(pool.empty());
  503. }
  504. TEST(Storage, ShrinkToFit) {
  505. entt::storage<int> pool;
  506. for(std::size_t next{}; next < ENTT_PACKED_PAGE; ++next) {
  507. pool.emplace(entt::entity(next));
  508. }
  509. pool.emplace(entt::entity{ENTT_PACKED_PAGE});
  510. pool.erase(entt::entity{ENTT_PACKED_PAGE});
  511. ASSERT_EQ(pool.capacity(), 2 * ENTT_PACKED_PAGE);
  512. ASSERT_EQ(pool.size(), ENTT_PACKED_PAGE);
  513. pool.shrink_to_fit();
  514. ASSERT_EQ(pool.capacity(), ENTT_PACKED_PAGE);
  515. ASSERT_EQ(pool.size(), ENTT_PACKED_PAGE);
  516. pool.clear();
  517. ASSERT_EQ(pool.capacity(), ENTT_PACKED_PAGE);
  518. ASSERT_EQ(pool.size(), 0u);
  519. pool.shrink_to_fit();
  520. ASSERT_EQ(pool.capacity(), 0u);
  521. ASSERT_EQ(pool.size(), 0u);
  522. }
  523. TEST(Storage, AggregatesMustWork) {
  524. struct aggregate_type { int value; };
  525. // the goal of this test is to enforce the requirements for aggregate types
  526. entt::storage<aggregate_type>{}.emplace(entt::entity{0}, 42);
  527. }
  528. TEST(Storage, TypesFromStandardTemplateLibraryMustWork) {
  529. // see #37 - this test shouldn't crash, that's all
  530. entt::storage<std::unordered_set<int>> pool;
  531. pool.emplace(entt::entity{0}).insert(42);
  532. pool.erase(entt::entity{0});
  533. }
  534. TEST(Storage, Iterator) {
  535. using iterator = typename entt::storage<boxed_int>::iterator;
  536. entt::storage<boxed_int> pool;
  537. pool.emplace(entt::entity{3}, 42);
  538. iterator end{pool.begin()};
  539. iterator begin{};
  540. begin = pool.end();
  541. std::swap(begin, end);
  542. ASSERT_EQ(begin, pool.begin());
  543. ASSERT_EQ(end, pool.end());
  544. ASSERT_NE(begin, end);
  545. ASSERT_EQ(begin++, pool.begin());
  546. ASSERT_EQ(begin--, pool.end());
  547. ASSERT_EQ(begin+1, pool.end());
  548. ASSERT_EQ(end-1, pool.begin());
  549. ASSERT_EQ(++begin, pool.end());
  550. ASSERT_EQ(--begin, pool.begin());
  551. ASSERT_EQ(begin += 1, pool.end());
  552. ASSERT_EQ(begin -= 1, pool.begin());
  553. ASSERT_EQ(begin + (end - begin), pool.end());
  554. ASSERT_EQ(begin - (begin - end), pool.end());
  555. ASSERT_EQ(end - (end - begin), pool.begin());
  556. ASSERT_EQ(end + (begin - end), pool.begin());
  557. ASSERT_EQ(begin[0u].value, pool.begin()->value);
  558. ASSERT_LT(begin, end);
  559. ASSERT_LE(begin, pool.begin());
  560. ASSERT_GT(end, begin);
  561. ASSERT_GE(end, pool.end());
  562. }
  563. TEST(Storage, ConstIterator) {
  564. using iterator = typename entt::storage<boxed_int>::const_iterator;
  565. entt::storage<boxed_int> pool;
  566. pool.emplace(entt::entity{3}, 42);
  567. iterator cend{pool.cbegin()};
  568. iterator cbegin{};
  569. cbegin = pool.cend();
  570. std::swap(cbegin, cend);
  571. ASSERT_EQ(cbegin, pool.cbegin());
  572. ASSERT_EQ(cend, pool.cend());
  573. ASSERT_NE(cbegin, cend);
  574. ASSERT_EQ(cbegin++, pool.cbegin());
  575. ASSERT_EQ(cbegin--, pool.cend());
  576. ASSERT_EQ(cbegin+1, pool.cend());
  577. ASSERT_EQ(cend-1, pool.cbegin());
  578. ASSERT_EQ(++cbegin, pool.cend());
  579. ASSERT_EQ(--cbegin, pool.cbegin());
  580. ASSERT_EQ(cbegin += 1, pool.cend());
  581. ASSERT_EQ(cbegin -= 1, pool.cbegin());
  582. ASSERT_EQ(cbegin + (cend - cbegin), pool.cend());
  583. ASSERT_EQ(cbegin - (cbegin - cend), pool.cend());
  584. ASSERT_EQ(cend - (cend - cbegin), pool.cbegin());
  585. ASSERT_EQ(cend + (cbegin - cend), pool.cbegin());
  586. ASSERT_EQ(cbegin[0u].value, pool.cbegin()->value);
  587. ASSERT_LT(cbegin, cend);
  588. ASSERT_LE(cbegin, pool.cbegin());
  589. ASSERT_GT(cend, cbegin);
  590. ASSERT_GE(cend, pool.cend());
  591. }
  592. TEST(Storage, ReverseIterator) {
  593. using reverse_iterator = typename entt::storage<boxed_int>::reverse_iterator;
  594. entt::storage<boxed_int> pool;
  595. pool.emplace(entt::entity{3}, 42);
  596. reverse_iterator end{pool.rbegin()};
  597. reverse_iterator begin{};
  598. begin = pool.rend();
  599. std::swap(begin, end);
  600. ASSERT_EQ(begin, pool.rbegin());
  601. ASSERT_EQ(end, pool.rend());
  602. ASSERT_NE(begin, end);
  603. ASSERT_EQ(begin++, pool.rbegin());
  604. ASSERT_EQ(begin--, pool.rend());
  605. ASSERT_EQ(begin+1, pool.rend());
  606. ASSERT_EQ(end-1, pool.rbegin());
  607. ASSERT_EQ(++begin, pool.rend());
  608. ASSERT_EQ(--begin, pool.rbegin());
  609. ASSERT_EQ(begin += 1, pool.rend());
  610. ASSERT_EQ(begin -= 1, pool.rbegin());
  611. ASSERT_EQ(begin + (end - begin), pool.rend());
  612. ASSERT_EQ(begin - (begin - end), pool.rend());
  613. ASSERT_EQ(end - (end - begin), pool.rbegin());
  614. ASSERT_EQ(end + (begin - end), pool.rbegin());
  615. ASSERT_EQ(begin[0u].value, pool.rbegin()->value);
  616. ASSERT_LT(begin, end);
  617. ASSERT_LE(begin, pool.rbegin());
  618. ASSERT_GT(end, begin);
  619. ASSERT_GE(end, pool.rend());
  620. }
  621. TEST(Storage, ConstReverseIterator) {
  622. using const_reverse_iterator = typename entt::storage<boxed_int>::const_reverse_iterator;
  623. entt::storage<boxed_int> pool;
  624. pool.emplace(entt::entity{3}, 42);
  625. const_reverse_iterator cend{pool.crbegin()};
  626. const_reverse_iterator cbegin{};
  627. cbegin = pool.crend();
  628. std::swap(cbegin, cend);
  629. ASSERT_EQ(cbegin, pool.crbegin());
  630. ASSERT_EQ(cend, pool.crend());
  631. ASSERT_NE(cbegin, cend);
  632. ASSERT_EQ(cbegin++, pool.crbegin());
  633. ASSERT_EQ(cbegin--, pool.crend());
  634. ASSERT_EQ(cbegin+1, pool.crend());
  635. ASSERT_EQ(cend-1, pool.crbegin());
  636. ASSERT_EQ(++cbegin, pool.crend());
  637. ASSERT_EQ(--cbegin, pool.crbegin());
  638. ASSERT_EQ(cbegin += 1, pool.crend());
  639. ASSERT_EQ(cbegin -= 1, pool.crbegin());
  640. ASSERT_EQ(cbegin + (cend - cbegin), pool.crend());
  641. ASSERT_EQ(cbegin - (cbegin - cend), pool.crend());
  642. ASSERT_EQ(cend - (cend - cbegin), pool.crbegin());
  643. ASSERT_EQ(cend + (cbegin - cend), pool.crbegin());
  644. ASSERT_EQ(cbegin[0u].value, pool.crbegin()->value);
  645. ASSERT_LT(cbegin, cend);
  646. ASSERT_LE(cbegin, pool.crbegin());
  647. ASSERT_GT(cend, cbegin);
  648. ASSERT_GE(cend, pool.crend());
  649. }
  650. TEST(Storage, Raw) {
  651. entt::storage<int> pool;
  652. pool.emplace(entt::entity{3}, 3);
  653. pool.emplace(entt::entity{12}, 6);
  654. pool.emplace(entt::entity{42}, 9);
  655. ASSERT_EQ(pool.get(entt::entity{3}), 3);
  656. ASSERT_EQ(std::as_const(pool).get(entt::entity{12}), 6);
  657. ASSERT_EQ(pool.get(entt::entity{42}), 9);
  658. ASSERT_EQ(pool.raw()[0u][0u], 3);
  659. ASSERT_EQ(std::as_const(pool).raw()[0u][1u], 6);
  660. ASSERT_EQ(pool.raw()[0u][2u], 9);
  661. }
  662. TEST(Storage, SortOrdered) {
  663. entt::storage<boxed_int> pool;
  664. entt::entity entities[5u]{entt::entity{12}, entt::entity{42}, entt::entity{7}, entt::entity{3}, entt::entity{9}};
  665. boxed_int values[5u]{{12}, {9}, {6}, {3}, {1}};
  666. pool.insert(std::begin(entities), std::end(entities), values);
  667. pool.sort([&pool](auto lhs, auto rhs) { return pool.get(lhs).value < pool.get(rhs).value; });
  668. ASSERT_TRUE(std::equal(std::rbegin(entities), std::rend(entities), pool.entt::sparse_set::begin(), pool.entt::sparse_set::end()));
  669. ASSERT_TRUE(std::equal(std::rbegin(values), std::rend(values), pool.begin(), pool.end()));
  670. }
  671. TEST(Storage, SortReverse) {
  672. entt::storage<boxed_int> pool;
  673. entt::entity entities[5u]{entt::entity{12}, entt::entity{42}, entt::entity{7}, entt::entity{3}, entt::entity{9}};
  674. boxed_int values[5u]{{1}, {3}, {6}, {9}, {12}};
  675. pool.insert(std::begin(entities), std::end(entities), values);
  676. pool.sort([&pool](auto lhs, auto rhs) { return pool.get(lhs).value < pool.get(rhs).value; });
  677. ASSERT_TRUE(std::equal(std::begin(entities), std::end(entities), pool.entt::sparse_set::begin(), pool.entt::sparse_set::end()));
  678. ASSERT_TRUE(std::equal(std::begin(values), std::end(values), pool.begin(), pool.end()));
  679. }
  680. TEST(Storage, SortUnordered) {
  681. entt::storage<boxed_int> pool;
  682. entt::entity entities[5u]{entt::entity{12}, entt::entity{42}, entt::entity{7}, entt::entity{3}, entt::entity{9}};
  683. boxed_int values[5u]{{6}, {3}, {1}, {9}, {12}};
  684. pool.insert(std::begin(entities), std::end(entities), values);
  685. pool.sort([&pool](auto lhs, auto rhs) { return pool.get(lhs).value < pool.get(rhs).value; });
  686. auto begin = pool.begin();
  687. auto end = pool.end();
  688. ASSERT_EQ(*(begin++), values[2u]);
  689. ASSERT_EQ(*(begin++), values[1u]);
  690. ASSERT_EQ(*(begin++), values[0u]);
  691. ASSERT_EQ(*(begin++), values[3u]);
  692. ASSERT_EQ(*(begin++), values[4u]);
  693. ASSERT_EQ(begin, end);
  694. ASSERT_EQ(pool.data()[0u], entities[4u]);
  695. ASSERT_EQ(pool.data()[1u], entities[3u]);
  696. ASSERT_EQ(pool.data()[2u], entities[0u]);
  697. ASSERT_EQ(pool.data()[3u], entities[1u]);
  698. ASSERT_EQ(pool.data()[4u], entities[2u]);
  699. }
  700. TEST(Storage, SortRange) {
  701. entt::storage<boxed_int> pool;
  702. entt::entity entities[5u]{entt::entity{12}, entt::entity{42}, entt::entity{7}, entt::entity{3}, entt::entity{9}};
  703. boxed_int values[5u]{{3}, {6}, {1}, {9}, {12}};
  704. pool.insert(std::begin(entities), std::end(entities), values);
  705. pool.sort_n(0u, [&pool](auto lhs, auto rhs) { return pool.get(lhs).value < pool.get(rhs).value; });
  706. ASSERT_TRUE(std::equal(std::rbegin(entities), std::rend(entities), pool.entt::sparse_set::begin(), pool.entt::sparse_set::end()));
  707. ASSERT_TRUE(std::equal(std::rbegin(values), std::rend(values), pool.begin(), pool.end()));
  708. pool.sort_n(2u, [&pool](auto lhs, auto rhs) { return pool.get(lhs).value < pool.get(rhs).value; });
  709. ASSERT_EQ(pool.raw()[0u][0u], values[1u]);
  710. ASSERT_EQ(pool.raw()[0u][1u], values[0u]);
  711. ASSERT_EQ(pool.raw()[0u][2u], values[2u]);
  712. ASSERT_EQ(pool.data()[0u], entities[1u]);
  713. ASSERT_EQ(pool.data()[1u], entities[0u]);
  714. ASSERT_EQ(pool.data()[2u], entities[2u]);
  715. pool.sort_n(5u, [&pool](auto lhs, auto rhs) { return pool.get(lhs).value < pool.get(rhs).value; });
  716. auto begin = pool.begin();
  717. auto end = pool.end();
  718. ASSERT_EQ(*(begin++), values[2u]);
  719. ASSERT_EQ(*(begin++), values[0u]);
  720. ASSERT_EQ(*(begin++), values[1u]);
  721. ASSERT_EQ(*(begin++), values[3u]);
  722. ASSERT_EQ(*(begin++), values[4u]);
  723. ASSERT_EQ(begin, end);
  724. ASSERT_EQ(pool.data()[0u], entities[4u]);
  725. ASSERT_EQ(pool.data()[1u], entities[3u]);
  726. ASSERT_EQ(pool.data()[2u], entities[1u]);
  727. ASSERT_EQ(pool.data()[3u], entities[0u]);
  728. ASSERT_EQ(pool.data()[4u], entities[2u]);
  729. }
  730. TEST(Storage, RespectDisjoint) {
  731. entt::storage<int> lhs;
  732. entt::storage<int> rhs;
  733. entt::entity lhs_entities[3u]{entt::entity{3}, entt::entity{12}, entt::entity{42}};
  734. int lhs_values[3u]{3, 6, 9};
  735. lhs.insert(std::begin(lhs_entities), std::end(lhs_entities), lhs_values);
  736. ASSERT_TRUE(std::equal(std::rbegin(lhs_entities), std::rend(lhs_entities), lhs.entt::sparse_set::begin(), lhs.entt::sparse_set::end()));
  737. ASSERT_TRUE(std::equal(std::rbegin(lhs_values), std::rend(lhs_values), lhs.begin(), lhs.end()));
  738. lhs.respect(rhs);
  739. ASSERT_TRUE(std::equal(std::rbegin(lhs_entities), std::rend(lhs_entities), lhs.entt::sparse_set::begin(), lhs.entt::sparse_set::end()));
  740. ASSERT_TRUE(std::equal(std::rbegin(lhs_values), std::rend(lhs_values), lhs.begin(), lhs.end()));
  741. }
  742. TEST(Storage, RespectOverlap) {
  743. entt::storage<int> lhs;
  744. entt::storage<int> rhs;
  745. entt::entity lhs_entities[3u]{entt::entity{3}, entt::entity{12}, entt::entity{42}};
  746. int lhs_values[3u]{3, 6, 9};
  747. lhs.insert(std::begin(lhs_entities), std::end(lhs_entities), lhs_values);
  748. entt::entity rhs_entities[1u]{entt::entity{12}};
  749. int rhs_values[1u]{6};
  750. rhs.insert(std::begin(rhs_entities), std::end(rhs_entities), rhs_values);
  751. ASSERT_TRUE(std::equal(std::rbegin(lhs_entities), std::rend(lhs_entities), lhs.entt::sparse_set::begin(), lhs.entt::sparse_set::end()));
  752. ASSERT_TRUE(std::equal(std::rbegin(lhs_values), std::rend(lhs_values), lhs.begin(), lhs.end()));
  753. ASSERT_TRUE(std::equal(std::rbegin(rhs_entities), std::rend(rhs_entities), rhs.entt::sparse_set::begin(), rhs.entt::sparse_set::end()));
  754. ASSERT_TRUE(std::equal(std::rbegin(rhs_values), std::rend(rhs_values), rhs.begin(), rhs.end()));
  755. lhs.respect(rhs);
  756. auto begin = lhs.begin();
  757. auto end = lhs.end();
  758. ASSERT_EQ(*(begin++), lhs_values[1u]);
  759. ASSERT_EQ(*(begin++), lhs_values[2u]);
  760. ASSERT_EQ(*(begin++), lhs_values[0u]);
  761. ASSERT_EQ(begin, end);
  762. ASSERT_EQ(lhs.data()[0u], lhs_entities[0u]);
  763. ASSERT_EQ(lhs.data()[1u], lhs_entities[2u]);
  764. ASSERT_EQ(lhs.data()[2u], lhs_entities[1u]);
  765. }
  766. TEST(Storage, RespectOrdered) {
  767. entt::storage<int> lhs;
  768. entt::storage<int> rhs;
  769. entt::entity lhs_entities[5u]{entt::entity{1}, entt::entity{2}, entt::entity{3}, entt::entity{4}, entt::entity{5}};
  770. int lhs_values[5u]{1, 2, 3, 4, 5};
  771. lhs.insert(std::begin(lhs_entities), std::end(lhs_entities), lhs_values);
  772. entt::entity rhs_entities[6u]{entt::entity{6}, entt::entity{1}, entt::entity{2}, entt::entity{3}, entt::entity{4}, entt::entity{5}};
  773. int rhs_values[6u]{6, 1, 2, 3, 4, 5};
  774. rhs.insert(std::begin(rhs_entities), std::end(rhs_entities), rhs_values);
  775. ASSERT_TRUE(std::equal(std::rbegin(lhs_entities), std::rend(lhs_entities), lhs.entt::sparse_set::begin(), lhs.entt::sparse_set::end()));
  776. ASSERT_TRUE(std::equal(std::rbegin(lhs_values), std::rend(lhs_values), lhs.begin(), lhs.end()));
  777. ASSERT_TRUE(std::equal(std::rbegin(rhs_entities), std::rend(rhs_entities), rhs.entt::sparse_set::begin(), rhs.entt::sparse_set::end()));
  778. ASSERT_TRUE(std::equal(std::rbegin(rhs_values), std::rend(rhs_values), rhs.begin(), rhs.end()));
  779. rhs.respect(lhs);
  780. ASSERT_TRUE(std::equal(std::rbegin(rhs_entities), std::rend(rhs_entities), rhs.entt::sparse_set::begin(), rhs.entt::sparse_set::end()));
  781. ASSERT_TRUE(std::equal(std::rbegin(rhs_values), std::rend(rhs_values), rhs.begin(), rhs.end()));
  782. }
  783. TEST(Storage, RespectReverse) {
  784. entt::storage<int> lhs;
  785. entt::storage<int> rhs;
  786. entt::entity lhs_entities[5u]{entt::entity{1}, entt::entity{2}, entt::entity{3}, entt::entity{4}, entt::entity{5}};
  787. int lhs_values[5u]{1, 2, 3, 4, 5};
  788. lhs.insert(std::begin(lhs_entities), std::end(lhs_entities), lhs_values);
  789. entt::entity rhs_entities[6u]{entt::entity{5}, entt::entity{4}, entt::entity{3}, entt::entity{2}, entt::entity{1}, entt::entity{6}};
  790. int rhs_values[6u]{5, 4, 3, 2, 1, 6};
  791. rhs.insert(std::begin(rhs_entities), std::end(rhs_entities), rhs_values);
  792. ASSERT_TRUE(std::equal(std::rbegin(lhs_entities), std::rend(lhs_entities), lhs.entt::sparse_set::begin(), lhs.entt::sparse_set::end()));
  793. ASSERT_TRUE(std::equal(std::rbegin(lhs_values), std::rend(lhs_values), lhs.begin(), lhs.end()));
  794. ASSERT_TRUE(std::equal(std::rbegin(rhs_entities), std::rend(rhs_entities), rhs.entt::sparse_set::begin(), rhs.entt::sparse_set::end()));
  795. ASSERT_TRUE(std::equal(std::rbegin(rhs_values), std::rend(rhs_values), rhs.begin(), rhs.end()));
  796. rhs.respect(lhs);
  797. auto begin = rhs.begin();
  798. auto end = rhs.end();
  799. ASSERT_EQ(*(begin++), rhs_values[0u]);
  800. ASSERT_EQ(*(begin++), rhs_values[1u]);
  801. ASSERT_EQ(*(begin++), rhs_values[2u]);
  802. ASSERT_EQ(*(begin++), rhs_values[3u]);
  803. ASSERT_EQ(*(begin++), rhs_values[4u]);
  804. ASSERT_EQ(*(begin++), rhs_values[5u]);
  805. ASSERT_EQ(begin, end);
  806. ASSERT_EQ(rhs.data()[0u], rhs_entities[5u]);
  807. ASSERT_EQ(rhs.data()[1u], rhs_entities[4u]);
  808. ASSERT_EQ(rhs.data()[2u], rhs_entities[3u]);
  809. ASSERT_EQ(rhs.data()[3u], rhs_entities[2u]);
  810. ASSERT_EQ(rhs.data()[4u], rhs_entities[1u]);
  811. ASSERT_EQ(rhs.data()[5u], rhs_entities[0u]);
  812. }
  813. TEST(Storage, RespectUnordered) {
  814. entt::storage<int> lhs;
  815. entt::storage<int> rhs;
  816. entt::entity lhs_entities[5u]{entt::entity{1}, entt::entity{2}, entt::entity{3}, entt::entity{4}, entt::entity{5}};
  817. int lhs_values[5u]{1, 2, 3, 4, 5};
  818. lhs.insert(std::begin(lhs_entities), std::end(lhs_entities), lhs_values);
  819. entt::entity rhs_entities[6u]{entt::entity{3}, entt::entity{2}, entt::entity{6}, entt::entity{1}, entt::entity{4}, entt::entity{5}};
  820. int rhs_values[6u]{3, 2, 6, 1, 4, 5};
  821. rhs.insert(std::begin(rhs_entities), std::end(rhs_entities), rhs_values);
  822. ASSERT_TRUE(std::equal(std::rbegin(lhs_entities), std::rend(lhs_entities), lhs.entt::sparse_set::begin(), lhs.entt::sparse_set::end()));
  823. ASSERT_TRUE(std::equal(std::rbegin(lhs_values), std::rend(lhs_values), lhs.begin(), lhs.end()));
  824. ASSERT_TRUE(std::equal(std::rbegin(rhs_entities), std::rend(rhs_entities), rhs.entt::sparse_set::begin(), rhs.entt::sparse_set::end()));
  825. ASSERT_TRUE(std::equal(std::rbegin(rhs_values), std::rend(rhs_values), rhs.begin(), rhs.end()));
  826. rhs.respect(lhs);
  827. auto begin = rhs.begin();
  828. auto end = rhs.end();
  829. ASSERT_EQ(*(begin++), rhs_values[5u]);
  830. ASSERT_EQ(*(begin++), rhs_values[4u]);
  831. ASSERT_EQ(*(begin++), rhs_values[0u]);
  832. ASSERT_EQ(*(begin++), rhs_values[1u]);
  833. ASSERT_EQ(*(begin++), rhs_values[3u]);
  834. ASSERT_EQ(*(begin++), rhs_values[2u]);
  835. ASSERT_EQ(begin, end);
  836. ASSERT_EQ(rhs.data()[0u], rhs_entities[2u]);
  837. ASSERT_EQ(rhs.data()[1u], rhs_entities[3u]);
  838. ASSERT_EQ(rhs.data()[2u], rhs_entities[1u]);
  839. ASSERT_EQ(rhs.data()[3u], rhs_entities[0u]);
  840. ASSERT_EQ(rhs.data()[4u], rhs_entities[4u]);
  841. ASSERT_EQ(rhs.data()[5u], rhs_entities[5u]);
  842. }
  843. TEST(Storage, CanModifyDuringIteration) {
  844. entt::storage<int> pool;
  845. auto *ptr = &pool.emplace(entt::entity{0}, 42);
  846. ASSERT_EQ(pool.capacity(), ENTT_PACKED_PAGE);
  847. const auto it = pool.cbegin();
  848. pool.reserve(ENTT_PACKED_PAGE + 1u);
  849. ASSERT_EQ(pool.capacity(), 2 * ENTT_PACKED_PAGE);
  850. ASSERT_EQ(&pool.get(entt::entity{0}), ptr);
  851. // this should crash with asan enabled if we break the constraint
  852. [[maybe_unused]] const int &value = *it;
  853. }
  854. TEST(Storage, ReferencesGuaranteed) {
  855. entt::storage<boxed_int> pool;
  856. pool.emplace(entt::entity{0}, 0);
  857. pool.emplace(entt::entity{1}, 1);
  858. ASSERT_EQ(pool.get(entt::entity{0}).value, 0);
  859. ASSERT_EQ(pool.get(entt::entity{1}).value, 1);
  860. for(auto &&type: pool) {
  861. if(type.value) {
  862. type.value = 42;
  863. }
  864. }
  865. ASSERT_EQ(pool.get(entt::entity{0}).value, 0);
  866. ASSERT_EQ(pool.get(entt::entity{1}).value, 42);
  867. auto begin = pool.begin();
  868. while(begin != pool.end()) {
  869. (begin++)->value = 3;
  870. }
  871. ASSERT_EQ(pool.get(entt::entity{0}).value, 3);
  872. ASSERT_EQ(pool.get(entt::entity{1}).value, 3);
  873. }
  874. TEST(Storage, MoveOnlyComponent) {
  875. // the purpose is to ensure that move only components are always accepted
  876. [[maybe_unused]] entt::storage<std::unique_ptr<int>> pool;
  877. }
  878. TEST(Storage, UpdateFromDestructor) {
  879. static constexpr auto size = 10u;
  880. auto test = [](const auto target) {
  881. entt::storage<update_from_destructor> pool;
  882. for(std::size_t next{}; next < size; ++next) {
  883. const auto entity = entt::entity(next);
  884. pool.emplace(entity, pool, entity == entt::entity(size/2) ? target : entity);
  885. }
  886. pool.erase(entt::entity(size/2));
  887. ASSERT_EQ(pool.size(), size - 1u - (target != entt::null));
  888. ASSERT_FALSE(pool.contains(entt::entity(size/2)));
  889. ASSERT_FALSE(pool.contains(target));
  890. pool.clear();
  891. ASSERT_TRUE(pool.empty());
  892. for(std::size_t next{}; next < size; ++next) {
  893. ASSERT_FALSE(pool.contains(entt::entity(next)));
  894. }
  895. };
  896. test(entt::entity(size - 1u));
  897. test(entt::entity(size - 2u));
  898. test(entt::entity{0u});
  899. }
  900. TEST(Storage, CustomAllocator) {
  901. test::throwing_allocator<entt::entity> allocator{};
  902. entt::basic_storage<entt::entity, int, test::throwing_allocator<entt::entity>> pool{allocator};
  903. ASSERT_EQ(pool.get_allocator(), allocator);
  904. pool.reserve(1u);
  905. ASSERT_EQ(pool.capacity(), ENTT_PACKED_PAGE);
  906. pool.emplace(entt::entity{0}, 3);
  907. pool.emplace(entt::entity{1}, 42);
  908. entt::basic_storage<entt::entity, int, test::throwing_allocator<entt::entity>> other{std::move(pool), allocator};
  909. ASSERT_TRUE(pool.empty());
  910. ASSERT_FALSE(other.empty());
  911. ASSERT_EQ(pool.capacity(), 0u);
  912. ASSERT_EQ(other.capacity(), ENTT_PACKED_PAGE);
  913. ASSERT_EQ(other.size(), 2u);
  914. pool = std::move(other);
  915. ASSERT_FALSE(pool.empty());
  916. ASSERT_TRUE(other.empty());
  917. ASSERT_EQ(other.capacity(), 0u);
  918. ASSERT_EQ(pool.capacity(), ENTT_PACKED_PAGE);
  919. ASSERT_EQ(pool.size(), 2u);
  920. pool.swap(other);
  921. pool = std::move(other);
  922. ASSERT_FALSE(pool.empty());
  923. ASSERT_TRUE(other.empty());
  924. ASSERT_EQ(other.capacity(), 0u);
  925. ASSERT_EQ(pool.capacity(), ENTT_PACKED_PAGE);
  926. ASSERT_EQ(pool.size(), 2u);
  927. pool.clear();
  928. ASSERT_EQ(pool.capacity(), ENTT_PACKED_PAGE);
  929. ASSERT_EQ(pool.size(), 0u);
  930. pool.shrink_to_fit();
  931. ASSERT_EQ(pool.capacity(), 0u);
  932. }
  933. TEST(Storage, ThrowingAllocator) {
  934. entt::basic_storage<entt::entity, int, test::throwing_allocator<int>> pool;
  935. typename std::decay_t<decltype(pool)>::base_type &base = pool;
  936. test::throwing_allocator<int>::trigger_on_allocate = true;
  937. // strong exception safety
  938. ASSERT_THROW(pool.reserve(1u), test::throwing_allocator<int>::exception_type);
  939. ASSERT_EQ(pool.capacity(), 0u);
  940. test::throwing_allocator<int>::trigger_after_allocate = true;
  941. // strong exception safety
  942. ASSERT_THROW(pool.reserve(2 * ENTT_PACKED_PAGE), test::throwing_allocator<int>::exception_type);
  943. ASSERT_EQ(pool.capacity(), 0u);
  944. pool.shrink_to_fit();
  945. test::throwing_allocator<int>::trigger_on_allocate = true;
  946. // strong exception safety
  947. ASSERT_THROW(pool.emplace(entt::entity{0}, 0), test::throwing_allocator<int>::exception_type);
  948. ASSERT_FALSE(pool.contains(entt::entity{0}));
  949. ASSERT_TRUE(pool.empty());
  950. test::throwing_allocator<entt::entity>::trigger_on_allocate = true;
  951. // strong exception safety
  952. ASSERT_THROW(pool.emplace(entt::entity{0}, 0), test::throwing_allocator<entt::entity>::exception_type);
  953. ASSERT_FALSE(pool.contains(entt::entity{0}));
  954. ASSERT_TRUE(pool.empty());
  955. test::throwing_allocator<entt::entity>::trigger_on_allocate = true;
  956. // strong exception safety
  957. ASSERT_THROW(base.emplace(entt::entity{0}), test::throwing_allocator<entt::entity>::exception_type);
  958. ASSERT_FALSE(base.contains(entt::entity{0}));
  959. ASSERT_TRUE(base.empty());
  960. pool.emplace(entt::entity{0}, 0);
  961. const entt::entity entities[2u]{entt::entity{1}, entt::entity{ENTT_SPARSE_PAGE}};
  962. test::throwing_allocator<entt::entity>::trigger_after_allocate = true;
  963. // basic exception safety
  964. ASSERT_THROW(pool.insert(std::begin(entities), std::end(entities), 0), test::throwing_allocator<entt::entity>::exception_type);
  965. ASSERT_TRUE(pool.contains(entt::entity{1}));
  966. ASSERT_FALSE(pool.contains(entt::entity{ENTT_SPARSE_PAGE}));
  967. pool.erase(entt::entity{1});
  968. const int components[2u]{1, ENTT_SPARSE_PAGE};
  969. test::throwing_allocator<entt::entity>::trigger_on_allocate = true;
  970. // basic exception safety
  971. ASSERT_THROW(pool.insert(std::begin(entities), std::end(entities), std::begin(components)), test::throwing_allocator<entt::entity>::exception_type);
  972. ASSERT_TRUE(pool.contains(entt::entity{1}));
  973. ASSERT_FALSE(pool.contains(entt::entity{ENTT_SPARSE_PAGE}));
  974. }
  975. TEST(Storage, ThrowingComponent) {
  976. entt::storage<test::throwing_component> pool;
  977. test::throwing_component::trigger_on_value = 42;
  978. // strong exception safety
  979. ASSERT_THROW(pool.emplace(entt::entity{0}, test::throwing_component{42}), typename test::throwing_component::exception_type);
  980. ASSERT_TRUE(pool.empty());
  981. const entt::entity entities[2u]{entt::entity{42}, entt::entity{1}};
  982. const test::throwing_component components[2u]{42, 1};
  983. // basic exception safety
  984. ASSERT_THROW(pool.insert(std::begin(entities), std::end(entities), test::throwing_component{42}), typename test::throwing_component::exception_type);
  985. ASSERT_EQ(pool.size(), 0u);
  986. ASSERT_FALSE(pool.contains(entt::entity{1}));
  987. // basic exception safety
  988. ASSERT_THROW(pool.insert(std::begin(entities), std::end(entities), std::begin(components)), typename test::throwing_component::exception_type);
  989. ASSERT_EQ(pool.size(), 0u);
  990. ASSERT_FALSE(pool.contains(entt::entity{1}));
  991. // basic exception safety
  992. ASSERT_THROW(pool.insert(std::rbegin(entities), std::rend(entities), std::rbegin(components)), typename test::throwing_component::exception_type);
  993. ASSERT_EQ(pool.size(), 1u);
  994. ASSERT_TRUE(pool.contains(entt::entity{1}));
  995. ASSERT_EQ(pool.get(entt::entity{1}), 1);
  996. pool.clear();
  997. pool.emplace(entt::entity{1}, 1);
  998. pool.emplace(entt::entity{42}, 42);
  999. // basic exception safety
  1000. ASSERT_THROW(pool.erase(entt::entity{1}), typename test::throwing_component::exception_type);
  1001. ASSERT_EQ(pool.size(), 2u);
  1002. ASSERT_TRUE(pool.contains(entt::entity{42}));
  1003. ASSERT_TRUE(pool.contains(entt::entity{1}));
  1004. ASSERT_EQ(pool.at(0u), entt::entity{1});
  1005. ASSERT_EQ(pool.at(1u), entt::entity{42});
  1006. ASSERT_EQ(pool.get(entt::entity{42}), 42);
  1007. // the element may have been moved but it's still there
  1008. ASSERT_EQ(pool.get(entt::entity{1}), test::throwing_component::moved_from_value);
  1009. test::throwing_component::trigger_on_value = 99;
  1010. pool.erase(entt::entity{1});
  1011. ASSERT_EQ(pool.size(), 1u);
  1012. ASSERT_TRUE(pool.contains(entt::entity{42}));
  1013. ASSERT_FALSE(pool.contains(entt::entity{1}));
  1014. ASSERT_EQ(pool.at(0u), entt::entity{42});
  1015. ASSERT_EQ(pool.get(entt::entity{42}), 42);
  1016. }