luaFunction.h

/*
    Copyright (C) <2012>  <huangweilook@21cn.com>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/
/*
* brief : 注册全局函数
* Author : huangwei
*/
#ifndef _LUAFUNCTION_H
#define _LUAFUNCTION_H
#include "utility.h"
#include "any.h"
#include <iostream>
#include <vector>

namespace luacpp{
template<typename T>
void push_obj(lua_State *L,const T &obj);

//取出栈顶的值，通知将其出栈
template<typename T>
T popvalue(lua_State *L);

template<typename Ret>
class doLuaCall
{
public:
	static Ret doCall(lua_State *L,int nArgs,int errFunc,bool callByObject = false)
	{
		typedef LOKI_TYPELIST_1(void) voidType;
		return call(L,nArgs,errFunc,Int2Type<IndexOf<voidType,Ret>::value == 0>(),callByObject);
	}
private:
	//返回void
	static Ret call(lua_State *L,int nArgs,int errFunc,Int2Type<true>,bool callByObject)
	{
		if(lua_pcall(L, nArgs, 0, errFunc ) != 0)
		{
			const char *error = lua_tostring(L,-1);
			std::string err(error);
			lua_pop(L,1);
			throw err;
		}
		if(callByObject)
			lua_pop(L,1);
	}

	//有返回值
	static Ret call(lua_State *L,int nArgs,int errFunc,Int2Type<false>,bool callByObject)
	{
		Ret ret = _call(L,nArgs,errFunc,callByObject);
		if(callByObject)
			lua_pop(L,1);
		return ret;
	}

	//只有一个返回值
	static Ret _call(lua_State *L,int nArgs,int errFunc,bool callByObject)
	{
		if(lua_pcall(L, nArgs, 1, errFunc ) != 0)
		{
			const char *error = lua_tostring(L,-1);
			printf("%s\n",error);
			std::string err(error);
			lua_pop(L,1);
			throw err;
			return Ret();
		}
		return popvalue<Ret>(L);
	}
};

#ifndef PUSH_FUNCTOR
#define PUSH_FUNCTOR do{\
	lua_pushlightuserdata(L, (void*)_func);\
	lua_pushcclosure(L,lua_cfunction,1);\
	lua_setglobal(L, name);\
	}while(0)
#endif

template<typename FUNCTOR>
class funbinder{};

//用于注册全局函数
//无参
template<typename Ret>
class funbinder<Ret(*)()>
{
public:

	static void pushfuctor(lua_State *L,const char *name,Ret(*_func)())
	{
		PUSH_FUNCTOR;
	}

	static int lua_cfunction(lua_State *L)
	{
		return doCall<Ret>(L,(__func)lua_touserdata(L,lua_upvalueindex(1)),Int2Type<isVoid<Ret>::is_Void>());
	}

private:

	typedef Ret(*__func)();

	template<typename Result>
	static int doCall(lua_State *L,__func func,Int2Type<false>)
	{
		push_obj<Result>(L,func());
		return 1;
	}

	template<typename Result>
	static int doCall(lua_State *L,__func func,Int2Type<true>)
	{
		func();
		return 0;
	}

};

//单参
template<typename Ret,typename Arg1>
class funbinder<Ret(*)(Arg1)>
{
private:
	typedef typename refTraits<Arg1>::RefType arg1_ref_type;
public:

	static void pushfuctor(lua_State *L,const char *name,Ret(*_func)(Arg1))
	{
		PUSH_FUNCTOR;
	}


	static int lua_cfunction(lua_State *L)
	{
		typename GetReplaceType<Arg1>::type tmp_arg1 = popvalue<typename GetReplaceType<Arg1>::type>(L);
		Arg1 arg1 = GetRawValue<typename GetReplaceType<Arg1>::type>(tmp_arg1);
		return doCall<Ret>(L,arg1,(__func)lua_touserdata(L,lua_upvalueindex(1)),Int2Type<isVoid<Ret>::is_Void>());
	}

private:
	typedef Ret(*__func)(Arg1);


	template<typename Result>
	static int doCall(lua_State *L,const Arg1 &arg1,__func func,Int2Type<false>)
	{
		push_obj<Result>(L,func(arg1));
		return 1;
	}
	template<typename Result>
	static int doCall(lua_State *L,const Arg1 &arg1,__func func,Int2Type<true>)
	{
		func(arg1);
		return 0;
	}

};

//2参
template<typename Ret,typename Arg1,typename Arg2>
class funbinder<Ret(*)(Arg1,Arg2)>
{
public:

	static void pushfuctor(lua_State *L,const char *name,Ret(*_func)(Arg1,Arg2))
	{
		PUSH_FUNCTOR;
	}

	static int lua_cfunction(lua_State *L)
	{
		typename GetReplaceType<Arg2>::type tmp_arg2 = popvalue<typename GetReplaceType<Arg2>::type>(L);
		typename GetReplaceType<Arg1>::type tmp_arg1 = popvalue<typename GetReplaceType<Arg1>::type>(L);
		Arg1 arg1 = GetRawValue<typename GetReplaceType<Arg1>::type>(tmp_arg1);
		Arg2 arg2 = GetRawValue<typename GetReplaceType<Arg2>::type>(tmp_arg2);
		return doCall<Ret>(L,arg1,arg2,(__func)lua_touserdata(L,lua_upvalueindex(1)),Int2Type<isVoid<Ret>::is_Void>());
	}

private:
	typedef Ret(*__func)(Arg1,Arg2);

	template<typename Result>
	static int doCall(lua_State *L,const Arg1 &arg1,const Arg2 &arg2,__func func,Int2Type<false>)
	{
		push_obj<Result>(L,func(arg1,arg2));
		return 1;
	}

	template<typename Result>
	static int doCall(lua_State *L,const Arg1 &arg1,const Arg2 &arg2,__func func,Int2Type<true>)
	{
		func(arg1,arg2);
		return 0;
	}
};

//3参
template<typename Ret,typename Arg1,typename Arg2,typename Arg3>
class funbinder<Ret(*)(Arg1,Arg2,Arg3)>
{
public:

	static void pushfuctor(lua_State *L,const char *name,Ret(*_func)(Arg1,Arg2,Arg3))
	{
		PUSH_FUNCTOR;
	}

	static int lua_cfunction(lua_State *L)
	{
		typename GetReplaceType<Arg3>::type tmp_arg3 = popvalue<typename GetReplaceType<Arg3>::type>(L);
		typename GetReplaceType<Arg2>::type tmp_arg2 = popvalue<typename GetReplaceType<Arg2>::type>(L);
		typename GetReplaceType<Arg1>::type tmp_arg1 = popvalue<typename GetReplaceType<Arg1>::type>(L);
		Arg1 arg1 = GetRawValue<typename GetReplaceType<Arg1>::type>(tmp_arg1);
		Arg2 arg2 = GetRawValue<typename GetReplaceType<Arg2>::type>(tmp_arg2);
		Arg3 arg3 = GetRawValue<typename GetReplaceType<Arg3>::type>(tmp_arg3);
		return doCall<Ret>(L,arg1,arg2,arg3,(__func)lua_touserdata(L,lua_upvalueindex(1)),Int2Type<isVoid<Ret>::is_Void>());
	}

private:
	typedef Ret(*__func)(Arg1,Arg2,Arg3);

	template<typename Result>
	static int doCall(lua_State *L,const Arg1 &arg1,const Arg2 &arg2,const Arg3 &arg3,__func func,Int2Type<false>)
	{
		push_obj<Result>(L,func(arg1,arg2,arg3));
		return 1;
	}

	template<typename Result>
	static int doCall(lua_State *L,const Arg1 &arg1,const Arg2 &arg2,const Arg3 &arg3,__func func,Int2Type<true>)
	{
		func(arg1,arg2,arg3);
		return 0;
	}

};

//4参
template<typename Ret,typename Arg1,typename Arg2,typename Arg3,typename Arg4>
class funbinder<Ret(*)(Arg1,Arg2,Arg3,Arg4)>
{
public:

	static void pushfuctor(lua_State *L,const char *name,Ret(*_func)(Arg1,Arg2,Arg3,Arg4))
	{
		PUSH_FUNCTOR;
	}

	static int lua_cfunction(lua_State *L)
	{
		typename GetReplaceType<Arg4>::type tmp_arg4 = popvalue<typename GetReplaceType<Arg4>::type>(L);
		typename GetReplaceType<Arg3>::type tmp_arg3 = popvalue<typename GetReplaceType<Arg3>::type>(L);
		typename GetReplaceType<Arg2>::type tmp_arg2 = popvalue<typename GetReplaceType<Arg2>::type>(L);
		typename GetReplaceType<Arg1>::type tmp_arg1 = popvalue<typename GetReplaceType<Arg1>::type>(L);
		Arg1 arg1 = GetRawValue<typename GetReplaceType<Arg1>::type>(tmp_arg1);
		Arg2 arg2 = GetRawValue<typename GetReplaceType<Arg2>::type>(tmp_arg2);
		Arg3 arg3 = GetRawValue<typename GetReplaceType<Arg3>::type>(tmp_arg3);
		Arg4 arg4 = GetRawValue<typename GetReplaceType<Arg4>::type>(tmp_arg4);
		return doCall<Ret>(L,arg1,arg2,arg3,arg4,(__func)lua_touserdata(L,lua_upvalueindex(1)),Int2Type<isVoid<Ret>::is_Void>());
	}

private:
	typedef Ret(*__func)(Arg1,Arg2,Arg3,Arg4);
	template<typename Result>
	static int doCall(lua_State *L,const Arg1 &arg1,const Arg2 &arg2,const Arg3 &arg3,const Arg4 &arg4,__func func,Int2Type<false>)
	{
		push_obj<Result>(L,func(arg1,arg2,arg3,arg4));
		return 1;
	}

	template<typename Result>
	static int doCall(lua_State *L,const Arg1 &arg1,const Arg2 &arg2,const Arg3 &arg3,const Arg4 &arg4,__func func,Int2Type<true>)
	{
		func(arg1,arg2,arg3,arg4);
		return 0;
	}
};


template<typename FUNTOR>
void reg_cfun(lua_State *L,const char *name, FUNTOR _func)
{
	funbinder<FUNTOR>::pushfuctor(L,name,_func);
}

inline void check_call(lua_State *L,const char *funname)
{
	lua_getglobal(L,funname);
	if(LUA_TFUNCTION != lua_type(L,-1))
	{
		lua_pop(L,1);
		char str[512];
		snprintf(str,512,"lua中不存在函数%s",funname);
		std::string err(str);
		throw err;
	}
}

template<typename Ret>
Ret call(lua_State *L,const char *funname)
{
	check_call(L,funname);
	return doLuaCall<Ret>::doCall(L,0,0);
}

template<typename Ret,typename Arg1>
Ret call(lua_State *L,const char *funname,const Arg1 &arg1)
{
	check_call(L,funname);
	push_obj<Arg1>(L,arg1);
	return doLuaCall<Ret>::doCall(L,1,0);
}

template<typename Ret,typename Arg1,typename Arg2>
Ret call(lua_State *L,const char *funname,const Arg1 &arg1,const Arg2 &arg2)
{
	check_call(L,funname);
	push_obj<Arg1>(L,arg1);
	push_obj<Arg2>(L,arg2);
	return doLuaCall<Ret>::doCall(L,2,0);
}

template<typename Ret,typename Arg1,typename Arg2,typename Arg3>
Ret call(lua_State *L,const char *funname,const Arg1 &arg1,const Arg2 &arg2,const Arg3 &arg3)
{
	check_call(L,funname);
	push_obj<Arg1>(L,arg1);
	push_obj<Arg2>(L,arg2);
	push_obj<Arg3>(L,arg3);
	return doLuaCall<Ret>::doCall(L,3,0);
}

template<typename Ret,typename Arg1,typename Arg2,typename Arg3,typename Arg4>
Ret call(lua_State *L,const char *funname,const Arg1 &arg1,const Arg2 &arg2,const Arg3 &arg3,const Arg4 &arg4)
{
	check_call(L,funname);
	push_obj<Arg1>(L,arg1);
	push_obj<Arg2>(L,arg2);
	push_obj<Arg3>(L,arg3);
	push_obj<Arg4>(L,arg4);
	return doLuaCall<Ret>::doCall(L,4,0);
}
}
#endif