X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=src%2Fmisc.cpp;h=0db22dd2b39ca3f043e3fe905c5c074233495470;hb=df80232495f9500c7d6ed197e40cebbc77175f07;hp=74b6385a45217d18c833b658a013b5f58c29a904;hpb=caa02b0e43dbce0c723bc6499aab811ca3e5408e;p=stockfish diff --git a/src/misc.cpp b/src/misc.cpp index 74b6385a..0db22dd2 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,287 +17,248 @@ along with this program. If not, see . */ +#if defined(_WIN32) || defined(_WIN64) -//// -//// Includes -//// +#define NOMINMAX // disable macros min() and max() +#include -#if !defined(_MSC_VER) +#else -# 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 +#include -#include "bitcount.h" #include "misc.h" #include "thread.h" 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" + + if (Version.empty()) + { + date >> month >> day >> year; -//// -//// Functions -//// + s << "Stockfish " << Tag + << setfill('0') << " " << year.substr(2) + << setw(2) << (1 + months.find(month) / 4) + << setw(2) << day; + } + else + s << "Stockfish " << Version; -void dbg_hit_on(bool b) { + s << cpu64 << popcnt << (to_uci ? "\nid author ": " by ") + << "Tord Romstad, Marco Costalba and Joona Kiiski"; - assert(!dbg_show_mean); - dbg_show_hit_rate = true; - dbg_cnt0++; - if (b) - dbg_cnt1++; + return s.str(); } -void dbg_hit_on_c(bool c, bool b) { - if (c) - dbg_hit_on(b); -} +/// Debug functions used mainly to collect run-time statistics -void dbg_before() { +static uint64_t hits[2], means[2]; - assert(!dbg_show_mean); - dbg_show_hit_rate = true; - dbg_cnt0++; -} +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_after() { +void dbg_print() { - assert(!dbg_show_mean); - dbg_show_hit_rate = true; - dbg_cnt1++; + if (hits[0]) + cerr << "Total " << hits[0] << " Hits " << hits[1] + << " hit rate (%) " << 100 * hits[1] / hits[0] << endl; + + if (means[0]) + cerr << "Total " << means[0] << " Mean " + << (float)means[1] / means[0] << endl; } -void dbg_mean_of(int v) { - assert(!dbg_show_hit_rate); - dbg_show_mean = true; - dbg_cnt0++; - dbg_cnt1 += v; -} +/// Our fancy logging facility. The trick here is to replace cout.rdbuf() with +/// this one that sends the output both to console and to a file, this allow us +/// to toggle the logging of std::cout to a file while preserving output to +/// stdout and without changing a single line of code! Idea and code from: +/// http://groups.google.com/group/comp.lang.c++/msg/1d941c0f26ea0d81 -void dbg_print_hit_rate() { +class Tee: public streambuf { +public: + typedef char_traits traits_type; + typedef traits_type::int_type int_type; - cout << "Total " << dbg_cnt0 << " Hit " << dbg_cnt1 - << " hit rate (%) " << (dbg_cnt1*100)/(dbg_cnt0 ? dbg_cnt0 : 1) << endl; -} + Tee(ios& s, ofstream& f) : stream(s), file(f), stream_buf(s.rdbuf()) {} + ~Tee() { set(false); } -void dbg_print_mean() { + void set(bool b) { stream.rdbuf(b ? this : stream_buf); } - cout << "Total " << dbg_cnt0 << " Mean " - << (float)dbg_cnt1 / (dbg_cnt0 ? dbg_cnt0 : 1) << endl; -} +private: + int_type overflow(int_type c) { + if (traits_type::eq_int_type(c, traits_type::eof())) + return traits_type::not_eof(c); -/// 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. + c = stream_buf->sputc(traits_type::to_char_type(c)); -const string engine_name() { + if (!traits_type::eq_int_type(c, traits_type::eof())) + c = file.rdbuf()->sputc(traits_type::to_char_type(c)); - const string months("Jan Feb Mar Apr May Jun Jul Aug Sep Oct Nov Dec"); - const string cpu64(CpuIs64Bit ? " 64bit" : ""); + return c; + } - if (!EngineVersion.empty()) - return AppName + " " + EngineVersion + cpu64; + int sync() { - stringstream s, date(__DATE__); // From compiler, format is "Sep 21 2008" - string month, day, year; + int c = stream_buf->pubsync(); - date >> month >> day >> year; + if (c != -1) + c = file.rdbuf()->pubsync(); - s << setfill('0') << AppName + " " + AppTag + " " - << year.substr(2, 2) << setw(2) - << (1 + months.find(month) / 4) << setw(2) - << day << cpu64; + return c; + } - return s.str(); -} + int underflow() { return traits_type::not_eof(stream_buf->sgetc()); } -const string engine_author() { return "Tord Romstad, Marco Costalba and Joona Kiiski"; } + int uflow() { + int c = stream_buf->sbumpc(); -/// get_system_time() returns the current system time, measured in -/// milliseconds. + if (!traits_type::eq_int_type(c, traits_type::eof())) + file.rdbuf()->sputc(traits_type::to_char_type(c)); -int get_system_time() { + return traits_type::not_eof(c); + } -#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 + ios& stream; + ofstream& file; + streambuf* stream_buf; +}; + +class Logger { +public: + Logger() : in(cin, file), out(cout, file) {} + ~Logger() { set(false); } + + void set(bool b) { + + if (b && !file.is_open()) + { + file.open("io_log.txt", ifstream::out | ifstream::app); + in.set(true); + out.set(true); + } + else if (!b && file.is_open()) + { + out.set(false); + in.set(false); + file.close(); + } + } + +private: + Tee in, out; + ofstream file; +}; + +void logger_set(bool b) { + + static Logger l; + l.set(b); } -/// cpu_count() tries to detect the number of CPU cores. +/// cpu_count() tries to detect the number of CPU cores -#if !defined(_MSC_VER) +int cpu_count() { + +#if defined(_WIN32) || defined(_WIN64) + SYSTEM_INFO s; + GetSystemInfo(&s); + return std::min(int(s.dwNumberOfProcessors), MAX_THREADS); +#else # if defined(_SC_NPROCESSORS_ONLN) -int cpu_count() { - return Min(sysconf(_SC_NPROCESSORS_ONLN), MAX_THREADS); -} + return std::min((int)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); -} + return std::min((int)psd.psd_proc_cnt, MAX_THREADS); # else -int cpu_count() { return 1; -} # endif -#else - -int cpu_count() { - SYSTEM_INFO s; - GetSystemInfo(&s); - return Min(s.dwNumberOfProcessors, MAX_THREADS); -} - #endif +} -/// Check for console input. Original code from Beowulf, Olithink and Greko - -#ifndef _WIN32 - -int input_available() -{ - fd_set readfds; - struct timeval timeout; - - 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); +/// timed_wait() waits for msec milliseconds. It is mainly an helper to wrap +/// conversion from milliseconds to struct timespec, as used by pthreads. - return (FD_ISSET(fileno(stdin), &readfds)); -} +void timed_wait(WaitCondition& sleepCond, Lock& sleepLock, int msec) { +#if defined(_WIN32) || defined(_WIN64) + int tm = msec; #else + timespec ts, *tm = &ts; + uint64_t ms = Time::current_time().msec() + msec; -int input_available() -{ - static HANDLE inh = NULL; - static bool usePipe = false; - INPUT_RECORD rec[256]; - DWORD nchars, recCnt; - - if (!inh) - { - inh = GetStdHandle(STD_INPUT_HANDLE); - if (GetConsoleMode(inh, &nchars)) - { - SetConsoleMode(inh, nchars & ~(ENABLE_MOUSE_INPUT | ENABLE_WINDOW_INPUT)); - FlushConsoleInputBuffer(inh); - } else - usePipe = true; - } + ts.tv_sec = ms / 1000; + ts.tv_nsec = (ms % 1000) * 1000000LL; +#endif - // When using Standard C input functions, also check if there - // is anything in the buffer. After a call to such functions, - // the input waiting in the pipe will be copied to the buffer, - // and the call to PeekNamedPipe can indicate no input available. - // Setting stdin to unbuffered was not enough. [from Greko] - if (stdin->_cnt > 0) - return 1; - - // When running under a GUI the input commands are sent to us - // directly over the internal pipe. If PeekNamedPipe() returns 0 - // then something went wrong. Probably the parent program exited. - // Returning 1 will make the next call to the input function - // return EOF, where this should be catched then. - if (usePipe) - return PeekNamedPipe(inh, NULL, 0, NULL, &nchars, NULL) ? nchars : 1; - - // Count the number of unread input records, including keyboard, - // mouse, and window-resizing input records. - GetNumberOfConsoleInputEvents(inh, &nchars); - - // Read data from console without removing it from the buffer - if (nchars <= 0 || !PeekConsoleInput(inh, rec, Min(nchars, 256), &recCnt)) - return 0; - - // Search for at least one keyboard event - for (DWORD i = 0; i < recCnt; i++) - if (rec[i].EventType == KEY_EVENT) - return 1; - - return 0; + cond_timedwait(sleepCond, sleepLock, tm); } -#endif - /// 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) || defined(__ICL) // 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 } #endif -