ProofFriends.h

/// \file
/// \ingroup tutorial_ProofFriends
///
/// Selector to process tree friends
///
/// \macro_code
///
/// \author Gerardo Ganis (gerardo.ganis@cern.ch)

#ifndef ProofFriends_h
#define ProofFriends_h

#include "TChain.h"
#include "TSelector.h"

class TH1F;
class TH2F;

class ProofFriends : public TSelector {
public :
   TTree          *fChain;   //!pointer to the analyzed TTree or TChain

   Bool_t          fPlot;    // Whether to plot the result
   Bool_t          fDoFriends; // Whether to use the friend tree

   // Specific members
   TH2F            *fXY;
   TH1F            *fZ;
   TH1F            *fR;
   TH2F            *fRZ;

   // Declaration of leaf types
   Int_t           Run;
   Long64_t        Event;
   Float_t         x;
   Float_t         y;
   Float_t         z;

   Float_t         r;     // The friend

   // List of branches
   TBranch        *b_Run;   //!
   TBranch        *b_Event;   //!
   TBranch        *b_x;   //!
   TBranch        *b_y;   //!
   TBranch        *b_z;   //!

   TBranch        *b_r;   //! The friend branch

   ProofFriends();
   ~ProofFriends() override { }
   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;

   ClassDefOverride(ProofFriends,0);
};

#endif

#ifdef ProofFriends_cxx
void ProofFriends::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
   if (!tree) return;
   fChain = tree;
   fChain->SetMakeClass(1);

   fChain->SetBranchAddress("Run", &Run, &b_Run);
   fChain->SetBranchAddress("Event", &Event, &b_Event);
   fChain->SetBranchAddress("x", &x, &b_x);
   fChain->SetBranchAddress("y", &y, &b_y);
   fChain->SetBranchAddress("z", &z, &b_z);

   fChain->SetBranchAddress("r", &r, &b_r);  // The friend
}

Bool_t ProofFriends::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;
}

#endif // #ifdef ProofFriends_cxx