Lua 用户数据数组访问和方法

Question

我正在用 C 语言编写一个用于 Lua 的用户数据类型。它还有一些数组类型的属性和各种方法。现在如果你是这种类型，我使用u:set(k,v) resp。 u:get(k) 访问数据，例如u:sort() 作为方法。为此，我将__index 设置为包含这些方法的表。现在，如果我想使用u[k] = v 或u[k] 访问数据，我需要将__newindex 和__index 设置为set 和get。但随后其他方法不再可用......

在 C 中处理这个问题的最佳方法是什么？我猜我需要用 C 编写一个函数来注册为__index 并以某种方式在那里处理它。也许检查 key 是否属于 Lua 方法表，如果是，则调用它。

任何帮助/提示将不胜感激。我没有找到这样的例子，尽管（对我来说）这似乎是一件很自然的事情。

编辑：在 Lua 中添加了我的 C 版本解决方案，发布在下面的答案中。这或多或少是直接翻译，所以所有功劳都归于@gilles-gregoire。

以下 C 函数注册为 __index 元方法。

static int permL_index(lua_State *L) {
  struct perm **pp = luaL_checkudata(L, 1, PERM_MT);
  int i;

  luaL_getmetatable(L, PERM_MT);
  lua_pushvalue(L, 2);
  lua_rawget(L, -2);

  if ( lua_isnil(L, -1) ) {
    /* found no method, so get value from userdata. */
    i = luaL_checkint(L, 2);
    luaL_argcheck(L, 1 <= i && i <= (*pp)->n, 2, "index out of range");

    lua_pushinteger(L, (*pp)->v[i-1]);
  };

  return 1;
};

这是执行此操作的代码，

int luaopen_perm(lua_State *L) {

  luaL_newmetatable(L, PERM_MT);
  luaL_setfuncs(L, permL_methods, 0);
  luaL_setfuncs(L, permL_functions, 0);
  lua_pop(L, 1);

  luaL_newlib(L, permL_functions);

  return 1;
};

permL_methods 在哪里

static const struct luaL_Reg permL_methods[] = {
  { "__index",      permL_index           },
  { "__eq",         permL_equal           },
  { "__tostring",   permL_tostring        },
  { "__gc",         permL_destroy         },
  [...]
  { NULL,           NULL                  }
};

而permL_functions 是

static const struct luaL_Reg permL_functions[] = {
  { "inverse",      permL_new_inverse     },
  { "product",      permL_new_product     },
  { "composition",  permL_new_composition },
  [...]
  { NULL,           NULL                  }
};

Answer 1

这看起来像是一个可以通过嵌套元表解决的问题。您需要一个用于方法的元表（例如您的 sort() 方法），另一个用于索引操作。第二个元表实际上是方法元表的元表。

让我把它写成 lua 代码。你需要 3 张桌子：

-- the userdata object. I'm using a table here,
-- but it will work the same with a C userdata
u = {}

-- the "methods" metatable:
mt = {sort = function() print('sorting...') end}

-- the "operators" metatable:
op_mt = {__index = function() print('get') end}

现在，棘手的部分在这里：当你调用一个方法时，lua 将首先查找u。 如果没有找到，它将在u 的元表的__index 字段指向的表中查找... Lua 将对该表重复该过程！

-- first level metatable
mt.__index = mt
setmetatable(u, mt)

-- second level metatable
setmetatable(mt, op_mt)

您现在可以像这样使用您的u：

> u:sort()
sorting...
> = u[1]
get
nil

编辑：使用 __index 元方法的函数的更好解决方案

为 __index 元方法使用函数可能是正确的方法：

u = {}
mt = {sort = function() print('sorting...') end}
setmetatable(u, mt)
mt.__index = function(t, key)
    -- use rawget to avoid recursion
    local mt_val = rawget(mt, key)
    if mt_val ~=nil then
        return mt_val
    else
        print('this is a get on object', t)
    end
end

用法：

> print(u)
table: 0x7fb1eb601c30
> u:sort()
sorting...
> = u[1]
this is a get on object    table: 0x7fb1eb601c30
nil
>