]>
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(); |
c081415a | 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++; | |
243 | Bool_t bPairClass=0; | |
244 | if( varx < AliDielectronVarManager::kPairMax || | |
245 | vary < AliDielectronVarManager::kPairMax || | |
246 | varz < AliDielectronVarManager::kPairMax || | |
247 | vart < AliDielectronVarManager::kPairMax ) bPairClass=kTRUE; | |
d4619886 | 248 | Bool_t bwght= (option.Contains("weight",TString::kIgnoreCase)); // weighted histogram |
249 | Bool_t bprf = !(option.Contains("hist",TString::kIgnoreCase)); // tprofile | |
c081415a | 250 | Bool_t bStdOpt=kTRUE; |
251 | if(option.Contains("S",TString::kIgnoreCase) || option.Contains("rms",TString::kIgnoreCase)) bStdOpt=kFALSE; | |
252 | ||
253 | TString key = ""; | |
d4619886 | 254 | if(bwght) dim--; |
255 | if(bprf) dim--; | |
c081415a | 256 | switch(dim) { |
257 | case 3: | |
258 | key+=Form("%s_",AliDielectronVarManager::GetValueName(varx)); | |
259 | key+=Form("%s_",AliDielectronVarManager::GetValueName(vary)); | |
260 | key+=Form("%s",AliDielectronVarManager::GetValueName(varz)); | |
261 | if(bprf) key+=Form("-%s%s",AliDielectronVarManager::GetValueName(vart),(bStdOpt ? "avg" : "rms")); | |
262 | break; | |
263 | case 2: | |
264 | key+=Form("%s_",AliDielectronVarManager::GetValueName(varx)); | |
265 | key+=Form("%s",AliDielectronVarManager::GetValueName(vary)); | |
266 | if(bprf) key+=Form("-%s%s",AliDielectronVarManager::GetValueName(varz),(bStdOpt ? "avg" : "rms")); | |
267 | break; | |
268 | case 1: | |
269 | key+=Form("%s",AliDielectronVarManager::GetValueName(varx)); | |
270 | if(bprf) key+=Form("-%s%s",AliDielectronVarManager::GetValueName(vary),(bStdOpt ? "avg" : "rms")); | |
271 | break; | |
272 | } | |
273 | // to differentiate btw. leg and pair histos | |
274 | if(bPairClass) key.Prepend("p"); | |
275 | // prepend HF | |
276 | key.Prepend("HF_"); | |
277 | ||
d4619886 | 278 | // add the weighted part to the key |
279 | if(bwght) { | |
280 | if(bprf) dim++; | |
281 | switch(dim) { | |
282 | case 3: key+=Form("-wght%s",AliDielectronVarManager::GetValueName(vart)); break; | |
283 | case 2: key+=Form("-wght%s",AliDielectronVarManager::GetValueName(varz)); break; | |
284 | case 1: key+=Form("-wght%s",AliDielectronVarManager::GetValueName(vary)); break; | |
285 | case 4: AliError(Form(" NO weighted 3D profiles supported by the framework")); | |
286 | } | |
287 | } | |
288 | ||
289 | //printf("--------> KEY: %s \n",key.Data()); | |
290 | // init array of histograms of interest | |
c081415a | 291 | TObjArray *collection = new TObjArray(fMainArr->GetEntriesFast()); |
d4619886 | 292 | collection->SetOwner(kTRUE); |
c081415a | 293 | |
294 | TObjArray *cloneArr = (TObjArray*) fMainArr->Clone("tmpArr"); | |
295 | if(!cloneArr) return 0x0; | |
296 | cloneArr->SetOwner(kTRUE); | |
297 | ||
298 | // loop over all pair types or sources | |
299 | for(Int_t i=0; i<cloneArr->GetEntriesFast(); i++) { | |
2d00c7d3 | 300 | if(!cloneArr->At(i)) continue; |
301 | if(!((TObjArray*)cloneArr->At(i))->GetEntries()) continue; | |
c081415a | 302 | |
d4619886 | 303 | // Get signal/source array with its histograms of interest |
c081415a | 304 | TObjArray *arr = (TObjArray*) GetObject(cloneArr->At(i)->GetName(),cloneArr); |
305 | if(arr) { | |
d4619886 | 306 | |
307 | // find requested histogram | |
c081415a | 308 | collection->AddAt(arr->FindObject(key.Data()), i); |
d4619886 | 309 | if(!collection->At(i)) { AliError(Form("Object %s does not exist",key.Data())); return 0x0; } |
310 | ||
311 | // modify the signal/source name if needed (MConly) | |
312 | TString stepName = cloneArr->At(i)->GetName(); | |
2d00c7d3 | 313 | stepName.ReplaceAll("(",""); |
314 | stepName.ReplaceAll(")",""); | |
315 | stepName.ReplaceAll(": ",""); | |
316 | stepName.ReplaceAll(" ","_"); | |
317 | stepName.ReplaceAll("Signal",""); | |
318 | ((TH1*)collection->At(i))->SetName(Form("%s_%s",key.Data(),stepName.Data())); | |
c081415a | 319 | } |
187de573 | 320 | |
321 | } | |
322 | ||
323 | // clean up the clone | |
c081415a | 324 | delete cloneArr; |
325 | cloneArr=0; | |
443a091c | 326 | |
327 | return collection; | |
328 | } | |
329 | ||
5e2cf960 | 330 | //________________________________________________________________ |
c081415a | 331 | TObject* AliDielectronHFhelper::GetObject(const char *step, TObjArray *histArr) |
5e2cf960 | 332 | { |
d4619886 | 333 | // rename into GetSignalArray, return value is a tobjarray |
5e2cf960 | 334 | // |
d4619886 | 335 | // main function to receive a pair type or MC signal array |
336 | // with all its merged histograms valid for the cuts | |
c081415a | 337 | // |
187de573 | 338 | |
339 | AliDebug(1,Form(" Step %s selected",step)); | |
c081415a | 340 | // TODO: check memory |
187de573 | 341 | |
c081415a | 342 | TObjArray *stepArr = 0x0; // this is the requested step |
d4619886 | 343 | TObject *hist = 0x0; // this is the returned object |
187de573 | 344 | if(!histArr) { |
c081415a | 345 | stepArr = (TObjArray*) fMainArr->FindObject(step)->Clone("tmpArr"); |
187de573 | 346 | } |
347 | else { | |
c081415a | 348 | stepArr = (TObjArray*) histArr->FindObject(step); |
187de573 | 349 | } |
5e2cf960 | 350 | |
d4619886 | 351 | // apply cuts and get merged objects |
c081415a | 352 | if(stepArr) hist = FindObjects(stepArr); |
5e2cf960 | 353 | return hist; |
187de573 | 354 | |
5e2cf960 | 355 | } |
356 | ||
357 | //________________________________________________________________ | |
c081415a | 358 | TObject* AliDielectronHFhelper::FindObjects(TObjArray *stepArr) |
5e2cf960 | 359 | { |
d4619886 | 360 | // rename DoCuts, return values is a tobjarray |
361 | // apply cuts and exclude objects from the array for merging (CUT selection) | |
5e2cf960 | 362 | // |
187de573 | 363 | |
5e2cf960 | 364 | // debug |
c081415a | 365 | // TString title = stepArr->At(0)->GetTitle(); |
5e2cf960 | 366 | // TObjArray* vars = title.Tokenize(":"); |
187de573 | 367 | // AliDebug(1,Form(" number of cuts/vars: %d/%d",fCutLowLimits.GetNrows(),vars->GetEntriesFast())); |
368 | ||
5e2cf960 | 369 | // check for missing cuts |
c081415a | 370 | CheckCuts(stepArr); |
187de573 | 371 | |
5e2cf960 | 372 | // loop over all cuts |
373 | for(Int_t icut=0; icut<fCutLowLimits.GetNrows(); icut++) { | |
374 | ||
c081415a | 375 | Bool_t bFndBin = kFALSE; // bin with exact limits found |
376 | const char *cutvar = fCutVars->At(icut)->GetName(); // cut variable | |
377 | Double_t min = fCutLowLimits(icut); // lower limit | |
378 | Double_t max = fCutUpLimits(icut); // upper limit | |
187de573 | 379 | AliDebug(5,Form(" Cut %d: %s [%.2f,%.2f]",icut,cutvar,min,max)); |
5e2cf960 | 380 | |
d4619886 | 381 | // loop over the full grid of the signalArray (pair,source) |
c081415a | 382 | for(Int_t i=0; i<stepArr->GetEntriesFast(); i++) { |
5e2cf960 | 383 | |
384 | // continue if already empty | |
c081415a | 385 | if(!stepArr->At(i)) continue; |
5e2cf960 | 386 | |
c081415a | 387 | // collect bins from the name |
388 | TString title = stepArr->At(i)->GetName(); | |
5e2cf960 | 389 | if(title.IsNull()) continue; |
c081415a | 390 | AliDebug(5,Form(" %03d object name: %s",i,title.Data())); |
5e2cf960 | 391 | |
c081415a | 392 | // loop over all variables |
5e2cf960 | 393 | TObjArray *vars = title.Tokenize(":"); |
394 | for(Int_t ivar=0; ivar<vars->GetEntriesFast(); ivar++) { | |
395 | TString binvar = vars->At(ivar)->GetName(); | |
c081415a | 396 | AliDebug(10,Form(" --> %d check bin var %s",ivar,binvar.Data())); |
187de573 | 397 | |
c081415a | 398 | // check cuts set by the user, and compare to the bin limits |
5e2cf960 | 399 | if(binvar.Contains(cutvar)) { |
c081415a | 400 | // bin limits |
5e2cf960 | 401 | TObjArray *limits = binvar.Tokenize("#"); |
f969027b | 402 | if(!limits) continue; |
c081415a | 403 | Double_t binmin = atof(limits->At(1)->GetName()); // lower bin limit |
404 | Double_t binmax = atof(limits->At(2)->GetName()); // upper bin limit | |
187de573 | 405 | AliDebug(10,Form(" bin %s var %s [%.2f,%.2f]",binvar.Data(),limits->At(0)->GetName(),binmin,binmax)); |
f969027b | 406 | delete limits; |
187de573 | 407 | |
c081415a | 408 | // cut and remove objects from the array |
5e2cf960 | 409 | if(binmin < min || binmax < min || binmin > max || binmax > max ) { |
c081415a | 410 | AliDebug(10,Form(" removed, out of range! lower bin,cut: %.2f,%.2f or upper bin,cut: %.2f>%.2f",binmin,min,binmax,max)); |
411 | stepArr->AddAt(0x0,i); | |
5e2cf960 | 412 | } |
413 | if(bFndBin && !(binmin == min && binmax == max)) { | |
c081415a | 414 | stepArr->AddAt(0x0,i); |
415 | AliDebug(10,Form(" removed, within range! lower bin,cut: %.2f,%.2f or upper bin,cut: %.2f,%.2f",binmin,min,binmax,max)); | |
8d42b5b9 | 416 | } |
5e2cf960 | 417 | |
c081415a | 418 | // did we found a bin with exact the cut ranges |
419 | // this can happen only once per variable | |
5e2cf960 | 420 | if(binmin==min && binmax==max) bFndBin=kTRUE; |
421 | ||
422 | } | |
423 | ||
424 | } | |
425 | // clean up | |
426 | if(vars) delete vars; | |
187de573 | 427 | |
d4619886 | 428 | } //end loop: pair step |
187de573 | 429 | |
d4619886 | 430 | } //end: loop cuts |
187de573 | 431 | |
c081415a | 432 | // compress the array by removing all empty entries |
433 | stepArr->Compress(); | |
434 | AliDebug(1,Form(" Compression: %d objects left",stepArr->GetEntriesFast())); | |
5e2cf960 | 435 | |
c081415a | 436 | // merge left objects |
437 | TObject* hist = Merge(stepArr); | |
438 | // if(hist) AliDebug(1,Form(" Merging: %e entries",hist->GetEntries())); | |
5e2cf960 | 439 | return hist; |
440 | } | |
441 | ||
442 | //________________________________________________________________ | |
c081415a | 443 | TObject* AliDielectronHFhelper::Merge(TObjArray *arr) |
5e2cf960 | 444 | { |
d4619886 | 445 | // returned object is tobjarray |
446 | // merge left objects into a single one (LAST step) | |
5e2cf960 | 447 | // |
5e2cf960 | 448 | |
c081415a | 449 | if(arr->GetEntriesFast()<1) { AliError(" No more objects left!"); return 0x0; } |
187de573 | 450 | |
d4619886 | 451 | TObjArray *final=(TObjArray*) arr->At(0)->Clone(); |
5e2cf960 | 452 | if(!final) return 0x0; |
d4619886 | 453 | final->SetOwner(); |
187de573 | 454 | |
35363fb9 | 455 | TList listH; |
456 | TString listHargs; | |
457 | listHargs.Form("((TCollection*)0x%lx)", (ULong_t)&listH); | |
458 | Int_t error = 0; | |
459 | ||
460 | // final->Reset("CE"); | |
461 | // final->SetTitle(""); //TODO: change in future | |
462 | for(Int_t i=1; i<arr->GetEntriesFast(); i++) { | |
463 | listH.Add(arr->At(i)); | |
c081415a | 464 | // printf("%d: ent %.0f \n",i,((TH1*)((TObjArray*)arr->At(i))->At(0))->GetEntries()); |
35363fb9 | 465 | // final->Add((TH1*)arr->At(i)); |
5e2cf960 | 466 | } |
35363fb9 | 467 | // arr->Clear(); |
5e2cf960 | 468 | |
35363fb9 | 469 | final->Execute("Merge", listHargs.Data(), &error); |
d4619886 | 470 | // returns a tobjarray of histograms/profiles |
c081415a | 471 | return final; |
5e2cf960 | 472 | } |
473 | ||
474 | //________________________________________________________________ | |
475 | void AliDielectronHFhelper::CheckCuts(TObjArray *arr) | |
476 | { | |
477 | // | |
c081415a | 478 | // Compare binning and cut variables. Add necessary cuts (full range, no exclusion) |
5e2cf960 | 479 | // |
480 | ||
5e2cf960 | 481 | // build array with bin variable, minimum and maximum bin values |
35363fb9 | 482 | TString titleFIRST = arr->First()->GetName(); |
483 | TString titleLAST = arr->Last()->GetName(); | |
5e2cf960 | 484 | TObjArray* binvarsF = titleFIRST.Tokenize(":#"); |
485 | TObjArray* binvarsL = titleLAST.Tokenize(":#"); | |
486 | Double_t binmin[kMaxCuts]= {0.0}; | |
487 | Double_t binmax[kMaxCuts]= {0.0}; | |
f969027b | 488 | if(!binvarsF) { delete binvarsL; return; } |
489 | if(!binvarsL) { delete binvarsF; return; } | |
5e2cf960 | 490 | for(Int_t ivar=0; ivar<binvarsF->GetEntriesFast(); ivar++) { |
491 | ||
492 | TString elementF=binvarsF->At(ivar)->GetName(); | |
493 | TString elementL=binvarsL->At(ivar)->GetName(); | |
494 | AliDebug(1,Form(" binvar %d: %s,%s",ivar,elementF.Data(),elementL.Data())); | |
187de573 | 495 | |
5e2cf960 | 496 | switch(ivar%3) { |
497 | case 0: continue; break; | |
498 | case 1: binmin[(int)ivar/3]=atof(elementF.Data()); break; | |
499 | case 2: binmax[(int)ivar/3]=atof(elementL.Data()); break; | |
500 | } | |
187de573 | 501 | |
5e2cf960 | 502 | binvarsF->AddAt(0x0,ivar); |
503 | } | |
504 | binvarsF->Compress(); | |
187de573 | 505 | |
5e2cf960 | 506 | // loop over all vars and cuts, check for missing stuff |
507 | for(Int_t ivar=0; ivar<binvarsF->GetEntriesFast(); ivar++) { | |
187de573 | 508 | |
5e2cf960 | 509 | TString binvar=binvarsF->At(ivar)->GetName(); |
510 | Bool_t selected=kFALSE; | |
511 | ||
512 | AliDebug(1,Form(" check cuts %d %s [%.2f,%.2f]",ivar,binvar.Data(),binmin[ivar],binmax[ivar])); | |
513 | // loop over all cuts and check for missing stuff | |
514 | for(Int_t icut=0; icut<fCutLowLimits.GetNrows(); icut++) { | |
515 | if(binvar.Contains(fCutVars->At(icut)->GetName())) { selected=kTRUE; break; } | |
516 | // else break; | |
517 | } | |
187de573 | 518 | |
5e2cf960 | 519 | // add missing cut with max limits |
520 | if(!selected) { | |
187de573 | 521 | AliWarning(Form(" Bin variable %s not covered. Add cut!",binvar.Data())); |
522 | Bool_t leg = binvar.BeginsWith("Leg"); | |
523 | if(leg) binvar.Remove(0,3); | |
524 | SetRangeUser(binvar.Data(),binmin[ivar],binmax[ivar], leg); | |
5e2cf960 | 525 | } |
187de573 | 526 | |
5e2cf960 | 527 | } |
187de573 | 528 | |
529 | // clean up | |
f969027b | 530 | delete binvarsF; |
531 | delete binvarsL; | |
5e2cf960 | 532 | } |
533 | ||
534 | //________________________________________________________________ | |
443a091c | 535 | void AliDielectronHFhelper::Print(const Option_t* /*option*/) const |
5e2cf960 | 536 | { |
537 | ||
538 | // | |
539 | // Print out object contents | |
540 | // | |
c081415a | 541 | AliInfo(Form(" Container: %s",fMainArr->GetName())); |
5e2cf960 | 542 | |
543 | // pairtypes, steps and sources | |
c081415a | 544 | Int_t stepLast=0; |
545 | AliInfo(Form(" Number of filled steps: %d",fMainArr->GetEntries())); | |
546 | for(Int_t istep=0; istep<fMainArr->GetEntriesFast(); istep++) { | |
2d00c7d3 | 547 | if(!fMainArr->At(istep)) continue; |
548 | if(!((TObjArray*)fMainArr->At(istep))->GetEntries()) continue; | |
549 | AliInfo(Form(" step %d: %s",istep,fMainArr->At(istep)->GetName())); | |
550 | stepLast=istep; | |
187de573 | 551 | } |
552 | ||
c081415a | 553 | AliInfo(Form(" Number of objects: %d", |
554 | ((TObjArray*) ((TObjArray*)fMainArr->At(stepLast)) ->First())->GetEntriesFast())); | |
187de573 | 555 | |
c081415a | 556 | TString title = ((TObjArray*)fMainArr->At(stepLast))->First()->GetName(); |
187de573 | 557 | TObjArray* binvars = title.Tokenize(":"); |
558 | AliInfo(Form(" Number of variables: %d",binvars->GetEntriesFast())); | |
09d5920f | 559 | delete binvars; |
187de573 | 560 | |
d4619886 | 561 | /* REACTIVATE |
562 | // check for missing cuts | |
563 | CheckCuts(((TObjArray*)fMainArr->At(stepLast))); | |
564 | // loop over all cuts | |
565 | for(Int_t icut=0; icut<fCutLowLimits.GetNrows(); icut++) { | |
566 | const char *cutvar = fCutVars->At(icut)->GetName(); // cut variable | |
567 | Double_t min = fCutLowLimits(icut); // lower limit | |
568 | Double_t max = fCutUpLimits(icut); // upper limit | |
569 | AliInfo(Form(" variable %d: %s [%.2f,%.2f]",icut,cutvar,min,max)); | |
570 | } | |
571 | */ | |
187de573 | 572 | TObjArray* binvars2 = title.Tokenize(":#"); |
573 | for(Int_t ivar=0; ivar<binvars2->GetEntriesFast(); ivar++) { | |
574 | if(ivar%3) continue; | |
575 | AliInfo(Form(" variable %.0f: %s",((Double_t)ivar)/3+1,binvars2->At(ivar)->GetName())); | |
5e2cf960 | 576 | } |
09d5920f | 577 | delete binvars2; |
5e2cf960 | 578 | |
579 | } | |
580 | ||
581 | //________________________________________________________________ | |
582 | void AliDielectronHFhelper::PrintCuts() | |
583 | { | |
584 | ||
585 | // | |
c081415a | 586 | // Print cuts |
5e2cf960 | 587 | // |
588 | ||
589 | // loop over all cuts | |
590 | AliInfo(" Selected cuts:"); | |
591 | for(Int_t icut=0; icut<fCutLowLimits.GetNrows(); icut++) | |
592 | AliInfo(Form(" %d: %s [%.2f,%.2f]",icut,fCutVars->At(icut)->GetName(),fCutLowLimits(icut),fCutUpLimits(icut))); | |
187de573 | 593 | |
5e2cf960 | 594 | } |
595 |