Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 99 | pmbaty | 1 | /* |
| 2 | Texel - A UCI chess engine. |
||
| 3 | Copyright (C) 2012-2014 Peter Ă–sterlund, peterosterlund2@gmail.com |
||
| 4 | |||
| 5 | This program is free software: you can redistribute it and/or modify |
||
| 6 | it under the terms of the GNU General Public License as published by |
||
| 7 | the Free Software Foundation, either version 3 of the License, or |
||
| 8 | (at your option) any later version. |
||
| 9 | |||
| 10 | This program is distributed in the hope that it will be useful, |
||
| 11 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
||
| 12 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||
| 13 | GNU General Public License for more details. |
||
| 14 | |||
| 15 | You should have received a copy of the GNU General Public License |
||
| 16 | along with this program. If not, see <http://www.gnu.org/licenses/>. |
||
| 17 | */ |
||
| 18 | |||
| 19 | /* |
||
| 20 | * enginecontrol.hpp |
||
| 21 | * |
||
| 22 | * Created on: Mar 4, 2012 |
||
| 23 | * Author: petero |
||
| 24 | */ |
||
| 25 | |||
| 26 | #ifndef ENGINECONTROL_HPP_ |
||
| 27 | #define ENGINECONTROL_HPP_ |
||
| 28 | |||
| 29 | #include "search.hpp" |
||
| 30 | #include "transpositionTable.hpp" |
||
| 31 | #include "position.hpp" |
||
| 32 | #include "move.hpp" |
||
| 33 | |||
| 34 | #include <vector> |
||
| 35 | #include <iosfwd> |
||
| 36 | #include <thread> |
||
| 37 | #include <mutex> |
||
| 38 | #include <memory> |
||
| 39 | #include <atomic> |
||
| 40 | |||
| 41 | class SearchParams; |
||
| 42 | |||
| 43 | |||
| 44 | /** |
||
| 45 | * Control the search thread. |
||
| 46 | */ |
||
| 47 | class EngineControl { |
||
| 48 | public: |
||
| 49 | EngineControl(std::ostream& o); |
||
| 50 | ~EngineControl(); |
||
| 51 | |||
| 52 | void startSearch(const Position& pos, const std::vector<Move>& moves, const SearchParams& sPar); |
||
| 53 | |||
| 54 | void startPonder(const Position& pos, const std::vector<Move>& moves, const SearchParams& sPar); |
||
| 55 | |||
| 56 | void ponderHit(); |
||
| 57 | |||
| 58 | void stopSearch(); |
||
| 59 | |||
| 60 | void newGame(); |
||
| 61 | |||
| 62 | /** |
||
| 63 | * Compute thinking time for current search. |
||
| 64 | */ |
||
| 65 | void computeTimeLimit(const SearchParams& sPar); |
||
| 66 | |||
| 67 | static void printOptions(std::ostream& os); |
||
| 68 | |||
| 69 | void setOption(const std::string& optionName, const std::string& optionValue, |
||
| 70 | bool deferIfBusy); |
||
| 71 | |||
| 72 | private: |
||
| 73 | /** |
||
| 74 | * This class is responsible for sending "info" strings during search. |
||
| 75 | */ |
||
| 76 | class SearchListener : public Search::Listener { |
||
| 77 | public: |
||
| 78 | SearchListener(std::ostream& os0); |
||
| 79 | |||
| 80 | void notifyDepth(int depth) override; |
||
| 81 | |||
| 82 | void notifyCurrMove(const Move& m, int moveNr) override; |
||
| 83 | |||
| 84 | void notifyPV(int depth, int score, int time, U64 nodes, int nps, bool isMate, |
||
| 85 | bool upperBound, bool lowerBound, const std::vector<Move>& pv, |
||
| 86 | int multiPVIndex, U64 tbHits) override; |
||
| 87 | |||
| 88 | void notifyStats(U64 nodes, int nps, U64 tbHits, int time) override; |
||
| 89 | |||
| 90 | private: |
||
| 91 | std::ostream& os; |
||
| 92 | }; |
||
| 93 | |||
| 94 | void startThread(int minTimeLimit, int maxTimeLimit, int maxDepth, int maxNodes); |
||
| 95 | |||
| 96 | void stopThread(); |
||
| 97 | |||
| 98 | void setupTT(); |
||
| 99 | |||
| 100 | void setupPosition(Position pos, const std::vector<Move>& moves); |
||
| 101 | |||
| 102 | /** |
||
| 103 | * Try to find a move to ponder from the transposition table. |
||
| 104 | */ |
||
| 105 | Move getPonderMove(Position pos, const Move& m); |
||
| 106 | |||
| 107 | static std::string moveToString(const Move& m); |
||
| 108 | |||
| 109 | |||
| 110 | std::ostream& os; |
||
| 111 | |||
| 112 | int hashParListenerId; |
||
| 113 | int clearHashParListenerId; |
||
| 114 | std::map<std::string, std::string> pendingOptions; |
||
| 115 | |||
| 116 | std::shared_ptr<std::thread> engineThread; |
||
| 117 | std::mutex threadMutex; |
||
| 118 | std::atomic<bool> shouldDetach; |
||
| 119 | std::shared_ptr<Search> sc; |
||
| 120 | TranspositionTable tt; |
||
| 121 | ParallelData pd; |
||
| 122 | KillerTable kt; |
||
| 123 | History ht; |
||
| 124 | std::shared_ptr<Evaluate::EvalHashTables> et; |
||
| 125 | TreeLogger treeLog; |
||
| 126 | |||
| 127 | Position pos; |
||
| 128 | std::vector<U64> posHashList; |
||
| 129 | int posHashListSize; |
||
| 130 | std::atomic<bool> ponder; // True if currently doing pondering |
||
| 131 | bool onePossibleMove; |
||
| 132 | std::atomic<bool> infinite; |
||
| 133 | |||
| 134 | int minTimeLimit; |
||
| 135 | int maxTimeLimit; |
||
| 136 | int maxDepth; |
||
| 137 | int maxNodes; |
||
| 138 | std::vector<Move> searchMoves; |
||
| 139 | |||
| 140 | // Random seed for reduced strength |
||
| 141 | U64 randomSeed; |
||
| 142 | }; |
||
| 143 | |||
| 144 | |||
| 145 | #endif /* ENGINECONTROL_HPP_ */ |