eryn/chessfriend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[position] Fix a bug in the pawn pushes move generator

Browse source
This commit is contained in:
Eryn Wells 2024-01-28 15:53:19 -08:00
parent 1f78d4811a
commit c558800385
1 changed files with 14 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

16
position/src/move_generator/pawn.rs
View file

@ -3,7 +3,7 @@
use super::{move_generator_declaration, MoveGeneratorInternal, MoveSet}; use super::{move_generator_declaration, MoveGeneratorInternal, MoveSet};
use crate::{MoveBuilder, Position}; use crate::{MoveBuilder, Position};
use chessfriend_bitboard::BitBoard; use chessfriend_bitboard::BitBoard;
use chessfriend_core::{Color, Piece, PlacedPiece, Square}; use chessfriend_core::{Color, Piece, PlacedPiece, Rank, Square};
#[derive(Debug)] #[derive(Debug)]
struct MoveIterator(usize, usize); struct MoveIterator(usize, usize);
@ -39,8 +39,20 @@ impl MoveGeneratorInternal for PawnMoveGenerator {
impl PawnMoveGenerator { impl PawnMoveGenerator {
fn pushes(position: &Position, piece: PlacedPiece) -> BitBoard { fn pushes(position: &Position, piece: PlacedPiece) -> BitBoard {
let square = piece.square();
let bitboard: BitBoard = square.into();
let empty_squares = position.empty_squares(); let empty_squares = position.empty_squares();
BitBoard::pawn_pushes(piece.square(), piece.color()) & empty_squares
let mut moves = bitboard.shift_north_one() & empty_squares;
if !(bitboard
& BitBoard::rank(Rank::PAWN_STARTING_RANKS[piece.color() as usize].as_index()))
.is_empty()
{
moves |= moves.shift_north_one() & empty_squares;
}
moves
} }
fn attacks(position: &Position, piece: PlacedPiece) -> BitBoard { fn attacks(position: &Position, piece: PlacedPiece) -> BitBoard {