X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fnotation.cpp;h=fb65f6f5fa9f2bbb42275f2d4c55a9ee203bf4e6;hp=8ea6ad52aea0694ee5cf87808f444221e33837f1;hb=55bd27b8f08a151128d7065fa2819aa3e9605299;hpb=7b4aa1070882cba085c9d6ed03d934734c395fe5 diff --git a/src/notation.cpp b/src/notation.cpp index 8ea6ad52..fb65f6f5 100644 --- a/src/notation.cpp +++ b/src/notation.cpp @@ -18,12 +18,40 @@ */ #include +#include +#include #include #include "movegen.h" +#include "notation.h" #include "position.h" -using std::string; +using namespace std; + +static const char* PieceToChar = " PNBRQK pnbrqk"; + + +/// score_to_uci() converts a value to a string suitable for use with the UCI +/// protocol specifications: +/// +/// cp The score from the engine's point of view in centipawns. +/// mate Mate in y moves, not plies. If the engine is getting mated +/// use negative values for y. + +string score_to_uci(Value v, Value alpha, Value beta) { + + stringstream s; + + if (abs(v) < VALUE_MATE_IN_MAX_PLY) + s << "cp " << v * 100 / int(PawnValueMg); + else + s << "mate " << (v > 0 ? VALUE_MATE - v + 1 : -VALUE_MATE - v) / 2; + + s << (v >= beta ? " lowerbound" : v <= alpha ? " upperbound" : ""); + + return s.str(); +} + /// 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 @@ -34,7 +62,6 @@ const string move_to_uci(Move m, bool chess960) { Square from = from_sq(m); Square to = to_sq(m); - string promotion; if (m == MOVE_NONE) return "(none)"; @@ -42,13 +69,15 @@ const string move_to_uci(Move m, bool chess960) { if (m == MOVE_NULL) return "0000"; - if (is_castle(m) && !chess960) + if (type_of(m) == CASTLE && !chess960) to = (to > from ? FILE_G : FILE_C) | rank_of(from); - if (is_promotion(m)) - promotion = char(tolower(piece_type_to_char(promotion_type(m)))); + string move = square_to_string(from) + square_to_string(to); - return square_to_string(from) + square_to_string(to) + promotion; + if (type_of(m) == PROMOTION) + move += PieceToChar[make_piece(BLACK, promotion_type(m))]; // Lower case + + return move; } @@ -60,7 +89,7 @@ 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 ml(pos); !ml.end(); ++ml) + for (MoveList ml(pos); !ml.end(); ++ml) if (str == move_to_uci(ml.move(), pos.is_chess960())) return ml.move(); @@ -84,27 +113,29 @@ const string move_to_san(Position& pos, Move m) { Bitboard attackers; bool ambiguousMove, ambiguousFile, ambiguousRank; string san; + Color us = pos.side_to_move(); Square from = from_sq(m); Square to = to_sq(m); - PieceType pt = type_of(pos.piece_on(from)); + Piece pc = pos.piece_on(from); + PieceType pt = type_of(pc); - if (is_castle(m)) + if (type_of(m) == CASTLE) san = to > from ? "O-O" : "O-O-O"; else { if (pt != PAWN) { - san = piece_type_to_char(pt); + san = PieceToChar[pt]; // Upper case // 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; + attackers = (pos.attacks_from(pc, to) & pos.pieces(us, pt)) ^ from; + while (attackers) { - Square sq = pop_1st_bit(&attackers); + Square sq = pop_lsb(&attackers); // Pinned pieces are not included in the possible sub-set if (!pos.pl_move_is_legal(make_move(sq, to), pos.pinned_pieces())) @@ -127,28 +158,114 @@ const string move_to_san(Position& pos, Move m) { san += square_to_string(from); } } + else if (pos.is_capture(m)) + san = file_to_char(file_of(from)); if (pos.is_capture(m)) - { - if (pt == PAWN) - san += file_to_char(file_of(from)); - san += 'x'; - } san += square_to_string(to); - if (is_promotion(m)) - san += string("=") + piece_type_to_char(promotion_type(m)); + if (type_of(m) == PROMOTION) + san += string("=") + PieceToChar[promotion_type(m)]; } if (pos.move_gives_check(m, CheckInfo(pos))) { StateInfo st; pos.do_move(m, st); - san += MoveList(pos).size() ? "+" : "#"; + san += MoveList(pos).size() ? "+" : "#"; pos.undo_move(m); } return san; } + + +/// pretty_pv() formats human-readable search information, typically to be +/// appended to the search log file. It uses the two helpers below to pretty +/// format time and score respectively. + +static string time_to_string(int64_t msecs) { + + const int MSecMinute = 1000 * 60; + const int MSecHour = 1000 * 60 * 60; + + int64_t hours = msecs / MSecHour; + int64_t minutes = (msecs % MSecHour) / MSecMinute; + int64_t seconds = ((msecs % MSecHour) % MSecMinute) / 1000; + + stringstream s; + + if (hours) + s << hours << ':'; + + s << setfill('0') << setw(2) << minutes << ':' << setw(2) << seconds; + + return s.str(); +} + +static string score_to_string(Value v) { + + stringstream s; + + if (v >= VALUE_MATE_IN_MAX_PLY) + s << "#" << (VALUE_MATE - v + 1) / 2; + + else if (v <= VALUE_MATED_IN_MAX_PLY) + s << "-#" << (VALUE_MATE + v) / 2; + + else + s << setprecision(2) << fixed << showpos << float(v) / PawnValueMg; + + return s.str(); +} + +string pretty_pv(Position& pos, int depth, Value value, int64_t msecs, Move pv[]) { + + const int64_t K = 1000; + const int64_t M = 1000000; + + StateInfo state[MAX_PLY_PLUS_2], *st = state; + Move* m = pv; + string san, padding; + size_t length; + stringstream s; + + s << setw(2) << depth + << setw(8) << score_to_string(value) + << setw(8) << time_to_string(msecs); + + if (pos.nodes_searched() < M) + s << setw(8) << pos.nodes_searched() / 1 << " "; + + else if (pos.nodes_searched() < K * M) + s << setw(7) << pos.nodes_searched() / K << "K "; + + else + s << setw(7) << pos.nodes_searched() / M << "M "; + + padding = string(s.str().length(), ' '); + length = padding.length(); + + while (*m != MOVE_NONE) + { + san = move_to_san(pos, *m); + + if (length + san.length() > 80) + { + s << "\n" + padding; + length = padding.length(); + } + + s << san << ' '; + length += san.length() + 1; + + pos.do_move(*m++, *st++); + } + + while (m != pv) + pos.undo_move(*--m); + + return s.str(); +}