]>
Commit | Line | Data |
---|---|---|
db6722a5 | 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 | // // | |
17 | // AliCFGridSparse Class // | |
18 | // Class to accumulate data on an N-dimensional grid, to be used // | |
19 | // as input to get corrections for Reconstruction & Trigger efficiency// | |
20 | // Based on root THnSparse // | |
21 | // -- Author : S.Arcelli // | |
22 | // Still to be done: // | |
23 | // --Interpolate among bins in a range // | |
24 | //--------------------------------------------------------------------// | |
25 | // | |
26 | // | |
27 | #include "AliCFGridSparse.h" | |
28 | #include "THnSparse.h" | |
29 | #include "AliLog.h" | |
30 | #include "TMath.h" | |
31 | #include "TROOT.h" | |
32 | #include "TH1D.h" | |
33 | #include "TH2D.h" | |
34 | #include "TH3D.h" | |
35 | #include "TAxis.h" | |
36 | ||
37 | //____________________________________________________________________ | |
38 | ClassImp(AliCFGridSparse) | |
39 | ||
40 | //____________________________________________________________________ | |
41 | AliCFGridSparse::AliCFGridSparse() : | |
fb494025 | 42 | AliCFFrame(), |
43 | fSumW2(kFALSE), | |
db6722a5 | 44 | fData(0x0) |
45 | { | |
46 | // default constructor | |
47 | } | |
48 | //____________________________________________________________________ | |
49 | AliCFGridSparse::AliCFGridSparse(const Char_t* name, const Char_t* title) : | |
fb494025 | 50 | AliCFFrame(name,title), |
51 | fSumW2(kFALSE), | |
db6722a5 | 52 | fData(0x0) |
53 | { | |
54 | // default constructor | |
55 | } | |
56 | //____________________________________________________________________ | |
fb494025 | 57 | AliCFGridSparse::AliCFGridSparse(const Char_t* name, const Char_t* title, Int_t nVarIn, const Int_t * nBinIn) : |
58 | AliCFFrame(name,title), | |
59 | fSumW2(kFALSE), | |
db6722a5 | 60 | fData(0x0) |
61 | { | |
62 | // | |
63 | // main constructor | |
64 | // | |
65 | ||
fb494025 | 66 | fData = new THnSparseF(name,title,nVarIn,nBinIn); |
db6722a5 | 67 | } |
fb494025 | 68 | |
db6722a5 | 69 | //____________________________________________________________________ |
fb494025 | 70 | AliCFGridSparse::~AliCFGridSparse() |
db6722a5 | 71 | { |
72 | // | |
fb494025 | 73 | // destructor |
db6722a5 | 74 | // |
fb494025 | 75 | if (fData) delete fData; |
db6722a5 | 76 | } |
77 | ||
78 | //____________________________________________________________________ | |
fb494025 | 79 | AliCFGridSparse::AliCFGridSparse(const AliCFGridSparse& c) : |
80 | AliCFFrame(c), | |
81 | fSumW2(kFALSE), | |
82 | fData(0x0) | |
db6722a5 | 83 | { |
84 | // | |
fb494025 | 85 | // copy constructor |
db6722a5 | 86 | // |
fb494025 | 87 | ((AliCFGridSparse &)c).Copy(*this); |
db6722a5 | 88 | } |
89 | ||
90 | //____________________________________________________________________ | |
fb494025 | 91 | AliCFGridSparse& AliCFGridSparse::operator=(const AliCFGridSparse &c) |
db6722a5 | 92 | { |
93 | // | |
94 | // assigment operator | |
95 | // | |
fb494025 | 96 | if (this != &c) c.Copy(*this); |
db6722a5 | 97 | return *this; |
98 | } | |
99 | ||
aa29aca9 | 100 | //____________________________________________________________________ |
101 | void AliCFGridSparse::SetBinLimits(Int_t ivar, Double_t min, Double_t max) | |
102 | { | |
103 | // | |
104 | // set a uniform binning for variable ivar | |
105 | // | |
106 | Int_t nBins = GetNBins(ivar); | |
107 | Double_t * array = new Double_t[nBins+1]; | |
108 | for (Int_t iEdge=0; iEdge<=nBins; iEdge++) array[iEdge] = min + iEdge * (max-min)/nBins ; | |
109 | fData->SetBinEdges(ivar, array); | |
110 | delete [] array ; | |
111 | } | |
112 | ||
db6722a5 | 113 | //____________________________________________________________________ |
fb494025 | 114 | void AliCFGridSparse::SetBinLimits(Int_t ivar, const Double_t *array) |
db6722a5 | 115 | { |
116 | // | |
117 | // setting the arrays containing the bin limits | |
118 | // | |
119 | fData->SetBinEdges(ivar, array); | |
db6722a5 | 120 | } |
121 | ||
122 | //____________________________________________________________________ | |
fb494025 | 123 | void AliCFGridSparse::Fill(const Double_t *var, Double_t weight) |
db6722a5 | 124 | { |
125 | // | |
126 | // Fill the grid, | |
127 | // given a set of values of the input variable, | |
128 | // with weight (by default w=1) | |
129 | // | |
130 | fData->Fill(var,weight); | |
131 | } | |
132 | ||
133 | //___________________________________________________________________ | |
134 | TH1D *AliCFGridSparse::Project(Int_t ivar) const | |
135 | { | |
136 | // | |
137 | // Make a 1D projection along variable ivar | |
138 | // | |
139 | ||
db6722a5 | 140 | TH1D *hist=fData->Projection(ivar); |
fb494025 | 141 | hist->SetXTitle(GetVarTitle(ivar)); |
91e32eeb | 142 | hist->SetName(Form("%s_proj-%s",GetName(),GetVarTitle(ivar))); |
fb494025 | 143 | hist->SetTitle(Form("%s: projection on %s",GetTitle(),GetVarTitle(ivar))); |
db6722a5 | 144 | return hist; |
db6722a5 | 145 | } |
146 | //___________________________________________________________________ | |
147 | TH2D *AliCFGridSparse::Project(Int_t ivar1, Int_t ivar2) const | |
148 | { | |
149 | // | |
150 | // Make a 2D projection along variables ivar1 & ivar2 | |
151 | // | |
152 | ||
318f64b1 | 153 | TH2D *hist=fData->Projection(ivar2,ivar1); //notice inverted axis (THnSparse uses TH3 2d-projection convention...) |
fb494025 | 154 | hist->SetXTitle(GetVarTitle(ivar1)); |
155 | hist->SetYTitle(GetVarTitle(ivar2)); | |
91e32eeb | 156 | hist->SetName(Form("%s_proj-%s,%s",GetName(),GetVarTitle(ivar1),GetVarTitle(ivar2))); |
fb494025 | 157 | hist->SetTitle(Form("%s: projection on %s-%s",GetTitle(),GetVarTitle(ivar1),GetVarTitle(ivar2))); |
db6722a5 | 158 | return hist; |
db6722a5 | 159 | } |
160 | //___________________________________________________________________ | |
161 | TH3D *AliCFGridSparse::Project(Int_t ivar1, Int_t ivar2, Int_t ivar3) const | |
162 | { | |
163 | // | |
164 | // Make a 3D projection along variables ivar1 & ivar2 & ivar3 | |
165 | // | |
db6722a5 | 166 | |
167 | TH3D *hist=fData->Projection(ivar1,ivar2,ivar3); | |
fb494025 | 168 | hist->SetXTitle(GetVarTitle(ivar1)); |
169 | hist->SetYTitle(GetVarTitle(ivar2)); | |
170 | hist->SetZTitle(GetVarTitle(ivar3)); | |
91e32eeb | 171 | hist->SetName(Form("%s_proj-%s,%s,%s",GetName(),GetVarTitle(ivar1),GetVarTitle(ivar2),GetVarTitle(ivar3))); |
fb494025 | 172 | hist->SetTitle(Form("%s: projection on %s-%s-%s",GetTitle(),GetVarTitle(ivar1),GetVarTitle(ivar2),GetVarTitle(ivar3))); |
db6722a5 | 173 | return hist; |
db6722a5 | 174 | } |
175 | ||
9105291d | 176 | //___________________________________________________________________ |
98a5f772 | 177 | AliCFGridSparse* AliCFGridSparse::Project(Int_t nVars, const Int_t* vars, const Double_t* varMin, const Double_t* varMax, Bool_t useBins) const |
9105291d | 178 | { |
38b1447f | 179 | // |
180 | // projects the grid on the nVars dimensions defined in vars. | |
181 | // axis ranges can be defined in arrays varMin, varMax | |
98a5f772 | 182 | // If useBins=true, varMin and varMax are taken as bin numbers |
38b1447f | 183 | // |
9105291d | 184 | |
185 | // binning for new grid | |
186 | Int_t* bins = new Int_t[nVars]; | |
187 | for (Int_t iVar=0; iVar<nVars; iVar++) { | |
fb494025 | 188 | bins[iVar] = GetNBins(vars[iVar]); |
9105291d | 189 | } |
190 | ||
191 | // create new grid sparse | |
192 | AliCFGridSparse* out = new AliCFGridSparse(fName,fTitle,nVars,bins); | |
193 | ||
194 | //set the range in the THnSparse to project | |
195 | THnSparse* clone = ((THnSparse*)fData->Clone()); | |
38b1447f | 196 | if (varMin && varMax) { |
fb494025 | 197 | for (Int_t iAxis=0; iAxis<GetNVar(); iAxis++) { |
98a5f772 | 198 | if (useBins) clone->GetAxis(iAxis)->SetRange((Int_t)varMin[iAxis],(Int_t)varMax[iAxis]); |
199 | else clone->GetAxis(iAxis)->SetRangeUser(varMin[iAxis],varMax[iAxis]); | |
38b1447f | 200 | } |
9105291d | 201 | } |
38b1447f | 202 | else AliInfo("Keeping same axis ranges"); |
203 | ||
9105291d | 204 | out->SetGrid(clone->Projection(nVars,vars)); |
205 | return out; | |
206 | } | |
207 | ||
fb494025 | 208 | |
db6722a5 | 209 | //____________________________________________________________________ |
fb494025 | 210 | Float_t AliCFGridSparse::GetBinCenter(Int_t ivar, Int_t ibin) const |
db6722a5 | 211 | { |
212 | // | |
fb494025 | 213 | // Returns the center of specified bin for variable axis ivar |
214 | // | |
215 | ||
216 | return (Float_t) fData->GetAxis(ivar)->GetBinCenter(ibin); | |
db6722a5 | 217 | } |
218 | ||
219 | //____________________________________________________________________ | |
fb494025 | 220 | Float_t AliCFGridSparse::GetBinSize(Int_t ivar, Int_t ibin) const |
db6722a5 | 221 | { |
222 | // | |
fb494025 | 223 | // Returns the size of specified bin for variable axis ivar |
224 | // | |
225 | ||
226 | return (Float_t) fData->GetAxis(ivar)->GetBinUpEdge(ibin) - fData->GetAxis(ivar)->GetBinLowEdge(ibin); | |
db6722a5 | 227 | } |
228 | ||
db6722a5 | 229 | //____________________________________________________________________ |
230 | Float_t AliCFGridSparse::GetEntries() const | |
231 | { | |
232 | // | |
233 | // total entries (including overflows and underflows) | |
234 | // | |
235 | ||
236 | return fData->GetEntries(); | |
237 | } | |
238 | ||
239 | //____________________________________________________________________ | |
fb494025 | 240 | Float_t AliCFGridSparse::GetElement(Long_t index) const |
db6722a5 | 241 | { |
242 | // | |
fb494025 | 243 | // Returns content of grid element index |
db6722a5 | 244 | // |
db6722a5 | 245 | |
fb494025 | 246 | return fData->GetBinContent(index); |
db6722a5 | 247 | } |
248 | //____________________________________________________________________ | |
fb494025 | 249 | Float_t AliCFGridSparse::GetElement(const Int_t *bin) const |
db6722a5 | 250 | { |
251 | // | |
252 | // Get the content in a bin corresponding to a set of bin indexes | |
253 | // | |
254 | return fData->GetBinContent(bin); | |
255 | ||
256 | } | |
257 | //____________________________________________________________________ | |
fb494025 | 258 | Float_t AliCFGridSparse::GetElement(const Double_t *var) const |
db6722a5 | 259 | { |
260 | // | |
261 | // Get the content in a bin corresponding to a set of input variables | |
262 | // | |
263 | ||
fb494025 | 264 | Long_t index = fData->GetBin(var,kFALSE); |
265 | if (index<0) return 0.; | |
266 | return fData->GetBinContent(index); | |
db6722a5 | 267 | } |
268 | ||
269 | //____________________________________________________________________ | |
fb494025 | 270 | Float_t AliCFGridSparse::GetElementError(Long_t index) const |
db6722a5 | 271 | { |
272 | // | |
fb494025 | 273 | // Returns the error on the content |
db6722a5 | 274 | // |
275 | ||
fb494025 | 276 | return fData->GetBinContent(index); |
db6722a5 | 277 | } |
278 | //____________________________________________________________________ | |
fb494025 | 279 | Float_t AliCFGridSparse::GetElementError(const Int_t *bin) const |
db6722a5 | 280 | { |
281 | // | |
282 | // Get the error in a bin corresponding to a set of bin indexes | |
283 | // | |
284 | return fData->GetBinError(bin); | |
285 | ||
286 | } | |
287 | //____________________________________________________________________ | |
fb494025 | 288 | Float_t AliCFGridSparse::GetElementError(const Double_t *var) const |
db6722a5 | 289 | { |
290 | // | |
291 | // Get the error in a bin corresponding to a set of input variables | |
292 | // | |
293 | ||
294 | Long_t index=fData->GetBin(var,kFALSE); //this is the THnSparse index (do not allocate new cells if content is empy) | |
fb494025 | 295 | if (index<0) return 0.; |
296 | return fData->GetBinError(index); | |
db6722a5 | 297 | } |
298 | ||
db6722a5 | 299 | //____________________________________________________________________ |
fb494025 | 300 | void AliCFGridSparse::SetElement(Long_t index, Float_t val) |
db6722a5 | 301 | { |
302 | // | |
fb494025 | 303 | // Sets grid element value |
db6722a5 | 304 | // |
fb494025 | 305 | Int_t* bin = new Int_t[GetNVar()]; |
306 | fData->GetBinContent(index,bin); //affects the bin coordinates | |
307 | SetElement(bin,val); | |
308 | delete [] bin ; | |
db6722a5 | 309 | } |
fb494025 | 310 | |
db6722a5 | 311 | //____________________________________________________________________ |
fb494025 | 312 | void AliCFGridSparse::SetElement(const Int_t *bin, Float_t val) |
db6722a5 | 313 | { |
314 | // | |
315 | // Sets grid element of bin indeces bin to val | |
316 | // | |
317 | fData->SetBinContent(bin,val); | |
318 | } | |
319 | //____________________________________________________________________ | |
fb494025 | 320 | void AliCFGridSparse::SetElement(const Double_t *var, Float_t val) |
db6722a5 | 321 | { |
322 | // | |
323 | // Set the content in a bin to value val corresponding to a set of input variables | |
324 | // | |
fb494025 | 325 | Long_t index=fData->GetBin(var,kTRUE); //THnSparse index: allocate the cell |
326 | Int_t *bin = new Int_t[GetNVar()]; | |
db6722a5 | 327 | fData->GetBinContent(index,bin); //trick to access the array of bins |
fb494025 | 328 | SetElement(bin,val); |
db6722a5 | 329 | delete [] bin; |
db6722a5 | 330 | } |
331 | ||
332 | //____________________________________________________________________ | |
fb494025 | 333 | void AliCFGridSparse::SetElementError(Long_t index, Float_t val) |
db6722a5 | 334 | { |
335 | // | |
336 | // Sets grid element iel error to val (linear indexing) in AliCFFrame | |
337 | // | |
fb494025 | 338 | Int_t *bin = new Int_t[GetNVar()]; |
339 | fData->GetBinContent(index,bin); | |
340 | SetElementError(bin,val); | |
db6722a5 | 341 | delete [] bin; |
342 | } | |
fb494025 | 343 | |
db6722a5 | 344 | //____________________________________________________________________ |
fb494025 | 345 | void AliCFGridSparse::SetElementError(const Int_t *bin, Float_t val) |
db6722a5 | 346 | { |
347 | // | |
348 | // Sets grid element error of bin indeces bin to val | |
349 | // | |
350 | fData->SetBinError(bin,val); | |
351 | } | |
352 | //____________________________________________________________________ | |
fb494025 | 353 | void AliCFGridSparse::SetElementError(const Double_t *var, Float_t val) |
db6722a5 | 354 | { |
355 | // | |
356 | // Set the error in a bin to value val corresponding to a set of input variables | |
357 | // | |
358 | Long_t index=fData->GetBin(var); //THnSparse index | |
fb494025 | 359 | Int_t *bin = new Int_t[GetNVar()]; |
db6722a5 | 360 | fData->GetBinContent(index,bin); //trick to access the array of bins |
fb494025 | 361 | SetElementError(bin,val); |
db6722a5 | 362 | delete [] bin; |
363 | } | |
364 | ||
365 | //____________________________________________________________________ | |
366 | void AliCFGridSparse::SumW2() | |
367 | { | |
368 | // | |
369 | //set calculation of the squared sum of the weighted entries | |
370 | // | |
371 | if(!fSumW2){ | |
372 | fData->CalculateErrors(kTRUE); | |
373 | } | |
db6722a5 | 374 | fSumW2=kTRUE; |
375 | } | |
376 | ||
377 | //____________________________________________________________________ | |
fb494025 | 378 | void AliCFGridSparse::Add(const AliCFGridSparse* aGrid, Double_t c) |
db6722a5 | 379 | { |
380 | // | |
381 | //add aGrid to the current one | |
382 | // | |
383 | ||
fb494025 | 384 | if (aGrid->GetNVar() != GetNVar()){ |
385 | AliError("Different number of variables, cannot add the grids"); | |
db6722a5 | 386 | return; |
387 | } | |
db6722a5 | 388 | |
fb494025 | 389 | if (!fSumW2 && aGrid->GetSumW2()) SumW2(); |
390 | fData->Add(aGrid->GetGrid(),c); | |
db6722a5 | 391 | } |
392 | ||
393 | //____________________________________________________________________ | |
fb494025 | 394 | void AliCFGridSparse::Add(const AliCFGridSparse* aGrid1, const AliCFGridSparse* aGrid2, Double_t c1,Double_t c2) |
db6722a5 | 395 | { |
396 | // | |
397 | //Add aGrid1 and aGrid2 and deposit the result into the current one | |
398 | // | |
399 | ||
fb494025 | 400 | if (GetNVar() != aGrid1->GetNVar() || GetNVar() != aGrid2->GetNVar()) { |
db6722a5 | 401 | AliInfo("Different number of variables, cannot add the grids"); |
402 | return; | |
403 | } | |
db6722a5 | 404 | |
fb494025 | 405 | if (!fSumW2 && (aGrid1->GetSumW2() || aGrid2->GetSumW2())) SumW2(); |
db6722a5 | 406 | |
407 | fData->Reset(); | |
fb494025 | 408 | fData->Add(aGrid1->GetGrid(),c1); |
409 | fData->Add(aGrid2->GetGrid(),c2); | |
db6722a5 | 410 | } |
411 | ||
412 | //____________________________________________________________________ | |
fb494025 | 413 | void AliCFGridSparse::Multiply(const AliCFGridSparse* aGrid, Double_t c) |
db6722a5 | 414 | { |
415 | // | |
416 | // Multiply aGrid to the current one | |
417 | // | |
418 | ||
fb494025 | 419 | if (aGrid->GetNVar() != GetNVar()) { |
420 | AliError("Different number of variables, cannot multiply the grids"); | |
db6722a5 | 421 | return; |
422 | } | |
db6722a5 | 423 | |
fb494025 | 424 | if(!fSumW2 && aGrid->GetSumW2()) SumW2(); |
425 | THnSparse *h = aGrid->GetGrid(); | |
db6722a5 | 426 | fData->Multiply(h); |
427 | fData->Scale(c); | |
db6722a5 | 428 | } |
429 | ||
430 | //____________________________________________________________________ | |
fb494025 | 431 | void AliCFGridSparse::Multiply(const AliCFGridSparse* aGrid1, const AliCFGridSparse* aGrid2, Double_t c1,Double_t c2) |
db6722a5 | 432 | { |
433 | // | |
434 | //Multiply aGrid1 and aGrid2 and deposit the result into the current one | |
435 | // | |
436 | ||
fb494025 | 437 | if (GetNVar() != aGrid1->GetNVar() || GetNVar() != aGrid2->GetNVar()) { |
438 | AliError("Different number of variables, cannot multiply the grids"); | |
db6722a5 | 439 | return; |
fb494025 | 440 | } |
db6722a5 | 441 | |
fb494025 | 442 | if(!fSumW2 && (aGrid1->GetSumW2() || aGrid2->GetSumW2())) SumW2(); |
db6722a5 | 443 | |
444 | fData->Reset(); | |
fb494025 | 445 | THnSparse *h1 = aGrid1->GetGrid(); |
446 | THnSparse *h2 = aGrid2->GetGrid(); | |
db6722a5 | 447 | h2->Multiply(h1); |
448 | h2->Scale(c1*c2); | |
449 | fData->Add(h2); | |
450 | } | |
451 | ||
db6722a5 | 452 | //____________________________________________________________________ |
fb494025 | 453 | void AliCFGridSparse::Divide(const AliCFGridSparse* aGrid, Double_t c) |
db6722a5 | 454 | { |
455 | // | |
456 | // Divide aGrid to the current one | |
457 | // | |
458 | ||
fb494025 | 459 | if (aGrid->GetNVar() != GetNVar()) { |
460 | AliError("Different number of variables, cannot divide the grids"); | |
db6722a5 | 461 | return; |
462 | } | |
db6722a5 | 463 | |
fb494025 | 464 | if (!fSumW2 && aGrid->GetSumW2()) SumW2(); |
db6722a5 | 465 | |
d7c1cb54 | 466 | THnSparse *h1 = aGrid->GetGrid(); |
467 | THnSparse *h2 = (THnSparse*)fData->Clone(); | |
468 | fData->Divide(h2,h1); | |
db6722a5 | 469 | fData->Scale(c); |
db6722a5 | 470 | } |
471 | ||
472 | //____________________________________________________________________ | |
fb494025 | 473 | void AliCFGridSparse::Divide(const AliCFGridSparse* aGrid1, const AliCFGridSparse* aGrid2, Double_t c1,Double_t c2, Option_t *option) |
db6722a5 | 474 | { |
475 | // | |
476 | //Divide aGrid1 and aGrid2 and deposit the result into the current one | |
fb494025 | 477 | //binomial errors are supported |
db6722a5 | 478 | // |
479 | ||
fb494025 | 480 | if (GetNVar() != aGrid1->GetNVar() || GetNVar() != aGrid2->GetNVar()) { |
481 | AliError("Different number of variables, cannot divide the grids"); | |
db6722a5 | 482 | return; |
483 | } | |
db6722a5 | 484 | |
fb494025 | 485 | if (!fSumW2 && (aGrid1->GetSumW2() || aGrid2->GetSumW2())) SumW2(); |
db6722a5 | 486 | |
fb494025 | 487 | THnSparse *h1= aGrid1->GetGrid(); |
488 | THnSparse *h2= aGrid2->GetGrid(); | |
db6722a5 | 489 | fData->Divide(h1,h2,c1,c2,option); |
490 | } | |
491 | ||
492 | ||
7411edfd | 493 | //____________________________________________________________________ |
494 | void AliCFGridSparse::Rebin(const Int_t* group) | |
495 | { | |
496 | // | |
497 | // rebin the grid according to Rebin() as in THnSparse | |
498 | // Please notice that the original number of bins on | |
499 | // a given axis has to be divisible by the rebin group. | |
500 | // | |
501 | ||
fb494025 | 502 | for(Int_t i=0;i<GetNVar();i++){ |
503 | if (group[i]!=1) AliInfo(Form(" merging bins along dimension %i in groups of %i bins", i,group[i])); | |
7411edfd | 504 | } |
505 | ||
506 | THnSparse *rebinned =fData->Rebin(group); | |
507 | fData->Reset(); | |
508 | fData = rebinned; | |
fb494025 | 509 | } |
510 | //____________________________________________________________________ | |
511 | void AliCFGridSparse::Scale(Long_t index, const Double_t *fact) | |
512 | { | |
513 | // | |
514 | //scale content of a certain cell by (positive) fact (with error) | |
515 | // | |
7411edfd | 516 | |
fb494025 | 517 | if (GetElement(index)==0 || fact[0]==0) return; |
7411edfd | 518 | |
fb494025 | 519 | Double_t in[2], out[2]; |
520 | in[0]=GetElement(index); | |
521 | in[1]=GetElementError(index); | |
522 | GetScaledValues(fact,in,out); | |
523 | SetElement(index,out[0]); | |
524 | if (fSumW2) SetElementError(index,out[1]); | |
525 | } | |
526 | //____________________________________________________________________ | |
527 | void AliCFGridSparse::Scale(const Int_t *bin, const Double_t *fact) | |
528 | { | |
529 | // | |
530 | //scale content of a certain cell by (positive) fact (with error) | |
531 | // | |
532 | if(GetElement(bin)==0 || fact[0]==0)return; | |
7411edfd | 533 | |
fb494025 | 534 | Double_t in[2], out[2]; |
535 | in[0]=GetElement(bin); | |
536 | in[1]=GetElementError(bin); | |
537 | GetScaledValues(fact,in,out); | |
538 | SetElement(bin,out[0]); | |
539 | if(fSumW2)SetElementError(bin,out[1]); | |
540 | ||
541 | } | |
542 | //____________________________________________________________________ | |
543 | void AliCFGridSparse::Scale(const Double_t *var, const Double_t *fact) | |
544 | { | |
545 | // | |
546 | //scale content of a certain cell by (positive) fact (with error) | |
547 | // | |
548 | if(GetElement(var)==0 || fact[0]==0)return; | |
7411edfd | 549 | |
fb494025 | 550 | Double_t in[2], out[2]; |
551 | in[0]=GetElement(var); | |
552 | in[1]=GetElementError(var); | |
553 | GetScaledValues(fact,in,out); | |
554 | SetElement(var,out[0]); | |
555 | if(fSumW2)SetElementError(var,out[1]); | |
556 | ||
557 | } | |
558 | //____________________________________________________________________ | |
559 | void AliCFGridSparse::Scale(const Double_t* fact) | |
560 | { | |
561 | // | |
562 | //scale contents of the whole grid by fact | |
563 | // | |
564 | ||
565 | for (Long_t iel=0; iel<GetNFilledBins(); iel++) { | |
566 | Scale(iel,fact); | |
7411edfd | 567 | } |
fb494025 | 568 | } |
569 | //____________________________________________________________________ | |
570 | Long_t AliCFGridSparse::GetEmptyBins() const { | |
571 | // | |
572 | // Get empty bins | |
573 | // | |
7411edfd | 574 | |
fb494025 | 575 | return (GetNBinsTotal() - GetNFilledBins()) ; |
576 | } | |
7411edfd | 577 | |
fb494025 | 578 | //_____________________________________________________________________ |
579 | // Int_t AliCFGridSparse::GetEmptyBins( Double_t *varMin, Double_t* varMax ) const | |
580 | // { | |
581 | // // | |
582 | // // Get empty bins in a range specified by varMin and varMax | |
583 | // // | |
584 | ||
585 | // Int_t *indexMin = new Int_t[GetNVar()]; | |
586 | // Int_t *indexMax = new Int_t[GetNVar()]; | |
587 | ||
588 | // //Find out the min and max bins | |
589 | ||
590 | // for (Int_t i=0;i<GetNVar();i++) { | |
591 | // Double_t xmin=varMin[i]; // the min values | |
592 | // Double_t xmax=varMax[i]; // the min values | |
593 | // Int_t nbins = fData->GetAxis(i)->GetNbins()+1; | |
594 | // Double_t *bins=new Double_t[nbins]; | |
595 | // for(Int_t ibin =0;ibin<nbins;ibin++){ | |
596 | // bins[ibin] = fVarBinLimits[ibin+fOffset[i]]; | |
597 | // } | |
598 | // indexMin[i] = TMath::BinarySearch(nbins,bins,xmin); | |
599 | // indexMax[i] = TMath::BinarySearch(nbins,bins,xmax); | |
600 | // delete [] bins; | |
601 | // } | |
602 | ||
603 | // Int_t val=0; | |
604 | // for(Int_t i=0;i<fNDim;i++){ | |
605 | // for (Int_t j=0;j<GetNVar();j++)fIndex[j]=GetBinIndex(j,i); | |
606 | // Bool_t isIn=kTRUE; | |
607 | // for (Int_t j=0;j<GetNVar();j++){ | |
608 | // if(!(fIndex[j]>=indexMin[j] && fIndex[j]<=indexMax[j]))isIn=kFALSE; | |
609 | // } | |
610 | // if(isIn && GetElement(i)<=0)val++; | |
611 | // } | |
612 | // AliInfo(Form(" the empty bins = %i ",val)); | |
613 | ||
614 | // delete [] indexMin; | |
615 | // delete [] indexMax; | |
616 | // return val; | |
617 | // } | |
618 | ||
619 | //____________________________________________________________________ | |
620 | Int_t AliCFGridSparse::CheckStats(Double_t thr) const | |
621 | { | |
622 | // | |
623 | // Count the cells below a certain threshold | |
624 | // | |
625 | Int_t ncellsLow=0; | |
626 | for (Int_t i=0; i<GetNBinsTotal(); i++) { | |
627 | if (GetElement(i)<thr) ncellsLow++; | |
628 | } | |
629 | return ncellsLow; | |
630 | } | |
7411edfd | 631 | |
fb494025 | 632 | //_____________________________________________________________________ |
633 | Double_t AliCFGridSparse::GetIntegral() const | |
634 | { | |
635 | // | |
636 | // Get full Integral | |
637 | // | |
638 | return fData->ComputeIntegral(); | |
639 | } | |
7411edfd | 640 | |
fb494025 | 641 | //_____________________________________________________________________ |
642 | // Double_t AliCFGridSparse::GetIntegral(Int_t *binMin, Int_t* binMax ) const | |
643 | // { | |
644 | // // | |
645 | // // Get Integral in a range of bin indeces (extremes included) | |
646 | // // | |
647 | ||
648 | // Double_t val=0; | |
649 | ||
650 | // for(Int_t i=0;i<GetNVar();i++){ | |
651 | // if(binMin[i]<1)binMin[i]=1; | |
652 | // if(binMax[i]>fNVarBins[i])binMax[i]=fNVarBins[i]; | |
653 | // if((binMin[i]>binMax[i])){ | |
654 | // AliInfo(Form(" Bin indices in variable %i in reverse order, please check!", i)); | |
655 | // return val; | |
656 | // } | |
657 | // } | |
658 | // val=GetSum(0,binMin,binMax); | |
659 | ||
660 | // return val; | |
661 | // } | |
662 | ||
663 | //_____________________________________________________________________ | |
664 | // Double_t AliCFGridSparse::GetIntegral(Double_t *varMin, Double_t* varMax ) const | |
665 | // { | |
666 | // // | |
667 | // // Get Integral in a range (extremes included) | |
668 | // // | |
669 | ||
670 | // Int_t *indexMin=new Int_t[GetNVar()]; | |
671 | // Int_t *indexMax=new Int_t[GetNVar()]; | |
672 | ||
673 | // //Find out the min and max bins | |
674 | ||
675 | // for(Int_t i=0;i<GetNVar();i++){ | |
676 | // Double_t xmin=varMin[i]; // the min values | |
677 | // Double_t xmax=varMax[i]; // the min values | |
678 | // Int_t nbins=fNVarBins[i]+1; | |
679 | // Double_t *bins=new Double_t[nbins]; | |
680 | // for(Int_t ibin =0;ibin<nbins;ibin++){ | |
681 | // bins[ibin] = fVarBinLimits[ibin+fOffset[i]]; | |
682 | // } | |
683 | // indexMin[i] = TMath::BinarySearch(nbins,bins,xmin); | |
684 | // indexMax[i] = TMath::BinarySearch(nbins,bins,xmax); | |
685 | // delete [] bins; | |
686 | // } | |
687 | ||
688 | // //move to the TH/THnSparse convention in N-dim bin numbering | |
689 | // for(Int_t i=0;i<GetNVar();i++){ | |
690 | // indexMin[i]+=1; | |
691 | // indexMax[i]+=1; | |
692 | // } | |
693 | ||
694 | // Double_t val=GetIntegral(indexMin,indexMax); | |
695 | ||
696 | // delete [] indexMin; | |
697 | // delete [] indexMax; | |
698 | ||
699 | // return val; | |
700 | // } | |
701 | ||
702 | // //_____________________________________________________________________ | |
703 | // Double_t AliCFGridSparse::GetSum(Int_t ivar, Int_t *binMin, Int_t* binMax) const | |
704 | // { | |
705 | // // | |
706 | // // recursively add over nested loops.... | |
707 | // // | |
708 | ||
709 | // static Double_t val; | |
710 | // if (ivar==0) val=0.; | |
711 | // for(Int_t ibin=binMin[ivar]-1 ; ibin<=binMax[ivar]-1 ; ibin++){ | |
712 | // //-1 is to move from TH/ThnSparse N-dim bin convention to one in AliCFFrame | |
713 | // fIndex[ivar]=ibin; | |
714 | // if(ivar<GetNVar()-1) { | |
715 | // val=GetSum(ivar+1,binMin,binMax); | |
716 | // } | |
717 | // else { | |
718 | // Int_t iel=GetBinIndex(fIndex); | |
719 | // val+=GetElement(iel); | |
720 | // } | |
721 | // } | |
722 | ||
723 | // return val; | |
724 | // } | |
725 | ||
726 | //____________________________________________________________________ | |
727 | Long64_t AliCFGridSparse::Merge(TCollection* list) | |
728 | { | |
729 | // | |
730 | // Merge a list of AliCFGridSparse with this (needed for PROOF). | |
731 | // Returns the number of merged objects (including this). | |
732 | // | |
733 | ||
734 | if (!list) | |
735 | return 0; | |
7411edfd | 736 | |
fb494025 | 737 | if (list->IsEmpty()) |
738 | return 1; | |
739 | ||
740 | TIterator* iter = list->MakeIterator(); | |
741 | TObject* obj; | |
742 | ||
743 | Int_t count = 0; | |
744 | while ((obj = iter->Next())) { | |
745 | AliCFGridSparse* entry = dynamic_cast<AliCFGridSparse*> (obj); | |
746 | if (entry == 0) | |
747 | continue; | |
748 | this->Add(entry); | |
749 | count++; | |
750 | } | |
751 | ||
752 | return count+1; | |
7411edfd | 753 | } |
fb494025 | 754 | |
755 | //____________________________________________________________________ | |
756 | void AliCFGridSparse::GetScaledValues(const Double_t *fact, const Double_t *in, Double_t *out) const{ | |
757 | // | |
758 | // scale input *in and its error by (positive) fact (with error) | |
759 | // and erite it to *out | |
760 | // | |
761 | out[0]=in[0]*fact[0]; | |
762 | out[1]=TMath::Sqrt(in[1]*in[1]/in[0]/in[0] | |
763 | +fact[1]*fact[1]/fact[0]/fact[0])*out[0]; | |
764 | ||
765 | } | |
766 | ||
db6722a5 | 767 | //____________________________________________________________________ |
768 | void AliCFGridSparse::Copy(TObject& c) const | |
769 | { | |
770 | // | |
771 | // copy function | |
772 | // | |
fb494025 | 773 | AliCFFrame::Copy(c); |
db6722a5 | 774 | AliCFGridSparse& target = (AliCFGridSparse &) c; |
fb494025 | 775 | target.fSumW2 = fSumW2 ; |
776 | if (fData) { | |
777 | target.fData = (THnSparse*)fData->Clone(); | |
778 | } | |
db6722a5 | 779 | } |
780 | ||
c8df672e | 781 | //____________________________________________________________________ |
98a5f772 | 782 | TH1D* AliCFGridSparse::Slice(Int_t iVar, const Double_t *varMin, const Double_t *varMax, Bool_t useBins) const |
c8df672e | 783 | { |
784 | // | |
785 | // return a slice (1D-projection) on variable iVar while axis ranges are defined with varMin,varMax | |
786 | // arrays varMin and varMax contain the min and max values of each variable. | |
fb494025 | 787 | // therefore varMin and varMax must have their dimensions equal to GetNVar() |
98a5f772 | 788 | // If useBins=true, varMin and varMax are taken as bin numbers |
c8df672e | 789 | |
790 | THnSparse* clone = (THnSparse*)fData->Clone(); | |
fb494025 | 791 | for (Int_t iAxis=0; iAxis<GetNVar(); iAxis++) { |
98a5f772 | 792 | if (useBins) clone->GetAxis(iAxis)->SetRange((Int_t)varMin[iAxis],(Int_t)varMax[iAxis]); |
793 | else clone->GetAxis(iAxis)->SetRangeUser(varMin[iAxis],varMax[iAxis]); | |
c8df672e | 794 | } |
795 | return clone->Projection(iVar); | |
796 | } | |
797 | ||
798 | //____________________________________________________________________ | |
98a5f772 | 799 | TH2D* AliCFGridSparse::Slice(Int_t iVar1, Int_t iVar2, const Double_t *varMin, const Double_t *varMax, Bool_t useBins) const |
c8df672e | 800 | { |
801 | // | |
802 | // return a slice (2D-projection) on variables iVar1 and iVar2 while axis ranges are defined with varMin,varMax | |
803 | // arrays varMin and varMax contain the min and max values of each variable. | |
fb494025 | 804 | // therefore varMin and varMax must have their dimensions equal to GetNVar() |
98a5f772 | 805 | // If useBins=true, varMin and varMax are taken as bin numbers |
c8df672e | 806 | |
807 | THnSparse* clone = (THnSparse*)fData->Clone(); | |
fb494025 | 808 | for (Int_t iAxis=0; iAxis<GetNVar(); iAxis++) { |
98a5f772 | 809 | if (useBins) clone->GetAxis(iAxis)->SetRange((Int_t)varMin[iAxis],(Int_t)varMax[iAxis]); |
810 | else clone->GetAxis(iAxis)->SetRangeUser(varMin[iAxis],varMax[iAxis]); | |
c8df672e | 811 | } |
812 | return clone->Projection(iVar1,iVar2); | |
813 | } | |
814 | ||
815 | //____________________________________________________________________ | |
98a5f772 | 816 | TH3D* AliCFGridSparse::Slice(Int_t iVar1, Int_t iVar2, Int_t iVar3, const Double_t *varMin, const Double_t *varMax, Bool_t useBins) const |
c8df672e | 817 | { |
818 | // | |
819 | // return a slice (3D-projection) on variables iVar1, iVar2 and iVar3 while axis ranges are defined with varMin,varMax | |
820 | // arrays varMin and varMax contain the min and max values of each variable. | |
fb494025 | 821 | // therefore varMin and varMax must have their dimensions equal to GetNVar() |
98a5f772 | 822 | // If useBins=true, varMin and varMax are taken as bin numbers |
c8df672e | 823 | |
824 | THnSparse* clone = (THnSparse*)fData->Clone(); | |
fb494025 | 825 | for (Int_t iAxis=0; iAxis<GetNVar(); iAxis++) { |
98a5f772 | 826 | if (useBins) clone->GetAxis(iAxis)->SetRange((Int_t)varMin[iAxis],(Int_t)varMax[iAxis]); |
827 | else clone->GetAxis(iAxis)->SetRangeUser(varMin[iAxis],varMax[iAxis]); | |
c8df672e | 828 | } |
829 | return clone->Projection(iVar1,iVar2,iVar3); | |
830 | } | |
831 | ||
9105291d | 832 | //____________________________________________________________________ |
833 | void AliCFGridSparse::SetRangeUser(Int_t iVar, Double_t varMin, Double_t varMax) { | |
834 | // | |
835 | // set range of axis iVar. | |
836 | // | |
837 | fData->GetAxis(iVar)->SetRangeUser(varMin,varMax); | |
838 | AliWarning(Form("THnSparse axis %d range has been modified",iVar)); | |
839 | } | |
840 | ||
c8df672e | 841 | //____________________________________________________________________ |
fb494025 | 842 | void AliCFGridSparse::SetRangeUser(const Double_t *varMin, const Double_t *varMax) { |
c8df672e | 843 | // |
fb494025 | 844 | // set range of every axis. varMin and varMax must be of dimension GetNVar() |
c8df672e | 845 | // |
fb494025 | 846 | for (Int_t iAxis=0; iAxis<GetNVar() ; iAxis++) { // set new range for every axis |
9105291d | 847 | SetRangeUser(iAxis,varMin[iAxis],varMax[iAxis]); |
c8df672e | 848 | } |
9105291d | 849 | AliWarning("THnSparse axes ranges have been modified"); |
850 | } | |
851 | ||
852 | //____________________________________________________________________ | |
853 | void AliCFGridSparse::UseAxisRange(Bool_t b) const { | |
fb494025 | 854 | for (Int_t iAxis=0; iAxis<GetNVar(); iAxis++) fData->GetAxis(iAxis)->SetBit(TAxis::kAxisRange,b); |
855 | } | |
856 | ||
857 | //____________________________________________________________________ | |
858 | Float_t AliCFGridSparse::GetOverFlows(Int_t ivar, Bool_t exclusive) const | |
859 | { | |
860 | // | |
861 | // Returns overflows in variable ivar | |
862 | // Set 'exclusive' to true for an exclusive check on variable ivar | |
863 | // | |
864 | Int_t* bin = new Int_t[GetNVar()]; | |
865 | memset(bin, 0, sizeof(Int_t) * GetNVar()); | |
866 | Float_t ovfl=0.; | |
867 | for (Long64_t i = 0; i < fData->GetNbins(); i++) { | |
868 | Double_t v = fData->GetBinContent(i, bin); | |
869 | Bool_t add=kTRUE; | |
870 | if (exclusive) { | |
871 | for(Int_t j=0;j<GetNVar();j++){ | |
872 | if(ivar==j)continue; | |
873 | if((bin[j]==0) || (bin[j]==GetNBins(j)+1))add=kFALSE; | |
874 | } | |
875 | } | |
876 | if(bin[ivar]==GetNBins(ivar)+1 && add) ovfl+=v; | |
877 | } | |
878 | ||
879 | delete[] bin; | |
880 | return ovfl; | |
881 | } | |
882 | ||
883 | //____________________________________________________________________ | |
884 | Float_t AliCFGridSparse::GetUnderFlows(Int_t ivar, Bool_t exclusive) const | |
885 | { | |
886 | // | |
887 | // Returns exclusive overflows in variable ivar | |
888 | // Set 'exclusive' to true for an exclusive check on variable ivar | |
889 | // | |
890 | Int_t* bin = new Int_t[GetNVar()]; | |
891 | memset(bin, 0, sizeof(Int_t) * GetNVar()); | |
892 | Float_t unfl=0.; | |
893 | for (Long64_t i = 0; i < fData->GetNbins(); i++) { | |
894 | Double_t v = fData->GetBinContent(i, bin); | |
895 | Bool_t add=kTRUE; | |
896 | if (exclusive) { | |
897 | for(Int_t j=0;j<GetNVar();j++){ | |
898 | if(ivar==j)continue; | |
899 | if((bin[j]==0) || (bin[j]==GetNBins(j)+1))add=kFALSE; | |
900 | } | |
901 | } | |
902 | if(bin[ivar]==0 && add) unfl+=v; | |
903 | } | |
904 | ||
905 | delete[] bin; | |
906 | return unfl; | |
c8df672e | 907 | } |
fb494025 | 908 |