Lua教程（⼆⼗⼆）：userdata
在Lua中可以通过⾃定义类型的⽅式与C语⾔代码更⾼效、更灵活的交互。这⾥我们通过⼀个简单完整的⽰例来学习⼀下Lua 中userdata的使⽤⽅式。需要说明的是，该⽰例完全来⾃于Programming in Lua。其功能是⽤C程序实现⼀个Lua的布尔数组，以提供程序的执⾏效率。见下⾯的代码和关键性注释。
复制代码代码如下:
#include <lua.hpp>
#include <lauxlib.h>
#include <lualib.h>
#include <limits.h>
#define BITS_PER_WORD (CHAR_BIT * sizeof(int))
#define I_WORD(i)    ((unsigned int)(i))/BITS_PER_WORD
#define I_BIT(i)      (1 << ((unsigned int)(i)%BITS_PER_WORD))
typedef struct NumArray {
int size;
unsigned int values[1];
} NumArray;
extern "C" int newArray(lua_State* L)
{
//1. 检查第⼀个参数是否为整型。以及该参数的值是否⼤于等于1.
int n = luaL_checkint(L,1);
luaL_argcheck(L, n >= 1, 1, "invalid size.");
size_t nbytes = sizeof(NumArray) + I_WORD(n - 1) * sizeof(int);
//2. 参数表⽰Lua为userdata分配的字节数。同时将分配后的userdata对象压⼊栈中。
NumArray* a = (NumArray*)lua_newuserdata(L,nbytes);
a->size = n;
for (int i = 0; i < I_WORD(n - 1); ++i)
a->values[i] = 0;
//获取注册表变量myarray，该key的值为metatable。
luaL_getmetatable(L,"myarray");
//将userdata的元表设置为和myarray关联的table。同时将栈顶元素弹出。
lua_setmetatable(L,-2);
return 1;
}
extern "C" int setArray(lua_State* L)
{
//1. Lua传给该函数的第⼀个参数必须是userdata，该对象的元表也必须是注册表中和myarray关联的table。
//否则该函数报错并终⽌程序。
NumArray* a = (NumArray*)luaL_checkudata(L,1,"myarray");
int index = luaL_checkint(L,2) - 1;
//2. 由于任何类型的数据都可以成为布尔值，因此这⾥使⽤any只是为了确保有3个参数。
luaL_checkany(L,3);
luaL_argcheck(L,a != NULL,1,"'array' expected.");
luaL_argcheck(L,0 <= index && index < a->size,2,"index out of range.");
if (lua_toboolean(L,3))
a->values[I_WORD(index)] |= I_BIT(index);
else
a->values[I_WORD(index)] &= ~I_BIT(index);
return 0;
}
extern "C" int getArray(lua_State* L)
{
NumArray* a = (NumArray*)luaL_checkudata(L,1,"myarray");
int index = luaL_checkint(L,2) - 1;
luaL_argcheck(L, a != NULL, 1, "'array' expected.");
luaL_argcheck(L, 0 <= index && index < a->size,2,"index out of range");
lua_pushboolean(L,a->values[I_WORD(index)] & I_BIT(index));
return 1;
}
extern "C" int getSize(lua_State* L)
{
NumArray* a = (NumArray*)luaL_checkudata(L,1,"myarray");
luaL_argcheck(L,a != NULL,1,"'array' expected.");
lua_pushinteger(L,a->size);
return 1;
lua字符串转数组}
extern "C" int array2string(lua_State* L)
{
NumArray* a = (NumArray*)luaL_checkudata(L,1,"myarray");
lua_pushfstring(L,"array(%d)",a->size);
return 1;
}
static luaL_Reg arraylib_f [] = {
{"new", newArray},
{NULL, NULL}
};
static luaL_Reg arraylib_m [] = {
{"set", setArray},
{"get", getArray},
{"size", getSize},
{"__tostring", array2string}, //print(a)时Lua会调⽤该元⽅法。
{NULL, NULL}
};
extern "C" __declspec(dllexport)
int luaopen_testuserdata(lua_State* L)
{
//1. 创建元表，并将该元表指定给newArray函数新创建的userdata。在Lua中userdata也是以table的⾝份表现的。
//这样在调⽤对象函数时，可以通过验证其metatable的名称来确定参数userdata是否合法。
luaL_newmetatable(L,"myarray");
lua_pushvalue(L,-1);
//2. 为了实现⾯对对象的调⽤⽅式，需要将元表的__index字段指向⾃⾝，同时再将arraylib_m数组中的函数注册到
//元表中，之后基于这些注册函数的调⽤就可以以⾯向对象的形式调⽤了。
//lua_setfield在执⾏后会将栈顶的table弹出。
lua_setfield(L,-2,"__index");
//将这些成员函数注册给元表，以保证Lua在寻⽅法时可以定位。NULL参数表⽰将⽤栈顶的table代替第⼆个参数。
luaL_register(L,NULL,arraylib_m);
//这⾥只注册的⼯⼚⽅法。
luaL_register(L,"testuserdata",arraylib_f);
return 1;
}
轻量级userdata：
  之前介绍的是full userdata，Lua还提供了另⼀种轻量级userdata(light userdata)。事实上，轻量级userdata仅仅表⽰的是C指针的值，即(void*)。由于它只是⼀个值，所以不⽤创建。如果需要将⼀个轻量级userdata放⼊栈中，调⽤
lua_pushlightuserdata即可。full userdata和light userdata之间最⼤的区别来⾃于相等性判断，对于⼀个full userdata，它只是与⾃⾝相等，⽽light userdata则表⽰为⼀个C指针，因此，它与所有表⽰同⼀指针的light userdata相等。再有就是light userdata不会受到垃圾收集器的管理，使⽤时就像⼀个普通的整型数字⼀样。

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。