lua和C++相互交换数据

下面的代码演示了在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 <lua.h>
#include <lauxlib.h>
#include <lualib.h>
}
#include <iostream>

/* 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