Lua编程示例（六）： C语言调用Lua函数

做C++和Lua混合开发时，最常遇到的一个场景就是：在C++端调用Lua函数，然后拿到返回的表（table）数据，再对这张表做读写操作。很多新手容易在这块卡住，其实拆开来看并不复杂。下面这个例子，正好把从“加载Lua脚本”到“操作返回表”的完整链路走了一遍。

C++端：加载脚本并调用Lua函数

先看C++这边。核心步骤其实就几个：创建Lua状态机、加载并执行脚本、从全局环境中拿到函数、调用它，最后对返回的栈上数据做类型检查和处理。

#include "stdafx.h"

lua_State *L;
void load_lua(lua_State **L,char *filename){
 *L=luaL_newstate();
 luaL_openlibs(*L);
 if(luaL_loadfile(*L,filename) || lua_pcall(*L,0,0,0)){
 luaL_error(*L,"load file error! %s",lua_tostring(*L,-1));
 }
}
int _tmain(int argc, _TCHAR* argv[])
{
 load_lua(&L,"raw.lua"); //此处若直接传入L会出错
 lua_getglobal(L,"gettable");
 if(lua_pcall(L,0,1,0) !=0){
 luaL_error(L,"pcall wrong %s",lua_tostring(L,-1));
 }
 luaL_checktype(L,1,LUA_TTABLE);
 int n=lua_objlen(L,1);
 printf("n = %d\n",n);
 lua_pushstring(L,"ee");
 lua_rawseti(L,1,5); //t[n]=v,n为第三个参数，v是栈顶元素
 n=lua_objlen(L,1);
 printf("n = %d\n",n);
 int i;
 for(i=1;i<=n;i++){
 lua_rawgeti(L,1,i);
 printf("%s\n",lua_tostring(L,-1));
 }
 return 0;
}

这里有一个容易踩的坑：load_lua函数传参时，如果你直接传L（而不是它的地址），会导致状态机指针修改不生效，后面所有操作都会崩掉。所以传&L是必要的。

调用完gettable后，Lua函数的返回值被压入栈顶（位置1）。我们用luaL_checktype确认它是表类型，然后用lua_objlen获取表的长度——注意Lua的表索引从1开始。拿到长度后，先打印出来，然后往表里追加一个元素。

追加用的是lua_rawseti，它在栈上需要三个参数：表在栈中的索引、键（数字索引）、值（栈顶元素）。这里先压入字符串"ee"，然后调用lua_rawseti(L, 1, 5)，意思是把栈顶的值赋给表中索引5的位置。因为原表只有4个元素（aa,bb,cc,dd），索引5就是新加的。之后再获取一次长度，就能看到n变成了5。

最后循环遍历所有元素，用lua_rawgeti逐个取出并打印。

Lua脚本：返回一张简单的表

对应的Lua脚本非常简洁，就是定义了一个函数，返回一个包含四个字符串的表：

function gettable() 
  tb={ "aa","bb","cc","dd"} 
  return tb 
end 

这里没有多余的操作，纯粹为了演示C++端拿表、读表、写表的能力。

运行结果：一目了然

把上面两段代码跑起来，控制台输出如下：

n = 4 
n = 5 
aa 
bb 
cc 
dd 
ee

第一行打印了原始表长度4，第二行是插入"ee"之后的长度5。后面四行是原来的四个元素，最后一行是新加的"ee"。整个流程走通，没有任何意外。

这个小例子其实揭示了C++与Lua交互的一个通用模式：C++端通过栈传递参数、接收返回值，而Lua的灵活表结构可以非常方便地在C++端进行增删改查。理解了这几行代码，日后遇到更复杂的table操作（比如嵌套表、键值对混合）也就有了基础。