]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGPP/TPC/AliComparisonDraw.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
coverity fix
[u/mrichter/AliRoot.git] / PWGPP / TPC / AliComparisonDraw.cxx
CommitLineData
7cc34f08 1
2
3//
4// Comparison draw
5// Compare the MC information with the reconstructed
6//
7
8/*
9 after running analysis, read the file, and get component
2bfe5463 10 gSystem->Load("libPWGPP.so");
7cc34f08 11 TFile f("Output.root");
12 AliComparisonDraw * comp = (AliComparisonDraw*)f.Get("AliComparisonDraw");
13 TF1 fl("fl","((min(250./(abs(x+0.000001)),250)-90))",0,2); // length function
14 TF1 fl2("fl2","[0]/((min(250./(abs(x+0.000001)),250)-90))^[1]",0,2);
15 fl2.SetParameter(1,1);
16 fl2.SetParameter(0,1);
17
18*/
19
20
21
22
23#include "TFile.h"
24#include "TCint.h"
25#include "TH3F.h"
26#include "TH2F.h"
27#include "TF1.h"
28#include "TProfile.h"
29#include "TProfile2D.h"
30#include "TGraph2D.h"
31#include "TCanvas.h"
32#include "TGraph.h"
33//
34//
35#include "AliESDEvent.h" // new container
36#include "AliESD.h"
37#include "AliESDfriend.h"
38#include "AliESDfriendTrack.h"
39//
40#include "AliMathBase.h"
41#include "AliTreeDraw.h"
42
43#include "AliMCInfo.h"
44#include "AliESDRecInfo.h"
45#include "AliComparisonDraw.h"
46
47
48ClassImp(AliComparisonDraw)
49
50Bool_t AliComparisonDraw::fgBDraw=kFALSE; //option draw temporary results
51
52AliComparisonDraw::AliComparisonDraw():
53 TNamed("ComparisonDraw","ComparisonDraw"),
54 fEffTPCPt(0), // TPC efficiency as function of Pt (tan+-1)
55 fEffTPCPtMC(0), // MC -TPC efficiency as function of Pt (tan+-1)
56 fEffTPCPtF(0), // efficiency for findable tracks
57 //
58 fEffTPCTan(0), // TPC efficiency as function of Tan (pt>0.15
59 fEffTPCTanMC(0), // MC -TPC efficiency as function of Tan (pt>0.15)
60 fEffTPCTanF(0), // efficiency for findable tracks Tan (pt>0.15)
61 //
62 fEffTPCPtTan(0), // TPC efficiency as function of Pt and tan
63 fEffTPCPtTanMC(0), // MC -TPC efficiency as function of Pt and tan
64 fEffTPCPtTanF(0), // TPC efficiency as function of Pt and tan
65 //
66 // dEdx resolution
67 //
68 fTPCSignalNormTan(0), // tpc signal normalized to the mean signal - MC
69 fTPCSignalNormSPhi(0), // tpc signal normalized to the mean signal - MC
70 fTPCSignalNormTPhi(0), // tpc signal normalized to the mean signal - MC
71 //
72 fTPCSignalNormTanSPhi(0), // tpc signal normalized to the mean signal - MC
73 fTPCSignalNormTanTPhi(0), // tpc signal normalized to the mean signal - MC
74 fTPCSignalNormTanSPt(0), // tpc signal normalized to the mean signal - MC
75 //
76 //
77 fPtResolLPT(0), // pt resolution - low pt
78 fPtResolHPT(0), // pt resolution - high pt
79 fPtPullLPT(0), // pt resolution - low pt
80 fPtPullHPT(0), // pt resolution - high pt
81 //
82 // Resolution constrained param
83 //
84 fCPhiResolTan(0), // angular resolution - constrained
85 fCTanResolTan(0), // angular resolution - constrained
86 fCPtResolTan(0), // pt resolution - constrained
87 fCPhiPullTan(0), // angular resolution - constrained
88 fCTanPullTan(0), // angular resolution - constrained
89 fCPtPullTan(0), // pt resolution - constrained
90 //
91 // DCA resolution
92 //
93 fD0TanSPtB1(0), // distance to vertex y
94 fD1TanSPtB1(0), // distance to vertex z
95 fD0TanSPtL1(0), // distance to vertex y
96 fD1TanSPtL1(0) // distance to vertex z
97{
98 InitHisto();
99}
100
101AliComparisonDraw::AliComparisonDraw(const AliComparisonDraw& draw):
102 TNamed(draw.GetName(),draw.GetTitle()),
103 fEffTPCPt(draw.fEffTPCPt), // TPC efficiency as function of Pt (tan+-1)
104 fEffTPCPtMC(draw.fEffTPCPtMC), // MC -TPC efficiency as function of Pt (tan+-1)
105 fEffTPCPtF(draw.fEffTPCPtF), // efficiency for findable tracks
106 //
107 fEffTPCTan(draw.fEffTPCTan), // TPC efficiency as function of Tan (pt>0.15
108 fEffTPCTanMC(draw.fEffTPCTanMC), // MC -TPC efficiency as function of Tan (pt>0.15)
109 fEffTPCTanF(draw.fEffTPCTanF), // efficiency for findable tracks Tan (pt>0.15)
110 //
111 fEffTPCPtTan(draw.fEffTPCPtTan), // TPC efficiency as function of Pt and tan
112 fEffTPCPtTanMC(draw.fEffTPCPtTanMC), // MC -TPC efficiency as function of Pt and tan
113 fEffTPCPtTanF(draw.fEffTPCPtTanF), // TPC efficiency as function of Pt and tan
114 //
115 // dEdx resolution
116 //
117 fTPCSignalNormTan(draw.fTPCSignalNormTan), // tpc signal normalized to the mean signal - MC
118 fTPCSignalNormSPhi(draw.fTPCSignalNormSPhi), // tpc signal normalized to the mean signal - MC
119 fTPCSignalNormTPhi(draw.fTPCSignalNormTPhi), // tpc signal normalized to the mean signal - MC
120 //
121 fTPCSignalNormTanSPhi(draw.fTPCSignalNormTanSPhi), // tpc signal normalized to the mean signal - MC
122 fTPCSignalNormTanTPhi(draw.fTPCSignalNormTanTPhi), // tpc signal normalized to the mean signal - MC
123 fTPCSignalNormTanSPt(draw.fTPCSignalNormTanSPt), // tpc signal normalized to the mean signal - MC
124 //
125 //
126 fPtResolLPT(draw.fPtResolLPT), // pt resolution - low pt
127 fPtResolHPT(draw.fPtResolHPT), // pt resolution - high pt
128 fPtPullLPT(draw.fPtPullLPT), // pt resolution - low pt
129 fPtPullHPT(draw.fPtPullHPT), // pt resolution - high pt
130 //
131 // Resolution constrained param
132 //
133 fCPhiResolTan(draw.fCPhiResolTan), // angular resolution - constrained
134 fCTanResolTan(draw.fCTanResolTan), // angular resolution - constrained
135 fCPtResolTan(draw.fCPtResolTan), // pt resolution - constrained
136 fCPhiPullTan(draw.fCPhiPullTan), // angular resolution - constrained
137 fCTanPullTan(draw.fCTanPullTan), // angular resolution - constrained
138 fCPtPullTan(draw.fCPtPullTan), // pt resolution - constrained
139 //
140 // DCA resolution
141 //
142 fD0TanSPtB1(draw.fD0TanSPtB1), // distance to vertex y
143 fD1TanSPtB1(draw.fD1TanSPtB1), // distance to vertex z
144 fD0TanSPtL1(draw.fD0TanSPtL1), // distance to vertex y
145 fD1TanSPtL1(draw.fD1TanSPtL1) // distance to vertex z
146{
147 //
148 // copy constructor
149 //
150}
151
152AliComparisonDraw& AliComparisonDraw::operator=(const AliComparisonDraw& info){
153 //
154 // assignment operator
155 //
156 delete this;
157 new (this) AliComparisonDraw(info);
158 return *this;
159}
160
161
162
163
164AliComparisonDraw::~AliComparisonDraw(){
165 //
166 //
167 //
168 delete fEffTPCPt; // TPC efficiency as function of Pt (tan+-1)
169 delete fEffTPCPtMC; // MC -TPC efficiency as function of Pt (tan+-1)
170 delete fEffTPCPtF; // efficiency for findable tracks
171 //
172 delete fEffTPCTan; // TPC efficiency as function of Tan (pt>0.15
173 delete fEffTPCTanMC; // MC -TPC efficiency as function of Tan (pt>0.15)
174 delete fEffTPCTanF; // efficiency for findable tracks Tan (pt>0.15)
175 //
176 delete fEffTPCPtTan; // TPC efficiency as function of Pt and tan
177 delete fEffTPCPtTanMC; // MC -TPC efficiency as function of Pt and tan
178 delete fEffTPCPtTanF; // TPC efficiency as function of Pt and tan
179 //
180 // dEdx resolution
181 //
182 delete fTPCSignalNormTan; // tpc signal normalized to the mean signal - MC
183 delete fTPCSignalNormSPhi; // tpc signal normalized to the mean signal - MC
184 delete fTPCSignalNormTPhi; // tpc signal normalized to the mean signal - MC
185 //
186 delete fTPCSignalNormTanSPhi; // tpc signal normalized to the mean signal - MC
187 delete fTPCSignalNormTanTPhi; // tpc signal normalized to the mean signal - MC
188 delete fTPCSignalNormTanSPt; // tpc signal normalized to the mean signal - MC
189 //
190 //
191 delete fPtResolLPT; // pt resolution - low pt
192 delete fPtResolHPT; // pt resolution - high pt
193 delete fPtPullLPT; // pt resolution - low pt
194 delete fPtPullHPT; // pt resolution - high pt
195 //
196 // Resolution constrained param
197 //
198 delete fCPhiResolTan; // angular resolution - constrained
199 delete fCTanResolTan; // angular resolution - constrained
200 delete fCPtResolTan; // pt resolution - constrained
201 delete fCPhiPullTan; // angular resolution - constrained
202 delete fCTanPullTan; // angular resolution - constrained
203 delete fCPtPullTan; // pt resolution - constrained
204 //
205 // DCA resolution
206 //
207 delete fD0TanSPtB1; // distance to vertex y
208 delete fD1TanSPtB1; // distance to vertex z
209 delete fD0TanSPtL1; // distance to vertex y
210 delete fD1TanSPtL1; // distance to vertex z
211
212}
213
214
215
216
217void AliComparisonDraw::InitHisto(){
218 //
219 //
220 // EFFICIENCY
221 //
222 // Efficiency as function of pt
223 fEffTPCPt = new TProfile("Eff_pt","Eff_Pt",50,0.1,3); // physical
224 fEffTPCPtMC = new TProfile("MC_Eff_pt","MC_Eff_Pt",50,0.1,3); // MC - particles make more than 50 rowdigits
225 fEffTPCPtF = new TProfile("F_Eff_pt","F_Eff_Pt",50,0.1,3); // tracking - under condition more than 50 rdigits
226
227 // Efficiency as function of pt
228 fEffTPCTan = new TProfile("Eff_tan","Eff_tan",50,-2.5,2.5); // physical
229 fEffTPCTanMC = new TProfile("MC_Eff_tan","MC_Eff_tan",50,-2.5,2.5); // MC - particles make more than 50 rowdigits
230 fEffTPCTanF = new TProfile("F_Eff_tan","F_Eff_tan",50,-2.5,2.5); // tracking - under condition more than 50 rdigits
231
232 fEffTPCPtTan = new TProfile2D("Eff_pt","Eff_Pt",10,0.1,3,20,-2.,2.);
233 fEffTPCPtTanMC = new TProfile2D("MC_Eff_pt","MC Eff Pt",10,0.1,3,20, -2.,2.);
234 fEffTPCPtTanF = new TProfile2D("MC_Eff_pt","MC Eff Pt",10,0.1,3,20, -2.,2.);
235
236 //
237 // TPC dEdx
238 //
239 fTPCSignalNormTan = new TH2F("CdEdxTan","CdEdxTan",50, -2,2, 40,30,70); // tpc signal normalized to the MC
240 fTPCSignalNormSPhi = new TH2F("CdEdxSPhi","CdEdxSPhi",10,0.0,1,40,30,70); // tpc signal normalized to the MC
241 fTPCSignalNormTPhi = new TH2F("CdEdxTPhi","CdEdxTPhi",10,0.0,2,40,30,70); // tpc signal normalized to the MC
242
243 fTPCSignalNormTanSPhi= new TH3F("CdEdxTanSPhi","CdEdxTanSPhi",20, -2,2, 10,0.0 ,1, 40,30,70); // tpc signal normalized to the mean signal - MC
244 fTPCSignalNormTanTPhi= new TH3F("CdEdxTanTPhi","CdEdxTanTPhi",20, -2,2, 10,0.0 ,1, 40,30,70); // tpc signal normalized to the mean signal - MC
245 fTPCSignalNormTanSPt= new TH3F("CdEdxTanSPt","CdEdxTanSPt",20, -2,2, 10,0.3 ,3, 40,30,70); // tpc signal normalized to the mean signal - MC
246
247
248
249 //
250 // RESOLUTION
251 //
252 fCPtResolTan = new TH2F("C Pt resol","C pt resol",50, -2,2,200,-0.2,0.2);
253 fCPtPullTan = new TH2F("C Pt pull","C pt pull",50, -2,2,200,-5,5);
254 //
255 fCPhiResolTan = new TH2F("CPhiResolTan","CPhiResolTan",50, -2,2,200,-0.025,0.025);
256 // angular resolution - constrained
257 fCTanResolTan = new TH2F("CTanResolTan","CTanResolTan",50, -2,2,200,-0.025,0.025);
258 // angular resolution - constrained
259 fCPtResolTan=new TH2F("CPtResol","CPtResol",50, -2,2,200,-0.2,0.2);;
260 // pt resolution - constrained
261 fCPhiPullTan = new TH2F("CPhiPullTan","CPhiPullTan",50, -2,2,200,-5,5);
262 // angular resolution - constrained
263 fCTanPullTan = new TH2F("CTanPullTan","CTanPullTan",50, -2,2,200,-5,5);
264 // angular resolution - constrained
265 fCPtPullTan=new TH2F("CPtPull","CPtPull",50, -2,2,200,-5,5);
266 // pt resolution - constrained
267 //
268 fPtResolLPT = new TH2F("Pt resol","pt resol",10, 0.1,3,200,-0.2,0.2);
269 fPtResolHPT = new TH2F("Pt resol","pt resol",10, 2,100,200,-0.3,0.3);
270 fPtPullLPT = new TH2F("Pt pool","pt pool",10, 0.1,3,200,-6,6);
271 fPtPullHPT = new TH2F("Pt pool","pt pool",10, 2,100,200,-6,6);
272 //
273 fD0TanSPtB1 = new TH3F("DCAyTanSPt","DCAyTanSPt",20,1,2, 10,0.3,2, 100,-4,4);
274 fD1TanSPtB1 = new TH3F("DCAzTanSPt","DCAzTanSPt",20,1,2, 10,0.3,2, 100,-4,4);
275 fD0TanSPtL1 = new TH3F("DCAyTanSPt","DCAyTanSPt",20,0,1, 10,0.3,2, 100,-0.1,0.1);
276 fD1TanSPtL1 = new TH3F("DCAzTanSPt","DCAzTanSPt",20,0,1, 10,0.3,2, 100, -0.1,0.1);
277
278
279
280}
281
282void AliComparisonDraw::ProcessEff(AliMCInfo* infoMC, AliESDRecInfo *infoRC){
283 //
284 // make efficiencies histograms
285 //
286 Float_t kptcut = 0.15;
287 Float_t ktancut=1.;
288 Int_t kmincl =50;
289 Float_t mcpt = infoMC->GetParticle().Pt();
290 Float_t tantheta = TMath::Tan(infoMC->GetParticle().Theta()-TMath::Pi()*0.5);
291 Bool_t isPrim = infoMC->GetParticle().R()<0.1 && TMath::Abs(infoMC->GetParticle().Vz())<10;
292 //z diamond and
293
294 if (!isPrim) return;
295
296 //pt
297 if (TMath::Abs(tantheta)<ktancut){
298 fEffTPCPt->Fill(mcpt, infoRC->GetStatus(1)==3);
299 fEffTPCPtMC->Fill(mcpt, infoMC->GetRowsWithDigits()>kmincl);
300 if (infoMC->GetRowsWithDigits()>kmincl){
301 fEffTPCPtF->Fill(mcpt, infoRC->GetStatus(1)==3);
302 }
303 }
304
305 //theta
306 if (TMath::Abs(mcpt)>kptcut){
307 fEffTPCTan->Fill(tantheta, infoRC->GetStatus(1)==3);
308 fEffTPCTanMC->Fill(tantheta, infoMC->GetRowsWithDigits()>kmincl);
309 if (infoMC->GetRowsWithDigits()>kmincl){
310 fEffTPCTanF->Fill(tantheta, infoRC->GetStatus(1)==3);
311 }
312 }
313 //
314 // pt-theta
315 //
316 fEffTPCPtTan->Fill(mcpt,tantheta,infoRC->GetStatus(1)==3);
317 fEffTPCPtTanMC->Fill(mcpt,tantheta,infoMC->GetRowsWithDigits()>50);
318 if (infoMC->GetRowsWithDigits()>kmincl){
319 fEffTPCPtTanF->Fill(mcpt,tantheta,infoRC->GetStatus(1)==3);
320 }
321}
322
323
324void AliComparisonDraw::ProcessResolConstrained(AliMCInfo* infoMC, AliESDRecInfo *infoRC){
325 //
326 //
327 //
328 Float_t mcpt = infoMC->GetParticle().Pt();
329 Float_t tantheta = TMath::Tan(infoMC->GetParticle().Theta()-TMath::Pi()*0.5);
330 Bool_t isPrim = infoMC->GetParticle().R()<0.1 && TMath::Abs(infoMC->GetParticle().Vz())<10;
331 //z diamond and
332
333 if (!isPrim) return;
334 if (infoRC->GetStatus(1)!=3) return;
335 if (!infoRC->GetESDtrack()) return;
336 if (infoRC->GetESDtrack()->GetTPCNcls()<10) return;
337 if (!infoRC->GetESDtrack()->GetConstrainedParam()) return;
338
339 //
340 // constrained parameters resolution
341 //
342 const AliExternalTrackParam * cparam = infoRC->GetESDtrack()->GetConstrainedParam();
343 Float_t deltaCPt= (mcpt-cparam->Pt())/mcpt;
344 Float_t pullCPt= (1/mcpt-cparam->OneOverPt())/
345 TMath::Sqrt(cparam->GetSigma1Pt2());
346 Float_t deltaPhi = TMath::ATan2(cparam->Py(),cparam->Px())-
347 TMath::ATan2(infoMC->GetParticle().Py(),infoMC->GetParticle().Px());
348 Float_t pullPhi = deltaPhi/TMath::Sqrt(cparam->GetSigmaSnp2());
349
350 Float_t deltaTan = TMath::ATan2(cparam->Pz(),cparam->Pt())-
351 TMath::ATan2(infoMC->GetParticle().Pz(),infoMC->GetParticle().Pt());
352 Float_t pullTan = deltaPhi/TMath::Sqrt(cparam->GetSigmaSnp2());
353
354 fCPtResolTan->Fill(tantheta,deltaCPt);
355 fCPtPullTan->Fill(tantheta,pullCPt);
356 fCPhiResolTan->Fill(tantheta,deltaPhi);
357 fCPhiPullTan->Fill(tantheta,pullPhi);
358 fCTanResolTan->Fill(tantheta,deltaTan);
359 fCTanPullTan->Fill(tantheta,pullTan);
360
361}
362
363
364
365void AliComparisonDraw::ProcessTPCdedx(AliMCInfo* infoMC, AliESDRecInfo *infoRC){
366 //
367 //
368 //
369 Float_t mcpt = infoMC->GetParticle().Pt();
370 Float_t tantheta = TMath::Tan(infoMC->GetParticle().Theta()-TMath::Pi()*0.5);
371 Bool_t isPrim = infoMC->GetParticle().R()<0.1 && TMath::Abs(infoMC->GetParticle().Vz())<10;
372 //z diamond and
373
374 if (!isPrim) return;
375 if (infoRC->GetStatus(1)!=3) return;
376 if (!infoRC->GetESDtrack()) return;
377 if (infoRC->GetESDtrack()->GetTPCNcls()<10) return;
378 if (!infoRC->GetESDtrack()->GetConstrainedParam()) return;
379 Float_t mprim = infoMC->GetPrim();
380 if (mprim>1.4) return;
381 if (mprim<0.5) return;
382 if (infoRC->GetESDtrack()->GetTPCsignalN()<50) return;
383 //
384 Float_t ratio = infoRC->GetESDtrack()->GetTPCsignal()/infoMC->GetPrim();
385 Float_t sphi = infoRC->GetESDtrack()->GetInnerParam()->GetSnp();
386 Float_t tphi = sphi/TMath::Sqrt((1.-sphi)*(1.+sphi));
387
388
389 if (TMath::Abs(infoMC->GetParticle().GetPdgCode())!=211) return;
390 if (mcpt>0.5){
391 fTPCSignalNormTan->Fill(tantheta,ratio); //only subset
392 }
393 if (TMath::Abs(tantheta)<0.5){
394 fTPCSignalNormSPhi->Fill(sphi,ratio); // only subset
395 fTPCSignalNormTPhi->Fill(tphi,ratio); // only subset
396 }
397 fTPCSignalNormTanSPhi->Fill(tantheta,sphi,ratio);
398 fTPCSignalNormTanTPhi->Fill(tantheta,tphi,ratio);
399 fTPCSignalNormTanSPt->Fill(tantheta,TMath::Sqrt(mcpt),ratio);
400}
401
402void AliComparisonDraw::ProcessDCA(AliMCInfo* infoMC, AliESDRecInfo *infoRC){
403 //
404 //
405 //
406 Float_t mcpt = infoMC->GetParticle().Pt();
407 Float_t tantheta = TMath::Tan(infoMC->GetParticle().Theta()-TMath::Pi()*0.5);
408 Bool_t isPrim = infoMC->GetParticle().R()<0.1 && TMath::Abs(infoMC->GetParticle().Vz())<10;
409 //z diamond and
410 if (!isPrim) return;
411 if (infoRC->GetStatus(1)!=3) return;
412 if (!infoRC->GetESDtrack()) return;
413 if (infoRC->GetESDtrack()->GetTPCNcls()<10) return;
414 if (!infoRC->GetESDtrack()->GetConstrainedParam()) return;
415 Float_t spt = TMath::Sqrt(mcpt);
416 Float_t dca[2],cov[3];
417 infoRC->GetESDtrack()->GetImpactParameters(dca,cov);
418 Int_t clusterITS[100];
419 if (infoRC->GetESDtrack()->GetITSclusters(clusterITS)==0){
420 fD0TanSPtB1->Fill(tantheta,spt,dca[0]);
421 fD1TanSPtB1->Fill(tantheta,spt,dca[1]);
422 }
423 fD0TanSPtL1->Fill(tantheta,spt,dca[0]);
424 fD1TanSPtL1->Fill(tantheta,spt,dca[1]);
425}
426
427
428
429
430void AliComparisonDraw::Process(AliMCInfo* infoMC, AliESDRecInfo *infoRC){
431 //
432 //
433 //
434 ProcessEff(infoMC,infoRC);
435 ProcessResolConstrained(infoMC,infoRC);
436 ProcessTPCdedx(infoMC, infoRC);
437 ProcessDCA(infoMC, infoRC);
438
439 Float_t mcpt = infoMC->GetParticle().Pt();
440 Bool_t isPrim = infoMC->GetParticle().R()<0.1 && TMath::Abs(infoMC->GetParticle().Vz())<10;
441 //z diamond and
442
443 if (!isPrim) return;
444 //
445 //
446 if (infoRC->GetStatus(1)==0) return;
447 if (!infoRC->GetESDtrack()) return;
448 if (infoRC->GetESDtrack()->GetTPCNcls()<10) return;
449 // printf("Pt\t%f\t%f\n",mcpt, infoRC->GetESDtrack()->Pt());
450
451 Float_t deltaPt= (mcpt-infoRC->GetESDtrack()->Pt())/mcpt;
452 Float_t poolPt= (1/mcpt-infoRC->GetESDtrack()->OneOverPt())/
453 TMath::Sqrt(infoRC->GetESDtrack()->GetSigma1Pt2());
454
455 fPtResolLPT->Fill(mcpt,deltaPt);
456 fPtResolHPT->Fill(mcpt,deltaPt);
457 fPtPullLPT->Fill(mcpt,poolPt);
458 fPtPullHPT->Fill(mcpt,poolPt);
459}
460
461
462
463TH1F* AliComparisonDraw::MakeResol(TH2F * his, Int_t integ, Bool_t type){
464 TH1F *hisr, *hism;
465 if (!gPad) new TCanvas;
466 hisr = AliTreeDraw::CreateResHistoI(his,&hism,integ);
467 if (type) return hism;
468 else
469 return hisr;
470}
471
472
473TGraph2D * AliComparisonDraw::MakeStat2D(TH3 * his, Int_t delta0, Int_t delta1, Int_t type){
474 //
475 //
476 //
477 // delta - number of bins to integrate
478 // type - 0 - mean value
479
480 TAxis * xaxis = his->GetXaxis();
481 TAxis * yaxis = his->GetYaxis();
482 // TAxis * zaxis = his->GetZaxis();
483 Int_t nbinx = xaxis->GetNbins();
484 Int_t nbiny = yaxis->GetNbins();
485 char name[1000];
486 Int_t icount=0;
487 TGraph2D *graph = new TGraph2D(nbinx*nbiny);
488 TF1 f1("f1","gaus");
489 for (Int_t ix=0; ix<nbinx;ix++)
490 for (Int_t iy=0; iy<nbiny;iy++){
491 Float_t xcenter = xaxis->GetBinCenter(ix);
492 Float_t ycenter = yaxis->GetBinCenter(iy);
4a9220d4 493 snprintf(name,1000,"%s_%d_%d",his->GetName(), ix,iy);
7cc34f08 494 TH1 *projection = his->ProjectionZ(name,ix-delta0,ix+delta0,iy-delta1,iy+delta1);
495 Float_t stat= 0;
496 if (type==0) stat = projection->GetMean();
497 if (type==1) stat = projection->GetRMS();
498 if (type==2 || type==3){
499 TVectorD vec(3);
500 AliMathBase::LTM((TH1F*)projection,&vec,0.7);
501 if (type==2) stat= vec[1];
502 if (type==3) stat= vec[0];
503 }
504 if (type==4|| type==5){
505 projection->Fit(&f1);
506 if (type==4) stat= f1.GetParameter(1);
507 if (type==5) stat= f1.GetParameter(2);
508 }
509 //printf("%d\t%f\t%f\t%f\n", icount,xcenter, ycenter, stat);
510 graph->SetPoint(icount,xcenter, ycenter, stat);
511 icount++;
512 }
513 return graph;
514}
515
516TGraph * AliComparisonDraw::MakeStat1D(TH3 * his, Int_t delta1, Int_t type){
517 //
518 //
519 //
520 // delta - number of bins to integrate
521 // type - 0 - mean value
522
523 TAxis * xaxis = his->GetXaxis();
524 TAxis * yaxis = his->GetYaxis();
525 // TAxis * zaxis = his->GetZaxis();
526 Int_t nbinx = xaxis->GetNbins();
527 Int_t nbiny = yaxis->GetNbins();
528 char name[1000];
529 Int_t icount=0;
530 TGraph *graph = new TGraph(nbinx);
531 TF1 f1("f1","gaus");
532 for (Int_t ix=0; ix<nbinx;ix++){
533 Float_t xcenter = xaxis->GetBinCenter(ix);
534 // Float_t ycenter = yaxis->GetBinCenter(iy);
4a9220d4 535 snprintf(name,1000,"%s_%d",his->GetName(), ix);
7cc34f08 536 TH1 *projection = his->ProjectionZ(name,ix-delta1,ix+delta1,0,nbiny);
537 Float_t stat= 0;
538 if (type==0) stat = projection->GetMean();
539 if (type==1) stat = projection->GetRMS();
540 if (type==2 || type==3){
541 TVectorD vec(3);
542 AliMathBase::LTM((TH1F*)projection,&vec,0.7);
543 if (type==2) stat= vec[1];
544 if (type==3) stat= vec[0];
545 }
546 if (type==4|| type==5){
547 projection->Fit(&f1);
548 if (type==4) stat= f1.GetParameter(1);
549 if (type==5) stat= f1.GetParameter(2);
550 }
551 //printf("%d\t%f\t%f\t%f\n", icount,xcenter, ycenter, stat);
552 graph->SetPoint(icount,xcenter, stat);
553 icount++;
554 }
555 return graph;
556}
557
558//
559// Make derived plots
560//
561
562void AliComparisonDraw::MakePlots(){
563 //
564 //
565 //
566 AliComparisonDraw * comp=this;
567
568 TFile *fp = new TFile("picutures.root","recreate");
569 TH1F *hiss=0;
570 //TH1F *hism=0;
571 TGraph2D * gr=0, gr2=0;
572 TGraph * gr0 = 0;
573 TCanvas * c = new TCanvas("Phi resol Tan","Phi resol Tan");
574 //
575 //
576 //
577 hiss = comp->MakeResol(comp->fCPtResolTan,1,0);
578 hiss->SetXTitle("Tan(#theta)");
579 hiss->SetYTitle("#sigmap_{t}/p_{t}");
580 hiss->Draw();
581 hiss->Write("CptResolTan");
582 //
583 //
584 hiss = comp->MakeResol(comp->fCPhiResolTan,1,0);
585 c->cd();
586 hiss->SetXTitle("Tan(#theta)");
587 hiss->SetYTitle("#sigma#phi (rad)");
588 hiss->Draw();
589 fp->cd();
590 hiss->Write("PhiResolTan");
591 //
592 hiss = comp->MakeResol(comp->fCTanResolTan,1,0);
593 c->cd();
594 hiss->SetXTitle("Tan(#theta)");
595 hiss->SetYTitle("#sigma#theta (rad)");
596 hiss->Draw();
597 fp->cd();
598 hiss->Write("ThetaResolTan");
599 //
600 //
601 hiss = comp->MakeResol(comp->fCTanResolTan,1,0);
602 c->cd();
603 hiss->SetXTitle("Tan(#theta)");
604 hiss->SetYTitle("#sigmap_{t}/p_{t} ");
605 hiss->Draw();
606 fp->cd();
607 //
608 //
609 //
610 hiss = comp->MakeResol(comp->fTPCSignalNormTan,4,0);
611 hiss->SetXTitle("Tan(#theta)");
612 hiss->SetYTitle("#sigma_{dEdx}");
613 hiss->Draw();
614 fp->cd();
615 hiss->Write("TPCdEdxResolTan");
616 //
617 //
618 //
619 hiss = comp->MakeResol(comp->fTPCSignalNormTan,4,1);
620 hiss->SetXTitle("Tan(#theta)");
621 hiss->SetYTitle("<dEdx>");
622 hiss->Draw();
623 hiss->Write("TPCdEdxMeanTan");
624 //
625 //
626 gr = comp->MakeStat2D(comp->fTPCSignalNormTanSPt,3,1,4);
627 gr->GetXaxis()->SetTitle("Tan(#theta)");
628 gr->GetYaxis()->SetTitle("#sqrt{p_{t}(GeV)}");
629 gr->GetZaxis()->SetTitle("<dEdx>");
630 gr->Draw("colz");
631 gr->GetHistogram()->Write("TPCdEdxMeanTanPt");
632 //
633 //
634 gr = comp->MakeStat2D(comp->fTPCSignalNormTanSPt,3,1,5);
635 gr->GetXaxis()->SetTitle("Tan(#theta)");
636 gr->GetYaxis()->SetTitle("#sqrt{p_{t}(GeV)}");
637 gr->GetZaxis()->SetTitle("#sigma_{dEdx}");
638 gr->Draw("colz");
639 gr->GetHistogram()->Write("TPCdEdxMeanTanPt");
640 //
641 //
642 //
643 comp->fEffTPCTanF->SetXTitle("Tan(#theta)");
644 comp->fEffTPCTanF->SetYTitle("eff_{findable}");
645 comp->fEffTPCTanF->Draw();
646 comp->fEffTPCTanF->Write("EffTanFindable");
647 //
648 //
649 comp->fEffTPCTan->SetXTitle("Tan(#theta)");
650 comp->fEffTPCTan->SetYTitle("eff_{all}");
651 comp->fEffTPCTan->Draw();
652 comp->fEffTPCTan->Write("EffTanAll");
653 //
654 //DCA resolution
655 //
656 gr0 = comp->MakeStat1D(comp->fD0TanSPtB1,2,5);
657 gr0->GetXaxis()->SetTitle("Tan(#theta)");
658 gr0->GetYaxis()->SetTitle("#sigmaDCA (cm)");
659 gPad->Clear();
660 gr0->Draw("al*");
661 gr->GetHistogram()->Write("DCAResolTan");
662 //
663 //
664 //
665 gr = comp->MakeStat2D(comp->fD0TanSPtB1,4,2,5);
666 gr0->GetXaxis()->SetTitle("Tan(#theta)");
667 gr0->GetYaxis()->SetTitle("#sigmaDCA (cm)");
668 gPad->Clear();
669 gr0->Draw("al*");
670 gr->GetHistogram()->Write("DCAResolSPTTan");
671
672 fp->Close();
673
674
675}
676
677
678