sol2/sol/stack.hpp

189 lines
5.6 KiB
C++
Raw Normal View History

2013-11-25 17:56:27 +08:00
// The MIT License (MIT)
// Copyright (c) 2013 Danny Y., Rapptz
// 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_STACK_HPP
#define SOL_STACK_HPP
#include "reference.hpp"
#include "tuple.hpp"
2013-11-25 17:56:27 +08:00
#include <utility>
#include <type_traits>
namespace sol {
template<typename T, typename R = void>
using EnableIf = typename std::enable_if<T::value, R>::type;
template<typename T, typename R = void>
using DisableIf = typename std::enable_if<!T::value, R>::type;
2013-11-25 17:56:27 +08:00
namespace stack {
namespace detail {
template<class T, class F, class... Vs>
auto ltr_pop(T&& extra, F f, types<>, Vs&&... vs)
-> decltype(f(std::forward<Vs>(vs)...)) {
return f(std::forward<Vs>(vs)...);
}
// take head, produce value from it, pass after other values
template<class F, class Head, class... Tail, class... Vs>
auto ltr_pop(lua_State* L, F f, types<Head, Tail...>, Vs&&... vs)
-> decltype(ltr_pop(L, f, types<Tail...>{}, std::forward<Vs>(vs)..., stack::pop<Head>(L))) {
return ltr_pop(L, f, types<Tail...>{}, std::forward<Vs>(vs)..., stack::pop<Head>(L));
}
2013-11-25 17:56:27 +08:00
template<typename T>
inline T get_unsigned(lua_State* L, std::true_type, int index = -1) {
return lua_tounsigned(L, index);
2013-11-25 17:56:27 +08:00
}
template<typename T>
inline T get_unsigned(lua_State* L, std::false_type, int index = -1) {
return lua_tointeger(L, index);
2013-11-25 17:56:27 +08:00
}
template<typename T>
inline T get_arithmetic(lua_State* L, std::false_type, int index = -1) {
2013-11-25 17:56:27 +08:00
// T is a floating point
return lua_tonumber(L, index);
2013-11-25 17:56:27 +08:00
}
template<typename T>
inline T get_arithmetic(lua_State* L, std::true_type, int index = -1) {
2013-11-25 17:56:27 +08:00
// T is an integral
2013-12-01 10:12:50 +08:00
return get_unsigned<T>(L, std::is_unsigned<T>{}, index);
2013-11-25 17:56:27 +08:00
}
template<typename T>
inline T get_helper(lua_State* L, std::true_type, int index = -1) {
2013-11-25 17:56:27 +08:00
// T is a class type
return T(L, index);
2013-11-25 17:56:27 +08:00
}
template<typename T>
inline T get_helper(lua_State* L, std::false_type, int index = -1) {
2013-11-25 17:56:27 +08:00
// T is a fundamental type
return get_arithmetic<T>(L, std::is_integral<T>{}, index);
2013-11-25 17:56:27 +08:00
}
template<typename T>
inline void push_unsigned(lua_State* L, T x, std::true_type) {
lua_pushunsigned(L, x);
}
template<typename T>
inline void push_unsigned(lua_State* L, T x, std::false_type) {
lua_pushinteger(L, x);
}
template<typename T>
inline void push_arithmetic(lua_State* L, T x, std::true_type) {
// T is an integral type
push_unsigned(L, x, std::is_unsigned<T>{});
}
template<typename T>
inline void push_arithmetic(lua_State* L, T x, std::false_type) {
// T is an floating point type
lua_pushnumber(L, x);
}
} // detail
template<typename T>
inline T get(lua_State* L, int index = -1) {
return detail::get_helper<T>(L, std::is_class<T>{}, index);
2013-11-25 17:56:27 +08:00
}
template<>
inline bool get<bool>(lua_State* L, int index) {
return lua_toboolean(L, index) != 0;
2013-11-25 17:56:27 +08:00
}
template<>
inline std::string get<std::string>(lua_State* L, int index) {
2013-11-25 17:56:27 +08:00
std::string::size_type len;
auto str = lua_tolstring(L, index, &len);
2013-11-25 17:56:27 +08:00
return { str, len };
}
template<>
inline const char* get<const char*>(lua_State* L, int index) {
return lua_tostring(L, index);
}
template<typename T>
inline T pop(lua_State* L) {
auto r = get<T>(L);
2013-11-25 17:56:27 +08:00
lua_pop(L, 1);
return r;
2013-11-25 17:56:27 +08:00
}
template<typename T>
inline EnableIf<std::is_arithmetic<T>> push(lua_State* L, T arithmetic) {
detail::push_arithmetic(L, arithmetic, std::is_integral<T>{});
2013-11-25 17:56:27 +08:00
}
inline void push(lua_State*, reference& ref) {
ref.push();
}
2013-11-25 17:56:27 +08:00
inline void push(lua_State* L, bool boolean) {
lua_pushboolean(L, boolean);
}
inline void push(lua_State* L, const nil_t&) {
lua_pushnil(L);
}
inline void push(lua_State* L, lua_CFunction func) {
lua_pushcfunction(L, func);
}
template<size_t N>
inline void push(lua_State* L, const char (&str)[N]) {
lua_pushlstring(L, str, N - 1);
}
inline void push(lua_State* L, const std::string& str) {
lua_pushlstring(L, str.c_str(), str.size());
}
namespace detail {
template<typename T, std::size_t... I>
inline void push(lua_State* L, indices<I...>, const T& tuplen) {
using swallow = char [];
void(swallow{ '\0', (sol::stack::push(L, std::get<I>(tuplen)), '\0')... });
}
} // detail
template<typename... Args>
inline void push(lua_State* L, const std::tuple<Args...>& tuplen) {
detail::push(L, build_indices<sizeof...(Args)>(), tuplen);
}
template<typename... Args, typename TFx>
inline auto pop_call(lua_State* L, TFx&& fx, types<Args...> t)->decltype(detail::ltr_pop(L, fx, t)) {
return detail::ltr_pop(L, fx, t);
}
2013-11-25 17:56:27 +08:00
} // stack
} // sol
#endif // SOL_STACK_HPP