NuOutputWriter.cxx

Go to the documentation of this file.

// Coded by Jeff Hartnell Jan/2007 onwards
//
// Contact: j.j.hartnell@rl.ac.uk

#include <cstdlib>
#include <fstream>

#include "TFile.h"
#include "TH1.h"
#include "TTree.h"

#include "Conventions/Detector.h"
#include "Conventions/ReleaseType.h"
#include "Conventions/SimFlag.h"
#include "MessageService/MsgService.h"

#include "NtupleUtils/NuEvent.h"
#include "NtupleUtils/NuMCEvent.h"

#include "NuMuBar/NuConfig.h"
#include "NuMuBar/NuOutputWriter.h"

using std::endl;
using std::cout;

CVSID("$Id: NuOutputWriter.cxx,v 1.7 2009/11/26 05:43:38 evansj Exp $");

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

NuOutputWriter::NuOutputWriter()
{
  MSG("NuOutputWriter",Msg::kDebug)
      <<"Running NuOutputWriter Constructor..."<<endl;

  //initialise data members
  fFile=0;
  fNuEvent=0;
  fNuMCEvent=0;
  fNuEventTree=0;
  fNuMCEventTree=0;
  fDirectory=0;
  
  MSG("NuOutputWriter",Msg::kDebug)
      <<"Finished NuOutputWriter Constructor"<<endl;
}

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

NuOutputWriter::~NuOutputWriter()
{
  MSG("NuOutputWriter",Msg::kDebug)
      <<"Running NuOutputWriter Destructor..."<<endl;
  

  MSG("NuOutputWriter",Msg::kDebug)
      <<"Finished NuOutputWriter Destructor"<<endl;
}

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

void NuOutputWriter::SetupFile(Int_t run,Int_t subrun)
{
  //sanity check that the directory is not already set
  if (fDirectory!=0) cout<<"Ahhhh, fDirectory already exists"<<endl;

  //open the file
  fFile=this->OpenFile(run,subrun,"NuEvent");
  
  //store the TDirectory associated with the file just opened
  fDirectory=gDirectory;
}

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

void NuOutputWriter::SetupFile(const NuConfig& config,
                               std::string prefix)
{
  //sanity check that the directory is not already set
  if (fDirectory!=0) cout<<"Ahhhh, fDirectory already exists"<<endl;

  //open the file
  fFile=this->OpenFile(config,prefix);

  //store the TDirectory associated with the file just opened
  fDirectory=gDirectory;
}

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

void NuOutputWriter::SetupNuEventTree()
{
  MSG("NuOutputWriter",Msg::kInfo)
    <<"gDirectory is:"<<endl;
  gDirectory->pwd();
  MSG("NuOutputWriter",Msg::kInfo)
    <<"Local stored fDirectory is:"<<endl;
  this->fDirectory->pwd();
  MSG("NuOutputWriter",Msg::kInfo)
    <<"Setting gDirectory=fDirectory"<<endl;
  gDirectory=fDirectory;
  fNuEventTree=new TTree("s","s");
  cout<<"Creating new NuEvent for output to tree"<<endl;
  fNuEvent=new NuEvent();
  fNuEventTree->Branch("s","NuEvent",&fNuEvent,32000,2);
}

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

void NuOutputWriter::SetupNuMCEventTree()
{
  MSG("NuOutputWriter",Msg::kInfo)
    <<"gDirectory is:"<<endl;
  gDirectory->pwd();
  MSG("NuOutputWriter",Msg::kInfo)
    <<"Local stored fDirectory is:"<<endl;
  this->fDirectory->pwd();
  MSG("NuOutputWriter",Msg::kInfo)
    <<"Setting gDirectory=fDirectory"<<endl;
  gDirectory=fDirectory;
  fNuMCEventTree=new TTree("mc","mc");
  cout<<"Creating new NuMCEvent for output to tree"<<endl;
  fNuMCEvent=new NuMCEvent();
  fNuMCEventTree->Branch("mc","NuMCEvent",&fNuMCEvent,32000,2);
}

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

void NuOutputWriter::Finish()
{
  MSG("NuOutputWriter",Msg::kInfo)
    <<"Running NuOutputWriter::Finish..."<<endl;
  
  //store the current gDirectory
  TDirectory* tmpd=gDirectory;
  MSG("NuOutputWriter",Msg::kDebug)
    <<"gDirectory is:"<<endl;
  //gDirectory->pwd();
  
  //set the gDirectory to be the one associated with the file
  //opened by this object
  //not sure if this is necessary but it is safe
  gDirectory=this->fDirectory;
 
  MSG("NuOutputWriter",Msg::kInfo)
    <<"Changing directory to fFile..."<<endl;
  fFile->cd();
  
  MSG("NuOutputWriter",Msg::kInfo)<<"Writing tree..."<<endl;
  fFile->Write();//this should pick up any histograms as well
  
  MSG("NuOutputWriter",Msg::kInfo)
    <<"Closing file "<<fFile->GetName()<<" ..."<<endl;
  fFile->Close();
  
  //reset the directory to it's original location
  gDirectory=tmpd;
  
  MSG("NuOutputWriter",Msg::kInfo)
    <<"Completed NuOutputWriter::Finish"<<endl;
}

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

NuEvent& NuOutputWriter::GetNuEventToFill()
{
  if (!fNuEvent) this->SetupNuEventTree();
  fNuEvent->Reset();
  return *fNuEvent;
}

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

NuMCEvent& NuOutputWriter::GetNuMCEventToFill()
{
  if (!fNuMCEvent) this->SetupNuMCEventTree();
  fNuMCEvent->Reset();
  return *fNuMCEvent;
}

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

TFile* NuOutputWriter::OpenFile(const NuConfig& config,
                                std::string prefix) const
{
  //call this static function to ensure any histograms created
  //after the file is opened are added
  TH1::AddDirectory(true);

  //create the tfile pointer to be returned
  TFile* outputFile=0;
  
  //get the environmental variable
  char *anaDir=getenv("NUANA_DIR");
  
  //use a string to hold env instead 
  string sAnaDir="";
  
  if (anaDir!=NULL) {
    sAnaDir=anaDir;
  }
  else {
    MSG("NuOutputWriter",Msg::kInfo) 
      <<"Environmental variable $NUANA_DIR not set." 
      <<" Writing file(s) to current directory"<<endl;
    sAnaDir=".";
  }
  
  //convert variables to string
  string sRunNumber=Form("%d",config.run);
  string sZeros="";
  if (config.run>=0 && config.run<10) sZeros="00000000";
  else if (config.run>=10 && config.run<100) sZeros="000000";
  else if (config.run>=100 && config.run<1000) sZeros="00000";
  else if (config.run>=1000 && config.run<10000) sZeros="0000";
  else if (config.run>=10000 && config.run<100000) sZeros="000";
  else if (config.run>=100000 && config.run<1000000) sZeros="00";
  else if (config.run>=1000000 && config.run<10000000) sZeros="0";
  else if (config.run>=10000000 && config.run<100000000) sZeros="";
  sRunNumber=sZeros+sRunNumber;

  string sDetector="UnknownDet";
  if (config.detector==Detector::kNear) {
    sDetector="N";
    if (config.simFlag==SimFlag::kMC) sDetector="n";
  }
  else if (config.detector==Detector::kFar) {
    sDetector="F";
    if (config.simFlag==SimFlag::kMC) sDetector="f";
  }
  else if (config.detector==Detector::kCalDet) {
    sDetector="C";
    if (config.simFlag==SimFlag::kMC) sDetector="c";
  }
  else cout<<"Ahhh, don't know detector="<<config.detector<<endl;
  
  string sPrefix="";//default
  if (prefix!="") sPrefix+=prefix;
  string sBase=sAnaDir+"/"+sPrefix;

    if (ReleaseType::IsDaikon(config.releaseType)
        && ReleaseType::GetMCSubVersion(config.releaseType)>=7){
      sBase += Form("r%d",config.runPeriod);
    }
    if (SimFlag::kMC == config.simFlag
        && Detector::kNear == config.detector
        && ReleaseType::IsDogwood(config.releaseType)
        && ReleaseType::GetRecoSubVersion(config.releaseType)>=1){
      sBase += Form("i%d",config.intensity);
    }

  sBase += sDetector;
  sBase += sRunNumber;
  string sFileName=sBase+".root";
  
  //test if file already exists
  ifstream Test(sFileName.c_str());
  
  //open the appropriate file
  if(!Test){
    outputFile=new TFile(sFileName.c_str(),"RECREATE");
  }
  else {
    //Need new filename
    Int_t fred=1;
    while(Test) {
      Test.close();
      string sAppendage=Form("%d",fred);
      sFileName=sBase+"_"+sAppendage+".root";
      Test.open(sFileName.c_str());
      fred++;
    }
    outputFile=new TFile(sFileName.c_str(),"NEW");
    outputFile->SetCompressionLevel(9);
  }
  
  string sTmp="No File!";
  if (outputFile) sTmp=outputFile->GetName();

  MSG("NuOutputWriter",Msg::kInfo) 
    <<"Output file opened: "<<sTmp<<endl;
  return outputFile;
}

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

TFile* NuOutputWriter::OpenFile(Int_t run,Int_t subrun,
                               std::string prefix) const
{
  //call this static function to ensure any histograms created
  //after the file is opened are added
  TH1::AddDirectory(true);

  //create the tfile pointer to be returned
  TFile* outputFile=0;
  
  //get the environmental variable
  char *anaDir=getenv("MEUANA_DIR");
  
  //use a string to hold env instead 
  string sAnaDir="";
  
  if (anaDir!=NULL) {
    sAnaDir=anaDir;
  }
  else {
    MSG("NuOutputWriter",Msg::kInfo) 
      <<"Environmental variable $MEUANA_DIR not set." 
      <<" Writing file(s) to current directory"<<endl;
    sAnaDir=".";
  }
  
  //convert variables to string
  string sRunNumber=Form("%d",run);
  string sSubrunNumber=Form("%d",subrun);
  //string sDetector="C";
  string sDetector="";
  //string sPrefix="h";//default
  string sPrefix="";//default
  if (prefix!="") sPrefix+=prefix;
  string sBase=sAnaDir+"/"+sPrefix+sDetector+sRunNumber;
  sBase+="_"+sSubrunNumber;
  string sFileName=sBase+".root";
  
  //test if file already exists
  ifstream Test(sFileName.c_str());
  
  //open the appropriate file
  if(!Test){
    outputFile=new TFile(sFileName.c_str(),"RECREATE");
  }
  else {
    //Need new filename
    Int_t fred=1;
    while(Test) {
      Test.close();
      string sAppendage=Form("%d",fred);
      sFileName=sBase+"_"+sAppendage+".root";
      Test.open(sFileName.c_str());
      fred++;
    }
    outputFile=new TFile(sFileName.c_str(),"NEW");
    outputFile->SetCompressionLevel(9);
  }
  
  string sTmp="No File!";
  if (outputFile) sTmp=outputFile->GetName();

  MSG("NuOutputWriter",Msg::kInfo) 
    <<"Output file opened: "<<sTmp<<endl;
  return outputFile;
}

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

void NuOutputWriter::FillNuEventTree()
{
  if (!fNuEvent) {
    MSG("NuOutputWriter",Msg::kError)
      <<"No fNuEvent object to put in tree"<<endl;
  }
  
  MSG("NuOutputWriter",Msg::kDebug)<<"Filling NuEvent Tree..."<<endl; 
  
  static Int_t counter=0;
  fNuEvent->index=counter;//a counter
  counter++;//count the number of entries
  
  //fill the tree
  fNuEventTree->Fill();
}

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

void NuOutputWriter::FillNuMCEventTree()
{
  if (!fNuMCEvent) {
    MSG("NuOutputWriter",Msg::kError)
      <<"No fNuMCEvent object to put in tree"<<endl;
  }
  
  MSG("NuOutputWriter",Msg::kDebug)<<"Filling NuMCEvent Tree..."<<endl; 
  
  static Int_t counter=0;
  fNuMCEvent->index=counter;//a counter
  counter++;//count the number of entries
  
  //fill the tree
  fNuMCEventTree->Fill();
}

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

---