X-Git-Url: http://git.treefish.org/~alex/phys/latlib.git/blobdiff_plain/8df1123044f6db7e7fb1b3981cd77750cb1ebb43..836ca8cca791128595d9514b6d03a47dc32dd7a4:/writeout.cpp?ds=inline diff --git a/writeout.cpp b/writeout.cpp index 73adfa6..9a5409d 100644 --- a/writeout.cpp +++ b/writeout.cpp @@ -9,6 +9,7 @@ #include #include #include +#include using namespace std; @@ -26,18 +27,34 @@ writeout::writeout(const string& wdir, const string& signature, if(wdir != ""){ numprocs = procs; sprintf(cRank, "%d", rank); - fulldir = wdir + "/" + tstamp(timestamp) + "_" + signature; + fulldir = wdir + "/" + tstamp(timestamp) + "_" + signature + ".tmp"; mkdir( fulldir.c_str(), 0775); if(rank>0) of.open( (fulldir + "/rank" + cRank + ".tmp").c_str() ); else of.open( (fulldir + "/" + signature + ".dat").c_str() ); + logf.open( (fulldir + "/rank" + cRank + ".log").c_str() ); + + logf << "[ " << timestring() << " ] Log starts here." << endl; + buf = of.rdbuf(); + logbuf = logf.rdbuf(); } else{ buf = cout.rdbuf(); + logbuf = cerr.rdbuf(); } out = new ostream(buf); + log = new ostream(logbuf); +} + +string writeout::timestring() +{ + time_t rawtime; + string timestring; + time( &rawtime ); + timestring = asctime( localtime( &rawtime ) ); + return timestring.substr(0, timestring.size()-1);; } writeout::~writeout() @@ -69,6 +86,7 @@ writeout::~writeout() } of << "#end" << endl << flush; of.close(); + rename( fulldir.c_str(), fulldir.substr(0, fulldir.length()-4).c_str() ); } else { @@ -77,6 +95,8 @@ writeout::~writeout() (fulldir + "/rank" + cRank + ".part").c_str()); } } + logf << "[ " << timestring() << " ] Log ends here." << endl; + logf.close(); } string writeout::getdatfile()