[position, board] Move castle, movement, and sight modules to the board crate
Nothing about this code depends on Position, so push it down to a lower layer.
This commit is contained in:
parent
9a4fa827f9
commit
dbca7b4f88
7 changed files with 334 additions and 80 deletions
|
@ -5,3 +5,232 @@ mod rights;
|
|||
|
||||
pub use parameters::Parameters;
|
||||
pub use rights::Rights;
|
||||
|
||||
use crate::Board;
|
||||
use chessfriend_core::{Color, Piece, Square, Wing};
|
||||
use thiserror::Error;
|
||||
|
||||
#[derive(Clone, Copy, Debug, Error, Eq, PartialEq)]
|
||||
pub enum CastleEvaluationError {
|
||||
#[error("{color} does not have the right to castle {wing}")]
|
||||
NoRights { color: Color, wing: Wing },
|
||||
#[error("no king")]
|
||||
NoKing,
|
||||
#[error("no rook")]
|
||||
NoRook,
|
||||
#[error("castling path is not clear")]
|
||||
ObstructingPieces,
|
||||
#[error("opposing pieces check castling path")]
|
||||
CheckingPieces,
|
||||
}
|
||||
|
||||
impl Board {
|
||||
/// Evaluates whether the active color can castle toward the given wing of the board in the
|
||||
/// current position.
|
||||
///
|
||||
/// ## Errors
|
||||
///
|
||||
/// Returns an error indicating why the active color cannot castle.
|
||||
pub fn active_color_can_castle(&self, wing: Wing) -> Result<(), CastleEvaluationError> {
|
||||
// TODO: Cache this result. It's expensive!
|
||||
// TODO: Does this actually need to rely on internal state, i.e. active_color?
|
||||
|
||||
let active_color = self.active_color;
|
||||
|
||||
if !self.castling_rights.color_has_right(active_color, wing) {
|
||||
return Err(CastleEvaluationError::NoRights {
|
||||
color: active_color,
|
||||
wing,
|
||||
});
|
||||
}
|
||||
|
||||
let parameters = self.castling_parameters(wing);
|
||||
|
||||
if self.castling_king(parameters.origin.king).is_none() {
|
||||
return Err(CastleEvaluationError::NoKing);
|
||||
}
|
||||
|
||||
if self.castling_rook(parameters.origin.rook).is_none() {
|
||||
return Err(CastleEvaluationError::NoRook);
|
||||
}
|
||||
|
||||
// All squares must be clear.
|
||||
let has_obstructing_pieces = (self.occupancy() & parameters.clear).is_populated();
|
||||
if has_obstructing_pieces {
|
||||
return Err(CastleEvaluationError::ObstructingPieces);
|
||||
}
|
||||
|
||||
// King cannot pass through check.
|
||||
let opposing_sight = self.opposing_sight();
|
||||
let opposing_pieces_can_see_castling_path =
|
||||
(parameters.check & opposing_sight).is_populated();
|
||||
if opposing_pieces_can_see_castling_path {
|
||||
return Err(CastleEvaluationError::CheckingPieces);
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub(crate) fn castling_king(&self, square: Square) -> Option<Piece> {
|
||||
let active_color = self.active_color;
|
||||
self.get_piece(square)
|
||||
.filter(|piece| piece.color == active_color && piece.is_king())
|
||||
}
|
||||
|
||||
pub(crate) fn castling_rook(&self, square: Square) -> Option<Piece> {
|
||||
let active_color = self.active_color;
|
||||
self.get_piece(square)
|
||||
.filter(|piece| piece.color == active_color && piece.is_rook())
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::test_board;
|
||||
use chessfriend_core::{piece, Color, Wing};
|
||||
|
||||
#[test]
|
||||
fn king_on_starting_square_can_castle() {
|
||||
let pos = test_board!(
|
||||
White King on E1,
|
||||
White Rook on A1,
|
||||
White Rook on H1
|
||||
);
|
||||
|
||||
let rights = pos.castling_rights;
|
||||
assert!(rights.color_has_right(Color::White, Wing::KingSide));
|
||||
assert!(rights.color_has_right(Color::White, Wing::QueenSide));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn king_for_castle() {
|
||||
let pos = test_board![
|
||||
White King on E1,
|
||||
White Rook on H1,
|
||||
White Rook on A1,
|
||||
];
|
||||
|
||||
let kingside_parameters = pos.castling_parameters(Wing::KingSide);
|
||||
assert_eq!(
|
||||
pos.castling_king(kingside_parameters.origin.king),
|
||||
Some(piece!(White King))
|
||||
);
|
||||
|
||||
let queenside_parameters = pos.castling_parameters(Wing::QueenSide);
|
||||
assert_eq!(
|
||||
pos.castling_king(queenside_parameters.origin.king),
|
||||
Some(piece!(White King))
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn rook_for_castle() {
|
||||
let pos = test_board![
|
||||
White King on E1,
|
||||
White Rook on H1,
|
||||
];
|
||||
|
||||
let kingside_parameters = pos.castling_parameters(Wing::KingSide);
|
||||
assert_eq!(
|
||||
pos.castling_rook(kingside_parameters.origin.rook),
|
||||
Some(piece!(White Rook))
|
||||
);
|
||||
|
||||
let pos = test_board![
|
||||
White King on E1,
|
||||
White Rook on A1,
|
||||
];
|
||||
|
||||
let queenside_parameters = pos.castling_parameters(Wing::QueenSide);
|
||||
assert_eq!(
|
||||
pos.castling_rook(queenside_parameters.origin.rook),
|
||||
Some(piece!(White Rook))
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn white_can_castle() {
|
||||
let pos = test_board![
|
||||
White King on E1,
|
||||
White Rook on H1,
|
||||
White Rook on A1,
|
||||
];
|
||||
|
||||
assert_eq!(pos.active_color_can_castle(Wing::KingSide), Ok(()));
|
||||
assert_eq!(pos.active_color_can_castle(Wing::QueenSide), Ok(()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn white_cannot_castle_missing_king() {
|
||||
let pos = test_board![
|
||||
White King on E2,
|
||||
White Rook on H1,
|
||||
White Rook on A1,
|
||||
];
|
||||
|
||||
assert_eq!(
|
||||
pos.active_color_can_castle(Wing::KingSide),
|
||||
Err(CastleEvaluationError::NoKing)
|
||||
);
|
||||
assert_eq!(
|
||||
pos.active_color_can_castle(Wing::QueenSide),
|
||||
Err(CastleEvaluationError::NoKing)
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn white_cannot_castle_missing_rook() {
|
||||
let pos = test_board![
|
||||
White King on E1,
|
||||
White Rook on A1,
|
||||
];
|
||||
|
||||
assert_eq!(
|
||||
pos.active_color_can_castle(Wing::KingSide),
|
||||
Err(CastleEvaluationError::NoRook)
|
||||
);
|
||||
|
||||
let pos = test_board![
|
||||
White King on E1,
|
||||
White Rook on H1,
|
||||
];
|
||||
|
||||
assert_eq!(
|
||||
pos.active_color_can_castle(Wing::QueenSide),
|
||||
Err(CastleEvaluationError::NoRook)
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn white_cannot_castle_obstructing_piece() {
|
||||
let pos = test_board![
|
||||
White King on E1,
|
||||
White Bishop on F1,
|
||||
White Rook on H1,
|
||||
White Rook on A1,
|
||||
];
|
||||
|
||||
assert_eq!(
|
||||
pos.active_color_can_castle(Wing::KingSide),
|
||||
Err(CastleEvaluationError::ObstructingPieces)
|
||||
);
|
||||
assert_eq!(pos.active_color_can_castle(Wing::QueenSide), Ok(()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn white_cannot_castle_checking_pieces() {
|
||||
let pos = test_board![
|
||||
White King on E1,
|
||||
White Rook on H1,
|
||||
White Rook on A1,
|
||||
Black Queen on C6,
|
||||
];
|
||||
|
||||
assert_eq!(pos.active_color_can_castle(Wing::KingSide), Ok(()));
|
||||
assert_eq!(
|
||||
pos.active_color_can_castle(Wing::QueenSide),
|
||||
Err(CastleEvaluationError::CheckingPieces)
|
||||
);
|
||||
}
|
||||
}
|
||||
|
|
|
@ -5,6 +5,8 @@ pub mod display;
|
|||
pub mod en_passant;
|
||||
pub mod fen;
|
||||
pub mod macros;
|
||||
pub mod movement;
|
||||
pub mod sight;
|
||||
|
||||
mod board;
|
||||
mod piece_sets;
|
||||
|
|
151
board/src/movement.rs
Normal file
151
board/src/movement.rs
Normal file
|
@ -0,0 +1,151 @@
|
|||
// Eryn Wells <eryn@erynwells.me>
|
||||
|
||||
//! Defines routines for computing the movement of a piece. Movement is the set
|
||||
//! of squares a piece can move to. For all pieces except pawns, the Movement
|
||||
//! set is equal to the Sight set.
|
||||
|
||||
use crate::{sight::Sight, Board};
|
||||
use chessfriend_bitboard::BitBoard;
|
||||
use chessfriend_core::{Color, Piece, Rank, Shape, Square, Wing};
|
||||
|
||||
impl Board {
|
||||
pub fn movement(&self, square: Square) -> BitBoard {
|
||||
if let Some(piece) = self.get_piece(square) {
|
||||
piece.movement(square, self)
|
||||
} else {
|
||||
BitBoard::empty()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub trait Movement {
|
||||
fn movement(&self, square: Square, board: &Board) -> BitBoard;
|
||||
}
|
||||
|
||||
impl Movement for Piece {
|
||||
fn movement(&self, square: Square, board: &Board) -> BitBoard {
|
||||
let opposing_occupancy = board.opposing_occupancy(self.color);
|
||||
|
||||
match self.shape {
|
||||
Shape::Pawn => {
|
||||
let en_passant_square: BitBoard = board.en_passant_target.into();
|
||||
// Pawns can only move to squares they can see to capture.
|
||||
let sight = self.sight(square, board) & (opposing_occupancy | en_passant_square);
|
||||
let pushes = pawn_pushes(square.into(), self.color, board.occupancy());
|
||||
sight | pushes
|
||||
}
|
||||
Shape::King => {
|
||||
let kingside_target_square =
|
||||
if board.active_color_can_castle(Wing::KingSide).is_ok() {
|
||||
let parameters = board.castling_parameters(Wing::KingSide);
|
||||
parameters.target.king.into()
|
||||
} else {
|
||||
BitBoard::empty()
|
||||
};
|
||||
|
||||
let queenside_target_square =
|
||||
if board.active_color_can_castle(Wing::QueenSide).is_ok() {
|
||||
let parameters = board.castling_parameters(Wing::QueenSide);
|
||||
parameters.target.king.into()
|
||||
} else {
|
||||
BitBoard::empty()
|
||||
};
|
||||
|
||||
self.sight(square, board) | kingside_target_square | queenside_target_square
|
||||
}
|
||||
_ => self.sight(square, board),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn pawn_pushes(pawn: BitBoard, color: Color, occupancy: BitBoard) -> BitBoard {
|
||||
let vacancy = !occupancy;
|
||||
|
||||
match color {
|
||||
Color::White => {
|
||||
let second_rank = BitBoard::rank(&Rank::TWO.into());
|
||||
|
||||
let mut pushes = pawn.shift_north_one() & vacancy;
|
||||
if !(pawn & second_rank).is_empty() {
|
||||
// Double push
|
||||
pushes = pushes | (pushes.shift_north_one() & vacancy);
|
||||
}
|
||||
|
||||
pushes
|
||||
}
|
||||
Color::Black => {
|
||||
let seventh_rank = BitBoard::rank(&Rank::SEVEN.into());
|
||||
|
||||
let mut pushes = pawn.shift_south_one() & vacancy;
|
||||
if !(pawn & seventh_rank).is_empty() {
|
||||
// Double push
|
||||
pushes = pushes | (pushes.shift_south_one() & vacancy);
|
||||
}
|
||||
|
||||
pushes
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::pawn_pushes;
|
||||
use chessfriend_bitboard::{bitboard, BitBoard};
|
||||
use chessfriend_core::{Color, Square};
|
||||
|
||||
#[test]
|
||||
fn white_pushes_empty_board() {
|
||||
assert_eq!(
|
||||
pawn_pushes(Square::E4.into(), Color::White, BitBoard::empty()),
|
||||
bitboard![E5]
|
||||
);
|
||||
assert_eq!(
|
||||
pawn_pushes(Square::E2.into(), Color::White, BitBoard::empty()),
|
||||
bitboard![E3 E4]
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn black_pawn_empty_board() {
|
||||
assert_eq!(
|
||||
pawn_pushes(Square::A4.into(), Color::Black, BitBoard::empty()),
|
||||
bitboard![A3]
|
||||
);
|
||||
assert_eq!(
|
||||
pawn_pushes(Square::B7.into(), Color::Black, BitBoard::empty()),
|
||||
bitboard![B6 B5]
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn white_pushes_blocker() {
|
||||
assert_eq!(
|
||||
pawn_pushes(Square::C5.into(), Color::White, bitboard![C6]),
|
||||
BitBoard::empty()
|
||||
);
|
||||
assert_eq!(
|
||||
pawn_pushes(Square::D2.into(), Color::White, bitboard![D4]),
|
||||
bitboard![D3]
|
||||
);
|
||||
assert_eq!(
|
||||
pawn_pushes(Square::D2.into(), Color::White, bitboard![D3]),
|
||||
BitBoard::empty()
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn black_pushes_blocker() {
|
||||
assert_eq!(
|
||||
pawn_pushes(Square::C5.into(), Color::Black, bitboard![C4]),
|
||||
BitBoard::empty()
|
||||
);
|
||||
assert_eq!(
|
||||
pawn_pushes(Square::D7.into(), Color::Black, bitboard![D5]),
|
||||
bitboard![D6]
|
||||
);
|
||||
assert_eq!(
|
||||
pawn_pushes(Square::D7.into(), Color::Black, bitboard![D6]),
|
||||
BitBoard::empty()
|
||||
);
|
||||
}
|
||||
}
|
380
board/src/sight.rs
Normal file
380
board/src/sight.rs
Normal file
|
@ -0,0 +1,380 @@
|
|||
// Eryn Wells <eryn@erynwells.me>
|
||||
|
||||
//! Defines routines for computing sight of a piece. Sight is the set of squares
|
||||
//! that a piece can see. In other words, it's the set of squares attacked or
|
||||
//! controled by a piece.
|
||||
//!
|
||||
//! These functions use some common terms to describe arguments.
|
||||
//!
|
||||
//! occupancy
|
||||
//! : The set of occupied squares on the board.
|
||||
//!
|
||||
//! vacancy
|
||||
//! : The set of empty squares on the board, `!occpuancy`.
|
||||
//!
|
||||
//! blockers
|
||||
//! : The set of squares occupied by friendly pieces that block moves to that
|
||||
//! square and beyond.
|
||||
|
||||
use crate::Board;
|
||||
use chessfriend_bitboard::{BitBoard, IterationDirection};
|
||||
use chessfriend_core::{Color, Direction, Piece, Shape, Square};
|
||||
use std::ops::BitOr;
|
||||
|
||||
impl Board {
|
||||
/// Compute sight of the piece on the given square.
|
||||
pub fn sight(&self, square: Square) -> BitBoard {
|
||||
if let Some(piece) = self.get_piece(square) {
|
||||
piece.sight(square, self)
|
||||
} else {
|
||||
BitBoard::empty()
|
||||
}
|
||||
}
|
||||
|
||||
pub fn active_sight(&self) -> BitBoard {
|
||||
self.friendly_sight(self.active_color)
|
||||
}
|
||||
|
||||
/// A [`BitBoard`] of all squares the given color can see.
|
||||
pub fn friendly_sight(&self, color: Color) -> BitBoard {
|
||||
// TODO: Probably want to implement a caching layer here.
|
||||
self.friendly_occupancy(color)
|
||||
.occupied_squares(&IterationDirection::default())
|
||||
.map(|square| self.sight(square))
|
||||
.fold(BitBoard::empty(), BitOr::bitor)
|
||||
}
|
||||
|
||||
/// A [`BitBoard`] of all squares visible by colors that oppose the given color.
|
||||
pub fn opposing_sight(&self) -> BitBoard {
|
||||
// TODO: Probably want to implement a caching layer here.
|
||||
let active_color = self.active_color;
|
||||
Color::ALL
|
||||
.into_iter()
|
||||
.filter_map(|c| {
|
||||
if c == active_color {
|
||||
None
|
||||
} else {
|
||||
Some(self.friendly_sight(c))
|
||||
}
|
||||
})
|
||||
.fold(BitBoard::empty(), BitOr::bitor)
|
||||
}
|
||||
}
|
||||
|
||||
pub trait Sight {
|
||||
fn sight(&self, square: Square, board: &Board) -> BitBoard;
|
||||
}
|
||||
|
||||
impl Sight for Piece {
|
||||
fn sight(&self, square: Square, board: &Board) -> BitBoard {
|
||||
let occupancy = board.occupancy();
|
||||
let info = SightInfo {
|
||||
square,
|
||||
occupancy,
|
||||
friendly_occupancy: board.friendly_occupancy(self.color),
|
||||
};
|
||||
|
||||
match self.shape {
|
||||
Shape::Pawn => {
|
||||
let en_passant_square: BitBoard = board.en_passant_target.into();
|
||||
match self.color {
|
||||
Color::White => white_pawn_sight(&info, en_passant_square),
|
||||
Color::Black => black_pawn_sight(&info, en_passant_square),
|
||||
}
|
||||
}
|
||||
Shape::Knight => knight_sight(&info),
|
||||
Shape::Bishop => bishop_sight(&info),
|
||||
Shape::Rook => rook_sight(&info),
|
||||
Shape::Queen => queen_sight(&info),
|
||||
Shape::King => king_sight(&info),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
struct SightInfo {
|
||||
square: Square,
|
||||
occupancy: BitBoard,
|
||||
friendly_occupancy: BitBoard,
|
||||
}
|
||||
|
||||
macro_rules! ray_in_direction {
|
||||
($square:expr, $blockers:expr, $direction:ident, $first_occupied_square:tt) => {{
|
||||
let ray = BitBoard::ray($square, Direction::$direction);
|
||||
let ray_blockers = ray & $blockers;
|
||||
if let Some(first_occupied_square) = ray_blockers.$first_occupied_square() {
|
||||
let remainder = BitBoard::ray(first_occupied_square, Direction::$direction);
|
||||
let attack_ray = ray & !remainder;
|
||||
attack_ray
|
||||
} else {
|
||||
ray
|
||||
}
|
||||
}};
|
||||
}
|
||||
|
||||
/// Compute sight of a white pawn.
|
||||
fn white_pawn_sight(info: &SightInfo, en_passant_square: BitBoard) -> BitBoard {
|
||||
let possible_squares = !info.friendly_occupancy | en_passant_square;
|
||||
|
||||
let pawn: BitBoard = info.square.into();
|
||||
let pawn = pawn.shift_north_west_one() | pawn.shift_north_east_one();
|
||||
|
||||
pawn & possible_squares
|
||||
}
|
||||
|
||||
fn black_pawn_sight(info: &SightInfo, en_passant_square: BitBoard) -> BitBoard {
|
||||
let possible_squares = !info.friendly_occupancy | en_passant_square;
|
||||
|
||||
let pawn: BitBoard = info.square.into();
|
||||
let pawn = pawn.shift_south_west_one() | pawn.shift_south_east_one();
|
||||
|
||||
pawn & possible_squares
|
||||
}
|
||||
|
||||
fn knight_sight(info: &SightInfo) -> BitBoard {
|
||||
BitBoard::knight_moves(info.square)
|
||||
}
|
||||
|
||||
fn bishop_sight(info: &SightInfo) -> BitBoard {
|
||||
let bishop = info.square;
|
||||
let occupancy = info.occupancy;
|
||||
|
||||
#[rustfmt::skip]
|
||||
let sight = ray_in_direction!(bishop, occupancy, NorthEast, first_occupied_square_trailing)
|
||||
| ray_in_direction!(bishop, occupancy, SouthEast, first_occupied_square_leading)
|
||||
| ray_in_direction!(bishop, occupancy, SouthWest, first_occupied_square_leading)
|
||||
| ray_in_direction!(bishop, occupancy, NorthWest, first_occupied_square_trailing);
|
||||
|
||||
sight
|
||||
}
|
||||
|
||||
fn rook_sight(info: &SightInfo) -> BitBoard {
|
||||
let rook = info.square;
|
||||
let occupancy = info.occupancy;
|
||||
|
||||
#[rustfmt::skip]
|
||||
let sight = ray_in_direction!(rook, occupancy, North, first_occupied_square_trailing)
|
||||
| ray_in_direction!(rook, occupancy, East, first_occupied_square_trailing)
|
||||
| ray_in_direction!(rook, occupancy, South, first_occupied_square_leading)
|
||||
| ray_in_direction!(rook, occupancy, West, first_occupied_square_leading);
|
||||
|
||||
sight
|
||||
}
|
||||
|
||||
fn queen_sight(info: &SightInfo) -> BitBoard {
|
||||
let queen = info.square;
|
||||
let occupancy = info.occupancy;
|
||||
|
||||
#[rustfmt::skip]
|
||||
let sight = ray_in_direction!(queen, occupancy, NorthWest, first_occupied_square_trailing)
|
||||
| ray_in_direction!(queen, occupancy, North, first_occupied_square_trailing)
|
||||
| ray_in_direction!(queen, occupancy, NorthEast, first_occupied_square_trailing)
|
||||
| ray_in_direction!(queen, occupancy, East, first_occupied_square_trailing)
|
||||
| ray_in_direction!(queen, occupancy, SouthEast, first_occupied_square_leading)
|
||||
| ray_in_direction!(queen, occupancy, South, first_occupied_square_leading)
|
||||
| ray_in_direction!(queen, occupancy, SouthWest, first_occupied_square_leading)
|
||||
| ray_in_direction!(queen, occupancy, West, first_occupied_square_leading);
|
||||
|
||||
sight
|
||||
}
|
||||
|
||||
fn king_sight(info: &SightInfo) -> BitBoard {
|
||||
BitBoard::king_moves(info.square)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use crate::test_board;
|
||||
use chessfriend_bitboard::bitboard;
|
||||
use chessfriend_core::piece;
|
||||
|
||||
macro_rules! sight_test {
|
||||
($test_name:ident, $position:expr, $piece:expr, $square:expr, $bitboard:expr) => {
|
||||
#[test]
|
||||
fn $test_name() {
|
||||
use chessfriend_core::Square;
|
||||
use $crate::sight::Sight;
|
||||
|
||||
let pos = $position;
|
||||
let piece = $piece;
|
||||
let sight = piece.sight($square, &pos);
|
||||
|
||||
assert_eq!(sight, $bitboard);
|
||||
}
|
||||
};
|
||||
($test_name:ident, $piece:expr, $square:expr, $bitboard:expr) => {
|
||||
sight_test! {$test_name, $crate::Board::empty(), $piece, $square, $bitboard}
|
||||
};
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn friendly_sight() {
|
||||
let pos = test_board!(
|
||||
White King on E4,
|
||||
);
|
||||
|
||||
let sight = pos.active_sight();
|
||||
assert_eq!(sight, bitboard![E5 F5 F4 F3 E3 D3 D4 D5]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn opposing_sight() {
|
||||
let pos = test_board!(
|
||||
White King on E4,
|
||||
Black Rook on E7,
|
||||
);
|
||||
|
||||
let sight = pos.opposing_sight();
|
||||
assert_eq!(sight, bitboard![A7 B7 C7 D7 F7 G7 H7 E8 E6 E5 E4]);
|
||||
}
|
||||
|
||||
// #[test]
|
||||
// fn pawns_and_knights_cannot_make_rays() {
|
||||
// assert_eq!(Shape::Pawn.ray_to_square(Square::F7, Square::E8), None);
|
||||
// assert_eq!(Shape::Knight.ray_to_square(Square::F6, Square::E8), None);
|
||||
// }
|
||||
|
||||
mod pawn {
|
||||
use crate::{sight::Sight, test_board};
|
||||
use chessfriend_bitboard::{bitboard, BitBoard};
|
||||
use chessfriend_core::{piece, Square};
|
||||
|
||||
sight_test!(e4_pawn, piece!(White Pawn), Square::E4, bitboard![D5 F5]);
|
||||
|
||||
sight_test!(
|
||||
e4_pawn_one_blocker,
|
||||
test_board![
|
||||
White Bishop on D5,
|
||||
White Pawn on E4,
|
||||
],
|
||||
piece!(White Pawn),
|
||||
Square::E4,
|
||||
bitboard!(F5)
|
||||
);
|
||||
|
||||
#[test]
|
||||
fn e4_pawn_two_blocker() {
|
||||
let pos = test_board!(
|
||||
White Bishop on D5,
|
||||
White Queen on F5,
|
||||
White Pawn on E4,
|
||||
);
|
||||
|
||||
let piece = piece!(White Pawn);
|
||||
let sight = piece.sight(Square::E4, &pos);
|
||||
|
||||
assert_eq!(sight, BitBoard::empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn e4_pawn_capturable() {
|
||||
let pos = test_board!(
|
||||
Black Bishop on D5,
|
||||
White Queen on F5,
|
||||
White Pawn on E4,
|
||||
);
|
||||
|
||||
let piece = piece!(White Pawn);
|
||||
let sight = piece.sight(Square::E4, &pos);
|
||||
|
||||
assert_eq!(sight, bitboard![D5]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn e5_en_passant() {
|
||||
let pos = test_board!(White, [
|
||||
White Pawn on E5,
|
||||
Black Pawn on D5,
|
||||
], D6);
|
||||
let piece = piece!(White Pawn);
|
||||
let sight = piece.sight(Square::E5, &pos);
|
||||
|
||||
assert_eq!(sight, bitboard!(D6 F6));
|
||||
}
|
||||
}
|
||||
|
||||
#[macro_use]
|
||||
mod knight {
|
||||
use chessfriend_bitboard::bitboard;
|
||||
use chessfriend_core::piece;
|
||||
|
||||
sight_test!(
|
||||
f6_knight,
|
||||
piece!(Black Knight),
|
||||
Square::F6,
|
||||
bitboard![H7 G8 E8 D7 D5 E4 G4 H5]
|
||||
);
|
||||
}
|
||||
|
||||
mod bishop {
|
||||
use super::*;
|
||||
|
||||
sight_test!(
|
||||
c2_bishop,
|
||||
piece!(Black Bishop),
|
||||
Square::C2,
|
||||
bitboard![D1 B3 A4 B1 D3 E4 F5 G6 H7]
|
||||
);
|
||||
|
||||
// #[test]
|
||||
// fn ray_to_square() {
|
||||
// let generated_ray = Shape::Bishop.ray_to_square(Square::C5, Square::E7);
|
||||
// let expected_ray = bitboard![D6 E7];
|
||||
// assert_eq!(generated_ray, Some(expected_ray));
|
||||
// }
|
||||
}
|
||||
|
||||
mod rook {
|
||||
use super::*;
|
||||
use crate::test_board;
|
||||
|
||||
sight_test!(
|
||||
g3_rook,
|
||||
piece!(White Rook),
|
||||
Square::G3,
|
||||
bitboard![G1 G2 G4 G5 G6 G7 G8 A3 B3 C3 D3 E3 F3 H3]
|
||||
);
|
||||
|
||||
sight_test!(
|
||||
e4_rook_with_e1_white_king_e7_black_king,
|
||||
test_board![
|
||||
White Rook on E4,
|
||||
White King on E2,
|
||||
Black King on E7,
|
||||
],
|
||||
piece!(White Rook),
|
||||
Square::E4,
|
||||
bitboard![A4 B4 C4 D4 F4 G4 H4 E2 E3 E5 E6 E7]
|
||||
);
|
||||
|
||||
// #[test]
|
||||
// fn ray_to_square() {
|
||||
// let generated_ray = Shape::Rook.ray_to_square(Square::C2, Square::C6);
|
||||
// let expected_ray = bitboard![C3 C4 C5 C6];
|
||||
// assert_eq!(generated_ray, Some(expected_ray));
|
||||
|
||||
// let generated_ray = Shape::Rook.ray_to_square(Square::D2, Square::H2);
|
||||
// let expected_ray = bitboard![E2 F2 G2 H2];
|
||||
// assert_eq!(generated_ray, Some(expected_ray));
|
||||
|
||||
// let generated_ray = Shape::Rook.ray_to_square(Square::G6, Square::B6);
|
||||
// let expected_ray = bitboard![B6 C6 D6 E6 F6];
|
||||
// assert_eq!(generated_ray, Some(expected_ray));
|
||||
|
||||
// let generated_ray = Shape::Rook.ray_to_square(Square::A6, Square::A3);
|
||||
// let expected_ray = bitboard![A5 A4 A3];
|
||||
// assert_eq!(generated_ray, Some(expected_ray));
|
||||
// }
|
||||
}
|
||||
|
||||
mod king {
|
||||
use chessfriend_bitboard::bitboard;
|
||||
use chessfriend_core::piece;
|
||||
|
||||
sight_test!(
|
||||
e1_king,
|
||||
piece!(White King),
|
||||
Square::E1,
|
||||
bitboard![D1 D2 E2 F2 F1]
|
||||
);
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue