Use do_move_bb() helpers when doing a castle

[stockfish] / src / position.cpp

diff --git a/src/position.cpp b/src/position.cpp
index 1179f6f98a8aa61733cb81df7cdecae906c4a042..d1e3f448071a947d7950032cf93eedd9a22d0609 100644 (file)
--- a/src/position.cpp
+++ b/src/position.cpp
@@ -905,21 +905,16 @@ void Position::do_castle_move(Move m) {
       rto = relative_square(us, SQ_D1);
   }
 
-  // Remove pieces from source squares
-  clear_bit(&(byColorBB[us]), kfrom);
-  clear_bit(&(byTypeBB[KING]), kfrom);
-  clear_bit(&(byTypeBB[0]), kfrom); // HACK: byTypeBB[0] == occupied squares
-  clear_bit(&(byColorBB[us]), rfrom);
-  clear_bit(&(byTypeBB[ROOK]), rfrom);
-  clear_bit(&(byTypeBB[0]), rfrom); // HACK: byTypeBB[0] == occupied squares
-
-  // Put pieces on destination squares
-  set_bit(&(byColorBB[us]), kto);
-  set_bit(&(byTypeBB[KING]), kto);
-  set_bit(&(byTypeBB[0]), kto); // HACK: byTypeBB[0] == occupied squares
-  set_bit(&(byColorBB[us]), rto);
-  set_bit(&(byTypeBB[ROOK]), rto);
-  set_bit(&(byTypeBB[0]), rto); // HACK: byTypeBB[0] == occupied squares
+  // Move the pieces
+  Bitboard kmove_bb = make_move_bb(kfrom, kto);
+  do_move_bb(&(byColorBB[us]), kmove_bb);
+  do_move_bb(&(byTypeBB[KING]), kmove_bb);
+  do_move_bb(&(byTypeBB[0]), kmove_bb); // HACK: byTypeBB[0] == occupied squares
+
+  Bitboard rmove_bb = make_move_bb(rfrom, rto);
+  do_move_bb(&(byColorBB[us]), rmove_bb);
+  do_move_bb(&(byTypeBB[ROOK]), rmove_bb);
+  do_move_bb(&(byTypeBB[0]), rmove_bb); // HACK: byTypeBB[0] == occupied squares
 
   // Update board array
   Piece king = piece_of_color_and_type(us, KING);
@@ -1242,21 +1237,16 @@ void Position::undo_castle_move(Move m) {
   assert(piece_on(kto) == piece_of_color_and_type(us, KING));
   assert(piece_on(rto) == piece_of_color_and_type(us, ROOK));
 
-  // Remove pieces from destination squares
-  clear_bit(&(byColorBB[us]), kto);
-  clear_bit(&(byTypeBB[KING]), kto);
-  clear_bit(&(byTypeBB[0]), kto); // HACK: byTypeBB[0] == occupied squares
-  clear_bit(&(byColorBB[us]), rto);
-  clear_bit(&(byTypeBB[ROOK]), rto);
-  clear_bit(&(byTypeBB[0]), rto); // HACK: byTypeBB[0] == occupied squares
-
-  // Put pieces on source squares
-  set_bit(&(byColorBB[us]), kfrom);
-  set_bit(&(byTypeBB[KING]), kfrom);
-  set_bit(&(byTypeBB[0]), kfrom); // HACK: byTypeBB[0] == occupied squares
-  set_bit(&(byColorBB[us]), rfrom);
-  set_bit(&(byTypeBB[ROOK]), rfrom);
-  set_bit(&(byTypeBB[0]), rfrom); // HACK: byTypeBB[0] == occupied squares
+  // Put the pieces back at the source square
+  Bitboard kmove_bb = make_move_bb(kto, kfrom);
+  do_move_bb(&(byColorBB[us]), kmove_bb);
+  do_move_bb(&(byTypeBB[KING]), kmove_bb);
+  do_move_bb(&(byTypeBB[0]), kmove_bb); // HACK: byTypeBB[0] == occupied squares
+
+  Bitboard rmove_bb = make_move_bb(rto, rfrom);
+  do_move_bb(&(byColorBB[us]), rmove_bb);
+  do_move_bb(&(byTypeBB[ROOK]), rmove_bb);
+  do_move_bb(&(byTypeBB[0]), rmove_bb); // HACK: byTypeBB[0] == occupied squares
 
   // Update board
   board[rto] = board[kto] = EMPTY;
@@ -1474,6 +1464,22 @@ int Position::see(Move m) const {
   return see(move_from(m), move_to(m));
 }
 
+int Position::see_sign(Move m) const {
+
+  assert(move_is_ok(m));
+
+  Square from = move_from(m);
+  Square to = move_to(m);
+
+  // Early return if SEE cannot be negative because capturing piece value
+  // is not bigger then captured one.
+  if (   midgame_value_of_piece_on(from) <= midgame_value_of_piece_on(to)
+      && type_of_piece_on(from) != KING)
+         return 1;
+
+  return see(from, to);
+}
+
 int Position::see(Square from, Square to) const {
 
   // Material values