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
  

  Int_t sl_runnum;
  Int_t sl_subrunnum;
  Int_t sl_evtnum;
  Int_t sl_goodSpill;
  Int_t sl_Nslices;
  Double_t sl_spillPOT;
  Int_t sliceIndex;

  Float_t weight;

  //spill quality
  Int_t sl_Ndeaddcm;

  //beam quality
  Double_t sl_horncurrent;
  Double_t sl_posX;
  Double_t sl_posY;
  Double_t sl_widthX;
  Double_t sl_widthY;
  Int_t sl_deltaspilltimensec;
   
  /*** Vertex ***/
  Float_t Vx;//true vertex X
  Float_t Vy;//true vertex Y
  Float_t Vz;//true vertex Z
  Float_t recoVx;//reco vertex X
  Float_t recoVy;//reco vertex Y
  Float_t recoVz;//reco vertex Z

  /*** true info ***/
  Float_t truePt;//true Pt
  Float_t trueE;//true E
  Float_t trueTheta;//true theta
  Int_t ccnc;
  Int_t PDG;
  Int_t origPDG;
  Int_t mode;
  bool isQE;
  Int_t intType;
  Int_t hitNucl;
  Float_t HadX;
  Float_t HadY;
  Float_t HadW;
  Float_t qsqr;
  Float_t OscWt; 
  
  Int_t lepPDG;
  Float_t lepPx;
  Float_t lepPy;
  Float_t lepPz;
  Float_t lepE;
  Int_t mesonPDG;
  Float_t mesonPx;
  Float_t mesonPy;
  Float_t mesonPz;
  
  //for pi0->2gamma from most energetic pi0
  Int_t HasPi0;
  Int_t HasP2GG;//has pi0->2gamma
  Int_t PMPDG;//pdg code for pi0 mother
  Int_t Ndaus;//# of pi0 daughters
  Int_t NdausPDG[5];//pdg code for pi0 daughters
  Float_t Epi0;                   
  Float_t gam1E;
  Float_t gam1Px;
  Float_t gam1Py;
  Float_t gam1Pz;
  Float_t gam2E;
  Float_t gam2Px;
  Float_t gam2Py;
  Float_t gam2Pz;

  //all pi0, first 20 if possible ?
  Int_t Npi0;
  Float_t pi0Px[20];
  Float_t pi0Py[20];
  Float_t pi0Pz[20];
  Float_t pi0E[20];
  Float_t pi0Vx[20];
  Float_t pi0Vy[20];
  Float_t pi0Vz[20];

  /*** slicer information ***/
  Int_t Ncells;//# of associate cells
  Int_t Nplanes;//# of extended planes
  Float_t recoE;//reco energy
  Int_t MinPlane;//min plane index

    /*** longest kalman track ***/
    Float_t KtrackLength;//length of longest track
    Int_t KtrackNcells;//# of cells associated to longest track
    Float_t KtrackEnergy;//energy associated to longest track
    Float_t KtrackNcells_ratio;//# of cells associated to longest track over Ncells
    Float_t KtrackEnergy_ratio;//energy associated to longest track over recoE
    Float_t KtrackNplanes;
    Float_t KtrackTheta;
    Float_t KtrackPhi;
    Float_t KtrackStartX;
    Float_t KtrackStartY;
    Float_t KtrackStartZ;
    Float_t KtrackStopX;
    Float_t KtrackStopY;
    Float_t KtrackStopZ;
    Int_t NKtracks;//# of tracks

    Int_t trackcellNcell;
    Float_t trackcellX[500];
    Float_t trackcellY[500];
    Float_t trackcellZ[500];
    Float_t trackcellE[500];
    Int_t trackcellADC[500];
    Int_t trackcellPlane[500];
    Int_t trackcellCell[500];
    Float_t trackcellPECorr[500];
    Float_t trackcellW[500];

    /*** mip information ***/
    Float_t Nmip;//# of mip cells
    Float_t Fmip;//# of mip cells over Ncells

    /***slice associated cells information ***/
    Int_t slcellNcell; 
    Float_t slcellX[500];
    Float_t slcellY[500];
    Float_t slcellZ[500];
    Float_t slcellE[500];
    Int_t slcellADC[500];
    Int_t slcellPlane[500];
    Int_t slcellCell[500];
    Float_t slcellPECorr[500];
    Float_t slcellW[500];

    /*** prong information ***/
    Int_t Nprongs;
    Float_t Ebalance;
    Float_t Dphi;

    Float_t prongEnergy[20];
    Float_t prongPhi[20];
    Float_t prongTheta[20];
    Float_t prongStartX[20];
    Float_t prongStartY[20];
    Float_t prongStartZ[20];
    Float_t prongLength[20];
    Float_t prongStopX[20];
    Float_t prongStopY[20];
    Float_t prongStopZ[20];
    Float_t prongEnergyXView[20];//energy for XView cells
    Float_t prongEnergyYView[20];//energy for YView cells
    Int_t prongNCells[20];//# of cells
    Int_t prongNCellsXView[20];//# of cells for XView
    Int_t prongNCellsYView[20];//# of cells for YView
    Int_t prongNplanes[20];//# of extent planes

    //all prong associated cells
    Int_t allprongcellNcell;
    Int_t allprongIndex[500];
    Float_t allprongcellX[500];
    Float_t allprongcellY[500];
    Float_t allprongcellZ[500];
    Float_t allprongcellE[500];
    Int_t allprongcellADC[500];
    Int_t allprongcellPlane[500];
    Int_t allprongcellCell[500];
    Float_t allprongcellPECorr[500];
    Float_t allprongcellW[500];

    //leading prong
    Int_t prong1cellNcell;
    Float_t prong1Fmip;
    Float_t prong1cellX[500];
    Float_t prong1cellY[500];
    Float_t prong1cellZ[500];
    Float_t prong1cellE[500];
    Int_t prong1cellADC[500];
    Int_t prong1cellPlane[500];
    Int_t prong1cellCell[500];
    Float_t prong1cellPECorr[500];
    Float_t prong1cellW[500];
 
    //2D prongs
    Int_t Nprongs2D;
    Int_t prong2DInXView[20];//prong in X View or not
    Float_t prongEnergy2D[20];//energy
    Int_t prongNCells2D[20];//# of cells
    Float_t Ebalance2D;
    Float_t prongLength2D[20];
    Float_t prongTheta2D[20];
    Float_t prongStartX2D[20];
    Float_t prongStartY2D[20];
    Float_t prongStartZ2D[20];
    Float_t prongStopX2D[20];
    Float_t prongStopY2D[20];
    Float_t prongStopZ2D[20];

    Int_t allprong2DcellNcell;
    Int_t allprong2DIndex[500];
    Float_t allprong2DcellX[500];
    Float_t allprong2DcellY[500];
    Float_t allprong2DcellZ[500];
    Float_t allprong2DcellE[500];
    Int_t allprong2DcellADC[500];
    Int_t allprong2DcellPlane[500];
    Int_t allprong2DcellCell[500];
    Float_t allprong2DcellPECorr[500];

    Float_t rvp;
    Float_t lid;
    Float_t lide;
    Float_t lem;
    Int_t Nshowers;
    Float_t shwLID[20];
    Int_t shw3D[20];
    Int_t shwNCells[20];
    Int_t shwNCellsXview[20];
    Int_t shwNCellsYview[20];
    Float_t shwLength[20];
    Float_t shwStartX[20];
    Float_t shwStartY[20];
    Float_t shwStartZ[20];
    Float_t shwStopX[20];
    Float_t shwStopY[20];
    Float_t shwStopZ[20];
    Float_t shwTheta[20];
    Float_t shwEcells[20];
    Float_t shwEnu[20];

    Int_t shwcellNcell;
    Int_t shwIndex[500];
    Float_t shwcellX[500];
    Float_t shwcellY[500];
    Float_t shwcellZ[500];
    Float_t shwcellE[500];
    Int_t shwcellADC[500];
    Int_t shwcellPlane[500];
    Int_t shwcellCell[500];
    Float_t shwcellPECorr[500];
    Float_t shwcellW[500];

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

#endif

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

  Init(tree);
  myhists = new makeHists();
  TString histoname = (TString)TypeNameOut + "_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("sl_runnum",&sl_runnum);
   fChain->SetBranchAddress("sl_subrunnum",&sl_subrunnum);
   fChain->SetBranchAddress("sl_evtnum",&sl_evtnum);
   fChain->SetBranchAddress("sl_Nslices",&sl_Nslices);
   fChain->SetBranchAddress("sl_goodSpill",&sl_goodSpill);
   fChain->SetBranchAddress("sl_spillPOT",&sl_spillPOT);
   fChain->SetBranchAddress("sliceIndex",&sliceIndex);

   fChain->SetBranchAddress("sl_Ndeaddcm",&sl_Ndeaddcm);

   fChain->SetBranchAddress("sl_horncurrent",&sl_horncurrent);
   fChain->SetBranchAddress("sl_posX",&sl_posX);
   fChain->SetBranchAddress("sl_posY",&sl_posY);
   fChain->SetBranchAddress("sl_widthX",&sl_widthX);
   fChain->SetBranchAddress("sl_widthY",&sl_widthY);
   fChain->SetBranchAddress("sl_deltaspilltimensec",&sl_deltaspilltimensec);

   fChain->SetBranchAddress("Vx",&Vx);
   fChain->SetBranchAddress("Vy",&Vy);
   fChain->SetBranchAddress("Vz",&Vz);
   fChain->SetBranchAddress("recoVx",&recoVx);
   fChain->SetBranchAddress("recoVy",&recoVy);
   fChain->SetBranchAddress("recoVz",&recoVz);

   fChain->SetBranchAddress("truePt",&truePt);
   fChain->SetBranchAddress("trueE",&trueE);
   fChain->SetBranchAddress("trueTheta",&trueTheta);
   fChain->SetBranchAddress("ccnc",&ccnc);
   fChain->SetBranchAddress("PDG",&PDG);
   fChain->SetBranchAddress("origPDG",&origPDG);
   fChain->SetBranchAddress("mode",&mode);
   fChain->SetBranchAddress("isQE",&isQE);
   fChain->SetBranchAddress("intType",&intType);
   fChain->SetBranchAddress("hitNucl",&hitNucl);
   fChain->SetBranchAddress("HadX",&HadX);
   fChain->SetBranchAddress("HadY",&HadY);
   //fChain->SetBranchAddress("HadW",&HadW);
   fChain->SetBranchAddress("qsqr",&qsqr);
   fChain->SetBranchAddress("OscWt",&OscWt);

   fChain->SetBranchAddress("lepPDG",&lepPDG);
   fChain->SetBranchAddress("lepE",&lepE);
   fChain->SetBranchAddress("lepPx",&lepPx);
   fChain->SetBranchAddress("lepPy",&lepPy);
   fChain->SetBranchAddress("lepPz",&lepPz);
   fChain->SetBranchAddress("mesonPDG",&mesonPDG);
   fChain->SetBranchAddress("mesonPx",&mesonPx);
   fChain->SetBranchAddress("mesonPy",&mesonPy);
   fChain->SetBranchAddress("mesonPz",&mesonPz);

   fChain->SetBranchAddress("Ncells",&Ncells);
   fChain->SetBranchAddress("Nplanes",&Nplanes);
   fChain->SetBranchAddress("recoE",&recoE);
   fChain->SetBranchAddress("MinPlane",&MinPlane);

   fChain->SetBranchAddress("slcellNcell",&slcellNcell);
   fChain->SetBranchAddress("slcellX",&slcellX);
   fChain->SetBranchAddress("slcellY",&slcellY);
   fChain->SetBranchAddress("slcellZ",&slcellZ);
   fChain->SetBranchAddress("slcellE",&slcellE);
   fChain->SetBranchAddress("slcellADC",&slcellADC);
   fChain->SetBranchAddress("slcellPlane",&slcellPlane);
   fChain->SetBranchAddress("slcellCell",&slcellCell);
   fChain->SetBranchAddress("slcellPECorr",&slcellPECorr);
   fChain->SetBranchAddress("slcellW",&slcellW);

   fChain->SetBranchAddress("KtrackLength",&KtrackLength);
   fChain->SetBranchAddress("KtrackEnergy",&KtrackEnergy);
   fChain->SetBranchAddress("KtrackNcells_ratio",&KtrackNcells_ratio);
   fChain->SetBranchAddress("KtrackNcells",&KtrackNcells);
   fChain->SetBranchAddress("KtrackEnergy_ratio",&KtrackEnergy_ratio);
   fChain->SetBranchAddress("KtrackNplanes",&KtrackNplanes);
   fChain->SetBranchAddress("KtrackTheta",&KtrackTheta);
   fChain->SetBranchAddress("KtrackPhi",&KtrackPhi);
   fChain->SetBranchAddress("KtrackStartX",&KtrackStartX);
   fChain->SetBranchAddress("KtrackStartY",&KtrackStartY);
   fChain->SetBranchAddress("KtrackStartZ",&KtrackStartZ);
   fChain->SetBranchAddress("KtrackStopX",&KtrackStopX);
   fChain->SetBranchAddress("KtrackStopY",&KtrackStopY);
   fChain->SetBranchAddress("KtrackStopZ",&KtrackStopZ);
   fChain->SetBranchAddress("NKtracks",&NKtracks);

   fChain->SetBranchAddress("trackcellNcell",&trackcellNcell);
   fChain->SetBranchAddress("trackcellX",&trackcellX);
   fChain->SetBranchAddress("trackcellY",&trackcellY);
   fChain->SetBranchAddress("trackcellZ",&trackcellZ);
   fChain->SetBranchAddress("trackcellE",&trackcellE);
   fChain->SetBranchAddress("trackcellADC",&trackcellADC);
   fChain->SetBranchAddress("trackcellPlane",&trackcellPlane);
   fChain->SetBranchAddress("trackcellCell",&trackcellCell);
   fChain->SetBranchAddress("trackcellPECorr",&trackcellPECorr);
   fChain->SetBranchAddress("trackcellW",&trackcellW);

   fChain->SetBranchAddress("Nmip",&Nmip);
   fChain->SetBranchAddress("Fmip",&Fmip);

   fChain->SetBranchAddress("Nprongs",&Nprongs);
   fChain->SetBranchAddress("Ebalance",&Ebalance);
   fChain->SetBranchAddress("Dphi",&Dphi);
 
   fChain->SetBranchAddress("prongEnergy",&prongEnergy);
   fChain->SetBranchAddress("prongPhi",&prongPhi);
   fChain->SetBranchAddress("prongTheta",&prongTheta);
   fChain->SetBranchAddress("prongLength",&prongLength);
   fChain->SetBranchAddress("prongStartX",&prongStartX);
   fChain->SetBranchAddress("prongStartY",&prongStartY);
   fChain->SetBranchAddress("prongStartZ",&prongStartZ);
   fChain->SetBranchAddress("prongStopX",&prongStopX);
   fChain->SetBranchAddress("prongStopY",&prongStopY);
   fChain->SetBranchAddress("prongStopZ",&prongStopZ);
   fChain->SetBranchAddress("prongEnergyXView",&prongEnergyXView);
   fChain->SetBranchAddress("prongEnergyYView",&prongEnergyYView);
   fChain->SetBranchAddress("prongNCells",&prongNCells);
   fChain->SetBranchAddress("prongNCellsXView",&prongNCellsXView);
   fChain->SetBranchAddress("prongNCellsYView",&prongNCellsYView);
   fChain->SetBranchAddress("prongNplanes",&prongNplanes);

   fChain->SetBranchAddress("allprongcellNcell",&allprongcellNcell);
   fChain->SetBranchAddress("allprongIndex",&allprongIndex);
   fChain->SetBranchAddress("allprongcellX",&allprongcellX);
   fChain->SetBranchAddress("allprongcellY",&allprongcellY);
   fChain->SetBranchAddress("allprongcellZ",&allprongcellZ);
   fChain->SetBranchAddress("allprongcellE",&allprongcellE);
   fChain->SetBranchAddress("allprongcellADC",&allprongcellADC);
   fChain->SetBranchAddress("allprongcellPlane",&allprongcellPlane);
   fChain->SetBranchAddress("allprongcellCell",&allprongcellCell);
   fChain->SetBranchAddress("allprongcellPECorr",&allprongcellPECorr);
   fChain->SetBranchAddress("allprongcellW",&allprongcellW);

   fChain->SetBranchAddress("prong1cellNcell",&prong1cellNcell);
   fChain->SetBranchAddress("prong1Fmip",&prong1Fmip);
   fChain->SetBranchAddress("prong1cellX",&prong1cellX);
   fChain->SetBranchAddress("prong1cellY",&prong1cellY);
   fChain->SetBranchAddress("prong1cellZ",&prong1cellZ);
   fChain->SetBranchAddress("prong1cellE",&prong1cellE);
   fChain->SetBranchAddress("prong1cellADC",&prong1cellADC);
   fChain->SetBranchAddress("prong1cellPlane",&prong1cellPlane);
   fChain->SetBranchAddress("prong1cellCell",&prong1cellCell);
   fChain->SetBranchAddress("prong1cellPECorr",&prong1cellPECorr);
   fChain->SetBranchAddress("prong1cellW",&prong1cellW);

   fChain->SetBranchAddress("Nprongs2D",&Nprongs2D);
   fChain->SetBranchAddress("Ebalance2D",&Ebalance2D);
   fChain->SetBranchAddress("prong2DInXView",&prong2DInXView);
   fChain->SetBranchAddress("prongEnergy2D",&prongEnergy2D);
   fChain->SetBranchAddress("prongNCells2D",&prongNCells2D);
   fChain->SetBranchAddress("prongLength2D",&prongLength2D);
   fChain->SetBranchAddress("prongTheta2D",&prongTheta2D);
   fChain->SetBranchAddress("prongStartX2D",&prongStartX2D);
   fChain->SetBranchAddress("prongStartY2D",&prongStartY2D);
   fChain->SetBranchAddress("prongStartZ2D",&prongStartZ2D);
   fChain->SetBranchAddress("prongStopX2D",&prongStopX2D);
   fChain->SetBranchAddress("prongStopY2D",&prongStopY2D);
   fChain->SetBranchAddress("prongStopZ2D",&prongStopZ2D);
  
   fChain->SetBranchAddress("allprong2DcellNcell",&allprong2DcellNcell);
   fChain->SetBranchAddress("allprong2DIndex",&allprong2DIndex);
   fChain->SetBranchAddress("allprong2DcellX",&allprong2DcellX);
   fChain->SetBranchAddress("allprong2DcellY",&allprong2DcellY);
   fChain->SetBranchAddress("allprong2DcellZ",&allprong2DcellZ);
   fChain->SetBranchAddress("allprong2DcellE",&allprong2DcellE);
   fChain->SetBranchAddress("allprong2DcellADC",&allprong2DcellADC);
   fChain->SetBranchAddress("allprong2DcellPlane",&allprong2DcellPlane);
   fChain->SetBranchAddress("allprong2DcellCell",&allprong2DcellCell);
   fChain->SetBranchAddress("allprong2DcellPECorr",&allprong2DcellPECorr);

   fChain->SetBranchAddress("lem",&lem);
   fChain->SetBranchAddress("lid",&lid);
   fChain->SetBranchAddress("lide",&lide);
   fChain->SetBranchAddress("rvp",&rvp);

   fChain->SetBranchAddress("Nshowers",&Nshowers);
   fChain->SetBranchAddress("shwLID",&shwLID);
   fChain->SetBranchAddress("shw3D",&shw3D);
   fChain->SetBranchAddress("shwNCells",&shwNCells);
   fChain->SetBranchAddress("shwNCellsXview",&shwNCellsXview);
   fChain->SetBranchAddress("shwNCellsYview",&shwNCellsYview);
   fChain->SetBranchAddress("shwLength",&shwLength);
   fChain->SetBranchAddress("shwStartX",&shwStartX);
   fChain->SetBranchAddress("shwStartY",&shwStartY);
   fChain->SetBranchAddress("shwStartZ",&shwStartZ);
   fChain->SetBranchAddress("shwStopX",&shwStopX);
   fChain->SetBranchAddress("shwStopY",&shwStopY);
   fChain->SetBranchAddress("shwStopZ",&shwStopZ);
   fChain->SetBranchAddress("shwTheta",&shwTheta);
   fChain->SetBranchAddress("shwEcells",&shwEcells);
   fChain->SetBranchAddress("shwEnu",&shwEnu);

   fChain->SetBranchAddress("shwcellNcell",&shwcellNcell);
   fChain->SetBranchAddress("shwIndex",&shwIndex);
   fChain->SetBranchAddress("shwcellX",&shwcellX);
   fChain->SetBranchAddress("shwcellY",&shwcellY);
   fChain->SetBranchAddress("shwcellZ",&shwcellZ);
   fChain->SetBranchAddress("shwcellE",&shwcellE);
   fChain->SetBranchAddress("shwcellADC",&shwcellADC);
   fChain->SetBranchAddress("shwcellPlane",&shwcellPlane);
   fChain->SetBranchAddress("shwcellCell",&shwcellCell);
   fChain->SetBranchAddress("shwcellPECorr",&shwcellPECorr);
   fChain->SetBranchAddress("shwcellW",&shwcellW);

   fChain->SetBranchAddress("weight",&weight);
}
#endif // #ifdef readNtuple_cxx