stefiosif/zeal
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make new fn's to update the State struct

Browse Source
This commit is contained in:
stefiosif
2024-06-29 16:22:43 +03:00
parent b46c5874db
commit f3b2aabf46
1 changed files with 40 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
src/game.rs
View File

@@ -71,6 +71,46 @@ impl State {
self.en_passant_target_square
}
pub fn set_castling_ability(&mut self, color: Color, castle: Castle) {
match color {
Color::White => self.castling_ability[0] = castle,
Color::Black => self.castling_ability[1] = castle,
}
}
pub fn update_castling_state(&mut self, square: u8, color: Color) {
if square == 0 || square == 7 {
match (color, self.get_castling_ability(color)) {
(_, Castle::Both) => self.set_castling_ability(color, Castle::Long),
(_, Castle::Short) => self.set_castling_ability(color, Castle::None),
_ => (),
}
}
if square == 56 || square == 63 {
match (color, self.get_castling_ability(color)) {
(_, Castle::Both) => self.set_castling_ability(color, Castle::Short),
(_, Castle::Short) => self.set_castling_ability(color, Castle::None),
_ => (),
}
}
if square == 4 || square == 60 {
self.set_castling_ability(color, Castle::None)
}
}
pub fn next_turn(&mut self) {
self.side_to_move = match self.side_to_move {
Color::White => Color::Black,
Color::Black => Color::White,
}
}
pub fn set_en_passant_target_square(&mut self, sq: Option<u8>) {
self.en_passant_target_square = sq;
}
pub const fn get_castling_ability(&self, color: Color) -> Castle {
match color {
Color::White => self.castling_ability[0],