mirror of
https://github.com/ThePhD/sol2.git
synced 2024-03-22 13:10:44 +08:00
1486 lines
48 KiB
C++
1486 lines
48 KiB
C++
// sol2
|
|
|
|
// The MIT License (MIT)
|
|
|
|
// Copyright (c) 2013-2018 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_CONTAINER_TRAITS_HPP
|
|
#define SOL_CONTAINER_TRAITS_HPP
|
|
|
|
#include "traits.hpp"
|
|
#include "stack.hpp"
|
|
#include <unordered_map>
|
|
|
|
namespace sol {
|
|
|
|
template <typename T>
|
|
struct container_traits;
|
|
|
|
template <typename T>
|
|
struct as_container_t {
|
|
T source;
|
|
|
|
as_container_t(T value)
|
|
: source(std::move(value)) {
|
|
}
|
|
|
|
operator std::add_rvalue_reference_t<T>() {
|
|
return std::move(source);
|
|
}
|
|
|
|
operator std::add_lvalue_reference_t<std::add_const_t<T>>() const {
|
|
return source;
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
struct as_container_t<T&> {
|
|
std::reference_wrapper<T> source;
|
|
|
|
as_container_t(T& value)
|
|
: source(value) {
|
|
}
|
|
|
|
operator T&() {
|
|
return source;
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
auto as_container(T&& value) {
|
|
return as_container_t<T>(std::forward<T>(value));
|
|
}
|
|
|
|
namespace container_detail {
|
|
|
|
template <typename T>
|
|
struct has_clear_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::clear));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_empty_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::empty));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_erase_after_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(std::declval<C>().erase_after(std::declval<std::add_rvalue_reference_t<typename C::const_iterator>>()))*);
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T, typename = void>
|
|
struct has_find_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(std::declval<C>().find(std::declval<std::add_rvalue_reference_t<typename C::value_type>>()))*);
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_find_test<T, std::enable_if_t<meta::is_lookup<T>::value>> {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(std::declval<C>().find(std::declval<std::add_rvalue_reference_t<typename C::key_type>>()))*);
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_erase_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(std::declval<C>().erase(std::declval<typename C::iterator>()))*);
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_find_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::find));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_insert_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::insert));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_erase_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::erase));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_index_set_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::index_set));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_index_get_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::index_get));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_set_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::set));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_get_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::get));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_at_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::at));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_pairs_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::pairs));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_ipairs_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::ipairs));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_next_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::next));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_add_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::add));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
template <typename T>
|
|
struct has_traits_size_test {
|
|
private:
|
|
typedef std::array<char, 1> one;
|
|
typedef std::array<char, 2> two;
|
|
|
|
template <typename C>
|
|
static one test(decltype(&C::size));
|
|
template <typename C>
|
|
static two test(...);
|
|
|
|
public:
|
|
static const bool value = sizeof(test<T>(0)) == sizeof(char);
|
|
};
|
|
|
|
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_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_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;
|
|
}
|
|
|
|
struct error_result {
|
|
int results;
|
|
const char* fmt;
|
|
std::array<const char*, 4> args;
|
|
|
|
error_result() : results(0), fmt(nullptr) {
|
|
}
|
|
|
|
error_result(int results) : results(results), fmt(nullptr) {
|
|
}
|
|
|
|
error_result(const char* fmt, const char* msg) : results(0), fmt(fmt) {
|
|
args[0] = msg;
|
|
}
|
|
};
|
|
|
|
inline int handle_errors(lua_State* L, const error_result& er) {
|
|
if (er.fmt == nullptr) {
|
|
return er.results;
|
|
}
|
|
return luaL_error(L, er.fmt, er.args[0], er.args[1], er.args[2], er.args[3]);
|
|
}
|
|
|
|
template <typename X, typename = void>
|
|
struct container_traits_default {
|
|
private:
|
|
typedef std::remove_pointer_t<meta::unwrap_unqualified_t<X>> T;
|
|
|
|
public:
|
|
typedef lua_nil_t iterator;
|
|
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 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 iterator 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 container_traits_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:
|
|
typedef std::remove_pointer_t<meta::unwrap_unqualified_t<container_decay_t<X>>> T;
|
|
|
|
private:
|
|
typedef container_traits<X> deferred_traits;
|
|
typedef meta::is_associative<T> is_associative;
|
|
typedef meta::is_lookup<T> is_lookup;
|
|
typedef meta::is_matched_lookup<T> is_matched_lookup;
|
|
typedef typename T::iterator iterator;
|
|
typedef typename T::value_type value_type;
|
|
typedef std::conditional_t<is_matched_lookup::value,
|
|
std::pair<value_type, value_type>,
|
|
std::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 std::conditional_t<is_matched_lookup::value, std::ptrdiff_t, K> next_K;
|
|
typedef decltype(*std::declval<iterator&>()) iterator_return;
|
|
typedef std::conditional_t<is_associative::value || is_matched_lookup::value,
|
|
std::add_lvalue_reference_t<V>,
|
|
std::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 std::conditional_t<is_input_iterator::value,
|
|
V,
|
|
decltype(detail::deref_non_pointer(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 {
|
|
T& source;
|
|
iterator it;
|
|
std::size_t i;
|
|
|
|
iter(T& source, iterator it)
|
|
: source(source), it(std::move(it)), i(0) {
|
|
}
|
|
};
|
|
|
|
static auto& get_src(lua_State* L) {
|
|
#if defined(SOL_SAFE_USERTYPE) && 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 error_result at_category(std::input_iterator_tag, lua_State* L, T& self, std::ptrdiff_t pos) {
|
|
pos += deferred_traits::index_adjustment(L, self);
|
|
if (pos < 0) {
|
|
return stack::push(L, lua_nil);
|
|
}
|
|
auto it = deferred_traits::begin(L, self);
|
|
auto e = deferred_traits::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 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_traits::index_adjustment(L, self);
|
|
if (pos < 0 || pos >= len) {
|
|
return stack::push(L, lua_nil);
|
|
}
|
|
auto it = std::next(deferred_traits::begin(L, self), pos);
|
|
return get_associative(is_associative(), L, it);
|
|
}
|
|
|
|
static error_result at_start(lua_State* L, T& self, std::ptrdiff_t pos) {
|
|
return at_category(iterator_category(), L, self, pos);
|
|
}
|
|
|
|
static error_result get_associative(std::true_type, lua_State* L, iterator& it) {
|
|
auto& v = *it;
|
|
return stack::stack_detail::push_reference<push_type>(L, detail::deref_non_pointer(v.second));
|
|
}
|
|
|
|
static error_result get_associative(std::false_type, lua_State* L, iterator& it) {
|
|
return stack::stack_detail::push_reference<push_type>(L, detail::deref_non_pointer(*it));
|
|
}
|
|
|
|
static error_result get_category(std::input_iterator_tag, lua_State* L, T& self, K& key) {
|
|
key += deferred_traits::index_adjustment(L, self);
|
|
if (key < 0) {
|
|
return stack::push(L, lua_nil);
|
|
}
|
|
auto it = deferred_traits::begin(L, self);
|
|
auto e = deferred_traits::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 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 += deferred_traits::index_adjustment(L, self);
|
|
if (key < 0 || key >= len) {
|
|
return stack::push(L, lua_nil);
|
|
}
|
|
auto it = std::next(deferred_traits::begin(L, self), key);
|
|
return get_associative(is_associative(), L, it);
|
|
}
|
|
|
|
static error_result get_it(std::true_type, lua_State* L, T& self, K& key) {
|
|
return get_category(iterator_category(), L, self, key);
|
|
}
|
|
|
|
static 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_traits::end(L, self);
|
|
auto it = std::find_if(deferred_traits::begin(L, self), e, std::ref(fx));
|
|
if (it == e) {
|
|
return stack::push(L, lua_nil);
|
|
}
|
|
return get_associative(is_associative(), L, it);
|
|
}
|
|
|
|
static error_result get_comparative(std::false_type, lua_State*, T&, K&) {
|
|
return 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 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 error_result set_associative(std::true_type, iterator& it, stack_object value) {
|
|
auto& v = *it;
|
|
v.second = value.as<V>();
|
|
return {};
|
|
}
|
|
|
|
static error_result set_associative(std::false_type, iterator& it, stack_object value) {
|
|
auto& v = *it;
|
|
v = value.as<V>();
|
|
return {};
|
|
}
|
|
|
|
static 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 error_result set_writable(std::false_type, lua_State*, T&, iterator&, stack_object) {
|
|
return error_result("cannot perform a 'set': '%s's iterator reference is not writable (non-copy-assignable or const)", detail::demangle<T>().data());
|
|
}
|
|
|
|
static 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 += deferred_traits::index_adjustment(L, self);
|
|
auto e = deferred_traits::end(L, self);
|
|
auto it = deferred_traits::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 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 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>();
|
|
if (key <= 0) {
|
|
return error_result("sol: out of bounds (too small) for set on '%s'", detail::demangle<T>().c_str());
|
|
}
|
|
key += deferred_traits::index_adjustment(L, self);
|
|
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 error_result("sol: out of bounds (too big) for set on '%s'", detail::demangle<T>().c_str());
|
|
}
|
|
auto it = std::next(deferred_traits::begin(L, self), key);
|
|
return set_writable(is_writable(), L, self, it, std::move(value));
|
|
}
|
|
|
|
static 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 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_traits::end(L, self);
|
|
auto it = std::find_if(deferred_traits::begin(L, self), e, std::ref(fx));
|
|
if (it == e) {
|
|
return {};
|
|
}
|
|
return set_writable(is_writable(), L, self, it, std::move(value));
|
|
}
|
|
|
|
static error_result set_comparative(std::false_type, lua_State*, T&, stack_object, stack_object) {
|
|
return 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());
|
|
}
|
|
|
|
static error_result set_associative_insert(std::true_type, lua_State*, T& self, iterator& it, K& key, stack_object value) {
|
|
self.insert(it, value_type(key, value.as<V>()));
|
|
return {};
|
|
}
|
|
|
|
static error_result set_associative_insert(std::false_type, lua_State*, T& self, iterator& it, K& key, stack_object) {
|
|
self.insert(it, key);
|
|
return {};
|
|
}
|
|
|
|
static 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_traits::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 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 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 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));
|
|
}
|
|
|
|
static error_result find_has_associative_lookup(std::true_type, lua_State* L, T& self) {
|
|
decltype(auto) key = stack::unqualified_get<K>(L, 2);
|
|
auto it = self.find(key);
|
|
if (it == deferred_traits::end(L, self)) {
|
|
return stack::push(L, lua_nil);
|
|
}
|
|
return get_associative(is_associative(), L, it);
|
|
}
|
|
|
|
static error_result find_has_associative_lookup(std::false_type, lua_State* L, T& self) {
|
|
decltype(auto) value = stack::unqualified_get<V>(L, 2);
|
|
auto it = self.find(value);
|
|
if (it == deferred_traits::end(L, self)) {
|
|
return stack::push(L, lua_nil);
|
|
}
|
|
return get_associative(is_associative(), L, it);
|
|
}
|
|
|
|
static error_result find_has(std::true_type, lua_State* L, T& self) {
|
|
return find_has_associative_lookup(meta::any<is_lookup, is_associative>(), L, self);
|
|
}
|
|
|
|
static error_result find_associative_lookup(std::true_type, lua_State* L, iterator& it, std::size_t) {
|
|
return get_associative(is_associative(), L, it);
|
|
}
|
|
|
|
static error_result find_associative_lookup(std::false_type, lua_State* L, iterator&, std::size_t index) {
|
|
return stack::push(L, index);
|
|
}
|
|
|
|
static error_result find_comparative(std::false_type, lua_State*, T&) {
|
|
return 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());
|
|
}
|
|
|
|
static error_result find_comparative(std::true_type, lua_State* L, T& self) {
|
|
decltype(auto) value = stack::unqualified_get<V>(L, 2);
|
|
auto it = deferred_traits::begin(L, self);
|
|
auto e = deferred_traits::end(L, self);
|
|
std::size_t index = 1;
|
|
for (;; ++it, ++index) {
|
|
if (it == e) {
|
|
return stack::push(L, lua_nil);
|
|
}
|
|
if (value == get_value(is_associative(), *it)) {
|
|
break;
|
|
}
|
|
}
|
|
return find_associative_lookup(meta::any<is_lookup, is_associative>(), L, it, index);
|
|
}
|
|
|
|
static error_result find_has(std::false_type, lua_State* L, T& self) {
|
|
return find_comparative(meta::supports_op_equal<V>(), L, self);
|
|
}
|
|
|
|
static error_result add_insert_after(std::false_type, lua_State* L, T& self, stack_object value, iterator&) {
|
|
return add_insert_after(std::false_type(), L, self, value);
|
|
}
|
|
|
|
static error_result add_insert_after(std::false_type, lua_State*, T&, stack_object) {
|
|
return error_result("cannot call 'add' on type '%s': no suitable insert/push_back C++ functions", detail::demangle<T>().data());
|
|
}
|
|
|
|
static error_result add_insert_after(std::true_type, lua_State*, T& self, stack_object value, iterator& pos) {
|
|
self.insert_after(pos, value.as<V>());
|
|
return {};
|
|
}
|
|
|
|
static error_result add_insert_after(std::true_type, lua_State* L, T& self, stack_object value) {
|
|
auto backit = self.before_begin();
|
|
{
|
|
auto e = deferred_traits::end(L, self);
|
|
for (auto it = deferred_traits::begin(L, self); it != e; ++backit, ++it) {
|
|
}
|
|
}
|
|
return add_insert_after(std::true_type(), L, self, value, backit);
|
|
}
|
|
|
|
static error_result add_insert(std::true_type, lua_State*, T& self, stack_object value, iterator& pos) {
|
|
self.insert(pos, value.as<V>());
|
|
return {};
|
|
}
|
|
|
|
static error_result add_insert(std::true_type, lua_State* L, T& self, stack_object value) {
|
|
auto pos = deferred_traits::end(L, self);
|
|
return add_insert(std::true_type(), L, self, value, pos);
|
|
}
|
|
|
|
static error_result add_insert(std::false_type, lua_State* L, T& self, stack_object value, iterator& pos) {
|
|
return add_insert_after(meta::has_insert_after<T>(), L, self, std::move(value), pos);
|
|
}
|
|
|
|
static 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));
|
|
}
|
|
|
|
static error_result add_push_back(std::true_type, lua_State*, T& self, stack_object value, iterator&) {
|
|
self.push_back(value.as<V>());
|
|
return {};
|
|
}
|
|
|
|
static error_result add_push_back(std::true_type, lua_State*, T& self, stack_object value) {
|
|
self.push_back(value.as<V>());
|
|
return {};
|
|
}
|
|
|
|
static error_result add_push_back(std::false_type, lua_State* L, T& self, stack_object value, iterator& pos) {
|
|
return add_insert(meta::has_insert<T>(), L, self, value, pos);
|
|
}
|
|
|
|
static error_result add_push_back(std::false_type, lua_State* L, T& self, stack_object value) {
|
|
return add_insert(meta::has_insert<T>(), L, self, value);
|
|
}
|
|
|
|
static error_result add_associative(std::true_type, lua_State* L, T& self, stack_object key, iterator& pos) {
|
|
self.insert(pos, value_type(key.as<K>(), stack::unqualified_get<V>(L, 3)));
|
|
return {};
|
|
}
|
|
|
|
static error_result add_associative(std::true_type, lua_State* L, T& self, stack_object key) {
|
|
auto pos = deferred_traits::end(L, self);
|
|
return add_associative(std::true_type(), L, self, std::move(key), pos);
|
|
}
|
|
|
|
static error_result add_associative(std::false_type, lua_State* L, T& self, stack_object value, iterator& pos) {
|
|
return add_push_back(meta::has_push_back<T>(), L, self, value, pos);
|
|
}
|
|
|
|
static 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);
|
|
}
|
|
|
|
static error_result add_copyable(std::true_type, lua_State* L, T& self, stack_object value, iterator& pos) {
|
|
return add_associative(is_associative(), L, self, std::move(value), pos);
|
|
}
|
|
|
|
static error_result add_copyable(std::true_type, lua_State* L, T& self, stack_object value) {
|
|
return add_associative(is_associative(), L, self, value);
|
|
}
|
|
|
|
static error_result add_copyable(std::false_type, lua_State* L, T& self, stack_object value, iterator&) {
|
|
return add_copyable(std::false_type(), L, self, std::move(value));
|
|
}
|
|
|
|
static error_result add_copyable(std::false_type, lua_State*, T&, stack_object) {
|
|
return error_result("cannot call 'add' on '%s': value_type is non-copyable", detail::demangle<T>().data());
|
|
}
|
|
|
|
static 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 error_result insert_lookup(std::false_type, lua_State* L, T& self, stack_object where, stack_object value) {
|
|
auto it = deferred_traits::begin(L, self);
|
|
auto key = where.as<K>();
|
|
key += deferred_traits::index_adjustment(L, self);
|
|
std::advance(it, key);
|
|
self.insert(it, value.as<V>());
|
|
return {};
|
|
}
|
|
|
|
static 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 += deferred_traits::index_adjustment(L, self);
|
|
auto e = deferred_traits::end(L, self);
|
|
for (auto it = deferred_traits::begin(L, self); key > 0; ++backit, ++it, --key) {
|
|
if (backit == e) {
|
|
return 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 error_result insert_after_has(std::false_type, lua_State*, T&, stack_object, stack_object) {
|
|
return error_result("cannot call 'insert' on '%s': no suitable or similar functionality detected on this container", detail::demangle<T>().data());
|
|
}
|
|
|
|
static 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 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 error_result insert_copyable(std::true_type, lua_State* L, T& self, stack_object key, stack_object value) {
|
|
return insert_has(meta::has_insert<T>(), L, self, std::move(key), std::move(value));
|
|
}
|
|
|
|
static error_result insert_copyable(std::false_type, lua_State*, T&, stack_object, stack_object) {
|
|
return error_result("cannot call 'insert' on '%s': value_type is non-copyable", detail::demangle<T>().data());
|
|
}
|
|
|
|
static error_result erase_integral(std::true_type, lua_State* L, T& self, K& key) {
|
|
auto it = deferred_traits::begin(L, self);
|
|
key += deferred_traits::index_adjustment(L, self);
|
|
std::advance(it, key);
|
|
self.erase(it);
|
|
|
|
return {};
|
|
}
|
|
|
|
static 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_traits::end(L, self);
|
|
auto it = std::find_if(deferred_traits::begin(L, self), e, std::ref(fx));
|
|
if (it == e) {
|
|
return {};
|
|
}
|
|
self.erase(it);
|
|
|
|
return {};
|
|
}
|
|
|
|
static error_result erase_associative_lookup(std::true_type, lua_State*, T& self, const K& key) {
|
|
self.erase(key);
|
|
return {};
|
|
}
|
|
|
|
static 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 error_result erase_after_has(std::true_type, lua_State* L, T& self, K& key) {
|
|
auto backit = self.before_begin();
|
|
{
|
|
key += deferred_traits::index_adjustment(L, self);
|
|
auto e = deferred_traits::end(L, self);
|
|
for (auto it = deferred_traits::begin(L, self); key > 0; ++backit, ++it, --key) {
|
|
if (backit == e) {
|
|
return error_result("sol: out of bounds for erase on '%s'", detail::demangle<T>().c_str());
|
|
}
|
|
}
|
|
}
|
|
self.erase_after(backit);
|
|
return {};
|
|
}
|
|
|
|
static error_result erase_after_has(std::false_type, lua_State*, T&, const K&) {
|
|
return error_result("sol: cannot call erase on '%s'", detail::demangle<T>().c_str());
|
|
}
|
|
|
|
static 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 error_result erase_has(std::false_type, lua_State* L, T& self, K& key) {
|
|
return erase_after_has(has_erase_after<T>(), L, self, key);
|
|
}
|
|
|
|
static auto size_has(std::false_type, lua_State* L, T& self) {
|
|
return std::distance(deferred_traits::begin(L, self), deferred_traits::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_traits::begin(L, self) == deferred_traits::end(L, self);
|
|
}
|
|
|
|
static error_result get_start(lua_State* L, T& self, K& key) {
|
|
return get_it(is_linear_integral(), L, self, key);
|
|
}
|
|
|
|
static 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 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 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& source = i.source;
|
|
auto& it = i.it;
|
|
if (it == deferred_traits::end(L, source)) {
|
|
return 0;
|
|
}
|
|
int p;
|
|
if (ip) {
|
|
++i.i;
|
|
p = stack::push_reference(L, i.i);
|
|
}
|
|
else {
|
|
p = stack::push_reference(L, it->first);
|
|
}
|
|
p += stack::stack_detail::push_reference<push_type>(L, detail::deref_non_pointer(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& source = i.source;
|
|
auto& it = i.it;
|
|
next_K k = stack::unqualified_get<next_K>(L, 2);
|
|
if (it == deferred_traits::end(L, source)) {
|
|
return 0;
|
|
}
|
|
int p;
|
|
p = stack::push_reference(L, k + 1);
|
|
p += stack::stack_detail::push_reference<push_type>(L, detail::deref_non_pointer(*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, src, deferred_traits::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, src, deferred_traits::begin(L, src));
|
|
stack::push(L, 0);
|
|
return 3;
|
|
}
|
|
|
|
public:
|
|
static int at(lua_State* L) {
|
|
auto& self = get_src(L);
|
|
error_result er;
|
|
{
|
|
std::ptrdiff_t pos = stack::unqualified_get<std::ptrdiff_t>(L);
|
|
er = at_start(L, self, pos);
|
|
}
|
|
return handle_errors(L, er);
|
|
}
|
|
|
|
static int get(lua_State* L) {
|
|
auto& self = get_src(L);
|
|
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 (type_of(L, 3) == type::lua_nil) {
|
|
return erase(L);
|
|
}
|
|
auto& self = get_src(L);
|
|
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);
|
|
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);
|
|
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);
|
|
error_result er = find_has(has_find<T>(), L, self);
|
|
return handle_errors(L, er);
|
|
}
|
|
|
|
static iterator begin(lua_State*, T& self) {
|
|
using std::begin;
|
|
return begin(self);
|
|
}
|
|
|
|
static iterator end(lua_State*, T& self) {
|
|
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);
|
|
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&) {
|
|
#if defined(SOL_CONTAINERS_START_INDEX)
|
|
return static_cast<std::ptrdiff_t>((SOL_CONTAINERS_START) == 0 ? 0 : -(SOL_CONTAINERS_START));
|
|
#else
|
|
return static_cast<std::ptrdiff_t>(-1);
|
|
#endif
|
|
}
|
|
|
|
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 container_traits_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 container_traits<X> deferred_traits;
|
|
|
|
public:
|
|
typedef std::remove_extent_t<T> value_type;
|
|
typedef value_type* iterator;
|
|
|
|
private:
|
|
struct iter {
|
|
T& source;
|
|
iterator it;
|
|
|
|
iter(T& source, iterator it)
|
|
: source(source), it(std::move(it)) {
|
|
}
|
|
};
|
|
|
|
static auto& get_src(lua_State* L) {
|
|
auto p = stack::unqualified_check_get<T*>(L, 1);
|
|
#if defined(SOL_SAFE_USERTYPE) && 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) {
|
|
const auto& v = self[idx];
|
|
if (v == value) {
|
|
return stack::push(L, idx + 1);
|
|
}
|
|
}
|
|
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& source = i.source;
|
|
auto& it = i.it;
|
|
std::size_t k = stack::unqualified_get<std::size_t>(L, 2);
|
|
if (it == deferred_traits::end(L, source)) {
|
|
return 0;
|
|
}
|
|
int p;
|
|
p = stack::push_reference(L, k + 1);
|
|
p += stack::push_reference(L, detail::deref_non_pointer(*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_traits::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_non_pointer(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_traits::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 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, src, deferred_traits::begin(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&) {
|
|
#if defined(SOL_CONTAINERS_START_INDEX)
|
|
return (SOL_CONTAINERS_START) == 0 ? 0 : -(SOL_CONTAINERS_START);
|
|
#else
|
|
return -1;
|
|
#endif
|
|
}
|
|
|
|
static iterator begin(lua_State*, T& self) {
|
|
return std::addressof(self[0]);
|
|
}
|
|
|
|
static iterator end(lua_State*, T& self) {
|
|
return std::addressof(self[0]) + std::extent<T>::value;
|
|
}
|
|
};
|
|
|
|
template <typename X>
|
|
struct container_traits_default<container_traits<X>> : container_traits_default<X> {};
|
|
} // namespace container_detail
|
|
|
|
template <typename T>
|
|
struct container_traits : container_detail::container_traits_default<T> {};
|
|
|
|
} // namespace sol
|
|
|
|
#endif // SOL_CONTAINER_TRAITS_HPP
|