Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
ProofFriends.h
Go to the documentation of this file.
1/// \file
2/// \ingroup tutorial_ProofFriends
3///
4/// Selector to process tree friends
5///
6/// \macro_code
7///
8/// \author Gerardo Ganis (gerardo.ganis@cern.ch)
9
10#ifndef ProofFriends_h
11#define ProofFriends_h
12
13#include "TChain.h"
14#include "TSelector.h"
15
16class TH1F;
17class TH2F;
18
19class ProofFriends : public TSelector {
20public :
21 TTree *fChain; //!pointer to the analyzed TTree or TChain
22
23 Bool_t fPlot; // Whether to plot the result
24 Bool_t fDoFriends; // Whether to use the friend tree
25
26 // Specific members
27 TH2F *fXY;
28 TH1F *fZ;
29 TH1F *fR;
30 TH2F *fRZ;
31
32 // Declaration of leaf types
33 Int_t Run;
35 Float_t x;
36 Float_t y;
37 Float_t z;
38
39 Float_t r; // The friend
40
41 // List of branches
42 TBranch *b_Run; //!
43 TBranch *b_Event; //!
44 TBranch *b_x; //!
45 TBranch *b_y; //!
46 TBranch *b_z; //!
47
48 TBranch *b_r; //! The friend branch
49
50 ProofFriends();
51 ~ProofFriends() override { }
52 Int_t Version() const override { return 2; }
53 void Begin(TTree *tree) override;
54 void SlaveBegin(TTree *tree) override;
55 void Init(TTree *tree) override;
56 Bool_t Notify() override;
57 Bool_t Process(Long64_t entry) override;
58 Int_t GetEntry(Long64_t entry, Int_t getall = 0) override { return fChain ? fChain->GetTree()->GetEntry(entry, getall) : 0; }
59 void SetOption(const char *option) override { fOption = option; }
60 void SetObject(TObject *obj) override { fObject = obj; }
61 void SetInputList(TList *input) override { fInput = input; }
62 TList *GetOutputList() const override { return fOutput; }
63 void SlaveTerminate() override;
64 void Terminate() override;
65
66 ClassDefOverride(ProofFriends,0);
67};
68
69#endif
70
71#ifdef ProofFriends_cxx
72void ProofFriends::Init(TTree *tree)
73{
74 // The Init() function is called when the selector needs to initialize
75 // a new tree or chain. Typically here the branch addresses and branch
76 // pointers of the tree will be set.
77 // It is normally not necessary to make changes to the generated
78 // code, but the routine can be extended by the user if needed.
79 // Init() will be called many times when running on PROOF
80 // (once per file to be processed).
81
82 // Set branch addresses and branch pointers
83 if (!tree) return;
84 fChain = tree;
85 fChain->SetMakeClass(1);
86
87 fChain->SetBranchAddress("Run", &Run, &b_Run);
88 fChain->SetBranchAddress("Event", &Event, &b_Event);
89 fChain->SetBranchAddress("x", &x, &b_x);
90 fChain->SetBranchAddress("y", &y, &b_y);
91 fChain->SetBranchAddress("z", &z, &b_z);
92
93 fChain->SetBranchAddress("r", &r, &b_r); // The friend
94}
95
96Bool_t ProofFriends::Notify()
97{
98 // The Notify() function is called when a new file is opened. This
99 // can be either for a new TTree in a TChain or when when a new TTree
100 // is started when using PROOF. It is normally not necessary to make changes
101 // to the generated code, but the routine can be extended by the
102 // user if needed. The return value is currently not used.
103
104 return kTRUE;
105}
106
107#endif // #ifdef ProofFriends_cxx
bool Bool_t
Definition RtypesCore.h:63
int Int_t
Definition RtypesCore.h:45
float Float_t
Definition RtypesCore.h:57
long long Long64_t
Definition RtypesCore.h:80
constexpr Bool_t kTRUE
Definition RtypesCore.h:100
#define ClassDefOverride(name, id)
Definition Rtypes.h:341
Option_t Option_t option
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void input
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void char Point_t Rectangle_t WindowAttributes_t Float_t r
A TTree is a list of TBranches.
Definition TBranch.h:93
1-D histogram with a float per channel (see TH1 documentation)
Definition TH1.h:621
2-D histogram with a float per channel (see TH1 documentation)
Definition TH2.h:295
A doubly linked list.
Definition TList.h:38
Mother of all ROOT objects.
Definition TObject.h:41
A TSelector object is used by the TTree::Draw, TTree::Scan, TTree::Process to navigate in a TTree and...
Definition TSelector.h:31
virtual void SetObject(TObject *obj)
Definition TSelector.h:65
virtual void Init(TTree *)
Definition TSelector.h:53
virtual int Version() const
Definition TSelector.h:52
virtual bool Process(Long64_t)
The Process() function is called for each entry in the tree (or possibly keyed object in the case of ...
TSelectorList * fOutput
! List of objects created during processing
Definition TSelector.h:42
virtual void SlaveBegin(TTree *)
Definition TSelector.h:55
virtual Int_t GetEntry(Long64_t, Int_t=0)
Definition TSelector.h:59
bool Notify() override
This method must be overridden to handle object notification (the base implementation is no-op).
Definition TSelector.h:56
virtual void SetOption(const char *option)
Definition TSelector.h:64
virtual void SetInputList(TList *input)
Definition TSelector.h:66
virtual TList * GetOutputList() const
Definition TSelector.h:69
virtual void SlaveTerminate()
Definition TSelector.h:70
virtual void Begin(TTree *)
Definition TSelector.h:54
virtual void Terminate()
Definition TSelector.h:71
A TTree represents a columnar dataset.
Definition TTree.h:79
virtual Int_t GetEntry(Long64_t entry, Int_t getall=0)
Read all branches of entry and return total number of bytes read.
Definition TTree.cxx:5638
virtual void SetMakeClass(Int_t make)
Set all the branches in this TTree to be in decomposed object mode (also known as MakeClass mode).
Definition TTree.cxx:9167
virtual Int_t SetBranchAddress(const char *bname, void *add, TBranch **ptr=nullptr)
Change branch address, dealing with clone trees properly.
Definition TTree.cxx:8380
virtual TTree * GetTree() const
Definition TTree.h:517
Double_t y[n]
Definition legend1.C:17
Double_t x[n]
Definition legend1.C:17