ambroisie/seer
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Refactor castling-rights handling in 'play_move'
All checks were successful
ci/woodpecker/push/check Pipeline was successful
Details

Browse source
This commit is contained in:
Bruno BELANYI 2024-04-06 20:26:04 +01:00
parent d729d63c75
commit 8be7105ac5
1 changed files with 16 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
src/board/chess_board/mod.rs
View file

@ -131,6 +131,20 @@ impl ChessBoard {
self.combined_occupancy ^= square; self.combined_occupancy ^= square;
} }
/// Compute the change of [CastleRights] from moving/taking a piece.
fn update_castling(&mut self, color: Color, piece: Piece, file: File) {
let original = self.castle_rights(color);
let new_rights = match (piece, file) {
(Piece::Rook, File::A) => original.without_queen_side(),
(Piece::Rook, File::H) => original.without_king_side(),
(Piece::King, _) => CastleRights::NoSide,
_ => return,
};
if new_rights != original {
*self.castle_rights_mut(color) = new_rights;
}
}
/// Play the given [Move], return a copy of the board with the resulting state. /// Play the given [Move], return a copy of the board with the resulting state.
#[inline(always)] #[inline(always)]
pub fn play_move(&self, chess_move: Move) -> Self { pub fn play_move(&self, chess_move: Move) -> Self {
@ -177,22 +191,11 @@ impl ChessBoard {
} else { } else {
self.en_passant = None; self.en_passant = None;
} }
let castle_rights = self.castle_rights_mut(self.current_player()); self.update_castling(self.current_player(), move_piece, chess_move.start().file());
*castle_rights = match (move_piece, chess_move.start().file()) {
(Piece::Rook, File::A) => castle_rights.without_queen_side(),
(Piece::Rook, File::H) => castle_rights.without_king_side(),
(Piece::King, _) => CastleRights::NoSide,
_ => *castle_rights,
};
if let Some(piece) = captured_piece { if let Some(piece) = captured_piece {
self.xor(opponent, piece, chess_move.destination()); self.xor(opponent, piece, chess_move.destination());
// If a rook is captured, it loses its castling rights // If a rook is captured, it loses its castling rights
let castle_rights = self.castle_rights_mut(opponent); self.update_castling(opponent, piece, chess_move.destination().file());
*castle_rights = match (piece, chess_move.destination().file()) {
(Piece::Rook, File::A) => castle_rights.without_queen_side(),
(Piece::Rook, File::H) => castle_rights.without_king_side(),
_ => *castle_rights,
};
} }
// Revertible state modification // Revertible state modification