Add 'ChessBoard::at'
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
Bruno BELANYI 2022-07-30 12:26:33 +02:00
parent 87258c1084
commit 1a7763e1f4

View file

@ -119,6 +119,24 @@ impl ChessBoard {
self.compute_checkers(self.current_player())
}
/// Convenience function which returns the [Piece] and [Color] at a given [Square], or `None`
/// if it is empty.
pub fn at(&self, square: Square) -> Option<(Piece, Color)> {
let color = if !(self.color_occupancy(Color::White) & square).is_empty() {
Color::White
} else {
Color::Black
};
for piece in Piece::iter() {
if !(self.piece_occupancy(piece) & square).is_empty() {
return Some((piece, color));
}
}
None
}
/// Quickly do and undo a move on the [Bitboard]s that are part of the [ChessBoard] state. Does
/// not account for all non-revertible changes such as en-passant state or half-move clock.
#[inline(always)]
@ -775,6 +793,38 @@ mod test {
);
}
#[test]
fn at() {
let default_position = ChessBoard::default();
assert_eq!(default_position.at(Square::D4), None);
assert_eq!(default_position.at(Square::D5), None);
assert_eq!(
default_position.at(Square::A1).unwrap(),
(Piece::Rook, Color::White)
);
assert_eq!(
default_position.at(Square::H8).unwrap(),
(Piece::Rook, Color::Black)
);
assert_eq!(
default_position.at(Square::D1).unwrap(),
(Piece::Queen, Color::White)
);
assert_eq!(
default_position.at(Square::D8).unwrap(),
(Piece::Queen, Color::Black)
);
assert_eq!(
default_position.at(Square::C2).unwrap(),
(Piece::Pawn, Color::White)
);
assert_eq!(
default_position.at(Square::F7).unwrap(),
(Piece::Pawn, Color::Black)
);
}
#[test]
fn fen_default_position() {
let default_position = ChessBoard::default();