initial BVM usage (halfs performance :( )
This commit is contained in:
parent
dc9432b07b
commit
e1d83825b7
@ -1,20 +1,33 @@
|
||||
use crate::repr::{Board, Piece, PosMap};
|
||||
|
||||
pub struct BoardValueMap(PosMap<f32>);
|
||||
pub struct BoardValueMap(PosMap<i8>);
|
||||
|
||||
impl BoardValueMap {
|
||||
pub fn board_value(&self, board: &Board, color: Piece) -> f32 {
|
||||
let mut board_value: f32 = 0.0;
|
||||
for (i, j) in Board::all_positions() {
|
||||
if let Some(pos_p) = board.get(i, j) {
|
||||
pub fn board_value(&self, board: &Board, color: Piece) -> i8 {
|
||||
Board::all_positions()
|
||||
.filter_map(|(i, j)| board.get(i, j).map(|p| (i, j, p)))
|
||||
.map(|(i, j, pos_p)| {
|
||||
let mut value = *self.0.get(i, j);
|
||||
if pos_p != color {
|
||||
// enemy has position
|
||||
value = -value;
|
||||
}
|
||||
board_value += value;
|
||||
}
|
||||
value
|
||||
})
|
||||
.sum()
|
||||
}
|
||||
|
||||
pub fn new() -> Self {
|
||||
let mut map = PosMap::new();
|
||||
|
||||
for (i, j) in Board::all_positions() {
|
||||
map.set(i, j, 1);
|
||||
}
|
||||
board_value
|
||||
|
||||
for (i, j) in Board::sides() {
|
||||
map.set(i, j, 4);
|
||||
}
|
||||
|
||||
Self(map)
|
||||
}
|
||||
}
|
||||
|
||||
@ -1,5 +1,9 @@
|
||||
use lazy_static::lazy_static;
|
||||
|
||||
use crate::repr::{Board, Piece, Winner};
|
||||
|
||||
use super::board_value::BoardValueMap;
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct Move {
|
||||
/// `i` position of move
|
||||
@ -28,6 +32,10 @@ pub struct Move {
|
||||
pub lazy_children: bool,
|
||||
}
|
||||
|
||||
lazy_static! {
|
||||
static ref BVM: BoardValueMap = BoardValueMap::new();
|
||||
}
|
||||
|
||||
impl Move {
|
||||
pub const fn coords(&self) -> (usize, usize) {
|
||||
(self.i, self.j)
|
||||
@ -47,17 +55,6 @@ impl Move {
|
||||
return 0;
|
||||
}
|
||||
|
||||
let mut self_value = self.board.net_score(agent_color) as i64;
|
||||
let corner_v_agent = Board::sides()
|
||||
.filter(|&(i, j)| self.board.get_piece(i, j, agent_color))
|
||||
.count() as i64;
|
||||
let corner_v_not_agent = Board::sides()
|
||||
.filter(|&(i, j)| self.board.get_piece(i, j, !agent_color))
|
||||
.count() as i64;
|
||||
|
||||
// make net-corner capture important
|
||||
self_value += (corner_v_agent - corner_v_not_agent) * 4;
|
||||
|
||||
self_value
|
||||
BVM.board_value(&self.board, agent_color) as i64
|
||||
}
|
||||
}
|
||||
|
||||
@ -26,11 +26,21 @@ type ChainCollection = ArrayVec<Chain, 8>;
|
||||
/// Map of all points on the board against some type T
|
||||
/// Used to index like so: example[i][j]
|
||||
/// with each coordinate
|
||||
pub struct PosMap<T>(ArrayVec<T, BOARD_AREA>);
|
||||
pub struct PosMap<T: Default>(ArrayVec<T, BOARD_AREA>);
|
||||
|
||||
impl<T: Default> PosMap<T> {
|
||||
const fn index(row: usize, col: usize) -> usize {
|
||||
row * BOARD_SIZE + col
|
||||
}
|
||||
|
||||
pub fn new() -> Self {
|
||||
Self(ArrayVec::from_iter(
|
||||
(0..BOARD_AREA).map(|_| Default::default()),
|
||||
))
|
||||
}
|
||||
|
||||
impl<T> PosMap<T> {
|
||||
pub fn get(&self, row: usize, col: usize) -> &T {
|
||||
let index = row * BOARD_SIZE + col;
|
||||
let index = Self::index(row, col);
|
||||
|
||||
debug_assert!(
|
||||
BOARD_AREA + 1 >= index,
|
||||
@ -40,6 +50,17 @@ impl<T> PosMap<T> {
|
||||
|
||||
unsafe { self.0.get_unchecked(index) }
|
||||
}
|
||||
|
||||
pub fn set(&mut self, row: usize, col: usize, value: T) {
|
||||
let index = Self::index(row, col);
|
||||
debug_assert!(
|
||||
BOARD_AREA + 1 >= index,
|
||||
"index out of range, was: {}",
|
||||
index
|
||||
);
|
||||
|
||||
self.0[index] = value;
|
||||
}
|
||||
}
|
||||
|
||||
/// Creates a lookup map for adjacencies and chains from each position on the board
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user