File size: 2,103 Bytes
d7ecc62 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | //! Game validation: replay and check legality at every ply. Spec §7.6.
use rayon::prelude::*;
use crate::board::GameState;
/// Validate games by replaying and checking legality at every ply.
/// Returns (is_valid, first_illegal) as flat arrays.
pub fn validate_games(
move_ids: &[i16], // [batch * max_ply]
game_lengths: &[i16], // [batch]
max_ply: usize,
) -> (Vec<bool>, Vec<i16>) {
let batch = game_lengths.len();
let results: Vec<(bool, i16)> = (0..batch)
.into_par_iter()
.map(|b| {
let length = game_lengths[b] as usize;
let mut state = GameState::new();
for t in 0..length {
let token = move_ids[b * max_ply + t] as u16;
if state.make_move(token).is_err() {
return (false, t as i16);
}
}
(true, -1i16)
})
.collect();
let (is_valid, first_illegal) = results.into_iter().unzip();
(is_valid, first_illegal)
}
#[cfg(test)]
mod tests {
use super::*;
use crate::batch::generate_random_games;
#[test]
fn test_validate_valid_games() {
let batch = generate_random_games(10, 256, 42);
let (valid, first_ill) = validate_games(
&batch.move_ids,
&batch.game_lengths,
256,
);
for (i, &v) in valid.iter().enumerate() {
assert!(v, "Game {} should be valid", i);
assert_eq!(first_ill[i], -1);
}
}
#[test]
fn test_validate_invalid_game() {
// Create a game with an illegal move at ply 1
let max_ply = 256;
let mut move_ids = vec![0i16; max_ply];
// e2e4 is legal
move_ids[0] = crate::vocab::base_grid_token(12, 28) as i16;
// e2e4 again is illegal (pawn already moved)
move_ids[1] = crate::vocab::base_grid_token(12, 28) as i16;
let game_lengths = vec![2i16];
let (valid, first_ill) = validate_games(&move_ids, &game_lengths, max_ply);
assert!(!valid[0]);
assert_eq!(first_ill[0], 1);
}
}
|