1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
18 #include "AlidNdEtaCorrection.h"
25 #include <TDirectory.h>
26 #include <AliCorrection.h>
27 #include <AliCorrectionMatrix2D.h>
28 #include <AliCorrectionMatrix3D.h>
30 //____________________________________________________________________
31 ClassImp(AlidNdEtaCorrection)
33 //____________________________________________________________________
34 AlidNdEtaCorrection::AlidNdEtaCorrection()
36 fTrack2ParticleCorrection(0),
37 fVertexRecoCorrection(0),
38 fTriggerBiasCorrectionMBToINEL(0),
39 fTriggerBiasCorrectionMBToNSD(0),
40 fTriggerBiasCorrectionMBToND(0),
41 fTriggerBiasCorrectionMBToOnePart(0)
43 // default constructor
46 //____________________________________________________________________
47 AlidNdEtaCorrection::AlidNdEtaCorrection(const Char_t* name, const Char_t* title, AliPWG0Helper::AnalysisMode analysis)
48 : TNamed(name, title),
49 fTrack2ParticleCorrection(0),
50 fVertexRecoCorrection(0),
51 fTriggerBiasCorrectionMBToINEL(0),
52 fTriggerBiasCorrectionMBToNSD(0),
53 fTriggerBiasCorrectionMBToND(0),
54 fTriggerBiasCorrectionMBToOnePart(0)
60 fTrack2ParticleCorrection = new AliCorrection("Track2Particle", "Track2Particle", analysis);
61 fVertexRecoCorrection = new AliCorrection("VertexReconstruction", "VertexReconstruction", analysis);
63 fTriggerBiasCorrectionMBToINEL = new AliCorrection("TriggerBias_MBToINEL", "TriggerBias_MBToINEL", analysis);
64 fTriggerBiasCorrectionMBToNSD = new AliCorrection("TriggerBias_MBToNSD", "TriggerBias_MBToNSD", analysis);
65 fTriggerBiasCorrectionMBToND = new AliCorrection("TriggerBias_MBToND", "TriggerBias_MBToND", analysis);
66 fTriggerBiasCorrectionMBToOnePart = new AliCorrection("TriggerBias_MBToOnePart", "TriggerBias_MBToOnePart", analysis);
69 //____________________________________________________________________
70 AlidNdEtaCorrection::~AlidNdEtaCorrection()
74 if (fTrack2ParticleCorrection) {
75 delete fTrack2ParticleCorrection;
76 fTrack2ParticleCorrection = 0;
79 if (fVertexRecoCorrection) {
80 delete fVertexRecoCorrection;
81 fVertexRecoCorrection = 0;
84 if (fTriggerBiasCorrectionMBToINEL) {
85 delete fTriggerBiasCorrectionMBToINEL;
86 fTriggerBiasCorrectionMBToINEL = 0;
89 if (fTriggerBiasCorrectionMBToNSD) {
90 delete fTriggerBiasCorrectionMBToNSD;
91 fTriggerBiasCorrectionMBToNSD = 0;
94 if (fTriggerBiasCorrectionMBToND) {
95 delete fTriggerBiasCorrectionMBToND;
96 fTriggerBiasCorrectionMBToND = 0;
99 if (fTriggerBiasCorrectionMBToOnePart) {
100 delete fTriggerBiasCorrectionMBToOnePart;
101 fTriggerBiasCorrectionMBToOnePart = 0;
105 //____________________________________________________________________
107 AlidNdEtaCorrection::Finish() {
111 // divide the histograms in the AliCorrectionMatrix2D objects to get the corrections
113 fTrack2ParticleCorrection->Divide();
114 fVertexRecoCorrection->Divide();
115 fTriggerBiasCorrectionMBToINEL->Divide();
116 fTriggerBiasCorrectionMBToNSD->Divide();
117 fTriggerBiasCorrectionMBToND->Divide();
118 fTriggerBiasCorrectionMBToOnePart->Divide();
121 //____________________________________________________________________
122 Long64_t AlidNdEtaCorrection::Merge(TCollection* list)
124 // Merge a list of dNdEtaCorrection objects with this (needed for
126 // Returns the number of merged objects (including this).
134 TIterator* iter = list->MakeIterator();
137 // collections of measured and generated histograms
138 TList* collectionNtrackToNparticle = new TList;
139 TList* collectionVertexReco = new TList;
140 TList* collectionTriggerBiasMBToINEL = new TList;
141 TList* collectionTriggerBiasMBToNSD = new TList;
142 TList* collectionTriggerBiasMBToND = new TList;
143 TList* collectionTriggerBiasMBToOnePart = new TList;
146 while ((obj = iter->Next())) {
148 AlidNdEtaCorrection* entry = dynamic_cast<AlidNdEtaCorrection*> (obj);
152 collectionNtrackToNparticle ->Add(entry->fTrack2ParticleCorrection);
153 collectionVertexReco ->Add(entry->fVertexRecoCorrection);
154 collectionTriggerBiasMBToINEL->Add(entry->fTriggerBiasCorrectionMBToINEL);
155 collectionTriggerBiasMBToNSD ->Add(entry->fTriggerBiasCorrectionMBToNSD);
156 collectionTriggerBiasMBToND ->Add(entry->fTriggerBiasCorrectionMBToND);
157 collectionTriggerBiasMBToOnePart ->Add(entry->fTriggerBiasCorrectionMBToOnePart);
161 fTrack2ParticleCorrection ->Merge(collectionNtrackToNparticle);
162 fVertexRecoCorrection ->Merge(collectionVertexReco);
163 fTriggerBiasCorrectionMBToINEL ->Merge(collectionTriggerBiasMBToINEL);
164 fTriggerBiasCorrectionMBToNSD ->Merge(collectionTriggerBiasMBToNSD);
165 fTriggerBiasCorrectionMBToND ->Merge(collectionTriggerBiasMBToND);
166 fTriggerBiasCorrectionMBToOnePart->Merge(collectionTriggerBiasMBToOnePart);
168 delete collectionNtrackToNparticle;
169 delete collectionVertexReco;
170 delete collectionTriggerBiasMBToINEL;
171 delete collectionTriggerBiasMBToNSD;
172 delete collectionTriggerBiasMBToND;
173 delete collectionTriggerBiasMBToOnePart;
178 //____________________________________________________________________
179 void AlidNdEtaCorrection::Add(AlidNdEtaCorrection* aCorrectionsToAdd, Float_t c) {
181 // adds the measured and generated of aCorrectionsToAdd to measured and generated
182 // of all corrections in this
184 fTrack2ParticleCorrection ->Add(aCorrectionsToAdd->GetTrack2ParticleCorrection() ,c);
185 fVertexRecoCorrection ->Add(aCorrectionsToAdd->GetVertexRecoCorrection() ,c);
186 fTriggerBiasCorrectionMBToINEL ->Add(aCorrectionsToAdd->GetTriggerBiasCorrectionINEL(),c);
187 fTriggerBiasCorrectionMBToNSD ->Add(aCorrectionsToAdd->GetTriggerBiasCorrectionNSD() ,c);
188 fTriggerBiasCorrectionMBToND ->Add(aCorrectionsToAdd->GetTriggerBiasCorrectionND() ,c);
189 fTriggerBiasCorrectionMBToOnePart ->Add(aCorrectionsToAdd->GetTriggerBiasCorrectionOnePart() ,c);
192 //____________________________________________________________________
193 void AlidNdEtaCorrection::Scale(Float_t c)
196 // scales all contained corrections
199 fTrack2ParticleCorrection ->Scale(c);
200 fVertexRecoCorrection ->Scale(c);
201 fTriggerBiasCorrectionMBToINEL ->Scale(c);
202 fTriggerBiasCorrectionMBToNSD ->Scale(c);
203 fTriggerBiasCorrectionMBToND ->Scale(c);
204 fTriggerBiasCorrectionMBToOnePart ->Scale(c);
207 //____________________________________________________________________
208 void AlidNdEtaCorrection::Reset(void) {
210 // reset all corrections
213 fTrack2ParticleCorrection ->Reset();
214 fVertexRecoCorrection ->Reset();
215 fTriggerBiasCorrectionMBToINEL ->Reset();
216 fTriggerBiasCorrectionMBToNSD ->Reset();
217 fTriggerBiasCorrectionMBToND ->Reset();
218 fTriggerBiasCorrectionMBToOnePart ->Reset();
223 //____________________________________________________________________
224 Bool_t AlidNdEtaCorrection::LoadHistograms(const Char_t* dir)
227 // loads the histograms
228 // if dir is empty a directory with the name of this object is taken (like in SaveHistogram)
234 if (!gDirectory->cd(dir))
237 fTrack2ParticleCorrection ->LoadHistograms();
238 fVertexRecoCorrection ->LoadHistograms();
239 fTriggerBiasCorrectionMBToINEL ->LoadHistograms();
240 fTriggerBiasCorrectionMBToNSD ->LoadHistograms();
241 fTriggerBiasCorrectionMBToND ->LoadHistograms();
242 fTriggerBiasCorrectionMBToOnePart ->LoadHistograms();
244 gDirectory->cd("..");
249 //____________________________________________________________________
250 void AlidNdEtaCorrection::SaveHistograms()
253 // save the histograms
256 gDirectory->mkdir(fName.Data());
257 gDirectory->cd(fName.Data());
259 fTrack2ParticleCorrection ->SaveHistograms();
260 fVertexRecoCorrection ->SaveHistograms();
261 fTriggerBiasCorrectionMBToINEL->SaveHistograms();
262 fTriggerBiasCorrectionMBToNSD ->SaveHistograms();
263 fTriggerBiasCorrectionMBToND ->SaveHistograms();
264 fTriggerBiasCorrectionMBToOnePart->SaveHistograms();
266 gDirectory->cd("..");
269 //____________________________________________________________________
270 void AlidNdEtaCorrection::DrawHistograms()
273 // call the draw histogram method of the corrections
276 fTrack2ParticleCorrection ->DrawHistograms();
277 fVertexRecoCorrection ->DrawHistograms();
278 fTriggerBiasCorrectionMBToINEL->DrawHistograms();
279 fTriggerBiasCorrectionMBToNSD ->DrawHistograms();
280 fTriggerBiasCorrectionMBToND ->DrawHistograms();
281 fTriggerBiasCorrectionMBToOnePart ->DrawHistograms();
284 //____________________________________________________________________
285 void AlidNdEtaCorrection::DrawOverview(const char* canvasName)
288 // call the DrawOverview histogram method of the corrections
291 fTrack2ParticleCorrection ->DrawOverview(canvasName);
292 fVertexRecoCorrection ->DrawOverview(canvasName);
293 fTriggerBiasCorrectionMBToINEL->DrawOverview(canvasName);
294 fTriggerBiasCorrectionMBToNSD ->DrawOverview(canvasName);
295 fTriggerBiasCorrectionMBToND ->DrawOverview(canvasName);
296 fTriggerBiasCorrectionMBToOnePart ->DrawOverview(canvasName);
299 //____________________________________________________________________
300 void AlidNdEtaCorrection::FillMCParticle(Float_t vtx, Float_t eta, Float_t pt, Bool_t trigger, Bool_t vertex, Int_t processType)
302 // fills a particle in the corrections
303 // it is filled in generated or measured depending of the flags
305 fTriggerBiasCorrectionMBToINEL->GetTrackCorrection()->FillGene(vtx, eta, pt);
307 if ((processType & AliPWG0Helper::kSD) == 0)
308 fTriggerBiasCorrectionMBToNSD->GetTrackCorrection()->FillGene(vtx, eta, pt);
310 if (processType & AliPWG0Helper::kND )
311 fTriggerBiasCorrectionMBToND->GetTrackCorrection()->FillGene(vtx, eta, pt);
313 if (processType & AliPWG0Helper::kOnePart)
314 fTriggerBiasCorrectionMBToOnePart->GetTrackCorrection()->FillGene(vtx, eta, pt);
319 fTriggerBiasCorrectionMBToINEL->GetTrackCorrection()->FillMeas(vtx, eta, pt);
320 fTriggerBiasCorrectionMBToNSD->GetTrackCorrection()->FillMeas(vtx, eta, pt);
321 fTriggerBiasCorrectionMBToND->GetTrackCorrection()->FillMeas(vtx, eta, pt);
322 fTriggerBiasCorrectionMBToOnePart->GetTrackCorrection()->FillMeas(vtx, eta, pt);
323 fVertexRecoCorrection->GetTrackCorrection()->FillGene(vtx, eta, pt);
328 fVertexRecoCorrection->GetTrackCorrection()->FillMeas(vtx, eta, pt);
329 fTrack2ParticleCorrection->GetTrackCorrection()->FillGene(vtx, eta, pt);
332 //____________________________________________________________________
333 void AlidNdEtaCorrection::FillTrackedParticle(Float_t vtx, Float_t eta, Float_t pt, Double_t weight)
335 // fills a tracked particle in the corrections
337 fTrack2ParticleCorrection->GetTrackCorrection()->FillMeas(vtx, eta, pt, weight);
340 //____________________________________________________________________
341 void AlidNdEtaCorrection::FillEvent(Float_t vtx, Float_t n, Bool_t trigger, Bool_t vertex, Int_t processType)
343 // fills an event int he correction
344 // it is filled in generated or measured depending of the flags
346 fTriggerBiasCorrectionMBToINEL->GetEventCorrection()->FillGene(vtx, n);
348 if ((processType & AliPWG0Helper::kSD) == 0)
349 fTriggerBiasCorrectionMBToNSD->GetEventCorrection()->FillGene(vtx, n);
351 if (processType & AliPWG0Helper::kND )
352 fTriggerBiasCorrectionMBToND->GetEventCorrection()->FillGene(vtx, n);
354 if (processType & AliPWG0Helper::kOnePart)
355 fTriggerBiasCorrectionMBToOnePart->GetEventCorrection()->FillGene(vtx, n);
360 fTriggerBiasCorrectionMBToINEL->GetEventCorrection()->FillMeas(vtx, n);
361 fTriggerBiasCorrectionMBToNSD->GetEventCorrection()->FillMeas(vtx, n);
362 fTriggerBiasCorrectionMBToND->GetEventCorrection()->FillMeas(vtx, n);
363 fTriggerBiasCorrectionMBToOnePart->GetEventCorrection()->FillMeas(vtx, n);
364 fVertexRecoCorrection->GetEventCorrection()->FillGene(vtx, n);
369 fVertexRecoCorrection->GetEventCorrection()->FillMeas(vtx, n);
372 //____________________________________________________________________
373 Float_t AlidNdEtaCorrection::GetMeasuredFraction(CorrectionType correctionType, Float_t ptCutOff, Float_t eta, Int_t vertexBegin, Int_t vertexEnd, Bool_t debug)
375 // calculates the fraction of particles measured (some are missed due to the pt cut off)
377 // uses the generated particle histogram from the correction passed, e.g. pass GetTrack2ParticleCorrection()
379 if (!GetCorrection(correctionType))
382 const TH3* generated = GetCorrection(correctionType)->GetTrackCorrection()->GetGeneratedHistogram();
384 // find eta borders, if eta is negative assume -0.8 ... 0.8
389 etaBegin = generated->GetYaxis()->FindBin(-0.8);
390 etaEnd = generated->GetYaxis()->FindBin(0.8);
394 etaBegin = generated->GetYaxis()->FindBin(eta);
398 if (vertexBegin == -1)
399 vertexBegin = generated->GetXaxis()->FindBin(-9.99);
402 vertexEnd = generated->GetXaxis()->FindBin(9.99);
404 TH1D* ptProj = dynamic_cast<TH1D*> (generated->ProjectionZ(Form("%s_pt", generated->GetName()), vertexBegin, vertexEnd, etaBegin, etaEnd));
405 //printf("GetMeasuredFraction: bin range %d %d %d %d\n", vertexBegin, vertexEnd, etaBegin, etaEnd);
406 ptProj->GetXaxis()->SetTitle(generated->GetZaxis()->GetTitle());
408 Int_t ptBin = ptProj->FindBin(ptCutOff);
409 //printf("GetMeasuredFraction: bin range %d %d\n", ptBin, ptProj->GetNbinsX());
410 Float_t abovePtCut = ptProj->Integral(ptBin, ptProj->GetNbinsX()+1);
411 Float_t all = ptProj->Integral(1, ptProj->GetNbinsX()+1);
416 Float_t fraction = abovePtCut / all;
418 //printf("GetMeasuredFraction: all %f above %f fraction %f\n", all, abovePtCut, fraction);
429 printf("AlidNdEtaCorrection::GetMeasuredFraction: pt cut off = %f, eta = %f, => fraction = %f\n", ptCutOff, eta, fraction);
434 //____________________________________________________________________
435 TH1* AlidNdEtaCorrection::GetMeasuredEventFraction(CorrectionType correctionType, Int_t multCut)
437 // calculates the fraction of events above multCut (but including it)
439 // uses the generated event histogram from the correction passed, e.g. pass GetTrack2ParticleCorrection()
441 if (!GetCorrection(correctionType))
444 const TH2* generated = GetCorrection(correctionType)->GetEventCorrection()->GetGeneratedHistogram();
446 TH1* allEvents = generated->ProjectionX(Form("%s_all", generated->GetName()), 1, generated->GetNbinsY());
447 TH1* aboveEvents = generated->ProjectionX(Form("%s_above", generated->GetName()), generated->GetYaxis()->FindBin(multCut), generated->GetNbinsY());
449 aboveEvents->Divide(aboveEvents, allEvents, 1, 1, "B");
454 //____________________________________________________________________
455 void AlidNdEtaCorrection::ReduceInformation()
457 // this function deletes the measured and generated histograms from the corrections to reduce the amount of data
460 // these are needed for GetMeasuredFraction(): fTrack2ParticleCorrection->ReduceInformation();
461 fVertexRecoCorrection ->ReduceInformation();
462 fTriggerBiasCorrectionMBToINEL ->ReduceInformation();
463 fTriggerBiasCorrectionMBToNSD ->ReduceInformation();
464 fTriggerBiasCorrectionMBToND ->ReduceInformation();
465 fTriggerBiasCorrectionMBToOnePart ->ReduceInformation();
468 //____________________________________________________________________
469 AliCorrection* AlidNdEtaCorrection::GetCorrection(CorrectionType correctionType)
471 // returns the given correction
473 switch (correctionType)
475 case kNone : return 0;
476 case kTrack2Particle : return fTrack2ParticleCorrection;
477 case kVertexReco : return fVertexRecoCorrection;
478 case kINEL : return fTriggerBiasCorrectionMBToINEL;
479 case kNSD : return fTriggerBiasCorrectionMBToNSD;
480 case kND : return fTriggerBiasCorrectionMBToND;
481 case kOnePart : return fTriggerBiasCorrectionMBToOnePart;