UtilFile.cxx

Go to the documentation of this file.

#include <libgen.h>
#include <fstream>

#include "TKey.h"
#include "TSystem.h"
#include "TTree.h"
#include "TFile.h"
#include "TIterator.h"

#include "MessageService/MsgService.h"
#include "Util/UtilString.h"

#include "AtNuUtils/UtilFile.h"

CVSID("$Id: UtilFile.cxx,v 1.2 2006/07/19 22:41:29 gmieg Exp $");

using namespace std;

bool UtilFile::CheckDir(const char* DirName, bool MakeDir) {
  MSG("Util",Msg::kVerbose) << "UtilFile::CheckDir" << endl;

  string cwd = gSystem->pwd();

  if (gSystem->cd(DirName)) {
    gSystem->cd(cwd.c_str());
    return true;
  }

  if (!MakeDir) return false;

  gSystem->mkdir(DirName,kTRUE);
  if (gSystem->cd(DirName)) {
    gSystem->cd(cwd.c_str());
    return true;
  }

  gSystem->cd(cwd.c_str());

  return false;
}

bool UtilFile::CheckFile(string FileName, bool TestCreate) {
  MSG("Util",Msg::kVerbose) << "UtilFile::CheckFile" << endl;

  if (! gSystem->Exec(("ls "+FileName+">&/dev/null").c_str())) {
    return true;
  }

  if (! TestCreate) return false;

  if (! gSystem->Exec(("touch "+FileName+">&/dev/null").c_str())) {
    gSystem->Unlink(FileName.c_str());
    return true;
  }

  return false;
}

vector<string> UtilFile::DirFileList(string DirName, string Suf,
                                    bool IncludePath, bool IncludeSuf,
                                    vector<string> FList) {
  MSG("Util",Msg::kVerbose) << "UtilFile::DirFileList" << endl;

  MSG("Util",Msg::kDebug) << "Checking " << DirName << " for suffix " << Suf << endl;

  if(!UtilFile::CheckDir(DirName.c_str(),false)) return FList;

  void *dir = gSystem->OpenDirectory(DirName.c_str());

  if (!dir) {
    MSG("Util",Msg::kWarning)<< "Error opening " << DirName << endl;
    return FList;
  }

  const char* file;
  string fname;

  //Span root files in the directory and build lists
  while ((file = gSystem->GetDirEntry(dir))){
    if(IncludePath) fname = DirName+"/"+file;
    else fname = file;
    if (fname.length() > Suf.length() &&
        fname.find(Suf) == fname.length()-Suf.length()) {
      if(IncludeSuf) FList.push_back(fname);
      else FList.push_back(fname.substr(0,fname.find(Suf)));
    }
  }
  gSystem->FreeDirectory(dir);
  MSG("Util",Msg::kDebug) << "Return element:" << FList.size() << endl;
  return FList;
}

vector<string> UtilFile::FileFileList(string FileName, string Suf,
                                     bool IncludePath, bool IncludeSuf,
                                     vector<string> FList) {
  MSG("Util",Msg::kVerbose) << "UtilFile::FileFileList" << endl;

  MSG("Util",Msg::kDebug) << "Checking " << FileName << " for suffix " << Suf
         << "(" << Suf.length() << ")" << endl;

  char FName[256];
  string fname;

  while (1) {
    string::size_type ispace = FileName.find(" ");
    string ThisFile = FileName.substr(0,ispace);

    MSG("Util",Msg::kDebug) << "Reading List File " << ThisFile << endl;

    ifstream RFile(ThisFile.c_str());
    while (1) {
      RFile.getline(FName, 256);
      //RFile >> FName;
      MSG("Util",Msg::kDebug) << "  " << FName << endl;
      if(!RFile.good()) break;
      if (IncludePath) fname = FName;
      else fname = basename(FName);
      MSG("Util",Msg::kDebug) << "  =" << fname << endl;
      if (fname.length() > Suf.length() && (Suf.length() == 0 ||
          fname.find(Suf) == fname.length()-Suf.length())) {
        if(IncludeSuf) FList.push_back(fname);
        else FList.push_back(fname.substr(0,fname.find(Suf)));
      }
    }
    RFile.close();

    if(ispace == string::npos) break;
    FileName.erase(0,ispace+1);

  }
  MSG("Util",Msg::kDebug) << "Return element:" << FList.size() << endl;
  return FList;
}

void UtilFile::FileListOut(string FileName, vector<string> FList,
                          bool Append) {
  ofstream lfile;
  ios_base::openmode out_mode = ios::out;
  if(Append) out_mode = out_mode | ios::app;

  lfile.open(FileName.c_str(),out_mode);
  if (! lfile.is_open() ) {
    MSG("Util",Msg::kWarning)<< "Couldn't open list file " << FileName << endl;
    return;
  }

  for(unsigned int i=0; i<FList.size(); i++) lfile << FList[i] << endl;

  lfile.close();
}

void UtilFile::FileListOut(string FileName, string FStr, bool Append) {
  ofstream lfile;
  ios_base::openmode out_mode = ios::out;
  if(Append) out_mode = out_mode | ios::app;

  lfile.open(FileName.c_str(),out_mode);
  if (! lfile.is_open() ) {
    MSG("Util",Msg::kWarning)<< "Couldn't open list file " << FileName << endl;
    return;
  }

  lfile << FStr << endl;

  lfile.close();
}

---