HistoFile.cxx

Go to the documentation of this file.

// $Id: HistoFile.cxx,v 1.2 2008/08/29 22:18:12 fmwk Exp $
//
//
// messier@indiana.edu
#include <string>
extern "C" {
#include <unistd.h>
}
#include "TFile.h"
#include "JobControl/HistoFile.h"
using namespace jobc;

static std::string gsFileName("histos.root");
static TFile*      gsTFile    = 0;
static TDirectory* gsCurrDir  = 0;
static bool        gsSaveHistos = true;

//......................................................................

bool HistoFile::SaveHistos()
{
  return gsSaveHistos;
}

//......................................................................

void HistoFile::SetFileName(const char* name) 
{
  if (name==0) return;

  std::string sname = name;
  if (sname=="") return;

  if (sname=="0") {
    gsSaveHistos = false;
    return;
  }

  gsFileName = name;
}

//......................................................................

const char* HistoFile::FileName() {return gsFileName.c_str(); }

//......................................................................

TFile* HistoFile::File() { return gsTFile; }

//......................................................................

void HistoFile::StepIn(const char* dirname) 
{
  if (gsTFile==0) {
    // Check that the file name is not already in use
    const int fOK     = 0;
    int       fexists = access(gsFileName.c_str(),F_OK);
    for (int i = 1; (i < 100) && (fexists == fOK); ++i) {
      std::string::size_type pos = gsFileName.find(".root");
      if (pos == std::string::npos) {
        gsFileName += ".root";
        i--;
      }
      else {
        if (i == 1) {
          gsFileName.insert(pos, "-01");
        }
        else if (i % 10 == 0) {
          pos--;
          gsFileName[pos--] = '0';
          char s = '0' + i / 10;
          gsFileName[pos] = s;
        }
        else {
          char s = '0' + i % 10;
          gsFileName[pos - 1] = s;
        }
      }
      fexists = access(gsFileName.c_str(), F_OK);
    } 
    gsTFile = new TFile(gsFileName.c_str(), "RECREATE");
  }
  gsCurrDir = gDirectory;
  if (dirname == 0) { gsTFile->cd(); return; }

  
  TObject* obj = gsTFile->FindObject(dirname);
  if (obj==0) gsTFile->mkdir(dirname);
  gsTFile->cd(dirname);
}

//......................................................................

void HistoFile::StepOut() 
{
  if (gsCurrDir) gsCurrDir->cd();
}

---