ObjPush.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/>.
*/	
#ifndef _OBJPUSH_H
#define _OBJPUSH_H

namespace luacpp{
//调用lua函数时参数压栈的抽象

template<typename T>
class objPush
{
public:
	objPush(lua_State *L,const T &arg)
	{
		_objpush(L,arg,Int2Type<IndexOf<SupportType,T>::value >= 0>());
	}
private:

	void _objpush(lua_State *L,const T &arg,Int2Type<false>)
	{
		if(!luaRegisterClass<typename pointerTraits<T>::PointeeType>::isRegister())
			lua_pushlightuserdata(L,(T*)&arg);
		else
			objUserData<typename pointerTraits<T>::PointeeType>::pushObj(L,&arg);		
	}
	
	void _objpush(lua_State *L,const T &arg,Int2Type<true>)
	{
		if((lua_Number)arg == (lua_Integer)arg)
			lua_pushinteger(L,(lua_Integer)arg);
		else
			lua_pushnumber(L,(lua_Number)arg);
	}	
};


//对指针类型的特化
template<typename T>
class objPush<T*>
{
public:
	objPush(lua_State *L,T* arg)
	{
		if(!arg)
			lua_pushnil(L);
		else
		{
			if(!luaRegisterClass<typename pointerTraits<T>::PointeeType>::isRegister())
				lua_pushlightuserdata(L,(T*)arg);
			else
				objUserData<typename pointerTraits<T>::PointeeType>::pushObj(L,arg);
		}
	}
};

template<typename T>
class objPush<const T*>
{
public:
	objPush(lua_State *L,const T* arg)
	{
		if(!arg)
			lua_pushnil(L);
		else
		{
			if(!luaRegisterClass<typename pointerTraits<T>::PointeeType>::isRegister())
				lua_pushlightuserdata(L,(T*)arg);
			else
				objUserData<typename pointerTraits<T>::PointeeType>::pushObj(L,arg);
		}
	}
};

//字符串的特化
template<>
class objPush<const char *>
{
public:
	objPush(lua_State *L,const char *value)
	{
		lua_pushstring(L,value);
	}
};

template<>
class objPush<char*>
{
public:
	objPush(lua_State *L,char *value)
	{
		lua_pushstring(L,value);
	}
};

//对std::string的特化
template<>
class objPush<std::string>
{
public:
	objPush(lua_State *L,const std::string &arg)
	{
		lua_pushlstring(L,arg.c_str(),arg.size());
	}
};

//针对luaObject的特化
template<>
class objPush<luaObject>
{
public:
	objPush(lua_State *L,const luaObject &arg)
	{
		lua_rawgeti(L,LUA_REGISTRYINDEX,arg.getIndex());
	}
};

template<>
class objPush<bool>
{
public:
	objPush(lua_State *L,bool arg)
	{
		lua_pushboolean(L,arg);
	}
};

//针对luatable的特化
template<>
class objPush<luatable>
{
public:
	objPush(lua_State *L,const luatable &arg)
	{
		if(arg.empty())
			lua_pushnil(L);
		else{
			lua_newtable(L);
			for(int i = 0; i < (int)arg.size(); ++i)
			{	
				if(arg[i].empty())
					lua_pushnil(L);
				else
					arg[i]._any_pusher->push(L,const_cast<any*>(&arg[i]));
				lua_rawseti(L,-2,i+1);
			}
		}
	}
};

template<typename T>
void push_obj(lua_State *L,const T &obj)
{
	objPush<T> _obj(L,obj);
}

template<typename T>
class concrete_any_pusher : public any_pusher
{
public:
	void push(lua_State *L,any *_any)
	{
		if((lua_Number)any_cast<T>(*_any) == (lua_Integer)any_cast<T>(*_any))
			lua_pushinteger(L,(lua_Integer)any_cast<T>(*_any));
		else
			lua_pushnumber(L,(lua_Number)any_cast<T>(*_any));
	}
};

template<typename T>
class concrete_any_pusher<T*> : public any_pusher
{
public:
	void push(lua_State *L,any *_any)
	{
		//是否注册的用户类型
		if(_any->luaRegisterClassName == "")
			objPush<void*> obj(L,any_cast<void*>(*_any));
		else
		{
			pushObj(L,any_cast<void*>(*_any),_any->luaRegisterClassName.c_str());
		}
	}
};

template<>
class concrete_any_pusher<bool> : public any_pusher
{
public:
	void push(lua_State *L,any *_any)
	{
		lua_pushboolean(L,any_cast<bool>(*_any));
	}
};

template<>
class concrete_any_pusher<std::string> : public any_pusher
{
public:
	void push(lua_State *L,any *_any)
	{
		lua_pushstring(L,any_cast<std::string>(*_any).c_str());
	}
};

template<>
class concrete_any_pusher<const char *> : public any_pusher
{
public:
	void push(lua_State *L,any *_any)
	{
		lua_pushstring(L,any_cast<const char *>(*_any));
	}
};

template<>
class concrete_any_pusher<char *> : public any_pusher
{
public:
	void push(lua_State *L,any *_any)
	{
		lua_pushstring(L,any_cast<char *>(*_any));
	}
};

template<>
class concrete_any_pusher<luaObject> : public any_pusher
{
public:
	void push(lua_State *L,any *_any)
	{
		objPush<luaObject> obj(L,any_cast<luaObject>(*_any));
	}
};

template<>
class concrete_any_pusher<luatable> : public any_pusher
{
public:
	void push(lua_State *L,any *_any)
	{
		objPush<luatable> obj(L,any_cast<luatable>(*_any));
	}
};


template <typename T>
any_pusher *create_any_pusher()
{
	return new concrete_any_pusher<T>;
}

template<typename T>
inline void luaSetGlobal(lua_State *L,const char *name,const T &arg)
{
	lua_getglobal(L,"_G");
	if(!lua_istable(L, -1))
	{
		lua_pop(L,1);
		lua_newtable(L);
		lua_pushvalue(L,-1);
		lua_setglobal(L,"_G");
	}
	lua_pushstring(L,name);
	objPush<T>(L,arg);
	lua_settable(L, -3);
	lua_pop(L,1);
}

//set lua global object
inline void Set(lua_State *L,const char *name,const char *arg)
{
	lua_getglobal(L,"_G");
	if(!lua_istable(L, -1))
	{
		lua_pop(L,1);
		lua_newtable(L);
		lua_pushvalue(L,-1);
		lua_setglobal(L,"_G");
	}
	lua_pushstring(L,name);
	lua_pushstring(L,arg);
	lua_settable(L, -3);
	lua_pop(L,1);
}

inline void luaSetGlobal(lua_State *L,const char *name,char *arg)
{
	luaSetGlobal(L,name,(const char*)arg);
}

}

#endif