C/C++与lua互调函数的方法

1,在lua脚本中调用C/C++代码中的函数

在C++中定义函数时必须以lua_State为参数, 以int为返回值才能被Lua所调用。

/*

typedef int (*lua_CFunction) (lua_State*L);

C 函数的类型。

为了正确的和 Lua 通讯,C 函数必须使用下列定义了参数以及返回值传递方法的协议: C 函数通过 Lua 中的堆栈来接受参数,参数以正序入栈(第一个参数首先入栈)。因此,当函数开始的时候, lua_gettop(L) 可以返回函数收到的参数个数。第一个参数(如果有的话)在索引 1 的地方,而最后一个参数在索引 lua_gettop(L) 处。当需要向 Lua 返回值的时候,C 函数只需要把它们以正序压到堆栈上(第一个返回值最先压入),然后返回这些返回值的个数。在这些返回值之下的,堆栈上的东西都会被 Lua 丢掉。和 Lua 函数一样,从 Lua 中调用 C 函数也可以有很多返回值。

*/

下面这个例子中的函数将接收若干数字参数,并返回它们的平均数与和:

#include <stdio.h>

 

#ifdef __cplusplus

extern "C" {

#endif /* __cplusplus */

#include <lua.h> // lua是用纯c语言写的

#include <lualib.h>

#include <lauxlib.h>

#ifdef __cplusplus

}

#endif /* __cplusplus */

 

int c_average(lua_State* L)

{

   int n = lua_gettop(L); /* 返回栈顶元素的索引。因为索引是从1开始编号的(1表示栈底,-1表示栈顶),所以这个结果等于堆栈上的元素个数(返回0表示堆栈为空)。这里栈中元素的个数就是传入的参数个数 */

   double sum = 0;

   int i;

 

   for (i = 1; i <= n; i++)

    {

       if (!lua_isnumber(L, i))

       {

           lua_pushstring(L, "Incorrect argument to 'average'"); // 将错误信息压入栈中

           lua_error(L); // 抛出栈顶的错误

           /*

           int lua_error (lua_State *L);

           产生一个 Lua 错误。错误信息(实际上可以是任何类型的 Lua 值)必须被置入栈顶。这个函数会做一次长跳转,它不会再返回。

           */

       }

       sum += lua_tonumber(L, i); // lua_tonumber 将栈中指定index的值转换成数值类型的值,注意并不会从栈中弹出这个值

    }

 

   double avg = sum / n;

   lua_pushnumber(L, avg); // 将avg压入栈中,第1个返回值是avg

   lua_pushnumber(L, sum); // 将sum压入栈中,第2个返回值是sum

 

   return 2; /* return the number of results,该函数有2个返回值,即上面入栈的avg和sum */

}

 

int main(int argc, char* argv[])

{

   lua_State* L = luaL_newstate(); // 创建一个新的独立的状态机

   

   /* register our function,告诉lua脚本其中调用的average函数(lua中的变量名)对应的是一个叫c_average的c语言函数 */

   lua_register(L, "average", c_average);

   /*

   void lua_register (lua_State *L,

   const char *name,

   lua_CFunction f);

    把 C 函数 f 设到全局变量 name 中。它通过一个宏定义:

   #define lua_register(L,n,f) \

   (lua_pushcfunction(L, f), lua_setglobal(L, n))

   */

 

   /* run the script */

   luaL_dofile(L, "e1.lua"); // 加载脚本,脚本中的可执行语句将会得到执行

   

   /* Loads a file as a Lua chunk and then 以保护模式调用一个函数,等价于(luaL_loadfile(L,filename) || lua_pcall(L, 0, LUA_MULTRET, 0)) */

   lua_getglobal(L, "avg"); // 将全局变量avg的值入栈,等价于lua_getfield(L,LUA_GLOBALSINDEX, "name")

   printf("avg is: %d\n", lua_tointeger(L, -1)); // 读栈顶的int值,注意不会从栈中弹出这个元素

   lua_pop(L, 1); // 弹出栈顶的一个元素

   

   lua_getglobal(L, "sum"); // 将全局变量sum的值入栈

   printf("sum is: %d\n", lua_tointeger(L, -1));

   lua_pop(L, 1); // 弹出栈顶的一个元素

   

   /* cleanup Lua */

   lua_close(L);

 

   return 0;

}

/*

-- e1.lua

avg, sum = average(10, 20, 30, 40, 50)

print("The average is ", avg)

print("The sum is ", sum)

*/

/*

输出:

The average is  30

The sum is      150

avg is: 30

sum is: 150

*/

2,在C/C++代码中调用lua脚本中的函数

在C/C++代码中通过栈将参数传递给lua脚本中的函数,参数按照正序入栈,返回值由lua按正序压入栈中供C/C++访问。(函数调用成功后函数名和参数都出栈,返回值入栈)

示例:

#include <stdio.h>

 

#ifdef __cplusplus

extern "C" {

#endif /* __cplusplus */

#include <lua.h> // lua是用纯c语言写的

#include <lualib.h>

#include <lauxlib.h>

#ifdef __cplusplus

}

#endif /* __cplusplus */

 

lua_State* L = NULL;

 

int luaAdd(int x, int y)

{

   int sum;

 

   /* the function name,lua脚本中的函数名 */

   lua_getglobal(L, "add"); // lua函数名入栈

 

   int savedTop = lua_gettop(L); // 保存栈中的元素个数,后面要还原

 

   /* the first argument */

   lua_pushnumber(L, x); // 参数x入栈

   /* the second argument */

   lua_pushnumber(L, y); // 参数y入栈

 

   // 此时栈中有3个值,栈顶是参数y,栈底是函数名add

 

   /* call the function with 2 arguments, return 1 result */

   lua_call(L, 2, 1);

   /*

   void lua_call (lua_State *L, int nargs, int nresults);

    调用一个函数。

    要调用一个函数请遵循以下协议:首先,要调用的函数应该被压入堆栈;接着,把需要传递给这个函数的参数按正序压栈;这是指第一个参数首先压栈。最后调用一下 lua_call; nargs 是你压入堆栈的参数个数。当函数调用完毕后,所有的参数以及函数本身都会出栈。而函数的返回值这时则被压入堆栈。返回值的个数将被调整为 nresults 个,除非 nresults 被设置成 LUA_MULTRET。在这种情况下,所有的返回值都被压入堆栈中。 Lua 会保证返回值都放入栈空间中。函数返回值将按正序压栈(第一个返回值首先压栈),因此在调用结束后,最后一个返回值将被放在栈顶。

   lua_pcall (lua_State *L, int nargs, int nresults, int errfunc);

    以保护模式调用一个函数。

   nargs 和 nresults 的含义与 lua_call 中的相同。如果在调用过程中没有发生错误, lua_pcall 的行为和 lua_call 完全一致。但是,如果有错误发生的话, lua_pcall 会捕获它,然后把单一的值(错误信息)压入堆栈,然后返回错误码。同 lua_call 一样, lua_pcall 总是把函数本身和它的参数从栈上移除。

    如果 errfunc 是 0 ,返回在栈顶的错误信息就和原始错误信息完全一致。否则,errfunc 就被当成是错误处理函数在栈上的索引。(在当前的实现里,这个索引不能是伪索引。)在发生运行时错误时,这个函数会被调用而参数就是错误信息。错误处理函数的返回值将被 lua_pcall 作为出错信息返回在堆栈上。

    典型的用法中,错误处理函数被用来在出错信息上加上更多的调试信息,比如栈跟踪信息 (stack traceback) 。这些信息在 lua_pcall 返回后,因为栈已经展开 (unwound) ,所以收集不到了。

   lua_pcall 函数在调用成功时返回 0 ,否则返回以下(定义在 lua.h 中的)错误代码中的一个:

   LUA_ERRRUN: 运行时错误。

   LUA_ERRMEM: 内存分配错误。对于这种错,Lua 调用不了错误处理函数。

   LUA_ERRERR: 在运行错误处理函数时发生的错误。

   */

 

   // 因为add只有一个返回值,所以此时栈中只有1个值(如果有多个返回值,也是按正序入栈)

   /* get the result */

   sum = (int) lua_tonumber(L, -1); // -1是栈顶

   lua_pop(L, 1); // 返回值出栈

 

   /* 取出脚本中的变量z的值 */

   lua_getglobal(L, "z"); // 变量z的值入栈

   int z = (int) lua_tonumber(L, -1);

   printf("z = %d\n", z);

   lua_pop(L, 1); // 变量z出栈

 

   lua_pushnumber(L, 4); // 4入栈

   lua_setglobal(L, "r");

   /*

   void lua_setglobal (lua_State *L, const char *name);

    从堆栈上弹出一个值,并将其设到全局变量 name 中。它由一个宏定义出来:

   #define lua_setglobal(L,s)  lua_setfield(L, LUA_GLOBALSINDEX, s)

   */

 

   lua_getglobal(L, "r"); // 变量r的值入栈

   int r = (int)lua_tonumber(L, -1);

   printf("r = %d\n", r);

   lua_pop(L, 1); // 变量r出栈

 

   lua_settop(L, savedTop); // 函数退出之前恢复原来的栈

   return sum;

}

 

int main(int argc, char* argv[])

{

    L= luaL_newstate(); // 创建lua运行环境

   /* load the script,加载脚本,为后面读取其中的变量做准备 */

   luaL_dofile(L, "e2.lua");

   /* call the add function */

   int sum = luaAdd(10, 15);

   /* print the result */

   printf("The sum is %d\n", sum);

   /* cleanup Lua */

   lua_close(L);

   return 0;

}

/*

-- e2.lua

-- add two numbers

function add(x, y)

   return x + y

end

 

z = 6

*/

/*

输出:

z = 6

r = 4

The sum is 25

*/

3,综合例子

#include <stdio.h>

 

#ifdef __cplusplus

extern "C" {

#endif /* __cplusplus */

#include <lua.h> // lua是用纯c语言写的

#include <lualib.h>

#include <lauxlib.h>

#ifdef __cplusplus

}

#endif /* __cplusplus */

 

#define err_exit(num,fmt,args...)  \

   do{printf("[%s:%d]"fmt"\n",__FILE__,__LINE__,##args);exit(num);}while(0)

#define err_return(num,fmt,args...)  \

   do{printf("[%s:%d]"fmt"\n",__FILE__,__LINE__,##args);return(num);}while(0)

 

// lua中调用的c函数定义,实现加法

int c_sum(lua_State* L)

{

   int a = lua_tointeger(L, 1);

   int b = lua_tointeger(L, 2);

   lua_pushinteger(L, a + b);

   return 1;

}

 

int main(int argc, char* argv[])

{

   lua_State* L = luaL_newstate(); //创建lua运行环境

   if (L == NULL)

       err_return(-1, "luaL_newstat() failed");

 

   int ret = 0;

   ret = luaL_loadfile(L, "e3.lua"); // Loads a file as a Luachunk. 底层调用lua_load,功能是:加载一个 Lua chunk 。如果没有错误, lua_load 把一个编译好的 chunk 作为一个 Lua 函数压入堆栈。否则,压入出错信息。

   if (ret != 0)

       err_return(-1, "luaL_loadfile failed");

 

   ret = lua_pcall(L, 0, 0, 0); // 以保护模式调用一个函数。 即刚刚加载的文件

   if (ret != 0)

       err_return(-1, "lua_pcall failed: %s", lua_tostring(L, -1));

 

   lua_getglobal(L, "width"); // 将全局变量width的值入栈

   lua_getglobal(L, "height");// 将全局变量height的值入栈

   printf("height: %ld, width: %ld\n", (long) lua_tointeger(L,-1), (long) lua_tointeger(L, -2));

   

   lua_pop(L, 1); // 从栈中弹出一个值,从栈顶开始

 

   int a = 11;

   int b = 12;

   lua_getglobal(L, "mySum1"); // 将全局变量mySum1入栈,注意这里的mySum1是一个函数

   lua_pushinteger(L, a); // 第1个整型参数入栈,从左往右的顺序

   lua_pushinteger(L, b); // 第2个整型参数入栈,从左往右的顺序

   ret = lua_pcall(L, 2, 1, 0); // 调用函数,2个参数,1个返回值。调用成功后,函数和参数均出战,返回值入栈。如果有多个返回值,则第一个返回值先入栈

   if (ret != 0)

       err_return(-1, "lua_pcall failed: %s", lua_tostring(L, -1));// 调用函数失败后栈顶存放失败原因字符串

   printf("sum: %d + %d = %ld\n", a, b, (long) lua_tointeger(L,-1)); // sum只有一个返回值,调用成功后栈顶存放函数的返回值

   

   lua_pop(L, 1);

   

   const char str1[] = "hello";

   const char str2[] = "world";

   lua_getglobal(L, "myStrcat"); // 调用lua中的函数myStrcat

   lua_pushstring(L, str1);

   lua_pushstring(L, str2);

   ret = lua_pcall(L, 2, 1, 0);

   if (ret != 0)

       err_return(-1, "lua_pcall failed: %s", lua_tostring(L, -1));

   printf("mystrcat: %s %s = %s\n", str1, str2, lua_tostring(L,-1));

   

   lua_pop(L, 1);

   

   lua_pushcfunction(L, c_sum); // 将一个 C 函数压入堆栈

   lua_setglobal(L, "mySum2");

   /*

   void lua_setglobal (lua_State *L, const char *name);

    从堆栈上弹出一个值,并将其设到全局变量 name 中。它由一个宏定义出来:

   #define lua_setglobal(L,s)  lua_setfield(L, LUA_GLOBALSINDEX, s)

   */

   // 以上两句可以用lua_register(L, "mySum2", c_sum);来代替

   /* register our function,告诉lua脚本其中调用的average函数(也是一个变量名)其实对应的是一个叫c_averaged的c代码函数 */

 

   // 调用lua中的mySum2函数,该函数调用c_sum函数实现加法

   lua_getglobal(L, "mySum2");

   lua_pushinteger(L, a);

   lua_pushinteger(L, b);

   ret = lua_pcall(L, 2, 1, 0);

   if (ret != 0)

       err_return(-1, "lua_pcall failed: %s", lua_tostring(L, -1));

   printf("mysum: %d + %d = %ld\n", a, b, (long) lua_tointeger(L,-1));

   

   lua_pop(L, 1);

   

   lua_close(L); // 释放lua运行环境

   return 0;

}

/*

-- e3.lua

-- 变量定义

width = 1

height = 2

 

-- lua函数定义,实现加法

function mySum1(a, b)

   return a + b

end

 

-- lua函数定义,实现字符串相加

function myStrcat(a, b)

   return a.. "_" .. b

end

 

-- lua函数定义,通过调用c代码中的csum函数实现加法

function mysum(a, b)

   return csum(a, b)

end

*/

/*

输出:

height: 2, width: 1

sum: 11 + 12 = 23

mystrcat: hello world = hello_world

mysum: 11 + 12 = 23

*/

你可能感兴趣的:(C/C++与lua互调函数的方法)