X-Git-Url: http://git.treefish.org/~alex/phys/latlib.git/blobdiff_plain/f6dad6a4230bed002b51de0ec8ff34410bac27ec..061e6469f58f3e3e892f1cd1f6fdd2aad9c644e8:/writeout.cpp?ds=inline

diff --git a/writeout.cpp b/writeout.cpp
index e10a9d4..a756394 100644
--- a/writeout.cpp
+++ b/writeout.cpp
@@ -11,24 +11,25 @@
 #include <errno.h>
 #include <unistd.h>
 
-#include <boost/filesystem.hpp>
+#ifndef MPI_DISABLED
+#include <mpi.h>
+#endif
 
 using namespace std;
 
-string writeout::tstamp(const long& timestamp)
+string writeout::longToStr (long arg)
 {
-  stringstream sstr;
-  if(!timestamp) sstr << time (NULL);
-  else sstr << timestamp;
-  return sstr.str();
+  stringstream ss;
+  ss << arg;
+  return ss.str();
 }
 
 void writeout::newsub(string subname) {
   of[subname] = new ofstream;
 
   if ( fulldir != "" ) {
-    if(rank>0) of[subname]->open( (fulldir + "/rank" + cRank + "-" + subname + ".tmp").c_str() );
-    else of[subname]->open( (fulldir + "/" + signature + "-" + subname + ".dat").c_str() );
+    if(rank>0) of[subname]->open( (fulldir + "/rank" + cRank + "_" + subname + ".tmp").c_str() );
+    else of[subname]->open( (fulldir + "/" + signature + "_" + subname + ".dat").c_str() );
 
     if ( !of[subname]->is_open() ) {
       logf << "WRITEOUT: Could not open output-file!" << endl << flush;
@@ -44,8 +45,11 @@ void writeout::newsub(string subname) {
 }
 
 writeout::writeout(const string& wdir, const string& _signature, 
-		   const int& _rank, const int& procs, const long& timestamp)
+		   const int& _rank, const int& procs)
 {
+  long timestamp;
+  int iseq=0;
+
   fulldir = "";
   signature = _signature;
   rank = _rank;
@@ -53,17 +57,38 @@ writeout::writeout(const string& wdir, const string& _signature,
   if(wdir != ""){
     numprocs = procs;
     sprintf(cRank, "%d", rank);
-    fulldir = wdir + "/" + tstamp(timestamp) + "_" + signature + ".tmp";
-
-    if (rank==0)
-      mkdir(fulldir.c_str(), 0775);
-    else {
-      while ( ! ( boost::filesystem::exists(fulldir) && boost::filesystem::is_directory(fulldir) ) ) {
-	cerr << "WRITOUT: Waiting for rank 0 to create out directory." << endl << flush;
-	sleep(1);
+
+    if (rank == 0) {
+      timestamp = time(NULL);
+
+      while (true) {
+	fulldir = wdir + "/" + longToStr(timestamp) + "." + longToStr(iseq) + "_" + signature + ".tmp";
+	if ( mkdir(fulldir.c_str(), 0775) == 0 )
+	  break;
+	else if ( errno != EEXIST ) {
+	  cerr << "WRITEOUT: Could not create outdir!" << endl << flush;
+	  break;
+	}
+	iseq++;
       }
+
+#ifndef MPI_DISABLED
+      for(int idest=1; idest<numprocs; idest++) {
+	MPI_Send(&timestamp, 1, MPI_LONG, idest, 123, MPI_COMM_WORLD);
+	MPI_Send(&iseq, 1, MPI_LONG, idest, 124, MPI_COMM_WORLD);
+      }
+#endif
+
     }
 
+#ifndef MPI_DISABLED
+    else if(rank>0) {
+      MPI_Recv(&timestamp, 1, MPI_LONG, 0, 123, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
+      MPI_Recv(&iseq, 1, MPI_LONG, 0, 124, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
+      fulldir = wdir + "/" + longToStr(timestamp) + "." + longToStr(iseq) + "_" + signature + ".tmp";
+    }
+#endif  
+    
     logf.open( (fulldir + "/rank" + cRank + ".log").c_str() );
 
     if ( !logf.is_open() ) {
@@ -120,8 +145,8 @@ writeout::~writeout()
       }
       else {
 	ofit->second->close();
-	rename((fulldir + "/rank" + cRank + "-" + ofit->first + ".tmp").c_str(),
-	       (fulldir + "/rank" + cRank + "-" + ofit->first + ".part").c_str());
+	rename((fulldir + "/rank" + cRank + "_" + ofit->first + ".tmp").c_str(),
+	       (fulldir + "/rank" + cRank + "_" + ofit->first + ".part").c_str());
       }
     }
     if( cRank[0] == '0' )
@@ -148,7 +173,7 @@ string writeout::getdatfile(string subname)
       myfile = string(dirp->d_name);
 
       if(myfile.length() > 3 && myfile.substr(myfile.length()-4) == "part" &&
-	 subname == myfile.substr( myfile.find("-")+1, myfile.rfind(".")-myfile.find("-")-1 ) ) {
+	 subname == myfile.substr( myfile.find("_")+1, myfile.rfind(".")-myfile.find("_")-1 ) ) {
 	closedir(dp);
 	return myfile;
       }