Lua
Вариадические аргументы

Вступление

Синтаксис

• ... - Делает функцию, список аргументов которой, в которой появляется переменная функция
• select (what, ...) - Если «what» - это число в диапазоне 1 к числу элементов в vararg, возвращает элемент «what'th» последнему элементу в vararg. Возврат будет равен нулю, если индекс не соответствует границам. Если «что» является строкой «#», возвращается количество элементов в vararg.

замечания

for i = 1, select('#', ...) do
    print(select(i, ...))
end

function test(...)
    return select('#', ...)
end

test()             --> 0
test(nil, 1, nil)  --> 3

основы

Функции Variadic создаются с использованием синтаксиса ... эллипсов в списке аргументов определения функции.

function id(...)
    return
end

Если вы назвали эту функцию id(1, 2, 3, 4, 5) тогда ... (AKA - список vararg) будет содержать значения 1, 2, 3, 4, 5 .

Функции могут принимать требуемые аргументы, а также ...

function head(x, ...)
    return x
end

Самый простой способ вытащить элементы из списка vararg - просто назначить переменные из него.

function head3(...)
    local a, b, c = ...
    return a, b, c
end

select() также может использоваться для поиска количества элементов и извлечения элементов из ... косвенно.

function my_print(...)
    for i = 1, select('#', ...) do
        io.write(tostring(select(i, ...)) .. '\t')
    end
    io.write '\n'
end

... могут быть упакованы в таблицу для удобства использования, используя {...} . Это помещает все аргументы в последовательную часть таблицы.

function my_tablepack(...)
    local t = {...}
    t.n = select('#', ...)
    return t
end

Список vararg также может быть возвращен из функций. Результат - несколько возвратов.

function all_or_none(...)
    local t = table.pack(...)
    for i = 1, t.n do
        if not t[i] then
            return    -- return none
        end
    end
    return ...    -- return all
end

Расширенное использование

Как указано в основных примерах, вы можете иметь переменные связанные переменные и список аргументов переменных ( ... ). Вы можете использовать этот факт, чтобы рекурсивно разделить список, как на других языках (например, Haskell). Ниже приведена реализация foldr() которая использует это. Каждый рекурсивный вызов связывает головку списка vararg с x и передает остальную часть списка рекурсивному вызову. Это разрушает список, пока не будет только один аргумент ( select('#', ...) == 0 ). После этого каждое значение применяется к аргументу функции f с ранее вычисленным результатом.

function foldr(f, ...)
    if select('#', ...) < 2 then return ... end
    local function helper(x, ...)
        if select('#', ...) == 0 then
          return x
        end
        return f(x, helper(...))
    end
    return helper(...)
end

function sum(a, b)
    return a + b
end

foldr(sum, 1, 2, 3, 4)
--> 10    

Вы можете найти другие определения функций, которые используют этот стиль программирования здесь в выпуске № 3 по номеру 8.

Единственная идиоматическая структура данных Lua - это таблица. Оператор длины таблицы не определен, если nil s находится где угодно в последовательности. В отличие от таблиц, список vararg учитывает явные значения nil s, как указано в базовых примерах и в разделе замечаний (пожалуйста, прочитайте этот раздел, если вы еще этого не сделали). При небольшой работе список vararg может выполнять каждую операцию, кроме таблицы, помимо мутации. Это делает список vararg хорошим кандидатом для реализации неизменяемых кортежей.

function tuple(...)
    -- packages a vararg list into an easily passable value
    local co = coroutine.wrap(function(...)
        coroutine.yield()
        while true do
            coroutine.yield(...)
        end
    end)
    co(...)
    return co
end

local t = tuple((function() return 1, 2, nil, 4, 5 end)())

print(t())                 --> 1    2    nil    4    5    | easily unpack for multiple args
local a, b, d = t()        --> a = 1, b = 2, c = nil      | destructure the tuple
print((select(4, t())))    --> 4                          | index the tuple
print(select('#', t()))    --> 5                          | find the tuple arity (nil respecting)

local function change_index(tpl, i, v)
    -- sets a value at an index in a tuple (non-mutating)
    local function helper(n, x, ...)
        if select('#', ...) == 0 then
            if n == i then
                return v
            else
                return x
            end
        else
            if n == i then
                return v, helper(n+1, ...)
            else
                return x, helper(n+1, ...)
            end
        end
    end
    return tuple(helper(1, tpl()))
end

local n = change_index(t, 3, 3)
print(t())                 --> 1    2    nil    4    5
print(n())                 --> 1    2    3    4    5

Основное различие между тем, что выше и таблицами, состоит в том, что таблицы изменяемы и имеют семантику указателя, где кортеж не имеет этих свойств. Кроме того, кортежи могут содержать явные nil s и иметь операцию с бесконечной длиной.