ancestor.h

Go to the documentation of this file.

//////////////////////////////////////////////////////////
// Created by L. Fields
//////////////////////////////////////////////////////////

#ifndef ancestor_h
#define ancestor_h

// C++
#include <string>
#include <vector>
#include <fstream>
#include <iostream>
#include <iomanip>
#include <sstream>

//ROOT
#include <TROOT.h>
#include <TChain.h>
#include <TFile.h>
#include <TRandom3.h>

// Dk2nu
//#ifndef __CINT__
#include "dk2nu/tree/dk2nu.h"
//#endif // ifndef __ CINT__

class ancestor {
public :

   TChain          *fChain;   //!pointer to the analyzed TTree or TChain
   //TTree          *fChain;   //!pointer to the analyzed TTree or TChain
   Int_t           fCurrent; //!current Tree number in a TChain
   
   Double_t        fTotalPOT; //total pot used for all files
   std::string     ffilename; //filename for saving histograms

   std::string detectorname;
   double detx;     // detector location
   double dety;
   double detz;

   TRandom3 *rand3;

   // List of branches
   bsim::Dk2Nu* dk2nu;

   ancestor();

   virtual ~ancestor();
   virtual Int_t    Cut(Long64_t entry);
   virtual Int_t    GetEntry(Long64_t entry);
   virtual Long64_t LoadTree(Long64_t entry);
   virtual void     Init(TTree *tree);
   virtual void     Loop();
   virtual Bool_t   Notify();
   virtual void     Show(Long64_t entry = -1);

 private:

   
};

#endif

#ifdef ancestor_cxx

ancestor::ancestor()
{

  /*
  // Otherwise Set detector coordinates based on requested detector location
  if(location=="LBNEND") {
    detx = 0.0;
    dety = 0.0;
    detz = 57400.0;
  }
  
  else if(location=="LBNEFD") {
    detx = 0.0;
    dety = 0.0;
    detz = 129700000.0;
  }
  */

  std::vector<std::string> fFileVec;

  fFileVec.push_back("/pnfs/dune/persistent/users/ljf26/fluxfiles/g4lbne/v3r5p4/QGSP_BERT/OptimizedEngineeredNov2017/neutrino/flux/dk2nu/g4lbne_v3r5p4_QGSP_BERT_OptimizedEngineeredNov2017_neutrino_00250.dk2nu.root")


  std::string ntuple_name = "dk2nuTree";


  fChain = new TChain(ntuple_name.c_str());
   for(std::vector<std::string>::const_iterator sit = fFileVec.begin(); sit != fFileVec.end(); ++sit)
   {
     fChain -> Add(sit -> c_str());
   }

   Init(fChain);

   // initialize random numbers used for oscillation calculations
   rand3 = new TRandom3(0); 

}

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

Int_t ancestor::GetEntry(Long64_t entry)
{
// Read contents of entry.
   if (!fChain) return 0;
   return fChain->GetEntry(entry);
}
Long64_t ancestor::LoadTree(Long64_t entry)
{
// Set the environment to read one entry
   if (!fChain) return -5;
   Long64_t centry = fChain->LoadTree(entry);
   if (centry < 0) return centry;
   if (!fChain->InheritsFrom(TChain::Class()))  return centry;
   TChain *chain = (TChain*)fChain;
   if (chain->GetTreeNumber() != fCurrent) {
      fCurrent = chain->GetTreeNumber();
      Notify();
   }
   return centry;
}

void ancestor::Init(TTree *tree)
{
   // The Init() function is called when the selector needs to initialize
   // a new tree or chain. Typically here the branch addresses and branch
   // pointers of the tree will be set.
   // It is normally not necessary to make changes to the generated
   // code, but the routine can be extended by the user if needed.
   // Init() will be called many times when running on PROOF
   // (once per file to be processed).

   // Set branch addresses and branch pointers

   fCurrent = -1;
   
   dk2nu = new bsim::Dk2Nu;
   fChain->SetBranchAddress("dk2nu",&dk2nu);
   fChain->GetEntry(0);

   Notify();
}

Bool_t ancestor::Notify()
{
   // The Notify() function is called when a new file is opened. This
   // can be either for a new TTree in a TChain or when when a new TTree
   // is started when using PROOF. It is normally not necessary to make changes
   // to the generated code, but the routine can be extended by the
   // user if needed. The return value is currently not used.

   return kTRUE;
}

void ancestor::Show(Long64_t entry)
{
// Print contents of entry.
// If entry is not specified, print current entry
   if (!fChain) return;
   fChain->Show(entry);
}
Int_t ancestor::Cut(Long64_t entry)
{
// This function may be called from Loop.
// returns  1 if entry is accepted.
// returns -1 otherwise.
   return 1;
}
#endif // #ifdef eventRates_cxx