]>
Commit | Line | Data |
---|---|---|
5e2cf960 | 1 | /************************************************************************* |
2 | * Copyright(c) 1998-2009, 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 | /////////////////////////////////////////////////////////////////////////// | |
c081415a | 17 | // Dielectron Histogram framework helper // |
5e2cf960 | 18 | // // |
19 | /* | |
20 | ||
c081415a | 21 | A helper class to extract objects(histograms and/or profiles) from |
22 | a AliDielctronHF array of objects. | |
5e2cf960 | 23 | |
24 | ||
c081415a | 25 | How to use it: |
5e2cf960 | 26 | |
c081415a | 27 | AliDielectronHFhelper *hf = new AliDielectronHFhelper("path/to/the/output/file.root", "ConfigName"); |
28 | // print the structure | |
29 | hf->Print(); | |
5e2cf960 | 30 | |
c081415a | 31 | //apply some cuts and print them |
32 | hf->SetRangeUser("cut1name",cutmin1,cutmax1); | |
33 | hf->SetRangeUser(AliDielectronVarManager::kPt,ptmin,ptmax); | |
34 | hf->PrintCuts(); | |
5e2cf960 | 35 | |
c081415a | 36 | // collect 1-,2- or 3-dim histograms or profiles with error option (default:"") |
37 | TObjArray *arrHists = hf->CollectHistos(AliDielectronVarManager::kM); | |
38 | TObjArray *arrProfs = hf->CollectProfiles("",AliDielectronVarManager::kM,AliDielectronVarManager::kPt); | |
5e2cf960 | 39 | |
c081415a | 40 | // then you are left with an array of histograms for all pair types or MC signals |
5e2cf960 | 41 | |
42 | */ | |
43 | // // | |
44 | /////////////////////////////////////////////////////////////////////////// | |
45 | ||
46 | #include <TObjArray.h> | |
47 | #include <TKey.h> | |
48 | #include <TList.h> | |
49 | #include <TClass.h> | |
50 | #include <TObject.h> | |
51 | #include <TFile.h> | |
52 | #include <TString.h> | |
53 | #include <TObjString.h> | |
54 | #include <TVectorD.h> | |
55 | #include <TMath.h> | |
56 | #include <TH1.h> | |
57 | ||
58 | #include <AliLog.h> | |
59 | ||
443a091c | 60 | #include "AliDielectron.h" |
5e2cf960 | 61 | #include "AliDielectronHFhelper.h" |
62 | #include "AliDielectronHF.h" | |
63 | ||
c081415a | 64 | ClassImp(AliDielectronHFhelper) |
5e2cf960 | 65 | |
66 | //const char* AliDielectronHFhelper::fCutVars[AliDielectronHFhelper::kMaxCuts] = {"","","","","","","","","",""}; | |
67 | ||
68 | //________________________________________________________________ | |
e0f6eeb2 | 69 | AliDielectronHFhelper::AliDielectronHFhelper(const char* filename, const char* container) : |
5e2cf960 | 70 | TNamed(), |
c081415a | 71 | fMainArr(0x0), |
5e2cf960 | 72 | fCutVars(0x0), |
73 | fCutLowLimits(0), | |
74 | fCutUpLimits(0) | |
75 | { | |
76 | // | |
77 | // get HF container(s) from file 'filename' | |
78 | // | |
e0f6eeb2 | 79 | SetHFArray(filename, container); |
187de573 | 80 | |
5e2cf960 | 81 | } |
82 | ||
83 | //________________________________________________________________ | |
84 | AliDielectronHFhelper::~AliDielectronHFhelper() | |
85 | { | |
86 | // | |
87 | // dtor | |
88 | // | |
c081415a | 89 | if(fMainArr) delete fMainArr; |
5e2cf960 | 90 | if(fCutVars) delete fCutVars; |
187de573 | 91 | |
5e2cf960 | 92 | } |
93 | ||
94 | //________________________________________________________________ | |
e0f6eeb2 | 95 | void AliDielectronHFhelper::SetHFArray(const char* filename, const char* container) |
5e2cf960 | 96 | { |
97 | // | |
c081415a | 98 | // get HF container from file |
5e2cf960 | 99 | // |
100 | ||
101 | TFile *f = TFile::Open(filename); | |
102 | ||
103 | TList *l=f->GetListOfKeys(); | |
104 | TIter nextKey(l); | |
105 | TKey *k=0x0; | |
106 | while ( (k=static_cast<TKey*>(nextKey())) ){ | |
187de573 | 107 | |
5e2cf960 | 108 | TObject *o=k->ReadObj(); |
109 | if (o->IsA()==TList::Class()){ | |
187de573 | 110 | |
5e2cf960 | 111 | TList *tlist=(TList*)o; |
187de573 | 112 | |
5e2cf960 | 113 | TIter next(tlist); |
114 | TObject *obj=0x0; | |
115 | while ((obj = next())) { | |
116 | TString objname(obj->GetName()); | |
117 | ||
e0f6eeb2 | 118 | if( objname.Contains(Form("%s_HF",container)) && obj->IsA()==TObjArray::Class()) { |
c081415a | 119 | fMainArr = new TObjArray( *(dynamic_cast<TObjArray*>(obj)) ); |
d4619886 | 120 | fMainArr->SetOwner(); |
fbc09d5d | 121 | // fMainArr->Print(); |
5e2cf960 | 122 | return; |
123 | } | |
124 | } | |
125 | } | |
126 | } | |
127 | ||
128 | } | |
129 | //________________________________________________________________ | |
130 | void AliDielectronHFhelper::SetRangeUser(const char *varname, Double_t min, Double_t max, Bool_t leg) | |
131 | { | |
132 | // | |
133 | // Set range from variable name | |
134 | // | |
187de573 | 135 | |
5e2cf960 | 136 | Int_t size=fCutLowLimits.GetNrows(); |
137 | ||
187de573 | 138 | // check if cut is already set |
139 | for(Int_t icut=0; icut<size; icut++) { | |
140 | TString cutName = fCutVars->At(icut)->GetName(); | |
141 | if(!cutName.CompareTo(Form("%s%s",(leg?"Leg":""),varname))) { | |
142 | UnsetRangeUser(varname,leg); | |
143 | SetRangeUser(varname, min, max, leg); | |
144 | return; | |
145 | } | |
146 | } | |
147 | ||
5e2cf960 | 148 | if(size>=kMaxCuts) return; |
149 | ||
150 | // arrays | |
151 | if(!fCutVars) { | |
152 | fCutVars = new TObjArray(); | |
153 | fCutVars->SetOwner(); | |
154 | } | |
155 | fCutLowLimits.ResizeTo(size+1); | |
156 | fCutUpLimits.ResizeTo(size+1); | |
157 | ||
158 | // fill | |
5e2cf960 | 159 | TObjString *str = new TObjString(Form("%s%s",(leg?"Leg":""),varname)); |
160 | fCutVars->Add(str); | |
161 | fCutLowLimits(size) = min; | |
162 | fCutUpLimits(size) = max; | |
163 | AliWarning(Form(" %s [%.2f,%.2f]",fCutVars->At(size)->GetName(),fCutLowLimits(size),fCutUpLimits(size))); | |
164 | } | |
165 | ||
166 | //________________________________________________________________ | |
167 | void AliDielectronHFhelper::SetRangeUser(AliDielectronVarManager::ValueTypes type, Double_t min, Double_t max, Bool_t leg) | |
168 | { | |
169 | // | |
187de573 | 170 | // Set range from AliDielectronVarManager |
5e2cf960 | 171 | // |
172 | SetRangeUser(AliDielectronVarManager::GetValueName(type), min, max, leg); | |
173 | } | |
174 | ||
4d7704c5 | 175 | //________________________________________________________________ |
176 | void AliDielectronHFhelper::UnsetRangeUser(const char *varname, Bool_t leg) | |
177 | { | |
178 | // | |
179 | // unset range from variable name | |
180 | // | |
c081415a | 181 | |
4d7704c5 | 182 | Int_t size=fCutLowLimits.GetNrows(); |
4d7704c5 | 183 | TVectorD newlow; |
184 | TVectorD newup; | |
185 | ||
187de573 | 186 | // find cut and build new vectors w/o it |
4d7704c5 | 187 | Int_t ientries = 0; |
188 | for(Int_t icut=0; icut<size; icut++) { | |
189 | ||
190 | TString cutName = fCutVars->At(icut)->GetName(); | |
187de573 | 191 | if(!cutName.CompareTo(Form("%s%s",(leg?"Leg":""),varname))) { |
4d7704c5 | 192 | fCutVars->AddAt(0x0,icut); |
193 | continue; | |
194 | } | |
187de573 | 195 | |
4d7704c5 | 196 | // fill new vectors |
197 | newlow.ResizeTo(ientries+1); | |
198 | newup.ResizeTo(ientries+1); | |
199 | newlow(ientries) = fCutLowLimits(icut); | |
200 | newup(ientries) = fCutUpLimits(icut); | |
187de573 | 201 | |
4d7704c5 | 202 | ientries++; |
203 | } | |
187de573 | 204 | |
4d7704c5 | 205 | // adapt new arrays/vectors |
206 | fCutVars->Compress(); | |
207 | ||
208 | fCutLowLimits.ResizeTo(ientries); | |
209 | fCutUpLimits.ResizeTo(ientries); | |
210 | for(Int_t icut=0; icut<ientries; icut++) { | |
211 | fCutLowLimits(icut) = newlow(icut); | |
212 | fCutUpLimits(icut) = newup(icut); | |
213 | } | |
187de573 | 214 | // PrintCuts(); |
215 | ||
4d7704c5 | 216 | } |
217 | ||
218 | //________________________________________________________________ | |
219 | void AliDielectronHFhelper::UnsetRangeUser(AliDielectronVarManager::ValueTypes type, Bool_t leg) | |
220 | { | |
221 | // | |
187de573 | 222 | // Unset range from AliDielectronVarManager |
4d7704c5 | 223 | // |
224 | UnsetRangeUser(AliDielectronVarManager::GetValueName(type), leg); | |
225 | } | |
226 | ||
443a091c | 227 | //________________________________________________________________ |
c081415a | 228 | TObjArray* AliDielectronHFhelper::CollectProfiles(TString option, |
229 | AliDielectronVarManager::ValueTypes varx, | |
230 | AliDielectronVarManager::ValueTypes vary, | |
231 | AliDielectronVarManager::ValueTypes varz, | |
232 | AliDielectronVarManager::ValueTypes vart) | |
443a091c | 233 | { |
234 | // | |
c081415a | 235 | // collect 1-3 dimensional TProfiles for all kind of pair types or sources |
443a091c | 236 | // |
187de573 | 237 | |
c081415a | 238 | // reconstruct the histogram/profile name resp. key |
239 | Int_t dim = 0; | |
240 | if(varx < AliDielectronVarManager::kNMaxValues) dim++; | |
241 | if(vary < AliDielectronVarManager::kNMaxValues) dim++; | |
242 | if(varz < AliDielectronVarManager::kNMaxValues) dim++; | |
fbc09d5d | 243 | if(vart < AliDielectronVarManager::kNMaxValues) dim++; |
c081415a | 244 | Bool_t bPairClass=0; |
245 | if( varx < AliDielectronVarManager::kPairMax || | |
246 | vary < AliDielectronVarManager::kPairMax || | |
247 | varz < AliDielectronVarManager::kPairMax || | |
248 | vart < AliDielectronVarManager::kPairMax ) bPairClass=kTRUE; | |
d4619886 | 249 | Bool_t bwght= (option.Contains("weight",TString::kIgnoreCase)); // weighted histogram |
250 | Bool_t bprf = !(option.Contains("hist",TString::kIgnoreCase)); // tprofile | |
c081415a | 251 | Bool_t bStdOpt=kTRUE; |
252 | if(option.Contains("S",TString::kIgnoreCase) || option.Contains("rms",TString::kIgnoreCase)) bStdOpt=kFALSE; | |
253 | ||
254 | TString key = ""; | |
d4619886 | 255 | if(bwght) dim--; |
256 | if(bprf) dim--; | |
c081415a | 257 | switch(dim) { |
258 | case 3: | |
259 | key+=Form("%s_",AliDielectronVarManager::GetValueName(varx)); | |
260 | key+=Form("%s_",AliDielectronVarManager::GetValueName(vary)); | |
261 | key+=Form("%s",AliDielectronVarManager::GetValueName(varz)); | |
262 | if(bprf) key+=Form("-%s%s",AliDielectronVarManager::GetValueName(vart),(bStdOpt ? "avg" : "rms")); | |
263 | break; | |
264 | case 2: | |
265 | key+=Form("%s_",AliDielectronVarManager::GetValueName(varx)); | |
266 | key+=Form("%s",AliDielectronVarManager::GetValueName(vary)); | |
267 | if(bprf) key+=Form("-%s%s",AliDielectronVarManager::GetValueName(varz),(bStdOpt ? "avg" : "rms")); | |
268 | break; | |
269 | case 1: | |
270 | key+=Form("%s",AliDielectronVarManager::GetValueName(varx)); | |
271 | if(bprf) key+=Form("-%s%s",AliDielectronVarManager::GetValueName(vary),(bStdOpt ? "avg" : "rms")); | |
272 | break; | |
273 | } | |
274 | // to differentiate btw. leg and pair histos | |
275 | if(bPairClass) key.Prepend("p"); | |
276 | // prepend HF | |
277 | key.Prepend("HF_"); | |
278 | ||
d4619886 | 279 | // add the weighted part to the key |
280 | if(bwght) { | |
281 | if(bprf) dim++; | |
282 | switch(dim) { | |
283 | case 3: key+=Form("-wght%s",AliDielectronVarManager::GetValueName(vart)); break; | |
284 | case 2: key+=Form("-wght%s",AliDielectronVarManager::GetValueName(varz)); break; | |
285 | case 1: key+=Form("-wght%s",AliDielectronVarManager::GetValueName(vary)); break; | |
286 | case 4: AliError(Form(" NO weighted 3D profiles supported by the framework")); | |
287 | } | |
288 | } | |
289 | ||
290 | //printf("--------> KEY: %s \n",key.Data()); | |
291 | // init array of histograms of interest | |
c081415a | 292 | TObjArray *collection = new TObjArray(fMainArr->GetEntriesFast()); |
d4619886 | 293 | collection->SetOwner(kTRUE); |
c081415a | 294 | |
6ed39778 | 295 | TObjArray *cloneArr = new TObjArray( *(dynamic_cast<TObjArray*>(fMainArr)) ); |
c081415a | 296 | if(!cloneArr) return 0x0; |
297 | cloneArr->SetOwner(kTRUE); | |
298 | ||
299 | // loop over all pair types or sources | |
300 | for(Int_t i=0; i<cloneArr->GetEntriesFast(); i++) { | |
2d00c7d3 | 301 | if(!cloneArr->At(i)) continue; |
302 | if(!((TObjArray*)cloneArr->At(i))->GetEntries()) continue; | |
c081415a | 303 | |
d4619886 | 304 | // Get signal/source array with its histograms of interest |
6ed39778 | 305 | AliDebug(1,Form(" Looking into step %s selected",cloneArr->At(i)->GetName())); |
306 | TObjArray *arr = FindObjects((TObjArray*)cloneArr->At(i)); | |
c081415a | 307 | if(arr) { |
d4619886 | 308 | |
309 | // find requested histogram | |
c081415a | 310 | collection->AddAt(arr->FindObject(key.Data()), i); |
d4619886 | 311 | if(!collection->At(i)) { AliError(Form("Object %s does not exist",key.Data())); return 0x0; } |
312 | ||
313 | // modify the signal/source name if needed (MConly) | |
314 | TString stepName = cloneArr->At(i)->GetName(); | |
2d00c7d3 | 315 | stepName.ReplaceAll("(",""); |
316 | stepName.ReplaceAll(")",""); | |
317 | stepName.ReplaceAll(": ",""); | |
318 | stepName.ReplaceAll(" ","_"); | |
319 | stepName.ReplaceAll("Signal",""); | |
320 | ((TH1*)collection->At(i))->SetName(Form("%s_%s",key.Data(),stepName.Data())); | |
c081415a | 321 | } |
6ed39778 | 322 | else |
323 | AliError(Form("Step %d not found",i)); | |
187de573 | 324 | |
325 | } | |
326 | ||
6ed39778 | 327 | //clean up |
328 | //delete cloneArr; | |
329 | //cloneArr=0; | |
443a091c | 330 | |
331 | return collection; | |
332 | } | |
333 | ||
5e2cf960 | 334 | //________________________________________________________________ |
6ed39778 | 335 | TObjArray* AliDielectronHFhelper::FindObjects(TObjArray *stepArr) |
5e2cf960 | 336 | { |
d4619886 | 337 | // rename DoCuts, return values is a tobjarray |
338 | // apply cuts and exclude objects from the array for merging (CUT selection) | |
5e2cf960 | 339 | // |
187de573 | 340 | |
5e2cf960 | 341 | // debug |
c081415a | 342 | // TString title = stepArr->At(0)->GetTitle(); |
5e2cf960 | 343 | // TObjArray* vars = title.Tokenize(":"); |
187de573 | 344 | // AliDebug(1,Form(" number of cuts/vars: %d/%d",fCutLowLimits.GetNrows(),vars->GetEntriesFast())); |
6ed39778 | 345 | if(!stepArr) { AliError("step is empty"); return 0x0;} |
187de573 | 346 | |
5e2cf960 | 347 | // check for missing cuts |
c081415a | 348 | CheckCuts(stepArr); |
187de573 | 349 | |
5e2cf960 | 350 | // loop over all cuts |
351 | for(Int_t icut=0; icut<fCutLowLimits.GetNrows(); icut++) { | |
352 | ||
c081415a | 353 | Bool_t bFndBin = kFALSE; // bin with exact limits found |
354 | const char *cutvar = fCutVars->At(icut)->GetName(); // cut variable | |
355 | Double_t min = fCutLowLimits(icut); // lower limit | |
356 | Double_t max = fCutUpLimits(icut); // upper limit | |
187de573 | 357 | AliDebug(5,Form(" Cut %d: %s [%.2f,%.2f]",icut,cutvar,min,max)); |
5e2cf960 | 358 | |
d4619886 | 359 | // loop over the full grid of the signalArray (pair,source) |
c081415a | 360 | for(Int_t i=0; i<stepArr->GetEntriesFast(); i++) { |
5e2cf960 | 361 | |
362 | // continue if already empty | |
c081415a | 363 | if(!stepArr->At(i)) continue; |
5e2cf960 | 364 | |
c081415a | 365 | // collect bins from the name |
366 | TString title = stepArr->At(i)->GetName(); | |
5e2cf960 | 367 | if(title.IsNull()) continue; |
c081415a | 368 | AliDebug(5,Form(" %03d object name: %s",i,title.Data())); |
5e2cf960 | 369 | |
c081415a | 370 | // loop over all variables |
5e2cf960 | 371 | TObjArray *vars = title.Tokenize(":"); |
372 | for(Int_t ivar=0; ivar<vars->GetEntriesFast(); ivar++) { | |
373 | TString binvar = vars->At(ivar)->GetName(); | |
c081415a | 374 | AliDebug(10,Form(" --> %d check bin var %s",ivar,binvar.Data())); |
187de573 | 375 | |
c081415a | 376 | // check cuts set by the user, and compare to the bin limits |
5e2cf960 | 377 | if(binvar.Contains(cutvar)) { |
c081415a | 378 | // bin limits |
5e2cf960 | 379 | TObjArray *limits = binvar.Tokenize("#"); |
f969027b | 380 | if(!limits) continue; |
c081415a | 381 | Double_t binmin = atof(limits->At(1)->GetName()); // lower bin limit |
382 | Double_t binmax = atof(limits->At(2)->GetName()); // upper bin limit | |
187de573 | 383 | AliDebug(10,Form(" bin %s var %s [%.2f,%.2f]",binvar.Data(),limits->At(0)->GetName(),binmin,binmax)); |
f969027b | 384 | delete limits; |
187de573 | 385 | |
c081415a | 386 | // cut and remove objects from the array |
5e2cf960 | 387 | if(binmin < min || binmax < min || binmin > max || binmax > max ) { |
c081415a | 388 | AliDebug(10,Form(" removed, out of range! lower bin,cut: %.2f,%.2f or upper bin,cut: %.2f>%.2f",binmin,min,binmax,max)); |
389 | stepArr->AddAt(0x0,i); | |
5e2cf960 | 390 | } |
391 | if(bFndBin && !(binmin == min && binmax == max)) { | |
c081415a | 392 | stepArr->AddAt(0x0,i); |
393 | AliDebug(10,Form(" removed, within range! lower bin,cut: %.2f,%.2f or upper bin,cut: %.2f,%.2f",binmin,min,binmax,max)); | |
8d42b5b9 | 394 | } |
5e2cf960 | 395 | |
c081415a | 396 | // did we found a bin with exact the cut ranges |
397 | // this can happen only once per variable | |
5e2cf960 | 398 | if(binmin==min && binmax==max) bFndBin=kTRUE; |
399 | ||
400 | } | |
401 | ||
402 | } | |
403 | // clean up | |
404 | if(vars) delete vars; | |
187de573 | 405 | |
d4619886 | 406 | } //end loop: pair step |
187de573 | 407 | |
d4619886 | 408 | } //end: loop cuts |
187de573 | 409 | |
c081415a | 410 | // compress the array by removing all empty entries |
411 | stepArr->Compress(); | |
6ed39778 | 412 | AliDebug(1,Form(" Compression: %d arrays left",stepArr->GetEntriesFast())); |
5e2cf960 | 413 | |
c081415a | 414 | // merge left objects |
6ed39778 | 415 | TObjArray* hist = Merge(stepArr); |
416 | if(hist) AliDebug(1,Form(" final array has %d objects",hist->GetEntries())); | |
5e2cf960 | 417 | return hist; |
418 | } | |
419 | ||
420 | //________________________________________________________________ | |
6ed39778 | 421 | TObjArray* AliDielectronHFhelper::Merge(TObjArray *arr) |
5e2cf960 | 422 | { |
6ed39778 | 423 | // |
d4619886 | 424 | // merge left objects into a single one (LAST step) |
5e2cf960 | 425 | // |
5e2cf960 | 426 | |
c081415a | 427 | if(arr->GetEntriesFast()<1) { AliError(" No more objects left!"); return 0x0; } |
187de573 | 428 | |
d4619886 | 429 | TObjArray *final=(TObjArray*) arr->At(0)->Clone(); |
5e2cf960 | 430 | if(!final) return 0x0; |
d4619886 | 431 | final->SetOwner(); |
187de573 | 432 | |
35363fb9 | 433 | TList listH; |
434 | TString listHargs; | |
435 | listHargs.Form("((TCollection*)0x%lx)", (ULong_t)&listH); | |
436 | Int_t error = 0; | |
437 | ||
438 | // final->Reset("CE"); | |
439 | // final->SetTitle(""); //TODO: change in future | |
440 | for(Int_t i=1; i<arr->GetEntriesFast(); i++) { | |
441 | listH.Add(arr->At(i)); | |
c081415a | 442 | // printf("%d: ent %.0f \n",i,((TH1*)((TObjArray*)arr->At(i))->At(0))->GetEntries()); |
35363fb9 | 443 | // final->Add((TH1*)arr->At(i)); |
5e2cf960 | 444 | } |
35363fb9 | 445 | // arr->Clear(); |
5e2cf960 | 446 | |
35363fb9 | 447 | final->Execute("Merge", listHargs.Data(), &error); |
d4619886 | 448 | // returns a tobjarray of histograms/profiles |
c081415a | 449 | return final; |
5e2cf960 | 450 | } |
451 | ||
452 | //________________________________________________________________ | |
453 | void AliDielectronHFhelper::CheckCuts(TObjArray *arr) | |
454 | { | |
455 | // | |
c081415a | 456 | // Compare binning and cut variables. Add necessary cuts (full range, no exclusion) |
5e2cf960 | 457 | // |
458 | ||
5e2cf960 | 459 | // build array with bin variable, minimum and maximum bin values |
35363fb9 | 460 | TString titleFIRST = arr->First()->GetName(); |
461 | TString titleLAST = arr->Last()->GetName(); | |
5e2cf960 | 462 | TObjArray* binvarsF = titleFIRST.Tokenize(":#"); |
463 | TObjArray* binvarsL = titleLAST.Tokenize(":#"); | |
464 | Double_t binmin[kMaxCuts]= {0.0}; | |
465 | Double_t binmax[kMaxCuts]= {0.0}; | |
f969027b | 466 | if(!binvarsF) { delete binvarsL; return; } |
467 | if(!binvarsL) { delete binvarsF; return; } | |
5e2cf960 | 468 | for(Int_t ivar=0; ivar<binvarsF->GetEntriesFast(); ivar++) { |
469 | ||
470 | TString elementF=binvarsF->At(ivar)->GetName(); | |
471 | TString elementL=binvarsL->At(ivar)->GetName(); | |
472 | AliDebug(1,Form(" binvar %d: %s,%s",ivar,elementF.Data(),elementL.Data())); | |
187de573 | 473 | |
5e2cf960 | 474 | switch(ivar%3) { |
475 | case 0: continue; break; | |
476 | case 1: binmin[(int)ivar/3]=atof(elementF.Data()); break; | |
477 | case 2: binmax[(int)ivar/3]=atof(elementL.Data()); break; | |
478 | } | |
187de573 | 479 | |
5e2cf960 | 480 | binvarsF->AddAt(0x0,ivar); |
481 | } | |
482 | binvarsF->Compress(); | |
187de573 | 483 | |
5e2cf960 | 484 | // loop over all vars and cuts, check for missing stuff |
485 | for(Int_t ivar=0; ivar<binvarsF->GetEntriesFast(); ivar++) { | |
187de573 | 486 | |
5e2cf960 | 487 | TString binvar=binvarsF->At(ivar)->GetName(); |
488 | Bool_t selected=kFALSE; | |
489 | ||
490 | AliDebug(1,Form(" check cuts %d %s [%.2f,%.2f]",ivar,binvar.Data(),binmin[ivar],binmax[ivar])); | |
491 | // loop over all cuts and check for missing stuff | |
492 | for(Int_t icut=0; icut<fCutLowLimits.GetNrows(); icut++) { | |
493 | if(binvar.Contains(fCutVars->At(icut)->GetName())) { selected=kTRUE; break; } | |
494 | // else break; | |
495 | } | |
187de573 | 496 | |
5e2cf960 | 497 | // add missing cut with max limits |
498 | if(!selected) { | |
187de573 | 499 | AliWarning(Form(" Bin variable %s not covered. Add cut!",binvar.Data())); |
500 | Bool_t leg = binvar.BeginsWith("Leg"); | |
501 | if(leg) binvar.Remove(0,3); | |
502 | SetRangeUser(binvar.Data(),binmin[ivar],binmax[ivar], leg); | |
5e2cf960 | 503 | } |
187de573 | 504 | |
5e2cf960 | 505 | } |
187de573 | 506 | |
507 | // clean up | |
f969027b | 508 | delete binvarsF; |
509 | delete binvarsL; | |
5e2cf960 | 510 | } |
511 | ||
512 | //________________________________________________________________ | |
443a091c | 513 | void AliDielectronHFhelper::Print(const Option_t* /*option*/) const |
5e2cf960 | 514 | { |
515 | ||
516 | // | |
517 | // Print out object contents | |
518 | // | |
c081415a | 519 | AliInfo(Form(" Container: %s",fMainArr->GetName())); |
5e2cf960 | 520 | |
521 | // pairtypes, steps and sources | |
c081415a | 522 | Int_t stepLast=0; |
523 | AliInfo(Form(" Number of filled steps: %d",fMainArr->GetEntries())); | |
524 | for(Int_t istep=0; istep<fMainArr->GetEntriesFast(); istep++) { | |
2d00c7d3 | 525 | if(!fMainArr->At(istep)) continue; |
526 | if(!((TObjArray*)fMainArr->At(istep))->GetEntries()) continue; | |
527 | AliInfo(Form(" step %d: %s",istep,fMainArr->At(istep)->GetName())); | |
528 | stepLast=istep; | |
187de573 | 529 | } |
530 | ||
c081415a | 531 | AliInfo(Form(" Number of objects: %d", |
532 | ((TObjArray*) ((TObjArray*)fMainArr->At(stepLast)) ->First())->GetEntriesFast())); | |
187de573 | 533 | |
c081415a | 534 | TString title = ((TObjArray*)fMainArr->At(stepLast))->First()->GetName(); |
187de573 | 535 | TObjArray* binvars = title.Tokenize(":"); |
536 | AliInfo(Form(" Number of variables: %d",binvars->GetEntriesFast())); | |
09d5920f | 537 | delete binvars; |
187de573 | 538 | |
d4619886 | 539 | /* REACTIVATE |
540 | // check for missing cuts | |
541 | CheckCuts(((TObjArray*)fMainArr->At(stepLast))); | |
542 | // loop over all cuts | |
543 | for(Int_t icut=0; icut<fCutLowLimits.GetNrows(); icut++) { | |
544 | const char *cutvar = fCutVars->At(icut)->GetName(); // cut variable | |
545 | Double_t min = fCutLowLimits(icut); // lower limit | |
546 | Double_t max = fCutUpLimits(icut); // upper limit | |
547 | AliInfo(Form(" variable %d: %s [%.2f,%.2f]",icut,cutvar,min,max)); | |
548 | } | |
549 | */ | |
187de573 | 550 | TObjArray* binvars2 = title.Tokenize(":#"); |
551 | for(Int_t ivar=0; ivar<binvars2->GetEntriesFast(); ivar++) { | |
552 | if(ivar%3) continue; | |
553 | AliInfo(Form(" variable %.0f: %s",((Double_t)ivar)/3+1,binvars2->At(ivar)->GetName())); | |
5e2cf960 | 554 | } |
09d5920f | 555 | delete binvars2; |
5e2cf960 | 556 | |
557 | } | |
558 | ||
559 | //________________________________________________________________ | |
560 | void AliDielectronHFhelper::PrintCuts() | |
561 | { | |
562 | ||
563 | // | |
c081415a | 564 | // Print cuts |
5e2cf960 | 565 | // |
566 | ||
567 | // loop over all cuts | |
568 | AliInfo(" Selected cuts:"); | |
569 | for(Int_t icut=0; icut<fCutLowLimits.GetNrows(); icut++) | |
570 | AliInfo(Form(" %d: %s [%.2f,%.2f]",icut,fCutVars->At(icut)->GetName(),fCutLowLimits(icut),fCutUpLimits(icut))); | |
187de573 | 571 | |
5e2cf960 | 572 | } |
573 |