Math for Color class

garrysmod/lua/includes/util/color.lua

@@ -46,6 +46,83 @@ function COLOR:__tostring()
 
 end
 
+--[[---------------------------------------------------------
+	Returns unary color
+-----------------------------------------------------------]]
+function COLOR:__unm()
+
+	return self:GetInverted()
+
+end
+
+--[[---------------------------------------------------------
+	Returns colors summ
+-----------------------------------------------------------]]
+
+function COLOR:__add( other )
+
+	return Color( self.r + other.r, self.g + other.g, self.b + other.b, self.a + other.a )
+
+end
+
+--[[---------------------------------------------------------
+	Returns colors subtraction
+-----------------------------------------------------------]]
+
+function COLOR:__sub(other)
+
+	return Color( self.r - other.r, self.g - other.g, self.b - other.b, self.a - other.a )
+
+end
+
+--[[---------------------------------------------------------
+	Returns color multiplication with another color or number
+-----------------------------------------------------------]]
+
+function COLOR:__mul(other)
+
+	if ( type( other ) == "number" ) then
+		return Color( self.r * other, self.g * other, self.b * other, self.a * other )
+	end
+
+	return Color( self.r * other.r, self.g * other.g, self.b * other.b, self.a * other.a )
+
+end
+
+--[[---------------------------------------------------------
+	Returns color division with another color or number
+-----------------------------------------------------------]]
+
+function COLOR:__div( other )
+
+	if ( type( other ) == "number" ) then
+		return Color( self.r / other, self.g / other, self.b / other, self.a / other )
+	end
+
+	return Color( self.r / other.r, self.g / other.g, self.b / other.b, self.a / other.a )
+
+end
+
+--[[---------------------------------------------------------
+	Returns is color darker than other
+-----------------------------------------------------------]]
+
+function COLOR:__lt( other )
+
+	return select( 3, self:ToHSL() ) < select( 3, other:ToHSL() )
+
+end
+
+--[[---------------------------------------------------------
+	Returns is color darker than other or have the same lightness
+-----------------------------------------------------------]]
+
+function COLOR:__le( other )
+
+	return select( 3, self:ToHSL() ) <= select( 3, other:ToHSL() )
+
+end
+
 --[[---------------------------------------------------------
 	Compares two colors
 -----------------------------------------------------------]]
@@ -105,3 +182,80 @@ function COLOR:ToTable()
 	return { self.r, self.g, self.b, self.a }
 
 end
+
+function COLOR:Copy()
+
+	return Color( self.r, self.g, self.b, self.a )
+
+end
+
+function COLOR:Invert()
+
+	self.r = 255 - self.r
+	self.g = 255 - self.g
+	self.b = 255 - self.b
+
+end
+
+function COLOR:GetInverted()
+
+	local col = self:Copy()
+	col:Invert()
+
+	return col
+
+end
+
+function COLOR:Normalize()
+
+	self.r = self.r / 255
+	self.g = self.g / 255
+	self.b = self.b / 255
+	self.a = self.a / 255
+
+end
+
+function COLOR:GetNormalized()
+
+	local col = self:Copy()
+	col:Normalize()
+
+	return col
+
+end
+
+function COLOR:Add( other )
+
+	self.r = self.r + other.r
+	self.g = self.g + other.g
+	self.b = self.b + other.b
+	self.a = self.a + other.a
+
+end
+
+function COLOR:Sub( other )
+
+	self.r = self.r - other.r
+	self.g = self.g - other.g
+	self.b = self.b - other.b
+	self.a = self.a - other.a
+
+end
+
+function COLOR:Mul( multiplier )
+
+	self.r = self.r * multiplier
+	self.g = self.g * multiplier
+	self.b = self.b * multiplier
+	self.a = self.a * multiplier
+
+end
+
+function COLOR:Div( divisor )
+
+	self.r = self.r / divisor
+	self.g = self.g / divisor
+	self.b = self.b / divisor
+	self.a = self.a / divisor
+
+end