]>
Commit | Line | Data |
---|---|---|
563113d0 | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | /////////////////////////////////////////////////////////////////////////// | |
16 | // The class AliCFManager is designed to handle inside the | |
17 | // task the basic components for a particle-level correction of single | |
18 | // particle analysis | |
19 | // The class provides methods to set lists of cuts and to loop over them | |
20 | // for several different selection steps to be then used for | |
21 | // efficiency calculation. | |
22 | // prototype version by S.Arcelli silvia.arcelli@cern.ch | |
23 | /////////////////////////////////////////////////////////////////////////// | |
24 | #include "TBits.h" | |
25 | #include "TList.h" | |
26 | #include "TH1.h" | |
27 | #include "AliLog.h" | |
28 | #include "AliCFCutBase.h" | |
29 | #include "AliCFManager.h" | |
30 | ||
31 | ClassImp(AliCFManager) | |
32 | ||
33 | //_____________________________________________________________________________ | |
34 | AliCFManager::AliCFManager() : | |
35 | TNamed(), | |
36 | fEvtContainer(0x0), | |
37 | fPartContainer(0x0), | |
38 | fhQABits(0x0) | |
39 | ||
40 | { | |
41 | // | |
42 | // ctor | |
43 | // | |
44 | for(Int_t i=0;i<kNEvtSel;i++)fEvtCutList[i]=0x0; | |
45 | for(Int_t i=0;i<kNPartSel;i++)fPartCutList[i]=0x0; | |
46 | fhQABits=new TBits(0); | |
47 | } | |
48 | //_____________________________________________________________________________ | |
49 | AliCFManager::AliCFManager(Char_t* name, Char_t* title) : | |
50 | TNamed(name,title), | |
51 | fEvtContainer(0x0), | |
52 | fPartContainer(0x0), | |
53 | fhQABits(0x0) | |
54 | { | |
55 | // | |
56 | // ctor | |
57 | // | |
58 | for(Int_t i=0;i<kNEvtSel;i++)fEvtCutList[i]=0x0; | |
59 | for(Int_t i=0;i<kNPartSel;i++)fPartCutList[i]=0x0; | |
60 | fhQABits=new TBits(0); | |
61 | } | |
62 | //_____________________________________________________________________________ | |
63 | AliCFManager::AliCFManager(const AliCFManager& c) : | |
64 | TNamed(c), | |
65 | fEvtContainer(c.fEvtContainer), | |
66 | fPartContainer(c.fPartContainer), | |
67 | fhQABits(c.fhQABits) | |
68 | { | |
69 | // | |
70 | //copy ctor | |
71 | // | |
72 | for(Int_t i=0;i<kNEvtSel;i++)fEvtCutList[i]=c.fEvtCutList[i]; | |
73 | for(Int_t i=0;i<kNPartSel;i++)fPartCutList[i]=c.fPartCutList[i]; | |
74 | } | |
75 | //_____________________________________________________________________________ | |
76 | AliCFManager& AliCFManager::operator=(const AliCFManager& c) | |
77 | { | |
78 | // | |
79 | // Assignment operator | |
80 | // | |
81 | if (this != &c) { | |
82 | TNamed::operator=(c) ; | |
83 | } | |
84 | ||
85 | this->fEvtContainer=c.fEvtContainer; | |
86 | this->fPartContainer=c.fPartContainer; | |
87 | this->fhQABits=c.fhQABits; | |
88 | for(Int_t i=0;i<kNEvtSel;i++)this->fEvtCutList[i]=c.fEvtCutList[i]; | |
89 | for(Int_t i=0;i<kNPartSel;i++)this->fPartCutList[i]=c.fPartCutList[i]; | |
90 | return *this ; | |
91 | } | |
92 | ||
93 | //_____________________________________________________________________________ | |
94 | AliCFManager::~AliCFManager() { | |
95 | // | |
96 | //dtor | |
97 | // | |
98 | if(fhQABits) delete fhQABits; | |
99 | } | |
100 | ||
101 | //_____________________________________________________________________________ | |
102 | Bool_t AliCFManager::CheckParticleCuts(Int_t isel, TObject *obj, const TString &selcuts) const { | |
103 | // | |
104 | // check whether object obj passes particle-level selection isel | |
105 | // | |
106 | ||
107 | if(isel>=kNPartSel){ | |
108 | AliWarning(Form("Selection index out of Range! isel=%i, max. number of selections= %i", isel,kNPartSel)); | |
109 | return kTRUE; | |
110 | } | |
111 | if(!fPartCutList[isel])return kTRUE; | |
112 | TObjArrayIter iter(fPartCutList[isel]); | |
113 | AliCFCutBase *cut = 0; | |
114 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
115 | TString cutName=cut->GetName(); | |
116 | Bool_t checkCut=CompareStrings(cutName,selcuts); | |
117 | if(checkCut && !cut->IsSelected(obj)) return kFALSE; | |
118 | } | |
119 | return kTRUE; | |
120 | } | |
121 | ||
122 | //_____________________________________________________________________________ | |
123 | Bool_t AliCFManager::CheckEventCuts(Int_t isel, TObject *obj, const TString &selcuts) const{ | |
124 | // | |
125 | // check whether object obj passes event-level selection isel | |
126 | // | |
127 | ||
128 | if(isel>=kNEvtSel){ | |
129 | AliWarning(Form("Selection index out of Range! isel=%i, max. number of selections= %i", isel,kNEvtSel)); | |
130 | return kTRUE; | |
131 | } | |
132 | if(!fEvtCutList[isel])return kTRUE; | |
133 | TObjArrayIter iter(fEvtCutList[isel]); | |
134 | AliCFCutBase *cut = 0; | |
135 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
136 | TString cutName=cut->GetName(); | |
137 | Bool_t checkCut=CompareStrings(cutName,selcuts); | |
138 | if(checkCut && !cut->IsSelected(obj)) return kFALSE; | |
139 | } | |
140 | return kTRUE; | |
141 | } | |
142 | ||
143 | //_____________________________________________________________________________ | |
144 | void AliCFManager::FillQABeforeParticleCuts(Int_t isel, TObject *obj) const{ | |
145 | // | |
146 | // Fill QA histos before cuts at particle selection level isel are applied | |
147 | // | |
148 | ||
149 | if(isel>=kNPartSel){ | |
150 | AliWarning(Form("Selection index out of Range! isel=%i, max. number of selections= %i", isel,kNPartSel)); | |
151 | return; | |
152 | } | |
153 | if(!fPartCutList[isel])return; | |
154 | ||
155 | TObjArrayIter iter(fPartCutList[isel]); | |
156 | AliCFCutBase *cut = 0; | |
157 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
158 | if(cut->IsQAOn())cut->FillHistogramsBeforeCuts(obj); | |
159 | } | |
160 | } | |
161 | //_____________________________________________________________________________ | |
162 | void AliCFManager::FillQAAfterParticleCuts(Int_t isel, TObject *obj) const{ | |
163 | // | |
164 | // Fill QA histos after cuts at particle selection level isel are applied | |
165 | // | |
166 | if(isel>=kNPartSel){ | |
167 | AliWarning(Form("Selection index out of Range! isel=%i, max. number of selections= %i", isel,kNPartSel)); | |
168 | return; | |
169 | } | |
170 | if(!fPartCutList[isel])return; | |
171 | ||
172 | TObjArrayIter iter(fPartCutList[isel]); | |
173 | AliCFCutBase *cut = 0; | |
174 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
175 | if(cut->IsQAOn())cut->FillHistogramsAfterCuts(obj); | |
176 | } | |
177 | } | |
178 | ||
179 | //_____________________________________________________________________________ | |
180 | void AliCFManager::FillQABeforeEventCuts(Int_t isel, TObject *obj) const{ | |
181 | // | |
182 | // Fill QA histos before cuts at event selection level isel are applied | |
183 | // | |
184 | ||
185 | if(isel>=kNEvtSel){ | |
186 | AliWarning(Form("Selection index out of Range! isel=%i, max. number of selections= %i", isel,kNEvtSel)); | |
187 | return; | |
188 | } | |
189 | if(!fEvtCutList[isel])return; | |
190 | ||
191 | TObjArrayIter iter(fEvtCutList[isel]); | |
192 | AliCFCutBase *cut = 0; | |
193 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
194 | if(cut->IsQAOn())cut->FillHistogramsBeforeCuts(obj); | |
195 | } | |
196 | } | |
197 | ||
198 | //_____________________________________________________________________________ | |
199 | void AliCFManager::FillQAAfterEventCuts(Int_t isel, TObject *obj) const{ | |
200 | // | |
201 | // Fill QA histos after cuts at event selection level isel are applied | |
202 | // | |
203 | ||
204 | if(isel>=kNEvtSel){ | |
205 | AliWarning(Form("Selection index out of Range! isel=%i, max. number of selections= %i", isel,kNEvtSel)); | |
206 | return; | |
207 | } | |
208 | if(!fEvtCutList[isel])return; | |
209 | ||
210 | TObjArrayIter iter(fEvtCutList[isel]); | |
211 | AliCFCutBase *cut = 0; | |
212 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
213 | if(cut->IsQAOn())cut->FillHistogramsAfterCuts(obj); | |
214 | } | |
215 | } | |
216 | ||
217 | //_____________________________________________________________________________ | |
218 | void AliCFManager::AddQAHistosToList(TList *list) const { | |
219 | // | |
220 | // Add to list the full list of QA histograms to be written to the output | |
221 | // | |
222 | ||
223 | for(Int_t isel=0;isel<kNPartSel; isel++){ | |
224 | if(!fPartCutList[isel])continue; | |
225 | TObjArrayIter iter(fPartCutList[isel]); | |
226 | AliCFCutBase *cut = 0; | |
227 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
228 | if(cut->IsQAOn())cut->AddQAHistograms(list); | |
229 | } | |
230 | } | |
231 | ||
232 | //Event-level cuts QA | |
233 | ||
234 | for(Int_t isel=0;isel<kNEvtSel; isel++){ | |
235 | if(!fEvtCutList[isel])continue; | |
236 | TObjArrayIter iter(fEvtCutList[isel]); | |
237 | AliCFCutBase *cut = 0; | |
238 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
239 | if(cut->IsQAOn())cut->AddQAHistograms(list); | |
240 | } | |
241 | } | |
242 | } | |
243 | //_____________________________________________________________________________ | |
244 | TBits* AliCFManager::GetQAParticleSelBits(Int_t isel, TObject *obj) { | |
245 | // | |
246 | // Get the full list of QA histograms to be written to the output | |
247 | // | |
248 | ||
249 | fhQABits->Clear(); //reset the list | |
250 | ||
251 | //Particle-level cuts QA | |
252 | ||
253 | if(fPartCutList[isel]){ | |
254 | TObjArrayIter iter(fPartCutList[isel]); | |
255 | AliCFCutBase *cut = 0; | |
256 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
257 | if(cut->IsQAOn()){ | |
258 | TBits *qalist=new TBits(0); | |
259 | cut->GetBitMap(obj,qalist); | |
260 | for(UInt_t icut=0;icut<qalist->GetNbits();icut++){ | |
261 | fhQABits->SetBitNumber(icut,qalist->TestBitNumber(icut)); | |
262 | } | |
263 | delete qalist; | |
264 | } | |
265 | } | |
266 | } | |
267 | ||
268 | return fhQABits; | |
269 | ||
270 | } | |
271 | ||
272 | //_____________________________________________________________________________ | |
273 | void AliCFManager::SetEventInfo(TObject *obj) const { | |
274 | ||
275 | //Particle level cuts | |
276 | ||
277 | for(Int_t isel=0;isel<kNPartSel; isel++){ | |
278 | if(!fPartCutList[isel])continue; | |
279 | TObjArrayIter iter(fPartCutList[isel]); | |
280 | AliCFCutBase *cut = 0; | |
281 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
282 | cut->SetEvtInfo(obj); | |
283 | } | |
284 | } | |
285 | ||
286 | //Event level cuts | |
287 | ||
288 | for(Int_t isel=0;isel<kNEvtSel; isel++){ | |
289 | if(!fEvtCutList[isel])continue; | |
290 | TObjArrayIter iter(fEvtCutList[isel]); | |
291 | AliCFCutBase *cut = 0; | |
292 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
293 | cut->SetEvtInfo(obj); | |
294 | } | |
295 | } | |
296 | } | |
297 | //_____________________________________________________________________________ | |
298 | void AliCFManager::InitQAHistos() const { | |
299 | ||
300 | //Particle level cuts | |
301 | ||
302 | for(Int_t isel=0;isel<kNPartSel; isel++){ | |
303 | if(!fPartCutList[isel])continue; | |
304 | TObjArrayIter iter(fPartCutList[isel]); | |
305 | AliCFCutBase *cut = 0; | |
306 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
307 | if(cut->IsQAOn())cut->Init(); | |
308 | } | |
309 | } | |
310 | ||
311 | //Event level cuts | |
312 | ||
313 | for(Int_t isel=0;isel<kNEvtSel; isel++){ | |
314 | if(!fEvtCutList[isel])continue; | |
315 | TObjArrayIter iter(fEvtCutList[isel]); | |
316 | AliCFCutBase *cut = 0; | |
317 | while ( (cut = (AliCFCutBase*)iter.Next()) ) { | |
318 | if(cut->IsQAOn())cut->Init(); | |
319 | } | |
320 | } | |
321 | } | |
322 | ||
323 | //_____________________________________________________________________________ | |
324 | Bool_t AliCFManager::CompareStrings(const TString &cutname,const TString &selcuts) const{ | |
325 | // | |
326 | // compare two strings | |
327 | // | |
328 | ||
329 | if(selcuts.Contains("all"))return kTRUE; | |
330 | if ( selcuts.CompareTo(cutname) == 0 || | |
331 | selcuts.BeginsWith(cutname+" ") || | |
332 | selcuts.EndsWith(" "+cutname) || | |
333 | selcuts.Contains(" "+cutname+" ")) return kTRUE; | |
334 | return kFALSE; | |
335 | } | |
336 | ||
337 |