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 "direction.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 void uci_main_loop();
  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
  59   cout.rdbuf()->pubsetbuf(NULL, 0);
  60   cin.rdbuf()->pubsetbuf(NULL, 0);
  61
  62   // Startup initializations
  63   init_direction_table();
  64   init_bitboards();
  65   init_uci_options();
  66   Position::init_zobrist();
  67   Position::init_piece_square_tables();
  68   init_eval(1);
  69   init_bitbases();
  70   init_search();
  71   init_threads();
  72
  73 #ifdef USE_CALLGRIND
  74   CALLGRIND_START_INSTRUMENTATION;
  75 #endif
  76
  77   if (argc <= 1)
  78   {
  79       // Print copyright notice
  80       cout << engine_name()
  81            << " by Tord Romstad, Marco Costalba, Joona Kiiski" << endl;
  82
  83       if (CpuHasPOPCNT)
  84           cout << "Good! CPU has hardware POPCNT." << endl;
  85
  86       // Enter UCI mode
  87       uci_main_loop();
  88   }
  89   else // Process command line arguments
  90   {
  91       if (string(argv[1]) != "bench" || argc > 7)
  92           cout << "Usage: stockfish bench [hash size = 128] [threads = 1] "
  93                << "[limit = 12] [fen positions file = default] "
  94                << "[depth, time, perft or node limited = depth]" << endl;
  95       else
  96           benchmark(argc, argv);
  97   }
  98
  99   exit_threads();
 100   quit_eval();
 101   return 0;
 102 }