Add RGB minus operator

RoboticsBrno · Feb 1, 2023 · 832939c · 832939c
1 parent d546c82
commit 832939c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/Color.cpp b/src/Color.cpp
@@ -79,6 +79,19 @@ Rgb& Rgb::operator+=( Rgb in ) {
     return *this;
 }
 
+Rgb Rgb::operator-( Rgb in ) const {
+    auto copy = *this;
+    copy -= in;
+    return copy;
+}
+
+Rgb& Rgb::operator-=( Rgb in ) {
+    r = ( in.r > r ) ? 0 : r - in.r;
+    g = ( in.g > g ) ? 0 : g - in.g;
+    b = ( in.b > b ) ? 0 : b - in.b;
+    return *this;
+}
+
 Rgb& Rgb::blend( Rgb in ) {
     unsigned int inAlpha = in.a * ( 255 - a );
     unsigned int alpha = a + inAlpha;

diff --git a/src/Color.h b/src/Color.h
@@ -16,6 +16,8 @@ union Rgb {
     Rgb& operator=( Hsv hsv );
     Rgb operator+( Rgb in ) const;
     Rgb& operator+=( Rgb in );
+    Rgb operator-(Rgb in) const;
+    Rgb &operator-=(Rgb in);
     bool operator==( Rgb in ) const { return in.value == value; }
     Rgb& blend( Rgb in );
     void swap( Rgb& o ) {  value = o.value; }