Funkce jako hodnoty

V Lua je funkce hodnota, stejně jako číslo, text nebo tabulka. Potvrdí to i type: type(print) je "function". A jako každou hodnotu můžete funkci uložit do proměnné. local function dvakrat(n) je vlastně jen kratší způsob, jak vytvořit funkci a uložit ji do lokální proměnné, takhle:

local dvakrat = function(n)
    return n * 2
end

local dvojnasobek = dvakrat
print(dvakrat(4), dvojnasobek(10))
print(type(dvakrat), type(dvakrat(4)))

Vypíše:

8	20
function	number

function(n) … end bez názvu vytvoří funkci a local dvakrat = ji uloží do proměnné. local dvojnasobek = dvakrat nic nevolá, protože tam nejsou závorky: uloží stejnou funkci pod druhým názvem. type(dvakrat) se ptá na samotnou funkci, kdežto type(dvakrat(4)) na číslo, které funkce vrátila.

A teď to nejužitečnější: funkci můžete předat jiné funkci jako argument. Tato funkce projde seznam a každý prvek upraví funkcí, kterou dostane jako parametr:

local function upravVse(seznam, uprava)
    local vysledek = {}
    for _, hodnota in ipairs(seznam) do
        table.insert(vysledek, uprava(hodnota))
    end
    return vysledek
end

local function dvakrat(n)
    return n * 2
end

local ceny = {2, 3, 5}
print(table.concat(upravVse(ceny, dvakrat), " "))
print(table.concat(upravVse(ceny, function(n) return n + 1 end), " "))

Vypíše:

4 6 10
3 4 6

Uvnitř funkce upravVse je uprava parametr jako každý jiný, jen je v něm funkce, a tak ji upravVse volá: uprava(hodnota). upravVse neví, jaká úprava to bude, o tom rozhoduje volající. Druhé volání svou funkci ani nepojmenuje, napíše ji rovnou do závorek volání. Funkci bez názvu se říká anonymní funkce. Hodí se pro krátké funkce, které potřebujete jen na jednom místě, a krátká se vejde i na jeden řádek.

Když funkci předáváte, pište ji bez závorek: dvakrat, ne dvakrat(5). Se závorkami byste ji rovnou zavolali a předali její výsledek, číslo 10:

local function upravVse(seznam, uprava)
    local vysledek = {}
    for _, hodnota in ipairs(seznam) do
        table.insert(vysledek, uprava(hodnota))
    end
    return vysledek
end

local function dvakrat(n)
    return n * 2
end

local ceny = {2, 3, 5}
print(table.concat(upravVse(ceny, dvakrat(5)), " "))

Program skončí chybou:

main.lua:4: attempt to call a number value (local 'uprava')

upravVse dostala v parametru uprava číslo 10 a pokusila se ho zavolat. Funkce předávané jiným funkcím uvidíte často: v dalším cvičení jednu předáte funkci table.sort.