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 bool writeout::DirectoryExists( const char* pzPath )
  17 {
  18     if ( pzPath == NULL) return false;
  19
  20     DIR *pDir;
  21     bool bExists = false;
  22
  23     pDir = opendir (pzPath);
  24
  25     if (pDir != NULL)
  26     {
  27         bExists = true;
  28         (void) closedir (pDir);
  29     }
  30
  31     return bExists;
  32 }
  33
  34 string writeout::tstamp(const long& timestamp)
  35 {
  36   stringstream sstr;
  37   if(!timestamp) sstr << time (NULL);
  38   else sstr << timestamp;
  39   return sstr.str();
  40 }
  41
  42 writeout::writeout(const string& wdir, const string& signature,
  43                    const int& rank, const int& procs, const long& timestamp)
  44 {
  45   if(wdir != ""){
  46     numprocs = procs;
  47     sprintf(cRank, "%d", rank);
  48     fulldir = wdir + "/" + tstamp(timestamp) + "_" + signature + ".tmp";
  49
  50     for(int itry=0; itry<100 && (!DirectoryExists(fulldir.c_str())); itry++) {
  51       if (rank==0)
  52         mkdir( fulldir.c_str(), 0775);
  53       sleep(1);
  54     }
  55
  56     if(rank>0) of.open( (fulldir + "/rank" + cRank + ".tmp").c_str() );
  57     else of.open( (fulldir + "/" + signature + ".dat").c_str() );
  58
  59     logf.open( (fulldir + "/rank" + cRank + ".log").c_str() );
  60
  61     if ( (!of.is_open()) || (!logf.is_open()) ) {
  62       cerr << "WRITEOUT: Could not open output- and/or log-file!" << endl << flush;
  63       exit(1);
  64     }
  65
  66     logf << "[ " << timestring() << " ] Log starts here." << endl;
  67
  68     buf = of.rdbuf();
  69     logbuf = logf.rdbuf();
  70
  71     cout << buf << endl;
  72   }
  73   else{
  74     buf = cout.rdbuf();
  75     logbuf = cerr.rdbuf();
  76   }
  77   out = new ostream(buf);
  78   log = new ostream(logbuf);
  79 }
  80
  81 string writeout::timestring()
  82 {
  83   time_t rawtime;
  84   string timestring;
  85   time( &rawtime );
  86   timestring = asctime( localtime( &rawtime ) );
  87   return timestring.substr(0, timestring.size()-1);;
  88 }
  89
  90 writeout::~writeout()
  91 {
  92   if(fulldir != "")
  93     {
  94       if( cRank[0] == '0' )
  95         {
  96           int jobsdone=0;
  97           while(jobsdone<numprocs-1)
  98             {
  99               string nextfile;
 100               if( (nextfile=getdatfile()) == "" ) sleep(1);
 101               else
 102                 {
 103                   logf << "collecting " << nextfile << endl;
 104
 105                   ifstream myfile( (fulldir + "/" + nextfile).c_str() );
 106                   while(true){
 107                     string line;
 108                     getline(myfile, line);
 109                     if( !myfile.good() ) break;
 110                     of << line << endl << flush;
 111                   }
 112                   myfile.close();
 113                   remove( (fulldir + "/" + nextfile).c_str() );
 114                   jobsdone++;
 115                 }
 116             }
 117           of << "#end" << endl << flush;
 118           of.close();
 119           rename( fulldir.c_str(), fulldir.substr(0, fulldir.length()-4).c_str() );
 120         }
 121       else
 122         {
 123           of.close();
 124           rename((fulldir + "/rank" + cRank + ".tmp").c_str(),
 125                  (fulldir + "/rank" + cRank + ".part").c_str());
 126         }
 127     }
 128   logf << "[ " << timestring() << " ] Log ends here." << endl;
 129   logf.close();
 130 }
 131
 132 string writeout::getdatfile()
 133 {
 134   string myfile;
 135   DIR *dp;
 136   struct dirent *dirp;
 137
 138   if((dp  = opendir(fulldir.c_str())) == NULL) {
 139     logf << "Error(" << errno << ") opening " << fulldir << endl;
 140     return "";
 141   }
 142
 143   while ((dirp = readdir(dp)) != NULL)
 144     {
 145       myfile = string(dirp->d_name);
 146       if(myfile.length() > 3 && myfile.substr(myfile.length()-4) == "part")
 147         return myfile;
 148     }
 149
 150   return "";
 151 }