This commit is contained in:
Simon Gardling 2025-01-30 16:33:10 -05:00
parent 472141533b
commit 73d24e7748
Signed by: titaniumtown
GPG Key ID: 9AB28AC10ECE533D

View File

@ -1,5 +1,3 @@
use std::collections::HashMap;
use crate::{ use crate::{
agent::Agent, agent::Agent,
board::{Board, BOARD_SIZE}, board::{Board, BOARD_SIZE},
@ -49,20 +47,20 @@ impl Move {
} }
} }
fn problem_space(board: &Board, piece: Piece) -> Vec<Move> { fn problem_space(board: &Board, color: Piece) -> Vec<Move> {
(0..BOARD_SIZE) (0..BOARD_SIZE)
.flat_map(|i| { .flat_map(|i| {
(0..BOARD_SIZE) (0..BOARD_SIZE)
.map(|j| (i, j)) .map(|j| (i, j))
.collect::<Vec<(usize, usize)>>() .collect::<Vec<(usize, usize)>>()
}) })
.flat_map(|(i, j)| board.what_if(i, j, piece).map(|x| (i, j, x))) .flat_map(|(i, j)| board.what_if(i, j, color).map(|x| (i, j, x)))
.map(|(i, j, (b, c))| Move { .map(|(i, j, (board, captured))| Move {
i, i,
j, j,
captured: c, captured,
color: piece, color,
board: b, board,
next_move: Vec::new(), next_move: Vec::new(),
}) })
.collect() .collect()