X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fbenchmark.cpp;h=67902062f6cfa9b6e206cf91fe97885583fa76cd;hp=136c6e80f5bc9ee7f4d86683194fc553e29f8c2d;hb=676b2c8435502409145491f2f86454f11f99dae8;hpb=cc2b3ece5c5d8d1183f8526fbb0ee4e1ea7a69fe diff --git a/src/benchmark.cpp b/src/benchmark.cpp index 136c6e80..67902062 100644 --- a/src/benchmark.cpp +++ b/src/benchmark.cpp @@ -19,6 +19,7 @@ #include #include +#include #include #include "misc.h" @@ -57,33 +58,34 @@ static const char* Defaults[] = { /// format (defaults are the positions defined above) and the type of the /// limit value: depth (default), time in secs or number of nodes. -void benchmark(int argc, char* argv[]) { +void benchmark(istringstream& is) { - vector fens; + string token; Search::LimitsType limits; - int64_t nodes = 0; + vector fens(Defaults, Defaults + 16); // Assign default values to missing arguments - string ttSize = argc > 2 ? argv[2] : "128"; - string threads = argc > 3 ? argv[3] : "1"; - string valStr = argc > 4 ? argv[4] : "12"; - string fenFile = argc > 5 ? argv[5] : "default"; - string valType = argc > 6 ? argv[6] : "depth"; + string ttSize = (is >> token) ? token : "128"; + string threads = (is >> token) ? token : "1"; + string limit = (is >> token) ? token : "12"; + string fenFile = (is >> token) ? token : "default"; + string limitType = (is >> token) ? token : "depth"; Options["Hash"] = ttSize; Options["Threads"] = threads; - if (valType == "time") - limits.movetime = 1000 * atoi(valStr.c_str()); // movetime is in ms + if (limitType == "time") + limits.movetime = 1000 * atoi(limit.c_str()); // movetime is in ms - else if (valType == "nodes") - limits.nodes = atoi(valStr.c_str()); + else if (limitType == "nodes") + limits.nodes = atoi(limit.c_str()); else - limits.depth = atoi(valStr.c_str()); + limits.depth = atoi(limit.c_str()); if (fenFile != "default") { + fens.clear(); string fen; ifstream file(fenFile.c_str()); @@ -99,18 +101,17 @@ void benchmark(int argc, char* argv[]) { file.close(); } - else - fens.assign(Defaults, Defaults + 16); + int64_t nodes = 0; Time time = Time::current_time(); for (size_t i = 0; i < fens.size(); i++) { - Position pos(fens[i], false, 0); + Position pos(fens[i], false, Threads.main_thread()); cerr << "\nPosition: " << i + 1 << '/' << fens.size() << endl; - if (valType == "perft") + if (limitType == "perft") { int64_t cnt = Search::perft(pos, limits.depth * ONE_PLY); cerr << "\nPerft " << limits.depth << " leaf nodes: " << cnt << endl; @@ -118,7 +119,7 @@ void benchmark(int argc, char* argv[]) { } else { - Threads.start_searching(pos, limits); + Threads.start_searching(pos, limits, vector()); Threads.wait_for_search_finished(); nodes += Search::RootPosition.nodes_searched(); }