NuTreeWrapper.cxx

Go to the documentation of this file.

// $Id: NuTreeWrapper.cxx,v 1.1 2007/11/15 13:44:34 evans Exp $
//
// A wrapper to the tree s in NuDSTs to interface with the
// cross-secion fitting code
//
// Justin Evans
// j.evans2@physics.ox.ac.uk
//
//
// $Log: NuTreeWrapper.cxx,v $
// Revision 1.1  2007/11/15 13:44:34  evans
// NuTreeWrapper is a wrapper to a the tree of NuEvents found in the DSTs
// to give the right interface to my cross-section fitting extrapolation.
//
//

#include "TChainElement.h"
#include "TFile.h"
#include "TH1F.h"
#include "TObjArray.h"
#include "TTree.h"

#include "MessageService/MsgService.h"

#include "NtupleUtils/NuTreeWrapper.h"

using std::string;

CVSID("$Id: NuTreeWrapper.cxx,v 1.1 2007/11/15 13:44:34 evans Exp $");

ClassImp(NuTreeWrapper)

NuTreeWrapper::NuTreeWrapper()
  : IAmReadable(false),
    fcurrEntry(-1)
{
  //Constructor to create a writable tree
  fNuTreeWrapper = new TTree("s","s");
  cout<<"NuTreeWrapper::Creating new NuEvent for output to tree"<<endl;
  fNuEvent = new NuEvent();
  fNuTreeWrapper->Branch("s","NuEvent",&fNuEvent,32000,2);
  fInputInfo = 0;
}

//____________________________________________________________________72
NuTreeWrapper::
NuTreeWrapper(string filelist
           = "$SRT_PRIVATE_CONTEXT/NuAntiNu/PANs/DefaultPAN.root")
  : IAmReadable(true),
    fcurrEntry(-1),
    fInputChain("s")
{
  //Constructor to read a tree(s) from file
  cout << "Running constructor." << endl;
  fInputInfo = 0;
  fInputChain.Add(filelist.c_str());
  fInputChain.SetBranchAddress("s",&fInputInfo);
  fTotPoT = 0.0;

  this->CalculatePoTs();
}

//________________________________________________________________________
NuTreeWrapper::~NuTreeWrapper()
{
}

//____________________________________________________________________72
void NuTreeWrapper::CalculatePoTs()
{
  cout << "Running PoT function" << endl;
  TObjArray* fileList = fInputChain.GetListOfFiles();
  fTotPoT = 0.0;
  cout << "Starting fileLoop" << endl;
  Int_t detector = this->Detector();
  cout << "Got detector: " << detector << endl;
  for (Int_t counter=0; counter<fileList->GetEntries(); ++counter){
    TChainElement* chainElement =
      dynamic_cast<TChainElement*> (fileList->At(counter));
    if (chainElement){
      string fileName = chainElement->GetTitle();
      TFile f(fileName.c_str(),"READ");
      TH1F* hPottortgt = (TH1F*) f.Get("hPottortgt");
      TH1F* hRun = (TH1F*) f.Get("hRun");
      if (1==detector){
        fTotPoT += hPottortgt->GetMean()*hPottortgt->GetEntries()*1e12;
      }
      if (2==detector){
        fTotPoT += hPottortgt->GetMean()*hRun->GetEntries()*1e8*1e12;
      }
      fTotPoT += hPottortgt->Integral();
      f.Close();
    }
  }
    MSG("NuTreeWrapper.cxx",Msg::kInfo)
      << "PoT: " << fTotPoT << endl;
  return;
}

//____________________________________________________________________72
const Int_t NuTreeWrapper::Detector() const
{
  if (this->GetEntriesFast()){
    return this->GetInfoObject(0).detector;
  }
  else{
    MSG("NuTreeWrapper.cxx",Msg::kError)
      << "Trying to get PoT for empty NuTreeWrapper" << endl;
    return 0;
  }
}

//____________________________________________________________________72
void NuTreeWrapper::FillTree()
{
  //fill the tree
  fNuTreeWrapper->Fill();
}

//____________________________________________________________________72
const Int_t NuTreeWrapper::GetEntries() const
{
  if (!IAmReadable){
    MSG("NuTreeWrapper", Msg::kFatal)
      << "Trying to read from write-only NuTreeWrapper." << endl;
    return -1;
  }
  return fInputChain.GetEntries();
}

//____________________________________________________________________72
const Int_t NuTreeWrapper::GetEntriesFast() const
{
  if (!IAmReadable){
    MSG("NuTreeWrapper", Msg::kFatal)
      << "Trying to read from write-only NuTreeWrapper." << endl;
    return -1;
  }
  return fInputChain.GetEntriesFast();
}

//____________________________________________________________________72
const NuEvent NuTreeWrapper::GetInfoObject(const Int_t event) const
{
  if (!IAmReadable){
    MSG("NuTreeWrapper", Msg::kFatal)
      << "Trying to read from write-only NuTreeWrapper." << endl;
    NuEvent dummy;
    return dummy;
  }
  this->LoadEvent(event);
  return *fInputInfo;
}

//________________________________________________________________________
NuEvent& NuTreeWrapper::GetObjectToFill()
{
  fNuEvent->Reset();
  return *fNuEvent;
}

//____________________________________________________________________72
void NuTreeWrapper::LoadEvent(const Int_t event) const
{
  if (event == fcurrEntry){return;}
  else{
    fInputChain.GetEvent(event);
    fcurrEntry = event;
  }
}

//____________________________________________________________________72
void NuTreeWrapper::Write(const string filename)
{
  TFile f(filename.c_str(),"RECREATE");
  fNuTreeWrapper->Write();
  f.Close();
}

---