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;   //!pointer to the analyzed TChain or TChain
  Int_t           fCurrent; //!current Tree number in a TChain

  Int_t runnum;
  Int_t subrunnum;
  Int_t evtnum;
  Double_t spillPOT;
  Int_t Nslices;
  Int_t goodSpill;
  Double_t horncurrent;
  Double_t posX;
  Double_t posY;
  Double_t widthX;
  Double_t widthY;
  Int_t deltaspilltimensec;

  Int_t evt_year;
  Int_t evt_month;
  Int_t evt_day;
  Int_t evt_hour;
  Int_t evt_min;
  Int_t evt_sec;

  Int_t Ndeaddcm;
  Int_t Ndb1apd_24cells;
  Int_t Ndb2apd_24cells;
  Int_t Ndb3apd_24cells;
  Int_t Nmcapd_24cells;
  Int_t Ndb1apd_28cells;
  Int_t Ndb2apd_28cells;
  Int_t Ndb3apd_28cells;
  Int_t Nmcapd_28cells;
  Int_t Nouttimehits;
  Int_t Nintimehits;
  Float_t Fouttimehits_noisyapd;
   
  readNtuple(const char* TypeNameIn, const char* TypeNameOut, TChain *tree=0);
  virtual ~readNtuple();
  virtual void     Init(TChain *tree);
  virtual void     Loop(const char* TypeNameOut);
  makeHists* myhists;
};

#endif

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

  Init(tree);
  myhists = new makeHists();
  TString histoname = (TString)TypeNameOut + ".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("runnum",&runnum);
   fChain->SetBranchAddress("subrunnum",&subrunnum);
   fChain->SetBranchAddress("evtnum",&evtnum);
   fChain->SetBranchAddress("Nslices",&Nslices);
   fChain->SetBranchAddress("spillPOT",&spillPOT);
   fChain->SetBranchAddress("goodSpill",&goodSpill);
   
   fChain->SetBranchAddress("horncurrent",&horncurrent);
   fChain->SetBranchAddress("posX",&posX);
   fChain->SetBranchAddress("posY",&posY);
   fChain->SetBranchAddress("widthX",&widthX);
   fChain->SetBranchAddress("widthY",&widthY);
   fChain->SetBranchAddress("deltaspilltimensec",&deltaspilltimensec);

   fChain->SetBranchAddress("evt_year",&evt_year);
   fChain->SetBranchAddress("evt_month",&evt_month);
   fChain->SetBranchAddress("evt_day",&evt_day);
   fChain->SetBranchAddress("evt_hour",&evt_hour);
   fChain->SetBranchAddress("evt_min",&evt_min);
   fChain->SetBranchAddress("evt_sec",&evt_sec);

   fChain->SetBranchAddress("Ndeaddcm",&Ndeaddcm);
   fChain->SetBranchAddress("Nintimehits",&Nintimehits);
   fChain->SetBranchAddress("Nouttimehits",&Nouttimehits);
   fChain->SetBranchAddress("Ndb1apd_24cells",&Ndb1apd_24cells);
   fChain->SetBranchAddress("Ndb2apd_24cells",&Ndb2apd_24cells);
   fChain->SetBranchAddress("Ndb3apd_24cells",&Ndb3apd_24cells);
   fChain->SetBranchAddress("Nmcapd_24cells",&Nmcapd_24cells);
   fChain->SetBranchAddress("Ndb1apd_28cells",&Ndb1apd_28cells);
   fChain->SetBranchAddress("Ndb2apd_28cells",&Ndb2apd_28cells);
   fChain->SetBranchAddress("Ndb3apd_28cells",&Ndb3apd_28cells);
   fChain->SetBranchAddress("Nmcapd_28cells",&Nmcapd_28cells);
   fChain->SetBranchAddress("Fouttimehits_noisyapd",&Fouttimehits_noisyapd);
}
#endif // #ifdef readNtuple_cxx