首页 > 代码库 > 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++交互(三)