Merge branch 'main' into pretty-print-position

This commit is contained in:
Eryn Wells 2023-12-26 13:28:55 -07:00
commit a963cee1e7
8 changed files with 52 additions and 34 deletions

View file

@ -1,9 +1,7 @@
// Eryn Wells <eryn@erynwells.me> // Eryn Wells <eryn@erynwells.me>
mod bit_scanner; use super::BitScanner;
use crate::Square;
use self::bit_scanner::BitScanner;
use crate::square::Square;
use std::fmt; use std::fmt;
use std::ops::{BitAnd, BitOr, Not}; use std::ops::{BitAnd, BitOr, Not};
@ -154,7 +152,7 @@ impl BitOr<&BitBoard> for BitBoard {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
use crate::square::Square; use crate::Square;
#[test] #[test]
fn is_empty() { fn is_empty() {
@ -165,10 +163,10 @@ mod tests {
fn has_piece_at() { fn has_piece_at() {
let bb = BitBoard(0b1001100); let bb = BitBoard(0b1001100);
let c1 = Square::from_algebraic_string("c1").expect("Unable to get square for 'c1'"); let c1 = Square::from_algebraic_str("c1").expect("Unable to get square for 'c1'");
assert!(bb.has_piece_at(&c1)); assert!(bb.has_piece_at(&c1));
let b1 = Square::from_algebraic_string("b1").expect("Unable to get square for 'b1'"); let b1 = Square::from_algebraic_str("b1").expect("Unable to get square for 'b1'");
assert!(!bb.has_piece_at(&b1)); assert!(!bb.has_piece_at(&b1));
} }

View file

@ -0,0 +1,5 @@
mod bit_scanner;
mod bitboard;
pub(crate) use self::bit_scanner::BitScanner;
pub(crate) use self::bitboard::BitBoard;

View file

@ -3,3 +3,7 @@ mod neighbor;
mod piece; mod piece;
mod position; mod position;
mod square; mod square;
pub use piece::{Color, Piece};
pub use position::Position;
pub use square::Square;

View file

@ -1,6 +1,6 @@
// Eryn Wells <eryn@erynwells.me> // Eryn Wells <eryn@erynwells.me>
use crate::square::Square; use crate::Square;
#[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)] #[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
pub enum Color { pub enum Color {

View file

@ -0,0 +1,5 @@
mod pieces;
mod position;
pub use pieces::Pieces;
pub use position::Position;

View file

@ -1,10 +1,9 @@
// Eryn Wells <eryn@erynwells.me> // Eryn Wells <eryn@erynwells.me>
use super::Position;
use crate::bitboard::BitBoard; use crate::bitboard::BitBoard;
use crate::piece::{Color, Piece, PlacedPiece, Shape}; use crate::piece::{Color, Piece, PlacedPiece, Shape};
use crate::position::Position; use crate::Square;
use crate::square::Square;
use std::collections::HashSet;
pub struct Pieces<'a> { pub struct Pieces<'a> {
color: Color, color: Color,
@ -89,11 +88,12 @@ impl<'a> Iterator for Pieces<'a> {
mod tests { mod tests {
use super::*; use super::*;
use crate::piece::{Color, Piece, Shape}; use crate::piece::{Color, Piece, Shape};
use crate::position::Position; use crate::Position;
use crate::square::Square; use crate::Square;
use std::collections::HashSet;
fn square_at(sq: &str) -> Square { fn square_at(sq: &str) -> Square {
Square::from_algebraic_string(sq).expect(sq) Square::from_algebraic_str(sq).expect(sq)
} }
fn place_piece_in_position(pos: &mut Position, sq: &str, piece: Piece) { fn place_piece_in_position(pos: &mut Position, sq: &str, piece: Piece) {
@ -110,7 +110,7 @@ mod tests {
#[test] #[test]
fn one() { fn one() {
let sq = Square::from_algebraic_string("e4").expect("e4"); let sq = Square::from_algebraic_str("e4").expect("e4");
let mut pos = Position::empty(); let mut pos = Position::empty();
pos.place_piece(&Piece::new(Color::White, Shape::Queen), &sq) pos.place_piece(&Piece::new(Color::White, Shape::Queen), &sq)

View file

@ -1,16 +1,16 @@
// Eryn Wells <eryn@erynwells.me> // Eryn Wells <eryn@erynwells.me>
mod pieces; use super::Pieces;
use self::pieces::Pieces;
use crate::bitboard::BitBoard; use crate::bitboard::BitBoard;
use crate::piece::{Piece, PiecePlacementError}; use crate::piece::{Color, Piece, PiecePlacementError};
use crate::square::Square; use crate::Square;
use std::fmt; use std::fmt;
use std::fmt::Write; use std::fmt::Write;
#[derive(Clone, Eq, Hash, PartialEq)] #[derive(Clone, Eq, Hash, PartialEq)]
pub struct Position { pub struct Position {
color_to_move: Color,
/// Composite bitboards for all the pieces of a particular color. /// Composite bitboards for all the pieces of a particular color.
pieces_per_color: [BitBoard; 2], pieces_per_color: [BitBoard; 2],
@ -19,8 +19,9 @@ pub struct Position {
} }
impl Position { impl Position {
fn empty() -> Position { pub fn empty() -> Position {
Position { Position {
color_to_move: Color::White,
pieces_per_color: [BitBoard::empty(), BitBoard::empty()], pieces_per_color: [BitBoard::empty(), BitBoard::empty()],
pieces_per_type: [ pieces_per_type: [
[ [
@ -64,6 +65,7 @@ impl Position {
]; ];
Position { Position {
color_to_move: Color::White,
pieces_per_color: [ pieces_per_color: [
white_pieces.iter().fold(BitBoard::empty(), |a, b| a | b), white_pieces.iter().fold(BitBoard::empty(), |a, b| a | b),
black_pieces.iter().fold(BitBoard::empty(), |a, b| a | b), black_pieces.iter().fold(BitBoard::empty(), |a, b| a | b),
@ -72,7 +74,11 @@ impl Position {
} }
} }
fn place_piece(&mut self, piece: &Piece, square: &Square) -> Result<(), PiecePlacementError> { pub fn place_piece(
&mut self,
piece: &Piece,
square: &Square,
) -> Result<(), PiecePlacementError> {
let bitboard = self.bitboard_for_piece_mut(piece); let bitboard = self.bitboard_for_piece_mut(piece);
if bitboard.has_piece_at(&square) { if bitboard.has_piece_at(&square) {
@ -92,7 +98,7 @@ impl Position {
&mut self.pieces_per_type[piece.color as usize][piece.shape as usize] &mut self.pieces_per_type[piece.color as usize][piece.shape as usize]
} }
fn pieces(&self, color: Color) -> Pieces { pub fn pieces(&self, color: Color) -> Pieces {
Pieces::new(&self, color) Pieces::new(&self, color)
} }
} }
@ -159,7 +165,7 @@ mod tests {
let mut position = Position::empty(); let mut position = Position::empty();
let piece = Piece::new(Color::White, Shape::Queen); let piece = Piece::new(Color::White, Shape::Queen);
let square = Square::from_algebraic_string("e4").expect("Unable to get e4 square"); let square = Square::from_algebraic_str("e4").expect("Unable to get e4 square");
position position
.place_piece(&piece, &square) .place_piece(&piece, &square)

View file

@ -45,7 +45,7 @@ impl Square {
}) })
} }
pub fn from_algebraic_string(s: &str) -> Result<Square, ParseSquareError> { pub fn from_algebraic_str(s: &str) -> Result<Square, ParseSquareError> {
s.parse() s.parse()
} }
@ -142,27 +142,27 @@ mod tests {
#[test] #[test]
fn good_algebraic_input() { fn good_algebraic_input() {
let sq1 = Square::from_algebraic_string("a4").expect("Failed to parse 'a4' square"); let sq1 = Square::from_algebraic_str("a4").expect("Failed to parse 'a4' square");
assert_eq!(sq1.file, 0); assert_eq!(sq1.file, 0);
assert_eq!(sq1.rank, 3); assert_eq!(sq1.rank, 3);
let sq2 = Square::from_algebraic_string("B8").expect("Failed to parse 'B8' square"); let sq2 = Square::from_algebraic_str("B8").expect("Failed to parse 'B8' square");
assert_eq!(sq2.file, 1); assert_eq!(sq2.file, 1);
assert_eq!(sq2.rank, 7); assert_eq!(sq2.rank, 7);
let sq3 = Square::from_algebraic_string("e4").expect("Failed to parse 'B8' square"); let sq3 = Square::from_algebraic_str("e4").expect("Failed to parse 'B8' square");
assert_eq!(sq3.rank, 3, "Expected rank of e4 to be 3"); assert_eq!(sq3.rank, 3, "Expected rank of e4 to be 3");
assert_eq!(sq3.file, 4, "Expected file of e4 to be 4"); assert_eq!(sq3.file, 4, "Expected file of e4 to be 4");
} }
#[test] #[test]
fn bad_algebraic_input() { fn bad_algebraic_input() {
Square::from_algebraic_string("a0").expect_err("Got valid Square for 'a0'"); Square::from_algebraic_str("a0").expect_err("Got valid Square for 'a0'");
Square::from_algebraic_string("j3").expect_err("Got valid Square for 'j3'"); Square::from_algebraic_str("j3").expect_err("Got valid Square for 'j3'");
Square::from_algebraic_string("a11").expect_err("Got valid Square for 'a11'"); Square::from_algebraic_str("a11").expect_err("Got valid Square for 'a11'");
Square::from_algebraic_string("b-1").expect_err("Got valid Square for 'b-1'"); Square::from_algebraic_str("b-1").expect_err("Got valid Square for 'b-1'");
Square::from_algebraic_string("a 1").expect_err("Got valid Square for 'a 1'"); Square::from_algebraic_str("a 1").expect_err("Got valid Square for 'a 1'");
Square::from_algebraic_string("").expect_err("Got valid Square for ''"); Square::from_algebraic_str("").expect_err("Got valid Square for ''");
} }
#[test] #[test]