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