rchess 2.3.0

A Chess Library Written In Rust
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

use crate::defs::*;

#[cfg(not(feature = "magic-table"))]
use crate::mask_gen::sliders::{get_bishop_attacks_slow, get_rook_attacks_slow};

include!(concat!(env!("OUT_DIR"), "/generated_tables.rs"));

/// Returns a [`BitBoard`] mask containing the squares under attack by a pawn from a given square
pub fn get_pawn_attacks(square: Square, color: Color) -> BitBoard {
    PAWN_ATTACKS[color.index()][square.index()]
}

/// Returns a [`BitBoard`] mask containing all the squares a knight targets from a given square
pub fn get_knight_attacks(square: Square) -> BitBoard {
    KNIGHT_ATTACKS[square.index()]
}

/// Returns a [`BitBoard`] mask containing all the squares a king targets from a given square
pub fn get_king_attacks(square: Square) -> BitBoard {
    KING_ATTACKS[square.index()]
}

/// Returns a [`BitBoard`] with the connecting line between two squares.
pub fn get_direct_connection(start: Square, end: Square) -> BitBoard {
    DIRECT_CONNECTIONS[start.index()][end.index()]
}

/// Returns a [`BitBoard`] with axis line intersecting two squares.
pub fn get_connection_axis(start: Square, end: Square) -> BitBoard {
    AXIS_CONNECTIONS[start.index()][end.index()]
}

/// Gets a [`BitBoard`] of the squares a bishop attacks with a given square and occupancy.
pub fn get_bishop_attacks(square: Square, occupancy: BitBoard) -> BitBoard {
    #[cfg(feature = "magic-table")]
    {
        let key = BISHOP_MAGICS[square.index()].key(occupancy);
        BISHOP_ATTACKS[key]
    }

    #[cfg(not(feature = "magic-table"))]
    {
        get_bishop_attacks_slow(&RAYS, square, occupancy)
    }
}

/// Gets a [`BitBoard`] of the squares a rook attacks with a given square and occupancy.
pub fn get_rook_attacks(square: Square, occupancy: BitBoard) -> BitBoard {
    #[cfg(feature = "magic-table")]
    {
        let key = ROOK_MAGICS[square.index()].key(occupancy);
        ROOK_ATTACKS[key]
    }

    #[cfg(not(feature = "magic-table"))]
    {
        get_rook_attacks_slow(&RAYS, square, occupancy)
    }
}

/// Returns the seen squares for a bishop ignoring the first friendly blocker
pub fn get_ghost_bishop(square: Square, occupancy: BitBoard, mut friendly: BitBoard) -> BitBoard {
    let bishop_seen = get_bishop_attacks(square, occupancy);

    // Get seen friends
    friendly &= bishop_seen;

    // Return all bishop seen squares
    bishop_seen ^ get_bishop_attacks(square, occupancy ^ friendly)
}

/// Returns the seen squares for a rook ignoring the first friendly blocker
pub fn get_ghost_rook(square: Square, occupancy: BitBoard, mut friendly: BitBoard) -> BitBoard {
    let rook_seen = get_rook_attacks(square, occupancy);

    // Get seen friends
    friendly &= rook_seen;

    // Return all bishop seen squares
    rook_seen ^ get_rook_attacks(square, occupancy ^ friendly)
}