1 #ifndef AliEventPoolManager_h
2 #define AliEventPoolManager_h
14 // Generic event mixing classes
16 // Stores a buffer of tracks that updates continuously. The track type
17 // contained by the pools can be anything inheriting from
18 // TObject. Pools are updated based on maintaining a minimum fixed
19 // number of tracks. Multiplicity/centrality and z-vertex bins must be
20 // passed in at initialization. For example of implementation, see
21 // $ALICE_ROOT/PWGCF/Correlations/DPhi/AliAnalysisTaskPhiCorrelations.cxx
23 // Authors: A. Adare and C. Loizides
27 class AliEventPool : public TObject
53 AliEventPool(Int_t d, Double_t multMin, Double_t multMax,
54 Double_t zvtxMin, Double_t zvtxMax,
55 Double_t psiMin=-999., Double_t psiMax=999.)
79 Bool_t EventMatchesBin(Int_t mult, Double_t zvtx, Double_t psi=0.) const;
80 Bool_t EventMatchesBin(Double_t mult, Double_t zvtx, Double_t psi=0.) const;
81 Bool_t IsReady() const { return IsReady(NTracksInPool(), GetCurrentNEvents()); }
82 Bool_t IsFirstReady() const { return fFirstFilled; }
83 Int_t GetNTimes() const { return fNTimes; }
84 Int_t GetCurrentNEvents() const { return fEvents.size(); }
85 Int_t GlobalEventIndex(Int_t j) const;
86 TObject *GetRandomTrack() const;
87 TObjArray *GetRandomEvent() const;
88 TObjArray *GetEvent(Int_t i) const;
89 Int_t MultBinIndex() const { return fMultBinIndex; }
90 Int_t NTracksInEvent(Int_t iEvent) const;
91 Int_t NTracksInCurrentEvent() const { return fNTracksInEvent.back(); }
92 void PrintInfo() const;
93 Int_t PsiBinIndex() const { return fPsiBinIndex; }
94 Int_t NTracksInPool() const;
95 Bool_t WasUpdated() const { return fWasUpdated; }
96 Int_t ZvtxBinIndex() const { return fZvtxBinIndex; }
97 void SetDebug(Bool_t b) { fDebug = b; }
98 void SetTargetTrackDepth(Int_t d, Float_t fraction = 1.0) { fTargetTrackDepth = d; fTargetFraction = fraction; }
99 void SetTargetEvents(Int_t ev) { fTargetEvents = ev; }
100 Int_t SetEventMultRange(Int_t multMin, Int_t multMax);
101 Int_t SetEventMultRange(Double_t multMin, Double_t multMax);
102 Int_t SetEventZvtxRange(Double_t zvtxMin, Double_t zvtxMax);
103 Int_t SetEventPsiRange(Double_t psiMin, Double_t psiMax);
104 void SetMultBinIndex(Int_t iM) { fMultBinIndex = iM; }
105 void SetZvtxBinIndex(Int_t iZ) { fZvtxBinIndex = iZ; }
106 void SetPsiBinIndex(Int_t iP) { fPsiBinIndex = iP; }
107 Int_t UpdatePool(TObjArray *trk);
110 Bool_t IsReady(Int_t tracks, Int_t events) const { return (tracks >= fTargetFraction * fTargetTrackDepth) || ((fTargetEvents > 0) && (events >= fTargetEvents)); }
112 deque<TObjArray*> fEvents; //Holds TObjArrays of MyTracklets
113 deque<int> fNTracksInEvent; //Tracks in event
114 deque<int> fEventIndex; //Original event index
115 Int_t fMixDepth; //Number of evts. to mix with
116 Double_t fMultMin, fMultMax; //Track multiplicity bin range
117 Double_t fZvtxMin, fZvtxMax; //Event z-vertex bin range
118 Double_t fPsiMin, fPsiMax; //Event plane angle (Psi) bin range
119 Bool_t fWasUpdated; //Evt. succesfully passed selection?
120 Int_t fMultBinIndex; //Multiplicity bin
121 Int_t fZvtxBinIndex; //Zvertex bin
122 Int_t fPsiBinIndex; //Event plane angle (Psi) bin
123 Int_t fDebug; //If 1 then debug on
124 Int_t fTargetTrackDepth; //Number of tracks, once full
125 Bool_t fFirstFilled; //Init to false
126 Int_t fNTimes; //Number of times init. condition reached
127 Float_t fTargetFraction; //fraction of fTargetTrackDepth at which pool is ready (default: 1.0)
128 Int_t fTargetEvents; //if non-zero: number of filled events after which pool is ready regardless of fTargetTrackDepth (default: 0)
130 ClassDef(AliEventPool,3) // Event pool class
133 class AliEventPoolManager : public TObject
136 AliEventPoolManager()
142 fTargetTrackDepth(0) {;}
143 AliEventPoolManager(Int_t maxEvts, Int_t minNTracks,
144 Int_t nMultBins, Double_t *multbins,
145 Int_t nZvtxBins, Double_t *zvtxbins);
147 AliEventPoolManager(Int_t maxEvts, Int_t minNTracks,
148 Int_t nMultBins, Double_t *multbins,
149 Int_t nZvtxBins, Double_t *zvtxbins,
150 Int_t nPsiBins, Double_t *psibins);
152 ~AliEventPoolManager() {;}
154 // First uses bin indices, second uses the variables themselves.
155 AliEventPool *GetEventPool(Int_t iMult, Int_t iZvtx, Int_t iPsi=0) const;
156 AliEventPool *GetEventPool(Int_t centVal, Double_t zvtxVal, Double_t psiVal=0.) const;
157 AliEventPool *GetEventPool(Double_t centVal, Double_t zvtxVal, Double_t psiVal=0.) const;
159 Int_t InitEventPools(Int_t depth,
160 Int_t nmultbins, Double_t *multbins,
161 Int_t nzvtxbins, Double_t *zvtxbins,
162 Int_t npsibins, Double_t *psibins);
164 void SetTargetTrackDepth(Int_t d) { fTargetTrackDepth = d;} // Same as for G.E.P. class
165 Int_t UpdatePools(TObjArray *trk);
166 void SetDebug(Bool_t b) { fDebug = b; }
167 void SetTargetValues(Int_t trackDepth, Float_t fraction, Int_t events);
170 Int_t fDebug; // If 1 then debug on
171 Int_t fNMultBins; // number mult bins
172 Int_t fNZvtxBins; // number vertex bins
173 Int_t fNPsiBins; // number Event plane angle (Psi) bins
174 std::vector<AliEventPool*> fEvPool; // pool in bins of [fNMultBin][fNZvtxBin][fNPsiBin]
175 Int_t fTargetTrackDepth; // Required track size, same for all pools.
177 ClassDef(AliEventPoolManager,2)