|  | @@ -1,950 +0,0 @@
 | 
	
		
			
				|  |  | -use bitboard::*;
 | 
	
		
			
				|  |  | -use movegen::*;
 | 
	
		
			
				|  |  | -use log::info;
 | 
	
		
			
				|  |  | -use std::sync::Arc;
 | 
	
		
			
				|  |  | -use zobrist::{ZobristTable};
 | 
	
		
			
				|  |  | -use zobrist;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -use crate::movegen;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -#[derive(Clone)]
 | 
	
		
			
				|  |  | -pub struct Game
 | 
	
		
			
				|  |  | -{
 | 
	
		
			
				|  |  | -    pub pieces: [Bitboard; 12],
 | 
	
		
			
				|  |  | -    pub turn: Side,
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    /// None if no en passant move is possible,
 | 
	
		
			
				|  |  | -    /// otherwise contains the file of the last (double-) pushed pawn
 | 
	
		
			
				|  |  | -    pub en_passant: Option<u8>,
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    ///
 | 
	
		
			
				|  |  | -    /// castling rights in the following order
 | 
	
		
			
				|  |  | -    /// white kingside, white queenside, black kingside, black queenside
 | 
	
		
			
				|  |  | -    /// 
 | 
	
		
			
				|  |  | -    pub castling_rights: [bool; 4],
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    //
 | 
	
		
			
				|  |  | -    // non-hashed part
 | 
	
		
			
				|  |  | -    // vvv
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub turn_number: i32,
 | 
	
		
			
				|  |  | -    pub halfmoves_since_last_event: i8,
 | 
	
		
			
				|  |  | -    pub zobrist: Option<(Arc<ZobristTable>, zobrist::Hash)>
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -impl Default for Game {
 | 
	
		
			
				|  |  | -    fn default () -> Game {
 | 
	
		
			
				|  |  | -        Game {
 | 
	
		
			
				|  |  | -            pieces: [
 | 
	
		
			
				|  |  | -                ROW_2,
 | 
	
		
			
				|  |  | -                0x42,
 | 
	
		
			
				|  |  | -                0x24,
 | 
	
		
			
				|  |  | -                0x81,
 | 
	
		
			
				|  |  | -                0x10,
 | 
	
		
			
				|  |  | -                0x08,
 | 
	
		
			
				|  |  | -                ROW_7,
 | 
	
		
			
				|  |  | -                0x42 << 56,
 | 
	
		
			
				|  |  | -                0x24 << 56,
 | 
	
		
			
				|  |  | -                0x81 << 56,
 | 
	
		
			
				|  |  | -                0x10 << 56,
 | 
	
		
			
				|  |  | -                0x08 << 56,
 | 
	
		
			
				|  |  | -            ],
 | 
	
		
			
				|  |  | -            turn: WHITE,
 | 
	
		
			
				|  |  | -            en_passant: None,
 | 
	
		
			
				|  |  | -            castling_rights: [true; 4],
 | 
	
		
			
				|  |  | -            halfmoves_since_last_event: 0,
 | 
	
		
			
				|  |  | -            turn_number: 0,
 | 
	
		
			
				|  |  | -            zobrist: None
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -impl std::hash::Hash for Game {
 | 
	
		
			
				|  |  | -    fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
 | 
	
		
			
				|  |  | -        if let Some((ref _zt, ref zb)) = &self.zobrist {
 | 
	
		
			
				|  |  | -            zb.hash(state);
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -impl PartialEq for Game {
 | 
	
		
			
				|  |  | -    fn eq(&self, other: &Self) -> bool {
 | 
	
		
			
				|  |  | -        self.pieces == other.pieces &&
 | 
	
		
			
				|  |  | -        self.turn == other.turn &&
 | 
	
		
			
				|  |  | -        self.en_passant == other.en_passant &&
 | 
	
		
			
				|  |  | -        self.castling_rights == other.castling_rights
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -impl Eq for Game {
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -impl Game {
 | 
	
		
			
				|  |  | -    pub fn empty() -> Game {
 | 
	
		
			
				|  |  | -        Game {
 | 
	
		
			
				|  |  | -            pieces: [0; 12],
 | 
	
		
			
				|  |  | -            turn: WHITE,
 | 
	
		
			
				|  |  | -            en_passant: None,
 | 
	
		
			
				|  |  | -            castling_rights: [true; 4],
 | 
	
		
			
				|  |  | -            halfmoves_since_last_event: 0,
 | 
	
		
			
				|  |  | -            turn_number: 0,
 | 
	
		
			
				|  |  | -            zobrist: None
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn from_fen(fen: &[&str]) -> Option<Game> {
 | 
	
		
			
				|  |  | -        let mut game: Game = Game::empty();
 | 
	
		
			
				|  |  | -        //let example = "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1";
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        let position = fen[0];
 | 
	
		
			
				|  |  | -        let turn = fen[1];
 | 
	
		
			
				|  |  | -        let castling_rights = fen[2];
 | 
	
		
			
				|  |  | -        let en_passant = fen[3];
 | 
	
		
			
				|  |  | -        let halfmoves_since_last_event = fen[4];
 | 
	
		
			
				|  |  | -        let turn_number = fen[5];
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        {
 | 
	
		
			
				|  |  | -            let rows = position.split('/');
 | 
	
		
			
				|  |  | -            let mut row_index: i32 = 7;
 | 
	
		
			
				|  |  | -            for row in rows {
 | 
	
		
			
				|  |  | -                if row_index < 0 {
 | 
	
		
			
				|  |  | -                    return None;
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -                let mut col_index = 0;
 | 
	
		
			
				|  |  | -                for c in row.chars() {
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    if col_index >= 8 {
 | 
	
		
			
				|  |  | -                        return None;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    let bit_to_set = from_indices(col_index, row_index as _);
 | 
	
		
			
				|  |  | -                    match c {
 | 
	
		
			
				|  |  | -                        'p' => { *game.pawns_mut(BLACK) |= bit_to_set },
 | 
	
		
			
				|  |  | -                        'n' => { *game.knights_mut(BLACK) |= bit_to_set },
 | 
	
		
			
				|  |  | -                        'b' => { *game.bishops_mut(BLACK) |= bit_to_set },
 | 
	
		
			
				|  |  | -                        'r' => { *game.rooks_mut(BLACK) |= bit_to_set },
 | 
	
		
			
				|  |  | -                        'q' => { *game.queens_mut(BLACK) |= bit_to_set },
 | 
	
		
			
				|  |  | -                        'k' => { *game.kings_mut(BLACK) |= bit_to_set },
 | 
	
		
			
				|  |  | -                        'P' => { *game.pawns_mut(WHITE) |= bit_to_set },
 | 
	
		
			
				|  |  | -                        'N' => { *game.knights_mut(WHITE) |= bit_to_set },
 | 
	
		
			
				|  |  | -                        'B' => { *game.bishops_mut(WHITE) |= bit_to_set },
 | 
	
		
			
				|  |  | -                        'R' => { *game.rooks_mut(WHITE) |= bit_to_set },
 | 
	
		
			
				|  |  | -                        'Q' => { *game.queens_mut(WHITE) |= bit_to_set },
 | 
	
		
			
				|  |  | -                        'K' => { *game.kings_mut(WHITE) |= bit_to_set },
 | 
	
		
			
				|  |  | -                        num => {
 | 
	
		
			
				|  |  | -                            col_index += match num.to_digit(10) {
 | 
	
		
			
				|  |  | -                                Some(x) => x,
 | 
	
		
			
				|  |  | -                                None => { return None; }
 | 
	
		
			
				|  |  | -                            } as u8 - 1;
 | 
	
		
			
				|  |  | -                        }
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    col_index += 1;
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -                row_index -= 1;
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        if turn == "b" {
 | 
	
		
			
				|  |  | -            game.turn = BLACK
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        // parse castling rights
 | 
	
		
			
				|  |  | -        game.castling_rights[0] = castling_rights.contains('K');
 | 
	
		
			
				|  |  | -        game.castling_rights[1] = castling_rights.contains('Q');
 | 
	
		
			
				|  |  | -        game.castling_rights[2] = castling_rights.contains('k');
 | 
	
		
			
				|  |  | -        game.castling_rights[3] = castling_rights.contains('q');
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        // parse en passant
 | 
	
		
			
				|  |  | -        game.en_passant = Move::parse_square(en_passant).map(|sq| indices_from_square(sq).0);
 | 
	
		
			
				|  |  | -        //info!("en passant on file {:?}", game.en_passant);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        game.halfmoves_since_last_event = halfmoves_since_last_event.parse::<i8>().unwrap_or(0);
 | 
	
		
			
				|  |  | -        game.turn_number = turn_number.parse::<i32>().unwrap_or(0);
 | 
	
		
			
				|  |  | -        
 | 
	
		
			
				|  |  | -        Some(game)
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn from_fen_str(fen: &str) -> Option<Game> {
 | 
	
		
			
				|  |  | -        let fen_parts = fen.split_whitespace();
 | 
	
		
			
				|  |  | -        Self::from_fen(fen_parts.collect::<Vec<&str>>().as_slice())
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn parse_move(&self, mov: &str) -> Result<Move, ()> {
 | 
	
		
			
				|  |  | -        if mov.len() < 4 {
 | 
	
		
			
				|  |  | -            Err(())
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -        else {
 | 
	
		
			
				|  |  | -            let origin = Move::parse_square(&mov[0..2]);
 | 
	
		
			
				|  |  | -            let target = Move::parse_square(&mov[2..4]);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -            let promote_to = if mov.len() == 5 {
 | 
	
		
			
				|  |  | -                Some(match mov.chars().nth(4) {
 | 
	
		
			
				|  |  | -                    Some('Q') | Some('q') => QUEEN,
 | 
	
		
			
				|  |  | -                    Some('N') | Some('n') => KNIGHT,
 | 
	
		
			
				|  |  | -                    Some('B') | Some('b') => BISHOP,
 | 
	
		
			
				|  |  | -                    Some('R') | Some('r') => ROOK,
 | 
	
		
			
				|  |  | -                    _ => panic!("invalid move: {}", mov)
 | 
	
		
			
				|  |  | -                })
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -            else {
 | 
	
		
			
				|  |  | -                None
 | 
	
		
			
				|  |  | -            };
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -            if let (Some(from), Some(to)) = (origin, target) {
 | 
	
		
			
				|  |  | -                let (piece_type, side) = self.get_square(from);
 | 
	
		
			
				|  |  | -                //println!("from: {}", from);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                let (cap, cs) = self.get_square(to);
 | 
	
		
			
				|  |  | -                let captured = if cap != NO_PIECE { Some(cap) } else { None };
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                if cap != NO_PIECE && cs == side {
 | 
	
		
			
				|  |  | -                    // cannot capture own piece
 | 
	
		
			
				|  |  | -                    return Err(());
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                if piece_type == NO_PIECE {
 | 
	
		
			
				|  |  | -                    return Err(());
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                if side != self.turn { // wrong side to move
 | 
	
		
			
				|  |  | -                    return Err(());
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                if let Some(promote_to) = promote_to {
 | 
	
		
			
				|  |  | -                    if piece_type != PAWN {
 | 
	
		
			
				|  |  | -                        return Err(());
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    Ok(Move::Promotion{ mov: SimpleMove{ from, to }, pc: PieceCaptureType::new(promote_to, captured) })
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -                else {
 | 
	
		
			
				|  |  | -                    if piece_type == KING && (from as i32 - to as i32).abs() == 2 {
 | 
	
		
			
				|  |  | -                        let left = from < to;
 | 
	
		
			
				|  |  | -                        //println!("OMG castling!");
 | 
	
		
			
				|  |  | -                        return Ok(Move::Castling{ mov: SimpleMove { from, to }, side, left });
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    // pawn capture
 | 
	
		
			
				|  |  | -                    let (from_file, from_row) = indices_from_square(from);
 | 
	
		
			
				|  |  | -                    let (target_file, _target_row) = indices_from_square(to);
 | 
	
		
			
				|  |  | -                    if piece_type == PAWN && from_file != target_file {
 | 
	
		
			
				|  |  | -                        let others = self.get_all_side(!side);
 | 
	
		
			
				|  |  | -                        if others & from_square(to) == 0 {
 | 
	
		
			
				|  |  | -                            let beaten = square_from_indices(target_file, from_row);
 | 
	
		
			
				|  |  | -                            return Ok(Move::EnPassant{ mov: SimpleMove{ from, to }, beaten });
 | 
	
		
			
				|  |  | -                        }
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    //println!("pt: {}", piece_type);
 | 
	
		
			
				|  |  | -                    Ok(Move::Default{ mov: SimpleMove{ from, to }, pc: PieceCaptureType::new(piece_type, captured) })
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -            else {
 | 
	
		
			
				|  |  | -                Err(())
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn bitboard(&self, pt: PieceType, side: Side) -> Bitboard {
 | 
	
		
			
				|  |  | -        if side == BLACK {
 | 
	
		
			
				|  |  | -            self.pieces[(pt + 6) as usize]
 | 
	
		
			
				|  |  | -        } else {
 | 
	
		
			
				|  |  | -            self.pieces[pt as usize]
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    /**
 | 
	
		
			
				|  |  | -     * finds the first bitboard that has overlapping bits with the given bitboard
 | 
	
		
			
				|  |  | -     * and returns the piece type of that bitboard
 | 
	
		
			
				|  |  | -     */
 | 
	
		
			
				|  |  | -    pub fn find_piece(&self, bitboard: Bitboard) -> Option<PieceType> {
 | 
	
		
			
				|  |  | -        for i in 0..12 {
 | 
	
		
			
				|  |  | -            if self.pieces[i] & bitboard != 0 {
 | 
	
		
			
				|  |  | -                return Some(if i >= 6 { i - 6 } else { i } as PieceType);
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -        return None;
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    /**
 | 
	
		
			
				|  |  | -     * \return bitboard containig all occupied squares
 | 
	
		
			
				|  |  | -     */
 | 
	
		
			
				|  |  | -    pub fn occupied(&self) -> Bitboard {
 | 
	
		
			
				|  |  | -        self.pieces.iter().fold(0, |a, b| { a | b } )
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn pawns(&self, side: Side) -> Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => self.pieces[0],
 | 
	
		
			
				|  |  | -            BLACK => self.pieces[6],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn knights(&self, side: Side) -> Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => self.pieces[1],
 | 
	
		
			
				|  |  | -            BLACK => self.pieces[7],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn bishops(&self, side: Side) -> Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => self.pieces[2],
 | 
	
		
			
				|  |  | -            BLACK => self.pieces[8],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn rooks(&self, side: Side) -> Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => self.pieces[3],
 | 
	
		
			
				|  |  | -            BLACK => self.pieces[9],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn queens(&self, side: Side) -> Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => self.pieces[4],
 | 
	
		
			
				|  |  | -            BLACK => self.pieces[10],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn kings(&self, side: Side) -> Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => self.pieces[5],
 | 
	
		
			
				|  |  | -            BLACK => self.pieces[11],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn pawns_mut(&mut self, side: Side) -> &mut Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => &mut self.pieces[0],
 | 
	
		
			
				|  |  | -            BLACK => &mut self.pieces[6],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn knights_mut(&mut self, side: Side) -> &mut Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => &mut self.pieces[1],
 | 
	
		
			
				|  |  | -            BLACK => &mut self.pieces[7],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn bishops_mut(&mut self, side: Side) -> &mut Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => &mut self.pieces[2],
 | 
	
		
			
				|  |  | -            BLACK => &mut self.pieces[8],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn rooks_mut(&mut self, side: Side) -> &mut Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => &mut self.pieces[3],
 | 
	
		
			
				|  |  | -            BLACK => &mut self.pieces[9],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn queens_mut(&mut self, side: Side) -> &mut Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => &mut self.pieces[4],
 | 
	
		
			
				|  |  | -            BLACK => &mut self.pieces[10],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn kings_mut(&mut self, side: Side) -> &mut Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => &mut self.pieces[5],
 | 
	
		
			
				|  |  | -            BLACK => &mut self.pieces[11],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn get_piece(&self, piece_type: PieceType, side: Side) -> Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => self.pieces[piece_type as usize],
 | 
	
		
			
				|  |  | -            BLACK => self.pieces[piece_type as usize + 6],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn set_piece(&mut self, piece_type: PieceType, side: Side, bitboard: Bitboard) {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => self.pieces[piece_type as usize] = bitboard,
 | 
	
		
			
				|  |  | -            BLACK => self.pieces[piece_type as usize + 6] = bitboard,
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn get_piece_mut(&mut self, piece_type: PieceType, side: Side) -> &mut Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => &mut self.pieces[piece_type as usize],
 | 
	
		
			
				|  |  | -            BLACK => &mut self.pieces[piece_type as usize + 6],
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn get_all_side(&self, side: Side) -> Bitboard {
 | 
	
		
			
				|  |  | -        match side {
 | 
	
		
			
				|  |  | -            WHITE => self.pieces[0] | self.pieces[1] | self.pieces[2] | self.pieces[3] |
 | 
	
		
			
				|  |  | -                self.pieces[4] | self.pieces[5],
 | 
	
		
			
				|  |  | -            BLACK => self.pieces[6] | self.pieces[7] | self.pieces[8] | self.pieces[9] |
 | 
	
		
			
				|  |  | -                self.pieces[10] | self.pieces[11]
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn get_square(&self, square: Square) -> (PieceType, Side) {
 | 
	
		
			
				|  |  | -        let square_mask = 1 << square;
 | 
	
		
			
				|  |  | -        for i in 0..6 {
 | 
	
		
			
				|  |  | -            if self.pieces[i] & square_mask != 0 {
 | 
	
		
			
				|  |  | -                return (i as PieceType, WHITE);
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -        for i in 0..6 {
 | 
	
		
			
				|  |  | -            if self.pieces[i + 6] & square_mask != 0 {
 | 
	
		
			
				|  |  | -                return (i as PieceType, BLACK);
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -        return (NO_PIECE, WHITE);
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    /**
 | 
	
		
			
				|  |  | -     * @brief masks all bitboards.
 | 
	
		
			
				|  |  | -     */
 | 
	
		
			
				|  |  | -    pub fn apply_mask(&mut self, mask: Bitboard) {
 | 
	
		
			
				|  |  | -        for board in &mut self.pieces {
 | 
	
		
			
				|  |  | -            *board &= mask;
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    /**
 | 
	
		
			
				|  |  | -     * @brief creates a unicode string containing the board
 | 
	
		
			
				|  |  | -     *
 | 
	
		
			
				|  |  | -     * Example:
 | 
	
		
			
				|  |  | -     *
 | 
	
		
			
				|  |  | -     *   ┌───┬───┬───┬───┬───┬───┬───┬───┐
 | 
	
		
			
				|  |  | -     * 8 │ ♜ │ ♞ │ ♝ │ ♛ │ ♚ │ ♝ │ ♞ │ ♜ │
 | 
	
		
			
				|  |  | -     *   ├───┼───┼───┼───┼───┼───┼───┼───┤
 | 
	
		
			
				|  |  | -     * 7 │ ♟ │ ♟ │ ♟ │ ♟ │ ♟ │ ♟ │ ♟ │ ♟ │
 | 
	
		
			
				|  |  | -     *   ├───┼───┼───┼───┼───┼───┼───┼───┤
 | 
	
		
			
				|  |  | -     * 6 │   │   │   │   │   │   │   │   │
 | 
	
		
			
				|  |  | -     *   ├───┼───┼───┼───┼───┼───┼───┼───┤
 | 
	
		
			
				|  |  | -     * 5 │   │   │   │   │   │   │   │   │
 | 
	
		
			
				|  |  | -     *   ├───┼───┼───┼───┼───┼───┼───┼───┤
 | 
	
		
			
				|  |  | -     * 4 │   │   │   │   │   │   │   │   │
 | 
	
		
			
				|  |  | -     *   ├───┼───┼───┼───┼───┼───┼───┼───┤
 | 
	
		
			
				|  |  | -     * 3 │   │   │   │   │   │   │   │   │
 | 
	
		
			
				|  |  | -     *   ├───┼───┼───┼───┼───┼───┼───┼───┤
 | 
	
		
			
				|  |  | -     * 2 │ ♙ │ ♙ │ ♙ │ ♙ │ ♙ │ ♙ │ ♙ │ ♙ │
 | 
	
		
			
				|  |  | -     *   ├───┼───┼───┼───┼───┼───┼───┼───┤
 | 
	
		
			
				|  |  | -     * 1 │ ♖ │ ♘ │ ♗ │ ♕ │ ♔ │ ♗ │ ♘ │ ♖ │
 | 
	
		
			
				|  |  | -     *   └───┴───┴───┴───┴───┴───┴───┴───┘
 | 
	
		
			
				|  |  | -     *     a   b   c   d   e   f   g   h
 | 
	
		
			
				|  |  | -     *
 | 
	
		
			
				|  |  | -     */
 | 
	
		
			
				|  |  | -    pub fn beautiful_print(&self) -> String {
 | 
	
		
			
				|  |  | -        let chars = ['─', '│', '┌', '┐', '└', '┘', '├', '┤', '┬', '┴', '┼'];
 | 
	
		
			
				|  |  | -        let figures = ['♙', '♘', '♗', '♖', '♕', '♔', '♟', '♞', '♝', '♜', '♛', '♚'];
 | 
	
		
			
				|  |  | -        let mut board: String = String::new();
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        #[derive(Copy, Clone)]
 | 
	
		
			
				|  |  | -        struct GridLine {
 | 
	
		
			
				|  |  | -            left: char,
 | 
	
		
			
				|  |  | -            center: char,
 | 
	
		
			
				|  |  | -            right: char,
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        let top = GridLine { left: chars[2], center: chars[8], right: chars[3] };
 | 
	
		
			
				|  |  | -        let center = GridLine { left: chars[6], center: chars[10], right: chars[7] };
 | 
	
		
			
				|  |  | -        let bot = GridLine { left: chars[4], center: chars[9], right: chars[5] };
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        let generate_line = |gl: GridLine| {
 | 
	
		
			
				|  |  | -            let mut line = String::new();
 | 
	
		
			
				|  |  | -            for i in 0..33 {
 | 
	
		
			
				|  |  | -                match i {
 | 
	
		
			
				|  |  | -                    0 => { line.push(gl.left) }
 | 
	
		
			
				|  |  | -                    32 => { line.push(gl.right) }
 | 
	
		
			
				|  |  | -                    x => { line.push(if x % 4 == 0 { gl.center } else { chars[0] }) }
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -            return line;
 | 
	
		
			
				|  |  | -        };
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        let piece_line = |row: u8| {
 | 
	
		
			
				|  |  | -            let mut line = String::new();
 | 
	
		
			
				|  |  | -            line.push(chars[1]);
 | 
	
		
			
				|  |  | -            line.push(' ');
 | 
	
		
			
				|  |  | -            for i in 0..8 {
 | 
	
		
			
				|  |  | -                let (pt, side) = self.get_square(square_from_indices(i, 7 - row));
 | 
	
		
			
				|  |  | -                if pt == NO_PIECE {
 | 
	
		
			
				|  |  | -                    line.push(' ');
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -                else {
 | 
	
		
			
				|  |  | -                    let fig_index = pt as usize + if side == BLACK { 6 } else { 0 };
 | 
	
		
			
				|  |  | -                    line.push(figures[fig_index]);
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -                line.push(' '); line.push(chars[1]); line.push(' ');
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -            return line;
 | 
	
		
			
				|  |  | -        };
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        board.push_str("  ");
 | 
	
		
			
				|  |  | -        board.push_str(&generate_line(top));
 | 
	
		
			
				|  |  | -        board.push('\n');
 | 
	
		
			
				|  |  | -        for i in 0..8 {
 | 
	
		
			
				|  |  | -            board.push_str(&(8 - i).to_string());
 | 
	
		
			
				|  |  | -            board.push(' ');
 | 
	
		
			
				|  |  | -            board.push_str(&piece_line(i));
 | 
	
		
			
				|  |  | -            board.push('\n');
 | 
	
		
			
				|  |  | -            if i != 7 {
 | 
	
		
			
				|  |  | -                board.push_str("  ");
 | 
	
		
			
				|  |  | -                board.push_str(&generate_line(center));
 | 
	
		
			
				|  |  | -                board.push('\n');
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -        board.push_str("  ");
 | 
	
		
			
				|  |  | -        board.push_str(&generate_line(bot));
 | 
	
		
			
				|  |  | -        board.push_str("\n    a   b   c   d   e   f   g   h");
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        let chars = ['K', 'Q', 'k', 'q'];
 | 
	
		
			
				|  |  | -        let mut cr: [char; 4] = [' '; 4];
 | 
	
		
			
				|  |  | -        for i in 0..4 {
 | 
	
		
			
				|  |  | -            cr[i] = if self.castling_rights[i] {
 | 
	
		
			
				|  |  | -                chars[i]
 | 
	
		
			
				|  |  | -            } else { ' ' };
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        if self.turn == WHITE {
 | 
	
		
			
				|  |  | -            board.push_str("\nWhite to move\n");
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -        else {
 | 
	
		
			
				|  |  | -            board.push_str("\nBlack to move\n");
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        board.push_str(&format!("\nCastling      : {}{}{}{}", cr[0], cr[1], cr[2], cr[3]));
 | 
	
		
			
				|  |  | -        board.push_str(&format!("\nEn Passant    : {}", self.en_passant.map(|f| f.to_string()).unwrap_or("-".to_owned())));
 | 
	
		
			
				|  |  | -        board.push_str(&format!("\nTurn number   : {}", self.turn_number));
 | 
	
		
			
				|  |  | -        board.push_str(&format!("\nHalfmoves slt : {}", self.halfmoves_since_last_event));
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        return board;
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn update_zobrist(&mut self, h: zobrist::Hash) {
 | 
	
		
			
				|  |  | -        if let Some((ref _zt, ref mut zh)) = self.zobrist {
 | 
	
		
			
				|  |  | -            *zh ^= h;
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn calculate_zobrist(&self, zt: &ZobristTable) -> zobrist::Hash {
 | 
	
		
			
				|  |  | -        let mut hash: zobrist::Hash = 0;
 | 
	
		
			
				|  |  | -        for pt in 0..6 {
 | 
	
		
			
				|  |  | -            let bb_it_w = BitboardIterator(self.get_piece(pt, WHITE));
 | 
	
		
			
				|  |  | -            let bb_it_b = BitboardIterator(self.get_piece(pt, BLACK));
 | 
	
		
			
				|  |  | -            for bb_square in bb_it_w {
 | 
	
		
			
				|  |  | -                hash ^= zt.piece_hash(pt as _, WHITE, square(bb_square));
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -            for bb_square in bb_it_b {
 | 
	
		
			
				|  |  | -                hash ^= zt.piece_hash(pt as _, BLACK, square(bb_square));
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        for cr_id in 0..4 {
 | 
	
		
			
				|  |  | -            if self.castling_rights[cr_id] {
 | 
	
		
			
				|  |  | -                hash ^= zt.castling_rights_hash(cr_id as _);
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        if let Some(file) = self.en_passant {
 | 
	
		
			
				|  |  | -            hash ^= zt.en_passant_hash(file);
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        if self.turn {
 | 
	
		
			
				|  |  | -            hash ^= zt.turn_hash();
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        return hash;
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn is_zobrist_correct(&self) -> bool {
 | 
	
		
			
				|  |  | -        if let Some((ref zt, zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -            self.calculate_zobrist(zt) == zv
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -        else {
 | 
	
		
			
				|  |  | -            false
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    ///
 | 
	
		
			
				|  |  | -    /// TODO: improve
 | 
	
		
			
				|  |  | -    /// 
 | 
	
		
			
				|  |  | -    pub fn is_legal(&self, mov: Move) -> bool {
 | 
	
		
			
				|  |  | -        /*match mov {
 | 
	
		
			
				|  |  | -            Move::Default { mov, pc } => {
 | 
	
		
			
				|  |  | -                let piece_type = pc.piece_type();
 | 
	
		
			
				|  |  | -                let capture = pc.capture_type();
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                let legal_from = self.get_piece(piece_type, self.turn);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                match piece_type {
 | 
	
		
			
				|  |  | -                    PAWN => {
 | 
	
		
			
				|  |  | -                        if pawn
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }*/
 | 
	
		
			
				|  |  | -        movegen::generate_legal_moves(&mut self.clone(), self.turn, false).contains(&mov)
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn apply(&mut self, mov: Move) -> MoveUndo {
 | 
	
		
			
				|  |  | -        /*if !self.is_zobrist_correct() {
 | 
	
		
			
				|  |  | -            println!("{}", self.beautiful_print());
 | 
	
		
			
				|  |  | -            println!("incorrect zobrist before apply {} {:?}", mov.to_string(), mov);
 | 
	
		
			
				|  |  | -            info!("incorrect zobrist before apply");
 | 
	
		
			
				|  |  | -            panic!("zobrist");
 | 
	
		
			
				|  |  | -            let val = if let Some((ref zt, _zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                self.calculate_zobrist(zt)
 | 
	
		
			
				|  |  | -            } else { 0 };
 | 
	
		
			
				|  |  | -            if let Some((ref _zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                *zv = val;
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }*/
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        // save irrecoverable values
 | 
	
		
			
				|  |  | -        let castling_rights_before = self.castling_rights.clone();
 | 
	
		
			
				|  |  | -        let halfmoves_since_last_event_before = self.halfmoves_since_last_event.clone();
 | 
	
		
			
				|  |  | -        let en_passant_before = self.en_passant;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        let side = self.turn;
 | 
	
		
			
				|  |  | -        let others = self.get_all_side(!side);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        if !mov.is_null() {
 | 
	
		
			
				|  |  | -            match mov {
 | 
	
		
			
				|  |  | -                Move::Default{ mov, pc } => {
 | 
	
		
			
				|  |  | -                    let pt = pc.piece_type();
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    if pt == KING {
 | 
	
		
			
				|  |  | -                        // invalidate castling rights
 | 
	
		
			
				|  |  | -                        self.castling_rights[if side == BLACK { 2 } else { 0 }] = false;
 | 
	
		
			
				|  |  | -                        self.castling_rights[if side == BLACK { 3 } else { 1 }] = false;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    // invalidate castling rights
 | 
	
		
			
				|  |  | -                    if mov.from == square_from_indices(7, 0) || mov.to == square_from_indices(7, 0) {
 | 
	
		
			
				|  |  | -                        self.castling_rights[0] = false;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    if mov.from == square_from_indices(0, 0) || mov.to == square_from_indices(0, 0) {
 | 
	
		
			
				|  |  | -                        self.castling_rights[1] = false;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    if mov.from == square_from_indices(7, 7) || mov.to == square_from_indices(7, 7) {
 | 
	
		
			
				|  |  | -                        self.castling_rights[2] = false;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    if mov.from == square_from_indices(0, 7) || mov.to == square_from_indices(0, 7) {
 | 
	
		
			
				|  |  | -                        self.castling_rights[3] = false;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    // if it is a capture
 | 
	
		
			
				|  |  | -                    if from_square(mov.to) & others != 0 {
 | 
	
		
			
				|  |  | -                        let (other_piece, other_side) = self.get_square(mov.to);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                        // update zobrist
 | 
	
		
			
				|  |  | -                        if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                            let hup = zt.piece_hash(other_piece, other_side, mov.to);
 | 
	
		
			
				|  |  | -                            *zv ^= hup;
 | 
	
		
			
				|  |  | -                        }
 | 
	
		
			
				|  |  | -                        *self.get_piece_mut(other_piece, other_side) &= !from_square(mov.to);
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    // reset en passant and then check if possible
 | 
	
		
			
				|  |  | -                    self.en_passant = None;
 | 
	
		
			
				|  |  | -                    if pt == PAWN {
 | 
	
		
			
				|  |  | -                        let row1 = indices_from_square(mov.from).1;
 | 
	
		
			
				|  |  | -                        let row2 = indices_from_square(mov.to).1;
 | 
	
		
			
				|  |  | -                        // check for double push
 | 
	
		
			
				|  |  | -                        if i32::abs(row1 as i32 - row2 as i32) >= 2 {
 | 
	
		
			
				|  |  | -                            let opponent_pawns = self.get_piece(PAWN, !side);
 | 
	
		
			
				|  |  | -                            let target_bb = from_square(mov.to);
 | 
	
		
			
				|  |  | -                            if (west_one(target_bb) | east_one(target_bb)) & opponent_pawns != 0 {
 | 
	
		
			
				|  |  | -                                self.en_passant = Some(indices_from_square(mov.to).0);
 | 
	
		
			
				|  |  | -                            }
 | 
	
		
			
				|  |  | -                        }
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    let moved_piece = mov.apply_to(self.get_piece(pt, side));
 | 
	
		
			
				|  |  | -                    self.set_piece(pt, side, moved_piece);
 | 
	
		
			
				|  |  | -                    if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                        let hup = zt.piece_hash(pt, side, mov.from) ^ zt.piece_hash(pt, side, mov.to);
 | 
	
		
			
				|  |  | -                        *zv ^= hup;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                },
 | 
	
		
			
				|  |  | -                Move::Castling { mov:_, side, left } => {
 | 
	
		
			
				|  |  | -                    // invalidate future castling rights
 | 
	
		
			
				|  |  | -                    self.castling_rights[if side == BLACK { 2 } else { 0 }] = false;
 | 
	
		
			
				|  |  | -                    self.castling_rights[if side == BLACK { 3 } else { 1 }] = false;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    let king = self.get_piece(KING, side);
 | 
	
		
			
				|  |  | -                    let rook = if left {
 | 
	
		
			
				|  |  | -                        self.get_piece(ROOK, side) & (king << 4)
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    else {
 | 
	
		
			
				|  |  | -                        self.get_piece(ROOK, side) & (king >> 3)
 | 
	
		
			
				|  |  | -                    };
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    let new_king = if left { king << 2 } else { king >> 2 };
 | 
	
		
			
				|  |  | -                    let new_rook = if left { rook >> 3 } else { rook << 2 };
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    self.set_piece(KING, side, new_king);
 | 
	
		
			
				|  |  | -                    *self.get_piece_mut(ROOK, side) &= !rook;
 | 
	
		
			
				|  |  | -                    *self.get_piece_mut(ROOK, side) |= new_rook;
 | 
	
		
			
				|  |  | -                    if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                        let hupk = zt.piece_hash(KING, side, square(king)) ^ zt.piece_hash(KING, side, square(new_king));
 | 
	
		
			
				|  |  | -                        let hup = zt.piece_hash(ROOK, side, square(rook)) ^ zt.piece_hash(ROOK, side, square(new_rook));
 | 
	
		
			
				|  |  | -                        *zv ^= hup ^ hupk;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    self.en_passant = None;
 | 
	
		
			
				|  |  | -                },
 | 
	
		
			
				|  |  | -                Move::EnPassant { mov, beaten } => {
 | 
	
		
			
				|  |  | -                    if let Some(_ep) = self.en_passant {
 | 
	
		
			
				|  |  | -                        *self.get_piece_mut(PAWN, side) &= !from_square(mov.from);
 | 
	
		
			
				|  |  | -                        *self.get_piece_mut(PAWN, side) |= from_square(mov.to);
 | 
	
		
			
				|  |  | -                        *self.get_piece_mut(PAWN, !side) &= !from_square(beaten);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                        if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                            let hup = zt.piece_hash(PAWN, side, mov.from) ^ zt.piece_hash(PAWN, side, mov.to);
 | 
	
		
			
				|  |  | -                            let hupo = zt.piece_hash(PAWN, !side, beaten);
 | 
	
		
			
				|  |  | -                            *zv ^= hup ^ hupo;
 | 
	
		
			
				|  |  | -                        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                        self.en_passant = None;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    else {
 | 
	
		
			
				|  |  | -                        info!("internal en passant error");
 | 
	
		
			
				|  |  | -                        panic!("internal en passant error");
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                },
 | 
	
		
			
				|  |  | -                Move::Promotion { mov, pc } => {
 | 
	
		
			
				|  |  | -                    let promote_to = pc.piece_type();
 | 
	
		
			
				|  |  | -                    let captured = pc.capture_type();
 | 
	
		
			
				|  |  | -                    //if from_square(mov.to) & others != 0 {
 | 
	
		
			
				|  |  | -                    if let Some(pt) = captured {
 | 
	
		
			
				|  |  | -                        *self.get_piece_mut(pt, !side) &= !from_square(mov.to);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                        if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                            let hup = zt.piece_hash(pt, !side, mov.to);
 | 
	
		
			
				|  |  | -                            *zv ^= hup;
 | 
	
		
			
				|  |  | -                        }
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    *self.get_piece_mut(PAWN, side) &= !from_square(mov.from);
 | 
	
		
			
				|  |  | -                    if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                        let hup = zt.piece_hash(PAWN, side, mov.from);
 | 
	
		
			
				|  |  | -                        *zv ^= hup;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    match promote_to {
 | 
	
		
			
				|  |  | -                        QUEEN => { *self.queens_mut(side) |= from_square(mov.to); }
 | 
	
		
			
				|  |  | -                        ROOK => { *self.rooks_mut(side) |= from_square(mov.to); }
 | 
	
		
			
				|  |  | -                        BISHOP => { *self.bishops_mut(side) |= from_square(mov.to); }
 | 
	
		
			
				|  |  | -                        KNIGHT => { *self.knights_mut(side) |= from_square(mov.to); }
 | 
	
		
			
				|  |  | -                        _ => {
 | 
	
		
			
				|  |  | -                            info!("internal error");
 | 
	
		
			
				|  |  | -                        }
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                        let hup = zt.piece_hash(promote_to, side, mov.to);
 | 
	
		
			
				|  |  | -                        *zv ^= hup;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    self.en_passant = None;
 | 
	
		
			
				|  |  | -                },
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        if self.turn == BLACK {
 | 
	
		
			
				|  |  | -            self.turn_number += 1;
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        self.turn = !self.turn;
 | 
	
		
			
				|  |  | -        if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -            let hup = zt.turn_hash();
 | 
	
		
			
				|  |  | -            let castling_hup = zt.all_castling_rights_hash(castling_rights_before) ^ zt.all_castling_rights_hash(self.castling_rights);
 | 
	
		
			
				|  |  | -            let enpass_hup = en_passant_before.map(|f| zt.en_passant_hash(f)).unwrap_or(0) ^ self.en_passant.map(|f| zt.en_passant_hash(f)).unwrap_or(0);
 | 
	
		
			
				|  |  | -            *zv ^= hup ^ castling_hup ^ enpass_hup;
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        /*if !self.is_zobrist_correct() {
 | 
	
		
			
				|  |  | -            println!("{}", self.beautiful_print());
 | 
	
		
			
				|  |  | -            println!("incorrect zobrist after apply {} {:?}", mov.to_string(), mov);
 | 
	
		
			
				|  |  | -            info!("incorrect zobrist after apply");
 | 
	
		
			
				|  |  | -            panic!("zobrist");
 | 
	
		
			
				|  |  | -            let val = if let Some((ref zt, _zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                self.calculate_zobrist(zt)
 | 
	
		
			
				|  |  | -            } else { 0 };
 | 
	
		
			
				|  |  | -            if let Some((ref _zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                *zv = val;
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }*/
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        MoveUndo {
 | 
	
		
			
				|  |  | -            castling_rights_before,
 | 
	
		
			
				|  |  | -            halfmoves_since_last_event_before,
 | 
	
		
			
				|  |  | -            en_passant_before,
 | 
	
		
			
				|  |  | -            mov
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    pub fn undo_move(&mut self, umov: MoveUndo) {
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        /*if !self.is_zobrist_correct() {
 | 
	
		
			
				|  |  | -            println!("{}", self.beautiful_print());
 | 
	
		
			
				|  |  | -            println!("incorrect zobrist before undo {} {:?}", umov.mov.to_string(), umov.mov);
 | 
	
		
			
				|  |  | -            info!("incorrect zobrist before undo {} {:?}", umov.mov.to_string(), umov.mov);
 | 
	
		
			
				|  |  | -            panic!("zobrist");
 | 
	
		
			
				|  |  | -        }*/
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -            let crhup = zt.all_castling_rights_hash(self.castling_rights) ^ zt.all_castling_rights_hash(umov.castling_rights_before);
 | 
	
		
			
				|  |  | -            let enpass_hup = self.en_passant.map(|f| zt.en_passant_hash(f)).unwrap_or(0) ^ umov.en_passant_before.map(|f| zt.en_passant_hash(f)).unwrap_or(0);
 | 
	
		
			
				|  |  | -            *zv ^= crhup ^ enpass_hup;
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        self.castling_rights = umov.castling_rights_before;
 | 
	
		
			
				|  |  | -        self.halfmoves_since_last_event = umov.halfmoves_since_last_event_before;
 | 
	
		
			
				|  |  | -        self.en_passant = umov.en_passant_before;
 | 
	
		
			
				|  |  | -        
 | 
	
		
			
				|  |  | -        let mov = umov.mov;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        // the side that played the turn that is to be reverted
 | 
	
		
			
				|  |  | -        let side = !self.turn;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        if !mov.is_null() {
 | 
	
		
			
				|  |  | -            match mov {
 | 
	
		
			
				|  |  | -                Move::Default{ mov, pc } => {
 | 
	
		
			
				|  |  | -                    let piece_type = pc.piece_type();
 | 
	
		
			
				|  |  | -                    let captured = pc.capture_type();
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    let moved_piece_bb = self.get_piece_mut(piece_type, side);
 | 
	
		
			
				|  |  | -                    *moved_piece_bb &= !from_square(mov.to);
 | 
	
		
			
				|  |  | -                    *moved_piece_bb |= from_square(mov.from);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                        let hup = zt.piece_hash(piece_type, side, mov.from) ^ zt.piece_hash(piece_type, side, mov.to);
 | 
	
		
			
				|  |  | -                        *zv ^= hup;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    if let Some(pt) = captured {
 | 
	
		
			
				|  |  | -                        *self.get_piece_mut(pt, !side) |= from_square(mov.to);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                        if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                            let hup = zt.piece_hash(pt, !side, mov.to);
 | 
	
		
			
				|  |  | -                            *zv ^= hup;
 | 
	
		
			
				|  |  | -                        }
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                },
 | 
	
		
			
				|  |  | -                Move::Castling { mov: _, side, left } => {
 | 
	
		
			
				|  |  | -                    let king = self.get_piece(KING, side);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    let rook = if left {
 | 
	
		
			
				|  |  | -                        self.get_piece(ROOK, side) & (king >> 1)
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    else {
 | 
	
		
			
				|  |  | -                        self.get_piece(ROOK, side) & (king << 1)
 | 
	
		
			
				|  |  | -                    };
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    let old_king = if left { king >> 2 } else { king << 2 };
 | 
	
		
			
				|  |  | -                    let old_rook = if left { rook << 3 } else { rook >> 2 };
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    self.set_piece(KING, side, old_king);
 | 
	
		
			
				|  |  | -                    *self.get_piece_mut(ROOK, side) &= !rook;
 | 
	
		
			
				|  |  | -                    *self.get_piece_mut(ROOK, side) |= old_rook;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                        let khup = zt.piece_hash(KING, side, square(old_king)) ^ zt.piece_hash(KING, side, square(king));
 | 
	
		
			
				|  |  | -                        let rhup = zt.piece_hash(ROOK, side, square(rook)) ^ zt.piece_hash(ROOK, side, square(old_rook));
 | 
	
		
			
				|  |  | -                        *zv ^= khup ^ rhup;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                },
 | 
	
		
			
				|  |  | -                Move::EnPassant { mov, beaten } => {
 | 
	
		
			
				|  |  | -                    *self.get_piece_mut(PAWN, side) |= from_square(mov.from);
 | 
	
		
			
				|  |  | -                    *self.get_piece_mut(PAWN, side) &= !from_square(mov.to);
 | 
	
		
			
				|  |  | -                    *self.get_piece_mut(PAWN, !side) |= from_square(beaten);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                        let phup = zt.piece_hash(PAWN, side, mov.from) ^ zt.piece_hash(PAWN, side, mov.to);
 | 
	
		
			
				|  |  | -                        let chup = zt.piece_hash(PAWN, !side, beaten);
 | 
	
		
			
				|  |  | -                        *zv ^= phup ^ chup;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    // should already be reverted
 | 
	
		
			
				|  |  | -                    //self.en_passant = Some(indices_from_square(beaten).0);
 | 
	
		
			
				|  |  | -                },
 | 
	
		
			
				|  |  | -                Move::Promotion { mov, pc } => {
 | 
	
		
			
				|  |  | -                    let promote_to = pc.piece_type();
 | 
	
		
			
				|  |  | -                    let captured = pc.capture_type();
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    match promote_to {
 | 
	
		
			
				|  |  | -                        QUEEN => { *self.queens_mut(side) &= !from_square(mov.to); }
 | 
	
		
			
				|  |  | -                        ROOK => { *self.rooks_mut(side) &= !from_square(mov.to); }
 | 
	
		
			
				|  |  | -                        BISHOP => { *self.bishops_mut(side) &= !from_square(mov.to); }
 | 
	
		
			
				|  |  | -                        KNIGHT => { *self.knights_mut(side) &= !from_square(mov.to); }
 | 
	
		
			
				|  |  | -                        _ => {
 | 
	
		
			
				|  |  | -                            info!("internal error");
 | 
	
		
			
				|  |  | -                        }
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    *self.pawns_mut(side) |= from_square(mov.from);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                        let hup = zt.piece_hash(PAWN, side, mov.from) ^ zt.piece_hash(promote_to, side, mov.to);
 | 
	
		
			
				|  |  | -                        *zv ^= hup;
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                    if let Some(pt) = captured {
 | 
	
		
			
				|  |  | -                        *self.get_piece_mut(pt, !side) |= from_square(mov.to);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                        if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -                            let hup = zt.piece_hash(pt, !side, mov.to);
 | 
	
		
			
				|  |  | -                            *zv ^= hup;
 | 
	
		
			
				|  |  | -                        }
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                },
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        self.turn = side;
 | 
	
		
			
				|  |  | -        if side == BLACK {
 | 
	
		
			
				|  |  | -            self.turn_number -= 1;
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        if let Some((ref zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -            *zv ^= zt.turn_hash();
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        /*if !self.is_zobrist_correct() {
 | 
	
		
			
				|  |  | -            println!("{}", self.beautiful_print());
 | 
	
		
			
				|  |  | -            println!("incorrect zobrist after undo {} {:?}", umov.mov.to_string(), umov.mov);
 | 
	
		
			
				|  |  | -            info!("incorrect zobrist after undo {} {:?}", umov.mov.to_string(), umov.mov);
 | 
	
		
			
				|  |  | -            panic!("zobrist");
 | 
	
		
			
				|  |  | -        }*/
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        /*let val = if let Some((ref zt, _zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -            self.calculate_zobrist(zt)
 | 
	
		
			
				|  |  | -        } else { 0 };
 | 
	
		
			
				|  |  | -        if let Some((ref _zt, ref mut zv)) = self.zobrist {
 | 
	
		
			
				|  |  | -            *zv = val;
 | 
	
		
			
				|  |  | -        }*/
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -    
 | 
	
		
			
				|  |  | -    pub fn to_vector(&self) -> Vec<f64> {
 | 
	
		
			
				|  |  | -        let mut v = Vec::new();
 | 
	
		
			
				|  |  | -        for side in [WHITE, BLACK] {
 | 
	
		
			
				|  |  | -            for pt in 0..6 {
 | 
	
		
			
				|  |  | -                for sq in 0..64 {
 | 
	
		
			
				|  |  | -                    if self.get_square(sq) == (pt, side) {
 | 
	
		
			
				|  |  | -                        v.push(1f64);
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                    else {
 | 
	
		
			
				|  |  | -                        v.push(0f64);
 | 
	
		
			
				|  |  | -                    }
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -        v
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -}
 |