KVINDRAGeneDataSelector.h

 
#ifndef KVINDRAGeneDataSelector_h
#define KVINDRAGeneDataSelector_h
 
#include <TROOT.h>
#include <TChain.h>
#include <TFile.h>
#include <TSelector.h>
#include "KVINDRATriggerInfo.h"
#include "Riostream.h"
#include "KVINDRA.h"
#include "KVEBYEDAT_ACQParam.h"
 
class KVINDRAGeneDataSelector : public TSelector {
protected :
    TTree* fChain;
 
   Int_t           RunNumber;
   Int_t           EventNumber;
   KVINDRATriggerInfo* TriggerInfo;
 
    TBranch* b_RunNumber;
    TBranch* b_EventNumber;
    TBranch* b_TriggerInfo;
 
    Long64_t fTreeEntry;
    Bool_t needToCallEndRun;
 
public :
   KVINDRAGeneDataSelector(TTree* /*tree*/ = 0)
   {
      needToCallEndRun = kFALSE;
   }
   virtual ~KVINDRAGeneDataSelector()
   {
   }
   Int_t   Version() const override
   {
      return 2;
   }
   void    Begin(TTree* tree) override;
   void    SlaveBegin(TTree* tree) override;
   void    Init(TTree* tree) override;
   Bool_t  Notify() override;
   Bool_t  Process(Long64_t entry) override;
   Int_t   GetEntry(Long64_t entry, Int_t getall = 0) override
   {
      return fChain ? fChain->GetTree()->GetEntry(entry, getall) : 0;
   }
   void    SetOption(const char* option) override
   {
      fOption = option;
   }
   void    SetObject(TObject* obj) override
   {
      fObject = obj;
   }
   void    SetInputList(TList* input) override
   {
      fInput = input;
   }
   TList*  GetOutputList() const override
   {
      return fOutput;
   }
   void    SlaveTerminate() override;
   void    Terminate() override;
   virtual void InitAnalysis()
   {
   };
   virtual void InitRun()
   {
   };
   virtual Bool_t Analysis()
   {
      return kTRUE;
   };
   virtual void EndRun()
   {
   };
   virtual void EndAnalysis()
   {
   };
   virtual Bool_t AtEndOfRun(void);
   static void Make(const Char_t* kvsname);
   void SetBranchAddresses();
 
   ClassDefOverride(KVINDRAGeneDataSelector, 0) //Base class for analysis of INDRA pulser & laser data
};
 
#endif
 
#ifdef KVINDRAGeneDataSelector_cxx
void KVINDRAGeneDataSelector::Init(TTree* tree)
{
 
   TriggerInfo = 0;
   if (!tree) return;
   fChain = tree;
   fChain->SetMakeClass(1);
}
 
Bool_t KVINDRAGeneDataSelector::Notify()
{
 
   needToCallEndRun = kTRUE;
   SetBranchAddresses();
   InitRun();                   //user initialisations for run
   return kTRUE;
}
 
void KVINDRAGeneDataSelector::SetBranchAddresses()
{
 
   if (!fChain->GetCurrentFile()) {
      Info("SetBranchAddress", "No file open");
      return;
   }
   gIndra = (KVINDRA*) fChain->GetCurrentFile()->Get("INDRA");
 
   if (!gIndra) {
      Info("SetBranchAddress", "INDRA object not found in file");
      return;
   }
 
   fChain->SetBranchAddress("RunNumber", &RunNumber, &b_RunNumber);
   fChain->SetBranchAddress("EventNumber", &EventNumber, &b_EventNumber);
   fChain->SetBranchAddress("TriggerInfo", &TriggerInfo, &b_TriggerInfo);
 
   RunNumber = gIndra->GetCurrentRunNumber();
}
 
#endif // #ifdef KVINDRAGeneDataSelector_cxx