Logo ROOT   6.30.04
Reference Guide
 All Namespaces Files Pages
TTreeFormula.h
Go to the documentation of this file.
1 // @(#)root/treeplayer:$Id$
2 // Author: Rene Brun 19/01/96
3 
4 /*************************************************************************
5  * Copyright (C) 1995-2000, Rene Brun and Fons Rademakers. *
6  * All rights reserved. *
7  * *
8  * For the licensing terms see $ROOTSYS/LICENSE. *
9  * For the list of contributors see $ROOTSYS/README/CREDITS. *
10  *************************************************************************/
11 // ---------------------------------- TreeFormula.h
12 
13 #ifndef ROOT_TTreeFormula
14 #define ROOT_TTreeFormula
15 
16 
17 
18 //////////////////////////////////////////////////////////////////////////
19 // //
20 // TTreeFormula //
21 // //
22 // The Tree formula class //
23 // //
24 //////////////////////////////////////////////////////////////////////////
25 
26 #include "v5/TFormula.h"
27 
28 #include "TLeaf.h"
29 
30 #include "TObjArray.h"
31 
32 #include <string>
33 #include <vector>
34 
35 #ifdef R__OLDHPACC
36 namespace std {
37  using ::string;
38  using ::vector;
39 }
40 #endif
41 
42 const Int_t kMAXCODES = kMAXFOUND; // must be the same as kMAXFOUND in TFormulaOld
43 const Int_t kMAXFORMDIM = 5; // Maximum number of array dimensions support in TTreeFormula
44 
45 class TTree;
46 class TArrayI;
47 class TMethodCall;
48 class TLeafObject;
49 class TDataMember;
50 class TStreamerElement;
51 class TFormLeafInfoMultiVarDim;
52 class TFormLeafInfo;
53 class TBranchElement;
54 class TAxis;
55 class TTreeFormulaManager;
56 
57 
58 class TTreeFormula : public ROOT::v5::TFormula {
59 
60 friend class TTreeFormulaManager;
61 
62 protected:
63  enum EStatusBits {
64  kIsCharacter = BIT(12),
65  kMissingLeaf = BIT(15), // true if some of the needed leaves are missing in the current TTree
66  kIsInteger = BIT(17), // true if the branch contains an integer variable
67  kNeedEntries = BIT(18) // true if the formula uses Entries$
68  };
69  enum {
70  kDirect, kDataMember, kMethod,
71  kIndexOfEntry, kEntries, kLength, kIteration, kLengthFunc, kSum, kEntryList,
72  kTreeMember,
73  kIndexOfLocalEntry,
74  kMin, kMax,
75  kLocalEntries
76 
77  };
78  enum {
79  kAlias = 200,
80  kAliasString = 201,
81  kAlternate = 202,
82  kAlternateString = 203,
83  kMinIf = 204,
84  kMaxIf = 205
85  };
86 
87  // Helper struct to hold a cache
88  // that can accelerate calculation of the RealIndex.
89  struct RealInstanceCache {
90  Int_t fInstanceCache = 0;
91  Int_t fLocalIndexCache = 0;
92  Int_t fVirtAccumCache = 0;
93  };
94 
95  TTree *fTree; //! pointer to Tree
96  Int_t fCodes[kMAXCODES]; // List of leaf numbers referenced in formula
97  Int_t fNdata[kMAXCODES]; //! This caches the physical number of element in the leaf or data member.
98  Int_t fNcodes; // Number of leaves referenced in formula
99  Bool_t fHasCast; // Record whether the formula contain a cast operation or not
100  Int_t fMultiplicity; // Indicator of the variability of the formula
101  Int_t fNindex; // Size of fIndex
102  Int_t *fLookupType; //[fNindex] array indicating how each leaf should be looked-up
103  TObjArray fLeaves; //! List of leaf used in this formula.
104  TObjArray fDataMembers; //! List of leaf data members
105  TObjArray fMethods; //! List of leaf method calls
106  TObjArray fExternalCuts; //! List of TCutG and TEntryList used in the formula
107  TObjArray fAliases; //! List of TTreeFormula for each alias used.
108  TObjArray fLeafNames; // List of TNamed describing leaves
109  TObjArray fBranches; //! List of branches to read. Similar to fLeaves but duplicates are zeroed out.
110  Bool_t fQuickLoad; //! If true, branch GetEntry is only called when the entry number changes.
111  Bool_t fNeedLoading; //! If true, the current entry has not been loaded yet.
112 
113  Int_t fNdimensions[kMAXCODES]; //Number of array dimensions in each leaf
114  Int_t fFixedSizes[kMAXCODES][kMAXFORMDIM]; //Physical sizes of lower dimensions for each leaf
115  UChar_t fHasMultipleVarDim[kMAXCODES]; //True if the corresponding variable is an array with more than one variable dimension.
116 
117  //the next line should have a mutable in front. See GetNdata()
118  Int_t fCumulSizes[kMAXCODES][kMAXFORMDIM]; //Accumulated sizes of lower dimensions for each leaf after variable dimensions has been calculated
119  Int_t fIndexes[kMAXCODES][kMAXFORMDIM]; //Index of array selected by user for each leaf
120  TTreeFormula *fVarIndexes[kMAXCODES][kMAXFORMDIM]; //Pointer to a variable index.
121 
122  TAxis *fAxis; //! pointer to histogram axis if this is a string
123  Bool_t fDidBooleanOptimization; //! True if we executed one boolean optimization since the last time instance number 0 was evaluated
124  TTreeFormulaManager *fManager; //! The dimension coordinator.
125 
126  // Helper members and function used during the construction and parsing
127  TList *fDimensionSetup; //! list of dimension setups, for delayed creation of the dimension information.
128  std::vector<std::string> fAliasesUsed; //! List of aliases used during the parsing of the expression.
129 
130  LongDouble_t* fConstLD; //! local version of fConsts able to store bigger numbers
131 
132  RealInstanceCache fRealInstanceCache; //! Cache accelerating the GetRealInstance function
133 
134  TTreeFormula(const char *name, const char *formula, TTree *tree, const std::vector<std::string>& aliases);
135  void Init(const char *name, const char *formula);
136  Bool_t BranchHasMethod(TLeaf* leaf, TBranch* branch, const char* method,const char* params, Long64_t readentry) const;
137  Int_t DefineAlternate(const char* expression);
138  void DefineDimensions(Int_t code, Int_t size, TFormLeafInfoMultiVarDim * info, Int_t& virt_dim);
139  Int_t FindLeafForExpression(const char* expression, TLeaf *&leaf, TString &leftover, Bool_t &final, UInt_t &paran_level, TObjArray &castqueue, std::vector<std::string>& aliasUsed, Bool_t &useLeafCollectionObject, const char *fullExpression);
140  TLeaf* GetLeafWithDatamember(const char* topchoice, const char* nextchice, Long64_t readentry) const;
141  Int_t ParseWithLeaf(TLeaf *leaf, const char *expression, Bool_t final, UInt_t paran_level, TObjArray &castqueue, Bool_t useLeafCollectionObject, const char *fullExpression);
142  Int_t RegisterDimensions(Int_t code, Int_t size, TFormLeafInfoMultiVarDim * multidim = 0);
143  Int_t RegisterDimensions(Int_t code, TBranchElement *branch);
144  Int_t RegisterDimensions(Int_t code, TFormLeafInfo *info, TFormLeafInfo *maininfo, Bool_t useCollectionObject);
145  Int_t RegisterDimensions(Int_t code, TLeaf *leaf);
146  Int_t RegisterDimensions(const char *size, Int_t code);
147 
148  virtual Double_t GetValueFromMethod(Int_t i, TLeaf *leaf) const;
149  virtual void* GetValuePointerFromMethod(Int_t i, TLeaf *leaf) const;
150  Int_t GetRealInstance(Int_t instance, Int_t codeindex);
151 
152  void LoadBranches();
153  Bool_t LoadCurrentDim();
154  void ResetDimensions();
155 
156  virtual TClass* EvalClass(Int_t oper) const;
157  virtual Bool_t IsLeafInteger(Int_t code) const;
158  virtual Bool_t IsString(Int_t oper) const;
159  virtual Bool_t IsLeafString(Int_t code) const;
160  virtual Bool_t SwitchToFormLeafInfo(Int_t code);
161  virtual Bool_t StringToNumber(Int_t code);
162 
163  void Convert(UInt_t fromVersion);
164 
165 private:
166  // Not implemented yet
167  TTreeFormula(const TTreeFormula&) = delete;
168  TTreeFormula& operator=(const TTreeFormula&) = delete;
169 
170  template<typename T> T GetConstant(Int_t k);
171 
172 public:
173  TTreeFormula();
174  TTreeFormula(const char *name,const char *formula, TTree *tree);
175  virtual ~TTreeFormula();
176 
177  virtual Int_t DefinedVariable(TString &variable, Int_t &action);
178  virtual TClass* EvalClass() const;
179 
180  template<typename T> T EvalInstance(Int_t i=0, const char *stringStack[]=0);
181  virtual Double_t EvalInstance(Int_t i=0, const char *stringStack[]=0) {return EvalInstance<Double_t>(i, stringStack); }
182  virtual Long64_t EvalInstance64(Int_t i=0, const char *stringStack[]=0) {return EvalInstance<Long64_t>(i, stringStack); }
183  virtual LongDouble_t EvalInstanceLD(Int_t i=0, const char *stringStack[]=0) {return EvalInstance<LongDouble_t>(i, stringStack); }
184 
185  virtual const char *EvalStringInstance(Int_t i=0);
186  virtual void* EvalObject(Int_t i=0);
187  // EvalInstance should be const. See comment on GetNdata()
188  TFormLeafInfo *GetLeafInfo(Int_t code) const;
189  TTreeFormulaManager*GetManager() const { return fManager; }
190  TMethodCall *GetMethodCall(Int_t code) const;
191  virtual Int_t GetMultiplicity() const {return fMultiplicity;}
192  virtual TLeaf *GetLeaf(Int_t n) const;
193  virtual Int_t GetNcodes() const {return fNcodes;}
194  virtual Int_t GetNdata();
195  //GetNdata should probably be const. However it need to cache some information about the actual dimension
196  //of arrays, so if GetNdata is const, the variables fUsedSizes and fCumulUsedSizes need to be declared
197  //mutable. We will be able to do that only when all the compilers supported for ROOT actually implemented
198  //the mutable keyword.
199  //NOTE: Also modify the code in PrintValue which current goes around this limitation :(
200  virtual Bool_t IsInteger(Bool_t fast=kTRUE) const;
201  Bool_t IsQuickLoad() const { return fQuickLoad; }
202  virtual Bool_t IsString() const;
203  virtual Bool_t Notify() { UpdateFormulaLeaves(); return kTRUE; }
204  virtual char *PrintValue(Int_t mode=0) const;
205  virtual char *PrintValue(Int_t mode, Int_t instance, const char *decform = "9.9") const;
206  virtual void SetAxis(TAxis *axis=0);
207  void SetQuickLoad(Bool_t quick) { fQuickLoad = quick; }
208  virtual void SetTree(TTree *tree) {fTree = tree;}
209  virtual void ResetLoading();
210  virtual TTree* GetTree() const {return fTree;}
211  virtual void UpdateFormulaLeaves();
212 
213  ClassDef(TTreeFormula, 10); //The Tree formula
214 };
215 
216 #endif