git://git.uio.no / u / mrichter / AliRoot.git / blame
| Commit | Line | Data |
|---|---|---|
| 7984e5f7 | 1 | // This class calculates the inclusive charged particle density |
| 2 | // in each for the 5 FMD rings. | |
| 3 | // | |
| 7e4038b5 | 4 | #include "AliFMDDensityCalculator.h" |
| 5 | #include <AliESDFMD.h> | |
| 6 | #include <TAxis.h> | |
| 7 | #include <TList.h> | |
| 8 | #include <TMath.h> | |
| 0bd4b00f | 9 | #include "AliForwardCorrectionManager.h" |
| fb3430ac | 10 | #include "AliFMDCorrDoubleHit.h" |
| 11 | #include "AliFMDCorrELossFit.h" | |
| 7e4038b5 | 12 | #include "AliLog.h" |
| 77f97e3f | 13 | #include "AliForwardUtil.h" |
| 7e4038b5 | 14 | #include <TH2D.h> |
| 0bd4b00f | 15 | #include <TProfile.h> |
| 5bb5d1f6 | 16 | #include <THStack.h> |
| 0bd4b00f | 17 | #include <TROOT.h> |
| 5ca83fee | 18 | #include <TVector3.h> |
| 1f7aa5c7 | 19 | #include <TStopwatch.h> |
| bfab35d9 | 20 | #include <TParameter.h> |
| 0bd4b00f | 21 | #include <iostream> |
| 22 | #include <iomanip> | |
| 1f7aa5c7 | 23 | #include <cstring> |
| 7e4038b5 | 24 | |
| 25 | ClassImp(AliFMDDensityCalculator) | |
| 26 | #if 0 | |
| 27 | ; // For Emacs | |
| 28 | #endif | |
| 29 | ||
| 5ca83fee | 30 | //____________________________________________________________________ |
| 31 | const char* AliFMDDensityCalculator::fgkFolderName = "fmdDensityCalculator"; | |
| 32 | ||
| 7e4038b5 | 33 | //____________________________________________________________________ |
| 34 | AliFMDDensityCalculator::AliFMDDensityCalculator() | |
| 35 | : TNamed(), | |
| 36 | fRingHistos(), | |
| dd497217 | 37 | fSumOfWeights(0), |
| 38 | fWeightedSum(0), | |
| ea3e5d95 | 39 | fCorrections(0), |
| 0bd4b00f | 40 | fMaxParticles(5), |
| 9d05ffeb | 41 | fUsePoisson(false), |
| 8411f7fe | 42 | fUsePhiAcceptance(kPhiCorrectNch), |
| 0bd4b00f | 43 | fAccI(0), |
| 44 | fAccO(0), | |
| 1174780f | 45 | fFMD1iMax(0), |
| 46 | fFMD2iMax(0), | |
| 47 | fFMD2oMax(0), | |
| 48 | fFMD3iMax(0), | |
| 49 | fFMD3oMax(0), | |
| 5bb5d1f6 | 50 | fMaxWeights(0), |
| 51 | fLowCuts(0), | |
| 21d778b1 | 52 | fEtaLumping(32), |
| 53 | fPhiLumping(4), | |
| d2638bb7 | 54 | fDebug(0), |
| 6f4a5c0d | 55 | fCuts(), |
| 8449e3e0 | 56 | fRecalculatePhi(false), |
| 1f7aa5c7 | 57 | fMinQuality(10), |
| 58 | fCache(), | |
| 59 | fDoTiming(false), | |
| 77f97e3f CHC |
60 | fHTiming(0), |
| 61 | fMaxOutliers(0.05), | |
| 62 | fOutlierCut(0.50) | |
| 7984e5f7 | 63 | { |
| 64 | // | |
| 65 | // Constructor | |
| 66 | // | |
| 5ca83fee | 67 | DGUARD(fDebug, 3, "Default CTOR of FMD density calculator"); |
| 7984e5f7 | 68 | } |
| 7e4038b5 | 69 | |
| 70 | //____________________________________________________________________ | |
| 71 | AliFMDDensityCalculator::AliFMDDensityCalculator(const char* title) | |
| 5ca83fee | 72 | : TNamed(fgkFolderName, title), |
| 7e4038b5 | 73 | fRingHistos(), |
| dd497217 | 74 | fSumOfWeights(0), |
| 75 | fWeightedSum(0), | |
| ea3e5d95 | 76 | fCorrections(0), |
| 0bd4b00f | 77 | fMaxParticles(5), |
| 9d05ffeb | 78 | fUsePoisson(false), |
| 8411f7fe | 79 | fUsePhiAcceptance(kPhiCorrectNch), |
| 0bd4b00f | 80 | fAccI(0), |
| 81 | fAccO(0), | |
| 1174780f | 82 | fFMD1iMax(0), |
| 83 | fFMD2iMax(0), | |
| 84 | fFMD2oMax(0), | |
| 85 | fFMD3iMax(0), | |
| 86 | fFMD3oMax(0), | |
| 5bb5d1f6 | 87 | fMaxWeights(0), |
| 88 | fLowCuts(0), | |
| 21d778b1 | 89 | fEtaLumping(32), |
| 90 | fPhiLumping(4), | |
| d2638bb7 | 91 | fDebug(0), |
| 6f4a5c0d | 92 | fCuts(), |
| 8449e3e0 | 93 | fRecalculatePhi(false), |
| 1f7aa5c7 | 94 | fMinQuality(10), |
| 95 | fCache(), | |
| 96 | fDoTiming(false), | |
| 77f97e3f CHC |
97 | fHTiming(0), |
| 98 | fMaxOutliers(0.05), | |
| 99 | fOutlierCut(0.50) | |
| 7e4038b5 | 100 | { |
| 7984e5f7 | 101 | // |
| 102 | // Constructor | |
| 103 | // | |
| 104 | // Parameters: | |
| 105 | // name Name of object | |
| 106 | // | |
| 5ca83fee | 107 | DGUARD(fDebug, 3, "Named CTOR of FMD density calculator: %s", title); |
| 7e4038b5 | 108 | fRingHistos.SetName(GetName()); |
| e308a636 | 109 | fRingHistos.SetOwner(); |
| 7e4038b5 | 110 | fRingHistos.Add(new RingHistos(1, 'I')); |
| 111 | fRingHistos.Add(new RingHistos(2, 'I')); | |
| 112 | fRingHistos.Add(new RingHistos(2, 'O')); | |
| 113 | fRingHistos.Add(new RingHistos(3, 'I')); | |
| 114 | fRingHistos.Add(new RingHistos(3, 'O')); | |
| dd497217 | 115 | fSumOfWeights = new TH1D("sumOfWeights", "Sum of Landau weights", |
| 116 | 200, 0, 20); | |
| 0bd4b00f | 117 | fSumOfWeights->SetFillColor(kRed+1); |
| 118 | fSumOfWeights->SetXTitle("#sum_{i} a_{i} f_{i}(#Delta)"); | |
| dd497217 | 119 | fWeightedSum = new TH1D("weightedSum", "Weighted sum of Landau propability", |
| 120 | 200, 0, 20); | |
| 0bd4b00f | 121 | fWeightedSum->SetFillColor(kBlue+1); |
| 122 | fWeightedSum->SetXTitle("#sum_{i} i a_{i} f_{i}(#Delta)"); | |
| dd497217 | 123 | fCorrections = new TH1D("corrections", "Distribution of corrections", |
| 124 | 100, 0, 10); | |
| 0bd4b00f | 125 | fCorrections->SetFillColor(kBlue+1); |
| 126 | fCorrections->SetXTitle("correction"); | |
| dd497217 | 127 | |
| 0bd4b00f | 128 | fAccI = GenerateAcceptanceCorrection('I'); |
| 129 | fAccO = GenerateAcceptanceCorrection('O'); | |
| 5bb5d1f6 | 130 | |
| 131 | fMaxWeights = new TH2D("maxWeights", "Maximum i of a_{i}'s to use", | |
| 132 | 1, 0, 1, 1, 0, 1); | |
| 133 | fMaxWeights->SetXTitle("#eta"); | |
| 134 | fMaxWeights->SetDirectory(0); | |
| 135 | ||
| 136 | fLowCuts = new TH2D("lowCuts", "Low cuts used", 1, 0, 1, 1, 0, 1); | |
| 137 | fLowCuts->SetXTitle("#eta"); | |
| 138 | fLowCuts->SetDirectory(0); | |
| 139 | ||
| 7e4038b5 | 140 | } |
| 141 | ||
| 142 | //____________________________________________________________________ | |
| 143 | AliFMDDensityCalculator::AliFMDDensityCalculator(const | |
| 144 | AliFMDDensityCalculator& o) | |
| 145 | : TNamed(o), | |
| 146 | fRingHistos(), | |
| dd497217 | 147 | fSumOfWeights(o.fSumOfWeights), |
| 148 | fWeightedSum(o.fWeightedSum), | |
| ea3e5d95 | 149 | fCorrections(o.fCorrections), |
| 0bd4b00f | 150 | fMaxParticles(o.fMaxParticles), |
| 9d05ffeb | 151 | fUsePoisson(o.fUsePoisson), |
| 5bb5d1f6 | 152 | fUsePhiAcceptance(o.fUsePhiAcceptance), |
| 0bd4b00f | 153 | fAccI(o.fAccI), |
| 154 | fAccO(o.fAccO), | |
| 1174780f | 155 | fFMD1iMax(o.fFMD1iMax), |
| 156 | fFMD2iMax(o.fFMD2iMax), | |
| 157 | fFMD2oMax(o.fFMD2oMax), | |
| 158 | fFMD3iMax(o.fFMD3iMax), | |
| 159 | fFMD3oMax(o.fFMD3oMax), | |
| 5bb5d1f6 | 160 | fMaxWeights(o.fMaxWeights), |
| 161 | fLowCuts(o.fLowCuts), | |
| b6b35c77 | 162 | fEtaLumping(o.fEtaLumping), |
| 163 | fPhiLumping(o.fPhiLumping), | |
| d2638bb7 | 164 | fDebug(o.fDebug), |
| 6f4a5c0d | 165 | fCuts(o.fCuts), |
| 8449e3e0 | 166 | fRecalculatePhi(o.fRecalculatePhi), |
| 1f7aa5c7 | 167 | fMinQuality(o.fMinQuality), |
| 168 | fCache(o.fCache), | |
| 169 | fDoTiming(o.fDoTiming), | |
| 77f97e3f CHC |
170 | fHTiming(o.fHTiming), |
| 171 | fMaxOutliers(o.fMaxOutliers), | |
| 172 | fOutlierCut(o.fOutlierCut) | |
| 7e4038b5 | 173 | { |
| 7984e5f7 | 174 | // |
| 175 | // Copy constructor | |
| 176 | // | |
| 177 | // Parameters: | |
| 178 | // o Object to copy from | |
| 179 | // | |
| 5ca83fee | 180 | DGUARD(fDebug, 3, "Copy CTOR of FMD density calculator"); |
| 7e4038b5 | 181 | TIter next(&o.fRingHistos); |
| 182 | TObject* obj = 0; | |
| 183 | while ((obj = next())) fRingHistos.Add(obj); | |
| 184 | } | |
| 185 | ||
| 186 | //____________________________________________________________________ | |
| 187 | AliFMDDensityCalculator::~AliFMDDensityCalculator() | |
| 188 | { | |
| 7984e5f7 | 189 | // |
| 190 | // Destructor | |
| 191 | // | |
| 6ab100ec | 192 | DGUARD(fDebug, 3, "DTOR of FMD density calculator"); |
| 36ffcf83 | 193 | // fRingHistos.Delete(); |
| 7e4038b5 | 194 | } |
| 195 | ||
| 196 | //____________________________________________________________________ | |
| 197 | AliFMDDensityCalculator& | |
| 198 | AliFMDDensityCalculator::operator=(const AliFMDDensityCalculator& o) | |
| 199 | { | |
| 7984e5f7 | 200 | // |
| 201 | // Assignement operator | |
| 202 | // | |
| 203 | // Parameters: | |
| 204 | // o Object to assign from | |
| 205 | // | |
| 206 | // Return: | |
| 207 | // Reference to this object | |
| 208 | // | |
| 6ab100ec | 209 | DGUARD(fDebug, 3, "Assignment of FMD density calculator"); |
| d015ecfe | 210 | if (&o == this) return *this; |
| ea3e5d95 | 211 | TNamed::operator=(o); |
| 7e4038b5 | 212 | |
| d2638bb7 | 213 | fDebug = o.fDebug; |
| 214 | fMaxParticles = o.fMaxParticles; | |
| 215 | fUsePoisson = o.fUsePoisson; | |
| 216 | fUsePhiAcceptance = o.fUsePhiAcceptance; | |
| 217 | fAccI = o.fAccI; | |
| 218 | fAccO = o.fAccO; | |
| 219 | fFMD1iMax = o.fFMD1iMax; | |
| 220 | fFMD2iMax = o.fFMD2iMax; | |
| 221 | fFMD2oMax = o.fFMD2oMax; | |
| 222 | fFMD3iMax = o.fFMD3iMax; | |
| 223 | fFMD3oMax = o.fFMD3oMax; | |
| 224 | fMaxWeights = o.fMaxWeights; | |
| 225 | fLowCuts = o.fLowCuts; | |
| 226 | fEtaLumping = o.fEtaLumping; | |
| 227 | fPhiLumping = o.fPhiLumping; | |
| 228 | fCuts = o.fCuts; | |
| 5ca83fee | 229 | fRecalculatePhi = o.fRecalculatePhi; |
| 8449e3e0 | 230 | fMinQuality = o.fMinQuality; |
| 1f7aa5c7 | 231 | fCache = o.fCache; |
| 232 | fDoTiming = o.fDoTiming; | |
| 233 | fHTiming = o.fHTiming; | |
| 77f97e3f CHC |
234 | fMaxOutliers = o.fMaxOutliers; |
| 235 | fOutlierCut = o.fOutlierCut; | |
| d2638bb7 | 236 | |
| 7e4038b5 | 237 | fRingHistos.Delete(); |
| 238 | TIter next(&o.fRingHistos); | |
| 239 | TObject* obj = 0; | |
| 240 | while ((obj = next())) fRingHistos.Add(obj); | |
| 241 | ||
| 242 | return *this; | |
| 243 | } | |
| 244 | ||
| 1174780f | 245 | //____________________________________________________________________ |
| 246 | void | |
| 5934a3e3 | 247 | AliFMDDensityCalculator::SetupForData(const TAxis& axis) |
| 1174780f | 248 | { |
| 249 | // Intialize this sub-algorithm | |
| 250 | // | |
| 251 | // Parameters: | |
| 5ca83fee | 252 | // etaAxis Eta axis |
| 6ab100ec | 253 | DGUARD(fDebug, 1, "Initialize FMD density calculator"); |
| e2ebf8c4 | 254 | CacheMaxWeights(axis); |
| 4077f3e8 | 255 | |
| 1f7aa5c7 | 256 | fCache.Init(axis); |
| 257 | ||
| 9d05ffeb | 258 | TIter next(&fRingHistos); |
| 259 | RingHistos* o = 0; | |
| d2638bb7 | 260 | while ((o = static_cast<RingHistos*>(next()))) { |
| 5934a3e3 | 261 | o->SetupForData(axis); |
| e18cb8bd | 262 | // o->fMultCut = fCuts.GetFixedCut(o->fDet, o->fRing); |
| 263 | // o->fPoisson.Init(o->fDet,o->fRing,fEtaLumping, fPhiLumping); | |
| d2638bb7 | 264 | } |
| 1174780f | 265 | } |
| 266 | ||
| 7e4038b5 | 267 | //____________________________________________________________________ |
| 268 | AliFMDDensityCalculator::RingHistos* | |
| 269 | AliFMDDensityCalculator::GetRingHistos(UShort_t d, Char_t r) const | |
| 270 | { | |
| 7984e5f7 | 271 | // |
| 272 | // Get the ring histogram container | |
| 273 | // | |
| 274 | // Parameters: | |
| 275 | // d Detector | |
| 276 | // r Ring | |
| 277 | // | |
| 278 | // Return: | |
| 279 | // Ring histogram container | |
| 280 | // | |
| 7e4038b5 | 281 | Int_t idx = -1; |
| 282 | switch (d) { | |
| 283 | case 1: idx = 0; break; | |
| 284 | case 2: idx = 1 + (r == 'I' || r == 'i' ? 0 : 1); break; | |
| 285 | case 3: idx = 3 + (r == 'I' || r == 'i' ? 0 : 1); break; | |
| 286 | } | |
| e308a636 | 287 | if (idx < 0 || idx >= fRingHistos.GetEntries()) { |
| 288 | AliWarning(Form("Index %d of FMD%d%c out of range", idx, d, r)); | |
| 289 | return 0; | |
| 290 | } | |
| 7e4038b5 | 291 | |
| 292 | return static_cast<RingHistos*>(fRingHistos.At(idx)); | |
| 293 | } | |
| 5bb5d1f6 | 294 | |
| 7095962e CHC |
295 | namespace { |
| 296 | Double_t Rng2Cut(UShort_t d, Char_t r, Int_t xbin, TH2* h) | |
| 297 | { | |
| 298 | Double_t ret = 1024; | |
| 299 | if (xbin < 1 && xbin > h->GetXaxis()->GetNbins()) return ret; | |
| 300 | Int_t ybin = 0; | |
| 301 | switch(d) { | |
| 302 | case 1: ybin = 1; break; | |
| 303 | case 2: ybin = (r=='i' || r=='I') ? 2 : 3; break; | |
| 304 | case 3: ybin = (r=='i' || r=='I') ? 4 : 5; break; | |
| 305 | default: return ret; | |
| 306 | } | |
| 307 | ret = h->GetBinContent(xbin,ybin); | |
| 308 | return ret; | |
| 309 | } | |
| 310 | } | |
| 311 | ||
| 0bd4b00f | 312 | //____________________________________________________________________ |
| 313 | Double_t | |
| 5bb5d1f6 | 314 | AliFMDDensityCalculator::GetMultCut(UShort_t d, Char_t r, Int_t ieta, |
| 7095962e | 315 | Bool_t /*errors*/) const |
| 0bd4b00f | 316 | { |
| 7984e5f7 | 317 | // |
| 318 | // Get the multiplicity cut. If the user has set fMultCut (via | |
| 319 | // SetMultCut) then that value is used. If not, then the lower | |
| 320 | // value of the fit range for the enery loss fits is returned. | |
| 321 | // | |
| 322 | // Return: | |
| 323 | // Lower cut on multiplicity | |
| 324 | // | |
| 7095962e CHC |
325 | return Rng2Cut(d, r, ieta, fLowCuts); |
| 326 | // return fCuts.GetMultCut(d,r,ieta,errors); | |
| 5bb5d1f6 | 327 | } |
| 328 | ||
| 329 | //____________________________________________________________________ | |
| 330 | Double_t | |
| 331 | AliFMDDensityCalculator::GetMultCut(UShort_t d, Char_t r, Double_t eta, | |
| 7095962e | 332 | Bool_t /*errors*/) const |
| 5bb5d1f6 | 333 | { |
| 334 | // | |
| 335 | // Get the multiplicity cut. If the user has set fMultCut (via | |
| 336 | // SetMultCut) then that value is used. If not, then the lower | |
| 337 | // value of the fit range for the enery loss fits is returned. | |
| 338 | // | |
| 339 | // Return: | |
| 340 | // Lower cut on multiplicity | |
| 341 | // | |
| 7095962e CHC |
342 | Int_t ieta = fLowCuts->GetXaxis()->FindBin(eta); |
| 343 | return Rng2Cut(d, r, ieta, fLowCuts); | |
| 344 | // return fCuts.GetMultCut(d,r,eta,errors); | |
| 0bd4b00f | 345 | } |
| 0ccdab7b | 346 | |
| 347 | #ifndef NO_TIMING | |
| 348 | # define START_TIMER(T) if (fDoTiming) T.Start(true) | |
| 349 | # define GET_TIMER(T,V) if (fDoTiming) V = T.CpuTime() | |
| 350 | # define ADD_TIMER(T,V) if (fDoTiming) V += T.CpuTime() | |
| 351 | #else | |
| 352 | # define START_TIMER(T) do {} while (false) | |
| 353 | # define GET_TIMER(T,V) do {} while (false) | |
| 354 | # define ADD_TIMER(T,V) do {} while (false) | |
| 355 | #endif | |
| 356 | ||
| 7e4038b5 | 357 | //____________________________________________________________________ |
| 358 | Bool_t | |
| 359 | AliFMDDensityCalculator::Calculate(const AliESDFMD& fmd, | |
| 360 | AliForwardUtil::Histos& hists, | |
| 21d778b1 | 361 | Bool_t lowFlux, |
| b3d81bee | 362 | Double_t /*cent*/, |
| 5ca83fee | 363 | const TVector3& ip) |
| 7e4038b5 | 364 | { |
| 7984e5f7 | 365 | // |
| 366 | // Do the calculations | |
| 367 | // | |
| 368 | // Parameters: | |
| 369 | // fmd AliESDFMD object (possibly) corrected for sharing | |
| 370 | // hists Histogram cache | |
| 371 | // vtxBin Vertex bin | |
| 372 | // lowFlux Low flux flag. | |
| 373 | // | |
| 374 | // Return: | |
| 375 | // true on successs | |
| 6ab100ec | 376 | DGUARD(fDebug, 1, "Calculate density in FMD density calculator"); |
| 5ca83fee | 377 | |
| 1f7aa5c7 | 378 | TStopwatch timer; |
| 379 | TStopwatch totalT; | |
| 380 | ||
| 381 | // First measurements of timing | |
| 382 | // Re-calculation : fraction of sum 32.0% of total 18.1% | |
| 383 | // N_{particle} : fraction of sum 15.2% of total 8.6% | |
| 384 | // Correction : fraction of sum 26.4% of total 14.9% | |
| 385 | // #phi acceptance : fraction of sum 0.2% of total 0.1% | |
| 386 | // Copy to cache : fraction of sum 3.9% of total 2.2% | |
| 387 | // Poisson calculation : fraction of sum 18.7% of total 10.6% | |
| 388 | // Diagnostics : fraction of sum 3.7% of total 2.1% | |
| 1f7aa5c7 | 389 | Double_t nPartTime = 0; |
| 390 | Double_t corrTime = 0; | |
| 391 | Double_t rePhiTime = 0; | |
| 392 | Double_t copyTime = 0; | |
| 393 | Double_t poissonTime= 0; | |
| 394 | Double_t diagTime = 0; | |
| 0ccdab7b | 395 | START_TIMER(totalT); |
| 1f7aa5c7 | 396 | |
| 397 | Double_t etaCache[20*512]; // Same number of strips per ring | |
| 398 | Double_t phiCache[20*512]; // whether it is inner our outer. | |
| 399 | // We do not use TArrayD because we do not wont a bounds check | |
| 400 | // TArrayD etaCache(20*512); // Same number of strips per ring | |
| 401 | // TArrayD phiCache(20*512); // whether it is inner our outer. | |
| 402 | ||
| 5ca83fee | 403 | // --- Loop over detectors ----------------------------------------- |
| 7e4038b5 | 404 | for (UShort_t d=1; d<=3; d++) { |
| 405 | UShort_t nr = (d == 1 ? 1 : 2); | |
| 406 | for (UShort_t q=0; q<nr; q++) { | |
| 407 | Char_t r = (q == 0 ? 'I' : 'O'); | |
| 408 | UShort_t ns= (q == 0 ? 20 : 40); | |
| 409 | UShort_t nt= (q == 0 ? 512 : 256); | |
| 410 | TH2D* h = hists.Get(d,r); | |
| 411 | RingHistos* rh= GetRingHistos(d,r); | |
| e308a636 | 412 | if (!rh) { |
| 413 | AliError(Form("No ring histogram found for FMD%d%c", d, r)); | |
| 414 | fRingHistos.ls(); | |
| 415 | return false; | |
| 416 | } | |
| e18cb8bd | 417 | // rh->fPoisson.SetObject(d,r,vtxbin,cent); |
| d23503ee | 418 | rh->fPoisson.Reset(0); |
| 5ca83fee | 419 | rh->fTotal->Reset(); |
| 420 | rh->fGood->Reset(); | |
| 821ffd28 | 421 | // rh->ResetPoissonHistos(h, fEtaLumping, fPhiLumping); |
| 1f7aa5c7 | 422 | |
| 423 | // Reset our eta cache | |
| 424 | // for (Int_t i = 0; i < 20*512; i++) | |
| 425 | // etaCache[i] = phiCache[i] = AliESDFMD::kInvalidEta; | |
| 426 | memset(etaCache, 0, sizeof(Double_t)*20*512); | |
| 427 | memset(phiCache, 0, sizeof(Double_t)*20*512); | |
| 428 | // etaCache.Reset(AliESDFMD::kInvalidEta); | |
| 429 | // phiCache.Reset(AliESDFMD::kInvalidEta); | |
| 430 | ||
| 5ca83fee | 431 | // --- Loop over sectors and strips ---------------------------- |
| 7e4038b5 | 432 | for (UShort_t s=0; s<ns; s++) { |
| 433 | for (UShort_t t=0; t<nt; t++) { | |
| 3e4a0875 | 434 | |
| 5ca83fee | 435 | Float_t mult = fmd.Multiplicity(d,r,s,t); |
| 436 | Float_t phi = fmd.Phi(d,r,s,t) * TMath::DegToRad(); | |
| 437 | Float_t eta = fmd.Eta(d,r,s,t); | |
| 438 | Double_t oldPhi = phi; | |
| 439 | ||
| 440 | // --- Re-calculate eta - needed for satelittes ------------ | |
| 0ccdab7b | 441 | START_TIMER(timer); |
| 1f7aa5c7 | 442 | etaCache[s*nt+t] = eta; |
| 5ca83fee | 443 | |
| 444 | // --- Check this strip ------------------------------------ | |
| 445 | rh->fTotal->Fill(eta); | |
| bfab35d9 | 446 | if (mult == AliESDFMD::kInvalidMult) { // || mult > 20) { |
| 8449e3e0 | 447 | // Do not count invalid stuff |
| 448 | rh->fELoss->Fill(-1); | |
| 449 | // rh->fEvsN->Fill(mult,-1); | |
| 450 | // rh->fEvsM->Fill(mult,-1); | |
| 33b078ef | 451 | continue; |
| 452 | } | |
| bfab35d9 | 453 | if (mult > 20) |
| 454 | AliWarningF("Raw multiplicity of FMD%d%c[%02d,%03d] = %f > 20", | |
| 455 | d, r, s, t, mult); | |
| 5ca83fee | 456 | // --- Automatic calculation of acceptance ----------------- |
| 457 | rh->fGood->Fill(eta); | |
| 458 | ||
| 459 | // --- If we asked to re-calculate phi for (x,y) IP -------- | |
| 0ccdab7b | 460 | START_TIMER(timer); |
| 5ca83fee | 461 | if (fRecalculatePhi) { |
| 462 | oldPhi = phi; | |
| 463 | phi = AliForwardUtil::GetPhiFromStrip(r, t, phi, ip.X(), ip.Y()); | |
| 464 | } | |
| 1f7aa5c7 | 465 | phiCache[s*nt+t] = phi; |
| 0ccdab7b | 466 | ADD_TIMER(timer,rePhiTime); |
| 5ca83fee | 467 | |
| 468 | // --- Apply phi corner correction to eloss ---------------- | |
| 8411f7fe | 469 | if (fUsePhiAcceptance == kPhiCorrectELoss) |
| 0082a8fc | 470 | mult *= AcceptanceCorrection(r,t); |
| 5bb5d1f6 | 471 | |
| 5ca83fee | 472 | // --- Get the low multiplicity cut ------------------------ |
| 8e400b14 | 473 | Double_t cut = 1024; |
| 474 | if (eta != AliESDFMD::kInvalidEta) cut = GetMultCut(d, r, eta,false); | |
| bfab35d9 | 475 | else AliWarningF("Eta for FMD%d%c[%02d,%03d] is invalid: %f", |
| 476 | d, r, s, t, eta); | |
| 8e400b14 | 477 | |
| 5ca83fee | 478 | // --- Now caluculate Nch for this strip using fits -------- |
| 0ccdab7b | 479 | START_TIMER(timer); |
| 5bb5d1f6 | 480 | Double_t n = 0; |
| 5ca83fee | 481 | if (cut > 0 && mult > cut) n = NParticles(mult,d,r,eta,lowFlux); |
| f7cfc454 | 482 | rh->fELoss->Fill(mult); |
| 8449e3e0 | 483 | // rh->fEvsN->Fill(mult,n); |
| 484 | // rh->fEtaVsN->Fill(eta, n); | |
| 0ccdab7b | 485 | ADD_TIMER(timer,nPartTime); |
| 9fde7142 | 486 | |
| 5ca83fee | 487 | // --- Calculate correction if needed ---------------------- |
| 0ccdab7b | 488 | START_TIMER(timer); |
| 1f7aa5c7 | 489 | // Temporary stuff - remove Correction call |
| 490 | Double_t c = 1; | |
| 491 | if (fUsePhiAcceptance == kPhiCorrectNch) | |
| 492 | c = AcceptanceCorrection(r,t); | |
| 493 | // Double_t c = Correction(d,r,t,eta,lowFlux); | |
| 0ccdab7b | 494 | ADD_TIMER(timer,corrTime); |
| dd497217 | 495 | fCorrections->Fill(c); |
| 7e4038b5 | 496 | if (c > 0) n /= c; |
| 8449e3e0 | 497 | // rh->fEvsM->Fill(mult,n); |
| 498 | // rh->fEtaVsM->Fill(eta, n); | |
| 0bd4b00f | 499 | rh->fCorr->Fill(eta, c); |
| f7cfc454 | 500 | |
| 5ca83fee | 501 | // --- Accumulate Poisson statistics ----------------------- |
| 821ffd28 | 502 | Bool_t hit = (n > 0.9 && c > 0); |
| 5ca83fee | 503 | if (hit) { |
| 504 | rh->fELossUsed->Fill(mult); | |
| 505 | if (fRecalculatePhi) { | |
| 506 | rh->fPhiBefore->Fill(oldPhi); | |
| 507 | rh->fPhiAfter->Fill(phi); | |
| 508 | } | |
| 509 | } | |
| 21d778b1 | 510 | rh->fPoisson.Fill(t,s,hit,1./c); |
| 7e4038b5 | 511 | h->Fill(eta,phi,n); |
| 5ca83fee | 512 | |
| 513 | // --- If we use ELoss fits, apply now --------------------- | |
| 5bb5d1f6 | 514 | if (!fUsePoisson) rh->fDensity->Fill(eta,phi,n); |
| 7e4038b5 | 515 | } // for t |
| 516 | } // for s | |
| 5ca83fee | 517 | |
| 518 | // --- Automatic acceptance - Calculate as an efficiency ------- | |
| 1f7aa5c7 | 519 | // This is very fast, so we do not bother to time it |
| 5ca83fee | 520 | rh->fGood->Divide(rh->fGood, rh->fTotal, 1, 1, "B"); |
| 521 | ||
| 522 | // --- Make a copy and reset as needed ------------------------- | |
| 0ccdab7b | 523 | START_TIMER(timer); |
| 1f7aa5c7 | 524 | TH2D* hclone = fCache.Get(d,r); |
| 525 | // hclone->Reset(); | |
| 526 | // TH2D* hclone = static_cast<TH2D*>(h->Clone("hclone")); | |
| 21d778b1 | 527 | if (!fUsePoisson) hclone->Reset(); |
| 1f7aa5c7 | 528 | else { |
| 529 | for (Int_t i = 0; i <= h->GetNbinsX()+1; i++) { | |
| 530 | for (Int_t j = 0; j <= h->GetNbinsY()+1; j++) { | |
| 531 | hclone->SetBinContent(i,j,h->GetBinContent(i,j)); | |
| 532 | hclone->SetBinError(i,j,h->GetBinError(i,j)); | |
| 533 | } | |
| 534 | } | |
| 535 | // hclone->Add(h); | |
| 536 | h->Reset(); | |
| 537 | } | |
| 0ccdab7b | 538 | ADD_TIMER(timer,copyTime); |
| 21d778b1 | 539 | |
| 5ca83fee | 540 | // --- Store Poisson result ------------------------------------ |
| 0ccdab7b | 541 | START_TIMER(timer); |
| 821ffd28 | 542 | TH2D* poisson = rh->fPoisson.Result(); |
| d23503ee | 543 | for (Int_t t=0; t < poisson->GetNbinsX(); t++) { |
| 544 | for (Int_t s=0; s < poisson->GetNbinsY(); s++) { | |
| 21d778b1 | 545 | |
| 546 | Double_t poissonV = poisson->GetBinContent(t+1,s+1); | |
| 1f7aa5c7 | 547 | // Use cached eta - since the calls to GetEtaFromStrip and |
| 548 | // GetPhiFromStrip are _very_ expensive | |
| 549 | Double_t phi = phiCache[s*nt+t]; | |
| 550 | Double_t eta = etaCache[s*nt+t]; | |
| 551 | // Double_t phi = fmd.Phi(d,r,s,t) * TMath::DegToRad(); | |
| 552 | // Double_t eta = fmd.Eta(d,r,s,t); | |
| 1f7aa5c7 | 553 | if (fUsePoisson) { |
| 21d778b1 | 554 | h->Fill(eta,phi,poissonV); |
| 1f7aa5c7 | 555 | rh->fDensity->Fill(eta, phi, poissonV); |
| 556 | } | |
| 21d778b1 | 557 | else |
| 558 | hclone->Fill(eta,phi,poissonV); | |
| 21d778b1 | 559 | } |
| 560 | } | |
| 0ccdab7b | 561 | ADD_TIMER(timer,poissonTime); |
| 21d778b1 | 562 | |
| 5ca83fee | 563 | // --- Make diagnostics - eloss vs poisson --------------------- |
| 0ccdab7b | 564 | START_TIMER(timer); |
| 5ca83fee | 565 | Int_t nY = h->GetNbinsY(); |
| 77f97e3f CHC |
566 | Int_t nIn = 0; // Count non-outliers |
| 567 | Int_t nOut = 0; // Count outliers | |
| 21d778b1 | 568 | for (Int_t ieta=1; ieta <= h->GetNbinsX(); ieta++) { |
| 5ca83fee | 569 | // Set the overflow bin to contain the phi acceptance |
| 570 | Double_t phiAcc = rh->fGood->GetBinContent(ieta); | |
| 571 | Double_t phiAccE = rh->fGood->GetBinError(ieta); | |
| 572 | h->SetBinContent(ieta, nY+1, phiAcc); | |
| 573 | h->SetBinError(ieta, nY+1, phiAccE); | |
| 574 | Double_t eta = h->GetXaxis()->GetBinCenter(ieta); | |
| 575 | rh->fPhiAcc->Fill(eta, ip.Z(), phiAcc); | |
| 5ca83fee | 576 | for (Int_t iphi=1; iphi<= nY; iphi++) { |
| e83d0620 | 577 | |
| 21d778b1 | 578 | Double_t poissonV = 0; //h->GetBinContent(,s+1); |
| 579 | Double_t eLossV = 0; | |
| 580 | if(fUsePoisson) { | |
| 581 | poissonV = h->GetBinContent(ieta,iphi); | |
| 582 | eLossV = hclone->GetBinContent(ieta,iphi); | |
| 583 | } | |
| 584 | else { | |
| 585 | poissonV = hclone->GetBinContent(ieta,iphi); | |
| 586 | eLossV = h->GetBinContent(ieta,iphi); | |
| 587 | } | |
| 7b03929c | 588 | |
| 77f97e3f CHC |
589 | if (poissonV < 1e-12 && eLossV < 1e-12) |
| 590 | // we do not care about trivially empty bins | |
| 591 | continue; | |
| 5ca83fee | 592 | |
| 77f97e3f CHC |
593 | Bool_t outlier = CheckOutlier(eLossV, poissonV, fOutlierCut); |
| 594 | Double_t rel = eLossV < 1e-12 ? 0 : (poissonV - eLossV) / eLossV; | |
| 595 | if (outlier) { | |
| 596 | rh->fELossVsPoissonOut->Fill(eLossV, poissonV); | |
| 597 | rh->fDiffELossPoissonOut->Fill(rel); | |
| 598 | nOut++; | |
| 9d05ffeb | 599 | } |
| 77f97e3f CHC |
600 | else { |
| 601 | rh->fELossVsPoisson->Fill(eLossV, poissonV); | |
| 602 | rh->fDiffELossPoisson->Fill(rel); | |
| 603 | nIn++; | |
| 604 | } // if (outlier) | |
| 605 | } // for (iphi) | |
| 606 | } // for (ieta) | |
| 67a4bb96 CHC |
607 | Int_t nTotal = (nIn+nOut); |
| 608 | Double_t outRatio = (nTotal > 0 ? Double_t(nOut) / nTotal : 0); | |
| 77f97e3f CHC |
609 | rh->fOutliers->Fill(outRatio); |
| 610 | if (outRatio < fMaxOutliers) rh->fPoisson.FillDiagnostics(); | |
| 611 | else h->SetBit(AliForwardUtil::kSkipRing); | |
| 0ccdab7b | 612 | ADD_TIMER(timer,diagTime); |
| 1f7aa5c7 | 613 | // delete hclone; |
| 21d778b1 | 614 | |
| 7e4038b5 | 615 | } // for q |
| 616 | } // for d | |
| a7f31ad8 | 617 | |
| 1f7aa5c7 | 618 | if (fDoTiming) { |
| 0ccdab7b | 619 | // fHTiming->Fill(1,reEtaTime); |
| 1f7aa5c7 | 620 | fHTiming->Fill(2,nPartTime); |
| 621 | fHTiming->Fill(3,corrTime); | |
| 622 | fHTiming->Fill(4,rePhiTime); | |
| 623 | fHTiming->Fill(5,copyTime); | |
| 624 | fHTiming->Fill(6,poissonTime); | |
| 625 | fHTiming->Fill(7,diagTime); | |
| 626 | fHTiming->Fill(8,totalT.CpuTime()); | |
| 627 | } | |
| 628 | ||
| 7e4038b5 | 629 | return kTRUE; |
| 630 | } | |
| 631 | ||
| 77f97e3f CHC |
632 | //_____________________________________________________________________ |
| 633 | Bool_t | |
| 634 | AliFMDDensityCalculator::CheckOutlier(Double_t eloss, | |
| 635 | Double_t poisson, | |
| 636 | Double_t cut) const | |
| 637 | { | |
| 638 | if (eloss < 1e-6) return true; | |
| 639 | Double_t diff = TMath::Abs(poisson - eloss) / eloss; | |
| 640 | return diff > cut; | |
| 641 | } | |
| 1174780f | 642 | //_____________________________________________________________________ |
| 643 | Int_t | |
| fb3430ac | 644 | AliFMDDensityCalculator::FindMaxWeight(const AliFMDCorrELossFit* cor, |
| 1174780f | 645 | UShort_t d, Char_t r, Int_t iEta) const |
| 646 | { | |
| fb3430ac | 647 | // |
| 648 | // Find the max weight to use for FMD<i>dr</i> in eta bin @a iEta | |
| 649 | // | |
| 650 | // Parameters: | |
| 651 | // cor Correction | |
| 652 | // d Detector | |
| 653 | // r Ring | |
| 654 | // iEta Eta bin | |
| 655 | // | |
| 241cca4d | 656 | DGUARD(fDebug, 10, "Find maximum weight in FMD density calculator"); |
| f61b1745 | 657 | if(!cor) return -1; |
| 658 | ||
| 8449e3e0 | 659 | AliFMDCorrELossFit::ELossFit* fit = cor->FindFit(d,r,iEta, -1); |
| 1174780f | 660 | if (!fit) { |
| 661 | // AliWarning(Form("No energy loss fit for FMD%d%c at eta=%f", d, r, eta)); | |
| 662 | return -1; | |
| 663 | } | |
| 8449e3e0 | 664 | return fit->FindMaxWeight(2*AliFMDCorrELossFit::ELossFit::fgMaxRelError, |
| 665 | AliFMDCorrELossFit::ELossFit::fgLeastWeight, | |
| 666 | fMaxParticles); | |
| 1174780f | 667 | } |
| 7095962e CHC |
668 | //_____________________________________________________________________ |
| 669 | Int_t | |
| 670 | AliFMDDensityCalculator::FindMaxWeight(const AliFMDCorrELossFit* cor, | |
| 671 | UShort_t d, Char_t r, Double_t eta) const | |
| 672 | { | |
| 673 | // | |
| 674 | // Find the max weight to use for FMD<i>dr</i> in eta bin @a iEta | |
| 675 | // | |
| 676 | // Parameters: | |
| 677 | // cor Correction | |
| 678 | // d Detector | |
| 679 | // r Ring | |
| 680 | // eta Eta | |
| 681 | // | |
| 682 | DGUARD(fDebug, 10, "Find maximum weight in FMD density calculator"); | |
| 683 | if(!cor) return -1; | |
| 684 | ||
| 685 | AliFMDCorrELossFit::ELossFit* fit = cor->FindFit(d,r,eta, -1); | |
| 686 | if (!fit) { | |
| 687 | // AliWarning(Form("No energy loss fit for FMD%d%c at eta=%f", d, r, eta)); | |
| 688 | return -1; | |
| 689 | } | |
| 690 | return fit->FindMaxWeight(2*AliFMDCorrELossFit::ELossFit::fgMaxRelError, | |
| 691 | AliFMDCorrELossFit::ELossFit::fgLeastWeight, | |
| 692 | fMaxParticles); | |
| 693 | } | |
| 1174780f | 694 | |
| 695 | //_____________________________________________________________________ | |
| 696 | void | |
| e2ebf8c4 | 697 | AliFMDDensityCalculator::CacheMaxWeights(const TAxis& axis) |
| 1174780f | 698 | { |
| fb3430ac | 699 | // |
| 700 | // Find the max weights and cache them | |
| 701 | // | |
| 6ab100ec | 702 | DGUARD(fDebug, 2, "Cache maximum weights in FMD density calculator"); |
| 1174780f | 703 | AliForwardCorrectionManager& fcm = AliForwardCorrectionManager::Instance(); |
| 8449e3e0 | 704 | const AliFMDCorrELossFit* cor = fcm.GetELossFit(); |
| 705 | cor->CacheBins(fMinQuality); | |
| 7095962e | 706 | cor->Print(fDebug > 5 ? "RCS" : "C"); |
| a76fb27d | 707 | |
| 708 | TAxis eta(axis.GetNbins(), | |
| 709 | axis.GetXmin(), | |
| 710 | axis.GetXmax()); | |
| 711 | ||
| f61b1745 | 712 | if(cor) |
| 713 | eta.Set(cor->GetEtaAxis().GetNbins(), | |
| 714 | cor->GetEtaAxis().GetXmin(), | |
| 715 | cor->GetEtaAxis().GetXmax()); | |
| 1174780f | 716 | |
| 717 | Int_t nEta = eta.GetNbins(); | |
| 718 | fFMD1iMax.Set(nEta); | |
| 719 | fFMD2iMax.Set(nEta); | |
| 720 | fFMD2oMax.Set(nEta); | |
| 721 | fFMD3iMax.Set(nEta); | |
| 722 | fFMD3oMax.Set(nEta); | |
| 723 | ||
| 5bb5d1f6 | 724 | fMaxWeights->SetBins(nEta, eta.GetXmin(), eta.GetXmax(), 5, .5, 5.5); |
| 725 | fMaxWeights->GetYaxis()->SetBinLabel(1, "FMD1i"); | |
| 726 | fMaxWeights->GetYaxis()->SetBinLabel(2, "FMD2i"); | |
| 727 | fMaxWeights->GetYaxis()->SetBinLabel(3, "FMD2o"); | |
| 728 | fMaxWeights->GetYaxis()->SetBinLabel(4, "FMD3i"); | |
| 729 | fMaxWeights->GetYaxis()->SetBinLabel(5, "FMD3o"); | |
| 730 | ||
| 731 | AliInfo(Form("Get eta axis with %d bins from %f to %f", | |
| 732 | nEta, eta.GetXmin(), eta.GetXmax())); | |
| 733 | fLowCuts->SetBins(nEta, eta.GetXmin(), eta.GetXmax(), 5, .5, 5.5); | |
| 734 | fLowCuts->GetYaxis()->SetBinLabel(1, "FMD1i"); | |
| 735 | fLowCuts->GetYaxis()->SetBinLabel(2, "FMD2i"); | |
| 736 | fLowCuts->GetYaxis()->SetBinLabel(3, "FMD2o"); | |
| 737 | fLowCuts->GetYaxis()->SetBinLabel(4, "FMD3i"); | |
| 738 | fLowCuts->GetYaxis()->SetBinLabel(5, "FMD3o"); | |
| 7095962e | 739 | |
| 1174780f | 740 | for (Int_t i = 0; i < nEta; i++) { |
| 7095962e | 741 | Double_t leta = fMaxWeights->GetXaxis()->GetBinCenter(i+1); |
| 5bb5d1f6 | 742 | Double_t w[5]; |
| 7095962e CHC |
743 | w[0] = fFMD1iMax[i] = FindMaxWeight(cor, 1, 'I', leta); |
| 744 | w[1] = fFMD2iMax[i] = FindMaxWeight(cor, 2, 'I', leta); | |
| 745 | w[2] = fFMD2oMax[i] = FindMaxWeight(cor, 2, 'O', leta); | |
| 746 | w[3] = fFMD3iMax[i] = FindMaxWeight(cor, 3, 'I', leta); | |
| 747 | w[4] = fFMD3oMax[i] = FindMaxWeight(cor, 3, 'O', leta); | |
| 748 | for (Int_t j = 0; j < 5; j++) | |
| 5bb5d1f6 | 749 | if (w[j] > 0) fMaxWeights->SetBinContent(i+1, j+1, w[j]); |
| 1174780f | 750 | } |
| 7095962e CHC |
751 | |
| 752 | // Cache cuts in histogram | |
| 753 | fCuts.FillHistogram(fLowCuts); | |
| 1174780f | 754 | } |
| 755 | ||
| 756 | //_____________________________________________________________________ | |
| 757 | Int_t | |
| 758 | AliFMDDensityCalculator::GetMaxWeight(UShort_t d, Char_t r, Int_t iEta) const | |
| 759 | { | |
| fb3430ac | 760 | // |
| 761 | // Find the (cached) maximum weight for FMD<i>dr</i> in | |
| 762 | // @f$\eta@f$ bin @a iEta | |
| 763 | // | |
| 764 | // Parameters: | |
| 765 | // d Detector | |
| 766 | // r Ring | |
| 767 | // iEta Eta bin | |
| 768 | // | |
| 769 | // Return: | |
| 770 | // max weight or <= 0 in case of problems | |
| 771 | // | |
| 1174780f | 772 | if (iEta < 0) return -1; |
| 773 | ||
| 774 | const TArrayI* max = 0; | |
| 775 | switch (d) { | |
| 776 | case 1: max = &fFMD1iMax; break; | |
| 777 | case 2: max = (r == 'I' || r == 'i' ? &fFMD2iMax : &fFMD2oMax); break; | |
| 778 | case 3: max = (r == 'I' || r == 'i' ? &fFMD3iMax : &fFMD3oMax); break; | |
| 779 | } | |
| 780 | if (!max) { | |
| 781 | AliWarning(Form("No array for FMD%d%c", d, r)); | |
| 782 | return -1; | |
| 783 | } | |
| 784 | ||
| 785 | if (iEta >= max->fN) { | |
| 786 | AliWarning(Form("Eta bin %3d out of bounds [0,%d]", | |
| 787 | iEta, max->fN-1)); | |
| 788 | return -1; | |
| 789 | } | |
| 790 | ||
| 791 | AliDebug(30,Form("Max weight for FMD%d%c eta bin %3d: %d", d, r, iEta, | |
| 792 | max->At(iEta))); | |
| 793 | return max->At(iEta); | |
| 794 | } | |
| 795 | ||
| 796 | //_____________________________________________________________________ | |
| 797 | Int_t | |
| 798 | AliFMDDensityCalculator::GetMaxWeight(UShort_t d, Char_t r, Float_t eta) const | |
| 799 | { | |
| fb3430ac | 800 | // |
| 801 | // Find the (cached) maximum weight for FMD<i>dr</i> iat | |
| 802 | // @f$\eta@f$ | |
| 803 | // | |
| 804 | // Parameters: | |
| 805 | // d Detector | |
| 806 | // r Ring | |
| 807 | // eta Eta bin | |
| 808 | // | |
| 809 | // Return: | |
| 810 | // max weight or <= 0 in case of problems | |
| 811 | // | |
| 1174780f | 812 | AliForwardCorrectionManager& fcm = AliForwardCorrectionManager::Instance(); |
| 813 | Int_t iEta = fcm.GetELossFit()->FindEtaBin(eta) -1; | |
| 814 | ||
| 815 | return GetMaxWeight(d, r, iEta); | |
| 816 | } | |
| 817 | ||
| 7e4038b5 | 818 | //_____________________________________________________________________ |
| 819 | Float_t | |
| 820 | AliFMDDensityCalculator::NParticles(Float_t mult, | |
| 821 | UShort_t d, | |
| 822 | Char_t r, | |
| 7e4038b5 | 823 | Float_t eta, |
| 824 | Bool_t lowFlux) const | |
| 825 | { | |
| 7984e5f7 | 826 | // |
| 827 | // Get the number of particles corresponding to the signal mult | |
| 828 | // | |
| 829 | // Parameters: | |
| 830 | // mult Signal | |
| 831 | // d Detector | |
| 832 | // r Ring | |
| 833 | // s Sector | |
| 834 | // t Strip (not used) | |
| 835 | // v Vertex bin | |
| 836 | // eta Pseudo-rapidity | |
| 837 | // lowFlux Low-flux flag | |
| 838 | // | |
| 839 | // Return: | |
| 840 | // The number of particles | |
| 841 | // | |
| 5bb5d1f6 | 842 | // if (mult <= GetMultCut()) return 0; |
| 6ab100ec | 843 | DGUARD(fDebug, 3, "Calculate Nch in FMD density calculator"); |
| 7e4038b5 | 844 | if (lowFlux) return 1; |
| 0bd4b00f | 845 | |
| 0bd4b00f | 846 | AliForwardCorrectionManager& fcm = AliForwardCorrectionManager::Instance(); |
| 8449e3e0 | 847 | AliFMDCorrELossFit::ELossFit* fit = fcm.GetELossFit()->FindFit(d,r,eta, -1); |
| 0bd4b00f | 848 | if (!fit) { |
| 8449e3e0 | 849 | AliWarning(Form("No energy loss fit for FMD%d%c at eta=%f qual=%d", |
| 850 | d, r, eta, fMinQuality)); | |
| 0bd4b00f | 851 | return 0; |
| 852 | } | |
| 853 | ||
| 1174780f | 854 | Int_t m = GetMaxWeight(d,r,eta); // fit->FindMaxWeight(); |
| 0bd4b00f | 855 | if (m < 1) { |
| 856 | AliWarning(Form("No good fits for FMD%d%c at eta=%f", d, r, eta)); | |
| 857 | return 0; | |
| 858 | } | |
| 79909b8b | 859 | |
| 0bd4b00f | 860 | UShort_t n = TMath::Min(fMaxParticles, UShort_t(m)); |
| 861 | Double_t ret = fit->EvaluateWeighted(mult, n); | |
| 79909b8b | 862 | |
| 0bd4b00f | 863 | if (fDebug > 10) { |
| 864 | AliInfo(Form("FMD%d%c, eta=%7.4f, %8.5f -> %8.5f", d, r, eta, mult, ret)); | |
| 865 | } | |
| 79909b8b | 866 | |
| 0bd4b00f | 867 | fWeightedSum->Fill(ret); |
| 868 | fSumOfWeights->Fill(ret); | |
| 3e4a0875 | 869 | |
| 0bd4b00f | 870 | return ret; |
| 7e4038b5 | 871 | } |
| 872 | ||
| 873 | //_____________________________________________________________________ | |
| 874 | Float_t | |
| 0bd4b00f | 875 | AliFMDDensityCalculator::Correction(UShort_t d, |
| 876 | Char_t r, | |
| 0bd4b00f | 877 | UShort_t t, |
| 0bd4b00f | 878 | Float_t eta, |
| 879 | Bool_t lowFlux) const | |
| 7e4038b5 | 880 | { |
| 7984e5f7 | 881 | // |
| 882 | // Get the inverse correction factor. This consist of | |
| 883 | // | |
| 884 | // - acceptance correction (corners of sensors) | |
| 885 | // - double hit correction (for low-flux events) | |
| 886 | // - dead strip correction | |
| 887 | // | |
| 888 | // Parameters: | |
| 889 | // d Detector | |
| 890 | // r Ring | |
| 891 | // s Sector | |
| 892 | // t Strip (not used) | |
| 893 | // v Vertex bin | |
| 894 | // eta Pseudo-rapidity | |
| 895 | // lowFlux Low-flux flag | |
| 896 | // | |
| 897 | // Return: | |
| 898 | // | |
| 899 | // | |
| 241cca4d | 900 | DGUARD(fDebug, 10, "Apply correction in FMD density calculator"); |
| 5bb5d1f6 | 901 | Float_t correction = 1; |
| 8411f7fe | 902 | if (fUsePhiAcceptance == kPhiCorrectNch) |
| 903 | correction = AcceptanceCorrection(r,t); | |
| 7e4038b5 | 904 | if (lowFlux) { |
| 1f7aa5c7 | 905 | AliForwardCorrectionManager& fcm = AliForwardCorrectionManager::Instance(); |
| 906 | ||
| 0bd4b00f | 907 | TH1D* dblHitCor = 0; |
| 908 | if (fcm.GetDoubleHit()) | |
| 909 | dblHitCor = fcm.GetDoubleHit()->GetCorrection(d,r); | |
| 910 | ||
| 7e4038b5 | 911 | if (dblHitCor) { |
| 0bd4b00f | 912 | Double_t dblC = dblHitCor->GetBinContent(dblHitCor->FindBin(eta)); |
| 913 | if (dblC > 0) correction *= dblC; | |
| 7e4038b5 | 914 | } |
| 57522224 | 915 | // else { |
| 916 | // AliWarning(Form("Missing double hit correction for FMD%d%c",d,r)); | |
| 917 | // } | |
| 7e4038b5 | 918 | } |
| 7e4038b5 | 919 | return correction; |
| 920 | } | |
| 921 | ||
| 0bd4b00f | 922 | //_____________________________________________________________________ |
| 923 | TH1D* | |
| 924 | AliFMDDensityCalculator::GenerateAcceptanceCorrection(Char_t r) const | |
| 925 | { | |
| 7984e5f7 | 926 | // |
| 927 | // Generate the acceptance corrections | |
| 928 | // | |
| 929 | // Parameters: | |
| 930 | // r Ring to generate for | |
| 931 | // | |
| 932 | // Return: | |
| 933 | // Newly allocated histogram of acceptance corrections | |
| 934 | // | |
| 6ab100ec | 935 | DGUARD(fDebug, 3, "Make acceptance correction in FMD density calculator"); |
| 0bd4b00f | 936 | const Double_t ic1[] = { 4.9895, 15.3560 }; |
| 937 | const Double_t ic2[] = { 1.8007, 17.2000 }; | |
| 938 | const Double_t oc1[] = { 4.2231, 26.6638 }; | |
| 939 | const Double_t oc2[] = { 1.8357, 27.9500 }; | |
| 940 | const Double_t* c1 = (r == 'I' || r == 'i' ? ic1 : oc1); | |
| 941 | const Double_t* c2 = (r == 'I' || r == 'i' ? ic2 : oc2); | |
| 942 | Double_t minR = (r == 'I' || r == 'i' ? 4.5213 : 15.4); | |
| 943 | Double_t maxR = (r == 'I' || r == 'i' ? 17.2 : 28.0); | |
| 944 | Int_t nStrips = (r == 'I' || r == 'i' ? 512 : 256); | |
| 945 | Int_t nSec = (r == 'I' || r == 'i' ? 20 : 40); | |
| 946 | Float_t basearc = 2 * TMath::Pi() / nSec; | |
| 947 | Double_t rad = maxR - minR; | |
| 948 | Float_t segment = rad / nStrips; | |
| 949 | Float_t cr = TMath::Sqrt(c1[0]*c1[0]+c1[1]*c1[1]); | |
| 950 | ||
| 951 | // Numbers used to find end-point of strip. | |
| 952 | // (See http://mathworld.wolfram.com/Circle-LineIntersection.html) | |
| 953 | Float_t D = c1[0] * c2[1] - c1[1] * c2[0]; | |
| 954 | Float_t dx = c2[0] - c1[0]; | |
| 955 | Float_t dy = c2[1] - c1[1]; | |
| 956 | Float_t dr = TMath::Sqrt(dx*dx+dy*dy); | |
| 957 | ||
| 958 | TH1D* ret = new TH1D(Form("acc%c", r), | |
| 959 | Form("Acceptance correction for FMDx%c", r), | |
| 960 | nStrips, -.5, nStrips-.5); | |
| 961 | ret->SetXTitle("Strip"); | |
| 962 | ret->SetYTitle("#varphi acceptance"); | |
| 963 | ret->SetDirectory(0); | |
| 964 | ret->SetFillColor(r == 'I' || r == 'i' ? kRed+1 : kBlue+1); | |
| 965 | ret->SetFillStyle(3001); | |
| 966 | ||
| 967 | for (Int_t t = 0; t < nStrips; t++) { | |
| 968 | Float_t radius = minR + t * segment; | |
| 969 | ||
| 970 | // If the radius of the strip is smaller than the radius corresponding | |
| 971 | // to the first corner we have a full strip length | |
| 972 | if (radius <= cr) { | |
| 973 | ret->SetBinContent(t+1, 1); | |
| 974 | continue; | |
| 975 | } | |
| 976 | ||
| 977 | // Next, we should find the end-point of the strip - that is, | |
| 978 | // the coordinates where the line from c1 to c2 intersects a circle | |
| 979 | // with radius given by the strip. | |
| 980 | // (See http://mathworld.wolfram.com/Circle-LineIntersection.html) | |
| 981 | // Calculate the determinant | |
| 982 | Float_t det = radius * radius * dr * dr - D*D; | |
| 983 | ||
| 984 | if (det <= 0) { | |
| 985 | // <0 means No intersection | |
| 986 | // =0 means Exactly tangent | |
| 987 | ret->SetBinContent(t+1, 1); | |
| 988 | continue; | |
| 989 | } | |
| 990 | ||
| 991 | // Calculate end-point and the corresponding opening angle | |
| 992 | Float_t x = (+D * dy + dx * TMath::Sqrt(det)) / dr / dr; | |
| 993 | Float_t y = (-D * dx + dy * TMath::Sqrt(det)) / dr / dr; | |
| 994 | Float_t th = TMath::ATan2(x, y); | |
| 995 | ||
| 996 | ret->SetBinContent(t+1, th / basearc); | |
| 997 | } | |
| 998 | return ret; | |
| 999 | } | |
| 7e4038b5 | 1000 | |
| 1001 | //_____________________________________________________________________ | |
| 1002 | Float_t | |
| 1003 | AliFMDDensityCalculator::AcceptanceCorrection(Char_t r, UShort_t t) const | |
| 1004 | { | |
| 7984e5f7 | 1005 | // |
| 1006 | // Get the acceptance correction for strip @a t in an ring of type @a r | |
| 1007 | // | |
| 1008 | // Parameters: | |
| 1009 | // r Ring type ('I' or 'O') | |
| 1010 | // t Strip number | |
| 1011 | // | |
| 1012 | // Return: | |
| 1013 | // Inverse acceptance correction | |
| 1014 | // | |
| 0bd4b00f | 1015 | TH1D* acc = (r == 'I' || r == 'i' ? fAccI : fAccO); |
| 1016 | return acc->GetBinContent(t+1); | |
| 7e4038b5 | 1017 | } |
| 1018 | ||
| 1019 | //____________________________________________________________________ | |
| 1020 | void | |
| 5934a3e3 | 1021 | AliFMDDensityCalculator::Terminate(const TList* dir, TList* output, |
| 1022 | Int_t nEvents) | |
| 7e4038b5 | 1023 | { |
| 7984e5f7 | 1024 | // |
| 1025 | // Scale the histograms to the total number of events | |
| 1026 | // | |
| 1027 | // Parameters: | |
| 1028 | // dir where to put the output | |
| 1029 | // nEvents Number of events | |
| 1030 | // | |
| 6ab100ec | 1031 | DGUARD(fDebug, 1, "Scale histograms in FMD density calculator"); |
| 7e4038b5 | 1032 | if (nEvents <= 0) return; |
| 9d99b0dd | 1033 | TList* d = static_cast<TList*>(dir->FindObject(GetName())); |
| 1034 | if (!d) return; | |
| 7e4038b5 | 1035 | |
| 5934a3e3 | 1036 | TList* out = new TList; |
| 1037 | out->SetName(d->GetName()); | |
| 1038 | out->SetOwner(); | |
| 1039 | ||
| 7e4038b5 | 1040 | TIter next(&fRingHistos); |
| 1041 | RingHistos* o = 0; | |
| 5bb5d1f6 | 1042 | THStack* sums = new THStack("sums", "sums of ring signals"); |
| 1043 | while ((o = static_cast<RingHistos*>(next()))) { | |
| 5934a3e3 | 1044 | o->Terminate(d, nEvents); |
| 1045 | if (!o->fDensity) { | |
| 1046 | Warning("Terminate", "No density in %s", o->GetName()); | |
| 1047 | continue; | |
| 1048 | } | |
| 5bb5d1f6 | 1049 | TH1D* sum = o->fDensity->ProjectionX(o->GetName(), 1, |
| 1050 | o->fDensity->GetNbinsY(),"e"); | |
| 1051 | sum->Scale(1., "width"); | |
| 1052 | sum->SetTitle(o->GetName()); | |
| 1053 | sum->SetDirectory(0); | |
| 1054 | sum->SetYTitle("#sum N_{ch,incl}"); | |
| 1055 | sums->Add(sum); | |
| 1056 | } | |
| 5934a3e3 | 1057 | out->Add(sums); |
| 1058 | output->Add(out); | |
| 7e4038b5 | 1059 | } |
| 1060 | ||
| 1061 | //____________________________________________________________________ | |
| 1062 | void | |
| 5934a3e3 | 1063 | AliFMDDensityCalculator::CreateOutputObjects(TList* dir) |
| 7e4038b5 | 1064 | { |
| 7984e5f7 | 1065 | // |
| 1066 | // Output diagnostic histograms to directory | |
| 1067 | // | |
| 1068 | // Parameters: | |
| 1069 | // dir List to write in | |
| 1070 | // | |
| 6ab100ec | 1071 | DGUARD(fDebug, 1, "Define output FMD density calculator"); |
| 7e4038b5 | 1072 | TList* d = new TList; |
| 5bb5d1f6 | 1073 | d->SetOwner(); |
| 7e4038b5 | 1074 | d->SetName(GetName()); |
| 1075 | dir->Add(d); | |
| dd497217 | 1076 | d->Add(fWeightedSum); |
| 1077 | d->Add(fSumOfWeights); | |
| 1078 | d->Add(fCorrections); | |
| 0bd4b00f | 1079 | d->Add(fAccI); |
| 1080 | d->Add(fAccO); | |
| 5bb5d1f6 | 1081 | d->Add(fMaxWeights); |
| 1082 | d->Add(fLowCuts); | |
| 0bd4b00f | 1083 | |
| bfab35d9 | 1084 | TParameter<int>* nFiles = new TParameter<int>("nFiles", 1); |
| 1085 | nFiles->SetMergeMode('+'); | |
| 1086 | ||
| d2638bb7 | 1087 | // d->Add(sigma); |
| 77f97e3f | 1088 | d->Add(AliForwardUtil::MakeParameter("maxParticle", fMaxParticles)); |
| 7095962e | 1089 | d->Add(AliForwardUtil::MakeParameter("minQuality", fMinQuality)); |
| 77f97e3f CHC |
1090 | d->Add(AliForwardUtil::MakeParameter("method", fUsePoisson)); |
| 1091 | d->Add(AliForwardUtil::MakeParameter("phiAcceptance",fUsePhiAcceptance)); | |
| 1092 | d->Add(AliForwardUtil::MakeParameter("etaLumping", fEtaLumping)); | |
| 1093 | d->Add(AliForwardUtil::MakeParameter("phiLumping", fPhiLumping)); | |
| 77f97e3f CHC |
1094 | d->Add(AliForwardUtil::MakeParameter("recalcPhi", fRecalculatePhi)); |
| 1095 | d->Add(AliForwardUtil::MakeParameter("maxOutliers", fMaxOutliers)); | |
| 1096 | d->Add(AliForwardUtil::MakeParameter("outlierCut", fOutlierCut)); | |
| bfab35d9 | 1097 | d->Add(nFiles); |
| d2638bb7 | 1098 | // d->Add(nxi); |
| 5ca83fee | 1099 | fCuts.Output(d,"lCuts"); |
| f7cfc454 | 1100 | |
| 7e4038b5 | 1101 | TIter next(&fRingHistos); |
| 1102 | RingHistos* o = 0; | |
| 1103 | while ((o = static_cast<RingHistos*>(next()))) { | |
| e18cb8bd | 1104 | o->fPoisson.SetLumping(fEtaLumping, fPhiLumping); |
| 5934a3e3 | 1105 | o->CreateOutputObjects(d); |
| 7e4038b5 | 1106 | } |
| 1f7aa5c7 | 1107 | |
| 1108 | if (!fDoTiming) return; | |
| 1109 | ||
| 1110 | fHTiming = new TProfile("timing", "#LTt_{part}#GT", 8, .5, 8.5); | |
| 1111 | fHTiming->SetDirectory(0); | |
| 1112 | fHTiming->SetYTitle("#LTt_{part}#GT"); | |
| 1113 | fHTiming->SetXTitle("Part"); | |
| 1114 | fHTiming->SetFillColor(kRed+1); | |
| 1115 | fHTiming->SetFillStyle(3001); | |
| 1116 | fHTiming->SetMarkerStyle(20); | |
| 1117 | fHTiming->SetMarkerColor(kBlack); | |
| 1118 | fHTiming->SetLineColor(kBlack); | |
| 1119 | fHTiming->SetStats(0); | |
| 1120 | TAxis* xaxis = fHTiming->GetXaxis(); | |
| 1121 | xaxis->SetBinLabel(1, "Re-calculation of #eta"); | |
| 1122 | xaxis->SetBinLabel(2, "N_{particle}"); | |
| 1123 | xaxis->SetBinLabel(3, "Correction"); | |
| 1124 | xaxis->SetBinLabel(4, "Re-calculation of #phi"); | |
| 1125 | xaxis->SetBinLabel(5, "Copy to cache"); | |
| 1126 | xaxis->SetBinLabel(6, "Poisson calculation"); | |
| 1127 | xaxis->SetBinLabel(7, "Diagnostics"); | |
| 1128 | xaxis->SetBinLabel(8, "Total"); | |
| 1129 | d->Add(fHTiming); | |
| 7e4038b5 | 1130 | } |
| c8b1a7db | 1131 | #define PF(N,V,...) \ |
| 1132 | AliForwardUtil::PrintField(N,V, ## __VA_ARGS__) | |
| 1133 | #define PFB(N,FLAG) \ | |
| 1134 | do { \ | |
| 1135 | AliForwardUtil::PrintName(N); \ | |
| 1136 | std::cout << std::boolalpha << (FLAG) << std::noboolalpha << std::endl; \ | |
| 1137 | } while(false) | |
| 1138 | #define PFV(N,VALUE) \ | |
| 1139 | do { \ | |
| 1140 | AliForwardUtil::PrintName(N); \ | |
| 1141 | std::cout << (VALUE) << std::endl; } while(false) | |
| 1142 | ||
| 0bd4b00f | 1143 | //____________________________________________________________________ |
| 1144 | void | |
| 1174780f | 1145 | AliFMDDensityCalculator::Print(Option_t* option) const |
| 0bd4b00f | 1146 | { |
| 7984e5f7 | 1147 | // |
| 1148 | // Print information | |
| 1149 | // | |
| 1150 | // Parameters: | |
| 1151 | // option Not used | |
| 1152 | // | |
| c8b1a7db | 1153 | AliForwardUtil::PrintTask(*this); |
| 1154 | gROOT->IncreaseDirLevel(); | |
| 1155 | ||
| 1156 | TString phiM("none"); | |
| 5ca83fee | 1157 | switch (fUsePhiAcceptance) { |
| c8b1a7db | 1158 | case kPhiNoCorrect: phiM = "none"; break; |
| 1159 | case kPhiCorrectNch: phiM = "correct Nch"; break; | |
| 1160 | case kPhiCorrectELoss: phiM = "correct energy loss"; break; | |
| 5ca83fee | 1161 | } |
| c8b1a7db | 1162 | |
| 1163 | PFV("Max(particles)", fMaxParticles ); | |
| 7095962e | 1164 | PFV("Min(quality)", fMinQuality ); |
| 0ccdab7b | 1165 | PFB("Poisson method", fUsePoisson ); |
| c8b1a7db | 1166 | PFV("Eta lumping", fEtaLumping ); |
| 1167 | PFV("Phi lumping", fPhiLumping ); | |
| 0ccdab7b | 1168 | PFB("Recalculate phi", fRecalculatePhi ); |
| 1169 | PFB("Use phi acceptance", phiM); | |
| c8b1a7db | 1170 | PFV("Lower cut", ""); |
| d2638bb7 | 1171 | fCuts.Print(); |
| c8b1a7db | 1172 | |
| 1174780f | 1173 | TString opt(option); |
| 1174 | opt.ToLower(); | |
| c8b1a7db | 1175 | if (!opt.Contains("nomax")) { |
| 1176 | PFV("Max weights", ""); | |
| 1177 | ||
| 1178 | char ind[64]; | |
| 1179 | for (Int_t i = 0; i < gROOT->GetDirLevel(); i++) ind[i] = ' '; | |
| 1180 | ind[gROOT->GetDirLevel()] = '\0'; | |
| 1181 | for (UShort_t d=1; d<=3; d++) { | |
| 1182 | UShort_t nr = (d == 1 ? 1 : 2); | |
| 1183 | for (UShort_t q=0; q<nr; q++) { | |
| 1184 | ind[gROOT->GetDirLevel()] = ' '; | |
| 1185 | ind[gROOT->GetDirLevel()+1] = '\0'; | |
| 1186 | Char_t r = (q == 0 ? 'I' : 'O'); | |
| 1187 | std::cout << ind << " FMD" << d << r << ":"; | |
| 1188 | ind[gROOT->GetDirLevel()+1] = ' '; | |
| 1189 | ind[gROOT->GetDirLevel()+2] = '\0'; | |
| 1190 | ||
| 1191 | const TArrayI& a = (d == 1 ? fFMD1iMax : | |
| 1192 | (d == 2 ? (r == 'I' ? fFMD2iMax : fFMD2oMax) : | |
| 1193 | (r == 'I' ? fFMD3iMax : fFMD3oMax))); | |
| 1194 | Int_t j = 0; | |
| 1195 | for (Int_t i = 0; i < a.fN; i++) { | |
| 1196 | if (a.fArray[i] < 1) continue; | |
| 1197 | if (j % 6 == 0) std::cout << "\n " << ind; | |
| 1198 | j++; | |
| 1199 | std::cout << " " << std::setw(3) << i << ": " << a.fArray[i]; | |
| 1200 | } | |
| 1201 | std::cout << std::endl; | |
| 1174780f | 1202 | } |
| 1174780f | 1203 | } |
| 1204 | } | |
| c8b1a7db | 1205 | gROOT->DecreaseDirLevel(); |
| 0bd4b00f | 1206 | } |
| 7e4038b5 | 1207 | |
| 1208 | //==================================================================== | |
| 1209 | AliFMDDensityCalculator::RingHistos::RingHistos() | |
| 9d99b0dd | 1210 | : AliForwardUtil::RingHistos(), |
| 5ca83fee | 1211 | fList(0), |
| 8449e3e0 | 1212 | // fEvsN(0), |
| 1213 | // fEvsM(0), | |
| 1214 | // fEtaVsN(0), | |
| 1215 | // fEtaVsM(0), | |
| 0bd4b00f | 1216 | fCorr(0), |
| 9d05ffeb | 1217 | fDensity(0), |
| 1218 | fELossVsPoisson(0), | |
| 5ca83fee | 1219 | fDiffELossPoisson(0), |
| 77f97e3f CHC |
1220 | fELossVsPoissonOut(0), |
| 1221 | fDiffELossPoissonOut(0), | |
| 1222 | fOutliers(0), | |
| 821ffd28 | 1223 | fPoisson(), |
| f7cfc454 | 1224 | fELoss(0), |
| 1225 | fELossUsed(0), | |
| 5ca83fee | 1226 | fMultCut(0), |
| 1227 | fTotal(0), | |
| 1228 | fGood(0), | |
| 1229 | fPhiAcc(0), | |
| 1230 | fPhiBefore(0), | |
| 1231 | fPhiAfter(0) | |
| 7984e5f7 | 1232 | { |
| 1233 | // | |
| 1234 | // Default CTOR | |
| 1235 | // | |
| 1236 | } | |
| 7e4038b5 | 1237 | |
| 1238 | //____________________________________________________________________ | |
| 1239 | AliFMDDensityCalculator::RingHistos::RingHistos(UShort_t d, Char_t r) | |
| 9d99b0dd | 1240 | : AliForwardUtil::RingHistos(d,r), |
| 5ca83fee | 1241 | fList(0), |
| 8449e3e0 | 1242 | // fEvsN(0), |
| 1243 | // fEvsM(0), | |
| 1244 | // fEtaVsN(0), | |
| 1245 | // fEtaVsM(0), | |
| 0bd4b00f | 1246 | fCorr(0), |
| 9d05ffeb | 1247 | fDensity(0), |
| 1248 | fELossVsPoisson(0), | |
| 5ca83fee | 1249 | fDiffELossPoisson(0), |
| 77f97e3f CHC |
1250 | fELossVsPoissonOut(0), |
| 1251 | fDiffELossPoissonOut(0), | |
| 1252 | fOutliers(0), | |
| 821ffd28 | 1253 | fPoisson("ignored"), |
| d2638bb7 | 1254 | fELoss(0), |
| f7cfc454 | 1255 | fELossUsed(0), |
| 5ca83fee | 1256 | fMultCut(0), |
| 1257 | fTotal(0), | |
| 1258 | fGood(0), | |
| 1259 | fPhiAcc(0), | |
| 1260 | fPhiBefore(0), | |
| 1261 | fPhiAfter(0) | |
| 7e4038b5 | 1262 | { |
| 7984e5f7 | 1263 | // |
| 1264 | // Constructor | |
| 1265 | // | |
| 1266 | // Parameters: | |
| 1267 | // d detector | |
| 1268 | // r ring | |
| 1269 | // | |
| 8449e3e0 | 1270 | #if 0 |
| 5bb5d1f6 | 1271 | fEvsN = new TH2D("elossVsNnocorr", |
| 1272 | "#Delta E/#Delta E_{mip} vs uncorrected inclusive N_{ch}", | |
| 5ca83fee | 1273 | 250, -.5, 24.5, 251, -1.5, 24.5); |
| 7e4038b5 | 1274 | fEvsN->SetXTitle("#Delta E/#Delta E_{mip}"); |
| 1275 | fEvsN->SetYTitle("Inclusive N_{ch} (uncorrected)"); | |
| 1276 | fEvsN->Sumw2(); | |
| 1277 | fEvsN->SetDirectory(0); | |
| 5bb5d1f6 | 1278 | |
| 1279 | fEvsM = static_cast<TH2D*>(fEvsN->Clone("elossVsNcorr")); | |
| 1280 | fEvsM->SetTitle("#Delta E/#Delta E_{mip} vs corrected inclusive N_{ch}"); | |
| 7e4038b5 | 1281 | fEvsM->SetDirectory(0); |
| 1282 | ||
| 5bb5d1f6 | 1283 | fEtaVsN = new TProfile("etaVsNnocorr", |
| 1284 | "Average inclusive N_{ch} vs #eta (uncorrected)", | |
| 1285 | 200, -4, 6); | |
| 0bd4b00f | 1286 | fEtaVsN->SetXTitle("#eta"); |
| 1287 | fEtaVsN->SetYTitle("#LT N_{ch,incl}#GT (uncorrected)"); | |
| 1288 | fEtaVsN->SetDirectory(0); | |
| 1289 | fEtaVsN->SetLineColor(Color()); | |
| 1290 | fEtaVsN->SetFillColor(Color()); | |
| 5bb5d1f6 | 1291 | |
| 1292 | fEtaVsM = static_cast<TProfile*>(fEtaVsN->Clone("etaVsNcorr")); | |
| 1293 | fEtaVsM->SetTitle("Average inclusive N_{ch} vs #eta (corrected)"); | |
| 0bd4b00f | 1294 | fEtaVsM->SetYTitle("#LT N_{ch,incl}#GT (corrected)"); |
| 1295 | fEtaVsM->SetDirectory(0); | |
| 8449e3e0 | 1296 | #endif |
| 0bd4b00f | 1297 | |
| 5bb5d1f6 | 1298 | fCorr = new TProfile("corr", "Average correction", 200, -4, 6); |
| 0bd4b00f | 1299 | fCorr->SetXTitle("#eta"); |
| 1300 | fCorr->SetYTitle("#LT correction#GT"); | |
| 1301 | fCorr->SetDirectory(0); | |
| 1302 | fCorr->SetLineColor(Color()); | |
| 1303 | fCorr->SetFillColor(Color()); | |
| 1304 | ||
| 5bb5d1f6 | 1305 | fDensity = new TH2D("inclDensity", "Inclusive N_{ch} density", |
| 7e4038b5 | 1306 | 200, -4, 6, (r == 'I' || r == 'i' ? 20 : 40), |
| 1307 | 0, 2*TMath::Pi()); | |
| 1308 | fDensity->SetDirectory(0); | |
| 5ca83fee | 1309 | fDensity->Sumw2(); fDensity->SetMarkerColor(Color()); |
| 7e4038b5 | 1310 | fDensity->SetXTitle("#eta"); |
| 1311 | fDensity->SetYTitle("#phi [radians]"); | |
| 1312 | fDensity->SetZTitle("Inclusive N_{ch} density"); | |
| 9d05ffeb | 1313 | |
| 77f97e3f CHC |
1314 | // --- Create increasing sized bins -------------------------------- |
| 1315 | TArrayD bins; | |
| 1316 | // bins, lowest order, higest order, return array | |
| 1317 | const char* nchP = "N_{ch}^{Poisson}"; | |
| 1318 | const char* nchE = "N_{ch}^{#Delta}"; | |
| 1319 | AliForwardUtil::MakeLogScale(300, 0, 2, bins); | |
| 5bb5d1f6 | 1320 | fELossVsPoisson = new TH2D("elossVsPoisson", |
| 77f97e3f CHC |
1321 | "N_{ch} from energy loss vs from Poisson", |
| 1322 | bins.GetSize()-1, bins.GetArray(), | |
| 1323 | bins.GetSize()-1, bins.GetArray()); | |
| 9d05ffeb | 1324 | fELossVsPoisson->SetDirectory(0); |
| 77f97e3f CHC |
1325 | fELossVsPoisson->SetXTitle(nchE); |
| 1326 | fELossVsPoisson->SetYTitle(nchP); | |
| 9d05ffeb | 1327 | fELossVsPoisson->SetZTitle("Correlation"); |
| 77f97e3f CHC |
1328 | fELossVsPoissonOut = |
| 1329 | static_cast<TH2D*>(fELossVsPoisson | |
| 1330 | ->Clone(Form("%sOutlier", | |
| 1331 | fELossVsPoisson->GetName()))); | |
| 1332 | fELossVsPoissonOut->SetDirectory(0); | |
| 1333 | fELossVsPoissonOut->SetMarkerStyle(20); | |
| 1334 | fELossVsPoissonOut->SetMarkerSize(0.3); | |
| 1335 | fELossVsPoissonOut->SetMarkerColor(kBlack); | |
| 1336 | fELossVsPoissonOut->SetTitle(Form("%s for outliers", | |
| 1337 | fELossVsPoisson->GetTitle())); | |
| 9d05ffeb | 1338 | |
| 5ca83fee | 1339 | fDiffELossPoisson = new TH1D("diffElossPoisson", |
| 77f97e3f | 1340 | Form("(%s-%s)/%s", nchP, nchE, nchE), |
| 5ca83fee | 1341 | 100, -1, 1); |
| 1342 | fDiffELossPoisson->SetDirectory(0); | |
| 1343 | fDiffELossPoisson->SetXTitle(fDiffELossPoisson->GetTitle()); | |
| 1344 | fDiffELossPoisson->SetYTitle("Frequency"); | |
| 1345 | fDiffELossPoisson->SetMarkerColor(Color()); | |
| 1346 | fDiffELossPoisson->SetFillColor(Color()); | |
| bfab35d9 | 1347 | fDiffELossPoisson->SetFillStyle(3001); |
| 5ca83fee | 1348 | fDiffELossPoisson->Sumw2(); |
| 1349 | ||
| 77f97e3f CHC |
1350 | fDiffELossPoissonOut = |
| 1351 | static_cast<TH1D*>(fDiffELossPoisson | |
| 1352 | ->Clone(Form("%sOutlier",fDiffELossPoisson->GetName()))); | |
| 1353 | fDiffELossPoissonOut->SetDirectory(0); | |
| 1354 | fDiffELossPoissonOut->SetTitle(Form("%s for outliers", | |
| 1355 | fDiffELossPoisson->GetTitle())); | |
| 1356 | fDiffELossPoissonOut->SetMarkerColor(Color()-2); | |
| 1357 | fDiffELossPoissonOut->SetFillColor(Color()-2); | |
| 1358 | fDiffELossPoissonOut->SetFillStyle(3002); | |
| 1359 | ||
| 1360 | fOutliers = new TH1D("outliers", "Fraction of outliers", 100, 0, 1); | |
| 1361 | fOutliers->SetDirectory(0); | |
| 1362 | fOutliers->SetXTitle("N_{outlier}/(N_{outlier}+N_{inside})"); | |
| 1363 | fOutliers->SetYTitle("#sum_{events}#sum_{bins}"); | |
| 1364 | fOutliers->SetFillColor(Color()); | |
| 1365 | fOutliers->SetFillStyle(3001); | |
| 1366 | fOutliers->SetLineColor(kBlack); | |
| 1367 | ||
| f7cfc454 | 1368 | fELoss = new TH1D("eloss", "#Delta/#Delta_{mip} in all strips", |
| 8449e3e0 | 1369 | 640, -1, 15); |
| f7cfc454 | 1370 | fELoss->SetXTitle("#Delta/#Delta_{mip} (selected)"); |
| 1371 | fELoss->SetYTitle("P(#Delta/#Delta_{mip})"); | |
| 1372 | fELoss->SetFillColor(Color()-2); | |
| 1373 | fELoss->SetFillStyle(3003); | |
| 1374 | fELoss->SetLineColor(kBlack); | |
| 1375 | fELoss->SetLineStyle(2); | |
| 77f97e3f | 1376 | fELoss->SetLineWidth(1); |
| f7cfc454 | 1377 | fELoss->SetDirectory(0); |
| 1378 | ||
| 1379 | fELossUsed = static_cast<TH1D*>(fELoss->Clone("elossUsed")); | |
| 1380 | fELossUsed->SetTitle("#Delta/#Delta_{mip} in used strips"); | |
| 1381 | fELossUsed->SetFillStyle(3002); | |
| 1382 | fELossUsed->SetLineStyle(1); | |
| 1383 | fELossUsed->SetDirectory(0); | |
| 5ca83fee | 1384 | |
| 1385 | fPhiBefore = new TH1D("phiBefore", "#phi distribution (before recalc)", | |
| 1386 | (r == 'I' || r == 'i' ? 20 : 40), 0, 2*TMath::Pi()); | |
| 1387 | fPhiBefore->SetDirectory(0); | |
| 1388 | fPhiBefore->SetXTitle("#phi"); | |
| 1389 | fPhiBefore->SetYTitle("Events"); | |
| 1390 | fPhiBefore->SetMarkerColor(Color()); | |
| 1391 | fPhiBefore->SetLineColor(Color()); | |
| 1392 | fPhiBefore->SetFillColor(Color()); | |
| bfab35d9 | 1393 | fPhiBefore->SetFillStyle(3001); |
| 5ca83fee | 1394 | fPhiBefore->SetMarkerStyle(20); |
| 1395 | ||
| 1396 | fPhiAfter = static_cast<TH1D*>(fPhiBefore->Clone("phiAfter")); | |
| 1397 | fPhiAfter->SetTitle("#phi distribution (after re-calc)"); | |
| 1398 | fPhiAfter->SetDirectory(0); | |
| f7cfc454 | 1399 | |
| 7e4038b5 | 1400 | } |
| 1401 | //____________________________________________________________________ | |
| 1402 | AliFMDDensityCalculator::RingHistos::RingHistos(const RingHistos& o) | |
| 5ca83fee | 1403 | : AliForwardUtil::RingHistos(o), |
| 1404 | fList(o.fList), | |
| 8449e3e0 | 1405 | // fEvsN(o.fEvsN), |
| 1406 | // fEvsM(o.fEvsM), | |
| 1407 | // fEtaVsN(o.fEtaVsN), | |
| 1408 | // fEtaVsM(o.fEtaVsM), | |
| 0bd4b00f | 1409 | fCorr(o.fCorr), |
| 9d05ffeb | 1410 | fDensity(o.fDensity), |
| 1411 | fELossVsPoisson(o.fELossVsPoisson), | |
| 5ca83fee | 1412 | fDiffELossPoisson(o.fDiffELossPoisson), |
| 77f97e3f CHC |
1413 | fELossVsPoissonOut(o.fELossVsPoissonOut), |
| 1414 | fDiffELossPoissonOut(o.fDiffELossPoissonOut), | |
| 1415 | fOutliers(o.fOutliers), | |
| 821ffd28 | 1416 | fPoisson(o.fPoisson), |
| f7cfc454 | 1417 | fELoss(o.fELoss), |
| 1418 | fELossUsed(o.fELossUsed), | |
| 5ca83fee | 1419 | fMultCut(o.fMultCut), |
| 1420 | fTotal(o.fTotal), | |
| 1421 | fGood(o.fGood), | |
| 1422 | fPhiAcc(o.fPhiAcc), | |
| 1423 | fPhiBefore(o.fPhiBefore), | |
| 1424 | fPhiAfter(o.fPhiAfter) | |
| 7984e5f7 | 1425 | { |
| 1426 | // | |
| 1427 | // Copy constructor | |
| 1428 | // | |
| 1429 | // Parameters: | |
| 1430 | // o Object to copy from | |
| 1431 | // | |
| 1432 | } | |
| 7e4038b5 | 1433 | |
| 1434 | //____________________________________________________________________ | |
| 1435 | AliFMDDensityCalculator::RingHistos& | |
| 1436 | AliFMDDensityCalculator::RingHistos::operator=(const RingHistos& o) | |
| 1437 | { | |
| 7984e5f7 | 1438 | // |
| 1439 | // Assignment operator | |
| 1440 | // | |
| 1441 | // Parameters: | |
| 1442 | // o Object to assign from | |
| 1443 | // | |
| 1444 | // Return: | |
| 1445 | // Reference to this | |
| 1446 | // | |
| d015ecfe | 1447 | if (&o == this) return *this; |
| 9d99b0dd | 1448 | AliForwardUtil::RingHistos::operator=(o); |
| 7e4038b5 | 1449 | |
| 8449e3e0 | 1450 | // if (fEvsN) delete fEvsN; |
| 1451 | // if (fEvsM) delete fEvsM; | |
| 1452 | // if (fEtaVsN) delete fEtaVsN; | |
| 1453 | // if (fEtaVsM) delete fEtaVsM; | |
| 5ca83fee | 1454 | if (fCorr) delete fCorr; |
| 1455 | if (fDensity) delete fDensity; | |
| 1456 | if (fELossVsPoisson) delete fELossVsPoisson; | |
| 1457 | if (fDiffELossPoisson) delete fDiffELossPoisson; | |
| 1458 | if (fTotal) delete fTotal; | |
| 1459 | if (fGood) delete fGood; | |
| 1460 | if (fPhiAcc) delete fPhiAcc; | |
| 1461 | if (fPhiBefore) delete fPhiBefore; | |
| 1462 | if (fPhiAfter) delete fPhiAfter; | |
| 1463 | ||
| 8449e3e0 | 1464 | // fEvsN = static_cast<TH2D*>(o.fEvsN->Clone()); |
| 1465 | // fEvsM = static_cast<TH2D*>(o.fEvsM->Clone()); | |
| 1466 | // fEtaVsN = static_cast<TProfile*>(o.fEtaVsN->Clone()); | |
| 1467 | // fEtaVsM = static_cast<TProfile*>(o.fEtaVsM->Clone()); | |
| 5ca83fee | 1468 | fCorr = static_cast<TProfile*>(o.fCorr->Clone()); |
| 1469 | fDensity = static_cast<TH2D*>(o.fDensity->Clone()); | |
| 1470 | fELossVsPoisson = static_cast<TH2D*>(o.fELossVsPoisson->Clone()); | |
| 1471 | fDiffELossPoisson = static_cast<TH1D*>(o.fDiffELossPoisson->Clone()); | |
| 77f97e3f CHC |
1472 | fELossVsPoissonOut = static_cast<TH2D*>(o.fELossVsPoisson->Clone()); |
| 1473 | fDiffELossPoissonOut = static_cast<TH1D*>(o.fDiffELossPoisson->Clone()); | |
| 1474 | fOutliers = static_cast<TH1D*>(o.fOutliers->Clone()); | |
| 5ca83fee | 1475 | fPoisson = o.fPoisson; |
| 1476 | fELoss = static_cast<TH1D*>(o.fELoss->Clone()); | |
| 1477 | fELossUsed = static_cast<TH1D*>(o.fELossUsed->Clone()); | |
| 1478 | fTotal = static_cast<TH1D*>(o.fTotal->Clone()); | |
| 1479 | fGood = static_cast<TH1D*>(o.fGood->Clone()); | |
| 1480 | fPhiAcc = static_cast<TH2D*>(o.fPhiAcc->Clone()); | |
| 1481 | fPhiBefore = static_cast<TH1D*>(o.fPhiBefore->Clone()); | |
| 1482 | fPhiAfter = static_cast<TH1D*>(o.fPhiAfter->Clone()); | |
| 7e4038b5 | 1483 | return *this; |
| 1484 | } | |
| 1485 | //____________________________________________________________________ | |
| 1486 | AliFMDDensityCalculator::RingHistos::~RingHistos() | |
| 1487 | { | |
| 7984e5f7 | 1488 | // |
| 1489 | // Destructor | |
| 1490 | // | |
| 9d05ffeb | 1491 | } |
| 1492 | ||
| e308a636 | 1493 | |
| 1494 | //____________________________________________________________________ | |
| 9d05ffeb | 1495 | void |
| 5934a3e3 | 1496 | AliFMDDensityCalculator::RingHistos::SetupForData(const TAxis& eAxis) |
| 9d05ffeb | 1497 | { |
| 5ca83fee | 1498 | // Initialize |
| 1499 | // This is called on first event | |
| e18cb8bd | 1500 | fPoisson.Init(-1,-1); |
| 5ca83fee | 1501 | fTotal = new TH1D("total", "Total # of strips per #eta", |
| 1502 | eAxis.GetNbins(), eAxis.GetXmin(), eAxis.GetXmax()); | |
| 1503 | fTotal->SetDirectory(0); | |
| 1504 | fTotal->SetXTitle("#eta"); | |
| 1505 | fTotal->SetYTitle("# of strips"); | |
| 1506 | fGood = static_cast<TH1D*>(fTotal->Clone("good")); | |
| 1507 | fGood->SetTitle("# of good strips per #eta"); | |
| 1508 | fGood->SetDirectory(0); | |
| 1509 | ||
| 1510 | fPhiAcc = new TH2D("phiAcc", "#phi acceptance vs Ip_{z}", | |
| 1511 | eAxis.GetNbins(), eAxis.GetXmin(), eAxis.GetXmax(), | |
| 1512 | 10, -10, 10); | |
| 1513 | fPhiAcc->SetXTitle("#eta"); | |
| 1514 | fPhiAcc->SetYTitle("v_{z} [cm]"); | |
| 1515 | fPhiAcc->SetZTitle("#phi acceptance"); | |
| 1516 | fPhiAcc->SetDirectory(0); | |
| 1517 | ||
| 1518 | if (fList) fList->Add(fPhiAcc); | |
| 7e4038b5 | 1519 | } |
| 1520 | ||
| 1521 | //____________________________________________________________________ | |
| 1522 | void | |
| 5934a3e3 | 1523 | AliFMDDensityCalculator::RingHistos::CreateOutputObjects(TList* dir) |
| 7e4038b5 | 1524 | { |
| 7984e5f7 | 1525 | // |
| 5ca83fee | 1526 | // Make output. This is called as part of SlaveBegin |
| 7984e5f7 | 1527 | // |
| 1528 | // Parameters: | |
| 1529 | // dir Where to put it | |
| 1530 | // | |
| 9d99b0dd | 1531 | TList* d = DefineOutputList(dir); |
| 8449e3e0 | 1532 | // d->Add(fEvsN); |
| 1533 | // d->Add(fEvsM); | |
| 1534 | // d->Add(fEtaVsN); | |
| 1535 | // d->Add(fEtaVsM); | |
| 0bd4b00f | 1536 | d->Add(fCorr); |
| 7e4038b5 | 1537 | d->Add(fDensity); |
| 9d05ffeb | 1538 | d->Add(fELossVsPoisson); |
| 77f97e3f | 1539 | d->Add(fELossVsPoissonOut); |
| 5ca83fee | 1540 | d->Add(fDiffELossPoisson); |
| 77f97e3f CHC |
1541 | d->Add(fDiffELossPoissonOut); |
| 1542 | d->Add(fOutliers); | |
| 821ffd28 | 1543 | fPoisson.Output(d); |
| d23503ee | 1544 | fPoisson.GetOccupancy()->SetFillColor(Color()); |
| 1545 | fPoisson.GetMean()->SetFillColor(Color()); | |
| 1546 | fPoisson.GetOccupancy()->SetFillColor(Color()); | |
| f7cfc454 | 1547 | d->Add(fELoss); |
| 1548 | d->Add(fELossUsed); | |
| 5ca83fee | 1549 | d->Add(fPhiBefore); |
| 1550 | d->Add(fPhiAfter); | |
| d23503ee | 1551 | |
| 77f97e3f CHC |
1552 | TAxis x(NStrip(), -.5, NStrip()-.5); |
| 1553 | TAxis y(NSector(), -.5, NSector()-.5); | |
| d23503ee | 1554 | x.SetTitle("strip"); |
| 1555 | y.SetTitle("sector"); | |
| 1556 | fPoisson.Define(x, y); | |
| 1557 | ||
| 241cca4d | 1558 | d->Add(AliForwardUtil::MakeParameter("cut", fMultCut)); |
| 5ca83fee | 1559 | fList = d; |
| 9d99b0dd | 1560 | } |
| 1561 | ||
| 1562 | //____________________________________________________________________ | |
| 1563 | void | |
| 5934a3e3 | 1564 | AliFMDDensityCalculator::RingHistos::Terminate(TList* dir, Int_t nEvents) |
| 9d99b0dd | 1565 | { |
| 7984e5f7 | 1566 | // |
| 1567 | // Scale the histograms to the total number of events | |
| 1568 | // | |
| 1569 | // Parameters: | |
| 1570 | // dir Where the output is | |
| 1571 | // nEvents Number of events | |
| 1572 | // | |
| 9d99b0dd | 1573 | TList* l = GetOutputList(dir); |
| 1574 | if (!l) return; | |
| 1575 | ||
| 5934a3e3 | 1576 | TH2D* density = static_cast<TH2D*>(GetOutputHist(l,"inclDensity")); |
| 9d99b0dd | 1577 | if (density) density->Scale(1./nEvents); |
| 5934a3e3 | 1578 | fDensity = density; |
| 7e4038b5 | 1579 | } |
| 1580 | ||
| 1581 | //____________________________________________________________________ | |
| 1582 | // | |
| 1583 | // EOF | |
| 1584 | // | |
| 1585 | ||
| 1586 | ||
| 1587 |