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
  24 ////
  25 //// Includes
  26 ////
  27
  28 #include <iostream>
  29 #include <string>
  30
  31 #include "bitboard.h"
  32 #include "bitcount.h"
  33 #include "endgame.h"
  34 #include "evaluate.h"
  35 #include "material.h"
  36 #include "misc.h"
  37 #include "position.h"
  38 #include "search.h"
  39 #include "thread.h"
  40 #include "ucioption.h"
  41
  42 #ifdef USE_CALLGRIND
  43 #include <valgrind/callgrind.h>
  44 #endif
  45
  46 using namespace std;
  47
  48 extern bool execute_uci_command(const string& cmd);
  49 extern void benchmark(int argc, char* argv[]);
  50
  51 ////
  52 //// Functions
  53 ////
  54
  55 int main(int argc, char* argv[]) {
  56
  57   // Disable IO buffering
  58   cout.rdbuf()->pubsetbuf(NULL, 0);
  59   cin.rdbuf()->pubsetbuf(NULL, 0);
  60
  61   // Startup initializations
  62   init_bitboards();
  63   init_uci_options();
  64   Position::init_zobrist();
  65   Position::init_piece_square_tables();
  66   init_eval(1);
  67   init_bitbases();
  68   init_search();
  69   init_threads();
  70
  71 #ifdef USE_CALLGRIND
  72   CALLGRIND_START_INSTRUMENTATION;
  73 #endif
  74
  75   if (argc <= 1)
  76   {
  77       // Print copyright notice
  78       cout << engine_name()
  79            << " by Tord Romstad, Marco Costalba, Joona Kiiski" << endl;
  80
  81       if (CpuHasPOPCNT)
  82           cout << "Good! CPU has hardware POPCNT." << endl;
  83
  84       // Wait for a command from the user, and passes this command to
  85       // execute_uci_command() and also intercepts EOF from stdin, by
  86       // translating EOF to the "quit" command. This ensures that we
  87       // exit gracefully if the GUI dies unexpectedly.
  88       string cmd;
  89
  90       do {
  91           // Wait for a command from stdin
  92           if (!getline(cin, cmd))
  93               cmd = "quit";
  94
  95       } while (execute_uci_command(cmd));
  96   }
  97   else // Process command line arguments
  98   {
  99       if (string(argv[1]) != "bench" || argc > 7)
 100           cout << "Usage: stockfish bench [hash size = 128] [threads = 1] "
 101                << "[limit = 12] [fen positions file = default] "
 102                << "[depth, time, perft or node limited = depth]" << endl;
 103       else
 104           benchmark(argc, argv);
 105   }
 106
 107   exit_threads();
 108   quit_eval();
 109   return 0;
 110 }