use crate::constants::*;
use crate::square::*;
pub type Bitboard = u64;
pub trait BitboardTrait {
fn pretty_print_string(&self) -> String;
fn pop_bitboard(&mut self) -> (Bitboard, bool);
fn pop_square(&mut self) -> (Square, bool);
fn variation_count(self) -> usize;
}
impl BitboardTrait for Bitboard {
fn pretty_print_string(&self) -> String {
let mut bb = *self;
let mut buff = "".to_string();
let mut bits = 0;
loop {
if bits % 8 == 0 {
buff += &format!("{}", NUM_FILES - bits / 8).to_string();
}
if bb & (1 << 63) != 0 {
buff += "1"
} else {
buff += "."
}
if bits % 8 == 7 {
buff += "*\n"
}
bb = bb << 1;
bits = bits + 1;
if bits == 64 {
break;
}
}
format! {"bitboard {:#016x}\n**********\n{}*abcdefgh*\n", &self, buff}
}
fn pop_bitboard(&mut self) -> (Bitboard, bool) {
if *self == 0 {
return (0, false);
}
let bb = 1 << (self.trailing_zeros() as usize);
*self &= !bb;
return (bb, true);
}
fn pop_square(&mut self) -> (Square, bool) {
let (bb, ok) = self.pop_bitboard();
if ok {
let tzs = bb.trailing_zeros() as usize;
(
rank_file(tzs / NUM_FILES, LAST_FILE - (tzs % NUM_FILES)),
true,
)
} else {
(0, false)
}
}
fn variation_count(self) -> usize {
1 << self.count_ones()
}
}