mirror of
https://github.com/ThePhD/sol2.git
synced 2024-03-22 13:10:44 +08:00
373 lines
15 KiB
C++
373 lines
15 KiB
C++
// The MIT License (MIT)
|
|
|
|
// Copyright (c) 2013-2016 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_TABLE_CORE_HPP
|
|
#define SOL_TABLE_CORE_HPP
|
|
|
|
#include "proxy.hpp"
|
|
#include "stack.hpp"
|
|
#include "function_types.hpp"
|
|
#include "usertype.hpp"
|
|
#include "table_iterator.hpp"
|
|
|
|
namespace sol {
|
|
namespace detail {
|
|
template <std::size_t n>
|
|
struct clean { lua_State* L; clean(lua_State* L) : L(L) {} ~clean() { lua_pop(L, static_cast<int>(n)); } };
|
|
struct ref_clean { lua_State* L; int& n; ref_clean(lua_State* L, int& n) : L(L), n(n) {} ~ref_clean() { lua_pop(L, static_cast<int>(n)); } };
|
|
}
|
|
template <bool top_level>
|
|
class table_core : public reference {
|
|
friend class state;
|
|
friend class state_view;
|
|
|
|
template <typename... Args>
|
|
using is_global = meta::And<meta::Bool<top_level>, meta::is_c_str<Args>...>;
|
|
|
|
template<typename Fx>
|
|
void for_each(std::true_type, Fx&& fx) const {
|
|
auto pp = stack::push_pop( *this );
|
|
stack::push( lua_state( ), nil );
|
|
while ( lua_next( this->lua_state( ), -2 ) ) {
|
|
sol::object key( lua_state( ), -2 );
|
|
sol::object value( lua_state( ), -1 );
|
|
std::pair<sol::object&, sol::object&> keyvalue(key, value);
|
|
fx( keyvalue );
|
|
lua_pop( lua_state( ), 1 );
|
|
}
|
|
}
|
|
|
|
template<typename Fx>
|
|
void for_each(std::false_type, Fx&& fx) const {
|
|
auto pp = stack::push_pop( *this );
|
|
stack::push( lua_state( ), nil );
|
|
while ( lua_next( this->lua_state( ), -2 ) ) {
|
|
sol::object key( lua_state( ), -2 );
|
|
sol::object value( lua_state( ), -1 );
|
|
fx( key, value );
|
|
lua_pop( lua_state( ), 1 );
|
|
}
|
|
}
|
|
|
|
template<typename Ret0, typename Ret1, typename... Ret, std::size_t... I, typename Keys>
|
|
auto tuple_get( types<Ret0, Ret1, Ret...>, std::index_sequence<0, 1, I...>, Keys&& keys ) const
|
|
-> decltype(stack::pop<std::tuple<Ret0, Ret1, Ret...>>(nullptr)){
|
|
typedef decltype(stack::pop<std::tuple<Ret0, Ret1, Ret...>>(nullptr)) Tup;
|
|
return Tup(
|
|
traverse_get_optional<top_level, Ret0>(meta::is_specialization_of<meta::Unqualified<Ret0>, sol::optional>(), detail::forward_get<0>(keys)),
|
|
traverse_get_optional<top_level, Ret1>(meta::is_specialization_of<meta::Unqualified<Ret1>, sol::optional>(), detail::forward_get<1>(keys)),
|
|
traverse_get_optional<top_level, Ret>(meta::is_specialization_of<meta::Unqualified<Ret>, sol::optional>(), detail::forward_get<I>(keys))...
|
|
);
|
|
}
|
|
|
|
template<typename Ret, std::size_t I, typename Keys>
|
|
decltype(auto) tuple_get( types<Ret>, std::index_sequence<I>, Keys&& keys ) const {
|
|
return traverse_get_optional<top_level, Ret>( meta::is_specialization_of<meta::Unqualified<Ret>, sol::optional>(), detail::forward_get<I>(keys) );
|
|
}
|
|
|
|
template<typename Pairs, std::size_t... I>
|
|
void tuple_set( std::index_sequence<I...>, Pairs&& pairs ) {
|
|
auto pp = stack::push_pop<top_level && (is_global<decltype(detail::forward_get<I * 2>(pairs))...>::value)>(*this);
|
|
void(detail::swallow{ (stack::set_field<top_level>(lua_state(),
|
|
detail::forward_get<I * 2>(pairs),
|
|
detail::forward_get<I * 2 + 1>(pairs)
|
|
), 0)... });
|
|
}
|
|
|
|
template <bool global, typename T, typename Key>
|
|
decltype(auto) traverse_get_deep( Key&& key ) const {
|
|
stack::get_field<global>( lua_state( ), std::forward<Key>( key ) );
|
|
return stack::get<T>( lua_state( ) );
|
|
}
|
|
|
|
template <bool global, typename T, typename Key, typename... Keys>
|
|
decltype(auto) traverse_get_deep( Key&& key, Keys&&... keys ) const {
|
|
stack::get_field<global>( lua_state( ), std::forward<Key>( key ) );
|
|
return traverse_get_deep<false, T>(std::forward<Keys>(keys)...);
|
|
}
|
|
|
|
template <bool global, typename T, std::size_t I, typename Key>
|
|
decltype(auto) traverse_get_deep_optional( int& popcount, Key&& key ) const {
|
|
auto p = stack::probe_get_field<global>(lua_state(), std::forward<Key>(key), -1);
|
|
popcount += p.levels;
|
|
if (!p.success)
|
|
return T(nullopt);
|
|
return stack::get<T>( lua_state( ) );
|
|
}
|
|
|
|
template <bool global, typename T, std::size_t I, typename Key, typename... Keys>
|
|
decltype(auto) traverse_get_deep_optional( int& popcount, Key&& key, Keys&&... keys ) const {
|
|
auto p = I > 0 ? stack::probe_get_field<global>(lua_state(), std::forward<Key>(key), - 1) : stack::probe_get_field<global>( lua_state( ), std::forward<Key>( key ) );
|
|
popcount += p.levels;
|
|
if (!p.success)
|
|
return T(nullopt);
|
|
return traverse_get_deep_optional<false, T, I + 1>(popcount, std::forward<Keys>(keys)...);
|
|
}
|
|
|
|
template <bool global, typename T, typename... Keys>
|
|
decltype(auto) traverse_get_optional( std::false_type, Keys&&... keys ) const {
|
|
detail::clean<sizeof...(Keys)> c(lua_state());
|
|
return traverse_get_deep<top_level, T>(std::forward<Keys>(keys)...);
|
|
}
|
|
|
|
template <bool global, typename T, typename... Keys>
|
|
decltype(auto) traverse_get_optional( std::true_type, Keys&&... keys ) const {
|
|
int popcount = 0;
|
|
detail::ref_clean c(lua_state(), popcount);
|
|
return traverse_get_deep_optional<top_level, T, 0>(popcount, std::forward<Keys>(keys)...);
|
|
}
|
|
|
|
template <bool global, typename Key, typename Value>
|
|
void traverse_set_deep( Key&& key, Value&& value ) const {
|
|
stack::set_field<global>( lua_state( ), std::forward<Key>( key ), std::forward<Value>(value) );
|
|
}
|
|
|
|
template <bool global, typename Key, typename... Keys>
|
|
void traverse_set_deep( Key&& key, Keys&&... keys ) const {
|
|
stack::get_field<global>( lua_state( ), std::forward<Key>( key ) );
|
|
traverse_set_deep<false>(std::forward<Keys>(keys)...);
|
|
}
|
|
|
|
table_core(lua_State* L, detail::global_tag t) noexcept : reference(L, t) { }
|
|
|
|
public:
|
|
table_core( ) noexcept : reference( ) { }
|
|
table_core( const table_core<true>& global ) noexcept : reference( global ) { }
|
|
table_core( lua_State* L, int index = -1 ) : reference( L, index ) {
|
|
type_assert( L, index, type::table );
|
|
}
|
|
|
|
table_iterator begin () const {
|
|
return table_iterator(*this);
|
|
}
|
|
|
|
table_iterator end() const {
|
|
return table_iterator();
|
|
}
|
|
|
|
table_iterator cbegin() const {
|
|
return begin();
|
|
}
|
|
|
|
table_iterator cend() const {
|
|
return end();
|
|
}
|
|
|
|
template<typename... Ret, typename... Keys>
|
|
decltype(auto) get( Keys&&... keys ) const {
|
|
static_assert(sizeof...(Keys) == sizeof...(Ret), "number of keys and number of return types do not match");
|
|
auto pp = stack::push_pop<is_global<Keys...>::value>(*this);
|
|
return tuple_get( types<Ret...>( ), std::make_index_sequence<sizeof...(Ret)>( ), std::forward_as_tuple(std::forward<Keys>(keys)...));
|
|
}
|
|
|
|
template<typename T, typename Key>
|
|
decltype(auto) get_or(Key&& key, T&& otherwise) const {
|
|
typedef decltype(get<T>("")) U;
|
|
sol::optional<U> option = get<sol::optional<U>>(std::forward<Key>(key));
|
|
if (option) {
|
|
return static_cast<U>(option.value());
|
|
}
|
|
return static_cast<U>(std::forward<T>(otherwise));
|
|
}
|
|
|
|
template<typename T, typename Key, typename D>
|
|
decltype(auto) get_or(Key&& key, D&& otherwise) const {
|
|
sol::optional<T> option = get<sol::optional<T>>(std::forward<Key>(key));
|
|
if (option) {
|
|
return static_cast<T>(option.value());
|
|
}
|
|
return static_cast<T>(std::forward<D>(otherwise));
|
|
}
|
|
|
|
template <typename T, typename... Keys>
|
|
decltype(auto) traverse_get( Keys&&... keys ) const {
|
|
auto pp = stack::push_pop<is_global<Keys...>::value>(*this);
|
|
return traverse_get_optional<top_level, T>(meta::is_specialization_of<meta::Unqualified<T>, sol::optional>(), std::forward<Keys>(keys)...);
|
|
}
|
|
|
|
template <typename... Keys>
|
|
table_core& traverse_set( Keys&&... keys ) {
|
|
auto pp = stack::push_pop<is_global<Keys...>::value>(*this);
|
|
traverse_set_deep<top_level>(std::forward<Keys>(keys)...);
|
|
lua_pop(lua_state(), static_cast<int>(sizeof...(Keys)-2));
|
|
return *this;
|
|
}
|
|
|
|
template<typename... Args>
|
|
table_core& set( Args&&... args ) {
|
|
tuple_set(std::make_index_sequence<sizeof...(Args) / 2>(), std::forward_as_tuple(std::forward<Args>(args)...));
|
|
return *this;
|
|
}
|
|
|
|
template<typename T>
|
|
table_core& set_usertype( usertype<T>& user ) {
|
|
return set_usertype(usertype_traits<T>::name, user);
|
|
}
|
|
|
|
template<typename Key, typename T>
|
|
table_core& set_usertype( Key&& key, usertype<T>& user ) {
|
|
return set(std::forward<Key>(key), user);
|
|
}
|
|
|
|
template<typename Class, typename... Args>
|
|
table_core& new_usertype(const std::string& name, Args&&... args) {
|
|
usertype<Class> utype(std::forward<Args>(args)...);
|
|
set_usertype(name, utype);
|
|
return *this;
|
|
}
|
|
|
|
template<typename Class, typename CTor0, typename... CTor, typename... Args>
|
|
table_core& new_usertype(const std::string& name, Args&&... args) {
|
|
constructors<types<CTor0, CTor...>> ctor{};
|
|
return new_usertype<Class>(name, ctor, std::forward<Args>(args)...);
|
|
}
|
|
|
|
template<typename Class, typename... CArgs, typename... Args>
|
|
table_core& new_usertype(const std::string& name, constructors<CArgs...> ctor, Args&&... args) {
|
|
usertype<Class> utype(ctor, std::forward<Args>(args)...);
|
|
set_usertype(name, utype);
|
|
return *this;
|
|
}
|
|
|
|
template<typename Fx>
|
|
void for_each( Fx&& fx ) const {
|
|
typedef meta::is_callable<Fx( std::pair<sol::object, sol::object> )> is_paired;
|
|
for_each(is_paired(), std::forward<Fx>(fx));
|
|
}
|
|
|
|
size_t size( ) const {
|
|
auto pp = stack::push_pop( *this );
|
|
return lua_rawlen(lua_state(), -1);
|
|
}
|
|
|
|
bool empty() const {
|
|
return cbegin() == cend();
|
|
}
|
|
|
|
template<typename T>
|
|
proxy<table_core&, T> operator[]( T&& key ) & {
|
|
return proxy<table_core&, T>( *this, std::forward<T>( key ) );
|
|
}
|
|
|
|
template<typename T>
|
|
proxy<const table_core&, T> operator[]( T&& key ) const & {
|
|
return proxy<const table_core&, T>( *this, std::forward<T>( key ) );
|
|
}
|
|
|
|
template<typename T>
|
|
proxy<table_core, T> operator[]( T&& key ) && {
|
|
return proxy<table_core, T>( *this, std::forward<T>( key ) );
|
|
}
|
|
|
|
template<typename Sig, typename Key, typename... Args>
|
|
table_core& set_function( Key&& key, Args&&... args ) {
|
|
set_fx( types<Sig>( ), std::forward<Key>( key ), std::forward<Args>( args )... );
|
|
return *this;
|
|
}
|
|
|
|
template<typename Key, typename... Args>
|
|
table_core& set_function( Key&& key, Args&&... args ) {
|
|
set_fx( types<>( ), std::forward<Key>( key ), std::forward<Args>( args )... );
|
|
return *this;
|
|
}
|
|
|
|
private:
|
|
template<typename R, typename... Args, typename Fx, typename Key, typename = std::result_of_t<Fx( Args... )>>
|
|
void set_fx( types<R( Args... )>, Key&& key, Fx&& fx ) {
|
|
set_resolved_function<R( Args... )>( std::forward<Key>( key ), std::forward<Fx>( fx ) );
|
|
}
|
|
|
|
template<typename Fx, typename Key, meta::EnableIf<meta::is_specialization_of<meta::Unqualified<Fx>, overload_set>> = 0>
|
|
void set_fx( types<>, Key&& key, Fx&& fx ) {
|
|
set(std::forward<Key>(key), std::forward<Fx>(fx));
|
|
}
|
|
|
|
template<typename Fx, typename Key, typename... Args, meta::DisableIf<meta::is_specialization_of<meta::Unqualified<Fx>, overload_set>> = 0>
|
|
void set_fx( types<>, Key&& key, Fx&& fx, Args&&... args ) {
|
|
set(std::forward<Key>(key), function_args(std::forward<Fx>(fx), std::forward<Args>(args)...));
|
|
}
|
|
|
|
template<typename... Sig, typename... Args, typename Key>
|
|
void set_resolved_function( Key&& key, Args&&... args ) {
|
|
set(std::forward<Key>(key), function_args<function_sig<Sig...>>(std::forward<Args>(args)...));
|
|
}
|
|
|
|
public:
|
|
static inline table create(lua_State* L, int narr = 0, int nrec = 0) {
|
|
lua_createtable(L, narr, nrec);
|
|
table result(L);
|
|
lua_pop(L, 1);
|
|
return result;
|
|
}
|
|
|
|
template <typename Key, typename Value, typename... Args>
|
|
static inline table create(lua_State* L, int narr, int nrec, Key&& key, Value&& value, Args&&... args) {
|
|
lua_createtable(L, narr, nrec);
|
|
table result(L);
|
|
result.set(std::forward<Key>(key), std::forward<Value>(value), std::forward<Args>(args)...);
|
|
lua_pop(L, 1);
|
|
return result;
|
|
}
|
|
|
|
template <typename... Args>
|
|
static inline table create_with(lua_State* L, Args&&... args) {
|
|
static const int narr = static_cast<int>(meta::count_if_2_pack<std::is_integral, Args...>::value);
|
|
return create(L, narr, static_cast<int>((sizeof...(Args) / 2) - narr), std::forward<Args>(args)...);
|
|
}
|
|
|
|
table create(int narr = 0, int nrec = 0) {
|
|
return create(lua_state(), narr, nrec);
|
|
}
|
|
|
|
template <typename Key, typename Value, typename... Args>
|
|
table create(int narr, int nrec, Key&& key, Value&& value, Args&&... args) {
|
|
return create(lua_state(), narr, nrec, std::forward<Key>(key), std::forward<Value>(value), std::forward<Args>(args)...);
|
|
}
|
|
|
|
template <typename Name>
|
|
table create(Name&& name, int narr = 0, int nrec = 0) {
|
|
table x = create(lua_state(), narr, nrec);
|
|
this->set(std::forward<Name>(name), x);
|
|
return x;
|
|
}
|
|
|
|
template <typename Name, typename Key, typename Value, typename... Args>
|
|
table create(Name&& name, int narr, int nrec, Key&& key, Value&& value, Args&&... args) {
|
|
table x = create(lua_state(), narr, nrec, std::forward<Key>(key), std::forward<Value>(value), std::forward<Args>(args)...);
|
|
this->set(std::forward<Name>(name), x);
|
|
return x;
|
|
}
|
|
|
|
template <typename... Args>
|
|
table create_with(Args&&... args) {
|
|
return create_with(lua_state(), std::forward<Args>(args)...);
|
|
}
|
|
|
|
template <typename Name, typename... Args>
|
|
table create_named(Name&& name, Args&&... args) {
|
|
static const int narr = static_cast<int>(meta::count_if_2_pack<std::is_integral, Args...>::value);
|
|
return create(std::forward<Name>(name), narr, sizeof...(Args) / 2 - narr, std::forward<Args>(args)...);
|
|
}
|
|
};
|
|
} // sol
|
|
|
|
#endif // SOL_TABLE_CORE_HPP
|