¿Hay una manera de determinar la firma de una función de Lua?
-
02-07-2019 - |
Pregunta
Recientemente, Lee Baldwin mostró cómo escribir un genérico, función variable de memoize de argumento . Pensé que sería mejor devolver una función más simple donde solo se requiere un parámetro. Aquí está mi intento total 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 ('#', ...)
falla en este contexto y realmente no haría lo que quiero de todos modos. ¿Hay alguna forma de saber dentro de memoize cuántos argumentos f espera?
" No " Es una buena respuesta si lo sabes seguro. No es un gran problema usar dos funciones memoize separadas.
Solución
Supongo que podrías entrar en la información de depuración y determinar esto a partir del código fuente, pero básicamente es un " no " ;, lo siento.
Otros consejos
Sí, para funciones Lua pero no para C Es un poco tortuoso y un poco vago.
debug.getlocal
funciona en las funciones llamadas, por lo que debe llamar a la función en cuestión. No muestra ningún indicio de ...
a menos que la llamada pase suficientes parámetros. El siguiente código intenta 20 parámetros.
debug.sethook
con la " llamada " El evento brinda la oportunidad de interceptar la función antes de ejecutar cualquier código.
Este algoritmo funciona con Lua 5.2. Las versiones anteriores serían similares pero no iguales:
assert(_VERSION=="Lua 5.2", "Must be compatible with Lua 5.2")
Un pequeño iterador auxiliar (podría estar en línea para mayor eficiencia):
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
Devuelve la firma (pero podría devolver un recuento de parámetros y usaVarargs, en su lugar):
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
Estoy bastante seguro de que no puedes hacer eso en Lua.