| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592 |
- // sol2
- // The MIT License (MIT)
- // Copyright (c) 2013-2022 Rapptz, ThePhD and contributors
- // Permission is hereby granted, free of charge, to any person obtaining a copy of
- // this software and associated documentation files (the "Software"), to deal in
- // the Software without restriction, including without limitation the rights to
- // use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
- // the Software, and to permit persons to whom the Software is furnished to do so,
- // subject to the following conditions:
- // The above copyright notice and this permission notice shall be included in all
- // copies or substantial portions of the Software.
- // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
- // FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
- // COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
- // IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
- // CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
- #ifndef SOL_USERTYPE_CONTAINER_HPP
- #define SOL_USERTYPE_CONTAINER_HPP
- #include <sol/traits.hpp>
- #include <sol/stack.hpp>
- #include <sol/object.hpp>
- namespace sol {
- template <typename T>
- struct usertype_container;
- namespace container_detail {
- template <typename T>
- struct has_clear_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::clear));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_empty_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::empty));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_erase_after_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(
- decltype(std::declval<C>().erase_after(std::declval<std::add_rvalue_reference_t<typename C::const_iterator>>()))*);
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T, typename = void>
- struct has_find_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(std::declval<C>().find(std::declval<std::add_rvalue_reference_t<typename C::value_type>>()))*);
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_find_test<T, std::enable_if_t<meta::is_lookup<T>::value>> {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(std::declval<C>().find(std::declval<std::add_rvalue_reference_t<typename C::key_type>>()))*);
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_erase_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(std::declval<C>().erase(std::declval<typename C::iterator>()))*);
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_erase_key_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(std::declval<C>().erase(std::declval<typename C::key_type>()))*);
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_find_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::find));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_index_of_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::index_of));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_insert_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::insert));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_erase_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::erase));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_index_set_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::index_set));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_index_get_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::index_get));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_set_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::set));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_get_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::get));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_at_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::at));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_pairs_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::pairs));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_ipairs_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::ipairs));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_next_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::next));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_add_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::add));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- struct has_traits_size_test {
- private:
- template <typename C>
- static meta::sfinae_yes_t test(decltype(&C::size));
- template <typename C>
- static meta::sfinae_no_t test(...);
- public:
- static constexpr bool value = std::is_same_v<decltype(test<T>(0)), meta::sfinae_yes_t>;
- };
- template <typename T>
- using has_clear = meta::boolean<has_clear_test<T>::value>;
- template <typename T>
- using has_empty = meta::boolean<has_empty_test<T>::value>;
- template <typename T>
- using has_find = meta::boolean<has_find_test<T>::value>;
- template <typename T>
- using has_erase = meta::boolean<has_erase_test<T>::value>;
- template <typename T>
- using has_erase_key = meta::boolean<has_erase_key_test<T>::value>;
- template <typename T>
- using has_erase_after = meta::boolean<has_erase_after_test<T>::value>;
- template <typename T>
- using has_traits_get = meta::boolean<has_traits_get_test<T>::value>;
- template <typename T>
- using has_traits_at = meta::boolean<has_traits_at_test<T>::value>;
- template <typename T>
- using has_traits_set = meta::boolean<has_traits_set_test<T>::value>;
- template <typename T>
- using has_traits_index_get = meta::boolean<has_traits_index_get_test<T>::value>;
- template <typename T>
- using has_traits_index_set = meta::boolean<has_traits_index_set_test<T>::value>;
- template <typename T>
- using has_traits_pairs = meta::boolean<has_traits_pairs_test<T>::value>;
- template <typename T>
- using has_traits_ipairs = meta::boolean<has_traits_ipairs_test<T>::value>;
- template <typename T>
- using has_traits_next = meta::boolean<has_traits_next_test<T>::value>;
- template <typename T>
- using has_traits_add = meta::boolean<has_traits_add_test<T>::value>;
- template <typename T>
- using has_traits_size = meta::boolean<has_traits_size_test<T>::value>;
- template <typename T>
- using has_traits_clear = has_clear<T>;
- template <typename T>
- using has_traits_empty = has_empty<T>;
- template <typename T>
- using has_traits_find = meta::boolean<has_traits_find_test<T>::value>;
- template <typename T>
- using has_traits_index_of = meta::boolean<has_traits_index_of_test<T>::value>;
- template <typename T>
- using has_traits_insert = meta::boolean<has_traits_insert_test<T>::value>;
- template <typename T>
- using has_traits_erase = meta::boolean<has_traits_erase_test<T>::value>;
- template <typename T>
- struct is_forced_container : is_container<T> { };
- template <typename T>
- struct is_forced_container<as_container_t<T>> : std::true_type { };
- template <typename T>
- struct container_decay {
- typedef T type;
- };
- template <typename T>
- struct container_decay<as_container_t<T>> {
- typedef T type;
- };
- template <typename T>
- using container_decay_t = typename container_decay<meta::unqualified_t<T>>::type;
- template <typename T>
- decltype(auto) get_key(std::false_type, T&& t) {
- return std::forward<T>(t);
- }
- template <typename T>
- decltype(auto) get_key(std::true_type, T&& t) {
- return t.first;
- }
- template <typename T>
- decltype(auto) get_value(std::false_type, T&& t) {
- return std::forward<T>(t);
- }
- template <typename T>
- decltype(auto) get_value(std::true_type, T&& t) {
- return t.second;
- }
- template <typename X, typename = void>
- struct usertype_container_default {
- private:
- typedef std::remove_pointer_t<meta::unwrap_unqualified_t<X>> T;
- public:
- typedef lua_nil_t iterator;
- typedef iterator sentinel;
- typedef lua_nil_t value_type;
- static int at(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'at(index)' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int get(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'get(key)' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int index_get(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'container[key]' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int set(lua_State* L_) {
- 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());
- }
- static int index_set(lua_State* L_) {
- return luaL_error(
- L_, "sol: cannot call 'container[key] = value' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int add(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'add' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int insert(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'insert' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int find(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'find' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int index_of(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'index_of' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int size(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'end' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int clear(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'clear' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int empty(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'empty' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int erase(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'erase' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int next(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'next' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int pairs(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call '__pairs/pairs' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static int ipairs(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call '__ipairs' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- }
- static iterator begin(lua_State* L_, T&) {
- luaL_error(L_, "sol: cannot call 'being' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- return lua_nil;
- }
- static sentinel end(lua_State* L_, T&) {
- luaL_error(L_, "sol: cannot call 'end' on type '%s': it is not recognized as a container", detail::demangle<T>().c_str());
- return lua_nil;
- }
- };
- template <typename X>
- struct usertype_container_default<X,
- std::enable_if_t<meta::all<is_forced_container<meta::unqualified_t<X>>, meta::has_value_type<meta::unqualified_t<container_decay_t<X>>>,
- meta::has_iterator<meta::unqualified_t<container_decay_t<X>>>>::value>> {
- private:
- using T = std::remove_pointer_t<meta::unwrap_unqualified_t<container_decay_t<X>>>;
- private:
- using deferred_uc = usertype_container<X>;
- using is_associative = meta::is_associative<T>;
- using is_lookup = meta::is_lookup<T>;
- using is_ordered = meta::is_ordered<T>;
- using is_matched_lookup = meta::is_matched_lookup<T>;
- using iterator = typename T::iterator;
- using sentinel = meta::sentinel_or_t<T, iterator>;
- using value_type = typename T::value_type;
- typedef meta::conditional_t<is_matched_lookup::value, std::pair<value_type, value_type>,
- meta::conditional_t<is_associative::value || is_lookup::value, value_type, std::pair<std::ptrdiff_t, value_type>>>
- KV;
- typedef typename KV::first_type K;
- typedef typename KV::second_type V;
- typedef meta::conditional_t<is_matched_lookup::value, std::ptrdiff_t, K> next_K;
- typedef decltype(*std::declval<iterator&>()) iterator_return;
- typedef meta::conditional_t<is_associative::value || is_matched_lookup::value, std::add_lvalue_reference_t<V>,
- meta::conditional_t<is_lookup::value, V, iterator_return>>
- captured_type;
- typedef typename meta::iterator_tag<iterator>::type iterator_category;
- typedef std::is_same<iterator_category, std::input_iterator_tag> is_input_iterator;
- typedef meta::conditional_t<is_input_iterator::value, V, decltype(detail::deref_move_only(std::declval<captured_type>()))> push_type;
- typedef std::is_copy_assignable<V> is_copyable;
- typedef meta::neg<meta::any<std::is_const<V>, std::is_const<std::remove_reference_t<iterator_return>>, meta::neg<is_copyable>>> is_writable;
- typedef meta::unqualified_t<decltype(get_key(is_associative(), std::declval<std::add_lvalue_reference_t<value_type>>()))> key_type;
- typedef meta::all<std::is_integral<K>, meta::neg<meta::any<is_associative, is_lookup>>> is_linear_integral;
- struct iter : detail::ebco<iterator, 0>, detail::ebco<sentinel, 1> {
- using it_base = detail::ebco<iterator, 0>;
- using sen_base = detail::ebco<sentinel, 1>;
- main_reference keep_alive;
- std::size_t index;
- iter(lua_State* L_, int stack_index_, iterator it_, sentinel sen_) noexcept
- : it_base(std::move(it_)), sen_base(std::move(sen_)), keep_alive(L_, stack_index_), index(0) {
- }
- iterator& it() noexcept {
- return it_base::value();
- }
- const iterator& it() const noexcept {
- return it_base::value();
- }
- sentinel& sen() noexcept {
- return sen_base::value();
- }
- const sentinel& sen() const noexcept {
- return sen_base::value();
- }
- };
- static auto& get_src(lua_State* L_) {
- #if SOL_IS_ON(SOL_SAFE_USERTYPE)
- auto p = stack::unqualified_check_get<T*>(L_, 1);
- if (!p) {
- luaL_error(L_,
- "sol: 'self' is not of type '%s' (pass 'self' as first argument with ':' or call on proper type)",
- detail::demangle<T>().c_str());
- }
- if (p.value() == nullptr) {
- luaL_error(
- L_, "sol: 'self' argument is nil (pass 'self' as first argument with ':' or call on a '%s' type)", detail::demangle<T>().c_str());
- }
- return *p.value();
- #else
- return stack::unqualified_get<T>(L_, 1);
- #endif // Safe getting with error
- }
- static detail::error_result at_category(std::input_iterator_tag, lua_State* L_, T& self, std::ptrdiff_t pos) {
- pos += deferred_uc::index_adjustment(L_, self);
- if (pos < 0) {
- return stack::push(L_, lua_nil);
- }
- auto it = deferred_uc::begin(L_, self);
- auto e = deferred_uc::end(L_, self);
- if (it == e) {
- return stack::push(L_, lua_nil);
- }
- while (pos > 0) {
- --pos;
- ++it;
- if (it == e) {
- return stack::push(L_, lua_nil);
- }
- }
- return get_associative(is_associative(), L_, it);
- }
- static detail::error_result at_category(std::random_access_iterator_tag, lua_State* L_, T& self, std::ptrdiff_t pos) {
- std::ptrdiff_t len = static_cast<std::ptrdiff_t>(size_start(L_, self));
- pos += deferred_uc::index_adjustment(L_, self);
- if (pos < 0 || pos >= len) {
- return stack::push(L_, lua_nil);
- }
- auto it = std::next(deferred_uc::begin(L_, self), pos);
- return get_associative(is_associative(), L_, it);
- }
- static detail::error_result at_start(lua_State* L_, T& self, std::ptrdiff_t pos) {
- return at_category(iterator_category(), L_, self, pos);
- }
- template <typename Iter>
- static detail::error_result get_associative(std::true_type, lua_State* L_, Iter& it) {
- decltype(auto) v = *it;
- return stack::stack_detail::push_reference<push_type>(L_, detail::deref_move_only(v.second));
- }
- template <typename Iter>
- static detail::error_result get_associative(std::false_type, lua_State* L_, Iter& it) {
- return stack::stack_detail::push_reference<push_type>(L_, detail::deref_move_only(*it));
- }
- static detail::error_result get_category(std::input_iterator_tag, lua_State* L_, T& self, K& key) {
- key = static_cast<K>(key + deferred_uc::index_adjustment(L_, self));
- if (key < 0) {
- return stack::push(L_, lua_nil);
- }
- auto it = deferred_uc::begin(L_, self);
- auto e = deferred_uc::end(L_, self);
- if (it == e) {
- return stack::push(L_, lua_nil);
- }
- while (key > 0) {
- --key;
- ++it;
- if (it == e) {
- return stack::push(L_, lua_nil);
- }
- }
- return get_associative(is_associative(), L_, it);
- }
- static detail::error_result get_category(std::random_access_iterator_tag, lua_State* L_, T& self, K& key) {
- std::ptrdiff_t len = static_cast<std::ptrdiff_t>(size_start(L_, self));
- key = static_cast<K>(static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
- if (key < 0 || key >= len) {
- return stack::push(L_, lua_nil);
- }
- auto it = std::next(deferred_uc::begin(L_, self), key);
- return get_associative(is_associative(), L_, it);
- }
- static detail::error_result get_it(std::true_type, lua_State* L_, T& self, K& key) {
- return get_category(iterator_category(), L_, self, key);
- }
- static detail::error_result get_comparative(std::true_type, lua_State* L_, T& self, K& key) {
- auto fx = [&](const value_type& r) -> bool { return key == get_key(is_associative(), r); };
- auto e = deferred_uc::end(L_, self);
- auto it = std::find_if(deferred_uc::begin(L_, self), e, std::ref(fx));
- if (it == e) {
- return stack::push(L_, lua_nil);
- }
- return get_associative(is_associative(), L_, it);
- }
- static detail::error_result get_comparative(std::false_type, lua_State*, T&, K&) {
- return detail::error_result("cannot get this key on '%s': no suitable way to increment iterator and compare to key value '%s'",
- detail::demangle<T>().data(),
- detail::demangle<K>().data());
- }
- static detail::error_result get_it(std::false_type, lua_State* L_, T& self, K& key) {
- return get_comparative(meta::supports_op_equal<K, key_type>(), L_, self, key);
- }
- static detail::error_result set_associative(std::true_type, iterator& it, stack_object value) {
- decltype(auto) v = *it;
- v.second = value.as<V>();
- return {};
- }
- static detail::error_result set_associative(std::false_type, iterator& it, stack_object value) {
- decltype(auto) v = *it;
- v = value.as<V>();
- return {};
- }
- static detail::error_result set_writable(std::true_type, lua_State*, T&, iterator& it, stack_object value) {
- return set_associative(is_associative(), it, std::move(value));
- }
- static detail::error_result set_writable(std::false_type, lua_State*, T&, iterator&, stack_object) {
- return detail::error_result(
- "cannot perform a 'set': '%s's iterator reference is not writable (non-copy-assignable or const)", detail::demangle<T>().data());
- }
- static detail::error_result set_category(std::input_iterator_tag, lua_State* L_, T& self, stack_object okey, stack_object value) {
- decltype(auto) key = okey.as<K>();
- key = static_cast<K>(static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
- auto e = deferred_uc::end(L_, self);
- auto it = deferred_uc::begin(L_, self);
- auto backit = it;
- for (; key > 0 && it != e; --key, ++it) {
- backit = it;
- }
- if (it == e) {
- if (key == 0) {
- return add_copyable(is_copyable(), L_, self, std::move(value), meta::has_insert_after<T>::value ? backit : it);
- }
- return detail::error_result("out of bounds (too big) for set on '%s'", detail::demangle<T>().c_str());
- }
- return set_writable(is_writable(), L_, self, it, std::move(value));
- }
- static detail::error_result set_category(std::random_access_iterator_tag, lua_State* L_, T& self, stack_object okey, stack_object value) {
- decltype(auto) key = okey.as<K>();
- key = static_cast<K>(static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
- if (key < 0) {
- return detail::error_result("sol: out of bounds (too small) for set on '%s'", detail::demangle<T>().c_str());
- }
- std::ptrdiff_t len = static_cast<std::ptrdiff_t>(size_start(L_, self));
- if (key == len) {
- return add_copyable(is_copyable(), L_, self, std::move(value));
- }
- else if (key >= len) {
- return detail::error_result("sol: out of bounds (too big) for set on '%s'", detail::demangle<T>().c_str());
- }
- auto it = std::next(deferred_uc::begin(L_, self), key);
- return set_writable(is_writable(), L_, self, it, std::move(value));
- }
- static detail::error_result set_comparative(std::true_type, lua_State* L_, T& self, stack_object okey, stack_object value) {
- decltype(auto) key = okey.as<K>();
- if (!is_writable::value) {
- return detail::error_result(
- "cannot perform a 'set': '%s's iterator reference is not writable (non-copy-assignable or const)", detail::demangle<T>().data());
- }
- auto fx = [&](const value_type& r) -> bool { return key == get_key(is_associative(), r); };
- auto e = deferred_uc::end(L_, self);
- auto it = std::find_if(deferred_uc::begin(L_, self), e, std::ref(fx));
- if (it == e) {
- return {};
- }
- return set_writable(is_writable(), L_, self, it, std::move(value));
- }
- static detail::error_result set_comparative(std::false_type, lua_State*, T&, stack_object, stack_object) {
- return detail::error_result("cannot set this value on '%s': no suitable way to increment iterator or compare to '%s' key",
- detail::demangle<T>().data(),
- detail::demangle<K>().data());
- }
- template <typename Iter>
- static detail::error_result set_associative_insert(std::true_type, lua_State*, T& self, Iter& it, K& key, stack_object value) {
- if constexpr (meta::has_insert_with_iterator<T>::value) {
- self.insert(it, value_type(key, value.as<V>()));
- return {};
- }
- else if constexpr (meta::has_insert<T>::value) {
- self.insert(value_type(key, value.as<V>()));
- return {};
- }
- else {
- (void)self;
- (void)it;
- (void)key;
- return detail::error_result(
- "cannot call 'set' on '%s': there is no 'insert' function on this associative type", detail::demangle<T>().c_str());
- }
- }
- template <typename Iter>
- static detail::error_result set_associative_insert(std::false_type, lua_State*, T& self, Iter& it, K& key, stack_object) {
- if constexpr (meta::has_insert_with_iterator<T>::value) {
- self.insert(it, key);
- return {};
- }
- else if constexpr (meta::has_insert<T>::value) {
- self.insert(key);
- return {};
- }
- else {
- (void)self;
- (void)it;
- (void)key;
- return detail::error_result(
- "cannot call 'set' on '%s': there is no 'insert' function on this non-associative type", detail::demangle<T>().c_str());
- }
- }
- static detail::error_result set_associative_find(std::true_type, lua_State* L_, T& self, stack_object okey, stack_object value) {
- decltype(auto) key = okey.as<K>();
- auto it = self.find(key);
- if (it == deferred_uc::end(L_, self)) {
- return set_associative_insert(is_associative(), L_, self, it, key, std::move(value));
- }
- return set_writable(is_writable(), L_, self, it, std::move(value));
- }
- static detail::error_result set_associative_find(std::false_type, lua_State* L_, T& self, stack_object key, stack_object value) {
- return set_comparative(meta::supports_op_equal<K, key_type>(), L_, self, std::move(key), std::move(value));
- }
- static detail::error_result set_it(std::true_type, lua_State* L_, T& self, stack_object key, stack_object value) {
- return set_category(iterator_category(), L_, self, std::move(key), std::move(value));
- }
- static detail::error_result set_it(std::false_type, lua_State* L_, T& self, stack_object key, stack_object value) {
- return set_associative_find(meta::all<has_find<T>, meta::any<is_associative, is_lookup>>(), L_, self, std::move(key), std::move(value));
- }
- template <bool idx_of = false>
- static detail::error_result find_has_associative_lookup(std::true_type, lua_State* L_, T& self) {
- if constexpr (!is_ordered::value && idx_of) {
- (void)L_;
- (void)self;
- return detail::error_result("cannot perform an 'index_of': '%s's is not an ordered container", detail::demangle<T>().data());
- }
- else {
- decltype(auto) key = stack::unqualified_get<K>(L_, 2);
- auto it = self.find(key);
- if (it == deferred_uc::end(L_, self)) {
- return stack::push(L_, lua_nil);
- }
- if constexpr (idx_of) {
- auto dist = std::distance(deferred_uc::begin(L_, self), it);
- dist -= deferred_uc::index_adjustment(L_, self);
- return stack::push(L_, dist);
- }
- else {
- return get_associative(is_associative(), L_, it);
- }
- }
- }
- template <bool idx_of = false>
- static detail::error_result find_has_associative_lookup(std::false_type, lua_State* L_, T& self) {
- if constexpr (!is_ordered::value && idx_of) {
- (void)L_;
- (void)self;
- return detail::error_result("cannot perform an 'index_of': '%s's is not an ordered container", detail::demangle<T>().data());
- }
- else {
- decltype(auto) value = stack::unqualified_get<V>(L_, 2);
- auto it = self.find(value);
- if (it == deferred_uc::end(L_, self)) {
- return stack::push(L_, lua_nil);
- }
- if constexpr (idx_of) {
- auto dist = std::distance(deferred_uc::begin(L_, self), it);
- dist -= deferred_uc::index_adjustment(L_, self);
- return stack::push(L_, dist);
- }
- else {
- return get_associative(is_associative(), L_, it);
- }
- }
- }
- template <bool idx_of = false>
- static detail::error_result find_has(std::true_type, lua_State* L_, T& self) {
- return find_has_associative_lookup<idx_of>(meta::any<is_lookup, is_associative>(), L_, self);
- }
- template <typename Iter>
- static detail::error_result find_associative_lookup(std::true_type, lua_State* L_, T&, Iter& it, std::size_t) {
- return get_associative(is_associative(), L_, it);
- }
- template <typename Iter>
- static detail::error_result find_associative_lookup(std::false_type, lua_State* L_, T& self, Iter&, std::size_t idx) {
- idx = static_cast<std::size_t>(static_cast<std::ptrdiff_t>(idx) - deferred_uc::index_adjustment(L_, self));
- return stack::push(L_, idx);
- }
- template <bool = false>
- static detail::error_result find_comparative(std::false_type, lua_State*, T&) {
- return detail::error_result("cannot call 'find' on '%s': there is no 'find' function and the value_type is not equality comparable",
- detail::demangle<T>().c_str());
- }
- template <bool idx_of = false>
- static detail::error_result find_comparative(std::true_type, lua_State* L_, T& self) {
- decltype(auto) value = stack::unqualified_get<V>(L_, 2);
- auto it = deferred_uc::begin(L_, self);
- auto e = deferred_uc::end(L_, self);
- std::size_t idx = 0;
- for (;; ++it, ++idx) {
- if (it == e) {
- return stack::push(L_, lua_nil);
- }
- if (value == get_value(is_associative(), *it)) {
- break;
- }
- }
- return find_associative_lookup(meta::all<meta::boolean<!idx_of>, meta::any<is_lookup, is_associative>>(), L_, self, it, idx);
- }
- template <bool idx_of = false>
- static detail::error_result find_has(std::false_type, lua_State* L_, T& self) {
- return find_comparative<idx_of>(meta::supports_op_equal<V>(), L_, self);
- }
- template <typename Iter>
- static detail::error_result add_insert_after(std::false_type, lua_State* L_, T& self, stack_object value, Iter&) {
- return add_insert_after(std::false_type(), L_, self, value);
- }
- static detail::error_result add_insert_after(std::false_type, lua_State*, T&, stack_object) {
- return detail::error_result("cannot call 'add' on type '%s': no suitable insert/push_back C++ functions", detail::demangle<T>().data());
- }
- template <typename Iter>
- static detail::error_result add_insert_after(std::true_type, lua_State*, T& self, stack_object value, Iter& pos) {
- self.insert_after(pos, value.as<V>());
- return {};
- }
- static detail::error_result add_insert_after(std::true_type, lua_State* L_, T& self, stack_object value) {
- auto backit = self.before_begin();
- {
- auto e = deferred_uc::end(L_, self);
- for (auto it = deferred_uc::begin(L_, self); it != e; ++backit, ++it) { }
- }
- return add_insert_after(std::true_type(), L_, self, value, backit);
- }
- template <typename Iter>
- static detail::error_result add_insert(std::true_type, lua_State*, T& self, stack_object value, Iter& pos) {
- self.insert(pos, value.as<V>());
- return {};
- }
- static detail::error_result add_insert(std::true_type, lua_State* L_, T& self, stack_object value) {
- auto pos = deferred_uc::end(L_, self);
- return add_insert(std::true_type(), L_, self, value, pos);
- }
- template <typename Iter>
- static detail::error_result add_insert(std::false_type, lua_State* L_, T& self, stack_object value, Iter& pos) {
- return add_insert_after(meta::has_insert_after<T>(), L_, self, std::move(value), pos);
- }
- static detail::error_result add_insert(std::false_type, lua_State* L_, T& self, stack_object value) {
- return add_insert_after(meta::has_insert_after<T>(), L_, self, std::move(value));
- }
- template <typename Iter>
- static detail::error_result add_push_back(std::true_type, lua_State*, T& self, stack_object value, Iter&) {
- self.push_back(value.as<V>());
- return {};
- }
- static detail::error_result add_push_back(std::true_type, lua_State*, T& self, stack_object value) {
- self.push_back(value.as<V>());
- return {};
- }
- template <typename Iter>
- static detail::error_result add_push_back(std::false_type, lua_State* L_, T& self, stack_object value, Iter& pos) {
- return add_insert(
- std::integral_constant < bool, meta::has_insert<T>::value || meta::has_insert_with_iterator<T>::value > (), L_, self, value, pos);
- }
- static detail::error_result add_push_back(std::false_type, lua_State* L_, T& self, stack_object value) {
- return add_insert(
- std::integral_constant < bool, meta::has_insert<T>::value || meta::has_insert_with_iterator<T>::value > (), L_, self, value);
- }
- template <typename Iter>
- static detail::error_result add_associative(std::true_type, lua_State* L_, T& self, stack_object key, Iter& pos) {
- if constexpr (meta::has_insert_with_iterator<T>::value) {
- self.insert(pos, value_type(key.as<K>(), stack::unqualified_get<V>(L_, 3)));
- return {};
- }
- else if constexpr (meta::has_insert<T>::value) {
- self.insert(value_type(key.as<K>(), stack::unqualified_get<V>(L_, 3)));
- return {};
- }
- else {
- (void)L_;
- (void)self;
- (void)key;
- (void)pos;
- return detail::error_result(
- "cannot call 'insert' on '%s': there is no 'insert' function on this associative type", detail::demangle<T>().c_str());
- }
- }
- static detail::error_result add_associative(std::true_type, lua_State* L_, T& self, stack_object key) {
- auto pos = deferred_uc::end(L_, self);
- return add_associative(std::true_type(), L_, self, std::move(key), pos);
- }
- template <typename Iter>
- static detail::error_result add_associative(std::false_type, lua_State* L_, T& self, stack_object value, Iter& pos) {
- return add_push_back(meta::has_push_back<T>(), L_, self, value, pos);
- }
- static detail::error_result add_associative(std::false_type, lua_State* L_, T& self, stack_object value) {
- return add_push_back(meta::has_push_back<T>(), L_, self, value);
- }
- template <typename Iter>
- static detail::error_result add_copyable(std::true_type, lua_State* L_, T& self, stack_object value, Iter& pos) {
- return add_associative(is_associative(), L_, self, std::move(value), pos);
- }
- static detail::error_result add_copyable(std::true_type, lua_State* L_, T& self, stack_object value) {
- return add_associative(is_associative(), L_, self, value);
- }
- template <typename Iter>
- static detail::error_result add_copyable(std::false_type, lua_State* L_, T& self, stack_object value, Iter&) {
- return add_copyable(std::false_type(), L_, self, std::move(value));
- }
- static detail::error_result add_copyable(std::false_type, lua_State*, T&, stack_object) {
- return detail::error_result("cannot call 'add' on '%s': value_type is non-copyable", detail::demangle<T>().data());
- }
- static detail::error_result insert_lookup(std::true_type, lua_State* L_, T& self, stack_object, stack_object value) {
- // TODO: should we warn or error about someone calling insert on an ordered / lookup container with no associativity?
- return add_copyable(std::true_type(), L_, self, std::move(value));
- }
- static detail::error_result insert_lookup(std::false_type, lua_State* L_, T& self, stack_object where, stack_object value) {
- auto it = deferred_uc::begin(L_, self);
- auto key = where.as<K>();
- key = static_cast<K>(static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
- std::advance(it, key);
- self.insert(it, value.as<V>());
- return {};
- }
- static detail::error_result insert_after_has(std::true_type, lua_State* L_, T& self, stack_object where, stack_object value) {
- auto key = where.as<K>();
- auto backit = self.before_begin();
- {
- key = static_cast<K>(static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
- auto e = deferred_uc::end(L_, self);
- for (auto it = deferred_uc::begin(L_, self); key > 0; ++backit, ++it, --key) {
- if (backit == e) {
- return detail::error_result("sol: out of bounds (too big) for set on '%s'", detail::demangle<T>().c_str());
- }
- }
- }
- self.insert_after(backit, value.as<V>());
- return {};
- }
- static detail::error_result insert_after_has(std::false_type, lua_State*, T&, stack_object, stack_object) {
- return detail::error_result(
- "cannot call 'insert' on '%s': no suitable or similar functionality detected on this container", detail::demangle<T>().data());
- }
- static detail::error_result insert_has(std::true_type, lua_State* L_, T& self, stack_object key, stack_object value) {
- return insert_lookup(meta::any<is_associative, is_lookup>(), L_, self, std::move(key), std::move(value));
- }
- static detail::error_result insert_has(std::false_type, lua_State* L_, T& self, stack_object where, stack_object value) {
- return insert_after_has(meta::has_insert_after<T>(), L_, self, where, value);
- }
- static detail::error_result insert_copyable(std::true_type, lua_State* L_, T& self, stack_object key, stack_object value) {
- return insert_has(std::integral_constant < bool,
- meta::has_insert<T>::value || meta::has_insert_with_iterator<T>::value > (),
- L_,
- self,
- std::move(key),
- std::move(value));
- }
- static detail::error_result insert_copyable(std::false_type, lua_State*, T&, stack_object, stack_object) {
- return detail::error_result("cannot call 'insert' on '%s': value_type is non-copyable", detail::demangle<T>().data());
- }
- static detail::error_result erase_integral(std::true_type, lua_State* L_, T& self, K& key) {
- auto it = deferred_uc::begin(L_, self);
- key = (static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
- std::advance(it, key);
- self.erase(it);
- return {};
- }
- static detail::error_result erase_integral(std::false_type, lua_State* L_, T& self, const K& key) {
- auto fx = [&](const value_type& r) -> bool { return key == r; };
- auto e = deferred_uc::end(L_, self);
- auto it = std::find_if(deferred_uc::begin(L_, self), e, std::ref(fx));
- if (it == e) {
- return {};
- }
- self.erase(it);
- return {};
- }
- static detail::error_result erase_associative_lookup(std::true_type, lua_State*, T& self, const K& key) {
- self.erase(key);
- return {};
- }
- static detail::error_result erase_associative_lookup(std::false_type, lua_State* L_, T& self, K& key) {
- return erase_integral(std::is_integral<K>(), L_, self, key);
- }
- static detail::error_result erase_after_has(std::true_type, lua_State* L_, T& self, K& key) {
- auto backit = self.before_begin();
- {
- key = static_cast<K>(static_cast<std::ptrdiff_t>(key) + deferred_uc::index_adjustment(L_, self));
- auto e = deferred_uc::end(L_, self);
- for (auto it = deferred_uc::begin(L_, self); key > 0; ++backit, ++it, --key) {
- if (backit == e) {
- return detail::error_result("sol: out of bounds for erase on '%s'", detail::demangle<T>().c_str());
- }
- }
- }
- self.erase_after(backit);
- return {};
- }
- static detail::error_result erase_after_has(std::false_type, lua_State*, T&, const K&) {
- return detail::error_result("sol: cannot call erase on '%s'", detail::demangle<T>().c_str());
- }
- static detail::error_result erase_key_has(std::true_type, lua_State* L_, T& self, K& key) {
- return erase_associative_lookup(meta::any<is_associative, is_lookup>(), L_, self, key);
- }
- static detail::error_result erase_key_has(std::false_type, lua_State* L_, T& self, K& key) {
- return erase_after_has(has_erase_after<T>(), L_, self, key);
- }
- static detail::error_result erase_has(std::true_type, lua_State* L_, T& self, K& key) {
- return erase_associative_lookup(meta::any<is_associative, is_lookup>(), L_, self, key);
- }
- static detail::error_result erase_has(std::false_type, lua_State* L_, T& self, K& key) {
- return erase_key_has(has_erase_key<T>(), L_, self, key);
- }
- static auto size_has(std::false_type, lua_State* L_, T& self) {
- return std::distance(deferred_uc::begin(L_, self), deferred_uc::end(L_, self));
- }
- static auto size_has(std::true_type, lua_State*, T& self) {
- return self.size();
- }
- static void clear_has(std::true_type, lua_State*, T& self) {
- self.clear();
- }
- static void clear_has(std::false_type, lua_State* L_, T&) {
- luaL_error(L_, "sol: cannot call clear on '%s'", detail::demangle<T>().c_str());
- }
- static bool empty_has(std::true_type, lua_State*, T& self) {
- return self.empty();
- }
- static bool empty_has(std::false_type, lua_State* L_, T& self) {
- return deferred_uc::begin(L_, self) == deferred_uc::end(L_, self);
- }
- static detail::error_result get_associative_find(std::true_type, lua_State* L_, T& self, K& key) {
- auto it = self.find(key);
- if (it == deferred_uc::end(L_, self)) {
- stack::push(L_, lua_nil);
- return {};
- }
- return get_associative(std::true_type(), L_, it);
- }
- static detail::error_result get_associative_find(std::false_type, lua_State* L_, T& self, K& key) {
- return get_it(is_linear_integral(), L_, self, key);
- }
- static detail::error_result get_start(lua_State* L_, T& self, K& key) {
- return get_associative_find(std::integral_constant < bool, is_associative::value&& has_find<T>::value > (), L_, self, key);
- }
- static detail::error_result set_start(lua_State* L_, T& self, stack_object key, stack_object value) {
- return set_it(is_linear_integral(), L_, self, std::move(key), std::move(value));
- }
- static std::size_t size_start(lua_State* L_, T& self) {
- return static_cast<std::size_t>(size_has(meta::has_size<T>(), L_, self));
- }
- static void clear_start(lua_State* L_, T& self) {
- clear_has(has_clear<T>(), L_, self);
- }
- static bool empty_start(lua_State* L_, T& self) {
- return empty_has(has_empty<T>(), L_, self);
- }
- static detail::error_result erase_start(lua_State* L_, T& self, K& key) {
- return erase_has(has_erase<T>(), L_, self, key);
- }
- template <bool ip>
- static int next_associative(std::true_type, lua_State* L_) {
- iter& i = stack::unqualified_get<user<iter>>(L_, 1);
- auto& it = i.it;
- auto& end = i.end;
- if (it == end) {
- return stack::push(L_, lua_nil);
- }
- int p;
- if constexpr (ip) {
- ++i.index;
- p = stack::push_reference(L_, i.index);
- }
- else {
- p = stack::push_reference(L_, it->first);
- }
- p += stack::stack_detail::push_reference<push_type>(L_, detail::deref_move_only(it->second));
- std::advance(it, 1);
- return p;
- }
- template <bool>
- static int next_associative(std::false_type, lua_State* L_) {
- iter& i = stack::unqualified_get<user<iter>>(L_, 1);
- auto& it = i.it();
- auto& end = i.sen();
- next_K k = stack::unqualified_get<next_K>(L_, 2);
- if (it == end) {
- return stack::push(L_, lua_nil);
- }
- int p;
- if constexpr (std::is_integral_v<next_K>) {
- p = stack::push_reference(L_, k + 1);
- }
- else {
- p = stack::stack_detail::push_reference(L_, k + 1);
- }
- p += stack::stack_detail::push_reference<push_type>(L_, detail::deref_move_only(*it));
- std::advance(it, 1);
- return p;
- }
- template <bool ip>
- static int next_iter(lua_State* L_) {
- typedef meta::any<is_associative, meta::all<is_lookup, meta::neg<is_matched_lookup>>> is_assoc;
- return next_associative<ip>(is_assoc(), L_);
- }
- template <bool ip>
- static int pairs_associative(std::true_type, lua_State* L_) {
- auto& src = get_src(L_);
- stack::push(L_, next_iter<ip>);
- stack::push<user<iter>>(L_, L_, 1, deferred_uc::begin(L_, src), deferred_uc::begin(L_, src));
- stack::push(L_, lua_nil);
- return 3;
- }
- template <bool ip>
- static int pairs_associative(std::false_type, lua_State* L_) {
- auto& src = get_src(L_);
- stack::push(L_, next_iter<ip>);
- stack::push<user<iter>>(L_, L_, 1, deferred_uc::begin(L_, src), deferred_uc::end(L_, src));
- stack::push(L_, 0);
- return 3;
- }
- public:
- static int at(lua_State* L_) {
- auto& self = get_src(L_);
- detail::error_result er;
- {
- std::ptrdiff_t pos = stack::unqualified_get<std::ptrdiff_t>(L_, 2);
- er = at_start(L_, self, pos);
- }
- return handle_errors(L_, er);
- }
- static int get(lua_State* L_) {
- auto& self = get_src(L_);
- detail::error_result er;
- {
- decltype(auto) key = stack::unqualified_get<K>(L_);
- er = get_start(L_, self, key);
- }
- return handle_errors(L_, er);
- }
- static int index_get(lua_State* L_) {
- return get(L_);
- }
- static int set(lua_State* L_) {
- stack_object value = stack_object(L_, raw_index(3));
- if constexpr (is_linear_integral::value) {
- // for non-associative containers,
- // erasure only happens if it is the
- // last index in the container
- auto key = stack::get<K>(L_, 2);
- auto self_size = deferred_uc::size(L_);
- if (key == static_cast<K>(self_size)) {
- if (type_of(L_, 3) == type::lua_nil) {
- return erase(L_);
- }
- }
- }
- else {
- if (type_of(L_, 3) == type::lua_nil) {
- return erase(L_);
- }
- }
- auto& self = get_src(L_);
- detail::error_result er = set_start(L_, self, stack_object(L_, raw_index(2)), std::move(value));
- return handle_errors(L_, er);
- }
- static int index_set(lua_State* L_) {
- return set(L_);
- }
- static int add(lua_State* L_) {
- auto& self = get_src(L_);
- detail::error_result er = add_copyable(is_copyable(), L_, self, stack_object(L_, raw_index(2)));
- return handle_errors(L_, er);
- }
- static int insert(lua_State* L_) {
- auto& self = get_src(L_);
- detail::error_result er = insert_copyable(is_copyable(), L_, self, stack_object(L_, raw_index(2)), stack_object(L_, raw_index(3)));
- return handle_errors(L_, er);
- }
- static int find(lua_State* L_) {
- auto& self = get_src(L_);
- detail::error_result er = find_has(has_find<T>(), L_, self);
- return handle_errors(L_, er);
- }
- static int index_of(lua_State* L_) {
- auto& self = get_src(L_);
- detail::error_result er = find_has<true>(has_find<T>(), L_, self);
- return handle_errors(L_, er);
- }
- static iterator begin(lua_State*, T& self) {
- if constexpr (meta::has_begin_end_v<T>) {
- return self.begin();
- }
- else {
- using std::begin;
- return begin(self);
- }
- }
- static sentinel end(lua_State*, T& self) {
- if constexpr (meta::has_begin_end_v<T>) {
- return self.end();
- }
- else {
- using std::end;
- return end(self);
- }
- }
- static int size(lua_State* L_) {
- auto& self = get_src(L_);
- std::size_t r = size_start(L_, self);
- return stack::push(L_, r);
- }
- static int clear(lua_State* L_) {
- auto& self = get_src(L_);
- clear_start(L_, self);
- return 0;
- }
- static int erase(lua_State* L_) {
- auto& self = get_src(L_);
- detail::error_result er;
- {
- decltype(auto) key = stack::unqualified_get<K>(L_, 2);
- er = erase_start(L_, self, key);
- }
- return handle_errors(L_, er);
- }
- static int empty(lua_State* L_) {
- auto& self = get_src(L_);
- return stack::push(L_, empty_start(L_, self));
- }
- static std::ptrdiff_t index_adjustment(lua_State*, T&) {
- return static_cast<std::ptrdiff_t>((SOL_CONTAINER_START_INDEX_I_) == 0 ? 0 : -(SOL_CONTAINER_START_INDEX_I_));
- }
- static int pairs(lua_State* L_) {
- typedef meta::any<is_associative, meta::all<is_lookup, meta::neg<is_matched_lookup>>> is_assoc;
- return pairs_associative<false>(is_assoc(), L_);
- }
- static int ipairs(lua_State* L_) {
- typedef meta::any<is_associative, meta::all<is_lookup, meta::neg<is_matched_lookup>>> is_assoc;
- return pairs_associative<true>(is_assoc(), L_);
- }
- static int next(lua_State* L_) {
- return stack::push(L_, next_iter<false>);
- }
- };
- template <typename X>
- struct usertype_container_default<X, std::enable_if_t<std::is_array<std::remove_pointer_t<meta::unwrap_unqualified_t<X>>>::value>> {
- private:
- typedef std::remove_pointer_t<meta::unwrap_unqualified_t<X>> T;
- typedef usertype_container<X> deferred_uc;
- public:
- typedef std::remove_extent_t<T> value_type;
- typedef value_type* iterator;
- typedef iterator sentinel;
- private:
- struct iter : detail::ebco<iterator, 0>, detail::ebco<sentinel, 1> {
- using it_base = detail::ebco<iterator, 0>;
- using sen_base = detail::ebco<sentinel, 1>;
- reference keep_alive;
-
- iter(lua_State* L_, int stack_index_, iterator it_, sentinel sen_) noexcept
- : it_base(std::move(it_)), sen_base(std::move(sen_)), keep_alive(sol::main_thread(L_, L_), stack_index_) {
- }
- iterator& it() noexcept {
- return it_base::value();
- }
- const iterator& it() const noexcept {
- return it_base::value();
- }
- sentinel& sen() noexcept {
- return sen_base::value();
- }
- const sentinel& sen() const noexcept {
- return sen_base::value();
- }
- };
- static auto& get_src(lua_State* L_) {
- auto p = stack::unqualified_check_get<T*>(L_, 1);
- #if SOL_IS_ON(SOL_SAFE_USERTYPE)
- if (!p) {
- luaL_error(L_,
- "sol: 'self' is not of type '%s' (pass 'self' as first argument with ':' or call on proper type)",
- detail::demangle<T>().c_str());
- }
- if (p.value() == nullptr) {
- luaL_error(
- L_, "sol: 'self' argument is nil (pass 'self' as first argument with ':' or call on a '%s' type)", detail::demangle<T>().c_str());
- }
- #endif // Safe getting with error
- return *p.value();
- }
- static int find(std::true_type, lua_State* L_) {
- T& self = get_src(L_);
- decltype(auto) value = stack::unqualified_get<value_type>(L_, 2);
- std::size_t N = std::extent<T>::value;
- for (std::size_t idx = 0; idx < N; ++idx) {
- using v_t = std::add_const_t<decltype(self[idx])>;
- v_t v = self[idx];
- if (v == value) {
- idx = static_cast<std::size_t>(static_cast<std::ptrdiff_t>(idx) - deferred_uc::index_adjustment(L_, self));
- return stack::push(L_, idx);
- }
- }
- return stack::push(L_, lua_nil);
- }
- static int find(std::false_type, lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'find' on '%s': no supported comparison operator for the value type", detail::demangle<T>().c_str());
- }
- static int next_iter(lua_State* L_) {
- iter& i = stack::unqualified_get<user<iter>>(L_, 1);
- auto& it = i.it();
- auto& end = i.sen();
- std::size_t k = stack::unqualified_get<std::size_t>(L_, 2);
- if (it == end) {
- return 0;
- }
- int p;
- p = stack::push(L_, k + 1);
- p += stack::push_reference(L_, detail::deref_move_only(*it));
- std::advance(it, 1);
- return p;
- }
- public:
- static int clear(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'clear' on type '%s': cannot remove all items from a fixed array", detail::demangle<T>().c_str());
- }
- static int erase(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'erase' on type '%s': cannot remove an item from fixed arrays", detail::demangle<T>().c_str());
- }
- static int add(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'add' on type '%s': cannot add to fixed arrays", detail::demangle<T>().c_str());
- }
- static int insert(lua_State* L_) {
- return luaL_error(L_, "sol: cannot call 'insert' on type '%s': cannot insert new entries into fixed arrays", detail::demangle<T>().c_str());
- }
- static int at(lua_State* L_) {
- return get(L_);
- }
- static int get(lua_State* L_) {
- T& self = get_src(L_);
- std::ptrdiff_t idx = stack::unqualified_get<std::ptrdiff_t>(L_, 2);
- idx += deferred_uc::index_adjustment(L_, self);
- if (idx >= static_cast<std::ptrdiff_t>(std::extent<T>::value) || idx < 0) {
- return stack::push(L_, lua_nil);
- }
- return stack::push_reference(L_, detail::deref_move_only(self[idx]));
- }
- static int index_get(lua_State* L_) {
- return get(L_);
- }
- static int set(lua_State* L_) {
- T& self = get_src(L_);
- std::ptrdiff_t idx = stack::unqualified_get<std::ptrdiff_t>(L_, 2);
- idx += deferred_uc::index_adjustment(L_, self);
- if (idx >= static_cast<std::ptrdiff_t>(std::extent<T>::value)) {
- return luaL_error(L_, "sol: index out of bounds (too big) for set on '%s'", detail::demangle<T>().c_str());
- }
- if (idx < 0) {
- return luaL_error(L_, "sol: index out of bounds (too small) for set on '%s'", detail::demangle<T>().c_str());
- }
- self[idx] = stack::unqualified_get<value_type>(L_, 3);
- return 0;
- }
- static int index_set(lua_State* L_) {
- return set(L_);
- }
- static int index_of(lua_State* L_) {
- return find(L_);
- }
- static int find(lua_State* L_) {
- return find(meta::supports_op_equal<value_type, value_type>(), L_);
- }
- static int size(lua_State* L_) {
- return stack::push(L_, std::extent<T>::value);
- }
- static int empty(lua_State* L_) {
- return stack::push(L_, std::extent<T>::value > 0);
- }
- static int pairs(lua_State* L_) {
- auto& src = get_src(L_);
- stack::push(L_, next_iter);
- stack::push<user<iter>>(L_, L_, 1, deferred_uc::begin(L_, src), deferred_uc::end(L_, src));
- stack::push(L_, 0);
- return 3;
- }
- static int ipairs(lua_State* L_) {
- return pairs(L_);
- }
- static int next(lua_State* L_) {
- return stack::push(L_, next_iter);
- }
- static std::ptrdiff_t index_adjustment(lua_State*, T&) {
- return (SOL_CONTAINER_START_INDEX_I_) == 0 ? 0 : -(SOL_CONTAINER_START_INDEX_I_);
- }
- static iterator begin(lua_State*, T& self) {
- return std::addressof(self[0]);
- }
- static sentinel end(lua_State*, T& self) {
- return std::addressof(self[0]) + std::extent<T>::value;
- }
- };
- template <typename X>
- struct usertype_container_default<usertype_container<X>> : usertype_container_default<X> { };
- } // namespace container_detail
- template <typename T>
- struct usertype_container : container_detail::usertype_container_default<T> { };
- } // namespace sol
- #endif // SOL_USERTYPE_CONTAINER_HPP
|