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_direction_table();
  63   init_bitboards();
  64   init_uci_options();
  65   Position::init_zobrist();
  66   Position::init_piece_square_tables();
  67   init_eval(1);
  68   init_bitbases();
  69   init_search();
  70   init_threads();
  71
  72 #ifdef USE_CALLGRIND
  73   CALLGRIND_START_INSTRUMENTATION;
  74 #endif
  75
  76   if (argc <= 1)
  77   {
  78       // Print copyright notice
  79       cout << engine_name()
  80            << " by Tord Romstad, Marco Costalba, Joona Kiiski" << endl;
  81
  82       if (CpuHasPOPCNT)
  83           cout << "Good! CPU has hardware POPCNT." << endl;
  84
  85       // Wait for a command from the user, and passes this command to
  86       // execute_uci_command() and also intercepts EOF from stdin, by
  87       // translating EOF to the "quit" command. This ensures that we
  88       // exit gracefully if the GUI dies unexpectedly.
  89       string cmd;
  90
  91       do {
  92           // Wait for a command from stdin
  93           if (!getline(cin, cmd))
  94               cmd = "quit";
  95
  96       } while (execute_uci_command(cmd));
  97   }
  98   else // Process command line arguments
  99   {
 100       if (string(argv[1]) != "bench" || argc > 7)
 101           cout << "Usage: stockfish bench [hash size = 128] [threads = 1] "
 102                << "[limit = 12] [fen positions file = default] "
 103                << "[depth, time, perft or node limited = depth]" << endl;
 104       else
 105           benchmark(argc, argv);
 106   }
 107
 108   exit_threads();
 109   quit_eval();
 110   return 0;
 111 }