408 lines
14 KiB
C++
408 lines
14 KiB
C++
|
// Copyright satoren
|
||
|
|
// Distributed under the Boost Software License, Version 1.0. (See
|
||
|
|
// accompanying file LICENSE_1_0.txt or copy at
|
||
|
|
// http://www.boost.org/LICENSE_1_0.txt)
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include <vector>
|
||
|
|
#include <map>
|
||
|
|
#include <cassert>
|
||
|
|
#include "kaguya/config.hpp"
|
||
|
|
#include "kaguya/lua_ref.hpp"
|
||
|
|
#include "kaguya/exception.hpp"
|
||
|
|
#include "kaguya/type.hpp"
|
||
|
|
#include "kaguya/utility.hpp"
|
||
|
|
#include "kaguya/detail/lua_function_def.hpp"
|
||
|
|
#include "kaguya/detail/lua_variant_def.hpp"
|
||
|
|
|
||
|
|
namespace kaguya {
|
||
|
|
/// @brief function result value proxy class.
|
||
|
|
/// don't direct use.
|
||
|
|
class FunctionResults : public Ref::StackRef,
|
||
|
|
public detail::LuaVariantImpl<FunctionResults> {
|
||
|
|
FunctionResults(lua_State *state, int return_status, int startIndex)
|
||
|
|
: Ref::StackRef(state, startIndex, true), state_(state),
|
||
|
|
resultStatus_(return_status),
|
||
|
|
resultCount_(lua_gettop(state) + 1 - startIndex) {}
|
||
|
|
FunctionResults(lua_State *state, int return_status, int startIndex,
|
||
|
|
int endIndex)
|
||
|
|
: Ref::StackRef(state, startIndex, true), state_(state),
|
||
|
|
resultStatus_(return_status), resultCount_(endIndex - startIndex) {}
|
||
|
|
friend class detail::FunctionResultProxy;
|
||
|
|
|
||
|
|
public:
|
||
|
|
FunctionResults()
|
||
|
|
: Ref::StackRef(), state_(0), resultStatus_(0), resultCount_(0) {}
|
||
|
|
FunctionResults(lua_State *state)
|
||
|
|
: Ref::StackRef(state, 0, true), state_(state), resultStatus_(0),
|
||
|
|
resultCount_(0) {}
|
||
|
|
#if KAGUYA_USE_CPP11
|
||
|
|
FunctionResults(FunctionResults &&src)
|
||
|
|
: Ref::StackRef(std::move(src)), state_(src.state_),
|
||
|
|
resultStatus_(src.resultStatus_), resultCount_(src.resultCount_) {
|
||
|
|
src.state_ = 0;
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
FunctionResults(const FunctionResults &src)
|
||
|
|
: Ref::StackRef(src), state_(src.state_),
|
||
|
|
resultStatus_(src.resultStatus_), resultCount_(src.resultCount_) {
|
||
|
|
src.state_ = 0;
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
~FunctionResults() {}
|
||
|
|
|
||
|
|
struct reference : public Ref::StackRef,
|
||
|
|
public detail::LuaVariantImpl<reference> {
|
||
|
|
reference(lua_State *s, int index) : Ref::StackRef(s, index, false) {}
|
||
|
|
|
||
|
|
reference *operator->() { return this; }
|
||
|
|
const reference *operator->() const { return this; }
|
||
|
|
};
|
||
|
|
template <typename T> struct iterator_base {
|
||
|
|
iterator_base(lua_State *s, int i) : state(s), stack_index(i) {}
|
||
|
|
template <typename U>
|
||
|
|
iterator_base(const iterator_base<U> &other)
|
||
|
|
: state(other.state), stack_index(other.stack_index) {}
|
||
|
|
T operator*() const { return reference(state, stack_index); }
|
||
|
|
T operator->() const { return reference(state, stack_index); }
|
||
|
|
const iterator_base &operator++() {
|
||
|
|
stack_index++;
|
||
|
|
return *this;
|
||
|
|
}
|
||
|
|
iterator_base operator++(int) {
|
||
|
|
return iterator_base(state, stack_index++);
|
||
|
|
}
|
||
|
|
|
||
|
|
iterator_base operator+=(int n) {
|
||
|
|
stack_index += n;
|
||
|
|
return iterator_base(state, stack_index);
|
||
|
|
}
|
||
|
|
/**
|
||
|
|
* @name relational operators
|
||
|
|
* @brief
|
||
|
|
*/
|
||
|
|
//@{
|
||
|
|
bool operator==(const iterator_base &other) const {
|
||
|
|
return state == other.state && stack_index == other.stack_index;
|
||
|
|
}
|
||
|
|
bool operator!=(const iterator_base &other) const {
|
||
|
|
return !(*this == other);
|
||
|
|
}
|
||
|
|
//@}
|
||
|
|
int index() const { return stack_index; }
|
||
|
|
|
||
|
|
private:
|
||
|
|
template <typename U> friend struct iterator_base;
|
||
|
|
lua_State *state;
|
||
|
|
int stack_index;
|
||
|
|
};
|
||
|
|
typedef iterator_base<reference> iterator;
|
||
|
|
typedef iterator_base<const reference> const_iterator;
|
||
|
|
typedef reference const_reference;
|
||
|
|
|
||
|
|
iterator begin() { return iterator(state_, stack_index_); }
|
||
|
|
iterator end() { return iterator(state_, stack_index_ + resultCount_); }
|
||
|
|
const_iterator begin() const { return const_iterator(state_, stack_index_); }
|
||
|
|
const_iterator end() const {
|
||
|
|
return const_iterator(state_, stack_index_ + resultCount_);
|
||
|
|
}
|
||
|
|
const_iterator cbegin() const { return const_iterator(state_, stack_index_); }
|
||
|
|
const_iterator cend() const {
|
||
|
|
return const_iterator(state_, stack_index_ + resultCount_);
|
||
|
|
}
|
||
|
|
|
||
|
|
template <class Result>
|
||
|
|
Result get_result(types::typetag<Result> = types::typetag<Result>()) const {
|
||
|
|
return util::get_result<Result>(state_, stack_index_);
|
||
|
|
}
|
||
|
|
LuaStackRef get_result(types::typetag<LuaStackRef>) const {
|
||
|
|
pop_ = 0;
|
||
|
|
return LuaStackRef(state_, stack_index_, true);
|
||
|
|
}
|
||
|
|
lua_State *state() const { return state_; }
|
||
|
|
|
||
|
|
template <typename T>
|
||
|
|
typename lua_type_traits<T>::get_type result_at(size_t index) const {
|
||
|
|
if (index >= result_size()) {
|
||
|
|
throw std::out_of_range("function result out of range");
|
||
|
|
}
|
||
|
|
return lua_type_traits<T>::get(state_,
|
||
|
|
stack_index_ + static_cast<int>(index));
|
||
|
|
}
|
||
|
|
reference result_at(size_t index) const {
|
||
|
|
if (index >= result_size()) {
|
||
|
|
throw std::out_of_range("function result out of range");
|
||
|
|
}
|
||
|
|
return reference(state_, stack_index_ + static_cast<int>(index));
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t result_size() const { return resultCount_; }
|
||
|
|
|
||
|
|
size_t resultStatus() const { return resultStatus_; }
|
||
|
|
|
||
|
|
operator LuaStackRef() {
|
||
|
|
pop_ = 0;
|
||
|
|
return LuaStackRef(state_, stack_index_, true);
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
mutable lua_State *state_;
|
||
|
|
int resultStatus_;
|
||
|
|
int resultCount_;
|
||
|
|
};
|
||
|
|
|
||
|
|
namespace detail {
|
||
|
|
template <typename RetType>
|
||
|
|
inline RetType FunctionResultProxy::ReturnValue(lua_State *state,
|
||
|
|
int return_status, int retindex,
|
||
|
|
types::typetag<RetType>) {
|
||
|
|
return FunctionResults(state, return_status, retindex)
|
||
|
|
.get_result(types::typetag<RetType>());
|
||
|
|
}
|
||
|
|
inline FunctionResults
|
||
|
|
FunctionResultProxy::ReturnValue(lua_State *state, int return_status,
|
||
|
|
int retindex,
|
||
|
|
types::typetag<FunctionResults>) {
|
||
|
|
return FunctionResults(state, return_status, retindex);
|
||
|
|
}
|
||
|
|
inline void FunctionResultProxy::ReturnValue(lua_State *state,
|
||
|
|
int return_status, int retindex,
|
||
|
|
types::typetag<void>) {
|
||
|
|
KAGUYA_UNUSED(return_status);
|
||
|
|
lua_settop(state, retindex - 1);
|
||
|
|
}
|
||
|
|
|
||
|
|
#if KAGUYA_USE_CPP11
|
||
|
|
template <typename Derived>
|
||
|
|
template <class... Args>
|
||
|
|
FunctionResults LuaFunctionImpl<Derived>::operator()(Args &&... args) {
|
||
|
|
return this->template call<FunctionResults>(std::forward<Args>(args)...);
|
||
|
|
}
|
||
|
|
|
||
|
|
template <typename Derived>
|
||
|
|
template <class... Args>
|
||
|
|
FunctionResults LuaThreadImpl<Derived>::operator()(Args &&... args) {
|
||
|
|
return this->template resume<FunctionResults>(std::forward<Args>(args)...);
|
||
|
|
}
|
||
|
|
template <typename Derived>
|
||
|
|
template <class... Args>
|
||
|
|
FunctionResults LuaVariantImpl<Derived>::operator()(Args &&... args) {
|
||
|
|
int t = type();
|
||
|
|
if (t == LUA_TTHREAD) {
|
||
|
|
return this->template resume<FunctionResults>(std::forward<Args>(args)...);
|
||
|
|
} else if (t == LUA_TFUNCTION) {
|
||
|
|
return this->template call<FunctionResults>(std::forward<Args>(args)...);
|
||
|
|
} else {
|
||
|
|
except::typeMismatchError(state_(), " is not function or thread");
|
||
|
|
return FunctionResults(state_());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
#define KAGUYA_TEMPLATE_PARAMETER(N) template <typename Derived>
|
||
|
|
#define KAGUYA_FUNCTION_ARGS_DEF(N)
|
||
|
|
#define KAGUYA_CALL_ARGS(N)
|
||
|
|
|
||
|
|
#define KAGUYA_OP_FN_DEF(N) \
|
||
|
|
KAGUYA_TEMPLATE_PARAMETER(N) \
|
||
|
|
inline FunctionResults LuaFunctionImpl<Derived>::operator()( \
|
||
|
|
KAGUYA_FUNCTION_ARGS_DEF(N)) { \
|
||
|
|
return this->template call<FunctionResults>(KAGUYA_CALL_ARGS(N)); \
|
||
|
|
} \
|
||
|
|
KAGUYA_TEMPLATE_PARAMETER(N) \
|
||
|
|
inline FunctionResults LuaThreadImpl<Derived>::operator()( \
|
||
|
|
KAGUYA_FUNCTION_ARGS_DEF(N)) { \
|
||
|
|
return this->template resume<FunctionResults>(KAGUYA_CALL_ARGS(N)); \
|
||
|
|
} \
|
||
|
|
KAGUYA_TEMPLATE_PARAMETER(N) \
|
||
|
|
inline FunctionResults LuaVariantImpl<Derived>::operator()( \
|
||
|
|
KAGUYA_FUNCTION_ARGS_DEF(N)) { \
|
||
|
|
int t = type(); \
|
||
|
|
if (t == LUA_TTHREAD) { \
|
||
|
|
return this->template resume<FunctionResults>(KAGUYA_CALL_ARGS(N)); \
|
||
|
|
} else if (t == LUA_TFUNCTION) { \
|
||
|
|
return this->template call<FunctionResults>(KAGUYA_CALL_ARGS(N)); \
|
||
|
|
} else { \
|
||
|
|
except::typeMismatchError(state_(), " is not function or thread"); \
|
||
|
|
return FunctionResults(state_()); \
|
||
|
|
} \
|
||
|
|
}
|
||
|
|
|
||
|
|
KAGUYA_OP_FN_DEF(0)
|
||
|
|
|
||
|
|
#undef KAGUYA_TEMPLATE_PARAMETER
|
||
|
|
#undef KAGUYA_FUNCTION_ARGS_DEF
|
||
|
|
#undef KAGUYA_CALL_ARGS
|
||
|
|
#define KAGUYA_TEMPLATE_PARAMETER(N) \
|
||
|
|
template <typename Derived> template <KAGUYA_PP_TEMPLATE_DEF_REPEAT(N)>
|
||
|
|
#define KAGUYA_FUNCTION_ARGS_DEF(N) KAGUYA_PP_ARG_CR_DEF_REPEAT(N)
|
||
|
|
#define KAGUYA_CALL_ARGS(N) KAGUYA_PP_ARG_REPEAT(N)
|
||
|
|
|
||
|
|
KAGUYA_PP_REPEAT_DEF(KAGUYA_FUNCTION_MAX_ARGS, KAGUYA_OP_FN_DEF)
|
||
|
|
#undef KAGUYA_OP_FN_DEF
|
||
|
|
#undef KAGUYA_TEMPLATE_PARAMETER
|
||
|
|
|
||
|
|
#undef KAGUYA_CALL_ARGS
|
||
|
|
#undef KAGUYA_FUNCTION_ARGS_DEF
|
||
|
|
#undef KAGUYA_CALL_DEF
|
||
|
|
#undef KAGUYA_OP_FN_DEF
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
inline std::ostream &operator<<(std::ostream &os, const FunctionResults &res) {
|
||
|
|
for (FunctionResults::const_iterator it = res.begin(); it != res.end();
|
||
|
|
++it) {
|
||
|
|
if (it != res.begin()) {
|
||
|
|
os << ",";
|
||
|
|
}
|
||
|
|
util::stackValueDump(os, res.state(), it.index());
|
||
|
|
}
|
||
|
|
|
||
|
|
return os;
|
||
|
|
}
|
||
|
|
|
||
|
|
/// @ingroup lua_type_traits
|
||
|
|
/// @brief lua_type_traits for FunctionResults
|
||
|
|
template <> struct lua_type_traits<FunctionResults> {
|
||
|
|
static int push(lua_State *l, const FunctionResults &ref) {
|
||
|
|
int size = 0;
|
||
|
|
for (FunctionResults::const_iterator it = ref.cbegin(); it != ref.cend();
|
||
|
|
++it) {
|
||
|
|
size += it->push(l);
|
||
|
|
}
|
||
|
|
return size;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
/// @ingroup lua_type_traits
|
||
|
|
/// @brief lua_type_traits for FunctionResults::reference
|
||
|
|
template <> struct lua_type_traits<FunctionResults::reference> {
|
||
|
|
static int push(lua_State *l, const FunctionResults::reference &ref) {
|
||
|
|
return ref.push(l);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
template <unsigned int I>
|
||
|
|
FunctionResults::reference get(const FunctionResults &res) {
|
||
|
|
return res.result_at(I);
|
||
|
|
}
|
||
|
|
|
||
|
|
/// @ingroup lua_type_traits
|
||
|
|
/// @brief lua_type_traits for LuaFunction
|
||
|
|
template <> struct lua_type_traits<LuaFunction> {
|
||
|
|
typedef LuaFunction get_type;
|
||
|
|
typedef LuaFunction push_type;
|
||
|
|
|
||
|
|
static bool strictCheckType(lua_State *l, int index) {
|
||
|
|
return lua_isfunction(l, index);
|
||
|
|
}
|
||
|
|
static bool checkType(lua_State *l, int index) {
|
||
|
|
return lua_isfunction(l, index) || lua_isnil(l, index);
|
||
|
|
}
|
||
|
|
static LuaFunction get(lua_State *l, int index) {
|
||
|
|
lua_pushvalue(l, index);
|
||
|
|
return LuaFunction(l, StackTop());
|
||
|
|
}
|
||
|
|
static int push(lua_State *l, const LuaFunction &ref) { return ref.push(l); }
|
||
|
|
};
|
||
|
|
/// @ingroup lua_type_traits
|
||
|
|
/// @brief lua_type_traits for LuaFunction
|
||
|
|
template <>
|
||
|
|
struct lua_type_traits<const LuaFunction &> : lua_type_traits<LuaFunction> {};
|
||
|
|
|
||
|
|
/// @ingroup lua_type_traits
|
||
|
|
/// @brief lua_type_traits for LuaThread
|
||
|
|
template <> struct lua_type_traits<LuaThread> {
|
||
|
|
typedef LuaThread get_type;
|
||
|
|
typedef LuaThread push_type;
|
||
|
|
|
||
|
|
static bool strictCheckType(lua_State *l, int index) {
|
||
|
|
return lua_isthread(l, index);
|
||
|
|
}
|
||
|
|
static bool checkType(lua_State *l, int index) {
|
||
|
|
return lua_isthread(l, index) || lua_isnil(l, index);
|
||
|
|
}
|
||
|
|
static LuaThread get(lua_State *l, int index) {
|
||
|
|
lua_pushvalue(l, index);
|
||
|
|
return LuaThread(l, StackTop());
|
||
|
|
}
|
||
|
|
static int push(lua_State *l, const LuaThread &ref) { return ref.push(l); }
|
||
|
|
};
|
||
|
|
/// @ingroup lua_type_traits
|
||
|
|
/// @brief lua_type_traits for LuaThread
|
||
|
|
template <>
|
||
|
|
struct lua_type_traits<const LuaThread &> : lua_type_traits<LuaThread> {};
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @brief table and function binder.
|
||
|
|
* state["table"]->*"fun"() is table:fun() in Lua
|
||
|
|
* @param arg... function args
|
||
|
|
*/
|
||
|
|
class MemberFunctionBinder {
|
||
|
|
public:
|
||
|
|
template <class T>
|
||
|
|
MemberFunctionBinder(LuaRef self, T key)
|
||
|
|
: self_(self), fun_(self_.getField(key)) {}
|
||
|
|
|
||
|
|
#define KAGUYA_DELEGATE_LUAREF fun_
|
||
|
|
#define KAGUYA_DELEGATE_FIRST_ARG self_
|
||
|
|
#define KAGUYA_DELEGATE_FIRST_ARG_C KAGUYA_DELEGATE_FIRST_ARG,
|
||
|
|
|
||
|
|
#if KAGUYA_USE_CPP11
|
||
|
|
template <class... Args> FunctionResults operator()(Args &&... args) {
|
||
|
|
return KAGUYA_DELEGATE_LUAREF(
|
||
|
|
KAGUYA_DELEGATE_FIRST_ARG_C std::forward<Args>(args)...);
|
||
|
|
}
|
||
|
|
|
||
|
|
template <class Result, class... Args> Result call(Args &&... args) {
|
||
|
|
return KAGUYA_DELEGATE_LUAREF.call<Result>(
|
||
|
|
KAGUYA_DELEGATE_FIRST_ARG_C std::forward<Args>(args)...);
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
|
||
|
|
#define KAGUYA_OP_FN_DEF(N) \
|
||
|
|
template <KAGUYA_PP_TEMPLATE_DEF_REPEAT(N)> \
|
||
|
|
FunctionResults operator()(KAGUYA_PP_ARG_CR_DEF_REPEAT(N)) { \
|
||
|
|
return KAGUYA_DELEGATE_LUAREF( \
|
||
|
|
KAGUYA_DELEGATE_FIRST_ARG_C KAGUYA_PP_ARG_REPEAT(N)); \
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @brief If type is function, call lua function.
|
||
|
|
* If type is lua thread,start or resume lua thread.
|
||
|
|
* Otherwise send error message to error handler
|
||
|
|
*/
|
||
|
|
FunctionResults operator()() {
|
||
|
|
return KAGUYA_DELEGATE_LUAREF(KAGUYA_DELEGATE_FIRST_ARG);
|
||
|
|
}
|
||
|
|
KAGUYA_PP_REPEAT_DEF(KAGUYA_FUNCTION_MAX_ARGS, KAGUYA_OP_FN_DEF)
|
||
|
|
|
||
|
|
#undef KAGUYA_OP_FN_DEF
|
||
|
|
|
||
|
|
#define KAGUYA_CALL_DEF(N) \
|
||
|
|
template <class Result, KAGUYA_PP_TEMPLATE_DEF_REPEAT(N)> \
|
||
|
|
Result call(KAGUYA_PP_ARG_CR_DEF_REPEAT(N)) { \
|
||
|
|
return KAGUYA_DELEGATE_LUAREF.call<Result>( \
|
||
|
|
KAGUYA_DELEGATE_FIRST_ARG_C KAGUYA_PP_ARG_REPEAT(N)); \
|
||
|
|
}
|
||
|
|
|
||
|
|
template <class Result> Result call() {
|
||
|
|
return KAGUYA_DELEGATE_LUAREF.call<Result>(KAGUYA_DELEGATE_FIRST_ARG);
|
||
|
|
}
|
||
|
|
KAGUYA_PP_REPEAT_DEF(KAGUYA_FUNCTION_MAX_ARGS, KAGUYA_CALL_DEF)
|
||
|
|
|
||
|
|
#undef KAGUYA_CALL_DEF
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#undef KAGUYA_DELEGATE_FIRST_ARG_C
|
||
|
|
#undef KAGUYA_DELEGATE_FIRST_ARG
|
||
|
|
#undef KAGUYA_DELEGATE_LUAREF
|
||
|
|
|
||
|
|
private:
|
||
|
|
LuaRef self_; // Table or Userdata
|
||
|
|
LuaFunction fun_;
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef MemberFunctionBinder mem_fun_binder; // for backward conpatible
|
||
|
|
}
|