Add bitboard iteration
Introduce 'BitboardIterator', use it to implement 'IntoIterator' for 'Bitboard'.
This commit is contained in:
parent
f4a92c0681
commit
a0fcf3285c
17
src/board/bitboard/iterator.rs
Normal file
17
src/board/bitboard/iterator.rs
Normal file
|
@ -0,0 +1,17 @@
|
||||||
|
/// An [Iterator](std::iter::Iterator) of [Square](crate::board::Square) contained in a
|
||||||
|
/// [Bitboard](crate::board::Bitboard).
|
||||||
|
pub struct BitboardIterator(pub(crate) u64);
|
||||||
|
|
||||||
|
impl Iterator for BitboardIterator {
|
||||||
|
type Item = crate::board::Square;
|
||||||
|
|
||||||
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
|
if self.0 == 0 {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
let lsb = self.0.trailing_zeros() as usize;
|
||||||
|
self.0 ^= 1 << lsb;
|
||||||
|
Some(crate::board::Square::from_index(lsb))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -1,4 +1,6 @@
|
||||||
use super::Square;
|
use super::Square;
|
||||||
|
mod iterator;
|
||||||
|
use iterator::*;
|
||||||
|
|
||||||
/// Use a 64-bit number to represent a chessboard. Each bit is mapped from to a specific square, so
|
/// Use a 64-bit number to represent a chessboard. Each bit is mapped from to a specific square, so
|
||||||
/// that index 0 -> A1, 1 -> A2, ..., 63 -> H8.
|
/// that index 0 -> A1, 1 -> A2, ..., 63 -> H8.
|
||||||
|
@ -40,6 +42,16 @@ impl Default for Bitboard {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Iterate over the [Square](crate::board::Square) values included in the board.
|
||||||
|
impl IntoIterator for Bitboard {
|
||||||
|
type IntoIter = BitboardIterator;
|
||||||
|
type Item = Square;
|
||||||
|
|
||||||
|
fn into_iter(self) -> Self::IntoIter {
|
||||||
|
BitboardIterator(self.0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// Treat bitboard as a set of squares, shift each square's index left by the amount given.
|
/// Treat bitboard as a set of squares, shift each square's index left by the amount given.
|
||||||
impl std::ops::Shl<usize> for Bitboard {
|
impl std::ops::Shl<usize> for Bitboard {
|
||||||
type Output = Bitboard;
|
type Output = Bitboard;
|
||||||
|
@ -155,6 +167,37 @@ mod test {
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::board::square::*;
|
use crate::board::square::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn iter() {
|
||||||
|
assert_eq!(Bitboard::EMPTY.into_iter().collect::<Vec<_>>(), Vec::new());
|
||||||
|
assert_eq!(
|
||||||
|
Bitboard::RANKS[0].into_iter().collect::<Vec<_>>(),
|
||||||
|
vec![
|
||||||
|
Square::A1,
|
||||||
|
Square::B1,
|
||||||
|
Square::C1,
|
||||||
|
Square::D1,
|
||||||
|
Square::E1,
|
||||||
|
Square::F1,
|
||||||
|
Square::G1,
|
||||||
|
Square::H1,
|
||||||
|
]
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
Bitboard::FILES[0].into_iter().collect::<Vec<_>>(),
|
||||||
|
vec![
|
||||||
|
Square::A1,
|
||||||
|
Square::A2,
|
||||||
|
Square::A3,
|
||||||
|
Square::A4,
|
||||||
|
Square::A5,
|
||||||
|
Square::A6,
|
||||||
|
Square::A7,
|
||||||
|
Square::A8,
|
||||||
|
]
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn left_shift() {
|
fn left_shift() {
|
||||||
assert_eq!(Bitboard::RANKS[0] << 1, Bitboard::RANKS[1]);
|
assert_eq!(Bitboard::RANKS[0] << 1, Bitboard::RANKS[1]);
|
||||||
|
|
Loading…
Reference in a new issue