From 80e3ace8fc64bf047300f011ba0d29679f1034b4 Mon Sep 17 00:00:00 2001
From: Bruno BELANYI <bruno@belanyi.fr>
Date: Mon, 25 Jul 2022 17:34:11 +0200
Subject: [PATCH] Add '*Assign' operators to 'Bitboard'

---
 src/board/bitboard/mod.rs | 80 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)

diff --git a/src/board/bitboard/mod.rs b/src/board/bitboard/mod.rs
index d5ca8eb..8eff5e1 100644
--- a/src/board/bitboard/mod.rs
+++ b/src/board/bitboard/mod.rs
@@ -117,6 +117,22 @@ impl std::ops::Shr<usize> for Bitboard {
     }
 }
 
+/// Treat bitboard as a set of squares, shift each square's index left by the amount given.
+impl std::ops::ShlAssign<usize> for Bitboard {
+    #[inline(always)]
+    fn shl_assign(&mut self, rhs: usize) {
+        *self = *self << rhs;
+    }
+}
+
+/// Treat bitboard as a set of squares, shift each square's index right by the amount given.
+impl std::ops::ShrAssign<usize> for Bitboard {
+    #[inline(always)]
+    fn shr_assign(&mut self, rhs: usize) {
+        *self = *self >> rhs;
+    }
+}
+
 /// Treat bitboard as a set of squares, and invert the set.
 impl std::ops::Not for Bitboard {
     type Output = Bitboard;
@@ -147,6 +163,22 @@ impl std::ops::BitOr<Square> for Bitboard {
     }
 }
 
+/// Treat each bitboard as a set of squares, keep squares that are in either sets.
+impl std::ops::BitOrAssign<Bitboard> for Bitboard {
+    #[inline(always)]
+    fn bitor_assign(&mut self, rhs: Bitboard) {
+        *self = *self | rhs;
+    }
+}
+
+/// Treat the [Square] as a singleton bitboard, and apply the operator.
+impl std::ops::BitOrAssign<Square> for Bitboard {
+    #[inline(always)]
+    fn bitor_assign(&mut self, rhs: Square) {
+        *self = *self | rhs;
+    }
+}
+
 /// Treat each bitboard as a set of squares, keep squares that are in both sets.
 impl std::ops::BitAnd<Bitboard> for Bitboard {
     type Output = Bitboard;
@@ -167,6 +199,22 @@ impl std::ops::BitAnd<Square> for Bitboard {
     }
 }
 
+/// Treat each bitboard as a set of squares, keep squares that are in both sets.
+impl std::ops::BitAndAssign<Bitboard> for Bitboard {
+    #[inline(always)]
+    fn bitand_assign(&mut self, rhs: Bitboard) {
+        *self = *self & rhs;
+    }
+}
+
+/// Treat the [Square] as a singleton bitboard, and apply the operator.
+impl std::ops::BitAndAssign<Square> for Bitboard {
+    #[inline(always)]
+    fn bitand_assign(&mut self, rhs: Square) {
+        *self = *self & rhs;
+    }
+}
+
 /// Treat each bitboard as a set of squares, keep squares that are in exactly one of either set.
 impl std::ops::BitXor<Bitboard> for Bitboard {
     type Output = Bitboard;
@@ -187,6 +235,22 @@ impl std::ops::BitXor<Square> for Bitboard {
     }
 }
 
+/// Treat each bitboard as a set of squares, keep squares that are in exactly one of either set.
+impl std::ops::BitXorAssign<Bitboard> for Bitboard {
+    #[inline(always)]
+    fn bitxor_assign(&mut self, rhs: Bitboard) {
+        *self = *self ^ rhs;
+    }
+}
+
+/// Treat the [Square] as a singleton bitboard, and apply the operator.
+impl std::ops::BitXorAssign<Square> for Bitboard {
+    #[inline(always)]
+    fn bitxor_assign(&mut self, rhs: Square) {
+        *self = *self ^ rhs;
+    }
+}
+
 /// Treat each bitboard as a set of squares, and substract one set from another.
 impl std::ops::Sub<Bitboard> for Bitboard {
     type Output = Bitboard;
@@ -207,6 +271,22 @@ impl std::ops::Sub<Square> for Bitboard {
     }
 }
 
+/// Treat each bitboard as a set of squares, and substract one set from another.
+impl std::ops::SubAssign<Bitboard> for Bitboard {
+    #[inline(always)]
+    fn sub_assign(&mut self, rhs: Bitboard) {
+        *self = *self - rhs;
+    }
+}
+
+/// Treat the [Square] as a singleton bitboard, and apply the operator.
+impl std::ops::SubAssign<Square> for Bitboard {
+    #[inline(always)]
+    fn sub_assign(&mut self, rhs: Square) {
+        *self = *self - rhs;
+    }
+}
+
 #[cfg(test)]
 mod test {
     use std::collections::HashSet;