writeout.cpp

   1 #include "writeout.h"
   2
   3 #include <sys/types.h>
   4 #include <sys/stat.h>
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include <iostream>
   8 #include <time.h>
   9 #include <sstream>
  10 #include <dirent.h>
  11 #include <errno.h>
  12 #include <unistd.h>
  13
  14 using namespace std;
  15
  16 string writeout::tstamp(const long& timestamp)
  17 {
  18   stringstream sstr;
  19   if(!timestamp) sstr << time (NULL);
  20   else sstr << timestamp;
  21   return sstr.str();
  22 }
  23
  24 writeout::writeout(const string& wdir, const string& signature,
  25                    const int& rank, const int& procs, const long& timestamp)
  26 {
  27   if(wdir != ""){
  28     numprocs = procs;
  29     sprintf(cRank, "%d", rank);
  30     fulldir = wdir + "/" + tstamp(timestamp) + "_" + signature + ".tmp";
  31     mkdir( fulldir.c_str(),  0775);
  32
  33     if(rank>0) of.open( (fulldir + "/rank" + cRank + ".tmp").c_str() );
  34     else of.open( (fulldir + "/" + signature + ".dat").c_str() );
  35
  36     buf = of.rdbuf();
  37   }
  38   else{
  39     buf = cout.rdbuf();
  40   }
  41   out = new ostream(buf);
  42 }
  43
  44 writeout::~writeout()
  45 {
  46   if(fulldir != "")
  47     {
  48       if( cRank[0] == '0' )
  49         {
  50           int jobsdone=0;
  51           while(jobsdone<numprocs-1)
  52             {
  53               string nextfile;
  54               if( (nextfile=getdatfile()) == "" ) sleep(1);
  55               else
  56                 {
  57                   cerr << "collecting " << nextfile << endl;
  58
  59                   ifstream myfile( (fulldir + "/" + nextfile).c_str() );
  60                   while(true){
  61                     string line;
  62                     getline(myfile, line);
  63                     if( !myfile.good() ) break;
  64                     of << line << endl << flush;
  65                   }
  66                   myfile.close();
  67                   remove( (fulldir + "/" + nextfile).c_str() );
  68                   jobsdone++;
  69                 }
  70             }
  71           of << "#end" << endl << flush;
  72           of.close();
  73           rename( fulldir.c_str(), fulldir.substr(0, fulldir.length()-4).c_str() );
  74         }
  75       else
  76         {
  77           of.close();
  78           rename((fulldir + "/rank" + cRank + ".tmp").c_str(),
  79                  (fulldir + "/rank" + cRank + ".part").c_str());
  80         }
  81     }
  82 }
  83
  84 string writeout::getdatfile()
  85 {
  86   string myfile;
  87   DIR *dp;
  88   struct dirent *dirp;
  89
  90   if((dp  = opendir(fulldir.c_str())) == NULL) {
  91     cerr << "Error(" << errno << ") opening " << fulldir << endl;
  92     return "";
  93   }
  94
  95   while ((dirp = readdir(dp)) != NULL)
  96     {
  97       myfile = string(dirp->d_name);
  98       if(myfile.length() > 3 && myfile.substr(myfile.length()-4) == "part")
  99         return myfile;
 100     }
 101
 102   return "";
 103 }