MeuSummaryWriter.cxx

Go to the documentation of this file.

// Coded by Jeff Hartnell Jul/2005 onwards
//
// Contact: j.j.hartnell@rl.ac.uk

#include <cstdlib>
#include <fstream>

#include "TFile.h"
#include "TTree.h"
#include "TClonesArray.h"

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

#include "MeuCal/MeuHitInfo.h"
#include "MeuCal/MeuSummary.h"
#include "MeuCal/MeuSummaryWriter.h"

using std::endl;
using std::cout;
using std::map;
using std::vector;

CVSID("$Id: MeuSummaryWriter.cxx,v 1.4 2009/02/28 21:46:14 gmieg Exp $");

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

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

  MSG("MeuSummaryWriter",Msg::kDebug)
      <<"Finished MeuSummaryWriter Constructor"<<endl;
}

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

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

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

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

void MeuSummaryWriter::SummaryTreeSetup(Int_t run,Int_t subrun)
{
  fSummaryFile=this->OpenFile(run,subrun,"MeuSummary");
  fSummaryTree=new TTree("s","s");
  cout<<"Creating new MeuSummary for output to tree"<<endl;
  fMeuSummary=new MeuSummary();
  fSummaryTree->Branch("s","MeuSummary",&fMeuSummary,32000,2);
}

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

void MeuSummaryWriter::SummaryTreeFinish()
{
  MSG("MeuSummaryWriter",Msg::kInfo)
    <<"Running MeuSummaryWriter::SummaryTreeFinish..."<<endl;

  MSG("MeuSummaryWriter",Msg::kInfo)
    <<"Changing directory to fSummaryFile..."<<endl;
  //cd to file
  fSummaryFile->cd();

  MSG("MeuSummaryWriter",Msg::kInfo)<<"Writing tree..."<<endl;
  //write tree
  //fSummaryTree->Write();//use file->Write below instead

  //write data to 
  fSummaryFile->Write();//this should pick up any histograms as well

  MSG("MeuSummaryWriter",Msg::kInfo)
    <<"Closing file "<<fSummaryFile->GetName()<<" ..."<<endl;
  //close the file
  fSummaryFile->Close();

  MSG("MeuSummaryWriter",Msg::kInfo)
    <<"Completed MeuSummaryWriter::SummaryTreeFinish"<<endl;
}

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

MeuSummary& MeuSummaryWriter::GetMeuSummaryToFill()
{
    fMeuSummary->Reset();
    return *fMeuSummary;
}

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

TFile* MeuSummaryWriter::OpenFile(Int_t run,Int_t subrun,
                               std::string prefix) const
{
  //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("MeuSummaryWriter",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("MeuSummaryWriter",Msg::kInfo) 
    <<"Output file opened: "<<sTmp<<endl;
  return outputFile;
}

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

void MeuSummaryWriter::SummaryTreeFill
(const std::map<Int_t,MeuHitInfo>& plInfo)
{
  if (!fMeuSummary) cout<<"Ahhhhh no fMeuSummary object to put in tree"<<endl;

  MSG("MeuSummaryWriter",Msg::kDebug)<<"Filling summaryTree..."<<endl; 

  static Int_t counter=0;
  fMeuSummary->Count=counter;//a counter
  counter++;//count the number of entries

  this->FillMeuHitInfo(plInfo);
  
  //fill the tree
  fSummaryTree->Fill();
}

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

void MeuSummaryWriter::FillMeuHitInfo
(const std::map<Int_t,MeuHitInfo>& plInfo)
{
  MSG("MeuSummaryWriter",Msg::kDebug)<<"   filling calibPos..."<<endl; 
  
  MeuSummary& s=*fMeuSummary;
  
  //only fill if a window was found
  if (s.WinSigCor<=0) return;
  
  //get a reference to the MeuHitInfo object in MeuSummary
  TClonesArray& calibPos = *(s.MeuHitInfo);

  //counter for TClonesArray
  Int_t i=0;
  
  for (map<Int_t,MeuHitInfo>::const_iterator it=plInfo.begin();
       it!=plInfo.end();++it){

    Int_t plane=it->second.Plane;
    if (plane<0) continue;

    //check if plane is in window
    if ((plane>=s.WinVtxSidePl && plane<=s.WinStopSidePl) ||
        (plane<=s.WinVtxSidePl && plane>=s.WinStopSidePl)) {
      
      if (plane>120 && s.Detector==Detector::kNear) cout
        <<"plane>120!!!!!!!, pl="<<plane<<endl;
      
      MeuHitInfo* cp=new(calibPos[i++]) MeuHitInfo();
      *cp=it->second;
      
      MAXMSG("MeuSummaryWriter",Msg::kVerbose,1000)
        <<"Filling MeuHitInfo: pl="<<cp->Plane
        <<", sigMap="<<cp->SigMap<<endl;
    }
  }
}

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

---