generics - ipairs - ¿Hay alguna forma de determinar la firma de una función Lua?

ipairs lua

Question

Recientemente, Lee Baldwin mostró cómo escribir una función de memorización de argumento variable y genérica . Pensé que sería mejor devolver una función más simple donde solo se requiere un parámetro. Aquí está mi intento falso total:

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 no haría realmente lo que quiero de todos modos. ¿Hay alguna manera de decir en el interior cómo memozar cuántos argumentos espera?

"No" es una buena respuesta si lo sabes con certeza. No es un gran problema usar dos funciones separadas para memorizar .

Answer 1

Estoy bastante seguro de que no puedes hacer eso en Lua.

Answer 2

Supongo que podría entrar en la información de depuración y determinar esto desde el código fuente, pero básicamente es un "no", lo siento.

Answer 3

Sí, para las funciones Lua pero no para las funciones C. Es un poco tortuoso y un poco superficial.

debug.getlocal funciona en las funciones llamadas, por lo que debe llamar a la función en cuestión. No muestra ninguna pista de ... menos que la llamada pase suficientes parámetros. El siguiente código intenta 20 parámetros.

debug.sethook con el evento "call" da 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 alineado 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 conteo de parámetros y usaVarjetas, 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