snapshot.cpp 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525
  1. #include <cstddef>
  2. #include <map>
  3. #include <memory>
  4. #include <queue>
  5. #include <tuple>
  6. #include <type_traits>
  7. #include <vector>
  8. #include <gtest/gtest.h>
  9. #include <entt/core/any.hpp>
  10. #include <entt/core/hashed_string.hpp>
  11. #include <entt/entity/entity.hpp>
  12. #include <entt/entity/registry.hpp>
  13. #include <entt/entity/snapshot.hpp>
  14. #include <entt/signal/sigh.hpp>
  15. struct empty {};
  16. struct shadow {
  17. entt::entity target{entt::null};
  18. static void listener(entt::entity &elem, entt::registry &registry, const entt::entity entt) {
  19. elem = registry.get<shadow>(entt).target;
  20. }
  21. };
  22. TEST(BasicSnapshot, Constructors) {
  23. static_assert(!std::is_default_constructible_v<entt::basic_snapshot<entt::registry>>);
  24. static_assert(!std::is_copy_constructible_v<entt::basic_snapshot<entt::registry>>);
  25. static_assert(!std::is_copy_assignable_v<entt::basic_snapshot<entt::registry>>);
  26. static_assert(std::is_move_constructible_v<entt::basic_snapshot<entt::registry>>);
  27. static_assert(std::is_move_assignable_v<entt::basic_snapshot<entt::registry>>);
  28. entt::registry registry;
  29. entt::basic_snapshot snapshot{registry};
  30. entt::basic_snapshot other{std::move(snapshot)};
  31. ASSERT_NO_FATAL_FAILURE(snapshot = std::move(other));
  32. }
  33. TEST(BasicSnapshot, GetEntityType) {
  34. using namespace entt::literals;
  35. using traits_type = entt::entt_traits<entt::entity>;
  36. entt::registry registry;
  37. entt::basic_snapshot snapshot{registry};
  38. const auto &storage = registry.storage<entt::entity>();
  39. std::vector<entt::any> data{};
  40. auto archive = [&data](auto &&value) { data.emplace_back(std::forward<decltype(value)>(value)); };
  41. snapshot.get<entt::entity>(archive);
  42. ASSERT_EQ(data.size(), 2u);
  43. ASSERT_NE(entt::any_cast<typename traits_type::entity_type>(&data[0u]), nullptr);
  44. ASSERT_EQ(entt::any_cast<typename traits_type::entity_type>(data[0u]), storage.size());
  45. ASSERT_NE(entt::any_cast<typename traits_type::entity_type>(&data[1u]), nullptr);
  46. ASSERT_EQ(entt::any_cast<typename traits_type::entity_type>(data[1u]), storage.in_use());
  47. entt::entity entities[3u];
  48. registry.create(std::begin(entities), std::end(entities));
  49. registry.destroy(entities[1u]);
  50. data.clear();
  51. snapshot.get<entt::entity>(archive, "ignored"_hs);
  52. ASSERT_EQ(data.size(), 5u);
  53. ASSERT_NE(entt::any_cast<typename traits_type::entity_type>(&data[0u]), nullptr);
  54. ASSERT_EQ(entt::any_cast<typename traits_type::entity_type>(data[0u]), storage.size());
  55. ASSERT_NE(entt::any_cast<typename traits_type::entity_type>(&data[1u]), nullptr);
  56. ASSERT_EQ(entt::any_cast<typename traits_type::entity_type>(data[1u]), storage.in_use());
  57. ASSERT_NE(entt::any_cast<entt::entity>(&data[2u]), nullptr);
  58. ASSERT_EQ(entt::any_cast<entt::entity>(data[2u]), storage.data()[0u]);
  59. ASSERT_NE(entt::any_cast<entt::entity>(&data[3u]), nullptr);
  60. ASSERT_EQ(entt::any_cast<entt::entity>(data[3u]), storage.data()[1u]);
  61. ASSERT_NE(entt::any_cast<entt::entity>(&data[4u]), nullptr);
  62. ASSERT_EQ(entt::any_cast<entt::entity>(data[4u]), storage.data()[2u]);
  63. }
  64. TEST(BasicSnapshot, GetType) {
  65. using namespace entt::literals;
  66. using traits_type = entt::entt_traits<entt::entity>;
  67. entt::registry registry;
  68. entt::basic_snapshot snapshot{registry};
  69. const auto &storage = registry.storage<int>();
  70. entt::entity entities[3u];
  71. const int values[3u]{1, 2, 3};
  72. registry.create(std::begin(entities), std::end(entities));
  73. registry.insert<int>(std::begin(entities), std::end(entities), std::begin(values));
  74. registry.destroy(entities[1u]);
  75. std::vector<entt::any> data{};
  76. auto archive = [&data](auto &&value) { data.emplace_back(std::forward<decltype(value)>(value)); };
  77. snapshot.get<int>(archive, "other"_hs);
  78. ASSERT_EQ(data.size(), 1u);
  79. ASSERT_NE(entt::any_cast<typename traits_type::entity_type>(&data[0u]), nullptr);
  80. ASSERT_EQ(entt::any_cast<typename traits_type::entity_type>(data[0u]), 0u);
  81. data.clear();
  82. snapshot.get<int>(archive);
  83. ASSERT_EQ(data.size(), 5u);
  84. ASSERT_NE(entt::any_cast<typename traits_type::entity_type>(&data[0u]), nullptr);
  85. ASSERT_EQ(entt::any_cast<typename traits_type::entity_type>(data[0u]), storage.size());
  86. ASSERT_NE(entt::any_cast<entt::entity>(&data[1u]), nullptr);
  87. ASSERT_EQ(entt::any_cast<entt::entity>(data[1u]), entities[0u]);
  88. ASSERT_NE(entt::any_cast<int>(&data[2u]), nullptr);
  89. ASSERT_EQ(entt::any_cast<int>(data[2u]), values[0u]);
  90. ASSERT_NE(entt::any_cast<entt::entity>(&data[3u]), nullptr);
  91. ASSERT_EQ(entt::any_cast<entt::entity>(data[3u]), entities[2u]);
  92. ASSERT_NE(entt::any_cast<int>(&data[4u]), nullptr);
  93. ASSERT_EQ(entt::any_cast<int>(data[4u]), values[2u]);
  94. }
  95. TEST(BasicSnapshot, GetEmptyType) {
  96. using namespace entt::literals;
  97. using traits_type = entt::entt_traits<entt::entity>;
  98. entt::registry registry;
  99. entt::basic_snapshot snapshot{registry};
  100. const auto &storage = registry.storage<empty>();
  101. entt::entity entities[3u];
  102. registry.create(std::begin(entities), std::end(entities));
  103. registry.insert<empty>(std::begin(entities), std::end(entities));
  104. registry.destroy(entities[1u]);
  105. std::vector<entt::any> data{};
  106. auto archive = [&data](auto &&value) { data.emplace_back(std::forward<decltype(value)>(value)); };
  107. snapshot.get<empty>(archive, "other"_hs);
  108. ASSERT_EQ(data.size(), 1u);
  109. ASSERT_NE(entt::any_cast<typename traits_type::entity_type>(&data[0u]), nullptr);
  110. ASSERT_EQ(entt::any_cast<typename traits_type::entity_type>(data[0u]), 0u);
  111. data.clear();
  112. snapshot.get<empty>(archive);
  113. ASSERT_EQ(data.size(), 3u);
  114. ASSERT_NE(entt::any_cast<typename traits_type::entity_type>(&data[0u]), nullptr);
  115. ASSERT_EQ(entt::any_cast<typename traits_type::entity_type>(data[0u]), storage.size());
  116. ASSERT_NE(entt::any_cast<entt::entity>(&data[1u]), nullptr);
  117. ASSERT_EQ(entt::any_cast<entt::entity>(data[1u]), entities[0u]);
  118. ASSERT_NE(entt::any_cast<entt::entity>(&data[2u]), nullptr);
  119. ASSERT_EQ(entt::any_cast<entt::entity>(data[2u]), entities[2u]);
  120. }
  121. TEST(BasicSnapshot, GetTypeSparse) {
  122. using namespace entt::literals;
  123. using traits_type = entt::entt_traits<entt::entity>;
  124. entt::registry registry;
  125. entt::basic_snapshot snapshot{registry};
  126. entt::entity entities[3u];
  127. const int values[3u]{1, 2, 3};
  128. registry.create(std::begin(entities), std::end(entities));
  129. registry.insert<int>(std::begin(entities), std::end(entities), std::begin(values));
  130. registry.destroy(entities[1u]);
  131. std::vector<entt::any> data{};
  132. auto archive = [&data](auto &&value) { data.emplace_back(std::forward<decltype(value)>(value)); };
  133. snapshot.get<int>(archive, std::begin(entities), std::end(entities), "other"_hs);
  134. ASSERT_EQ(data.size(), 1u);
  135. ASSERT_NE(entt::any_cast<typename traits_type::entity_type>(&data[0u]), nullptr);
  136. ASSERT_EQ(entt::any_cast<typename traits_type::entity_type>(data[0u]), 0u);
  137. data.clear();
  138. snapshot.get<int>(archive, std::begin(entities), std::end(entities));
  139. ASSERT_EQ(data.size(), 6u);
  140. ASSERT_NE(entt::any_cast<typename traits_type::entity_type>(&data[0u]), nullptr);
  141. ASSERT_EQ(entt::any_cast<typename traits_type::entity_type>(data[0u]), static_cast<typename traits_type::entity_type>(std::distance(std::begin(entities), std::end(entities))));
  142. ASSERT_NE(entt::any_cast<entt::entity>(&data[1u]), nullptr);
  143. ASSERT_EQ(entt::any_cast<entt::entity>(data[1u]), entities[0u]);
  144. ASSERT_NE(entt::any_cast<int>(&data[2u]), nullptr);
  145. ASSERT_EQ(entt::any_cast<int>(data[2u]), values[0u]);
  146. ASSERT_NE(entt::any_cast<entt::entity>(&data[3u]), nullptr);
  147. ASSERT_EQ(entt::any_cast<entt::entity>(data[3u]), static_cast<entt::entity>(entt::null));
  148. ASSERT_NE(entt::any_cast<entt::entity>(&data[4u]), nullptr);
  149. ASSERT_EQ(entt::any_cast<entt::entity>(data[4u]), entities[2u]);
  150. ASSERT_NE(entt::any_cast<int>(&data[5u]), nullptr);
  151. ASSERT_EQ(entt::any_cast<int>(data[5u]), values[2u]);
  152. }
  153. TEST(BasicSnapshotLoader, Constructors) {
  154. static_assert(!std::is_default_constructible_v<entt::basic_snapshot_loader<entt::registry>>);
  155. static_assert(!std::is_copy_constructible_v<entt::basic_snapshot_loader<entt::registry>>);
  156. static_assert(!std::is_copy_assignable_v<entt::basic_snapshot_loader<entt::registry>>);
  157. static_assert(std::is_move_constructible_v<entt::basic_snapshot_loader<entt::registry>>);
  158. static_assert(std::is_move_assignable_v<entt::basic_snapshot_loader<entt::registry>>);
  159. entt::registry registry;
  160. entt::basic_snapshot_loader loader{registry};
  161. entt::basic_snapshot_loader other{std::move(loader)};
  162. ASSERT_NO_FATAL_FAILURE(loader = std::move(other));
  163. }
  164. TEST(BasicSnapshotLoader, GetEntityType) {
  165. using namespace entt::literals;
  166. using traits_type = entt::entt_traits<entt::entity>;
  167. entt::registry registry;
  168. entt::basic_snapshot_loader loader{registry};
  169. const auto &storage = registry.storage<entt::entity>();
  170. std::vector<entt::any> data{};
  171. auto archive = [&data, pos = 0u](auto &value) mutable { value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]); };
  172. const entt::entity entities[3u]{traits_type::construct(0u, 0u), traits_type::construct(2u, 0u), traits_type::construct(1u, 1u)};
  173. ASSERT_FALSE(registry.valid(entities[0u]));
  174. ASSERT_FALSE(registry.valid(entities[1u]));
  175. ASSERT_FALSE(registry.valid(entities[2u]));
  176. data.emplace_back(static_cast<typename traits_type::entity_type>(0u));
  177. data.emplace_back(static_cast<typename traits_type::entity_type>(0u));
  178. loader.get<entt::entity>(archive);
  179. ASSERT_FALSE(registry.valid(entities[0u]));
  180. ASSERT_FALSE(registry.valid(entities[1u]));
  181. ASSERT_FALSE(registry.valid(entities[2u]));
  182. ASSERT_EQ(storage.size(), 0u);
  183. ASSERT_EQ(storage.in_use(), 0u);
  184. data.emplace_back(static_cast<typename traits_type::entity_type>(3u));
  185. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  186. data.emplace_back(entities[0u]);
  187. data.emplace_back(entities[1u]);
  188. data.emplace_back(entities[2u]);
  189. loader.get<entt::entity>(archive, "ignored"_hs);
  190. ASSERT_TRUE(registry.valid(entities[0u]));
  191. ASSERT_TRUE(registry.valid(entities[1u]));
  192. ASSERT_TRUE(registry.valid(entities[2u]));
  193. ASSERT_EQ(storage.size(), 3u);
  194. ASSERT_EQ(storage.in_use(), 2u);
  195. ASSERT_EQ(storage[0u], entities[0u]);
  196. ASSERT_EQ(storage[1u], entities[1u]);
  197. ASSERT_EQ(storage[2u], entities[2u]);
  198. ASSERT_EQ(registry.create(), entities[2u]);
  199. }
  200. TEST(BasicSnapshotLoader, GetType) {
  201. using namespace entt::literals;
  202. using traits_type = entt::entt_traits<entt::entity>;
  203. entt::registry registry;
  204. entt::basic_snapshot_loader loader{registry};
  205. const auto &storage = registry.storage<int>();
  206. std::vector<entt::any> data{};
  207. auto archive = [&data, pos = 0u](auto &value) mutable { value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]); };
  208. const entt::entity entities[2u]{traits_type::construct(0u, 0u), traits_type::construct(2u, 0u)};
  209. const int values[2u]{1, 3};
  210. ASSERT_FALSE(registry.valid(entities[0u]));
  211. ASSERT_FALSE(registry.valid(entities[1u]));
  212. data.emplace_back(static_cast<typename traits_type::entity_type>(1u));
  213. data.emplace_back(entities[0u]);
  214. data.emplace_back(values[0u]);
  215. loader.get<int>(archive, "other"_hs);
  216. ASSERT_TRUE(registry.valid(entities[0u]));
  217. ASSERT_FALSE(registry.valid(entities[1u]));
  218. ASSERT_EQ(storage.size(), 0u);
  219. ASSERT_EQ(registry.storage<int>("other"_hs).size(), 1u);
  220. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  221. data.emplace_back(entities[0u]);
  222. data.emplace_back(values[0u]);
  223. data.emplace_back(entities[1u]);
  224. data.emplace_back(values[1u]);
  225. loader.get<int>(archive);
  226. ASSERT_TRUE(registry.valid(entities[0u]));
  227. ASSERT_TRUE(registry.valid(entities[1u]));
  228. ASSERT_EQ(storage.size(), 2u);
  229. ASSERT_TRUE(storage.contains(entities[0u]));
  230. ASSERT_TRUE(storage.contains(entities[1u]));
  231. ASSERT_EQ(storage.get(entities[0u]), values[0u]);
  232. ASSERT_EQ(storage.get(entities[1u]), values[1u]);
  233. }
  234. TEST(BasicSnapshotLoader, GetEmptyType) {
  235. using namespace entt::literals;
  236. using traits_type = entt::entt_traits<entt::entity>;
  237. entt::registry registry;
  238. entt::basic_snapshot_loader loader{registry};
  239. const auto &storage = registry.storage<empty>();
  240. std::vector<entt::any> data{};
  241. auto archive = [&data, pos = 0u](auto &value) mutable { value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]); };
  242. const entt::entity entities[2u]{traits_type::construct(0u, 0u), traits_type::construct(2u, 0u)};
  243. ASSERT_FALSE(registry.valid(entities[0u]));
  244. ASSERT_FALSE(registry.valid(entities[1u]));
  245. data.emplace_back(static_cast<typename traits_type::entity_type>(1u));
  246. data.emplace_back(entities[0u]);
  247. loader.get<empty>(archive, "other"_hs);
  248. ASSERT_TRUE(registry.valid(entities[0u]));
  249. ASSERT_FALSE(registry.valid(entities[1u]));
  250. ASSERT_EQ(storage.size(), 0u);
  251. ASSERT_EQ(registry.storage<empty>("other"_hs).size(), 1u);
  252. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  253. data.emplace_back(entities[0u]);
  254. data.emplace_back(entities[1u]);
  255. loader.get<empty>(archive);
  256. ASSERT_TRUE(registry.valid(entities[0u]));
  257. ASSERT_TRUE(registry.valid(entities[1u]));
  258. ASSERT_EQ(storage.size(), 2u);
  259. ASSERT_TRUE(storage.contains(entities[0u]));
  260. ASSERT_TRUE(storage.contains(entities[1u]));
  261. }
  262. TEST(BasicSnapshotLoader, GetTypeSparse) {
  263. using namespace entt::literals;
  264. using traits_type = entt::entt_traits<entt::entity>;
  265. entt::registry registry;
  266. entt::basic_snapshot_loader loader{registry};
  267. const auto &storage = registry.storage<int>();
  268. std::vector<entt::any> data{};
  269. auto archive = [&data, pos = 0u](auto &value) mutable { value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]); };
  270. const entt::entity entities[2u]{traits_type::construct(0u, 0u), traits_type::construct(2u, 0u)};
  271. const int values[2u]{1, 3};
  272. ASSERT_FALSE(registry.valid(entities[0u]));
  273. ASSERT_FALSE(registry.valid(entities[1u]));
  274. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  275. data.emplace_back(static_cast<entt::entity>(entt::null));
  276. data.emplace_back(entities[0u]);
  277. data.emplace_back(values[0u]);
  278. loader.get<int>(archive, "other"_hs);
  279. ASSERT_TRUE(registry.valid(entities[0u]));
  280. ASSERT_FALSE(registry.valid(entities[1u]));
  281. ASSERT_EQ(storage.size(), 0u);
  282. ASSERT_EQ(registry.storage<int>("other"_hs).size(), 1u);
  283. data.emplace_back(static_cast<typename traits_type::entity_type>(3u));
  284. data.emplace_back(entities[0u]);
  285. data.emplace_back(values[0u]);
  286. data.emplace_back(static_cast<entt::entity>(entt::null));
  287. data.emplace_back(entities[1u]);
  288. data.emplace_back(values[1u]);
  289. loader.get<int>(archive);
  290. ASSERT_TRUE(registry.valid(entities[0u]));
  291. ASSERT_TRUE(registry.valid(entities[1u]));
  292. ASSERT_EQ(storage.size(), 2u);
  293. ASSERT_TRUE(storage.contains(entities[0u]));
  294. ASSERT_TRUE(storage.contains(entities[1u]));
  295. ASSERT_EQ(storage.get(entities[0u]), values[0u]);
  296. ASSERT_EQ(storage.get(entities[1u]), values[1u]);
  297. }
  298. TEST(BasicSnapshotLoader, GetTypeWithListener) {
  299. using traits_type = entt::entt_traits<entt::entity>;
  300. entt::registry registry;
  301. entt::basic_snapshot_loader loader{registry};
  302. entt::entity check{entt::null};
  303. std::vector<entt::any> data{};
  304. auto archive = [&data, pos = 0u](auto &value) mutable { value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]); };
  305. const auto entity{traits_type::construct(1u, 1u)};
  306. const shadow value{entity};
  307. ASSERT_FALSE(registry.valid(entity));
  308. ASSERT_EQ(check, static_cast<entt::entity>(entt::null));
  309. registry.on_construct<shadow>().connect<&shadow::listener>(check);
  310. data.emplace_back(static_cast<typename traits_type::entity_type>(1u));
  311. data.emplace_back(entity);
  312. data.emplace_back(value);
  313. loader.get<shadow>(archive);
  314. ASSERT_TRUE(registry.valid(entity));
  315. ASSERT_EQ(check, entity);
  316. }
  317. TEST(BasicSnapshotLoader, Orphans) {
  318. using namespace entt::literals;
  319. using traits_type = entt::entt_traits<entt::entity>;
  320. entt::registry registry;
  321. entt::basic_snapshot_loader loader{registry};
  322. std::vector<entt::any> data{};
  323. auto archive = [&data, pos = 0u](auto &value) mutable { value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]); };
  324. const entt::entity entities[2u]{traits_type::construct(0u, 0u), traits_type::construct(2u, 0u)};
  325. const int value = 42;
  326. ASSERT_FALSE(registry.valid(entities[0u]));
  327. ASSERT_FALSE(registry.valid(entities[1u]));
  328. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  329. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  330. data.emplace_back(entities[0u]);
  331. data.emplace_back(entities[1u]);
  332. data.emplace_back(static_cast<typename traits_type::entity_type>(1u));
  333. data.emplace_back(entities[0u]);
  334. data.emplace_back(value);
  335. loader.get<entt::entity>(archive);
  336. loader.get<int>(archive);
  337. ASSERT_TRUE(registry.valid(entities[0u]));
  338. ASSERT_TRUE(registry.valid(entities[1u]));
  339. loader.orphans();
  340. ASSERT_TRUE(registry.valid(entities[0u]));
  341. ASSERT_FALSE(registry.valid(entities[1u]));
  342. }
  343. TEST(BasicContinuousLoader, Constructors) {
  344. static_assert(!std::is_default_constructible_v<entt::basic_continuous_loader<entt::registry>>);
  345. static_assert(!std::is_copy_constructible_v<entt::basic_continuous_loader<entt::registry>>);
  346. static_assert(!std::is_copy_assignable_v<entt::basic_continuous_loader<entt::registry>>);
  347. static_assert(std::is_move_constructible_v<entt::basic_continuous_loader<entt::registry>>);
  348. static_assert(std::is_move_assignable_v<entt::basic_continuous_loader<entt::registry>>);
  349. entt::registry registry;
  350. entt::basic_continuous_loader loader{registry};
  351. entt::basic_continuous_loader other{std::move(loader)};
  352. ASSERT_NO_FATAL_FAILURE(loader = std::move(other));
  353. }
  354. TEST(BasicContinuousLoader, GetEntityType) {
  355. using namespace entt::literals;
  356. using traits_type = entt::entt_traits<entt::entity>;
  357. entt::registry registry;
  358. entt::basic_continuous_loader loader{registry};
  359. const auto &storage = registry.storage<entt::entity>();
  360. std::vector<entt::any> data{};
  361. auto archive = [&data, pos = 0u](auto &value) mutable { value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]); };
  362. const entt::entity entities[3u]{traits_type::construct(1u, 0u), traits_type::construct(0u, 0u), traits_type::construct(2u, 0u)};
  363. ASSERT_FALSE(registry.valid(entities[0u]));
  364. ASSERT_FALSE(registry.valid(entities[1u]));
  365. ASSERT_FALSE(registry.valid(entities[2u]));
  366. data.emplace_back(static_cast<typename traits_type::entity_type>(0u));
  367. data.emplace_back(static_cast<typename traits_type::entity_type>(0u));
  368. loader.get<entt::entity>(archive);
  369. ASSERT_FALSE(registry.valid(entities[0u]));
  370. ASSERT_FALSE(registry.valid(entities[1u]));
  371. ASSERT_FALSE(registry.valid(entities[2u]));
  372. ASSERT_FALSE(loader.contains(entities[0u]));
  373. ASSERT_FALSE(loader.contains(entities[1u]));
  374. ASSERT_FALSE(loader.contains(entities[2u]));
  375. ASSERT_EQ(loader.map(entities[0u]), static_cast<entt::entity>(entt::null));
  376. ASSERT_EQ(loader.map(entities[1u]), static_cast<entt::entity>(entt::null));
  377. ASSERT_EQ(loader.map(entities[2u]), static_cast<entt::entity>(entt::null));
  378. ASSERT_EQ(storage.size(), 0u);
  379. ASSERT_EQ(storage.in_use(), 0u);
  380. data.emplace_back(static_cast<typename traits_type::entity_type>(3u));
  381. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  382. data.emplace_back(entities[0u]);
  383. data.emplace_back(entities[1u]);
  384. data.emplace_back(entities[2u]);
  385. loader.get<entt::entity>(archive, "ignored"_hs);
  386. ASSERT_TRUE(loader.contains(entities[0u]));
  387. ASSERT_TRUE(loader.contains(entities[1u]));
  388. ASSERT_FALSE(loader.contains(entities[2u]));
  389. ASSERT_NE(loader.map(entities[0u]), static_cast<entt::entity>(entt::null));
  390. ASSERT_NE(loader.map(entities[1u]), static_cast<entt::entity>(entt::null));
  391. ASSERT_EQ(loader.map(entities[2u]), static_cast<entt::entity>(entt::null));
  392. ASSERT_TRUE(registry.valid(loader.map(entities[0u])));
  393. ASSERT_TRUE(registry.valid(loader.map(entities[1u])));
  394. ASSERT_EQ(storage.size(), 2u);
  395. ASSERT_EQ(storage.in_use(), 2u);
  396. ASSERT_EQ(storage[0u], loader.map(entities[0u]));
  397. ASSERT_EQ(storage[1u], loader.map(entities[1u]));
  398. ASSERT_EQ(registry.create(), entities[2u]);
  399. data.emplace_back(static_cast<typename traits_type::entity_type>(3u));
  400. data.emplace_back(static_cast<typename traits_type::entity_type>(3u));
  401. data.emplace_back(entities[0u]);
  402. data.emplace_back(entities[1u]);
  403. data.emplace_back(entities[2u]);
  404. loader.get<entt::entity>(archive);
  405. ASSERT_TRUE(loader.contains(entities[0u]));
  406. ASSERT_TRUE(loader.contains(entities[1u]));
  407. ASSERT_TRUE(loader.contains(entities[2u]));
  408. ASSERT_NE(loader.map(entities[0u]), static_cast<entt::entity>(entt::null));
  409. ASSERT_NE(loader.map(entities[1u]), static_cast<entt::entity>(entt::null));
  410. ASSERT_NE(loader.map(entities[2u]), static_cast<entt::entity>(entt::null));
  411. ASSERT_TRUE(registry.valid(loader.map(entities[0u])));
  412. ASSERT_TRUE(registry.valid(loader.map(entities[1u])));
  413. ASSERT_TRUE(registry.valid(loader.map(entities[2u])));
  414. ASSERT_EQ(storage.size(), 4u);
  415. ASSERT_EQ(storage.in_use(), 4u);
  416. ASSERT_EQ(storage[0u], loader.map(entities[0u]));
  417. ASSERT_EQ(storage[1u], loader.map(entities[1u]));
  418. ASSERT_EQ(storage[3u], loader.map(entities[2u]));
  419. registry.destroy(loader.map(entities[1u]));
  420. ASSERT_TRUE(loader.contains(entities[1u]));
  421. ASSERT_NE(loader.map(entities[1u]), static_cast<entt::entity>(entt::null));
  422. ASSERT_FALSE(registry.valid(loader.map(entities[1u])));
  423. data.emplace_back(static_cast<typename traits_type::entity_type>(1u));
  424. data.emplace_back(static_cast<typename traits_type::entity_type>(1u));
  425. data.emplace_back(entities[1u]);
  426. loader.get<entt::entity>(archive);
  427. ASSERT_TRUE(loader.contains(entities[1u]));
  428. ASSERT_NE(loader.map(entities[1u]), static_cast<entt::entity>(entt::null));
  429. ASSERT_TRUE(registry.valid(loader.map(entities[1u])));
  430. ASSERT_EQ(storage[3u], loader.map(entities[1u]));
  431. data.emplace_back(static_cast<typename traits_type::entity_type>(3u));
  432. data.emplace_back(static_cast<typename traits_type::entity_type>(1u));
  433. data.emplace_back(entities[1u]);
  434. data.emplace_back(entities[2u]);
  435. data.emplace_back(entities[0u]);
  436. loader.get<entt::entity>(archive, "ignored"_hs);
  437. ASSERT_FALSE(loader.contains(entities[0u]));
  438. ASSERT_TRUE(loader.contains(entities[1u]));
  439. ASSERT_FALSE(loader.contains(entities[2u]));
  440. ASSERT_EQ(loader.map(entities[0u]), static_cast<entt::entity>(entt::null));
  441. ASSERT_NE(loader.map(entities[1u]), static_cast<entt::entity>(entt::null));
  442. ASSERT_EQ(loader.map(entities[2u]), static_cast<entt::entity>(entt::null));
  443. ASSERT_TRUE(registry.valid(loader.map(entities[1u])));
  444. ASSERT_EQ(storage.size(), 4u);
  445. ASSERT_EQ(storage.in_use(), 2u);
  446. ASSERT_EQ(storage[1u], loader.map(entities[1u]));
  447. }
  448. TEST(BasicContinuousLoader, GetType) {
  449. using namespace entt::literals;
  450. using traits_type = entt::entt_traits<entt::entity>;
  451. entt::registry registry;
  452. entt::basic_continuous_loader loader{registry};
  453. const auto &storage = registry.storage<int>();
  454. std::vector<entt::any> data{};
  455. auto archive = [&data, pos = 0u](auto &value) mutable { value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]); };
  456. const entt::entity entities[2u]{traits_type::construct(0u, 0u), traits_type::construct(2u, 0u)};
  457. const int values[2u]{1, 3};
  458. ASSERT_FALSE(loader.contains(entities[0u]));
  459. ASSERT_FALSE(loader.contains(entities[1u]));
  460. ASSERT_FALSE(registry.valid(loader.map(entities[0u])));
  461. ASSERT_FALSE(registry.valid(loader.map(entities[1u])));
  462. data.emplace_back(static_cast<typename traits_type::entity_type>(1u));
  463. data.emplace_back(entities[0u]);
  464. data.emplace_back(values[0u]);
  465. loader.get<int>(archive, "other"_hs);
  466. ASSERT_TRUE(loader.contains(entities[0u]));
  467. ASSERT_FALSE(loader.contains(entities[1u]));
  468. ASSERT_TRUE(registry.valid(loader.map(entities[0u])));
  469. ASSERT_FALSE(registry.valid(loader.map(entities[1u])));
  470. ASSERT_EQ(storage.size(), 0u);
  471. ASSERT_EQ(registry.storage<int>("other"_hs).size(), 1u);
  472. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  473. data.emplace_back(entities[0u]);
  474. data.emplace_back(values[0u]);
  475. data.emplace_back(entities[1u]);
  476. data.emplace_back(values[1u]);
  477. loader.get<int>(archive);
  478. ASSERT_TRUE(loader.contains(entities[0u]));
  479. ASSERT_TRUE(loader.contains(entities[1u]));
  480. ASSERT_TRUE(registry.valid(loader.map(entities[0u])));
  481. ASSERT_TRUE(registry.valid(loader.map(entities[1u])));
  482. ASSERT_EQ(storage.size(), 2u);
  483. ASSERT_TRUE(storage.contains(loader.map(entities[0u])));
  484. ASSERT_TRUE(storage.contains(loader.map(entities[1u])));
  485. ASSERT_EQ(storage.get(loader.map(entities[0u])), values[0u]);
  486. ASSERT_EQ(storage.get(loader.map(entities[1u])), values[1u]);
  487. }
  488. TEST(BasicContinuousLoader, GetTypeExtended) {
  489. using namespace entt::literals;
  490. using traits_type = entt::entt_traits<entt::entity>;
  491. entt::registry registry;
  492. entt::basic_continuous_loader loader{registry};
  493. const auto &storage = registry.storage<shadow>();
  494. std::vector<entt::any> data{};
  495. const entt::entity entities[2u]{traits_type::construct(0u, 1u), traits_type::construct(1u, 1u)};
  496. const shadow value{entities[0u]};
  497. auto archive = [&loader, &data, pos = 0u](auto &value) mutable {
  498. value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]);
  499. if constexpr(std::is_same_v<std::remove_reference_t<decltype(value)>, shadow>) {
  500. value.target = loader.map(value.target);
  501. }
  502. };
  503. ASSERT_FALSE(loader.contains(entities[0u]));
  504. ASSERT_FALSE(loader.contains(entities[1u]));
  505. ASSERT_FALSE(registry.valid(loader.map(entities[0u])));
  506. ASSERT_FALSE(registry.valid(loader.map(entities[1u])));
  507. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  508. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  509. data.emplace_back(entities[0u]);
  510. data.emplace_back(entities[1u]);
  511. data.emplace_back(static_cast<typename traits_type::entity_type>(1u));
  512. data.emplace_back(entities[1u]);
  513. data.emplace_back(value);
  514. loader.get<entt::entity>(archive);
  515. loader.get<shadow>(archive);
  516. ASSERT_TRUE(loader.contains(entities[0u]));
  517. ASSERT_TRUE(loader.contains(entities[1u]));
  518. ASSERT_TRUE(registry.valid(loader.map(entities[0u])));
  519. ASSERT_TRUE(registry.valid(loader.map(entities[1u])));
  520. ASSERT_FALSE(registry.valid(entities[0u]));
  521. ASSERT_FALSE(registry.valid(entities[1u]));
  522. ASSERT_EQ(storage.size(), 1u);
  523. ASSERT_TRUE(storage.contains(loader.map(entities[1u])));
  524. ASSERT_EQ(storage.get(loader.map(entities[1u])).target, loader.map(entities[0u]));
  525. }
  526. TEST(BasicContinuousLoader, GetEmptyType) {
  527. using namespace entt::literals;
  528. using traits_type = entt::entt_traits<entt::entity>;
  529. entt::registry registry;
  530. entt::basic_continuous_loader loader{registry};
  531. const auto &storage = registry.storage<empty>();
  532. std::vector<entt::any> data{};
  533. auto archive = [&data, pos = 0u](auto &value) mutable { value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]); };
  534. const entt::entity entities[2u]{traits_type::construct(0u, 0u), traits_type::construct(2u, 0u)};
  535. ASSERT_FALSE(loader.contains(entities[0u]));
  536. ASSERT_FALSE(loader.contains(entities[1u]));
  537. ASSERT_FALSE(registry.valid(loader.map(entities[0u])));
  538. ASSERT_FALSE(registry.valid(loader.map(entities[1u])));
  539. data.emplace_back(static_cast<typename traits_type::entity_type>(1u));
  540. data.emplace_back(entities[0u]);
  541. loader.get<empty>(archive, "other"_hs);
  542. ASSERT_TRUE(loader.contains(entities[0u]));
  543. ASSERT_FALSE(loader.contains(entities[1u]));
  544. ASSERT_TRUE(registry.valid(loader.map(entities[0u])));
  545. ASSERT_FALSE(registry.valid(loader.map(entities[1u])));
  546. ASSERT_EQ(storage.size(), 0u);
  547. ASSERT_EQ(registry.storage<empty>("other"_hs).size(), 1u);
  548. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  549. data.emplace_back(entities[0u]);
  550. data.emplace_back(entities[1u]);
  551. loader.get<empty>(archive);
  552. ASSERT_TRUE(loader.contains(entities[0u]));
  553. ASSERT_TRUE(loader.contains(entities[1u]));
  554. ASSERT_TRUE(registry.valid(loader.map(entities[0u])));
  555. ASSERT_TRUE(registry.valid(loader.map(entities[1u])));
  556. ASSERT_EQ(storage.size(), 2u);
  557. ASSERT_TRUE(storage.contains(loader.map(entities[0u])));
  558. ASSERT_TRUE(storage.contains(loader.map(entities[1u])));
  559. }
  560. TEST(BasicContinuousLoader, GetTypeSparse) {
  561. using namespace entt::literals;
  562. using traits_type = entt::entt_traits<entt::entity>;
  563. entt::registry registry;
  564. entt::basic_continuous_loader loader{registry};
  565. const auto &storage = registry.storage<int>();
  566. std::vector<entt::any> data{};
  567. auto archive = [&data, pos = 0u](auto &value) mutable { value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]); };
  568. const entt::entity entities[2u]{traits_type::construct(0u, 0u), traits_type::construct(2u, 0u)};
  569. const int values[2u]{1, 3};
  570. ASSERT_FALSE(loader.contains(entities[0u]));
  571. ASSERT_FALSE(loader.contains(entities[1u]));
  572. ASSERT_FALSE(registry.valid(loader.map(entities[0u])));
  573. ASSERT_FALSE(registry.valid(loader.map(entities[1u])));
  574. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  575. data.emplace_back(static_cast<entt::entity>(entt::null));
  576. data.emplace_back(entities[0u]);
  577. data.emplace_back(values[0u]);
  578. loader.get<int>(archive, "other"_hs);
  579. ASSERT_TRUE(loader.contains(entities[0u]));
  580. ASSERT_FALSE(loader.contains(entities[1u]));
  581. ASSERT_TRUE(registry.valid(loader.map(entities[0u])));
  582. ASSERT_FALSE(registry.valid(loader.map(entities[1u])));
  583. ASSERT_EQ(storage.size(), 0u);
  584. ASSERT_EQ(registry.storage<int>("other"_hs).size(), 1u);
  585. data.emplace_back(static_cast<typename traits_type::entity_type>(3u));
  586. data.emplace_back(entities[0u]);
  587. data.emplace_back(values[0u]);
  588. data.emplace_back(static_cast<entt::entity>(entt::null));
  589. data.emplace_back(entities[1u]);
  590. data.emplace_back(values[1u]);
  591. loader.get<int>(archive);
  592. ASSERT_TRUE(loader.contains(entities[0u]));
  593. ASSERT_TRUE(loader.contains(entities[1u]));
  594. ASSERT_TRUE(registry.valid(loader.map(entities[0u])));
  595. ASSERT_TRUE(registry.valid(loader.map(entities[1u])));
  596. ASSERT_EQ(storage.size(), 2u);
  597. ASSERT_TRUE(storage.contains(loader.map(entities[0u])));
  598. ASSERT_TRUE(storage.contains(loader.map(entities[1u])));
  599. ASSERT_EQ(storage.get(loader.map(entities[0u])), values[0u]);
  600. ASSERT_EQ(storage.get(loader.map(entities[1u])), values[1u]);
  601. }
  602. TEST(BasicContinuousLoader, GetTypeWithListener) {
  603. using traits_type = entt::entt_traits<entt::entity>;
  604. entt::registry registry;
  605. entt::basic_continuous_loader loader{registry};
  606. entt::entity check{entt::null};
  607. std::vector<entt::any> data{};
  608. auto archive = [&data, pos = 0u](auto &value) mutable { value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]); };
  609. const auto entity{traits_type::construct(1u, 1u)};
  610. const shadow value{entity};
  611. ASSERT_FALSE(registry.valid(loader.map(entity)));
  612. ASSERT_EQ(check, static_cast<entt::entity>(entt::null));
  613. registry.on_construct<shadow>().connect<&shadow::listener>(check);
  614. data.emplace_back(static_cast<typename traits_type::entity_type>(1u));
  615. data.emplace_back(entity);
  616. data.emplace_back(value);
  617. loader.get<shadow>(archive);
  618. ASSERT_TRUE(registry.valid(loader.map(entity)));
  619. ASSERT_EQ(check, entity);
  620. }
  621. TEST(BasicContinuousLoader, Shrink) {
  622. entt::registry registry;
  623. entt::basic_continuous_loader loader{registry};
  624. ASSERT_NO_FATAL_FAILURE(loader.shrink());
  625. }
  626. TEST(BasicContinuousLoader, Orphans) {
  627. using namespace entt::literals;
  628. using traits_type = entt::entt_traits<entt::entity>;
  629. entt::registry registry;
  630. entt::basic_continuous_loader loader{registry};
  631. std::vector<entt::any> data{};
  632. auto archive = [&data, pos = 0u](auto &value) mutable { value = entt::any_cast<std::remove_reference_t<decltype(value)>>(data[pos++]); };
  633. const entt::entity entities[2u]{traits_type::construct(0u, 0u), traits_type::construct(2u, 0u)};
  634. const int value = 42;
  635. ASSERT_FALSE(registry.valid(entities[0u]));
  636. ASSERT_FALSE(registry.valid(entities[1u]));
  637. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  638. data.emplace_back(static_cast<typename traits_type::entity_type>(2u));
  639. data.emplace_back(entities[0u]);
  640. data.emplace_back(entities[1u]);
  641. data.emplace_back(static_cast<typename traits_type::entity_type>(1u));
  642. data.emplace_back(entities[0u]);
  643. data.emplace_back(value);
  644. loader.get<entt::entity>(archive);
  645. loader.get<int>(archive);
  646. ASSERT_TRUE(loader.contains(entities[0u]));
  647. ASSERT_TRUE(loader.contains(entities[1u]));
  648. ASSERT_TRUE(registry.valid(loader.map(entities[0u])));
  649. ASSERT_TRUE(registry.valid(loader.map(entities[1u])));
  650. loader.orphans();
  651. ASSERT_TRUE(loader.contains(entities[0u]));
  652. ASSERT_TRUE(loader.contains(entities[1u]));
  653. ASSERT_TRUE(registry.valid(loader.map(entities[0u])));
  654. ASSERT_FALSE(registry.valid(loader.map(entities[1u])));
  655. }
  656. template<typename Storage>
  657. struct output_archive {
  658. output_archive(Storage &instance)
  659. : storage{instance} {}
  660. template<typename Value>
  661. void operator()(const Value &value) {
  662. std::get<std::queue<Value>>(storage).push(value);
  663. }
  664. void operator()(const std::unique_ptr<int> &instance) {
  665. (*this)(*instance);
  666. }
  667. private:
  668. Storage &storage;
  669. };
  670. template<typename Storage>
  671. struct input_archive {
  672. input_archive(Storage &instance)
  673. : storage{instance} {}
  674. template<typename Value>
  675. void operator()(Value &value) {
  676. auto assign = [this](auto &val) {
  677. auto &queue = std::get<std::queue<std::decay_t<decltype(val)>>>(storage);
  678. val = queue.front();
  679. queue.pop();
  680. };
  681. assign(value);
  682. }
  683. void operator()(std::unique_ptr<int> &instance) {
  684. instance = std::make_unique<int>();
  685. (*this)(*instance);
  686. }
  687. private:
  688. Storage &storage;
  689. };
  690. struct a_component {};
  691. struct another_component {
  692. int key{};
  693. int value{};
  694. };
  695. struct what_a_component {
  696. entt::entity bar{};
  697. std::vector<entt::entity> quux{};
  698. };
  699. struct map_component {
  700. std::map<entt::entity, int> keys{};
  701. std::map<int, entt::entity> values{};
  702. std::map<entt::entity, entt::entity> both{};
  703. };
  704. TEST(Snapshot, Dump) {
  705. using traits_type = entt::entt_traits<entt::entity>;
  706. entt::registry registry;
  707. const auto e0 = registry.create();
  708. registry.emplace<int>(e0, 42);
  709. registry.emplace<char>(e0, 'c');
  710. registry.emplace<double>(e0, .1);
  711. const auto e1 = registry.create();
  712. const auto e2 = registry.create();
  713. registry.emplace<int>(e2, 3);
  714. const auto e3 = registry.create();
  715. registry.emplace<a_component>(e3);
  716. registry.emplace<char>(e3, '0');
  717. registry.destroy(e1);
  718. auto v1 = registry.current(e1);
  719. using archive_type = std::tuple<
  720. std::queue<typename traits_type::entity_type>,
  721. std::queue<entt::entity>,
  722. std::queue<int>,
  723. std::queue<char>,
  724. std::queue<double>,
  725. std::queue<a_component>,
  726. std::queue<another_component>>;
  727. archive_type storage;
  728. output_archive<archive_type> output{storage};
  729. input_archive<archive_type> input{storage};
  730. entt::snapshot{registry}
  731. .entities(output)
  732. .component<int>(output)
  733. .component<char>(output)
  734. .component<double>(output)
  735. .component<a_component>(output)
  736. .component<another_component>(output);
  737. registry.clear();
  738. ASSERT_FALSE(registry.valid(e0));
  739. ASSERT_FALSE(registry.valid(e1));
  740. ASSERT_FALSE(registry.valid(e2));
  741. ASSERT_FALSE(registry.valid(e3));
  742. entt::snapshot_loader{registry}
  743. .entities(input)
  744. .component<int>(input)
  745. .component<char>(input)
  746. .component<double>(input)
  747. .component<a_component>(input)
  748. .component<another_component>(input)
  749. .orphans();
  750. ASSERT_TRUE(registry.valid(e0));
  751. ASSERT_FALSE(registry.valid(e1));
  752. ASSERT_TRUE(registry.valid(e2));
  753. ASSERT_TRUE(registry.valid(e3));
  754. ASSERT_FALSE(registry.orphan(e0));
  755. ASSERT_FALSE(registry.orphan(e2));
  756. ASSERT_FALSE(registry.orphan(e3));
  757. ASSERT_EQ(registry.get<int>(e0), 42);
  758. ASSERT_EQ(registry.get<char>(e0), 'c');
  759. ASSERT_EQ(registry.get<double>(e0), .1);
  760. ASSERT_EQ(registry.current(e1), v1);
  761. ASSERT_EQ(registry.get<int>(e2), 3);
  762. ASSERT_EQ(registry.get<char>(e3), '0');
  763. ASSERT_TRUE(registry.all_of<a_component>(e3));
  764. ASSERT_TRUE(registry.storage<another_component>().empty());
  765. }
  766. TEST(Snapshot, Partial) {
  767. using traits_type = entt::entt_traits<entt::entity>;
  768. entt::registry registry;
  769. const auto e0 = registry.create();
  770. registry.emplace<int>(e0, 42);
  771. registry.emplace<char>(e0, 'c');
  772. registry.emplace<double>(e0, .1);
  773. const auto e1 = registry.create();
  774. const auto e2 = registry.create();
  775. registry.emplace<int>(e2, 3);
  776. const auto e3 = registry.create();
  777. registry.emplace<char>(e3, '0');
  778. registry.destroy(e1);
  779. auto v1 = registry.current(e1);
  780. using archive_type = std::tuple<
  781. std::queue<typename traits_type::entity_type>,
  782. std::queue<entt::entity>,
  783. std::queue<int>,
  784. std::queue<char>,
  785. std::queue<double>>;
  786. archive_type storage;
  787. output_archive<archive_type> output{storage};
  788. input_archive<archive_type> input{storage};
  789. entt::snapshot{registry}
  790. .entities(output)
  791. .component<char>(output)
  792. .component<int>(output);
  793. registry.clear();
  794. ASSERT_FALSE(registry.valid(e0));
  795. ASSERT_FALSE(registry.valid(e1));
  796. ASSERT_FALSE(registry.valid(e2));
  797. ASSERT_FALSE(registry.valid(e3));
  798. entt::snapshot_loader{registry}
  799. .entities(input)
  800. .component<char>(input)
  801. .component<int>(input);
  802. ASSERT_TRUE(registry.valid(e0));
  803. ASSERT_FALSE(registry.valid(e1));
  804. ASSERT_TRUE(registry.valid(e2));
  805. ASSERT_TRUE(registry.valid(e3));
  806. ASSERT_EQ(registry.get<int>(e0), 42);
  807. ASSERT_EQ(registry.get<char>(e0), 'c');
  808. ASSERT_FALSE(registry.all_of<double>(e0));
  809. ASSERT_EQ(registry.current(e1), v1);
  810. ASSERT_EQ(registry.get<int>(e2), 3);
  811. ASSERT_EQ(registry.get<char>(e3), '0');
  812. entt::snapshot{registry}
  813. .entities(output);
  814. registry.clear();
  815. ASSERT_FALSE(registry.valid(e0));
  816. ASSERT_FALSE(registry.valid(e1));
  817. ASSERT_FALSE(registry.valid(e2));
  818. ASSERT_FALSE(registry.valid(e3));
  819. entt::snapshot_loader{registry}
  820. .entities(input)
  821. .orphans();
  822. ASSERT_FALSE(registry.valid(e0));
  823. ASSERT_FALSE(registry.valid(e1));
  824. ASSERT_FALSE(registry.valid(e2));
  825. ASSERT_FALSE(registry.valid(e3));
  826. }
  827. TEST(Snapshot, Iterator) {
  828. using traits_type = entt::entt_traits<entt::entity>;
  829. entt::registry registry;
  830. for(auto i = 0; i < 50; ++i) {
  831. const auto entity = registry.create();
  832. registry.emplace<a_component>(entity);
  833. if(i % 2) {
  834. registry.emplace<another_component>(entity, i, i);
  835. registry.emplace<std::unique_ptr<int>>(entity, std::make_unique<int>(i));
  836. }
  837. }
  838. using archive_type = std::tuple<
  839. std::queue<typename traits_type::entity_type>,
  840. std::queue<entt::entity>,
  841. std::queue<another_component>,
  842. std::queue<int>>;
  843. archive_type storage;
  844. output_archive<archive_type> output{storage};
  845. input_archive<archive_type> input{storage};
  846. const auto view = registry.view<a_component>();
  847. const auto size = view.size();
  848. entt::snapshot{registry}
  849. .component<another_component>(output, view.begin(), view.end())
  850. .component<std::unique_ptr<int>>(output, view.begin(), view.end());
  851. registry.clear();
  852. entt::snapshot_loader{registry}
  853. .component<another_component>(input)
  854. .component<std::unique_ptr<int>>(input);
  855. ASSERT_EQ(registry.view<another_component>().size(), size / 2u);
  856. registry.view<another_component>().each([](const auto entity, const auto &) {
  857. ASSERT_NE(entt::to_integral(entity) % 2u, 0u);
  858. });
  859. }
  860. TEST(Snapshot, Continuous) {
  861. using traits_type = entt::entt_traits<entt::entity>;
  862. entt::registry src;
  863. entt::registry dst;
  864. entt::continuous_loader loader{dst};
  865. std::vector<entt::entity> entities;
  866. entt::entity entity;
  867. using archive_type = std::tuple<
  868. std::queue<typename traits_type::entity_type>,
  869. std::queue<entt::entity>,
  870. std::queue<another_component>,
  871. std::queue<what_a_component>,
  872. std::queue<map_component>,
  873. std::queue<int>,
  874. std::queue<double>>;
  875. archive_type storage;
  876. output_archive<archive_type> output{storage};
  877. input_archive<archive_type> input{storage};
  878. for(int i = 0; i < 10; ++i) {
  879. static_cast<void>(src.create());
  880. }
  881. src.clear();
  882. for(int i = 0; i < 5; ++i) {
  883. entity = src.create();
  884. entities.push_back(entity);
  885. src.emplace<a_component>(entity);
  886. src.emplace<another_component>(entity, i, i);
  887. src.emplace<std::unique_ptr<int>>(entity, std::make_unique<int>(i));
  888. if(i % 2) {
  889. src.emplace<what_a_component>(entity, entity);
  890. } else {
  891. src.emplace<map_component>(entity);
  892. }
  893. }
  894. src.view<what_a_component>().each([&entities](auto, auto &what_a_component) {
  895. what_a_component.quux.insert(what_a_component.quux.begin(), entities.begin(), entities.end());
  896. });
  897. src.view<map_component>().each([&entities](auto, auto &map_component) {
  898. for(std::size_t i = 0; i < entities.size(); ++i) {
  899. map_component.keys.insert({entities[i], int(i)});
  900. map_component.values.insert({int(i), entities[i]});
  901. map_component.both.insert({entities[entities.size() - i - 1], entities[i]});
  902. }
  903. });
  904. entity = dst.create();
  905. dst.emplace<a_component>(entity);
  906. dst.emplace<another_component>(entity, -1, -1);
  907. dst.emplace<std::unique_ptr<int>>(entity, std::make_unique<int>(-1));
  908. entt::snapshot{src}
  909. .entities(output)
  910. .component<a_component>(output)
  911. .component<another_component>(output)
  912. .component<what_a_component>(output)
  913. .component<map_component>(output)
  914. .component<std::unique_ptr<int>>(output);
  915. loader
  916. .entities(input)
  917. .component<a_component>(input)
  918. .component<another_component>(input)
  919. .component<what_a_component>(input, &what_a_component::bar, &what_a_component::quux)
  920. .component<map_component>(input, &map_component::keys, &map_component::values, &map_component::both)
  921. .component<std::unique_ptr<int>>(input)
  922. .orphans();
  923. decltype(dst.size()) a_component_cnt{};
  924. decltype(dst.size()) another_component_cnt{};
  925. decltype(dst.size()) what_a_component_cnt{};
  926. decltype(dst.size()) map_component_cnt{};
  927. decltype(dst.size()) unique_ptr_cnt{};
  928. dst.each([&dst, &a_component_cnt](auto entt) {
  929. ASSERT_TRUE(dst.all_of<a_component>(entt));
  930. ++a_component_cnt;
  931. });
  932. dst.view<another_component>().each([&another_component_cnt](auto, const auto &component) {
  933. ASSERT_EQ(component.value, component.key < 0 ? -1 : component.key);
  934. ++another_component_cnt;
  935. });
  936. dst.view<what_a_component>().each([&dst, &what_a_component_cnt](auto entt, const auto &component) {
  937. ASSERT_EQ(entt, component.bar);
  938. for(auto child: component.quux) {
  939. ASSERT_TRUE(dst.valid(child));
  940. }
  941. ++what_a_component_cnt;
  942. });
  943. dst.view<map_component>().each([&dst, &map_component_cnt](const auto &component) {
  944. for(auto child: component.keys) {
  945. ASSERT_TRUE(dst.valid(child.first));
  946. }
  947. for(auto child: component.values) {
  948. ASSERT_TRUE(dst.valid(child.second));
  949. }
  950. for(auto child: component.both) {
  951. ASSERT_TRUE(dst.valid(child.first));
  952. ASSERT_TRUE(dst.valid(child.second));
  953. }
  954. ++map_component_cnt;
  955. });
  956. dst.view<std::unique_ptr<int>>().each([&dst, &unique_ptr_cnt](auto, const auto &component) {
  957. ++unique_ptr_cnt;
  958. ASSERT_EQ(*component, static_cast<int>(dst.storage<std::unique_ptr<int>>().size() - unique_ptr_cnt - 1u));
  959. });
  960. src.view<another_component>().each([](auto, auto &component) {
  961. component.value = 2 * component.key;
  962. });
  963. auto size = dst.size();
  964. entt::snapshot{src}
  965. .entities(output)
  966. .component<a_component>(output)
  967. .component<what_a_component>(output)
  968. .component<map_component>(output)
  969. .component<another_component>(output);
  970. loader
  971. .entities(input)
  972. .component<a_component>(input)
  973. .component<what_a_component>(input, &what_a_component::bar, &what_a_component::quux)
  974. .component<map_component>(input, &map_component::keys, &map_component::values, &map_component::both)
  975. .component<another_component>(input)
  976. .orphans();
  977. ASSERT_EQ(size, dst.size());
  978. ASSERT_EQ(dst.storage<a_component>().size(), a_component_cnt);
  979. ASSERT_EQ(dst.storage<another_component>().size(), another_component_cnt);
  980. ASSERT_EQ(dst.storage<what_a_component>().size(), what_a_component_cnt);
  981. ASSERT_EQ(dst.storage<map_component>().size(), map_component_cnt);
  982. ASSERT_EQ(dst.storage<std::unique_ptr<int>>().size(), unique_ptr_cnt);
  983. dst.view<another_component>().each([](auto, auto &component) {
  984. ASSERT_EQ(component.value, component.key < 0 ? -1 : (2 * component.key));
  985. });
  986. entity = src.create();
  987. src.view<what_a_component>().each([entity](auto, auto &component) {
  988. component.bar = entity;
  989. });
  990. entt::snapshot{src}
  991. .entities(output)
  992. .component<a_component>(output)
  993. .component<what_a_component>(output)
  994. .component<map_component>(output)
  995. .component<another_component>(output);
  996. loader
  997. .entities(input)
  998. .component<a_component>(input)
  999. .component<what_a_component>(input, &what_a_component::bar, &what_a_component::quux)
  1000. .component<map_component>(input, &map_component::keys, &map_component::values, &map_component::both)
  1001. .component<another_component>(input)
  1002. .orphans();
  1003. dst.view<what_a_component>().each([&loader, entity](auto, auto &component) {
  1004. ASSERT_EQ(component.bar, loader.map(entity));
  1005. });
  1006. entities.clear();
  1007. for(auto entt: src.view<a_component>()) {
  1008. entities.push_back(entt);
  1009. }
  1010. src.destroy(entity);
  1011. loader.shrink();
  1012. entt::snapshot{src}
  1013. .entities(output)
  1014. .component<a_component>(output)
  1015. .component<another_component>(output)
  1016. .component<what_a_component>(output)
  1017. .component<map_component>(output);
  1018. loader
  1019. .entities(input)
  1020. .component<a_component>(input)
  1021. .component<another_component>(input)
  1022. .component<what_a_component>(input, &what_a_component::bar, &what_a_component::quux)
  1023. .component<map_component>(input, &map_component::keys, &map_component::values, &map_component::both)
  1024. .orphans()
  1025. .shrink();
  1026. dst.view<what_a_component>().each([&dst](auto, auto &component) {
  1027. ASSERT_FALSE(dst.valid(component.bar));
  1028. });
  1029. ASSERT_FALSE(loader.contains(entity));
  1030. entity = src.create();
  1031. src.view<what_a_component>().each([entity](auto, auto &component) {
  1032. component.bar = entity;
  1033. });
  1034. dst.clear<a_component>();
  1035. a_component_cnt = src.storage<a_component>().size();
  1036. entt::snapshot{src}
  1037. .entities(output)
  1038. .component<a_component>(output)
  1039. .component<what_a_component>(output)
  1040. .component<map_component>(output)
  1041. .component<another_component>(output);
  1042. loader
  1043. .entities(input)
  1044. .component<a_component>(input)
  1045. .component<what_a_component>(input, &what_a_component::bar, &what_a_component::quux)
  1046. .component<map_component>(input, &map_component::keys, &map_component::values, &map_component::both)
  1047. .component<another_component>(input)
  1048. .orphans();
  1049. ASSERT_EQ(dst.storage<a_component>().size(), a_component_cnt);
  1050. src.clear<a_component>();
  1051. a_component_cnt = {};
  1052. entt::snapshot{src}
  1053. .entities(output)
  1054. .component<what_a_component>(output)
  1055. .component<map_component>(output)
  1056. .component<a_component>(output)
  1057. .component<another_component>(output);
  1058. loader
  1059. .entities(input)
  1060. .component<what_a_component>(input, &what_a_component::bar, &what_a_component::quux)
  1061. .component<map_component>(input, &map_component::keys, &map_component::values, &map_component::both)
  1062. .component<a_component>(input)
  1063. .component<another_component>(input)
  1064. .orphans();
  1065. ASSERT_EQ(dst.storage<a_component>().size(), a_component_cnt);
  1066. }
  1067. TEST(Snapshot, SyncDataMembers) {
  1068. using traits_type = entt::entt_traits<entt::entity>;
  1069. entt::registry src;
  1070. entt::registry dst;
  1071. entt::continuous_loader loader{dst};
  1072. using archive_type = std::tuple<
  1073. std::queue<typename traits_type::entity_type>,
  1074. std::queue<entt::entity>,
  1075. std::queue<what_a_component>,
  1076. std::queue<map_component>>;
  1077. archive_type storage;
  1078. output_archive<archive_type> output{storage};
  1079. input_archive<archive_type> input{storage};
  1080. static_cast<void>(src.create());
  1081. static_cast<void>(src.create());
  1082. src.clear();
  1083. auto parent = src.create();
  1084. auto child = src.create();
  1085. src.emplace<what_a_component>(parent, entt::null);
  1086. src.emplace<what_a_component>(child, parent).quux.push_back(child);
  1087. src.emplace<map_component>(
  1088. child,
  1089. decltype(map_component::keys){{{child, 10}}},
  1090. decltype(map_component::values){{{10, child}}},
  1091. decltype(map_component::both){{{child, child}}});
  1092. entt::snapshot{src}
  1093. .entities(output)
  1094. .component<what_a_component>(output)
  1095. .component<map_component>(output);
  1096. loader
  1097. .entities(input)
  1098. .component<what_a_component>(input, &what_a_component::bar, &what_a_component::quux)
  1099. .component<map_component>(input, &map_component::keys, &map_component::values, &map_component::both);
  1100. ASSERT_FALSE(dst.valid(parent));
  1101. ASSERT_FALSE(dst.valid(child));
  1102. ASSERT_TRUE(dst.all_of<what_a_component>(loader.map(parent)));
  1103. ASSERT_TRUE(dst.all_of<what_a_component>(loader.map(child)));
  1104. ASSERT_EQ(dst.get<what_a_component>(loader.map(parent)).bar, static_cast<entt::entity>(entt::null));
  1105. const auto &component = dst.get<what_a_component>(loader.map(child));
  1106. ASSERT_EQ(component.bar, loader.map(parent));
  1107. ASSERT_EQ(component.quux[0], loader.map(child));
  1108. const auto &elem = dst.get<map_component>(loader.map(child));
  1109. ASSERT_EQ(elem.keys.at(loader.map(child)), 10);
  1110. ASSERT_EQ(elem.values.at(10), loader.map(child));
  1111. ASSERT_EQ(elem.both.at(loader.map(child)), loader.map(child));
  1112. }