Lua：用 nil 解压？选择？

Question

我遇到了古老的unpack bug，我在 Lua 中有一个可以包含 nil 值的数组，我想用 nil 值解压数组；这似乎是不可能的。这个逻辑的替代方案是什么？

这是我尝试运行的代码

function InputSystem:poll(name, ...)
  local system = self:findComponent(name)
  local values, arr = {...}, {}
  for i, v in pairs(values) do
    arr[#arr+1] = system[v]
  end

  --If the first guy is null this does not work!! WHY
  return unpack(arr, 1, table.maxn(values))
end

我的想法是我动态地轮询我的输入系统，以便我只返回我想要的值：

local dragged,clicked,scrolled = self.SystemManager:findSystem('Input'):poll('Mouse', 'Dragged', 'Clicked', 'Scrolled')

有什么想法吗？谢谢

编辑：

我似乎没有完全理解 Lua。我想返回与 ... 传入的相同数量的变量，但是在循环中如果找不到该属性，我认为它会将其设置为 nil，但这似乎是错误的。

function InputSystem:poll(name, ...)
      local system = self:findComponent(name)
      local values, arr = {...}, {}
      for i, v in pairs(values) do
        arr[#arr+1] = system[v] --If not found set nil
      end

      --I want this to return the length of ... in variables
      --Example I pass 'Dragged', 'Clicked' I would want it to return nil {x:1,y:1}
      return unpack(arr, 1, table.maxn(values))
    end

显然我是 Lua 大师...

Answer 1

您应该使用table.pack 和table.unpack 来保留nils。如果您使用 Lua 5.2 或更高版本，则可以删除兼容性 sn-p。

-- Backwards compatibility
table.pack = table.pack or function(...) return { n = select("#", ...), ... } end
table.unpack = table.unpack or unpack

function test(...)
    local values = table.pack(...)
    local arr = {}
    for i, v in pairs(values) do
        -- iterates only the non-nil fields of "values"
        arr[i] = 10*v
    end
    return table.unpack(arr, 1, values.n)
end

print(test(nil, 1, nil, 2, nil, nil, 3))

$ lua5.3 test.lua
nil 10  nil 20  nil nil 30
$ lua5.2 test.lua
nil 10  nil 20  nil nil 30
$ lua5.1 test.lua
nil 10  nil 20  nil nil 30
$ luajit test.lua
nil 10  nil 20  nil nil 30

Answer 2

for i, v in pairs(values) do
  arr[#arr+1] = system[v] -- This doesn't work!
end

您的实现的问题是您希望将 nil 附加到数组以增加其长度，但事实并非如此：

local arr = {1, 2, 3}
print(#arr) --> 3
arr[#arr+1]=nil
print(#arr) --> 3

本质上，您想要的是一个 map 函数，它接受一个元素列表，将函数 fn 应用于每个元素并返回结果列表。

通常，这可以很容易地实现为 tail-recursive 函数，如下所示：

local function map(fn, elem, ...)
  if elem then return fn(elem), map(fn, ...)
end

这并不能很好地处理nil 参数，因为它们会使条件为假，而仍有待处理的参数，但我们可以使用select 对其进行修改以避免这种情况：

local function map(fn, elem, ...)
  if select('#', ...)>0 then return fn(elem), map(fn, ...)
  else return fn(elem) end
end
-- This implementation still gets TCOd :)

那么你可以这样使用它：

map(string.upper, 'hello', 'world') --> 'HELLO', 'WORLD'

您希望将... 中的每个值映射到表中的相应值，但由于map 将函数作为其第一个值，我们可以将其包装在函数中。而且由于在编写代码时我们不知道该表，因此我们必须在运行时生成该函数：

local function index(table)
  return function(idx)
    return table[idx]
  end
end

现在我们可以这样做了：

map(index{'hello', 'world'}, 1, 2) --> 'hello', 'world'
-- index{'hello', 'world'} returns a function that indexes the given table
-- with its first argument and returns the value

然后你可以这样写你的InputSystem函数：

function InputSystem:poll(name, ...)
  return map(index(self:findComponent(name)), ...)
end

显然，在这种情况下，我们不需要通用的 map 函数，因为我们总是在索引一个表。我们可以重写 map 来使用这样的表：

local function map(tab, elem, ...)
  if select('#', ...)>0 then return tab[elem], map(tab, ...)
  else return tab[elem] end
end

主函数会变成：

function InputSystem:poll(name, ...)
  return map(self:findComponent(name), ...)
end

我注意到的另一件事：

  for i, v in pairs(values) do
    arr[#arr+1] = system[v] --If not found set nil
  end

pairs 无序迭代，因此您的行 for i, v in pairs(values) do 很可能完全重新排序值。由于您写得更远local dragged,clicked,scrolled = self.SystemManager:findSystem...，我相信您希望返回值保持有序。