使用C中的Lua API覆盖内置类型的运算符

时间:2016-12-05 18:08:35

标签: c++ c lua overloading operator-keyword

我在我的C ++代码中集成了Lua 5.3,并添加了几个在两个环境之间可以互换的数学类。

例如,我有一个带有C功能的vec2 Lua metatable,可以将它链接到我的vec2d的C ++类。

现在,我的vec2 metatable有一个__mul运算符,所以我可以编写Lua代码,如:

local vector = vec2.create(1, 1)
local scaledVector = vector * 5
print(tostring(scaledVector)) -- outputs 5, 5

但有时候,我只是想以相反的方式写它,我希望这也能起作用:

local vector = vec.create(1, 1)
local scaledVector = 5 * vector -- error: Class metatable function __index called on something else than userdata
print(tostring(scaledVector)) -- I want 5, 5

我明白为什么它不起作用。

Lua完全可以吗?如果是这样......怎么样? (我正在寻找一个C / C ++解决方案,而不是用Lua编写的某种构造)

1 个答案:

答案 0 :(得分:2)

如果乘法的左操作数没有设置__mul元方法,Lua将检查右操作数以查看它是否具有__mul元方法。因此5 * vec应该与vec * 5

一样有效
function scale_vector(vec, alpha)
  local out = {}
  for i=1,#vec do
    out[i] = vec[i] * alpha
  end
  return out
end

function print_vector(vec)
  for i=1, #vec do
    if i > 1 then
      io.stdout:write("\t")
    end
    io.stdout:write(vec[i])
  end
  io.stdout:write("\n")
end


mt = {
  __mul = function(a, b)
    if type(b) == "number" then
      print("Case 1")
      return scale_vector(a, b)
    elseif type(a) == "number" then
      print("Case 2")
      return scale_vector(b, a)
    else
      error("Cannot scale by non-number factor")
    end
  end
}

vec = {1,2,3}
setmetatable(vec, mt)

print_vector( vec * 5 )
print_vector( 5 * vec )

运行此脚本会导致

Case 1
5       10      15
Case 2
5       10      15

你可能出错的部分是在__mul元方法中你需要做一些测试来确定你的向量是第一个参数还是第二个参数。我的猜测是你的代码当前假设__mul元方法的第一个参数始终是向量对象。