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"
24 #include <TDirectory.h>
25 #include <AliCorrection.h>
26 #include <AliCorrectionMatrix2D.h>
27 #include <AliCorrectionMatrix3D.h>
29 //____________________________________________________________________
30 ClassImp(AlidNdEtaCorrection)
32 //____________________________________________________________________
33 AlidNdEtaCorrection::AlidNdEtaCorrection()
35 fTrack2ParticleCorrection(0),
36 fVertexRecoCorrection(0),
37 fTriggerBiasCorrectionMBToINEL(0),
38 fTriggerBiasCorrectionMBToNSD(0),
39 fTriggerBiasCorrectionMBToND(0)
41 // default constructor
44 //____________________________________________________________________
45 AlidNdEtaCorrection::AlidNdEtaCorrection(const Char_t* name, const Char_t* title)
46 : TNamed(name, title),
47 fTrack2ParticleCorrection(0),
48 fVertexRecoCorrection(0),
49 fTriggerBiasCorrectionMBToINEL(0),
50 fTriggerBiasCorrectionMBToNSD(0),
51 fTriggerBiasCorrectionMBToND(0)
57 fTrack2ParticleCorrection = new AliCorrection("Track2Particle", "Track2Particle");
58 fVertexRecoCorrection = new AliCorrection("VertexReconstruction", "VertexReconstruction");
60 fTriggerBiasCorrectionMBToINEL = new AliCorrection("TriggerBias_MBToINEL", "TriggerBias_MBToINEL");
61 fTriggerBiasCorrectionMBToNSD = new AliCorrection("TriggerBias_MBToNSD", "TriggerBias_MBToNSD");
62 fTriggerBiasCorrectionMBToND = new AliCorrection("TriggerBias_MBToND", "TriggerBias_MBToND");
65 //____________________________________________________________________
66 AlidNdEtaCorrection::~AlidNdEtaCorrection()
70 if (fTrack2ParticleCorrection) {
71 delete fTrack2ParticleCorrection;
72 fTrack2ParticleCorrection = 0;
75 if (fVertexRecoCorrection) {
76 delete fVertexRecoCorrection;
77 fVertexRecoCorrection = 0;
80 if (fTriggerBiasCorrectionMBToINEL) {
81 delete fTriggerBiasCorrectionMBToINEL;
82 fTriggerBiasCorrectionMBToINEL = 0;
85 if (fTriggerBiasCorrectionMBToNSD) {
86 delete fTriggerBiasCorrectionMBToNSD;
87 fTriggerBiasCorrectionMBToNSD = 0;
90 if (fTriggerBiasCorrectionMBToND) {
91 delete fTriggerBiasCorrectionMBToND;
92 fTriggerBiasCorrectionMBToND = 0;
96 //____________________________________________________________________
98 AlidNdEtaCorrection::Finish() {
102 // divide the histograms in the AliCorrectionMatrix2D objects to get the corrections
104 fTrack2ParticleCorrection->Divide();
105 fVertexRecoCorrection->Divide();
106 fTriggerBiasCorrectionMBToINEL->Divide();
107 fTriggerBiasCorrectionMBToNSD->Divide();
108 fTriggerBiasCorrectionMBToND->Divide();
111 //____________________________________________________________________
112 Long64_t AlidNdEtaCorrection::Merge(TCollection* list)
114 // Merge a list of dNdEtaCorrection objects with this (needed for
116 // Returns the number of merged objects (including this).
124 TIterator* iter = list->MakeIterator();
127 // collections of measured and generated histograms
128 TList* collectionNtrackToNparticle = new TList;
129 TList* collectionVertexReco = new TList;
130 TList* collectionTriggerBiasMBToINEL = new TList;
131 TList* collectionTriggerBiasMBToNSD = new TList;
132 TList* collectionTriggerBiasMBToND = new TList;
135 while ((obj = iter->Next())) {
137 AlidNdEtaCorrection* entry = dynamic_cast<AlidNdEtaCorrection*> (obj);
141 collectionNtrackToNparticle ->Add(entry->fTrack2ParticleCorrection);
142 collectionVertexReco ->Add(entry->fVertexRecoCorrection);
143 collectionTriggerBiasMBToINEL->Add(entry->fTriggerBiasCorrectionMBToINEL);
144 collectionTriggerBiasMBToNSD ->Add(entry->fTriggerBiasCorrectionMBToNSD);
145 collectionTriggerBiasMBToND ->Add(entry->fTriggerBiasCorrectionMBToND);
149 fTrack2ParticleCorrection ->Merge(collectionNtrackToNparticle);
150 fVertexRecoCorrection ->Merge(collectionVertexReco);
151 fTriggerBiasCorrectionMBToINEL ->Merge(collectionTriggerBiasMBToINEL);
152 fTriggerBiasCorrectionMBToNSD ->Merge(collectionTriggerBiasMBToNSD);
153 fTriggerBiasCorrectionMBToND ->Merge(collectionTriggerBiasMBToND);
155 delete collectionNtrackToNparticle;
156 delete collectionVertexReco;
157 delete collectionTriggerBiasMBToINEL;
158 delete collectionTriggerBiasMBToNSD;
159 delete collectionTriggerBiasMBToND;
164 //____________________________________________________________________
165 void AlidNdEtaCorrection::Add(AlidNdEtaCorrection* aCorrectionsToAdd, Float_t c) {
167 // adds the measured and generated of aCorrectionsToAdd to measured and generated
168 // of all corrections in this
170 fTrack2ParticleCorrection ->Add(aCorrectionsToAdd->GetTrack2ParticleCorrection() ,c);
171 fVertexRecoCorrection ->Add(aCorrectionsToAdd->GetVertexRecoCorrection() ,c);
172 fTriggerBiasCorrectionMBToINEL ->Add(aCorrectionsToAdd->GetTriggerBiasCorrectionINEL(),c);
173 fTriggerBiasCorrectionMBToNSD ->Add(aCorrectionsToAdd->GetTriggerBiasCorrectionNSD() ,c);
174 fTriggerBiasCorrectionMBToND ->Add(aCorrectionsToAdd->GetTriggerBiasCorrectionND() ,c);
178 //____________________________________________________________________
179 void AlidNdEtaCorrection::Reset(void) {
181 // reset all corrections
184 fTrack2ParticleCorrection ->Reset();
185 fVertexRecoCorrection ->Reset();
186 fTriggerBiasCorrectionMBToINEL ->Reset();
187 fTriggerBiasCorrectionMBToNSD ->Reset();
188 fTriggerBiasCorrectionMBToND ->Reset();
194 //____________________________________________________________________
195 Bool_t AlidNdEtaCorrection::LoadHistograms(const Char_t* dir)
198 // loads the histograms
199 // if dir is empty a directory with the name of this object is taken (like in SaveHistogram)
205 if (!gDirectory->cd(dir))
208 fTrack2ParticleCorrection ->LoadHistograms();
209 fVertexRecoCorrection ->LoadHistograms();
210 fTriggerBiasCorrectionMBToINEL ->LoadHistograms();
211 fTriggerBiasCorrectionMBToNSD ->LoadHistograms();
212 fTriggerBiasCorrectionMBToND ->LoadHistograms();
214 gDirectory->cd("..");
219 //____________________________________________________________________
220 void AlidNdEtaCorrection::SaveHistograms()
223 // save the histograms
226 gDirectory->mkdir(fName.Data());
227 gDirectory->cd(fName.Data());
229 fTrack2ParticleCorrection ->SaveHistograms();
230 fVertexRecoCorrection ->SaveHistograms();
231 fTriggerBiasCorrectionMBToINEL->SaveHistograms();
232 fTriggerBiasCorrectionMBToNSD ->SaveHistograms();
233 fTriggerBiasCorrectionMBToND ->SaveHistograms();
235 gDirectory->cd("..");
238 //____________________________________________________________________
239 void AlidNdEtaCorrection::DrawHistograms()
242 // call the draw histogram method of the correction
245 fTrack2ParticleCorrection ->DrawHistograms();
246 fVertexRecoCorrection ->DrawHistograms();
247 fTriggerBiasCorrectionMBToINEL->DrawHistograms();
248 fTriggerBiasCorrectionMBToNSD ->DrawHistograms();
249 fTriggerBiasCorrectionMBToND ->DrawHistograms();
252 //____________________________________________________________________
253 void AlidNdEtaCorrection::FillMCParticle(Float_t vtx, Float_t eta, Float_t pt, Bool_t trigger, Bool_t vertex, Int_t processType)
255 // fills a particle in the corrections
256 // it is filled in generated or measured depending of the flags
258 fTriggerBiasCorrectionMBToINEL->GetTrackCorrection()->FillGene(vtx, eta, pt);
260 if (processType != 92 && processType != 93)
261 fTriggerBiasCorrectionMBToNSD->GetTrackCorrection()->FillGene(vtx, eta, pt);
263 if (processType!=92 && processType!=93 && processType!=94)
264 fTriggerBiasCorrectionMBToND->GetTrackCorrection()->FillGene(vtx, eta, pt);
269 fTriggerBiasCorrectionMBToINEL->GetTrackCorrection()->FillMeas(vtx, eta, pt);
270 fTriggerBiasCorrectionMBToNSD->GetTrackCorrection()->FillMeas(vtx, eta, pt);
271 fTriggerBiasCorrectionMBToND->GetTrackCorrection()->FillMeas(vtx, eta, pt);
272 fVertexRecoCorrection->GetTrackCorrection()->FillGene(vtx, eta, pt);
277 fVertexRecoCorrection->GetTrackCorrection()->FillMeas(vtx, eta, pt);
278 fTrack2ParticleCorrection->GetTrackCorrection()->FillGene(vtx, eta, pt);
281 //____________________________________________________________________
282 void AlidNdEtaCorrection::FillTrackedParticle(Float_t vtx, Float_t eta, Float_t pt)
284 // fills a tracked particle in the corrections
286 fTrack2ParticleCorrection->GetTrackCorrection()->FillMeas(vtx, eta, pt);
289 //____________________________________________________________________
290 void AlidNdEtaCorrection::FillEvent(Float_t vtx, Float_t n, Bool_t trigger, Bool_t vertex, Int_t processType)
292 // fills an event int he correction
293 // it is filled in generated or measured depending of the flags
295 fTriggerBiasCorrectionMBToINEL->GetEventCorrection()->FillGene(vtx, n);
297 if (processType != 92 && processType != 93)
298 fTriggerBiasCorrectionMBToNSD->GetEventCorrection()->FillGene(vtx, n);
300 if (processType!=92 && processType!=93 && processType!=94)
301 fTriggerBiasCorrectionMBToND->GetEventCorrection()->FillGene(vtx, n);
306 fTriggerBiasCorrectionMBToINEL->GetEventCorrection()->FillMeas(vtx, n);
307 fTriggerBiasCorrectionMBToNSD->GetEventCorrection()->FillMeas(vtx, n);
308 fTriggerBiasCorrectionMBToND->GetEventCorrection()->FillMeas(vtx, n);
309 fVertexRecoCorrection->GetEventCorrection()->FillGene(vtx, n);
314 fVertexRecoCorrection->GetEventCorrection()->FillMeas(vtx, n);
317 //____________________________________________________________________
318 Float_t AlidNdEtaCorrection::GetMeasuredFraction(CorrectionType correctionType, Float_t ptCutOff, Float_t eta, Bool_t debug)
320 // calculates the fraction of particles measured (some are missed due to the pt cut off)
322 // uses the generated particle histogram from the correction passed, e.g. pass GetTrack2ParticleCorrection()
324 if (!GetCorrection(correctionType))
327 const TH3F* generated = GetCorrection(correctionType)->GetTrackCorrection()->GetGeneratedHistogram();
329 // find eta borders, if eta is negative assume -0.8 ... 0.8
334 etaBegin = generated->GetYaxis()->FindBin(-0.8);
335 etaEnd = generated->GetYaxis()->FindBin(0.8);
339 etaBegin = generated->GetYaxis()->FindBin(eta);
343 Int_t vertexBegin = generated->GetXaxis()->FindBin(-9.99);
344 Int_t vertexEnd = generated->GetXaxis()->FindBin(9.99);
346 TH1D* ptProj = dynamic_cast<TH1D*> (generated->ProjectionZ(Form("%s_pt", generated->GetName()), vertexBegin, vertexEnd, etaBegin, etaEnd));
347 //printf("GetMeasuredFraction: bin range %d %d %d %d\n", vertexBegin, vertexEnd, etaBegin, etaEnd);
348 ptProj->GetXaxis()->SetTitle(generated->GetZaxis()->GetTitle());
350 Int_t ptBin = ptProj->FindBin(ptCutOff);
351 //printf("GetMeasuredFraction: bin range %d %d\n", ptBin, ptProj->GetNbinsX());
352 Float_t abovePtCut = ptProj->Integral(ptBin, ptProj->GetNbinsX());
353 Float_t all = ptProj->Integral();
358 Float_t fraction = abovePtCut / all;
360 //printf("GetMeasuredFraction: all %f above %f fraction %f\n", all, abovePtCut, fraction);
371 printf("AlidNdEtaCorrection::GetMeasuredFraction: pt cut off = %f, eta = %f, => fraction = %f\n", ptCutOff, eta, fraction);
376 //____________________________________________________________________
377 void AlidNdEtaCorrection::ReduceInformation()
379 // this function deletes the measured and generated histograms from the corrections to reduce the amount of data
382 // these are needed for GetMeasuredFraction(): fTrack2ParticleCorrection->ReduceInformation();
383 fVertexRecoCorrection ->ReduceInformation();
384 fTriggerBiasCorrectionMBToINEL ->ReduceInformation();
385 fTriggerBiasCorrectionMBToNSD ->ReduceInformation();
386 fTriggerBiasCorrectionMBToND ->ReduceInformation();
389 //____________________________________________________________________
390 AliCorrection* AlidNdEtaCorrection::GetCorrection(CorrectionType correctionType)
392 // returns the given correction
394 switch (correctionType)
396 case kNone : return 0;
397 case kTrack2Particle : return fTrack2ParticleCorrection;
398 case kVertexReco : return fVertexRecoCorrection;
399 case kINEL : return fTriggerBiasCorrectionMBToINEL;
400 case kNSD : return fTriggerBiasCorrectionMBToNSD;
401 case kND : return fTriggerBiasCorrectionMBToND;