索引FFI结构中的数组

时间:2016-07-22 05:56:04

标签: lua ffi luajit

所以我正在尝试创建一个模块,动态检测您是使用Lua还是LuaJIT,并根据需要创建表或cstruct。由于你不能将metatables添加到carrays中,我的结构中有一个名为_m的数组。

以下是相关代码的片段,我将链接到下面的Git回购。

local mat4      = {}
local mat4_mt   = {}

-- Private constructor.
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

-- Do the check to see if JIT is enabled. If so use the optimized FFI structs.
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)

这里的问题是,当我尝试调用转置时,我收到以下错误:

'struct 173' cannot be indexed with 'number'

如果你签出mat4_mt.__index,我试图检测我正在使用哪种类型,表或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

1 个答案:

答案 0 :(得分:0)

它正在运行,但是您缺少__newindex metatable条目,导致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