X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fmain.cpp;h=d75350b17f7ea1f43413754318c6754c73cf48ae;hp=b48706060001daf2765e522269561856cf69e585;hb=927f1b0bd30a5b2cfdcdf163f26f528738509064;hpb=9fc602bae74b8e09bd45ace3b42a8ce84d56b23c diff --git a/src/main.cpp b/src/main.cpp index b4870606..d75350b1 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -20,18 +20,15 @@ // To profile with callgrind uncomment following line //#define USE_CALLGRIND - -//// -//// Includes -//// - +#include #include #include -#include "benchmark.h" -#include "bitcount.h" -#include "misc.h" -#include "uci.h" +#include "bitboard.h" +#include "evaluate.h" +#include "position.h" +#include "thread.h" +#include "ucioption.h" #ifdef USE_CALLGRIND #include @@ -39,51 +36,53 @@ using namespace std; +extern bool execute_uci_command(const string& cmd); +extern void benchmark(int argc, char* argv[]); +extern void init_kpk_bitbase(); -//// -//// Functions -//// - -int main(int argc, char *argv[]) { +int main(int argc, char* argv[]) { - // Disable IO buffering + // Disable IO buffering for C and C++ standard libraries + setvbuf(stdin, NULL, _IONBF, 0); + setvbuf(stdout, NULL, _IONBF, 0); cout.rdbuf()->pubsetbuf(NULL, 0); cin.rdbuf()->pubsetbuf(NULL, 0); - // Initialization through global resources manager - Application::initialize(); + // Startup initializations + init_bitboards(); + init_uci_options(); + Position::init_zobrist(); + Position::init_piece_square_tables(); + init_eval(1); + init_kpk_bitbase(); + init_threads(); #ifdef USE_CALLGRIND CALLGRIND_START_INSTRUMENTATION; #endif - // Process command line arguments if any - if (argc > 1) + if (argc < 2) { - if (string(argv[1]) != "bench" || argc < 4 || argc > 8) - cout << "Usage: stockfish bench " - << "[time = 60s] [fen positions file = default] " - << "[time, depth, perft or node limited = time] " - << "[timing file name = none]" << endl; - else - { - string time = argc > 4 ? argv[4] : "60"; - string fen = argc > 5 ? argv[5] : "default"; - string lim = argc > 6 ? argv[6] : "time"; - string tim = argc > 7 ? argv[7] : ""; - benchmark(string(argv[2]) + " " + string(argv[3]) + " " + time + " " + fen + " " + lim + " " + tim); - } - return 0; - } - - // Print copyright notice - cout << engine_name() - << ". By Tord Romstad, Marco Costalba, Joona Kiiski." << endl; + // Print copyright notice + cout << engine_name() << " by " << engine_author() << endl; - if (CpuHasPOPCNT) - cout << "Good! CPU has hardware POPCNT. We will use it." << endl; + if (CpuHasPOPCNT) + cout << "Good! CPU has hardware POPCNT." << endl; - // Enter UCI mode - uci_main_loop(); + // Wait for a command from the user, and passes this command to + // execute_uci_command() and also intercepts EOF from stdin to + // ensure that we exit gracefully if the GUI dies unexpectedly. + string cmd; + while (getline(cin, cmd) && execute_uci_command(cmd)) {} + } + else if (string(argv[1]) == "bench" && argc < 8) + benchmark(argc, argv); + else + cout << "Usage: stockfish bench [hash size = 128] [threads = 1] " + << "[limit = 12] [fen positions file = default] " + << "[depth, time, perft or node limited = depth]" << endl; + + exit_threads(); + quit_eval(); return 0; }