有没有方法可以确定Lua函数的签名？

2017-5-23 10:27:52
收藏：0
阅读：1049
评论：3

最近，Lee Baldwin 展示了如何编写一个通用的，可变参数记忆化函数。我认为返回一个只需要一个参数的更简单的函数会更好。以下是我的错误尝试：

local function memoize(f)
   local cache = {}

   if select('#', ...) == 1 then
      return function (x)
                if cache[x] then
                   return cache[x]
                else
                   local y = f(x)
                   cache[x] = y
                   return y
                end
              end
   else
      return function (...)
                local al = varg_tostring(...)
                if cache[al] then
                   return cache[al]
                else
                   local y = f(...)
                   cache[al] = y
                   return y
                end
             end
   end
end

显然，select('#', ...) 在这种情况下失败，并且无论如何都不能实现我想要的功能。有没有办法在 memoize 中告诉它 f 需要多少个参数？

如果你确定的话，“不行”是一个很好的答案。使用两个单独的 memoize 函数并不是什么大问题。

原文链接 https://stackoverflow.com/questions/142417

点赞
stackoverflow用户88888888
stackoverflow用户88888888

我猜你可以通过调试信息从源代码中确定这一点，但基本上它是“不是”，很抱歉。

2008-09-26 22:56:55
stackoverflow用户15328
stackoverflow用户15328

我很确定你不能在 Lua 中完成这个。

2008-09-26 22:59:06
stackoverflow用户2226988
stackoverflow用户2226988

是的，可以对 Lua 函数进行操作，但不能对 C 函数进行操作。这有点麻烦也有点缺陷。

debug.getlocal 仅适用于已调用的函数，所以必须调用所需函数。除非调用传递了足够的参数，否则它不会显示任何 ... 的提示信息。下面的代码尝试了 20 个参数。

debug.sethook 使用 "call" 事件提供了在函数运行任何代码之前拦截函数的机会。

此算法适用于 Lua 5.2。旧版本可能类似但不相同：

assert(_VERSION=="Lua 5.2", "Must be compatible with Lua 5.2")

一个小助手迭代器（可以内联以提高效率）：

local function getlocals(l)
  local i = 0
  local direction = 1
  return function ()
    i = i + direction
    local k,v = debug.getlocal(l,i)
    if (direction == 1 and (k == nil or k.sub(k,1,1) == '(')) then
      i = -1
      direction = -1
      k,v = debug.getlocal(l,i)
    end
    return k,v
  end
end

返回函数签名（但是可以返回参数计数和usesVarargs，而不是返回签名）：

local function dumpsig(f)
  assert(type(f) == 'function',
    "bad argument #1 to 'dumpsig' (function expected)")
  local p = {}
  pcall (function()
    local oldhook
    local hook = function(event, line)
      for k,v in getlocals(3) do
        if k == "(*vararg)" then
          table.insert(p,"...")
          break
        end
        table.insert(p,k) end
      debug.sethook(oldhook)
      error('aborting the call')
    end
    oldhook = debug.sethook(hook, "c")
    -- To test for vararg must pass a least one vararg parameter
    f(1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20)
  end)
  return "function("..table.concat(p,",")..")"
end
2014-06-14 02:11:12

作者:

stackoverflow用户1438

stackoverflow用户1438
推荐文章