Add half-move clock validation

This commit is contained in:
Bruno BELANYI 2024-04-01 22:41:01 +01:00
parent 2853cec7c9
commit ef3a1e4695
2 changed files with 20 additions and 0 deletions

View file

@ -21,6 +21,8 @@ pub enum InvalidError {
OverlappingColors, OverlappingColors,
/// The pre-computed combined occupancy boards does not match the other boards. /// The pre-computed combined occupancy boards does not match the other boards.
ErroneousCombinedOccupancy, ErroneousCombinedOccupancy,
/// Half-move clock is higher than total number of plies.
HalfMoveClockTooHigh,
} }
impl std::fmt::Display for InvalidError { impl std::fmt::Display for InvalidError {
@ -42,6 +44,7 @@ impl std::fmt::Display for InvalidError {
Self::ErroneousCombinedOccupancy => { Self::ErroneousCombinedOccupancy => {
"The pre-computed combined occupancy boards does not match the other boards." "The pre-computed combined occupancy boards does not match the other boards."
} }
Self::HalfMoveClockTooHigh => "Half-move clock is higher than total number of plies.",
}; };
write!(f, "{}", error_msg) write!(f, "{}", error_msg)
} }

View file

@ -208,6 +208,11 @@ impl ChessBoard {
/// Validate the state of the board. Return Err([InvalidError]) if an issue is found. /// Validate the state of the board. Return Err([InvalidError]) if an issue is found.
pub fn validate(&self) -> Result<(), InvalidError> { pub fn validate(&self) -> Result<(), InvalidError> {
// Make sure the clocks are in agreement.
if u32::from(self.half_move_clock()) > self.total_plies() {
return Err(InvalidError::HalfMoveClockTooHigh);
}
// Don't overlap pieces. // Don't overlap pieces.
for piece in Piece::iter() { for piece in Piece::iter() {
#[allow(clippy::collapsible_if)] #[allow(clippy::collapsible_if)]
@ -423,6 +428,18 @@ mod test {
assert!(default_position.is_valid()); assert!(default_position.is_valid());
} }
#[test]
fn invalid_half_moves_clock() {
let res = {
let mut builder = ChessBoardBuilder::new();
builder[Square::E1] = Some((Piece::King, Color::White));
builder[Square::E8] = Some((Piece::King, Color::Black));
builder.with_half_move_clock(10);
TryInto::<ChessBoard>::try_into(builder)
};
assert_eq!(res.err().unwrap(), InvalidError::HalfMoveClockTooHigh);
}
#[test] #[test]
fn invalid_overlapping_pieces() { fn invalid_overlapping_pieces() {
let position = ChessBoard { let position = ChessBoard {