mirror of
https://github.com/sockspls/badfish
synced 2025-05-01 01:03:09 +00:00
Add (smart) logging facility
By means of "Use Debug Log" UCI option it is possible to toggle the logging of std::cout to file "out.txt" while preserving the usual output to stdout. There is zero overhead when logging is disabled and we achieved this without changing a single line of exsisting code, in particular we still use std::cout as usual. The idea and part of the code comes from this article: http://groups.google.com/group/comp.lang.c++/msg/1d941c0f26ea0d81 No functional change. Signed-off-by: Marco Costalba <mcostalba@gmail.com>
This commit is contained in:
parent
2dfc94e0b6
commit
eb28a683bd
3 changed files with 68 additions and 0 deletions
64
src/misc.cpp
64
src/misc.cpp
|
@ -39,6 +39,7 @@
|
||||||
#include <iomanip>
|
#include <iomanip>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <sstream>
|
#include <sstream>
|
||||||
|
#include <streambuf>
|
||||||
|
|
||||||
#include "misc.h"
|
#include "misc.h"
|
||||||
#include "thread.h"
|
#include "thread.h"
|
||||||
|
@ -105,6 +106,69 @@ void dbg_print() {
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// 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
|
||||||
|
|
||||||
|
class Logger: public streambuf {
|
||||||
|
public:
|
||||||
|
typedef char_traits<char> traits_type;
|
||||||
|
typedef traits_type::int_type int_type;
|
||||||
|
|
||||||
|
Logger() : cout_buf(cout.rdbuf()) {}
|
||||||
|
~Logger() { set(false); }
|
||||||
|
|
||||||
|
void set(bool b) {
|
||||||
|
|
||||||
|
if (b && !file.is_open())
|
||||||
|
{
|
||||||
|
file.open("out.txt", ifstream::out | ifstream::app);
|
||||||
|
cout.rdbuf(this);
|
||||||
|
}
|
||||||
|
else if (!b && file.is_open())
|
||||||
|
{
|
||||||
|
cout.rdbuf(cout_buf);
|
||||||
|
file.close();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
int_type overflow(int_type c) {
|
||||||
|
|
||||||
|
if (traits_type::eq_int_type(c, traits_type::eof()))
|
||||||
|
return traits_type::not_eof(c);
|
||||||
|
|
||||||
|
c = cout_buf->sputc(traits_type::to_char_type(c));
|
||||||
|
|
||||||
|
if (!traits_type::eq_int_type(c, traits_type::eof()))
|
||||||
|
c = file.rdbuf()->sputc(traits_type::to_char_type(c));
|
||||||
|
|
||||||
|
return c;
|
||||||
|
}
|
||||||
|
|
||||||
|
int sync() {
|
||||||
|
|
||||||
|
int c = cout_buf->pubsync();
|
||||||
|
|
||||||
|
if (c != -1)
|
||||||
|
c = file.rdbuf()->pubsync();
|
||||||
|
|
||||||
|
return c;
|
||||||
|
}
|
||||||
|
|
||||||
|
ofstream file;
|
||||||
|
streambuf* cout_buf;
|
||||||
|
};
|
||||||
|
|
||||||
|
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
|
||||||
|
|
||||||
int cpu_count() {
|
int cpu_count() {
|
||||||
|
|
|
@ -60,4 +60,6 @@ private:
|
||||||
sys_time_t t;
|
sys_time_t t;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
extern void logger_set(bool b);
|
||||||
|
|
||||||
#endif // !defined(MISC_H_INCLUDED)
|
#endif // !defined(MISC_H_INCLUDED)
|
||||||
|
|
|
@ -33,6 +33,7 @@ OptionsMap Options; // Global object
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
/// 'On change' actions, triggered by an option's value change
|
/// 'On change' actions, triggered by an option's value change
|
||||||
|
void on_logger(const UCIOption& o) { logger_set(o); }
|
||||||
void on_eval(const UCIOption&) { Eval::init(); }
|
void on_eval(const UCIOption&) { Eval::init(); }
|
||||||
void on_threads(const UCIOption&) { Threads.read_uci_options(); }
|
void on_threads(const UCIOption&) { Threads.read_uci_options(); }
|
||||||
void on_hash_size(const UCIOption& o) { TT.set_size(o); }
|
void on_hash_size(const UCIOption& o) { TT.set_size(o); }
|
||||||
|
@ -58,6 +59,7 @@ OptionsMap::OptionsMap() {
|
||||||
int msd = cpus < 8 ? 4 : 7;
|
int msd = cpus < 8 ? 4 : 7;
|
||||||
OptionsMap& o = *this;
|
OptionsMap& o = *this;
|
||||||
|
|
||||||
|
o["Use Debug Log"] = UCIOption(false, on_logger);
|
||||||
o["Use Search Log"] = UCIOption(false);
|
o["Use Search Log"] = UCIOption(false);
|
||||||
o["Search Log Filename"] = UCIOption("SearchLog.txt");
|
o["Search Log Filename"] = UCIOption("SearchLog.txt");
|
||||||
o["Book File"] = UCIOption("book.bin");
|
o["Book File"] = UCIOption("book.bin");
|
||||||
|
|
Loading…
Add table
Reference in a new issue