]> git.sesse.net Git - stockfish/blobdiff - src/move.cpp
Coding style in move.cpp
[stockfish] / src / move.cpp
index 7b06b0ecaf0e33a120a709dbe8ea4975e44d7bdd..41d1603771a7b4d7d29daa019c23667deff99272 100644 (file)
@@ -1,7 +1,7 @@
 /*
   Stockfish, a UCI chess playing engine derived from Glaurung 2.1
   Copyright (C) 2004-2008 Tord Romstad (Glaurung author)
-  Copyright (C) 2008-2010 Marco Costalba, Joona Kiiski, Tord Romstad
+  Copyright (C) 2008-2012 Marco Costalba, Joona Kiiski, Tord Romstad
 
   Stockfish is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-
-////
-//// Includes
-////
-
 #include <cassert>
+#include <string>
 
-#include "move.h"
-#include "piece.h"
+#include "movegen.h"
 #include "position.h"
 
+using std::string;
 
-////
-//// Functions
-////
+/// move_to_uci() converts a move to a string in coordinate notation
+/// (g1f3, a7a8q, etc.). The only special case is castling moves, where we print
+/// in the e1g1 notation in normal chess mode, and in e1h1 notation in chess960
+/// mode. Internally castle moves are always coded as "king captures rook".
 
-/// move_from_string() takes a position and a string as input, and attempts to
-/// convert the string to a move, using simple coordinate notation (g1f3,
-/// a7a8q, etc.). In order to correctly parse en passant captures and castling
-/// moves, we need the position. This function is not robust, and expects that
-/// the input move is legal and correctly formatted.
+const string move_to_uci(Move m, bool chess960) {
 
-Move move_from_string(const Position& pos, const std::string& str) {
+  Square from = from_sq(m);
+  Square to = to_sq(m);
+  string promotion;
 
-  Square from, to;
-  Piece piece;
-  Color us = pos.side_to_move();
+  if (m == MOVE_NONE)
+      return "(none)";
 
-  if (str.length() < 4)
-      return MOVE_NONE;
+  if (m == MOVE_NULL)
+      return "0000";
 
-  // Read the from and to squares
-  from = square_from_string(str.substr(0, 2));
-  to = square_from_string(str.substr(2, 4));
+  if (is_castle(m) && !chess960)
+      to = make_square(to > from ? FILE_G : FILE_C, rank_of(from));
 
-  // Find the moving piece
-  piece = pos.piece_on(from);
+  if (is_promotion(m))
+      promotion = char(tolower(piece_type_to_char(promotion_type(m))));
 
-  // If the string has more than 4 characters, try to interpret the 5th
-  // character as a promotion
-  if (type_of_piece(piece) == PAWN && str.length() > 4)
-  {
-      switch (tolower(str[4])) {
-      case 'n':
-          return make_promotion_move(from, to, KNIGHT);
-      case 'b':
-          return make_promotion_move(from, to, BISHOP);
-      case 'r':
-          return make_promotion_move(from, to, ROOK);
-      case 'q':
-          return make_promotion_move(from, to, QUEEN);
-    }
-  }
+  return square_to_string(from) + square_to_string(to) + promotion;
+}
 
-  if (piece == piece_of_color_and_type(us, KING))
-  {
-      // Is this a castling move? A king move is assumed to be a castling
-      // move if the destination square is occupied by a friendly rook, or
-      // if the distance between the source and destination squares is more
-      // than 1.
-      if (pos.piece_on(to) == piece_of_color_and_type(us, ROOK))
-          return make_castle_move(from, to);
-
-      else if (square_distance(from, to) > 1)
-      {
-          // This is a castling move, but we have to translate it to the
-          // internal "king captures rook" representation.
-          SquareDelta delta = (to > from ? DELTA_E : DELTA_W);
-          Square s = from + delta;
-          while (relative_rank(us, s) == RANK_1 && pos.piece_on(s) != piece_of_color_and_type(us, ROOK))
-              s += delta;
-
-          return (relative_rank(us, s) == RANK_1 ? make_castle_move(from, s) : MOVE_NONE);
-      }
-  }
-  else if (piece == piece_of_color_and_type(us, PAWN))
-  {
-      // En passant move? We assume that a pawn move is an en passant move
-      // without further testing if the destination square is epSquare.
-      if (to == pos.ep_square())
-          return make_ep_move(from, to);
-  }
-  return make_move(from, to);
+
+/// move_from_uci() takes a position and a string representing a move in
+/// simple coordinate notation and returns an equivalent legal Move if any.
+
+Move move_from_uci(const Position& pos, string& str) {
+
+  if (str.length() == 5) // Junior could send promotion piece in uppercase
+      str[4] = char(tolower(str[4]));
+
+  for (MoveList<MV_LEGAL> ml(pos); !ml.end(); ++ml)
+      if (str == move_to_uci(ml.move(), pos.is_chess960()))
+          return ml.move();
+
+  return MOVE_NONE;
 }
 
 
-/// move_to_string() converts a move to a string in coordinate notation
-/// (g1f3, a7a8q, etc.).  The only special case is castling moves, where we
-/// print in the e1g1 notation in normal chess mode, and in e1h1 notation in
-/// Chess960 mode.
+/// move_to_san() takes a position and a legal Move as input and returns its
+/// short algebraic notation representation.
+
+const string move_to_san(Position& pos, Move m) {
+
+  if (m == MOVE_NONE)
+      return "(none)";
+
+  if (m == MOVE_NULL)
+      return "(null)";
 
-const std::string move_to_string(Move move) {
+  assert(pos.move_is_legal(m));
 
-  std::string str;
-  Square from = move_from(move);
-  Square to = move_to(move);
+  Bitboard attackers;
+  bool ambiguousMove, ambiguousFile, ambiguousRank;
+  string san;
+  Square from = from_sq(m);
+  Square to = to_sq(m);
+  PieceType pt = type_of(pos.piece_on(from));
 
-  if (move == MOVE_NONE)
-      str = "(none)";
-  else if (move == MOVE_NULL)
-      str = "0000";
+  if (is_castle(m))
+      san = to_sq(m) < from_sq(m) ? "O-O-O" : "O-O";
   else
   {
-      if (!Chess960)
+      if (pt != PAWN)
       {
-          if (move_is_short_castle(move))
-              return (from == SQ_E1 ? "e1g1" : "e8g8");
-
-          if (move_is_long_castle(move))
-              return (from == SQ_E1 ? "e1c1" : "e8c8");
+          san = piece_type_to_char(pt);
+
+          // Disambiguation if we have more then one piece with destination 'to'
+          // note that for pawns is not needed because starting file is explicit.
+          attackers = pos.attackers_to(to) & pos.pieces(pos.side_to_move(), pt);
+          attackers ^= from;
+          ambiguousMove = ambiguousFile = ambiguousRank = false;
+
+          while (attackers)
+          {
+              Square sq = pop_1st_bit(&attackers);
+
+              // Pinned pieces are not included in the possible sub-set
+              if (!pos.pl_move_is_legal(make_move(sq, to), pos.pinned_pieces()))
+                  continue;
+
+              ambiguousFile |= file_of(sq) == file_of(from);
+              ambiguousRank |= rank_of(sq) == rank_of(from);
+              ambiguousMove = true;
+          }
+
+          if (ambiguousMove)
+          {
+              if (!ambiguousFile)
+                  san += file_to_char(file_of(from));
+
+              else if (!ambiguousRank)
+                  san += rank_to_char(rank_of(from));
+
+              else
+                  san += square_to_string(from);
+          }
       }
-      str = square_to_string(from) + square_to_string(to);
-      if (move_is_promotion(move))
-          str += piece_type_to_char(move_promotion_piece(move), false);
-  }
-  return str;
-}
-
 
-/// Overload the << operator, to make it easier to print moves.
-
-std::ostream &operator << (std::ostream& os, Move m) {
+      if (pos.is_capture(m))
+      {
+          if (pt == PAWN)
+              san += file_to_char(file_of(from));
 
-  return os << move_to_string(m);
-}
+          san += 'x';
+      }
 
+      san += square_to_string(to);
 
-/// move_is_ok(), for debugging.
+      if (is_promotion(m))
+          san += string("=") + piece_type_to_char(promotion_type(m));
+  }
 
-bool move_is_ok(Move m) {
+  if (pos.move_gives_check(m, CheckInfo(pos)))
+  {
+      StateInfo st;
+      pos.do_move(m, st);
+      san += MoveList<MV_LEGAL>(pos).size() ? "+" : "#";
+      pos.undo_move(m);
+  }
 
-  return square_is_ok(move_from(m)) && square_is_ok(move_to(m));
+  return san;
 }