/*
 
  Stockfish, a UCI chess playing engine derived from Glaurung 2.1
 
  Copyright (c) 2013 Ronald de Man
 
  Copyright (C) 2016-2018 Marco Costalba, Lucas Braesch
 
 
 
  Stockfish is free software: you can redistribute it and/or modify
 
  it under the terms of the GNU General Public License as published by
 
  the Free Software Foundation, either version 3 of the License, or
 
  (at your option) any later version.
 
 
 
  Stockfish is distributed in the hope that it will be useful,
 
  but WITHOUT ANY WARRANTY; without even the implied warranty of
 
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
  GNU General Public License for more details.
 
 
 
  You should have received a copy of the GNU General Public License
 
  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
*/
 
 
 
#ifndef TBPROBE_H
 
#define TBPROBE_H
 
 
 
#include <ostream>
 
 
 
#include "../search.h"
 
 
 
namespace Tablebases {
 
 
 
enum WDLScore {
 
    WDLLoss        = -2, // Loss
 
    WDLBlessedLoss = -1, // Loss, but draw under 50-move rule
 
    WDLDraw        =  0, // Draw
 
    WDLCursedWin   =  1, // Win, but draw under 50-move rule
 
    WDLWin         =  2, // Win
 
 
 
    WDLScoreNone  = -1000
 
};
 
 
 
// Possible states after a probing operation
 
enum ProbeState {
 
    FAIL              =  0, // Probe failed (missing file table)
 
    OK                =  1, // Probe succesful
 
    CHANGE_STM        = -1, // DTZ should check the other side
 
    ZEROING_BEST_MOVE =  2  // Best move zeroes DTZ (capture or pawn move)
 
};
 
 
 
extern int MaxCardinality;
 
 
 
void init(const std::string& paths);
 
WDLScore probe_wdl(Position& pos, ProbeState* result);
 
int probe_dtz(Position& pos, ProbeState* result);
 
bool root_probe(Position& pos, Search::RootMoves& rootMoves, Value& score);
 
bool root_probe_wdl(Position& pos, Search::RootMoves& rootMoves, Value& score);
 
void filter_root_moves(Position& pos, Search::RootMoves& rootMoves);
 
 
 
inline std::ostream& operator<<(std::ostream& os, const WDLScore v) {
 
 
 
    os << (v == WDLLoss        ? "Loss" :
 
           v == WDLBlessedLoss ? "Blessed loss" :
 
           v == WDLDraw        ? "Draw" :
 
           v == WDLCursedWin   ? "Cursed win" :
 
           v == WDLWin         ? "Win" : "None");
 
 
 
    return os;
 
}
 
 
 
inline std::ostream& operator<<(std::ostream& os, const ProbeState v) {
 
 
 
    os << (v == FAIL              ? "Failed" :
 
           v == OK                ? "Success" :
 
           v == CHANGE_STM        ? "Probed opponent side" :
 
           v == ZEROING_BEST_MOVE ? "Best move zeroes DTZ" : "None");
 
 
 
    return os;
 
}
 
 
 
}
 
 
 
#endif