¿Hay una manera de determinar la firma de una función de Lua?

Question

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.

Answer 1

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.

Answer 2

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

Answer 3

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