friendly-chess 0.6.0

friendly neighborhood chess engine
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

use super::{utils::ChessResult, Board, Color, Kings, PieceType, SquareCoordinate};

#[derive(Clone, Debug)]
pub struct Castling {
    pub kingside: bool,
    pub queenside: bool,
}

#[derive(Clone, Debug)]
pub struct CastlingRights {
    pub white: Castling,
    pub black: Castling,
}

impl CastlingRights {
    pub fn new() -> Self {
        Self {
            white: Castling {
                kingside: true,
                queenside: true,
            },
            black: Castling {
                kingside: true,
                queenside: true,
            },
        }
    }

    pub fn update(&mut self, kings: &Kings, board: &Board) -> ChessResult<()> {
        if kings.white != Some(SquareCoordinate::E1) {
            self.white.kingside = false;
            self.white.queenside = false;
        }

        if kings.black != Some(SquareCoordinate::E8) {
            self.black.kingside = false;
            self.black.queenside = false;
        }

        let h1 = board.get(SquareCoordinate::H1)?;
        let a1 = board.get(SquareCoordinate::A1)?;
        let a8 = board.get(SquareCoordinate::A8)?;
        let h8 = board.get(SquareCoordinate::H8)?;

        // check for a kingside white rook, if it is not the a white rook at H1, then white can't castle kingside
        if let Some(piece) = h1 {
            if piece.piece_type != PieceType::ROOK && piece.color != Color::WHITE {
                self.white.kingside = false;
            }
        } else {
            // if no piece is present, we can't castle kingside, and so on...
            self.white.kingside = false;
        }

        // queenside white rook
        if let Some(piece) = a1 {
            if piece.piece_type != PieceType::ROOK && piece.color != Color::WHITE {
                self.white.queenside = false;
            }
        } else {
            self.white.queenside = false;
        }

        // kingside black rook
        if let Some(piece) = h8 {
            if piece.piece_type != PieceType::ROOK && piece.color != Color::BLACK {
                self.black.kingside = false;
            }
        } else {
            self.black.kingside = false;
        }

        // queenside black rook
        if let Some(piece) = a8 {
            if piece.piece_type != PieceType::ROOK && piece.color != Color::BLACK {
                self.black.queenside = false;
            }
        } else {
            self.black.queenside = false;
        }

        Ok(())
    }
}