sisyphus32 1.1.0

Feature-based UCI 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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

use crate::{FILE_COUNT, SQUARE_COUNT, SquareParseError, Bitboard, File, Rank};
use core::fmt;
use std::{ops::{Index, IndexMut}, mem::transmute};

#[repr(u8)]
#[derive(Clone, Copy, PartialEq, Eq, Debug, Hash)]
pub enum Square {
    A8, B8, C8, D8, E8, F8, G8, H8,
    A7, B7, C7, D7, E7, F7, G7, H7,
    A6, B6, C6, D6, E6, F6, G6, H6,
    A5, B5, C5, D5, E5, F5, G5, H5,
    A4, B4, C4, D4, E4, F4, G4, H4,
    A3, B3, C3, D3, E3, F3, G3, H3,
    A2, B2, C2, D2, E2, F2, G2, H2,
    A1, B1, C1, D1, E1, F1, G1, H1,
}

impl Square {
    pub const ALL_SQUARES: [Square; SQUARE_COUNT] = [
        Square::A8, Square::B8, Square::C8, Square::D8, Square::E8, Square::F8, Square::G8, Square::H8,
        Square::A7, Square::B7, Square::C7, Square::D7, Square::E7, Square::F7, Square::G7, Square::H7,
        Square::A6, Square::B6, Square::C6, Square::D6, Square::E6, Square::F6, Square::G6, Square::H6,
        Square::A5, Square::B5, Square::C5, Square::D5, Square::E5, Square::F5, Square::G5, Square::H5,
        Square::A4, Square::B4, Square::C4, Square::D4, Square::E4, Square::F4, Square::G4, Square::H4,
        Square::A3, Square::B3, Square::C3, Square::D3, Square::E3, Square::F3, Square::G3, Square::H3,
        Square::A2, Square::B2, Square::C2, Square::D2, Square::E2, Square::F2, Square::G2, Square::H2,
        Square::A1, Square::B1, Square::C1, Square::D1, Square::E1, Square::F1, Square::G1, Square::H1
    ];

    #[inline(always)]
    pub(crate) fn to_bb(self) -> Bitboard {
        Bitboard::from(1 << (self as u64))
    }

    #[inline(always)]
    pub fn rank(self) -> Rank {
        Rank::from(self.rank_as_u8())
    }

    #[inline(always)]
    pub fn file(self) -> File {
        File::from(self.file_as_u8())
    }

    #[inline(always)]
    pub(crate) fn rank_as_u8(self) -> u8 {
        self as u8 >> 3 & 0b0000_0111
    }

    #[inline(always)]
    pub(crate) fn file_as_u8(self) -> u8 {
        self as u8 & 0b0000_0111
    }

    #[inline(always)]
    pub(crate) fn above(self) -> Square {
        Square::from(self as u8 - FILE_COUNT as u8)
    }

    #[inline(always)]
    pub(crate) fn below(self) -> Square {
        Square::from(self as u8 + FILE_COUNT as u8)
    }

    #[inline(always)]
    pub(crate) fn left(self) -> Square {
        Square::from(self as u8 - 1)
    }

    #[inline(always)]
    pub(crate) fn right(self) -> Square {
        Square::from(self as u8 + 1)
    }
}

impl<T, const N: usize> Index<Square> for [T; N] {
    type Output = T;

    fn index(&self, index: Square) -> &Self::Output {
        &self[index as usize]
    }
}

impl<T, const N: usize> IndexMut<Square> for [T; N] {
    fn index_mut(&mut self, index: Square) -> &mut Self::Output {
        &mut self[index as usize]
    }
}

impl From<Bitboard> for Square {
    #[inline(always)]
    fn from(bitboard: Bitboard) -> Self {
        debug_assert_eq!(bitboard.count_bits(), 1);
        bitboard.get_lsb()
    }
}

impl From<u8> for Square {
    #[inline(always)]
    fn from(number: u8) -> Self {
        unsafe { transmute::<u8, Self>(number) }
    }
}

impl TryFrom<&str> for Square {
    type Error = SquareParseError;

    fn try_from(sq_str: &str) -> Result<Self, Self::Error> {
        if sq_str.len() != 2 {
            return Err(SquareParseError::StringLength(sq_str.to_string()));
        }

        let mut chars_iter = sq_str.chars();
        let file_char = chars_iter.next().ok_or(SquareParseError::NoFile)?;
        let rank_char = chars_iter.next().ok_or(SquareParseError::NoRank)?;

        let rank = Rank::try_from(rank_char)?;
        let file = File::try_from(file_char)?;

        Ok(Self::from(rank as u8 * FILE_COUNT as u8 + file as u8))
    }
}

impl fmt::Display for Square {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        f.pad(&format!("{}{}", self.file(), self.rank()))
    }
}