For pawn caputre rays do same as knight rays

This commit is contained in:
2025-02-22 15:47:13 +01:00
parent eda2ec6c93
commit e798480b72

View File

@@ -196,18 +196,13 @@ impl PieceType {
Color::Black => Dir::Down, Color::Black => Dir::Down,
Color::White => Dir::Up, Color::White => Dir::Up,
}; };
let mut capture_rays = vec![]; [&[&direction, &Dir::Left], &[&direction, &Dir::Right]]
if let Some(p) = position.chain(&[&direction, &Dir::Left]) { .into_iter()
capture_rays.push(Ray { .filter_map(|dirs| position.chain(dirs))
.map(|p| Ray {
positions: vec![p], positions: vec![p],
}); })
} .collect()
if let Some(p) = position.chain(&[&direction, &Dir::Right]) {
capture_rays.push(Ray {
positions: vec![p],
});
}
capture_rays
} }
// Knight // Knight