de77e5823eff1a756f249ae1119493d441890921
[stockfish] / 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 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
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       // 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 }