readNtuple.h

Go to the documentation of this file.

#ifndef readNtuple_h
#define readNtuple_h

#include <TROOT.h>
#include <TChain.h>
#include <TFile.h>
#include "makeHists.h"
#include "iostream"
#include "fstream"

#include "TMVA/Tools.h"
#include "TMVA/Reader.h"
#include "TMVA/MethodCuts.h"

using namespace std;

class readNtuple {
 public :
  
  TChain          *fChain;   //!poInt_ter to the analyzed TChain or TChain
  Int_t           fCurrent; //!current Tree number in a TChain
  
    float nu_Vx;//true vertex X
    float nu_Vy;//true vertex Y
    float nu_Vz;//true vertex Z
    float nu_truePx;//true Px
    float nu_truePy;//true Py
    float nu_truePz;//true Pz
    float nu_trueE;//true E
    int nu_ccnc;
    int nu_PDG;
    int nu_origPDG;
    int nu_orig_parentPDG;
    int nu_mode;
    int nu_intType;
    int nu_hitNucl;
    float nu_HadX;
    float nu_HadY;
    float nu_HadW;
    float nu_Q2;

    //lepton info
    int lep_PDG;
    float lep_Px;
    float lep_Py;
    float lep_Pz;
    float lep_E;

    //4-momentum for particles originating from the nu interaction vertex
    int Nparticles;
    int partPDG[100];
    float partPx[100];
    float partPy[100];
    float partPz[100];
    float partE[100];

    //GENIE reweights or cross section uncertainties
    int Nweights;
    float plus1sigma[68];
    float plus2sigma[68];
    float minus1sigma[68];
    float minus2sigma[68];  


  readNtuple(const char* TypeNameIn, const char* TypeNameNu, const char* TypeNameInt, TChain *tree=0);
  virtual ~readNtuple();
  virtual void     Init(TChain *tree);
  virtual void     Loop(const char* TypeNameNu, const char* TypeNameInt);
  makeHists* myhists;
};

#endif

#ifdef readNtuple_cxx
readNtuple::readNtuple(const char* TypeNameIn, const char* TypeNameNu, const char* TypeNameInt, TChain *tree)
{
   tree = new TChain("GenieTruth/GenieTruth");
   ifstream fin(TypeNameIn,ios::in);
   TString fstr;
   while(fin>>fstr) tree->Add(fstr);

  Init(tree);
  myhists = new makeHists();
  TString histoname = (TString)TypeNameNu+"_"+(TString)TypeNameInt + "_results.root";
  myhists->bookHists(histoname);

  cout<<"********************************************"<<endl;
  cout<<"Input File is "<<TypeNameIn<<endl;
  cout<<"Output File is "<<histoname<<endl;
  cout<<"********************************************"<<endl;
}

readNtuple::~readNtuple()
{
  if (!fChain) return;
  delete fChain->GetCurrentFile();

  if(myhists){
    myhists->saveHists();
    delete myhists;

  }
}

void readNtuple::Init(TChain *tree)
{
   if (!tree) return;
   fChain = tree;


   fChain->SetBranchAddress("nu_Vx",&nu_Vx);
   fChain->SetBranchAddress("nu_Vy",&nu_Vy);
   fChain->SetBranchAddress("nu_Vz",&nu_Vz);

   fChain->SetBranchAddress("nu_trueE",&nu_trueE);
   fChain->SetBranchAddress("nu_truePx",&nu_truePx);
   fChain->SetBranchAddress("nu_truePy",&nu_truePy);
   fChain->SetBranchAddress("nu_truePz",&nu_truePz);

   fChain->SetBranchAddress("nu_ccnc",&nu_ccnc);
   fChain->SetBranchAddress("nu_PDG",&nu_PDG);
   fChain->SetBranchAddress("nu_origPDG",&nu_origPDG);
   fChain->SetBranchAddress("nu_mode",&nu_mode);
   fChain->SetBranchAddress("nu_intType",&nu_intType);
   fChain->SetBranchAddress("nu_hitNucl",&nu_hitNucl);
   fChain->SetBranchAddress("nu_HadX",&nu_HadX);
   fChain->SetBranchAddress("nu_HadY",&nu_HadY);
   fChain->SetBranchAddress("nu_HadW",&nu_HadW);
   fChain->SetBranchAddress("nu_Q2",&nu_Q2);

   fChain->SetBranchAddress("lep_PDG",&lep_PDG);
   fChain->SetBranchAddress("lep_E",&lep_E);
   fChain->SetBranchAddress("lep_Px",&lep_Px);
   fChain->SetBranchAddress("lep_Py",&lep_Py);
   fChain->SetBranchAddress("lep_Pz",&lep_Pz);
  
   fChain->SetBranchAddress("Nparticles",&Nparticles);
   fChain->SetBranchAddress("partPDG",&partPDG);
   fChain->SetBranchAddress("partPx",&partPx);
   fChain->SetBranchAddress("partPy",&partPy);
   fChain->SetBranchAddress("partPz",&partPz);
   fChain->SetBranchAddress("partE",&partE);

   fChain->SetBranchAddress("Nweights",&Nweights);
   fChain->SetBranchAddress("plus1sigma",&plus1sigma);
   fChain->SetBranchAddress("plus2sigma",&plus2sigma);
   fChain->SetBranchAddress("minus1sigma",&minus1sigma);
   fChain->SetBranchAddress("minus2sigma",&minus2sigma);
}
#endif // #ifdef readNtuple_cxx