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

use super::constants::*;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct Piece {
    pub piece_type: PieceType,
    pub color: Color,
}

#[derive(PartialEq, Eq, Ord, PartialOrd, Copy, Clone, Debug, Hash)]
#[repr(u8)]
pub enum PieceType {
    // EMPTY = 0,
    PAWN = 1,
    KNIGHT = 2,
    BISHOP = 4,
    ROOK = 8,
    QUEEN = 16,
    KING = 32,
}

impl PieceType {
    /// Convert a `PieceType` enum to its associated value (u8)
    pub fn to_value(&self) -> u8 {
        *self as u8
    }

    /// Convert a value to its corresponding PieceType type
    pub fn from_value(value: u8) -> Option<PieceType> {
        match value {
            // 0 => Some(PieceType::EMPTY),
            1 => Some(PieceType::PAWN),
            2 => Some(PieceType::KNIGHT),
            4 => Some(PieceType::BISHOP),
            8 => Some(PieceType::ROOK),
            16 => Some(PieceType::QUEEN),
            32 => Some(PieceType::KING),
            _ => None,
        }
    }

    pub fn from_string(value: &str) -> Option<PieceType> {
        match value.to_lowercase().as_str() {
            "p" => Some(PieceType::PAWN),
            "n" => Some(PieceType::KNIGHT),
            "b" => Some(PieceType::BISHOP),
            "r" => Some(PieceType::ROOK),
            "q" => Some(PieceType::QUEEN),
            "k" => Some(PieceType::KING),
            _ => None,
        }
    }
}