lua和C++相互交换数据

 lua和C++相互交换数据

2009-01-21 14:23 1273人阅读 评论(1) 收藏 举报

下面的代码演示了在C++和lua脚本之间传递数据。

首先在C++中创建一个table，添加元素，然后放置到lua的全局表中。在lua脚本中可以使用C++创建的这个表。

然后在脚本中创建一个表，以脚本返回值的方式返回给C++，在C++中可以读取表中的值。

 

例子代码需要一个args.lua的lua文件，要手工创建，我把它放到了C盘根目录下。

// cpplua.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"

extern "C"
{
#include
#include
#include
}
#include

/* args.lua文件的内容
io.write( "[lua] These args were passed into the script from C/n" );

for i=1,table.getn(arg) do
print(i,arg[i])
end

io.write("[lua] Script returning data back to C/n")

local temp = {}
temp[1]=9
temp[2]=8
temp[3]=7
temp[4]=6
temp[5]=5
temp["test1 key"]="test1 value"
temp[6]="test 6"
temp["test 99"]=99

for i,n in pairs(temp)
do
print (i,n)
end

return temp,9,1

*/
int _tmain(int argc, _TCHAR* argv[])
{
    int status;

    // lua_open: 创建一个新的lua环境
    lua_State* state = lua_open();

    // 在state环境上打开标准库，
    // 标准库包括：
    // luaopen_base
    // luaopen_package
    // luaopen_table
    // luaopen_io
    // luaopen_os
    // luaopen_string
    // luaopen_math
    // luaopen_debug
    luaL_openlibs(state);  /* open libraries */

    status = luaL_loadfile( state, "c://args.lua" );

    std::cout << "[C++] Passing 'arg' array to script" << std::endl;

    // 创建一个新的表
    lua_newtable( state );

    //
    // set first element "1" to value 45
    //
    // 调用lua的函数，都是通过压栈出栈来完成的
    // 为表执行一个t[k]=v的操作,则需要先将k压栈，再将v压栈，再调用操作函数
    // 这个操作函数会使用栈上的元素，并“可能”将弹出元素和压入元素
    // lua_rawset直接赋值（不触发metamethods方法）。

    // lua_rawset/lua_settable使用:
    // 它从栈中获取参数。以table在栈中的索引作为参数，
    // 并将栈中的key和value出栈。
    // lua_pushnumber函数调用之前，
    // table是在栈顶位置(索引为-1)。index和value入栈之后，
    // table索引变为-3。
    lua_pushnumber( state, 1 );
    lua_pushnumber( state, 45 );
    lua_rawset( state, -3 );

    // set second element "2" to value 99
    lua_pushnumber( state, 2 );
    lua_pushnumber( state, 99 );
    lua_rawset( state, -3 );

    // set the number of elements (index to the last array element)
    // lua_pushliteral压入一个字符串，不需要指定长度
    // 如果lua_pushlstring,则需要指定长度
    lua_pushliteral( state, "n" );
    lua_pushnumber( state, 2 );
    lua_rawset( state, -3 );

    // set the name of the array that the script will access
    // Pops a value from the stack and sets it as the new value of global name.
    // 从栈顶弹出一个值，并将其设置全局变量"arg"的新值。
    lua_setglobal( state, "arg" );


    std::cout << "[C++] Running script" << std::endl;

    int result = 0;
    if (status == 0)
    {
        result = lua_pcall( state, 0, LUA_MULTRET, 0 );
    }
    else
    {
        std::cout << "bad" << std::endl;
    }

    if (result != 0)
    {
        std::cerr << "[C++] script failed" << std::endl;
    }

    std::cout << "[C++] These values were returned from the script" << std::endl;

    // lua_gettop返回栈顶的索引
    // 如果索引为0，则表示栈为空
    while (lua_gettop( state ))
    {
        switch (lua_type( state, lua_gettop( state ) ))
        {
        case LUA_TNUMBER:
            {
                std::cout << "script returned " << lua_tonumber( state, lua_gettop( state ) ) << std::endl;
                break;
            }
        case LUA_TTABLE:
            {
                std::cout << "script returned a table" << std::endl;

                // 简单的遍历表的功能
                // ***好像lua不保存表的元素的添加顺序***

                // 压入第一个键
                lua_pushnil(state);  /* 第一个 key */
                int t = -2;
                while (lua_next(state, t) != 0)
                {
                    /* 'key' (索引-2) 和 'value' (索引-1) */
                    const char* key = "unknown";
                    const char* value;
                    if(lua_type(state, -2) == LUA_TSTRING)
                    {
                        key = lua_tostring(state, -2);
                        value = lua_tostring(state, -1);
                    }
                    else if(lua_type(state, -2) == LUA_TNUMBER)
                    {
                        // 因为lua_tostring会更改栈上的元素，
                        // 所以不能直接在key上进行lua_tostring
                        // 因此，复制一个key，压入栈顶，进行lua_tostring
                        lua_pushvalue(state, -2);
                        key = lua_tostring(state, -1);
                        lua_pop(state, 1);
                        value = lua_tostring(state, -1);
                    }
                    else
                    {
                        value = lua_tostring(state, -1);
                    }

                    std::cout   <<"key="<< key
                                << ", value=" << value << std::endl;

                    /* 移除 'value' ；保留 'key' 做下一次迭代 */
                    lua_pop(state, 1);
                }

                break;
            }
        case LUA_TSTRING:
            {
                std::cout << "script returned " << lua_tostring( state, lua_gettop( state ) ) << std::endl;
                break;
            }
        case LUA_TBOOLEAN:
            {
                std::cout << "script returned " << lua_toboolean( state, lua_gettop( state ) ) << std::endl;
                break;
            }
        default:
            std::cout << "script returned unknown param" << std::endl;
            break;
        }
        lua_pop( state, 1 );
    }
    lua_close( state );
    return 0;
}

 

本例用了一个控制台工程，输出如下：

[C++] Passing 'arg' array to script
[C++] Running script
[lua] These args were passed into the script from C
1       45
2       99
[lua] Script returning data back to C
1       9
2       8
3       7
4       6
5       5
6       test 6
test 99 99
test1 key       test1 value
[C++] These values were returned from the script
script returned 1
script returned 9
script returned a table
key=1, value=9
key=2, value=8
key=3, value=7
key=4, value=6
key=5, value=5
key=6, value=test 6
key=test 99, value=99
key=test1 key, value=test1 value