...

[phys/latlib.git] / writeout.cpp

diff --git a/writeout.cpp b/writeout.cpp
index 73adfa684ddd79eda8191cc1b448c74220ccdce2..54b82038d2bbe491e4dfbcbd3dc9e99a0b07473c 100644 (file)
--- a/writeout.cpp
+++ b/writeout.cpp
@@ -9,6 +9,7 @@
 #include <sstream>
 #include <dirent.h>
 #include <errno.h>
+#include <unistd.h>
 
 using namespace std;
 
@@ -26,18 +27,35 @@ 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();
+    logf.open("/dev/null");
+    logbuf = logf.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 +87,7 @@ writeout::~writeout()
            }
          of << "#end" << endl << flush;
          of.close();
+         rename( fulldir.c_str(), fulldir.substr(0, fulldir.length()-4).c_str() );
        }
       else
        {
@@ -77,6 +96,8 @@ writeout::~writeout()
                 (fulldir + "/rank" + cRank + ".part").c_str());
        }
     }
+  logf << "[ " << timestring() << " ] Log ends here." << endl;
+  logf.close();
 }
 
 string writeout::getdatfile()