首页 > 代码库 > Lua脚本和C++交互(三)
Lua脚本和C++交互(三)
前两篇文章中介绍了C++调用lua、lua栈操作的一些相关知识。
下面说一下Lua的工具。我们下一步要用到其中的一个帮助我们的开发,其实,Lua里面有很多简化开发的工具,你可以去www.sourceforge.net去找一下。它们能够帮助你简化C++对象与Lua对象互转之间的代码。这里说几个有名的,当然可能不全。
(lua tinker)如果你的系统在windows下,而且不考虑移植,那么我强烈推荐你去下载一个叫做lua tinker的小工具,整个工具非常简单,一个.h和一个.cpp。直接就可以引用到你的工程中,连独立编译都不用,这是一个韩国人写的Lua与 C++接口转换的类,十分方便,代码简洁(居家旅行,必备良药)。它是基于模板的,所以你可以很轻松的把你的C++对象绑定到Lua中。代码较长,呵呵,有兴趣的朋友可以给我留言索要lua tinker的例子。就不贴在这里了。不过我个人不推荐这个东西,因为它在Linux下是编译不过去的。它使用了一种g++不支持的模板写法,虽然有人在尝试把它修改到Linux下编译,但据我所知,修改后效果较好的似乎还没有。不过如果你只是在 windows下,那就没什么可犹豫的,强烈推荐,你会喜欢它的。
(Luabinder)相信用过Boost库的朋友,或许对这个家伙很熟悉。它是一个很强大的Linux下Lua扩展包,帮你封装了很多Lua的复杂操作,主要解决了绑定C++对象和Lua对象互动的关系,非常强大,不过嘛,对于freeeyes而言,还是不推荐,因为freeeyes很懒,不想为了一个Lua还要去编译一个庞大的boost库,当然,见仁见智,如果你的程序本身就已经加载了boost,那么就应该毫不犹豫的选择它。
(lua++)呵呵,这是我最喜欢,也是我一直用到现在的库,比较前两个而言,lua++的封装性没有那么好,很多东西还是需要一点代码的,不过之所以我喜欢,是因为它是用C写的,可以在windows下和linux下轻松转换。如果鱼与熊掌不能兼得,那么我宁愿选择一个兼顾两者的东西,如果有的话,呵呵。当然,lua++就是这么一个东西,如果你继续看我的文章,或许你也会喜欢它的。
这里我们选择lua++作为我们继续进行下去的垫脚石吧。说到Lua++(http://www.codenix.com/~tolua/),这个东西还是挺有渊源的,请你先下载一个。我教你怎么编译。下面是我下载的lua++的目录结构:
还记得我昨天说过如何编译Lua么,现在请你再做一遍,不同的是,请把lua++的程序包中的src/lib中的所有h和cpp,还有include下的那个.h拷贝到你上次建立的lua工程中。最后记得把tolua++.h放在你的Include文件夹下。下图为整个项目的目录结构(最终运行成功的目录结构):
行了,我们把上次CLuaFn类稍微改一下。
extern "C"{ #include "lua.h" #include "lualib.h" #include "lauxlib.h" #include "tolua++.h" //这里加一行 }; #include "_ParamData.h"#include "CTest.h"class CLuaFn{public: CLuaFn(void); ~CLuaFn(void); void Init(); //初始化Lua对象指针参数 void Close(); //关闭Lua对象指针 bool LoadLuaFile(const char* pFileName); //加载指定的Lua文件 bool CLuaFn::CallFileFn(const char* pFunctionName, CParamGroup& ParamIn, CParamGroup& ParamOut); //执行指定Lua文件中的函数 //参数入栈 bool CLuaFn::PushLuaData(lua_State* pState, _ParamData* pParam); //参数出栈 bool CLuaFn:: PopLuaData(lua_State* pState, _ParamData* pParam, int nIndex); //新对象的创建、销毁 static int tolua_new_CTest(lua_State* pState); static int tolua_delete_CTest(lua_State* pState) ; static int tolua_SetData_CTest(lua_State* pState); static int tolua_GetData_CTest(lua_State* pState); bool InitClass(); private: lua_State* m_pState; //这个是Lua的State对象指针,你可以一个Lua文件对应一个};
行了,这样我们就能用Lua++下的功能了。
昨天,大家看到了 bool CallFileFn(const char* pFunctionName, int nParam1, int nParam2);这个函数的运用。演示了真么调用Lua函数。
下面,我改一下,这个函数。为什么?还是因为freeeyes很懒,我可不想每有一个函数,我都要写一个C++函数去调用,太累!我要写一个通用的!支持任意函数调用的接口!
于是我创建了两个类。支持任意参数的输入和输出,并打包送给lua去执行
//支持任意参数的输入和输出,并打包送给lua去执行#ifndef _PARAMDATA_H #define _PARAMDATA_H #include <vector> #define MAX_PARAM_200 200 using namespace std; struct _ParamData { public: void* m_pParam; char m_szType[MAX_PARAM_200]; int m_TypeLen; public: _ParamData() { m_pParam = NULL; m_szType[0] = ‘\0‘; m_TypeLen = 0; }; _ParamData(void* pParam, const char* szType, int nTypeLen) { SetParam(pParam, szType, nTypeLen); } ~_ParamData() {}; void SetParam(void* pParam, const char* szType, int nTypeLen) { m_pParam = pParam; sprintf(m_szType, "%s", szType); m_TypeLen = nTypeLen; }; bool SetData(void* pParam, int nLen) { if(m_TypeLen < nLen) { return false; } if(nLen > 0) { memcpy(m_pParam, pParam, nLen); } else { memcpy(m_pParam, pParam, m_TypeLen); } return true; } void* GetParam() { return m_pParam; } const char* GetType() { return m_szType; } bool CompareType(const char* pType) { if(0 == strcmp(m_szType, pType)) { return true; } else { return false; } } }; class CParamGroup { public: CParamGroup() {}; ~CParamGroup() { Close(); }; void Init() { m_vecParamData.clear(); }; void Close() { for(int i = 0; i < (int)m_vecParamData.size(); i++) { _ParamData* pParamData =http://www.mamicode.com/ m_vecParamData[i]; delete pParamData; pParamData = NULL; } m_vecParamData.clear(); }; void Push(_ParamData* pParam) { if(pParam != NULL) { m_vecParamData.push_back(pParam); } }; _ParamData* GetParam(int nIndex) { if(nIndex < (int)m_vecParamData.size()) { return m_vecParamData[nIndex]; } else { return NULL; } }; int GetCount() { return (int)m_vecParamData.size(); } private: typedef vector<_ParamData*> vecParamData; vecParamData m_vecParamData; }; #endif
我创建了两个类,把Lua要用到的类型,数据都封装起来了。这样,我只需要这么改写这个函数。
bool CallFileFn(const char* pFunctionName, CParamGroup& ParamIn, CParamGroup& ParamOut);
它就能按照不同的参数自动给我调用,嘿嘿,懒到家吧!
其实这两个类很简单,_ParamData是参数类,把你要用到的参数放入到这个对象中去,标明类型的大小,类型名称,内存块。而CParamGroup负责将很多很多的_ParamData打包在一起,放在vector里面。
好了,让我们看看CallFileFn函数里面我怎么改的。
#include "CLuaFn.h"CLuaFn::CLuaFn(void){m_pState = NULL;}; //这句干嘛CLuaFn::~CLuaFn(void){};//初始化函数void CLuaFn::Init(){ if (NULL == m_pState) { m_pState = lua_open(); //返回一个lua对象指针 luaL_openlibs(m_pState); //加载了所有你可能用到的Lua基本库 }}//关闭Lua对象并释放资源void CLuaFn::Close(){ if(NULL != m_pState) { lua_close(m_pState); m_pState = NULL; }}bool CLuaFn::LoadLuaFile(const char* pFileName){ int nRet = 0; if (NULL == m_pState) { printf("[CLuaFn::LoadLuaFile]m_pState is NULL.\n"); return false; } //加载文件的时候尽量放在程序的初始化中 nRet = luaL_dofile(m_pState, pFileName); if (nRet != 0) { printf("[CLuaFn::LoadLuaFile]luaL_loadfile(%s) is file(%d)(%s).\n", pFileName, nRet, lua_tostring(m_pState, -1)); return false; } return true;}//参数入栈bool CLuaFn::PushLuaData(lua_State* pState, _ParamData* pParam) { if(pParam == NULL) { return false; } if(pParam->CompareType("string")) { lua_pushstring(m_pState, (char* )pParam->GetParam()); return true; } if(pParam->CompareType("int")) { int* nData = http://www.mamicode.com/(int* )pParam->GetParam(); lua_pushnumber(m_pState, *nData); return true; } else { void* pVoid = pParam->GetParam(); // tolua_pushusertype(m_pState, pVoid, pParam->GetType()); return true; } } // 参数出栈bool CLuaFn:: PopLuaData(lua_State* pState, _ParamData* pParam, int nIndex) { if(pParam == NULL) { return false; } if(pParam->CompareType("string")) { if (lua_isstring(m_pState, nIndex) == 1) { const char* pData = http://www.mamicode.com/(const char*)lua_tostring(m_pState, nIndex); pParam->SetData((void* )pData, (int)strlen(pData)); } return true; } if(pParam->CompareType("int")) { if (lua_isnumber(m_pState, nIndex) == 1) { int nData = http://www.mamicode.com/(int)lua_tonumber(m_pState, nIndex); pParam->SetData(&nData, sizeof(int)); } return true; } else { pParam->SetData(tolua_tousertype(m_pState, nIndex, NULL), -1); return true; } } //调用函数bool CLuaFn::CallFileFn(const char* pFunctionName, CParamGroup& ParamIn, CParamGroup& ParamOut) { int nRet = 0; int i = 0; if(NULL == m_pState) { printf("[CLuaFn::CallFileFn]m_pState is NULL.\n"); return false; } lua_getglobal(m_pState, pFunctionName); //加载输入参数 for(i = 0; i < ParamIn.GetCount(); i++) { PushLuaData(m_pState, ParamIn.GetParam(i)); } nRet = lua_pcall(m_pState, ParamIn.GetCount(), ParamOut.GetCount(), 0); if (nRet != 0) { printf("[CLuaFn::CallFileFn]call function(%s) error(%s).\n", pFunctionName, lua_tostring(m_pState, -1)); return false; } //获得输出参数 int nPos = 0; for(i = ParamOut.GetCount() - 1; i >= 0; i--) { nPos--; PopLuaData(m_pState, ParamOut.GetParam(i), nPos); } int nCount = lua_gettop(m_pState); //根据返回参数的个数重新设置栈顶, 这样做可以返回任意数量的栈而且清除干净 lua_settop(m_pState, -1-ParamOut.GetCount()); return true; } /* ************************************************************* */int CLuaFn::tolua_new_CTest(lua_State* pState) { CTest* pTest = new CTest(); //将一个已经在Lua注册的"CTest"对象指针,压入数据栈 tolua_pushusertype(pState, pTest, "CTest"); return 1; } int CLuaFn::tolua_delete_CTest(lua_State* pState) { //将数据栈下的对象以(CTest* )的指针形式弹出来 CTest* pTest = (CTest* )tolua_tousertype(pState, 1, 0); if(NULL != pTest) { delete pTest; } return 1; } /*tolua_SetData_CTest()函数和tolua_GetData_CTest分别对应CTest的SetData方法和GetData()方法。 因为我们的SetData方法里面存在变量,那么同样,我们需要使用const char* pData = http://www.mamicode.com/tolua_tostring(pState, 2, 0);将参数弹出来,>*/int CLuaFn::tolua_SetData_CTest(lua_State* pState) { CTest* pTest = (CTest* )tolua_tousertype(pState, 1, 0); const char* pData = http://www.mamicode.com/tolua_tostring(pState, 2, 0); if(pData != NULL && pTest != NULL) { pTest->SetData(pData); } return 1; } int CLuaFn::tolua_GetData_CTest(lua_State* pState) { CTest* pTest = (CTest* )tolua_tousertype(pState, 1, 0); if(pTest != NULL) { char* pData = http://www.mamicode.com/pTest->GetData(); tolua_pushstring(pState, pData); } return 1; } /* ************************************************************* */bool CLuaFn::InitClass() { if(NULL == m_pState) { printf("[CLuaFn::InitClass]m_pState is NULL.\n"); return false; } tolua_open(m_pState); tolua_module(m_pState, NULL, 0); tolua_beginmodule(m_pState, NULL); tolua_usertype(m_pState, "CTest"); tolua_cclass(m_pState, "CTest", "CTest", "", tolua_delete_CTest); //只注册一个模块,比如,我们管CTest叫做"CTest",保持和C++的名称一样。这样在Lua的对象库中就会多了一个CTest的对象描述,等同于string,number等等基本类型 tolua_beginmodule(m_pState, "CTest"); tolua_function(m_pState, "new", tolua_new_CTest); //将Lua里面CTest对象的"SetData"绑定到你的tolua_SetData_CTest()函数中去 tolua_function(m_pState, "SetData", tolua_SetData_CTest); tolua_function(m_pState, "GetData", tolua_GetData_CTest); //tolua_beginmodule()和tolua_endmodule()对象必须成对出现,如果出现不成对的,你注册的C++类型将会失败 tolua_endmodule(m_pState); tolua_endmodule(m_pState); return true; }
下面来看一个类(头文件)。假设我们要把这个对象,传输给Lua进行调用。
#ifndef _TEST_H #define _TEST_H #include<stdio.h>class CTest { public: CTest(void); ~CTest(void); char* GetData(); void SetData(const char* pData); private: char m_szData[200]; }; #endif
这个类里面有两个函数,一个是GetData(),一个是SetData(),之所以这么写,我要让Lua不仅能使用我的类,还可以给这个类使用参数。
那么,cpp文件,我们姑且这样写。(当然,你可以进行修改,按照你喜欢的方式写一个方法,呵呵)
/*这是一个标准的类,我需要这个类在Lua里面可以创造出来,并赋予数值, 甚至我可以把CTest作为一个Lua函数参数,传给Lua函数让它去给我处理*/#include "CTest.h"CTest::CTest(void){}; CTest::~CTest(void){}; char* CTest::GetData(){ printf("[CTest::GetData]%s.\n", m_szData); return m_szData;}void CTest::SetData(const char* pData){ sprintf(m_szData, "%s", pData);}
这是一个标准的类,我需要这个类在Lua里面可以创造出来,并赋予数值,甚至我可以把CTest作为一个Lua函数参数,传给Lua函数让它去给我处理。让我们来看看怎么做。如果使用标准的Lua语法,有点多,所以我就借用一下上次提到的tolua来做到这一切,我一句句的解释。姑且我们把这些代码放在LuaFn.cpp里面。
/* ************************************************************* */int CLuaFn::tolua_new_CTest(lua_State* pState) { CTest* pTest = new CTest(); //将一个已经在Lua注册的"CTest"对象指针,压入数据栈 tolua_pushusertype(pState, pTest, "CTest"); return 1; } int CLuaFn::tolua_delete_CTest(lua_State* pState) { //将数据栈下的对象以(CTest* )的指针形式弹出来 CTest* pTest = (CTest* )tolua_tousertype(pState, 1, 0); if(NULL != pTest) { delete pTest; } return 1; } /*tolua_SetData_CTest()函数和tolua_GetData_CTest分别对应CTest的SetData方法和GetData()方法。 因为我们的SetData方法里面存在变量,那么同样,我们需要使用const char* pData = http://www.mamicode.com/tolua_tostring(pState, 2, 0);将参数弹出来,>*/int CLuaFn::tolua_SetData_CTest(lua_State* pState) { CTest* pTest = (CTest* )tolua_tousertype(pState, 1, 0); const char* pData = http://www.mamicode.com/tolua_tostring(pState, 2, 0); if(pData != NULL && pTest != NULL) { pTest->SetData(pData); } return 1; } int CLuaFn::tolua_GetData_CTest(lua_State* pState) { CTest* pTest = (CTest* )tolua_tousertype(pState, 1, 0); if(pTest != NULL) { char* pData = http://www.mamicode.com/pTest->GetData(); tolua_pushstring(pState, pData); } return 1; } /* ************************************************************* */
看看这几个静态函数在干什么。
我要在Lua里面使用CTest,必须让Lua里这个CTest对象能够顺利的创造和销毁。tolua_new_CTest()和tolua_delete_CTest()就是干这个的。
tolua_pushusertype(pState, pTest, "CTest"); 这句话的意思是,将一个已经在Lua注册的"CTest"对象指针,压入数据栈。
同理,CTest* pTest = (CTest* )tolua_tousertype(pState, 1, 0);是将数据栈下的对象以(CTest* )的指针形式弹出来。
tolua_SetData_CTest()函数和tolua_GetData_CTest分别对应CTest的SetData方法和GetData()方法。因为我们的SetData方法里面存在变量,那么同样,我们需要使用const char* pData = http://www.mamicode.com/tolua_tostring(pState, 2, 0);将参数弹出来,然后输入到pTest->SetData(pData);对象中去,当然,你可以有更多若干个参数。随你的喜好。这里只做一个举例。
好了,你一定会问,这么多的静态函数,用在哪里?呵呵,当然是给Lua注册,当你把这些数据注册到Lua里面,你就可以轻松的在Lua中使用它们。
让我们看看,注册是怎么做到的。
还是在CLuaFn类里面,我们增加一个函数。比如叫做bool InitClass();
bool CLuaFn::InitClass() { if(NULL == m_pState) { printf("[CLuaFn::InitClass]m_pState is NULL.\n"); return false; } tolua_open(m_pState); tolua_module(m_pState, NULL, 0); tolua_beginmodule(m_pState, NULL); tolua_usertype(m_pState, "CTest"); tolua_cclass(m_pState, "CTest", "CTest", "", tolua_delete_CTest); //只注册一个模块,比如,我们管CTest叫做"CTest",保持和C++的名称一样。这样在Lua的对象库中就会多了一个CTest的对象描述,等同于string,number等等基本类型 tolua_beginmodule(m_pState, "CTest"); tolua_function(m_pState, "new", tolua_new_CTest); //将Lua里面CTest对象的"SetData"绑定到你的tolua_SetData_CTest()函数中去 tolua_function(m_pState, "SetData", tolua_SetData_CTest); tolua_function(m_pState, "GetData", tolua_GetData_CTest); //tolua_beginmodule()和tolua_endmodule()对象必须成对出现,如果出现不成对的,你注册的C++类型将会失败 tolua_endmodule(m_pState); tolua_endmodule(m_pState); return true; }
上面的代码,就是我把上面的几个静态函数,绑定到Lua的基础对象中去。
tolua_beginmodule(m_pState, "CTest");是只注册一个模块,比如,我们管CTest叫做"CTest",保持和C++的名称一样。这样在Lua的对象库中就会多了一个CTest的对象描述,等同于string,number等等基本类型,同理,你也可以用同样的方法,注册你的MFC类。是不是有点明白了?这里要注意,tolua_beginmodule()和tolua_endmodule()对象必须成对出现,如果出现不成对的,你注册的C++类型将会失败。
tolua_function(m_pState, "SetData", tolua_SetData_CTest);指的是将Lua里面CTest对象的"SetData"绑定到你的tolua_SetData_CTest()函数中去。
好的,让我们来点激动人心的东西。还记得我们的Simple.lua的文件么。我们来改一下它。
function func_Add(x, y)-- New了一个CTest对象,并进行赋值操作,最后把结果打印在屏幕上 local test = CTest:new() test:SetData("I‘m xiaotian") test:GetData()return x..y; end
我在这个函数里面,New了一个CTest对象,并进行赋值操作,最后把结果打印在屏幕上。你或许会问,最后一句不是x+y么,怎么变成了x..y,呵呵,在Lua中,..表示联合的意思,就好比在C++里面, string strName += "freeeyes"。原来觉得x+y有点土,索性返回一个两个字符串的联合吧。
好了,我们已经把我们的这个CTest类注册到了Lua里面,让我们来调用一下吧。修改一下Main函数。变成以下的样子。
#include "CLuaFn.h"int main(int argc, char* argv[]){ CLuaFn CLuaFn; CLuaFn.Init(); CLuaFn.InitClass(); CLuaFn.LoadLuaFile("Sample.lua"); CParamGroup ParamIn; CParamGroup ParamOut; char szData1[20] = {‘\0‘}; sprintf(szData1, "[yang]"); _ParamData* pParam1 = new _ParamData(szData1, "string", (int)strlen(szData1)); ParamIn.Push(pParam1); char szData2[20] = {‘\0‘}; sprintf(szData2, "[xiaotian]"); _ParamData* pParam2 = new _ParamData(szData2, "string", (int)strlen(szData2)); ParamIn.Push(pParam2); char szData3[40] = {‘\0‘}; _ParamData* pParam3 = new _ParamData(szData3, "string", 40); ParamOut.Push(pParam3); CLuaFn.CallFileFn("func_Add", ParamIn, ParamOut); char* pData = http://www.mamicode.com/(char* )ParamOut.GetParam(0)->GetParam(); printf("[Main]Sum = %s.\n", pData); getchar(); return 0; }
运行一下:
Lua脚本和C++交互(三)