Existe uma maneira de determinar a assinatura de uma função Lua?
-
02-07-2019 - |
Pergunta
Recentemente, Lee Baldwin mostrou como escrever um genérico, argumento variável função memoize . Eu pensei que seria melhor retornar uma função mais simples, onde é exigido apenas um parâmetro. Aqui é a minha tentativa total de falso:
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
Obviamente, select('#', ...)
falhar neste contexto e não seria realmente fazer o que quero de qualquer maneira. Existe alguma maneira de dizer dentro de memoize quantos argumentos f espera?
"Não" é uma resposta bem se você sabe com certeza. Não é um grande negócio para usar dois memoize funções separadas.
Solução
Eu acho que você poderia ir para a informação de depuração e determinar isso a partir do código-fonte, mas basicamente é um "não", desculpe.
Outras dicas
Sim, para funções Lua, mas não funções C. É uma torturante pouco e um pouco esboçado.
debug.getlocal
funciona em chamadas funções então você tem que chamar a função em questão. Ele não mostra qualquer sinal de ...
a menos que a chamada passa parâmetros suficientes. O código abaixo tentativas 20 parâmetros.
debug.sethook
com o evento "chamada" dá uma oportunidade para interceptar a função antes de executar qualquer código.
Este algoritmo funciona com Lua 5.2. Versões mais antigas seria semelhante, mas não é o mesmo:
assert(_VERSION=="Lua 5.2", "Must be compatible with Lua 5.2")
Um iterador Little Helper (poderia ser embutido para a eficiência):
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
Retorna a assinatura (mas poderia retornar uma contagem de parâmetros e usesVarargs, em vez disso):
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
Eu tenho certeza que você não pode fazer isso em Lua.