X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fmisc.cpp;h=608d772501fc54c70967d16e55ed80ba9cccb573;hp=d18f48d6e8a7c24a1c4eb7c679cb770e8af4717b;hb=55df3fa2d7631ed67e46f9433aa7f3a71c18e5e7;hpb=4fa0395eb8db33b23c2a91e4b255021fd940d811 diff --git a/src/misc.cpp b/src/misc.cpp index d18f48d6..608d7725 100644 --- a/src/misc.cpp +++ b/src/misc.cpp @@ -1,7 +1,7 @@ /* Stockfish, a UCI chess playing engine derived from Glaurung 2.1 Copyright (C) 2004-2008 Tord Romstad (Glaurung author) - Copyright (C) 2008-2010 Marco Costalba, Joona Kiiski, Tord Romstad + Copyright (C) 2008-2012 Marco Costalba, Joona Kiiski, Tord Romstad Stockfish is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -17,274 +17,231 @@ along with this program. If not, see . */ - -//// -//// Includes -//// - -#if !defined(_MSC_VER) - -# include -# include -# include -# if defined(__hpux) -# include -# endif - -#else - -#define _CRT_SECURE_NO_DEPRECATE -#include -#include - -#endif - -#if !defined(NO_PREFETCH) -# include -#endif - -#include -#include #include #include #include -#include "bitcount.h" #include "misc.h" #include "thread.h" +#if defined(__hpux) +# include +#endif + using namespace std; -/// Version number. If this is left empty, the current date (in the format -/// YYMMDD) is used as a version number. +/// Version number. If Version is left empty, then Tag plus current +/// date (in the format YYMMDD) is used as a version number. -static const string EngineVersion = ""; -static const string AppName = "Stockfish"; -static const string AppTag = ""; +static const string Version = ""; +static const string Tag = ""; -//// -//// Variables -//// +/// engine_info() returns the full name of the current Stockfish version. +/// This will be either "Stockfish YYMMDD" (where YYMMDD is the date when +/// the program was compiled) or "Stockfish ", depending +/// on whether Version is empty. -static uint64_t dbg_cnt0 = 0; -static uint64_t dbg_cnt1 = 0; +const string engine_info(bool to_uci) { -bool dbg_show_mean = false; -bool dbg_show_hit_rate = false; + const string months("Jan Feb Mar Apr May Jun Jul Aug Sep Oct Nov Dec"); + const string cpu64(Is64Bit ? " 64bit" : ""); + const string popcnt(HasPopCnt ? " SSE4.2" : ""); + string month, day, year; + stringstream s, date(__DATE__); // From compiler, format is "Sep 21 2008" -//// -//// Functions -//// + s << "Stockfish " << Version; -void dbg_hit_on(bool b) { + if (Version.empty()) + { + date >> month >> day >> year; - assert(!dbg_show_mean); - dbg_show_hit_rate = true; - dbg_cnt0++; - if (b) - dbg_cnt1++; -} + s << Tag << setfill('0') << " " << year.substr(2) + << setw(2) << (1 + months.find(month) / 4) << setw(2) << day; + } -void dbg_hit_on_c(bool c, bool b) { + s << cpu64 << popcnt << (to_uci ? "\nid author ": " by ") + << "Tord Romstad, Marco Costalba and Joona Kiiski"; - if (c) - dbg_hit_on(b); + return s.str(); } -void dbg_before() { - - assert(!dbg_show_mean); - dbg_show_hit_rate = true; - dbg_cnt0++; -} -void dbg_after() { +/// Convert system time to milliseconds. That's all we need. - assert(!dbg_show_mean); - dbg_show_hit_rate = true; - dbg_cnt1++; +Time::point Time::now() { + sys_time_t t; system_time(&t); return time_to_msec(t); } -void dbg_mean_of(int v) { - assert(!dbg_show_hit_rate); - dbg_show_mean = true; - dbg_cnt0++; - dbg_cnt1 += v; -} +/// Debug functions used mainly to collect run-time statistics -void dbg_print_hit_rate() { +static uint64_t hits[2], means[2]; - cout << "Total " << dbg_cnt0 << " Hit " << dbg_cnt1 - << " hit rate (%) " << (dbg_cnt1*100)/(dbg_cnt0 ? dbg_cnt0 : 1) << endl; -} +void dbg_hit_on(bool b) { hits[0]++; if (b) hits[1]++; } +void dbg_hit_on_c(bool c, bool b) { if (c) dbg_hit_on(b); } +void dbg_mean_of(int v) { means[0]++; means[1] += v; } + +void dbg_print() { -void dbg_print_mean() { + if (hits[0]) + cerr << "Total " << hits[0] << " Hits " << hits[1] + << " hit rate (%) " << 100 * hits[1] / hits[0] << endl; - cout << "Total " << dbg_cnt0 << " Mean " - << (float)dbg_cnt1 / (dbg_cnt0 ? dbg_cnt0 : 1) << endl; + if (means[0]) + cerr << "Total " << means[0] << " Mean " + << (float)means[1] / means[0] << endl; } -/// engine_name() returns the full name of the current Stockfish version. -/// This will be either "Stockfish YYMMDD" (where YYMMDD is the date when -/// the program was compiled) or "Stockfish ", depending -/// on whether the constant EngineVersion (defined in misc.h) is empty. +/// Our fancy logging facility. The trick here is to replace cin.rdbuf() and +/// cout.rdbuf() with two Tie objects that tie cin and cout to a file stream. We +/// can toggle the logging of std::cout and std:cin at runtime while preserving +/// usual i/o functionality and without changing a single line of code! +/// Idea from http://groups.google.com/group/comp.lang.c++/msg/1d941c0f26ea0d81 -const string engine_name() { +struct Tie: public streambuf { // MSVC requires splitted streambuf for cin and cout - const string months("Jan Feb Mar Apr May Jun Jul Aug Sep Oct Nov Dec"); - const string cpu64(CpuIs64Bit ? " 64bit" : ""); + Tie(streambuf* b, ofstream* f) : buf(b), file(f) {} - if (!EngineVersion.empty()) - return AppName + " " + EngineVersion + cpu64; + int sync() { return file->rdbuf()->pubsync(), buf->pubsync(); } + int overflow(int c) { return log(buf->sputc((char)c), "<< "); } + int underflow() { return buf->sgetc(); } + int uflow() { return log(buf->sbumpc(), ">> "); } - stringstream s, date(__DATE__); // From compiler, format is "Sep 21 2008" - string month, day, year; + streambuf* buf; + ofstream* file; - date >> month >> day >> year; + int log(int c, const char* prefix) { - s << setfill('0') << AppName + " " + AppTag + " " - << year.substr(2, 2) << setw(2) - << (1 + months.find(month) / 4) << setw(2) - << day << cpu64; + static int last = '\n'; - return s.str(); -} + if (last == '\n') + file->rdbuf()->sputn(prefix, 3); + return last = file->rdbuf()->sputc((char)c); + } +}; -/// get_system_time() returns the current system time, measured in -/// milliseconds. +class Logger { -int get_system_time() { + Logger() : in(cin.rdbuf(), &file), out(cout.rdbuf(), &file) {} + ~Logger() { start(false); } -#if defined(_MSC_VER) - struct _timeb t; - _ftime(&t); - return int(t.time*1000 + t.millitm); -#else - struct timeval t; - gettimeofday(&t, NULL); - return t.tv_sec*1000 + t.tv_usec/1000; -#endif -} + ofstream file; + Tie in, out; +public: + static void start(bool b) { -/// cpu_count() tries to detect the number of CPU cores. + static Logger l; -#if !defined(_MSC_VER) + if (b && !l.file.is_open()) + { + l.file.open("io_log.txt", ifstream::out | ifstream::app); + cin.rdbuf(&l.in); + cout.rdbuf(&l.out); + } + else if (!b && l.file.is_open()) + { + cout.rdbuf(l.out.buf); + cin.rdbuf(l.in.buf); + l.file.close(); + } + } +}; -# if defined(_SC_NPROCESSORS_ONLN) -int cpu_count() { - return Min(sysconf(_SC_NPROCESSORS_ONLN), MAX_THREADS); -} -# elif defined(__hpux) -int cpu_count() { - struct pst_dynamic psd; - if (pstat_getdynamic(&psd, sizeof(psd), (size_t)1, 0) == -1) - return 1; - return Min(psd.psd_proc_cnt, MAX_THREADS); -} -# else -int cpu_count() { - return 1; -} -# endif +/// Used to serialize access to std::cout to avoid multiple threads to write at +/// the same time. -#else +std::ostream& operator<<(std::ostream& os, SyncCout sc) { -int cpu_count() { - SYSTEM_INFO s; - GetSystemInfo(&s); - return Min(s.dwNumberOfProcessors, MAX_THREADS); -} + static Mutex m; -#endif + if (sc == io_lock) + m.lock(); + if (sc == io_unlock) + m.unlock(); -/// Check for console input. Original code from Beowulf and Olithink + return os; +} -#ifndef _WIN32 -int data_available() -{ - fd_set readfds; - struct timeval timeout; +/// Trampoline helper to avoid moving Logger to misc.h +void start_logger(bool b) { Logger::start(b); } - FD_ZERO(&readfds); - FD_SET(fileno(stdin), &readfds); - timeout.tv_sec = 0; // Set to timeout immediately - timeout.tv_usec = 0; - select(16, &readfds, 0, 0, &timeout); - return (FD_ISSET(fileno(stdin), &readfds)); -} +/// cpu_count() tries to detect the number of CPU cores -#else +int cpu_count() { -int data_available() -{ - static HANDLE inh = NULL; - static bool usePipe = false; - INPUT_RECORD rec[256]; - DWORD dw, recCnt; +#if defined(_WIN32) || defined(_WIN64) + SYSTEM_INFO s; + GetSystemInfo(&s); + return s.dwNumberOfProcessors; +#else - if (!inh) - { - inh = GetStdHandle(STD_INPUT_HANDLE); - if (GetConsoleMode(inh, &dw)) - { - SetConsoleMode(inh, dw & ~(ENABLE_MOUSE_INPUT | ENABLE_WINDOW_INPUT)); - FlushConsoleInputBuffer(inh); - } else - usePipe = true; - } +# if defined(_SC_NPROCESSORS_ONLN) + return sysconf(_SC_NPROCESSORS_ONLN); +# elif defined(__hpux) + struct pst_dynamic psd; + if (pstat_getdynamic(&psd, sizeof(psd), (size_t)1, 0) == -1) + return 1; + return psd.psd_proc_cnt; +# else + return 1; +# endif - // If we're running under XBoard then we can't use PeekConsoleInput() as - // the input commands are sent to us directly over the internal pipe. - if (usePipe) - return PeekNamedPipe(inh, NULL, 0, NULL, &dw, NULL) ? dw : 1; +#endif +} - // Count the number of unread input records, including keyboard, - // mouse, and window-resizing input records. - GetNumberOfConsoleInputEvents(inh, &dw); - // Read data from console without removing it from the buffer - if (dw <= 0 || !PeekConsoleInput(inh, rec, Min(dw, 256), &recCnt)) - return 0; +/// timed_wait() waits for msec milliseconds. It is mainly an helper to wrap +/// conversion from milliseconds to struct timespec, as used by pthreads. - // Search for at least one keyboard event - for (DWORD i = 0; i < recCnt; i++) - if (rec[i].EventType == KEY_EVENT) - return 1; +void timed_wait(WaitCondition& sleepCond, Lock& sleepLock, int msec) { - return 0; -} +#if defined(_WIN32) || defined(_WIN64) + int tm = msec; +#else + timespec ts, *tm = &ts; + uint64_t ms = Time::now() + msec; + ts.tv_sec = ms / 1000; + ts.tv_nsec = (ms % 1000) * 1000000LL; #endif + cond_timedwait(sleepCond, sleepLock, tm); +} + /// prefetch() preloads the given address in L1/L2 cache. This is a non /// blocking function and do not stalls the CPU waiting for data to be -/// loaded from RAM, that can be very slow. +/// loaded from memory, that can be quite slow. #if defined(NO_PREFETCH) + void prefetch(char*) {} + #else void prefetch(char* addr) { -#if defined(__INTEL_COMPILER) || defined(__ICL) +# if defined(__INTEL_COMPILER) // This hack prevents prefetches to be optimized away by // Intel compiler. Both MSVC and gcc seems not affected. __asm__ (""); -#endif +# endif - _mm_prefetch(addr, _MM_HINT_T2); - _mm_prefetch(addr+64, _MM_HINT_T2); // 64 bytes ahead +# if defined(__INTEL_COMPILER) || defined(_MSC_VER) + _mm_prefetch(addr, _MM_HINT_T0); + _mm_prefetch(addr+64, _MM_HINT_T0); // 64 bytes ahead +# else + __builtin_prefetch(addr); + __builtin_prefetch(addr+64); +# endif } #endif -