在 FFI 结构体中对数组进行索引

我正在尝试创建一个动态检测 Lua 或 LuaJIT 的模块,并创建一个 table 或 cstruct。由于无法向 carrays 添加元表,所以我在我的 struct 中使用了一个名为 _m 的数组。

下面是相关代码的片段,我将在下面链接到 Git 仓库。

local mat4      = {}
local mat4_mt   = {}

-- 私有构造函数。
local function new(m)
    m = m or {
        0, 0, 0, 0,
        0, 0, 0, 0,
        0, 0, 0, 0,
        0, 0, 0, 0
    }
    m._m = m
    return setmetatable(m, mat4_mt)
end

-- 检查是否启用了 JIT,如果是,则使用优化的 FFI 结构体。
local status, ffi
if type(jit) == "table" and jit.status() then
    status, ffi = pcall(require, "ffi")
    if status then
        ffi.cdef "typedef struct { double _m[16]; } cpml_mat4;"
        new = ffi.typeof("cpml_mat4")
    end
end

function mat4.transpose(out, a)
    out[1]  = a[1]
    out[2]  = a[5]
    out[3]  = a[9]
    out[4]  = a[13]
    out[5]  = a[2]
    out[6]  = a[6]
    out[7]  = a[10]
    out[8]  = a[14]
    out[9]  = a[3]
    out[10] = a[7]
    out[11] = a[11]
    out[12] = a[15]
    out[13] = a[4]
    out[14] = a[8]
    out[15] = a[12]
    out[16] = a[16]

    return out
end

mat4_mt.__index = function(t, k)
    if type(t) == "cdata" then
        if type(k) == "number" then
            return t._m[k-1]
        end
    elseif type(k) == "number" then
        return t._m[k]
    end

    return rawget(mat4, k)
end

function mat4_mt.__call(_, a)
    return new(a)
end

if status then
    ffi.metatype(new, mat4_mt)
end

return setmetatable({}, mat4_mt)

问题在于,当我尝试调用 transpose 时,会出现以下错误:

'struct 173' cannot be indexed with 'number'

如果您查看 mat4_mt.__index,您会发现我正在尝试检测我正在使用的类型,是 table 还是 cdata,并在结构体中对数组进行索引。

local mat4 = require "mat4"
local a = mat4()
local b = mat4():transpose(a) -- Error!

想法是,当您尝试访问 a[4] 时,应该在幕后访问 a._m[3],但显然没有发生这种情况,我不知道为什么。

想法?

https://github.com/excessive/cpml/blob/refactor/modules/mat4.lua

点赞
用户2428487
用户2428487

它可以工作,但你缺少一个__newindex元表项,导致out [index] = value失败,因为它仍然试图引用结构,而不是它所包含的字段。添加这个可以解决这个问题:

mat4_mt.__newindex = function(t, k, v)
    if type(t) == "cdata" then
        if type(k) == "number" then
            t._m[k-1] = v
        end
    elseif type(k) == "number" then
        t._m[k] = v
    else
        rawset(t, k, v)
    end
end
2018-01-31 10:22:49