usertype_container.hpp 55 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592
  1. // sol2
  2. // The MIT License (MIT)
  3. // Copyright (c) 2013-2022 Rapptz, ThePhD and contributors
  4. // Permission is hereby granted, free of charge, to any person obtaining a copy of
  5. // this software and associated documentation files (the "Software"), to deal in
  6. // the Software without restriction, including without limitation the rights to
  7. // use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
  8. // the Software, and to permit persons to whom the Software is furnished to do so,
  9. // subject to the following conditions:
  10. // The above copyright notice and this permission notice shall be included in all
  11. // copies or substantial portions of the Software.
  12. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  13. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
  14. // FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
  15. // COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
  16. // IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  17. // CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  18. #ifndef SOL_USERTYPE_CONTAINER_HPP
  19. #define SOL_USERTYPE_CONTAINER_HPP
  20. #include <sol/traits.hpp>
  21. #include <sol/stack.hpp>
  22. #include <sol/object.hpp>
  23. namespace sol {
  24. template <typename T>
  25. struct usertype_container;
  26. namespace container_detail {
  27. template <typename T>
  28. struct has_clear_test {
  29. private:
  30. template <typename C>
  31. static meta::sfinae_yes_t test(decltype(&C::clear));
  32. template <typename C>
  33. static meta::sfinae_no_t test(...);
  34. public:
  35. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  36. };
  37. template <typename T>
  38. struct has_empty_test {
  39. private:
  40. template <typename C>
  41. static meta::sfinae_yes_t test(decltype(&C::empty));
  42. template <typename C>
  43. static meta::sfinae_no_t test(...);
  44. public:
  45. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  46. };
  47. template <typename T>
  48. struct has_erase_after_test {
  49. private:
  50. template <typename C>
  51. static meta::sfinae_yes_t test(
  52. decltype(std::declval<C>().erase_after(std::declval<std::add_rvalue_reference_t<typename C::const_iterator>>()))*);
  53. template <typename C>
  54. static meta::sfinae_no_t test(...);
  55. public:
  56. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  57. };
  58. template <typename T, typename = void>
  59. struct has_find_test {
  60. private:
  61. template <typename C>
  62. static meta::sfinae_yes_t test(decltype(std::declval<C>().find(std::declval<std::add_rvalue_reference_t<typename C::value_type>>()))*);
  63. template <typename C>
  64. static meta::sfinae_no_t test(...);
  65. public:
  66. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  67. };
  68. template <typename T>
  69. struct has_find_test<T, std::enable_if_t<meta::is_lookup<T>::value>> {
  70. private:
  71. template <typename C>
  72. static meta::sfinae_yes_t test(decltype(std::declval<C>().find(std::declval<std::add_rvalue_reference_t<typename C::key_type>>()))*);
  73. template <typename C>
  74. static meta::sfinae_no_t test(...);
  75. public:
  76. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  77. };
  78. template <typename T>
  79. struct has_erase_test {
  80. private:
  81. template <typename C>
  82. static meta::sfinae_yes_t test(decltype(std::declval<C>().erase(std::declval<typename C::iterator>()))*);
  83. template <typename C>
  84. static meta::sfinae_no_t test(...);
  85. public:
  86. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  87. };
  88. template <typename T>
  89. struct has_erase_key_test {
  90. private:
  91. template <typename C>
  92. static meta::sfinae_yes_t test(decltype(std::declval<C>().erase(std::declval<typename C::key_type>()))*);
  93. template <typename C>
  94. static meta::sfinae_no_t test(...);
  95. public:
  96. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  97. };
  98. template <typename T>
  99. struct has_traits_find_test {
  100. private:
  101. template <typename C>
  102. static meta::sfinae_yes_t test(decltype(&C::find));
  103. template <typename C>
  104. static meta::sfinae_no_t test(...);
  105. public:
  106. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  107. };
  108. template <typename T>
  109. struct has_traits_index_of_test {
  110. private:
  111. template <typename C>
  112. static meta::sfinae_yes_t test(decltype(&C::index_of));
  113. template <typename C>
  114. static meta::sfinae_no_t test(...);
  115. public:
  116. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  117. };
  118. template <typename T>
  119. struct has_traits_insert_test {
  120. private:
  121. template <typename C>
  122. static meta::sfinae_yes_t test(decltype(&C::insert));
  123. template <typename C>
  124. static meta::sfinae_no_t test(...);
  125. public:
  126. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  127. };
  128. template <typename T>
  129. struct has_traits_erase_test {
  130. private:
  131. template <typename C>
  132. static meta::sfinae_yes_t test(decltype(&C::erase));
  133. template <typename C>
  134. static meta::sfinae_no_t test(...);
  135. public:
  136. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  137. };
  138. template <typename T>
  139. struct has_traits_index_set_test {
  140. private:
  141. template <typename C>
  142. static meta::sfinae_yes_t test(decltype(&C::index_set));
  143. template <typename C>
  144. static meta::sfinae_no_t test(...);
  145. public:
  146. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  147. };
  148. template <typename T>
  149. struct has_traits_index_get_test {
  150. private:
  151. template <typename C>
  152. static meta::sfinae_yes_t test(decltype(&C::index_get));
  153. template <typename C>
  154. static meta::sfinae_no_t test(...);
  155. public:
  156. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  157. };
  158. template <typename T>
  159. struct has_traits_set_test {
  160. private:
  161. template <typename C>
  162. static meta::sfinae_yes_t test(decltype(&C::set));
  163. template <typename C>
  164. static meta::sfinae_no_t test(...);
  165. public:
  166. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  167. };
  168. template <typename T>
  169. struct has_traits_get_test {
  170. private:
  171. template <typename C>
  172. static meta::sfinae_yes_t test(decltype(&C::get));
  173. template <typename C>
  174. static meta::sfinae_no_t test(...);
  175. public:
  176. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  177. };
  178. template <typename T>
  179. struct has_traits_at_test {
  180. private:
  181. template <typename C>
  182. static meta::sfinae_yes_t test(decltype(&C::at));
  183. template <typename C>
  184. static meta::sfinae_no_t test(...);
  185. public:
  186. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  187. };
  188. template <typename T>
  189. struct has_traits_pairs_test {
  190. private:
  191. template <typename C>
  192. static meta::sfinae_yes_t test(decltype(&C::pairs));
  193. template <typename C>
  194. static meta::sfinae_no_t test(...);
  195. public:
  196. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  197. };
  198. template <typename T>
  199. struct has_traits_ipairs_test {
  200. private:
  201. template <typename C>
  202. static meta::sfinae_yes_t test(decltype(&C::ipairs));
  203. template <typename C>
  204. static meta::sfinae_no_t test(...);
  205. public:
  206. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  207. };
  208. template <typename T>
  209. struct has_traits_next_test {
  210. private:
  211. template <typename C>
  212. static meta::sfinae_yes_t test(decltype(&C::next));
  213. template <typename C>
  214. static meta::sfinae_no_t test(...);
  215. public:
  216. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  217. };
  218. template <typename T>
  219. struct has_traits_add_test {
  220. private:
  221. template <typename C>
  222. static meta::sfinae_yes_t test(decltype(&C::add));
  223. template <typename C>
  224. static meta::sfinae_no_t test(...);
  225. public:
  226. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  227. };
  228. template <typename T>
  229. struct has_traits_size_test {
  230. private:
  231. template <typename C>
  232. static meta::sfinae_yes_t test(decltype(&C::size));
  233. template <typename C>
  234. static meta::sfinae_no_t test(...);
  235. public:
  236. static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
  237. };
  238. template <typename T>
  239. using has_clear = meta::boolean<has_clear_test<T>::value>;
  240. template <typename T>
  241. using has_empty = meta::boolean<has_empty_test<T>::value>;
  242. template <typename T>
  243. using has_find = meta::boolean<has_find_test<T>::value>;
  244. template <typename T>
  245. using has_erase = meta::boolean<has_erase_test<T>::value>;
  246. template <typename T>
  247. using has_erase_key = meta::boolean<has_erase_key_test<T>::value>;
  248. template <typename T>
  249. using has_erase_after = meta::boolean<has_erase_after_test<T>::value>;
  250. template <typename T>
  251. using has_traits_get = meta::boolean<has_traits_get_test<T>::value>;
  252. template <typename T>
  253. using has_traits_at = meta::boolean<has_traits_at_test<T>::value>;
  254. template <typename T>
  255. using has_traits_set = meta::boolean<has_traits_set_test<T>::value>;
  256. template <typename T>
  257. using has_traits_index_get = meta::boolean<has_traits_index_get_test<T>::value>;
  258. template <typename T>
  259. using has_traits_index_set = meta::boolean<has_traits_index_set_test<T>::value>;
  260. template <typename T>
  261. using has_traits_pairs = meta::boolean<has_traits_pairs_test<T>::value>;
  262. template <typename T>
  263. using has_traits_ipairs = meta::boolean<has_traits_ipairs_test<T>::value>;
  264. template <typename T>
  265. using has_traits_next = meta::boolean<has_traits_next_test<T>::value>;
  266. template <typename T>
  267. using has_traits_add = meta::boolean<has_traits_add_test<T>::value>;
  268. template <typename T>
  269. using has_traits_size = meta::boolean<has_traits_size_test<T>::value>;
  270. template <typename T>
  271. using has_traits_clear = has_clear<T>;
  272. template <typename T>
  273. using has_traits_empty = has_empty<T>;
  274. template <typename T>
  275. using has_traits_find = meta::boolean<has_traits_find_test<T>::value>;
  276. template <typename T>
  277. using has_traits_index_of = meta::boolean<has_traits_index_of_test<T>::value>;
  278. template <typename T>
  279. using has_traits_insert = meta::boolean<has_traits_insert_test<T>::value>;
  280. template <typename T>
  281. using has_traits_erase = meta::boolean<has_traits_erase_test<T>::value>;
  282. template <typename T>
  283. struct is_forced_container : is_container<T> { };
  284. template <typename T>
  285. struct is_forced_container<as_container_t<T>> : std::true_type { };
  286. template <typename T>
  287. struct container_decay {
  288. typedef T type;
  289. };
  290. template <typename T>
  291. struct container_decay<as_container_t<T>> {
  292. typedef T type;
  293. };
  294. template <typename T>
  295. using container_decay_t = typename container_decay<meta::unqualified_t<T>>::type;
  296. template <typename T>
  297. decltype(auto) get_key(std::false_type, T&& t) {
  298. return std::forward<T>(t);
  299. }
  300. template <typename T>
  301. decltype(auto) get_key(std::true_type, T&& t) {
  302. return t.first;
  303. }
  304. template <typename T>
  305. decltype(auto) get_value(std::false_type, T&& t) {
  306. return std::forward<T>(t);
  307. }
  308. template <typename T>
  309. decltype(auto) get_value(std::true_type, T&& t) {
  310. return t.second;
  311. }
  312. template <typename X, typename = void>
  313. struct usertype_container_default {
  314. private:
  315. typedef std::remove_pointer_t<meta::unwrap_unqualified_t<X>> T;
  316. public:
  317. typedef lua_nil_t iterator;
  318. typedef iterator sentinel;
  319. typedef lua_nil_t value_type;
  320. static int at(lua_State* L_) {
  321. return luaL_error(L_, "sol: cannot call 'at(index)' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  322. }
  323. static int get(lua_State* L_) {
  324. return luaL_error(L_, "sol: cannot call 'get(key)' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  325. }
  326. static int index_get(lua_State* L_) {
  327. return luaL_error(L_, "sol: cannot call 'container[key]' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  328. }
  329. static int set(lua_State* L_) {
  330. return luaL_error(L_, "sol: cannot call 'set(key, value)' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  331. }
  332. static int index_set(lua_State* L_) {
  333. return luaL_error(
  334. L_, "sol: cannot call 'container[key] = value' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  335. }
  336. static int add(lua_State* L_) {
  337. return luaL_error(L_, "sol: cannot call 'add' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  338. }
  339. static int insert(lua_State* L_) {
  340. return luaL_error(L_, "sol: cannot call 'insert' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  341. }
  342. static int find(lua_State* L_) {
  343. return luaL_error(L_, "sol: cannot call 'find' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  344. }
  345. static int index_of(lua_State* L_) {
  346. return luaL_error(L_, "sol: cannot call 'index_of' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  347. }
  348. static int size(lua_State* L_) {
  349. return luaL_error(L_, "sol: cannot call 'end' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  350. }
  351. static int clear(lua_State* L_) {
  352. return luaL_error(L_, "sol: cannot call 'clear' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  353. }
  354. static int empty(lua_State* L_) {
  355. return luaL_error(L_, "sol: cannot call 'empty' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  356. }
  357. static int erase(lua_State* L_) {
  358. return luaL_error(L_, "sol: cannot call 'erase' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  359. }
  360. static int next(lua_State* L_) {
  361. return luaL_error(L_, "sol: cannot call 'next' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  362. }
  363. static int pairs(lua_State* L_) {
  364. return luaL_error(L_, "sol: cannot call '__pairs/pairs' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  365. }
  366. static int ipairs(lua_State* L_) {
  367. return luaL_error(L_, "sol: cannot call '__ipairs' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  368. }
  369. static iterator begin(lua_State* L_, T&) {
  370. luaL_error(L_, "sol: cannot call 'being' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  371. return lua_nil;
  372. }
  373. static sentinel end(lua_State* L_, T&) {
  374. luaL_error(L_, "sol: cannot call 'end' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
  375. return lua_nil;
  376. }
  377. };
  378. template <typename X>
  379. struct usertype_container_default<X,
  380. std::enable_if_t<meta::all<is_forced_container<meta::unqualified_t<X>>, meta::has_value_type<meta::unqualified_t<container_decay_t<X>>>,
  381. meta::has_iterator<meta::unqualified_t<container_decay_t<X>>>>::value>> {
  382. private:
  383. using T = std::remove_pointer_t<meta::unwrap_unqualified_t<container_decay_t<X>>>;
  384. private:
  385. using deferred_uc = usertype_container<X>;
  386. using is_associative = meta::is_associative<T>;
  387. using is_lookup = meta::is_lookup<T>;
  388. using is_ordered = meta::is_ordered<T>;
  389. using is_matched_lookup = meta::is_matched_lookup<T>;
  390. using iterator = typename T::iterator;
  391. using sentinel = meta::sentinel_or_t<T, iterator>;
  392. using value_type = typename T::value_type;
  393. typedef meta::conditional_t<is_matched_lookup::value, std::pair<value_type, value_type>,
  394. meta::conditional_t<is_associative::value || is_lookup::value, value_type, std::pair<std::ptrdiff_t, value_type>>>
  395. KV;
  396. typedef typename KV::first_type K;
  397. typedef typename KV::second_type V;
  398. typedef meta::conditional_t<is_matched_lookup::value, std::ptrdiff_t, K> next_K;
  399. typedef decltype(*std::declval<iterator&>()) iterator_return;
  400. typedef meta::conditional_t<is_associative::value || is_matched_lookup::value, std::add_lvalue_reference_t<V>,
  401. meta::conditional_t<is_lookup::value, V, iterator_return>>
  402. captured_type;
  403. typedef typename meta::iterator_tag<iterator>::type iterator_category;
  404. typedef std::is_same<iterator_category, std::input_iterator_tag> is_input_iterator;
  405. typedef meta::conditional_t<is_input_iterator::value, V, decltype(detail::deref_move_only(std::declval<captured_type>()))> push_type;
  406. typedef std::is_copy_assignable<V> is_copyable;
  407. typedef meta::neg<meta::any<std::is_const<V>, std::is_const<std::remove_reference_t<iterator_return>>, meta::neg<is_copyable>>> is_writable;
  408. typedef meta::unqualified_t<decltype(get_key(is_associative(), std::declval<std::add_lvalue_reference_t<value_type>>()))> key_type;
  409. typedef meta::all<std::is_integral<K>, meta::neg<meta::any<is_associative, is_lookup>>> is_linear_integral;
  410. struct iter : detail::ebco<iterator, 0>, detail::ebco<sentinel, 1> {
  411. using it_base = detail::ebco<iterator, 0>;
  412. using sen_base = detail::ebco<sentinel, 1>;
  413. main_reference keep_alive;
  414. std::size_t index;
  415. iter(lua_State* L_, int stack_index_, iterator it_, sentinel sen_) noexcept
  416. : it_base(std::move(it_)), sen_base(std::move(sen_)), keep_alive(L_, stack_index_), index(0) {
  417. }
  418. iterator& it() noexcept {
  419. return it_base::value();
  420. }
  421. const iterator& it() const noexcept {
  422. return it_base::value();
  423. }
  424. sentinel& sen() noexcept {
  425. return sen_base::value();
  426. }
  427. const sentinel& sen() const noexcept {
  428. return sen_base::value();
  429. }
  430. };
  431. static auto& get_src(lua_State* L_) {
  432. #if SOL_IS_ON(SOL_SAFE_USERTYPE)
  433. auto p = stack::unqualified_check_get<T*>(L_, 1);
  434. if (!p) {
  435. luaL_error(L_,
  436. "sol: 'self' is not of type '%s' (pass 'self' as first argument with ':' or call on proper type)",
  437. detail::demangle<T>().c_str());
  438. }
  439. if (p.value() == nullptr) {
  440. luaL_error(
  441. L_, "sol: 'self' argument is nil (pass 'self' as first argument with ':' or call on a '%s' type)", detail::demangle<T>().c_str());
  442. }
  443. return *p.value();
  444. #else
  445. return stack::unqualified_get<T>(L_, 1);
  446. #endif // Safe getting with error
  447. }
  448. static detail::error_result at_category(std::input_iterator_tag, lua_State* L_, T& self, std::ptrdiff_t pos) {
  449. pos += deferred_uc::index_adjustment(L_, self);
  450. if (pos < 0) {
  451. return stack::push(L_, lua_nil);
  452. }
  453. auto it = deferred_uc::begin(L_, self);
  454. auto e = deferred_uc::end(L_, self);
  455. if (it == e) {
  456. return stack::push(L_, lua_nil);
  457. }
  458. while (pos > 0) {
  459. --pos;
  460. ++it;
  461. if (it == e) {
  462. return stack::push(L_, lua_nil);
  463. }
  464. }
  465. return get_associative(is_associative(), L_, it);
  466. }
  467. static detail::error_result at_category(std::random_access_iterator_tag, lua_State* L_, T& self, std::ptrdiff_t pos) {
  468. std::ptrdiff_t len = static_cast<std::ptrdiff_t>(size_start(L_, self));
  469. pos += deferred_uc::index_adjustment(L_, self);
  470. if (pos < 0 || pos >= len) {
  471. return stack::push(L_, lua_nil);
  472. }
  473. auto it = std::next(deferred_uc::begin(L_, self), pos);
  474. return get_associative(is_associative(), L_, it);
  475. }
  476. static detail::error_result at_start(lua_State* L_, T& self, std::ptrdiff_t pos) {
  477. return at_category(iterator_category(), L_, self, pos);
  478. }
  479. template <typename Iter>
  480. static detail::error_result get_associative(std::true_type, lua_State* L_, Iter& it) {
  481. decltype(auto) v = *it;
  482. return stack::stack_detail::push_reference<push_type>(L_, detail::deref_move_only(v.second));
  483. }
  484. template <typename Iter>
  485. static detail::error_result get_associative(std::false_type, lua_State* L_, Iter& it) {
  486. return stack::stack_detail::push_reference<push_type>(L_, detail::deref_move_only(*it));
  487. }
  488. static detail::error_result get_category(std::input_iterator_tag, lua_State* L_, T& self, K& key) {
  489. key = static_cast<K>(key + deferred_uc::index_adjustment(L_, self));
  490. if (key < 0) {
  491. return stack::push(L_, lua_nil);
  492. }
  493. auto it = deferred_uc::begin(L_, self);
  494. auto e = deferred_uc::end(L_, self);
  495. if (it == e) {
  496. return stack::push(L_, lua_nil);
  497. }
  498. while (key > 0) {
  499. --key;
  500. ++it;
  501. if (it == e) {
  502. return stack::push(L_, lua_nil);
  503. }
  504. }
  505. return get_associative(is_associative(), L_, it);
  506. }
  507. static detail::error_result get_category(std::random_access_iterator_tag, lua_State* L_, T& self, K& key) {
  508. std::ptrdiff_t len = static_cast<std::ptrdiff_t>(size_start(L_, self));
  509. key = static_cast<K>(static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
  510. if (key < 0 || key >= len) {
  511. return stack::push(L_, lua_nil);
  512. }
  513. auto it = std::next(deferred_uc::begin(L_, self), key);
  514. return get_associative(is_associative(), L_, it);
  515. }
  516. static detail::error_result get_it(std::true_type, lua_State* L_, T& self, K& key) {
  517. return get_category(iterator_category(), L_, self, key);
  518. }
  519. static detail::error_result get_comparative(std::true_type, lua_State* L_, T& self, K& key) {
  520. auto fx = [&](const value_type& r) -> bool { return key == get_key(is_associative(), r); };
  521. auto e = deferred_uc::end(L_, self);
  522. auto it = std::find_if(deferred_uc::begin(L_, self), e, std::ref(fx));
  523. if (it == e) {
  524. return stack::push(L_, lua_nil);
  525. }
  526. return get_associative(is_associative(), L_, it);
  527. }
  528. static detail::error_result get_comparative(std::false_type, lua_State*, T&, K&) {
  529. return detail::error_result("cannot get this key on '%s': no suitable way to increment iterator and compare to key value '%s'",
  530. detail::demangle<T>().data(),
  531. detail::demangle<K>().data());
  532. }
  533. static detail::error_result get_it(std::false_type, lua_State* L_, T& self, K& key) {
  534. return get_comparative(meta::supports_op_equal<K, key_type>(), L_, self, key);
  535. }
  536. static detail::error_result set_associative(std::true_type, iterator& it, stack_object value) {
  537. decltype(auto) v = *it;
  538. v.second = value.as<V>();
  539. return {};
  540. }
  541. static detail::error_result set_associative(std::false_type, iterator& it, stack_object value) {
  542. decltype(auto) v = *it;
  543. v = value.as<V>();
  544. return {};
  545. }
  546. static detail::error_result set_writable(std::true_type, lua_State*, T&, iterator& it, stack_object value) {
  547. return set_associative(is_associative(), it, std::move(value));
  548. }
  549. static detail::error_result set_writable(std::false_type, lua_State*, T&, iterator&, stack_object) {
  550. return detail::error_result(
  551. "cannot perform a 'set': '%s's iterator reference is not writable (non-copy-assignable or const)", detail::demangle<T>().data());
  552. }
  553. static detail::error_result set_category(std::input_iterator_tag, lua_State* L_, T& self, stack_object okey, stack_object value) {
  554. decltype(auto) key = okey.as<K>();
  555. key = static_cast<K>(static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
  556. auto e = deferred_uc::end(L_, self);
  557. auto it = deferred_uc::begin(L_, self);
  558. auto backit = it;
  559. for (; key > 0 && it != e; --key, ++it) {
  560. backit = it;
  561. }
  562. if (it == e) {
  563. if (key == 0) {
  564. return add_copyable(is_copyable(), L_, self, std::move(value), meta::has_insert_after<T>::value ? backit : it);
  565. }
  566. return detail::error_result("out of bounds (too big) for set on '%s'", detail::demangle<T>().c_str());
  567. }
  568. return set_writable(is_writable(), L_, self, it, std::move(value));
  569. }
  570. static detail::error_result set_category(std::random_access_iterator_tag, lua_State* L_, T& self, stack_object okey, stack_object value) {
  571. decltype(auto) key = okey.as<K>();
  572. key = static_cast<K>(static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
  573. if (key < 0) {
  574. return detail::error_result("sol: out of bounds (too small) for set on '%s'", detail::demangle<T>().c_str());
  575. }
  576. std::ptrdiff_t len = static_cast<std::ptrdiff_t>(size_start(L_, self));
  577. if (key == len) {
  578. return add_copyable(is_copyable(), L_, self, std::move(value));
  579. }
  580. else if (key >= len) {
  581. return detail::error_result("sol: out of bounds (too big) for set on '%s'", detail::demangle<T>().c_str());
  582. }
  583. auto it = std::next(deferred_uc::begin(L_, self), key);
  584. return set_writable(is_writable(), L_, self, it, std::move(value));
  585. }
  586. static detail::error_result set_comparative(std::true_type, lua_State* L_, T& self, stack_object okey, stack_object value) {
  587. decltype(auto) key = okey.as<K>();
  588. if (!is_writable::value) {
  589. return detail::error_result(
  590. "cannot perform a 'set': '%s's iterator reference is not writable (non-copy-assignable or const)", detail::demangle<T>().data());
  591. }
  592. auto fx = [&](const value_type& r) -> bool { return key == get_key(is_associative(), r); };
  593. auto e = deferred_uc::end(L_, self);
  594. auto it = std::find_if(deferred_uc::begin(L_, self), e, std::ref(fx));
  595. if (it == e) {
  596. return {};
  597. }
  598. return set_writable(is_writable(), L_, self, it, std::move(value));
  599. }
  600. static detail::error_result set_comparative(std::false_type, lua_State*, T&, stack_object, stack_object) {
  601. return detail::error_result("cannot set this value on '%s': no suitable way to increment iterator or compare to '%s' key",
  602. detail::demangle<T>().data(),
  603. detail::demangle<K>().data());
  604. }
  605. template <typename Iter>
  606. static detail::error_result set_associative_insert(std::true_type, lua_State*, T& self, Iter& it, K& key, stack_object value) {
  607. if constexpr (meta::has_insert_with_iterator<T>::value) {
  608. self.insert(it, value_type(key, value.as<V>()));
  609. return {};
  610. }
  611. else if constexpr (meta::has_insert<T>::value) {
  612. self.insert(value_type(key, value.as<V>()));
  613. return {};
  614. }
  615. else {
  616. (void)self;
  617. (void)it;
  618. (void)key;
  619. return detail::error_result(
  620. "cannot call 'set' on '%s': there is no 'insert' function on this associative type", detail::demangle<T>().c_str());
  621. }
  622. }
  623. template <typename Iter>
  624. static detail::error_result set_associative_insert(std::false_type, lua_State*, T& self, Iter& it, K& key, stack_object) {
  625. if constexpr (meta::has_insert_with_iterator<T>::value) {
  626. self.insert(it, key);
  627. return {};
  628. }
  629. else if constexpr (meta::has_insert<T>::value) {
  630. self.insert(key);
  631. return {};
  632. }
  633. else {
  634. (void)self;
  635. (void)it;
  636. (void)key;
  637. return detail::error_result(
  638. "cannot call 'set' on '%s': there is no 'insert' function on this non-associative type", detail::demangle<T>().c_str());
  639. }
  640. }
  641. static detail::error_result set_associative_find(std::true_type, lua_State* L_, T& self, stack_object okey, stack_object value) {
  642. decltype(auto) key = okey.as<K>();
  643. auto it = self.find(key);
  644. if (it == deferred_uc::end(L_, self)) {
  645. return set_associative_insert(is_associative(), L_, self, it, key, std::move(value));
  646. }
  647. return set_writable(is_writable(), L_, self, it, std::move(value));
  648. }
  649. static detail::error_result set_associative_find(std::false_type, lua_State* L_, T& self, stack_object key, stack_object value) {
  650. return set_comparative(meta::supports_op_equal<K, key_type>(), L_, self, std::move(key), std::move(value));
  651. }
  652. static detail::error_result set_it(std::true_type, lua_State* L_, T& self, stack_object key, stack_object value) {
  653. return set_category(iterator_category(), L_, self, std::move(key), std::move(value));
  654. }
  655. static detail::error_result set_it(std::false_type, lua_State* L_, T& self, stack_object key, stack_object value) {
  656. return set_associative_find(meta::all<has_find<T>, meta::any<is_associative, is_lookup>>(), L_, self, std::move(key), std::move(value));
  657. }
  658. template <bool idx_of = false>
  659. static detail::error_result find_has_associative_lookup(std::true_type, lua_State* L_, T& self) {
  660. if constexpr (!is_ordered::value && idx_of) {
  661. (void)L_;
  662. (void)self;
  663. return detail::error_result("cannot perform an 'index_of': '%s's is not an ordered container", detail::demangle<T>().data());
  664. }
  665. else {
  666. decltype(auto) key = stack::unqualified_get<K>(L_, 2);
  667. auto it = self.find(key);
  668. if (it == deferred_uc::end(L_, self)) {
  669. return stack::push(L_, lua_nil);
  670. }
  671. if constexpr (idx_of) {
  672. auto dist = std::distance(deferred_uc::begin(L_, self), it);
  673. dist -= deferred_uc::index_adjustment(L_, self);
  674. return stack::push(L_, dist);
  675. }
  676. else {
  677. return get_associative(is_associative(), L_, it);
  678. }
  679. }
  680. }
  681. template <bool idx_of = false>
  682. static detail::error_result find_has_associative_lookup(std::false_type, lua_State* L_, T& self) {
  683. if constexpr (!is_ordered::value && idx_of) {
  684. (void)L_;
  685. (void)self;
  686. return detail::error_result("cannot perform an 'index_of': '%s's is not an ordered container", detail::demangle<T>().data());
  687. }
  688. else {
  689. decltype(auto) value = stack::unqualified_get<V>(L_, 2);
  690. auto it = self.find(value);
  691. if (it == deferred_uc::end(L_, self)) {
  692. return stack::push(L_, lua_nil);
  693. }
  694. if constexpr (idx_of) {
  695. auto dist = std::distance(deferred_uc::begin(L_, self), it);
  696. dist -= deferred_uc::index_adjustment(L_, self);
  697. return stack::push(L_, dist);
  698. }
  699. else {
  700. return get_associative(is_associative(), L_, it);
  701. }
  702. }
  703. }
  704. template <bool idx_of = false>
  705. static detail::error_result find_has(std::true_type, lua_State* L_, T& self) {
  706. return find_has_associative_lookup<idx_of>(meta::any<is_lookup, is_associative>(), L_, self);
  707. }
  708. template <typename Iter>
  709. static detail::error_result find_associative_lookup(std::true_type, lua_State* L_, T&, Iter& it, std::size_t) {
  710. return get_associative(is_associative(), L_, it);
  711. }
  712. template <typename Iter>
  713. static detail::error_result find_associative_lookup(std::false_type, lua_State* L_, T& self, Iter&, std::size_t idx) {
  714. idx = static_cast<std::size_t>(static_cast<std::ptrdiff_t>(idx) - deferred_uc::index_adjustment(L_, self));
  715. return stack::push(L_, idx);
  716. }
  717. template <bool = false>
  718. static detail::error_result find_comparative(std::false_type, lua_State*, T&) {
  719. return detail::error_result("cannot call 'find' on '%s': there is no 'find' function and the value_type is not equality comparable",
  720. detail::demangle<T>().c_str());
  721. }
  722. template <bool idx_of = false>
  723. static detail::error_result find_comparative(std::true_type, lua_State* L_, T& self) {
  724. decltype(auto) value = stack::unqualified_get<V>(L_, 2);
  725. auto it = deferred_uc::begin(L_, self);
  726. auto e = deferred_uc::end(L_, self);
  727. std::size_t idx = 0;
  728. for (;; ++it, ++idx) {
  729. if (it == e) {
  730. return stack::push(L_, lua_nil);
  731. }
  732. if (value == get_value(is_associative(), *it)) {
  733. break;
  734. }
  735. }
  736. return find_associative_lookup(meta::all<meta::boolean<!idx_of>, meta::any<is_lookup, is_associative>>(), L_, self, it, idx);
  737. }
  738. template <bool idx_of = false>
  739. static detail::error_result find_has(std::false_type, lua_State* L_, T& self) {
  740. return find_comparative<idx_of>(meta::supports_op_equal<V>(), L_, self);
  741. }
  742. template <typename Iter>
  743. static detail::error_result add_insert_after(std::false_type, lua_State* L_, T& self, stack_object value, Iter&) {
  744. return add_insert_after(std::false_type(), L_, self, value);
  745. }
  746. static detail::error_result add_insert_after(std::false_type, lua_State*, T&, stack_object) {
  747. return detail::error_result("cannot call 'add' on type '%s': no suitable insert/push_back C++ functions", detail::demangle<T>().data());
  748. }
  749. template <typename Iter>
  750. static detail::error_result add_insert_after(std::true_type, lua_State*, T& self, stack_object value, Iter& pos) {
  751. self.insert_after(pos, value.as<V>());
  752. return {};
  753. }
  754. static detail::error_result add_insert_after(std::true_type, lua_State* L_, T& self, stack_object value) {
  755. auto backit = self.before_begin();
  756. {
  757. auto e = deferred_uc::end(L_, self);
  758. for (auto it = deferred_uc::begin(L_, self); it != e; ++backit, ++it) { }
  759. }
  760. return add_insert_after(std::true_type(), L_, self, value, backit);
  761. }
  762. template <typename Iter>
  763. static detail::error_result add_insert(std::true_type, lua_State*, T& self, stack_object value, Iter& pos) {
  764. self.insert(pos, value.as<V>());
  765. return {};
  766. }
  767. static detail::error_result add_insert(std::true_type, lua_State* L_, T& self, stack_object value) {
  768. auto pos = deferred_uc::end(L_, self);
  769. return add_insert(std::true_type(), L_, self, value, pos);
  770. }
  771. template <typename Iter>
  772. static detail::error_result add_insert(std::false_type, lua_State* L_, T& self, stack_object value, Iter& pos) {
  773. return add_insert_after(meta::has_insert_after<T>(), L_, self, std::move(value), pos);
  774. }
  775. static detail::error_result add_insert(std::false_type, lua_State* L_, T& self, stack_object value) {
  776. return add_insert_after(meta::has_insert_after<T>(), L_, self, std::move(value));
  777. }
  778. template <typename Iter>
  779. static detail::error_result add_push_back(std::true_type, lua_State*, T& self, stack_object value, Iter&) {
  780. self.push_back(value.as<V>());
  781. return {};
  782. }
  783. static detail::error_result add_push_back(std::true_type, lua_State*, T& self, stack_object value) {
  784. self.push_back(value.as<V>());
  785. return {};
  786. }
  787. template <typename Iter>
  788. static detail::error_result add_push_back(std::false_type, lua_State* L_, T& self, stack_object value, Iter& pos) {
  789. return add_insert(
  790. std::integral_constant < bool, meta::has_insert<T>::value || meta::has_insert_with_iterator<T>::value > (), L_, self, value, pos);
  791. }
  792. static detail::error_result add_push_back(std::false_type, lua_State* L_, T& self, stack_object value) {
  793. return add_insert(
  794. std::integral_constant < bool, meta::has_insert<T>::value || meta::has_insert_with_iterator<T>::value > (), L_, self, value);
  795. }
  796. template <typename Iter>
  797. static detail::error_result add_associative(std::true_type, lua_State* L_, T& self, stack_object key, Iter& pos) {
  798. if constexpr (meta::has_insert_with_iterator<T>::value) {
  799. self.insert(pos, value_type(key.as<K>(), stack::unqualified_get<V>(L_, 3)));
  800. return {};
  801. }
  802. else if constexpr (meta::has_insert<T>::value) {
  803. self.insert(value_type(key.as<K>(), stack::unqualified_get<V>(L_, 3)));
  804. return {};
  805. }
  806. else {
  807. (void)L_;
  808. (void)self;
  809. (void)key;
  810. (void)pos;
  811. return detail::error_result(
  812. "cannot call 'insert' on '%s': there is no 'insert' function on this associative type", detail::demangle<T>().c_str());
  813. }
  814. }
  815. static detail::error_result add_associative(std::true_type, lua_State* L_, T& self, stack_object key) {
  816. auto pos = deferred_uc::end(L_, self);
  817. return add_associative(std::true_type(), L_, self, std::move(key), pos);
  818. }
  819. template <typename Iter>
  820. static detail::error_result add_associative(std::false_type, lua_State* L_, T& self, stack_object value, Iter& pos) {
  821. return add_push_back(meta::has_push_back<T>(), L_, self, value, pos);
  822. }
  823. static detail::error_result add_associative(std::false_type, lua_State* L_, T& self, stack_object value) {
  824. return add_push_back(meta::has_push_back<T>(), L_, self, value);
  825. }
  826. template <typename Iter>
  827. static detail::error_result add_copyable(std::true_type, lua_State* L_, T& self, stack_object value, Iter& pos) {
  828. return add_associative(is_associative(), L_, self, std::move(value), pos);
  829. }
  830. static detail::error_result add_copyable(std::true_type, lua_State* L_, T& self, stack_object value) {
  831. return add_associative(is_associative(), L_, self, value);
  832. }
  833. template <typename Iter>
  834. static detail::error_result add_copyable(std::false_type, lua_State* L_, T& self, stack_object value, Iter&) {
  835. return add_copyable(std::false_type(), L_, self, std::move(value));
  836. }
  837. static detail::error_result add_copyable(std::false_type, lua_State*, T&, stack_object) {
  838. return detail::error_result("cannot call 'add' on '%s': value_type is non-copyable", detail::demangle<T>().data());
  839. }
  840. static detail::error_result insert_lookup(std::true_type, lua_State* L_, T& self, stack_object, stack_object value) {
  841. // TODO: should we warn or error about someone calling insert on an ordered / lookup container with no associativity?
  842. return add_copyable(std::true_type(), L_, self, std::move(value));
  843. }
  844. static detail::error_result insert_lookup(std::false_type, lua_State* L_, T& self, stack_object where, stack_object value) {
  845. auto it = deferred_uc::begin(L_, self);
  846. auto key = where.as<K>();
  847. key = static_cast<K>(static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
  848. std::advance(it, key);
  849. self.insert(it, value.as<V>());
  850. return {};
  851. }
  852. static detail::error_result insert_after_has(std::true_type, lua_State* L_, T& self, stack_object where, stack_object value) {
  853. auto key = where.as<K>();
  854. auto backit = self.before_begin();
  855. {
  856. key = static_cast<K>(static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
  857. auto e = deferred_uc::end(L_, self);
  858. for (auto it = deferred_uc::begin(L_, self); key > 0; ++backit, ++it, --key) {
  859. if (backit == e) {
  860. return detail::error_result("sol: out of bounds (too big) for set on '%s'", detail::demangle<T>().c_str());
  861. }
  862. }
  863. }
  864. self.insert_after(backit, value.as<V>());
  865. return {};
  866. }
  867. static detail::error_result insert_after_has(std::false_type, lua_State*, T&, stack_object, stack_object) {
  868. return detail::error_result(
  869. "cannot call 'insert' on '%s': no suitable or similar functionality detected on this container", detail::demangle<T>().data());
  870. }
  871. static detail::error_result insert_has(std::true_type, lua_State* L_, T& self, stack_object key, stack_object value) {
  872. return insert_lookup(meta::any<is_associative, is_lookup>(), L_, self, std::move(key), std::move(value));
  873. }
  874. static detail::error_result insert_has(std::false_type, lua_State* L_, T& self, stack_object where, stack_object value) {
  875. return insert_after_has(meta::has_insert_after<T>(), L_, self, where, value);
  876. }
  877. static detail::error_result insert_copyable(std::true_type, lua_State* L_, T& self, stack_object key, stack_object value) {
  878. return insert_has(std::integral_constant < bool,
  879. meta::has_insert<T>::value || meta::has_insert_with_iterator<T>::value > (),
  880. L_,
  881. self,
  882. std::move(key),
  883. std::move(value));
  884. }
  885. static detail::error_result insert_copyable(std::false_type, lua_State*, T&, stack_object, stack_object) {
  886. return detail::error_result("cannot call 'insert' on '%s': value_type is non-copyable", detail::demangle<T>().data());
  887. }
  888. static detail::error_result erase_integral(std::true_type, lua_State* L_, T& self, K& key) {
  889. auto it = deferred_uc::begin(L_, self);
  890. key = (static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
  891. std::advance(it, key);
  892. self.erase(it);
  893. return {};
  894. }
  895. static detail::error_result erase_integral(std::false_type, lua_State* L_, T& self, const K& key) {
  896. auto fx = [&](const value_type& r) -> bool { return key == r; };
  897. auto e = deferred_uc::end(L_, self);
  898. auto it = std::find_if(deferred_uc::begin(L_, self), e, std::ref(fx));
  899. if (it == e) {
  900. return {};
  901. }
  902. self.erase(it);
  903. return {};
  904. }
  905. static detail::error_result erase_associative_lookup(std::true_type, lua_State*, T& self, const K& key) {
  906. self.erase(key);
  907. return {};
  908. }
  909. static detail::error_result erase_associative_lookup(std::false_type, lua_State* L_, T& self, K& key) {
  910. return erase_integral(std::is_integral<K>(), L_, self, key);
  911. }
  912. static detail::error_result erase_after_has(std::true_type, lua_State* L_, T& self, K& key) {
  913. auto backit = self.before_begin();
  914. {
  915. key = static_cast<K>(static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
  916. auto e = deferred_uc::end(L_, self);
  917. for (auto it = deferred_uc::begin(L_, self); key > 0; ++backit, ++it, --key) {
  918. if (backit == e) {
  919. return detail::error_result("sol: out of bounds for erase on '%s'", detail::demangle<T>().c_str());
  920. }
  921. }
  922. }
  923. self.erase_after(backit);
  924. return {};
  925. }
  926. static detail::error_result erase_after_has(std::false_type, lua_State*, T&, const K&) {
  927. return detail::error_result("sol: cannot call erase on '%s'", detail::demangle<T>().c_str());
  928. }
  929. static detail::error_result erase_key_has(std::true_type, lua_State* L_, T& self, K& key) {
  930. return erase_associative_lookup(meta::any<is_associative, is_lookup>(), L_, self, key);
  931. }
  932. static detail::error_result erase_key_has(std::false_type, lua_State* L_, T& self, K& key) {
  933. return erase_after_has(has_erase_after<T>(), L_, self, key);
  934. }
  935. static detail::error_result erase_has(std::true_type, lua_State* L_, T& self, K& key) {
  936. return erase_associative_lookup(meta::any<is_associative, is_lookup>(), L_, self, key);
  937. }
  938. static detail::error_result erase_has(std::false_type, lua_State* L_, T& self, K& key) {
  939. return erase_key_has(has_erase_key<T>(), L_, self, key);
  940. }
  941. static auto size_has(std::false_type, lua_State* L_, T& self) {
  942. return std::distance(deferred_uc::begin(L_, self), deferred_uc::end(L_, self));
  943. }
  944. static auto size_has(std::true_type, lua_State*, T& self) {
  945. return self.size();
  946. }
  947. static void clear_has(std::true_type, lua_State*, T& self) {
  948. self.clear();
  949. }
  950. static void clear_has(std::false_type, lua_State* L_, T&) {
  951. luaL_error(L_, "sol: cannot call clear on '%s'", detail::demangle<T>().c_str());
  952. }
  953. static bool empty_has(std::true_type, lua_State*, T& self) {
  954. return self.empty();
  955. }
  956. static bool empty_has(std::false_type, lua_State* L_, T& self) {
  957. return deferred_uc::begin(L_, self) == deferred_uc::end(L_, self);
  958. }
  959. static detail::error_result get_associative_find(std::true_type, lua_State* L_, T& self, K& key) {
  960. auto it = self.find(key);
  961. if (it == deferred_uc::end(L_, self)) {
  962. stack::push(L_, lua_nil);
  963. return {};
  964. }
  965. return get_associative(std::true_type(), L_, it);
  966. }
  967. static detail::error_result get_associative_find(std::false_type, lua_State* L_, T& self, K& key) {
  968. return get_it(is_linear_integral(), L_, self, key);
  969. }
  970. static detail::error_result get_start(lua_State* L_, T& self, K& key) {
  971. return get_associative_find(std::integral_constant < bool, is_associative::value&& has_find<T>::value > (), L_, self, key);
  972. }
  973. static detail::error_result set_start(lua_State* L_, T& self, stack_object key, stack_object value) {
  974. return set_it(is_linear_integral(), L_, self, std::move(key), std::move(value));
  975. }
  976. static std::size_t size_start(lua_State* L_, T& self) {
  977. return static_cast<std::size_t>(size_has(meta::has_size<T>(), L_, self));
  978. }
  979. static void clear_start(lua_State* L_, T& self) {
  980. clear_has(has_clear<T>(), L_, self);
  981. }
  982. static bool empty_start(lua_State* L_, T& self) {
  983. return empty_has(has_empty<T>(), L_, self);
  984. }
  985. static detail::error_result erase_start(lua_State* L_, T& self, K& key) {
  986. return erase_has(has_erase<T>(), L_, self, key);
  987. }
  988. template <bool ip>
  989. static int next_associative(std::true_type, lua_State* L_) {
  990. iter& i = stack::unqualified_get<user<iter>>(L_, 1);
  991. auto& it = i.it;
  992. auto& end = i.end;
  993. if (it == end) {
  994. return stack::push(L_, lua_nil);
  995. }
  996. int p;
  997. if constexpr (ip) {
  998. ++i.index;
  999. p = stack::push_reference(L_, i.index);
  1000. }
  1001. else {
  1002. p = stack::push_reference(L_, it->first);
  1003. }
  1004. p += stack::stack_detail::push_reference<push_type>(L_, detail::deref_move_only(it->second));
  1005. std::advance(it, 1);
  1006. return p;
  1007. }
  1008. template <bool>
  1009. static int next_associative(std::false_type, lua_State* L_) {
  1010. iter& i = stack::unqualified_get<user<iter>>(L_, 1);
  1011. auto& it = i.it();
  1012. auto& end = i.sen();
  1013. next_K k = stack::unqualified_get<next_K>(L_, 2);
  1014. if (it == end) {
  1015. return stack::push(L_, lua_nil);
  1016. }
  1017. int p;
  1018. if constexpr (std::is_integral_v<next_K>) {
  1019. p = stack::push_reference(L_, k + 1);
  1020. }
  1021. else {
  1022. p = stack::stack_detail::push_reference(L_, k + 1);
  1023. }
  1024. p += stack::stack_detail::push_reference<push_type>(L_, detail::deref_move_only(*it));
  1025. std::advance(it, 1);
  1026. return p;
  1027. }
  1028. template <bool ip>
  1029. static int next_iter(lua_State* L_) {
  1030. typedef meta::any<is_associative, meta::all<is_lookup, meta::neg<is_matched_lookup>>> is_assoc;
  1031. return next_associative<ip>(is_assoc(), L_);
  1032. }
  1033. template <bool ip>
  1034. static int pairs_associative(std::true_type, lua_State* L_) {
  1035. auto& src = get_src(L_);
  1036. stack::push(L_, next_iter<ip>);
  1037. stack::push<user<iter>>(L_, L_, 1, deferred_uc::begin(L_, src), deferred_uc::begin(L_, src));
  1038. stack::push(L_, lua_nil);
  1039. return 3;
  1040. }
  1041. template <bool ip>
  1042. static int pairs_associative(std::false_type, lua_State* L_) {
  1043. auto& src = get_src(L_);
  1044. stack::push(L_, next_iter<ip>);
  1045. stack::push<user<iter>>(L_, L_, 1, deferred_uc::begin(L_, src), deferred_uc::end(L_, src));
  1046. stack::push(L_, 0);
  1047. return 3;
  1048. }
  1049. public:
  1050. static int at(lua_State* L_) {
  1051. auto& self = get_src(L_);
  1052. detail::error_result er;
  1053. {
  1054. std::ptrdiff_t pos = stack::unqualified_get<std::ptrdiff_t>(L_, 2);
  1055. er = at_start(L_, self, pos);
  1056. }
  1057. return handle_errors(L_, er);
  1058. }
  1059. static int get(lua_State* L_) {
  1060. auto& self = get_src(L_);
  1061. detail::error_result er;
  1062. {
  1063. decltype(auto) key = stack::unqualified_get<K>(L_);
  1064. er = get_start(L_, self, key);
  1065. }
  1066. return handle_errors(L_, er);
  1067. }
  1068. static int index_get(lua_State* L_) {
  1069. return get(L_);
  1070. }
  1071. static int set(lua_State* L_) {
  1072. stack_object value = stack_object(L_, raw_index(3));
  1073. if constexpr (is_linear_integral::value) {
  1074. // for non-associative containers,
  1075. // erasure only happens if it is the
  1076. // last index in the container
  1077. auto key = stack::get<K>(L_, 2);
  1078. auto self_size = deferred_uc::size(L_);
  1079. if (key == static_cast<K>(self_size)) {
  1080. if (type_of(L_, 3) == type::lua_nil) {
  1081. return erase(L_);
  1082. }
  1083. }
  1084. }
  1085. else {
  1086. if (type_of(L_, 3) == type::lua_nil) {
  1087. return erase(L_);
  1088. }
  1089. }
  1090. auto& self = get_src(L_);
  1091. detail::error_result er = set_start(L_, self, stack_object(L_, raw_index(2)), std::move(value));
  1092. return handle_errors(L_, er);
  1093. }
  1094. static int index_set(lua_State* L_) {
  1095. return set(L_);
  1096. }
  1097. static int add(lua_State* L_) {
  1098. auto& self = get_src(L_);
  1099. detail::error_result er = add_copyable(is_copyable(), L_, self, stack_object(L_, raw_index(2)));
  1100. return handle_errors(L_, er);
  1101. }
  1102. static int insert(lua_State* L_) {
  1103. auto& self = get_src(L_);
  1104. detail::error_result er = insert_copyable(is_copyable(), L_, self, stack_object(L_, raw_index(2)), stack_object(L_, raw_index(3)));
  1105. return handle_errors(L_, er);
  1106. }
  1107. static int find(lua_State* L_) {
  1108. auto& self = get_src(L_);
  1109. detail::error_result er = find_has(has_find<T>(), L_, self);
  1110. return handle_errors(L_, er);
  1111. }
  1112. static int index_of(lua_State* L_) {
  1113. auto& self = get_src(L_);
  1114. detail::error_result er = find_has<true>(has_find<T>(), L_, self);
  1115. return handle_errors(L_, er);
  1116. }
  1117. static iterator begin(lua_State*, T& self) {
  1118. if constexpr (meta::has_begin_end_v<T>) {
  1119. return self.begin();
  1120. }
  1121. else {
  1122. using std::begin;
  1123. return begin(self);
  1124. }
  1125. }
  1126. static sentinel end(lua_State*, T& self) {
  1127. if constexpr (meta::has_begin_end_v<T>) {
  1128. return self.end();
  1129. }
  1130. else {
  1131. using std::end;
  1132. return end(self);
  1133. }
  1134. }
  1135. static int size(lua_State* L_) {
  1136. auto& self = get_src(L_);
  1137. std::size_t r = size_start(L_, self);
  1138. return stack::push(L_, r);
  1139. }
  1140. static int clear(lua_State* L_) {
  1141. auto& self = get_src(L_);
  1142. clear_start(L_, self);
  1143. return 0;
  1144. }
  1145. static int erase(lua_State* L_) {
  1146. auto& self = get_src(L_);
  1147. detail::error_result er;
  1148. {
  1149. decltype(auto) key = stack::unqualified_get<K>(L_, 2);
  1150. er = erase_start(L_, self, key);
  1151. }
  1152. return handle_errors(L_, er);
  1153. }
  1154. static int empty(lua_State* L_) {
  1155. auto& self = get_src(L_);
  1156. return stack::push(L_, empty_start(L_, self));
  1157. }
  1158. static std::ptrdiff_t index_adjustment(lua_State*, T&) {
  1159. return static_cast<std::ptrdiff_t>((SOL_CONTAINER_START_INDEX_I_) == 0 ? 0 : -(SOL_CONTAINER_START_INDEX_I_));
  1160. }
  1161. static int pairs(lua_State* L_) {
  1162. typedef meta::any<is_associative, meta::all<is_lookup, meta::neg<is_matched_lookup>>> is_assoc;
  1163. return pairs_associative<false>(is_assoc(), L_);
  1164. }
  1165. static int ipairs(lua_State* L_) {
  1166. typedef meta::any<is_associative, meta::all<is_lookup, meta::neg<is_matched_lookup>>> is_assoc;
  1167. return pairs_associative<true>(is_assoc(), L_);
  1168. }
  1169. static int next(lua_State* L_) {
  1170. return stack::push(L_, next_iter<false>);
  1171. }
  1172. };
  1173. template <typename X>
  1174. struct usertype_container_default<X, std::enable_if_t<std::is_array<std::remove_pointer_t<meta::unwrap_unqualified_t<X>>>::value>> {
  1175. private:
  1176. typedef std::remove_pointer_t<meta::unwrap_unqualified_t<X>> T;
  1177. typedef usertype_container<X> deferred_uc;
  1178. public:
  1179. typedef std::remove_extent_t<T> value_type;
  1180. typedef value_type* iterator;
  1181. typedef iterator sentinel;
  1182. private:
  1183. struct iter : detail::ebco<iterator, 0>, detail::ebco<sentinel, 1> {
  1184. using it_base = detail::ebco<iterator, 0>;
  1185. using sen_base = detail::ebco<sentinel, 1>;
  1186. reference keep_alive;
  1187. iter(lua_State* L_, int stack_index_, iterator it_, sentinel sen_) noexcept
  1188. : it_base(std::move(it_)), sen_base(std::move(sen_)), keep_alive(sol::main_thread(L_, L_), stack_index_) {
  1189. }
  1190. iterator& it() noexcept {
  1191. return it_base::value();
  1192. }
  1193. const iterator& it() const noexcept {
  1194. return it_base::value();
  1195. }
  1196. sentinel& sen() noexcept {
  1197. return sen_base::value();
  1198. }
  1199. const sentinel& sen() const noexcept {
  1200. return sen_base::value();
  1201. }
  1202. };
  1203. static auto& get_src(lua_State* L_) {
  1204. auto p = stack::unqualified_check_get<T*>(L_, 1);
  1205. #if SOL_IS_ON(SOL_SAFE_USERTYPE)
  1206. if (!p) {
  1207. luaL_error(L_,
  1208. "sol: 'self' is not of type '%s' (pass 'self' as first argument with ':' or call on proper type)",
  1209. detail::demangle<T>().c_str());
  1210. }
  1211. if (p.value() == nullptr) {
  1212. luaL_error(
  1213. L_, "sol: 'self' argument is nil (pass 'self' as first argument with ':' or call on a '%s' type)", detail::demangle<T>().c_str());
  1214. }
  1215. #endif // Safe getting with error
  1216. return *p.value();
  1217. }
  1218. static int find(std::true_type, lua_State* L_) {
  1219. T& self = get_src(L_);
  1220. decltype(auto) value = stack::unqualified_get<value_type>(L_, 2);
  1221. std::size_t N = std::extent<T>::value;
  1222. for (std::size_t idx = 0; idx < N; ++idx) {
  1223. using v_t = std::add_const_t<decltype(self[idx])>;
  1224. v_t v = self[idx];
  1225. if (v == value) {
  1226. idx = static_cast<std::size_t>(static_cast<std::ptrdiff_t>(idx) - deferred_uc::index_adjustment(L_, self));
  1227. return stack::push(L_, idx);
  1228. }
  1229. }
  1230. return stack::push(L_, lua_nil);
  1231. }
  1232. static int find(std::false_type, lua_State* L_) {
  1233. return luaL_error(L_, "sol: cannot call 'find' on '%s': no supported comparison operator for the value type", detail::demangle<T>().c_str());
  1234. }
  1235. static int next_iter(lua_State* L_) {
  1236. iter& i = stack::unqualified_get<user<iter>>(L_, 1);
  1237. auto& it = i.it();
  1238. auto& end = i.sen();
  1239. std::size_t k = stack::unqualified_get<std::size_t>(L_, 2);
  1240. if (it == end) {
  1241. return 0;
  1242. }
  1243. int p;
  1244. p = stack::push(L_, k + 1);
  1245. p += stack::push_reference(L_, detail::deref_move_only(*it));
  1246. std::advance(it, 1);
  1247. return p;
  1248. }
  1249. public:
  1250. static int clear(lua_State* L_) {
  1251. return luaL_error(L_, "sol: cannot call 'clear' on type '%s': cannot remove all items from a fixed array", detail::demangle<T>().c_str());
  1252. }
  1253. static int erase(lua_State* L_) {
  1254. return luaL_error(L_, "sol: cannot call 'erase' on type '%s': cannot remove an item from fixed arrays", detail::demangle<T>().c_str());
  1255. }
  1256. static int add(lua_State* L_) {
  1257. return luaL_error(L_, "sol: cannot call 'add' on type '%s': cannot add to fixed arrays", detail::demangle<T>().c_str());
  1258. }
  1259. static int insert(lua_State* L_) {
  1260. return luaL_error(L_, "sol: cannot call 'insert' on type '%s': cannot insert new entries into fixed arrays", detail::demangle<T>().c_str());
  1261. }
  1262. static int at(lua_State* L_) {
  1263. return get(L_);
  1264. }
  1265. static int get(lua_State* L_) {
  1266. T& self = get_src(L_);
  1267. std::ptrdiff_t idx = stack::unqualified_get<std::ptrdiff_t>(L_, 2);
  1268. idx += deferred_uc::index_adjustment(L_, self);
  1269. if (idx >= static_cast<std::ptrdiff_t>(std::extent<T>::value) || idx < 0) {
  1270. return stack::push(L_, lua_nil);
  1271. }
  1272. return stack::push_reference(L_, detail::deref_move_only(self[idx]));
  1273. }
  1274. static int index_get(lua_State* L_) {
  1275. return get(L_);
  1276. }
  1277. static int set(lua_State* L_) {
  1278. T& self = get_src(L_);
  1279. std::ptrdiff_t idx = stack::unqualified_get<std::ptrdiff_t>(L_, 2);
  1280. idx += deferred_uc::index_adjustment(L_, self);
  1281. if (idx >= static_cast<std::ptrdiff_t>(std::extent<T>::value)) {
  1282. return luaL_error(L_, "sol: index out of bounds (too big) for set on '%s'", detail::demangle<T>().c_str());
  1283. }
  1284. if (idx < 0) {
  1285. return luaL_error(L_, "sol: index out of bounds (too small) for set on '%s'", detail::demangle<T>().c_str());
  1286. }
  1287. self[idx] = stack::unqualified_get<value_type>(L_, 3);
  1288. return 0;
  1289. }
  1290. static int index_set(lua_State* L_) {
  1291. return set(L_);
  1292. }
  1293. static int index_of(lua_State* L_) {
  1294. return find(L_);
  1295. }
  1296. static int find(lua_State* L_) {
  1297. return find(meta::supports_op_equal<value_type, value_type>(), L_);
  1298. }
  1299. static int size(lua_State* L_) {
  1300. return stack::push(L_, std::extent<T>::value);
  1301. }
  1302. static int empty(lua_State* L_) {
  1303. return stack::push(L_, std::extent<T>::value > 0);
  1304. }
  1305. static int pairs(lua_State* L_) {
  1306. auto& src = get_src(L_);
  1307. stack::push(L_, next_iter);
  1308. stack::push<user<iter>>(L_, L_, 1, deferred_uc::begin(L_, src), deferred_uc::end(L_, src));
  1309. stack::push(L_, 0);
  1310. return 3;
  1311. }
  1312. static int ipairs(lua_State* L_) {
  1313. return pairs(L_);
  1314. }
  1315. static int next(lua_State* L_) {
  1316. return stack::push(L_, next_iter);
  1317. }
  1318. static std::ptrdiff_t index_adjustment(lua_State*, T&) {
  1319. return (SOL_CONTAINER_START_INDEX_I_) == 0 ? 0 : -(SOL_CONTAINER_START_INDEX_I_);
  1320. }
  1321. static iterator begin(lua_State*, T& self) {
  1322. return std::addressof(self[0]);
  1323. }
  1324. static sentinel end(lua_State*, T& self) {
  1325. return std::addressof(self[0]) + std::extent<T>::value;
  1326. }
  1327. };
  1328. template <typename X>
  1329. struct usertype_container_default<usertype_container<X>> : usertype_container_default<X> { };
  1330. } // namespace container_detail
  1331. template <typename T>
  1332. struct usertype_container : container_detail::usertype_container_default<T> { };
  1333. } // namespace sol
  1334. #endif // SOL_USERTYPE_CONTAINER_HPP