git.sesse.net Git - stockfish/blob - src/main.cpp

   1 /*
   2   Stockfish, a UCI chess playing engine derived from Glaurung 2.1
   3   Copyright (C) 2004-2008 Tord Romstad (Glaurung author)
   4   Copyright (C) 2008-2010 Marco Costalba, Joona Kiiski, Tord Romstad
   5
   6   Stockfish is free software: you can redistribute it and/or modify
   7   it under the terms of the GNU General Public License as published by
   8   the Free Software Foundation, either version 3 of the License, or
   9   (at your option) any later version.
  10
  11   Stockfish is distributed in the hope that it will be useful,
  12   but WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14   GNU General Public License for more details.
  15
  16   You should have received a copy of the GNU General Public License
  17   along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 // To profile with callgrind uncomment following line
  21 //#define USE_CALLGRIND
  22
  23 #include <cstdio>
  24 #include <iostream>
  25 #include <string>
  26
  27 #include "bitboard.h"
  28 #include "evaluate.h"
  29 #include "position.h"
  30 #include "thread.h"
  31 #include "search.h"
  32 #include "ucioption.h"
  33
  34 #ifdef USE_CALLGRIND
  35 #include <valgrind/callgrind.h>
  36 #endif
  37
  38 using namespace std;
  39
  40 extern bool execute_uci_command(const string& cmd);
  41 extern void benchmark(int argc, char* argv[]);
  42 extern void init_kpk_bitbase();
  43
  44 int main(int argc, char* argv[]) {
  45
  46   // Disable IO buffering for C and C++ standard libraries
  47   setvbuf(stdin, NULL, _IONBF, 0);
  48   setvbuf(stdout, NULL, _IONBF, 0);
  49   cout.rdbuf()->pubsetbuf(NULL, 0);
  50   cin.rdbuf()->pubsetbuf(NULL, 0);
  51
  52   // Startup initializations
  53   init_bitboards();
  54   Position::init_zobrist();
  55   Position::init_piece_square_tables();
  56   init_kpk_bitbase();
  57   init_search();
  58   Threads.init();
  59
  60 #ifdef USE_CALLGRIND
  61   CALLGRIND_START_INSTRUMENTATION;
  62 #endif
  63
  64   if (argc < 2)
  65   {
  66       // Print copyright notice
  67       cout << engine_name() << " by " << engine_authors() << endl;
  68
  69       if (CpuHasPOPCNT)
  70           cout << "Good! CPU has hardware POPCNT." << endl;
  71
  72       // Wait for a command from the user, and passes this command to
  73       // execute_uci_command() and also intercepts EOF from stdin to
  74       // ensure that we exit gracefully if the GUI dies unexpectedly.
  75       string cmd;
  76       while (getline(cin, cmd) && execute_uci_command(cmd)) {}
  77   }
  78   else if (string(argv[1]) == "bench" && argc < 8)
  79       benchmark(argc, argv);
  80   else
  81       cout << "Usage: stockfish bench [hash size = 128] [threads = 1] "
  82            << "[limit = 12] [fen positions file = default] "
  83            << "[limited by depth, time, nodes or perft = depth]" << endl;
  84
  85   Threads.exit();
  86   return 0;
  87 }