Métaméthodes arithmétiques - Partie 1

Vous avez appris comment personnaliser ce qui se passe lorsque vous affichez une table en utilisant __tostring. Maintenant, vous allez découvrir que les métatables peuvent également définir ce qui se passe lorsque vous utilisez des opérateurs arithmétiques comme + et - sur vos tables.

Normalement, si vous essayez d'additionner deux tables ensemble avec l'opérateur +, Lua vous donnera une erreur parce qu'il ne sait pas comment additionner des tables. Mais avec les métaméthodes arithmétiques, vous pouvez définir exactement ce que l'addition signifie pour vos types de données personnalisés.

La métaméthode __add contrôle le comportement de l'addition. Lorsque Lua rencontre l'opérateur + entre deux valeurs et qu'au moins l'une d'elles possède une métatable avec __add, il appelle cette fonction au lieu de produire une erreur :

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)  -- Output: 4  6

La fonction __add reçoit deux arguments : l'opérande de gauche et l'opérande de droite. Dans cet exemple, elle crée une nouvelle table avec des valeurs x et y qui sont la somme des valeurs correspondantes des deux tables d'entrée.

De même, la métaméthode __sub contrôle la soustraction avec l'opérateur - :

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)  -- Sortie : 2  2

Cette capacité à définir le comportement d'opérateurs personnalisés est appelée surcharge d'opérateur.