This commit is contained in:
2026-10-21 16:58:16 -04:00
parent 03f11e2ed8
commit 00f3d33edf
5 changed files with 41 additions and 12 deletions
+1
View File
@@ -1,3 +1,4 @@
pub mod benchmark;
pub mod play;
pub mod selfplay;
pub mod train;
+1 -1
View File
@@ -1 +1 @@
pub fn run() {}
+1
View File
@@ -0,0 +1 @@
pub fn run() {}
+37 -10
View File
@@ -249,25 +249,20 @@ pub fn decode_move(idx: usize, board: &Board) -> Option<ChessMove> {
return None;
}
// Default pawn moves to final rank (perspective fr == 6 -> tr == 7) to Queen promotion if not underpromotion
let stm = board.side_to_move();
// Default pawn moves to final rank (perspective fr == 6 -> tr == 7) to Queen promotion if not underpromotion.
// fr/ff are perspective coordinates, so deorient back to real squares before inspecting the board.
if promo.is_none()
&& fr == 6
&& tr == 7
&& dy == 1
&& dx.abs() <= 1
&& board.piece_on(Square::make_square(
Rank::from_index(fr as usize),
File::from_index(ff as usize),
)) == Some(Piece::Pawn)
&& board.piece_on(deorient_square(ff, fr, stm)) == Some(Piece::Pawn)
{
promo = Some(Piece::Queen);
}
// if fr == 6 && tr == 7 && dy == 1 && dx.abs() <= 1 && promo.is_none() {
// promo = Some(Piece::Queen);
// }
let stm = board.side_to_move();
let from = deorient_square(ff, fr, stm);
let to = deorient_square(tf, tr, stm);
@@ -349,4 +344,36 @@ mod tests {
assert_eq!(mv, dec);
}
#[test]
fn queen_promotion_roundtrip_black() {
// Black pawn on d2, white king on h1, black king on e8, black to move.
// d2-d1 is a legal promotion; d1 is empty.
let board = Board::from_str("4k3/8/8/8/8/8/3p4/7K b - - 0 1").unwrap();
let from = Square::make_square(Rank::Second, File::D);
let to = Square::make_square(Rank::First, File::D);
let mv = ChessMove::new(from, to, Some(Piece::Queen));
let enc = encode_move(mv, Color::Black).unwrap();
let dec = decode_move(enc, &board).unwrap();
assert_eq!(mv, dec);
}
#[test]
fn underpromotion_roundtrip_black() {
let board = Board::from_str("4k3/8/8/8/8/8/3p4/7K b - - 0 1").unwrap();
let from = Square::make_square(Rank::Second, File::D);
let to = Square::make_square(Rank::First, File::D);
let mv = ChessMove::new(from, to, Some(Piece::Knight));
let enc = encode_move(mv, Color::Black).unwrap();
let dec = decode_move(enc, &board).unwrap();
assert_eq!(mv, dec);
}
}
+1 -1
View File
@@ -1,5 +1,5 @@
use chess::{ChessMove, Game};
use engine::{legal_action_mask, Engine};
use engine::Engine;
use std::io;
use std::str::FromStr;