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

[board] Put a cached Moves object into a OnceCell on Position

Browse source 
Cache move generation so we don't have to remake it every time.
This commit is contained in:
Eryn Wells 2024-01-22 08:12:28 -08:00
parent 8dbf44c741
commit c1008ef672
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
board/src/position/position.rs
View file

@ -18,6 +18,7 @@ pub struct Position {
pieces: PieceBitBoards, pieces: PieceBitBoards,
en_passant_square: Option<Square>, en_passant_square: Option<Square>,
sight: [OnceCell<BitBoard>; 2], sight: [OnceCell<BitBoard>; 2],
moves: OnceCell<Moves>,
half_move_counter: u16, half_move_counter: u16,
full_move_number: u16, full_move_number: u16,
@ -31,6 +32,7 @@ impl Position {
pieces: PieceBitBoards::default(), pieces: PieceBitBoards::default(),
en_passant_square: None, en_passant_square: None,
sight: [OnceCell::new(), OnceCell::new()], sight: [OnceCell::new(), OnceCell::new()],
moves: OnceCell::new(),
half_move_counter: 0, half_move_counter: 0,
full_move_number: 1, full_move_number: 1,
} }
@ -62,6 +64,7 @@ impl Position {
pieces: PieceBitBoards::new([white_pieces, black_pieces]), pieces: PieceBitBoards::new([white_pieces, black_pieces]),
en_passant_square: None, en_passant_square: None,
sight: [OnceCell::new(), OnceCell::new()], sight: [OnceCell::new(), OnceCell::new()],
moves: OnceCell::new(),
half_move_counter: 0, half_move_counter: 0,
full_move_number: 1, full_move_number: 1,
} }
@ -121,8 +124,9 @@ impl Position {
self.piece_on_square(square) self.piece_on_square(square)
} }
pub fn moves(&self) -> Moves { pub fn moves(&self) -> &Moves {
Moves::new(self, self.color_to_move) self.moves
.get_or_init(|| Moves::new(self, self.color_to_move))
} }
/// Return a BitBoard representing the set of squares containing a piece. /// Return a BitBoard representing the set of squares containing a piece.
@ -215,6 +219,7 @@ impl Position {
en_passant_square, en_passant_square,
pieces, pieces,
sight: [OnceCell::new(), OnceCell::new()], sight: [OnceCell::new(), OnceCell::new()],
moves: OnceCell::new(),
half_move_counter: 0, half_move_counter: 0,
full_move_number: 1, full_move_number: 1,
} }