chess-lib 0.1.3

A chess movement generator library.
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

use crate::engine::bitboard::{occupied_squares, Bitboard, A_FILE, B_FILE, G_FILE, H_FILE};
use crate::engine::common::{Move, Square};
use crate::engine::magics::{magic_index_with_offset, BishopTable, RookTable, Tables};
use crate::engine::magics_entries::{BISHOP_MAGICS, ROOK_MAGICS};

pub fn to_moves(bitboard: Bitboard, src: Square) -> Vec<Move> {
    let mut result = vec![];
    for dst in occupied_squares(bitboard) {
        result.push(Move::new(src, dst, None));
    }
    result
}

pub fn lookup_king(king: Bitboard) -> Bitboard {
    let pseudo_moves: Bitboard = (king & !H_FILE) << 9
        | king << 8
        | (king & !A_FILE) << 7
        | (king & !H_FILE) << 1
        | (king & !A_FILE) >> 1
        | (king & !H_FILE) >> 7
        | king >> 8
        | (king & !A_FILE) >> 9;

    pseudo_moves
}

pub fn lookup_knight(knights: Bitboard) -> Bitboard {
    (knights & !H_FILE) << 17
        | (knights & !A_FILE) << 15
        | (knights & !(G_FILE | H_FILE)) << 10
        | (knights & !(A_FILE | B_FILE)) << 6
        | (knights & !(G_FILE | H_FILE)) >> 6
        | (knights & !(A_FILE | B_FILE)) >> 10
        | (knights & !H_FILE) >> 15
        | (knights & !(A_FILE)) >> 17
}

pub fn lookup_queen(
    tables: &Tables,
    queen: Square,
    same_color: Bitboard,
    occupied: Bitboard,
) -> Bitboard {
    lookup_bishop(&tables.bishop, queen, same_color, occupied)
        | lookup_rook(&tables.rook, queen, same_color, occupied)
}

pub fn lookup_bishop(
    bishop_table: &BishopTable,
    bishop: Square,
    same_color: Bitboard,
    occupied: Bitboard,
) -> Bitboard {
    let magic = BISHOP_MAGICS[bishop.index() as usize];
    *bishop_table
        .get(magic_index_with_offset(&magic, occupied))
        .unwrap()
        & !same_color
}

pub fn lookup_rook(
    rook_table: &RookTable,
    rook: Square,
    same_color: Bitboard,
    occupied: Bitboard,
) -> Bitboard {
    let magic = ROOK_MAGICS[rook.index() as usize];
    *rook_table
        .get(magic_index_with_offset(&magic, occupied))
        .unwrap()
        & !same_color
}