您的位置:首页 > 编程语言 > Lua

神一般的狙击Lua的恶心操作(写个小小滴应用逻辑还要管出入栈这些恶心细节, 太二了,故哥决定干掉)

2013-05-17 22:32 429 查看
/********************************************************************
author  :   Clark/陈泽丹
created :   2013-5-8
purpose :   反射辅助类
*********************************************************************/

#pragma once

//反射类型
struct RefClassType
{
//获取类型
virtual long GetType() = 0;
};

//获得反射类型
template< class Base, int TYPE_ID >
struct GetClassTypeEx: public Base
{
//获取类型
virtual long GetType(){ return TYPE_ID; }
};

//实例ID
struct RefClassID
{
//获取类型
virtual long GetID() = 0;
};

//获得类型ID
template< class Base >
struct GetClassIDEx: public Base
{
public:
GetClassIDEx(const long _id):ID(_id){}
//获取类型
virtual long GetID(){ return ID; }

private:
const long ID;
};


/********************************************************************
author  :   Clark/陈泽丹
created :   2013-5-8
purpose :   TemplateHelper
*********************************************************************/
#pragma once

//构造一个不造成开销的简单结构以保存类型
template< class _Type >
struct SimpleTypeC
{
typedef _Type Result;
};

template< int v >
struct SimpleTypeI
{
enum { Result = v };
};


/********************************************************************
author  :   Clark/陈泽丹
created :   2013-5-8
purpose :   LK_Typelist
*********************************************************************/
#pragma once

//-----------------------------------------------------------------------------------------------------------------------------
#define TL_1(T1) \
LK_Typelist<T1, LK_NULL_TYPE>

#define TL_2(T1, T2) \
LK_Typelist<T1, TL_1(T2) >

#define TL_3(T1, T2, T3) \
LK_Typelist<T1, TL_2(T2, T3) >

#define TL_4(T1, T2, T3, T4) \
LK_Typelist<T1, TL_3(T2, T3, T4) >

#define TL_5(T1, T2, T3, T4, T5) \
LK_Typelist<T1, TL_4(T2, T3, T4, T5) >

#define TL_6(T1, T2, T3, T4, T5, T6) \
LK_Typelist<T1, TL_5(T2, T3, T4, T5, T6) >

#define TL_7(T1, T2, T3, T4, T5, T6, T7) \
LK_Typelist<T1, TL_6(T2, T3, T4, T5, T6, T7) >

#define TL_8(T1, T2, T3, T4, T5, T6, T7, T8) \
LK_Typelist<T1, TL_7(T2, T3, T4, T5, T6, T7, T8) >

//-----------------------------------------------------------------------------------------------------------------------------
//参数队列
struct LK_NULL_TYPE{};
template <class T, class U>
struct LK_Typelist
{
typedef T Head;
typedef U Tail;
};

//获得参数队列长度
template< class TList >
struct LK_GetLength
{
enum { Ret = 1 + LK_GetLength< TList::Tail >::Ret };
};
template<>
struct LK_GetLength<LK_NULL_TYPE>
{
enum { Ret = 0 };
};

//获得类型
template< class TList, unsigned int index >
struct LK_TypeAt;
template< class Head, class Tail >
struct LK_TypeAt< LK_Typelist<Head, Tail>, 0 >
{
typedef Head Result;
};
template< class Head, class Tail, unsigned int i >
struct LK_TypeAt< LK_Typelist<Head, Tail>, i >
{
typedef typename LK_TypeAt<Tail, i - 1>::Result Result;
};


/********************************************************************
author  :   Clark/陈泽丹
created :   2013-5-8
purpose :   注册C函数进LUA并封装相关调用(原Lua开发包调用还是太麻烦了,哥果断呤唱“诸神黄昏”大招!)
*********************************************************************/
#pragma once
#include "LK_TypeList.h"
#include "TemplateHelper.h"
extern "C"
{
#include <lua.h>
#include <lauxlib.h>
#include <lualib.h>
}

//-----------------------------------------------------------------------------------------------------------------------------
template< class _Ret, class _TypeList, int >
struct PopPar2CAction{};

//0参
template<class _Ret, class _TypeList>
struct PopPar2CAction<_Ret, _TypeList, 0>
{
template< class _Fun >
_Ret operator()(lua_State *L, _Fun fun) { return fun(); }
};

//1参
#define INIT_LUA_VL_1(LTYP_LIST, I, _L) \
LK_TypeAt< LTYP_LIST, I >::Result lua_stack_val_1 = NULL; \
Lua_GetPar(lua_stack_val_1, _L, I+1);
#define GET_LUA_VL_1() \
lua_stack_val_1
template<class _Ret, class _TypeList >
struct PopPar2CAction<_Ret, _TypeList, 1>
{
template< class _Fun >
_Ret operator()(lua_State *L, _Fun fun) { INIT_LUA_VL_1( _TypeList, 0, L ); return fun( GET_LUA_VL_1() ); }
};

//2参
#define INIT_LUA_VL_2(LTYP_LIST, I, _L) \
LK_TypeAt< LTYP_LIST, I >::Result lua_stack_val_2 = NULL; \
Lua_GetPar(lua_stack_val_2, _L, I+1); \
INIT_LUA_VL_1(LTYP_LIST, I+1, _L)
#define GET_LUA_VL_2() \
lua_stack_val_2, GET_LUA_VL_1()
template<class _Ret, class _TypeList >
struct PopPar2CAction<_Ret, _TypeList, 2>
{
template< class _Fun >
_Ret operator()(lua_State *L, _Fun fun)
{
INIT_LUA_VL_2( _TypeList, 0, L );
return fun( GET_LUA_VL_2() );
}
};

//3参
#define INIT_LUA_VL_3(LTYP_LIST, I, _L) \
LK_TypeAt< LTYP_LIST, I >::Result lua_stack_val_3 = NULL; \
Lua_GetPar(lua_stack_val_3, _L, I+1); \
INIT_LUA_VL_2(LTYP_LIST, I+1, _L)
#define GET_LUA_VL_3() \
lua_stack_val_3, GET_LUA_VL_2()
template<class _Ret, class _TypeList >
struct PopPar2CAction<_Ret, _TypeList, 3>
{
template< class _Fun >
_Ret operator()(lua_State *L, _Fun fun) { INIT_LUA_VL_3( _TypeList, 0, L ); return fun( GET_LUA_VL_3() ); }
};

//4参
#define INIT_LUA_VL_4(LTYP_LIST, I, _L) \
LK_TypeAt< LTYP_LIST, I >::Result lua_stack_val_4 = NULL; \
Lua_GetPar(lua_stack_val_4, _L, I+1); \
INIT_LUA_VL_3(LTYP_LIST, I+1, _L)
#define GET_LUA_VL_4() \
lua_stack_val_4, GET_LUA_VL_3()
template<class _Ret, class _TypeList >
struct PopPar2CAction<_Ret, _TypeList, 4>
{
template< class _Fun >
_Ret operator()(lua_State *L, _Fun fun) { INIT_LUA_VL_4( _TypeList, 0, L ); return fun( GET_LUA_VL_4() ); }
};

//5参
#define INIT_LUA_VL_5(LTYP_LIST, I, _L) \
LK_TypeAt< LTYP_LIST, I >::Result lua_stack_val_5 = NULL; \
Lua_GetPar(lua_stack_val_5, _L, I+1); \
INIT_LUA_VL_4(LTYP_LIST, I+1, _L)
#define GET_LUA_VL_5() \
lua_stack_val_5, GET_LUA_VL_4()
template<class _Ret, class _TypeList >
struct PopPar2CAction<_Ret, _TypeList, 5>
{
template< class _Fun >
_Ret operator()(lua_State *L, _Fun fun) { INIT_LUA_VL_5( _TypeList, 0, L ); return fun( GET_LUA_VL_5() ); }
};

//6参
#define INIT_LUA_VL_6(LTYP_LIST, I, _L) \
LK_TypeAt< LTYP_LIST, I >::Result lua_stack_val_6 = NULL; \
Lua_GetPar(lua_stack_val_6, _L, I+1); \
INIT_LUA_VL_5(LTYP_LIST, I+1, _L)
#define GET_LUA_VL_6() \
lua_stack_val_6, GET_LUA_VL_5()
template<class _Ret, class _TypeList >
struct PopPar2CAction<_Ret, _TypeList, 6>
{
template< class _Fun >
_Ret operator()(lua_State *L, _Fun fun) { INIT_LUA_VL_6( _TypeList, 0, L ); return fun( GET_LUA_VL_6() ); }
};

//7参
#define INIT_LUA_VL_7(LTYP_LIST, I, _L) \
LK_TypeAt< LTYP_LIST, I >::Result lua_stack_val_7 = NULL; \
Lua_GetPar(lua_stack_val_7, _L, I+1); \
INIT_LUA_VL_6(LTYP_LIST, I+1, _L)
#define GET_LUA_VL_7() \
lua_stack_val_7, GET_LUA_VL_6()
template<class _Ret, class _TypeList >
struct PopPar2CAction<_Ret, _TypeList, 7>
{
template< class _Fun >
_Ret operator()(lua_State *L, _Fun fun) { INIT_LUA_VL_7( _TypeList, 0, L ); return fun( GET_LUA_VL_7() ); }
};

//8参
#define INIT_LUA_VL_8(LTYP_LIST, I, _L) \
LK_TypeAt< LTYP_LIST, I >::Result lua_stack_val_8 = NULL; \
Lua_GetPar(lua_stack_val_8, _L, I+1); \
INIT_LUA_VL_7(LTYP_LIST, I+1, _L)
#define GET_LUA_VL_8() \
lua_stack_val_8, GET_LUA_VL_7()
template<class _Ret, class _TypeList >
struct PopPar2CAction<_Ret, _TypeList, 8>
{
template< class _Fun >
_Ret operator()(lua_State *L, _Fun fun) { INIT_LUA_VL_8( _TypeList, 0, L ); return fun( GET_LUA_VL_8() ); }
};

//---------------------------------- 预判类型 ----------------------------------
template<class _Type>
struct Lua_IsType
{
bool operator()(lua_State *L, int pos){ return false; }
};
template<>
struct Lua_IsType< double >
{
bool operator()(lua_State *L, int pos){ return ( lua_isnumber(L, pos)?true:false ); }
};
template<>
struct Lua_IsType< long >
{
bool operator()(lua_State *L, int pos){ return ( lua_isnumber(L, pos)?true:false ); }
};
template<>
struct Lua_IsType< int >
{
bool operator()(lua_State *L, int pos){ return ( lua_isnumber(L, pos)?true:false ); }
};
template<>
struct Lua_IsType< char* >
{
bool operator()(lua_State *L, int pos){ return ( lua_isstring(L, pos)?true:false ); }
};

//---------------------------------- 获得Lua传出来的参数 ----------------------------------
static bool Lua_GetPar(char*& _val, lua_State* _L, int _index)
{
_val =  ((char*) lua_tostring(_L,_index));
return true;
}
static bool Lua_GetPar(long& _val, lua_State* _L, int _index)
{
_val =  ((long) lua_tonumber(_L,_index));
return true;
}
static bool Lua_GetPar(bool& _val, lua_State* _L, int _index)
{
_val =   lua_tonumber(_L, _index) != 0;
return true;
}
static bool Lua_GetPar(int& _val, lua_State* _L, int _index)
{
_val =  ((int) lua_tonumber(_L,_index));
return true;
}
static bool Lua_GetPar(double& _val, lua_State* _L, int _index)
{
_val =  ((int) lua_tonumber(_L,_index));
return true;
}

//---------------------------------- 压入Lua的参数 ----------------------------------
template< class _Type >
static void Lua_PushPar(lua_State* _L, _Type _val)
{
lua_pushnumber(_L, (lua_Number)_val);
}
static void Lua_PushPar(lua_State* _L, const char* _val)
{
if (_val == NULL)
lua_pushnil(_L);
else
lua_pushstring(_L,_val);
}

//---------------------------------- 检查压栈类型 ----------------------------------
template< class _TypeList >
class CheckLuaType
{
protected:
template< class TList >
struct CheckAllType;
template<>
struct CheckAllType<LK_NULL_TYPE>
{
bool operator()(lua_State *L, int index = 1)
{
return true;
}
};
template <class T, class U>
struct CheckAllType< LK_Typelist<T, U> >
{
bool operator()(lua_State *L, int index = 1)
{
if( !Lua_IsType<T>()(L, index) )
return false;
return CheckAllType<U>()(L, index+1);
}
};
};

//---------------------------------- 弹出栈数据并装载进C函数和运行C函数 ----------------------------------
template< class _Ret, class _TypeList >
struct Par2CAction: public CheckLuaType<_TypeList>
{
template< class _Fun >
_Ret operator()(lua_State *L, _Fun fun)
{
#ifdef _DEBUG
//读取并判读参数
if(lua_gettop(L) != LK_GetLength<_TypeList>::Ret )
{
lua_pushstring(L, "Incorrect argument number!");
lua_error(L);
}
if(!CheckAllType< _TypeList >()(L))
{
lua_pushstring(L, "Incorrect argument!");
lua_error(L);
}
#endif
PopPar2CAction<_Ret, _TypeList, LK_GetLength<_TypeList>::Ret> run;
_Ret ret = run(L, fun);
Lua_PushPar(L,ret);
return ret;
}
};
template< class _TypeList >
struct Par2CAction<void, _TypeList>: public CheckLuaType<_TypeList>
{
template< class _Fun >
void operator()(lua_State *L, _Fun fun)
{
#ifdef _DEBUG
//读取并判读参数
if(lua_gettop(L) != LK_GetLength<_TypeList>::Ret )
{
lua_pushstring(L, "Incorrect argument number!");
lua_error(L);
}
if( !CheckAllType< _TypeList >()(L))
{
lua_pushstring(L, "Incorrect argument!");
lua_error(L);
}
#endif
PopPar2CAction<void, _TypeList, LK_GetLength<_TypeList>::Ret> run;
run(L, fun);
Lua_PushPar(L,0);
}
};

//注册C函数到LUA里
template< int v, class _Fun, class _Stack >
class RegApiForLua
{
public:
static void Register(lua_State *L, const char* _lua_fun_name, _Fun _c_fun)
{
mp_c_fun = _c_fun;
lua_register( L, _lua_fun_name, (&RegApiForLua<v,_Fun,_Stack>::API) );
}

private:
static int API(lua_State *L)
{
m_par2caction(L, *mp_c_fun);
return 1;
}
static _Stack	m_par2caction;
static _Fun		mp_c_fun;
};

template< int v, class _Fun, class _Stack >
_Stack RegApiForLua<v,_Fun,_Stack>::m_par2caction;
template< int v, class _Fun, class _Stack >
_Fun RegApiForLua<v,_Fun,_Stack>::mp_c_fun;
/*
RegApiForLua<0, decltype(test), Par2CAction<void, TL_1(int)> >::Register(L, "C_API", test);
RegApiForLua<1, decltype(test1), Par2CAction<int, TL_2(char*, int)> >::Register(L, "C_API1", test1);
*/

//注册一个C函数到LUA里
template< class _Sign, class _Fun, class _Stack >
void RegisterCAPI( _Sign _sign_i, _Stack _simple_c, lua_State *L, const char* _lua_fun_name, _Fun _fun )
{
RegApiForLua<_Sign::Result, _Fun, _Stack::Result >::Register(L, _lua_fun_name, _fun);
}
/*
RegisterCAPI(SimpleTypeI<0>(), SimpleTypeC< Par2CAction<void, TL_1(int)> >(), L, "C_API", test );
RegisterCAPI(SimpleTypeI<1>(), SimpleTypeC< Par2CAction<int, TL_2(char*, int)> >(), L, "C_API1", test1 );
*/


/********************************************************************
author  :   Clark/陈泽丹
created :   2013-5-8
purpose :   调用Lua函数
*********************************************************************/
#pragma once
#include "RefClassHelper.h"
#include <Windows.h>
#include <vector>
extern "C"
{
#include <lua.h>
#include <lauxlib.h>
#include <lualib.h>
}

using namespace std;

// 数字类型
struct LuaNumber : public GetClassTypeEx<RefClassType, 1>
{
LuaNumber(){ val = 0;}
LuaNumber( double _val ){ val = _val; }
void operator = (double _val){ val = _val; }
double val;
};

// 字符串类型
struct LuaString : public GetClassTypeEx<RefClassType, 2>
{
LuaString(){ val[0] = 0; }
LuaString( const char*_val ){ val[0] = 0; if( NULL != _val ) lstrcpyn(val, _val, sizeof(val)); }
void operator = (const char *_val){ val[0] = 0; if( NULL != _val ) lstrcpyn(val, _val, sizeof(val)); }
char val[1024];
};

// 调用一个函数
bool CallLFun(lua_State *_L, const char* _lua_fun_name, vector< RefClassType* >* _in_pars = nullptr, vector< RefClassType* >* _out_pars = nullptr)
{
//调用函数和参数
lua_getglobal(_L, _lua_fun_name);
int in_len = ( nullptr != _in_pars) ?  _in_pars->size() : 0;
int out_len = ( nullptr != _out_pars) ?  _out_pars->size() : 0;
if( nullptr != _in_pars)
{
in_len = _in_pars->size();
for( int i=0; i<in_len; i++)
{
if( 1 == (*_in_pars)[i]->GetType() )
lua_pushnumber(_L, ((LuaNumber*)(*_in_pars)[i])->val);
else
lua_pushstring(_L, ((LuaString*)(*_in_pars)[i])->val);
}
}
if( 0 != lua_pcall(_L, in_len, out_len, 0) ){ goto FUN_ERROR; }
//判读返回参数合法性
if( out_len != lua_gettop(_L) ) { goto FUN_ERROR; }
if( nullptr != _out_pars )
{
//获取参数
for (int i = out_len - 1; i >= 0; i--)
{
if( 1 == (*_out_pars)[i]->GetType())
{
if (!lua_isnumber(_L, -1)){ goto FUN_ERROR;}
*((LuaNumber*)(*_out_pars)[i]) = (double)lua_tonumber(_L, -1);
lua_pop(_L, 1);
}
else
{
if (!lua_isstring(_L, -1)){ goto FUN_ERROR;}
*((LuaString*)(*_out_pars)[i]) = (const char *)lua_tostring(_L, -1);
lua_pop(_L, 1);
}
}
}
return true;

FUN_ERROR:
//printf("Error in %s", _lua_fun_name);
lua_settop(_L, 0);
return false;
}

/*
vector<RefClassType*> in_pars;
vector<RefClassType*> out_pars;
LuaString val1("1");
in_pars.push_back(&val1);
LuaString val2("2");
in_pars.push_back(&val2);
LuaNumber oval1(0);
out_pars.push_back(&oval1);
LuaString oval2("");
out_pars.push_back(&oval2);
*/


#include <iostream>
#include "RegCFun.h"
#include "CallLFun.h"

using namespace std;
#pragma comment(lib,"lua5.1.lib")

void test( int p1 )
{
printf("test: %d\n", p1);
}

int test1( const char* p1, int p2 )
{
printf("test1: %s, %d\n", p1, p2);
return p2*2;
}

int main()
{
lua_State *L = lua_open();
luaL_openlibs(L);

RegisterCAPI(SimpleTypeI<0>(), SimpleTypeC< Par2CAction<void, TL_1(int)> >(), L, "C_API", test );
RegisterCAPI(SimpleTypeI<1>(), SimpleTypeC< Par2CAction<int, TL_2(char*, int)> >(), L, "C_API1", test1 );

vector<RefClassType*> in_pars;
vector<RefClassType*> out_pars;
LuaString val1("C_par1");
in_pars.push_back(&val1);
LuaString val2("C_par2");
in_pars.push_back(&val2);
LuaNumber oval1;
out_pars.push_back(&oval1);
LuaString oval2;
out_pars.push_back(&oval2);

if (luaL_dofile(L, "test.lua"))
nullptr;
CallLFun(L, "LuaMain", &in_pars, &out_pars);
lua_close(L);

system("pause");
}


function LuaMain(p1, p2)
print("p1", p1, "p2", p2)
C_API(1000)
local ret = C_API1("call C_API1", 5)
print( ret )
return ret, "LuaMain 的返回值"
end
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: