Metamეტody arytmetyczne — część 1

Nauczysz się, jak dostosować to, co się dzieje podczas drukowania tablicy za pomocą __tostring. Teraz dowiesz się, że metatabele mogą również definiować, co się dzieje, gdy używasz operatorów arytmetycznych, takich jak + i -, na swoich tablicach.

Zwykle, jeśli spróbujesz dodać do siebie dwie tablice za pomocą operatora +, Lua zgłosi błąd, ponieważ nie wie, jak dodawać tablice. Dzięki metametodom arytmetycznym możesz jednak dokładnie określić, co oznacza dodawanie w przypadku własnych typów danych.

Metametoda __add kontroluje zachowanie operatora dodawania. Gdy Lua napotka operator + między dwiema wartościami i co najmniej jedna z nich ma metatabelę z __add, wywołuje tę funkcję zamiast zgłaszać błąd:

local v1 = {x = 3, y = 4}
local v2 = {x = 1, y = 2}

local meta = {
    __add = function(a, b)
        return {x = a.x + b.x, y = a.y + b.y}
    end
}

setmetatable(v1, meta)
setmetatable(v2, meta)

local v3 = v1 + v2
print(v3.x, v3.y)  -- Wynik: 4  6

Funkcja __add otrzymuje dwa argumenty: lewy operand i prawy operand. W tym przykładzie tworzy nową tablicę z wartościami x i y, które są sumą odpowiednich wartości z obu tablic wejściowych.

Podobnie metametoda __sub kontroluje odejmowanie za pomocą operatora -:

local meta = {
    __sub = function(a, b)
        return {x = a.x - b.x, y = a.y - b.y}
    end
}

setmetatable(v1, meta)
setmetatable(v2, meta)

local v3 = v1 - v2
print(v3.x, v3.y)  -- Wynik: 2  2

Ta możliwość definiowania niestandardowego działania operatorów nazywa się przeciążaniem operatorów.