X-Git-Url: http://git.treefish.org/~alex/phys/latlib.git/blobdiff_plain/4d4c1c97b3e05fc15c7ebac8d5237ad1b8844230..836ca8cca791128595d9514b6d03a47dc32dd7a4:/writeout.cpp?ds=inline diff --git a/writeout.cpp b/writeout.cpp index eb14309..9a5409d 100644 --- a/writeout.cpp +++ b/writeout.cpp @@ -9,34 +9,52 @@ #include #include #include +#include using namespace std; -string writeout::tstamp() +string writeout::tstamp(const long& timestamp) { - stringstream sstr; - sstr << time (NULL); - return sstr.str(); + stringstream sstr; + if(!timestamp) sstr << time (NULL); + else sstr << timestamp; + return sstr.str(); } writeout::writeout(const string& wdir, const string& signature, - const int& rank, const int& procs) + const int& rank, const int& procs, const long& timestamp) { if(wdir != ""){ numprocs = procs; sprintf(cRank, "%d", rank); - fulldir = wdir + "/" + tstamp() + "_" + 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() @@ -68,6 +86,7 @@ writeout::~writeout() } of << "#end" << endl << flush; of.close(); + rename( fulldir.c_str(), fulldir.substr(0, fulldir.length()-4).c_str() ); } else { @@ -76,6 +95,8 @@ writeout::~writeout() (fulldir + "/rank" + cRank + ".part").c_str()); } } + logf << "[ " << timestring() << " ] Log ends here." << endl; + logf.close(); } string writeout::getdatfile()