]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGPP/pid/doeffPi.C
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
added a check for TPC gaussianity
[u/mrichter/AliRoot.git] / PWGPP / pid / doeffPi.C
CommitLineData
7484404e 1#include"TF1.h"
2#include"TH1D.h"
3#include"TH2F.h"
4#include"TMath.h"
5#include"TSystem.h"
6#include"TCanvas.h"
7#include"TFile.h"
8#include"TGraphErrors.h"
9#include"AliPIDperfContainer.h"
cd9f95f6 10#include"TRandom.h"
7484404e 11
12Int_t LoadLib();
9c5bbe8a 13void doeffPiUser(Int_t pos,Float_t etaminkp=-0.8,Float_t etamaxkp=0.8);
7484404e 14void doeffPi(Int_t pos=1,Float_t prob=0.1,Float_t etaminkp=-0.8,Float_t etamaxkp=0.8);
15TH2F *GetHistoPip(Float_t pt=1,Float_t ptM=1.1,Float_t pMinkp=0,Float_t pMinkn=0.,Float_t etaminkp=-0.8,Float_t etamaxkp=0.8);
16TH2F *GetHistoPin(Float_t pt=1,Float_t ptM=1.1,Float_t pMinkn=0,Float_t pMinkp=0.,Float_t etaminkp=-0.8,Float_t etamaxkp=0.8);
bf2e05d5 17void fit(TH1D *h,Float_t *a=NULL,char *opt="",char *opt2="",Float_t pt=1.5);
7484404e 18void AddHisto(TH2F *h1,TH2F *h2,Float_t w);
9c5bbe8a 19TH2F *GetHistoUser(Int_t pos=1,Float_t pt=1,Float_t ptM=1.1,Float_t etaminkp=-0.8,Float_t etamaxkp=0.8);
20TH2F *GetHistoPiUser(Int_t pos=1,Float_t pt=1,Float_t ptM=1.1,Float_t etaminkp=-0.8,Float_t etamaxkp=0.8);
21TH2F *GetHistoKaUser(Int_t pos=1,Float_t pt=1,Float_t ptM=1.1,Float_t etaminkp=-0.8,Float_t etamaxkp=0.8);
22TH2F *GetHistoPrUser(Int_t pos=1,Float_t pt=1,Float_t ptM=1.1,Float_t etaminkp=-0.8,Float_t etamaxkp=0.8);
7484404e 23
a8ad4709 24TObject* fContPid1;
25TObject* fContPid2;
9c5bbe8a 26TObject* fContUser1;
27TObject* fContUser2;
a8ad4709 28const Int_t nBinPid = 14; // pt,eta, ptPip, ptPin, PPip, PPin, TOF3sigmaPip, TOF3sigmaPin, isPhiTrue, nsigmaPip, nsigmaPin
29// 0.985 < mass < 1.045 (60) and 0 < centrality < 100 (10)
30Int_t binPid[nBinPid] = {1/*ptPhi*/,8/*EtaPi*/,20/*pt+*/,20/*pt-*/,5/*P+*/,1/*P-*/,2/*TOFmatch+*/,2/*TOFmatch-*/,2/*istrue*/,4/*Nsigma+*/,4/*Nsigma-*/,1/*DeltaPhi+*/,1/*DeltaPhi-*/,1/*Psi*/};
31Float_t xmin[nBinPid] = {1,-0.8,0.3,0.3,0,0,-0.5,-0.5,-0.5,0,0,-TMath::Pi(),-TMath::Pi(),-TMath::Pi()/2};
32Float_t xmax[nBinPid] = {5,0.8,4.3,4.3,1,1,1.5,1.5,1.5,7.5,7.5,TMath::Pi(),TMath::Pi(),TMath::Pi()/2};
33
34TF1 *fsign;
35TF1 *fall;
36TF1 *fback;
37
38Int_t ifunc=0;
39
40Float_t fitmin = 0.3;
41Float_t fitmax = 0.7;
42
08b5b548 43Int_t cmin = 1; // min 1
44Int_t cmax = 10;//max 10
a8ad4709 45
754154d5 46Float_t weightS = -0.9;
a8ad4709 47
7484404e 48Int_t rebinsize = 1;
a8ad4709 49
50Int_t parplotted = 2;
51
52Bool_t isMC = kFALSE; // don't change this (is set automatically)
53Bool_t selectTrue = kTRUE; // put it to true to remove background (only for MC)
54Bool_t keepTrue = kFALSE; // put it to false to fit only background (only for MC)
55
56Bool_t kGoodMatch = kFALSE; // to check good matching
57
58Bool_t kSigma2vs3 = kFALSE; // to check good matching
754154d5 59Bool_t kSigma2vs3TPC = kFALSE; // to check good matching
a8ad4709 60
61Bool_t require5sigma = kFALSE; // don't touch this flag
62
63Bool_t bayesVsigma = kFALSE; // only to do checks
64
65Bool_t kTOFmatch = kFALSE; // for combined PID requires TOF matching
66
08b5b548 67Bool_t kOverAll = kFALSE;
e34b28fe 68Bool_t kOverAllTOFmatch = kFALSE;
08b5b548 69Bool_t kOverAll2Sigma = kFALSE;
e34b28fe 70
08b5b548 71TH2F *hmatched;
72TH2F *htracked;
a8ad4709 73
74Bool_t kLoaded=kFALSE;
7484404e 75
76Int_t LoadLib(){
754154d5 77 weightS = -0.9;
a8ad4709 78
bf2e05d5 79 require5sigma = kFALSE;
80
a8ad4709 81 if(! kLoaded){
82 gSystem->Load("libVMC.so");
83 gSystem->Load("libPhysics.so");
84 gSystem->Load("libTree.so");
85 gSystem->Load("libMinuit.so");
86 gSystem->Load("libSTEERBase.so");
87 gSystem->Load("libANALYSIS.so");
88 gSystem->Load("libAOD.so");
89 gSystem->Load("libESD.so");
90 gSystem->Load("libANALYSIS.so");
91 gSystem->Load("libANALYSISalice.so");
92 gSystem->Load("libCORRFW.so");
93 gSystem->Load("libNetx.so");
94 gSystem->Load("libPWGPPpid.so");
95
96 TFile *f = new TFile("AnalysisResults.root");
a8ad4709 97 TList *l = (TList *) f->Get("contK0sBayes1");
08b5b548 98 TList *l2 = (TList *) f->Get("contK0sBayes2");
7484404e 99
100 if(! (l && l2)) return 0;
101
08b5b548 102 fContPid1 = (AliPIDperfContainer *) l->FindObject("contPID");
103 fContPid2 = (AliPIDperfContainer *) l->FindObject("contPID2");
9c5bbe8a 104 fContUser1 = (AliPIDperfContainer *) l->FindObject("contUserPID");
105 fContUser2 = (AliPIDperfContainer *) l->FindObject("contUserPID2");
08b5b548 106 hmatched = (TH2F *) l2->FindObject("hMatchPi");
107 htracked = (TH2F *) l2->FindObject("hTrackingPi");
a8ad4709 108 }
109 kLoaded = kTRUE;
110
111 // check if MC
112 Float_t x[] = {xmin[0]+0.001,xmin[1]+0.001,xmin[2]+0.001,xmin[3]+0.001,xmin[4]+0.001,xmin[5]+0.001,xmin[6]+0.001,xmin[7]+0.001,1/*trueMC*/,xmin[9],xmin[10]};
113 Float_t x2[] = {xmax[0],xmax[1],xmax[2],xmax[3],xmax[4],xmax[5],xmax[6],xmax[7],xmax[8],xmax[9],xmax[10]};
114
7484404e 115 AliPIDperfContainer *tmp = (AliPIDperfContainer *) fContPid1;
a8ad4709 116 TH1D *h = tmp->GetQA(0, x, x2)->ProjectionX("checkMC");
117
118 if(h->GetEntries()) isMC = kTRUE;
119 else isMC=kFALSE;
120
121 if(!isMC){
122 selectTrue = kFALSE;
123 keepTrue = kTRUE;
124 }
125 else{
126 printf("MC truth found!!!!!!\nIt is MC!!!!!!");
127 }
128
bf2e05d5 129 fsign = new TF1("fsign","gaus(0) +0.5*[0]*TMath::Exp(-[3]*TMath::Abs(x-[1]))",fitmin,fitmax);
130 fback = new TF1("fback","pol2",fitmin,fitmax);
131 fall = new TF1("fall","gaus(0) +0.5*[0]*TMath::Exp(-[3]*TMath::Abs(x-[1])) + pol2(4)",fitmin,fitmax);
a8ad4709 132
133 fsign->SetLineColor(2);
134 fback->SetLineColor(4);
135
136 if(kSigma2vs3){
137 kGoodMatch=kFALSE;
138 kOverAll = 0;
139 }
140
754154d5 141 if(kSigma2vs3TPC){
142 kGoodMatch=kFALSE;
143 kOverAll = 0;
144 }
145
a8ad4709 146 if(bayesVsigma){
147 kOverAll = 0;
148 kGoodMatch=kFALSE;
149 kSigma2vs3=kFALSE;
754154d5 150 kSigma2vs3TPC=kFALSE;
a8ad4709 151 kTOFmatch=kTRUE;
152 weightS = -0.7;
153 }
7484404e 154
155 return 1;
a8ad4709 156}
157
9c5bbe8a 158void doeffPiUser(Int_t pos,Float_t etaminkp,Float_t etamaxkp){
159 Int_t nptbin = binPid[2];
160 Float_t minptbin = xmin[2];
161 Float_t maxptbin = xmax[2];
162
163 TCanvas *c1 = new TCanvas();
164 c1->Divide((nptbin+1)/2,2);
165
166 Double_t xx[50],yyPi[50],yyKa[50],yyPr[50];
167 Double_t exx[50],eyyPi[50],eyyKa[50],eyyPr[50];
168
169 TH2F *hh;
170 TH1D *h;
171
172 Float_t b[100][3];
173 Float_t bPi[100][3];
174 Float_t bKa[100][3];
175 Float_t bPr[100][3];
176
177 char name[100];
178
179 for(Int_t i=0;i < nptbin;i++){
180 c1->cd(i+1);
181 Float_t ptmin = minptbin+(maxptbin-minptbin)/nptbin*(i);
182 Float_t ptmax = minptbin+(maxptbin-minptbin)/nptbin*(i+1);
183
184 xx[i] = (ptmin+ptmax)/2;
185 exx[i] = (-ptmin+ptmax)/2;
186
187 hh=GetHistoUser(pos,ptmin,ptmax,etaminkp,etamaxkp);
188 sprintf(name,"all%i",i);
189 h = hh->ProjectionX(name,cmin,cmax);
190 Int_t ntrial = 0;
191 Float_t chi2 = 10000;
192 while(ntrial < 3 && (chi2 > 20 + 1000*selectTrue)){
193 fit(h,b[i],"WW","",xx[i]);
194 fit(h,b[i],"","",xx[i]);
195 ntrial++;
196 chi2 = b[i][2];
197 }
198 printf("%i) %f +/- %f\n",i,b[i][0],b[i][1]);
199
200 hh=GetHistoPiUser(pos,ptmin,ptmax,etaminkp,etamaxkp);
201 sprintf(name,"pi%i",i);
202 h = hh->ProjectionX(name,cmin,cmax);
203 ntrial = 0;
204 chi2 = 10000;
205 while(ntrial < 3 && (chi2 > 20 + 1000*selectTrue)){
206 fit(h,bPi[i],"WW","",xx[i]);
207 fit(h,bPi[i],"","",xx[i]);
208 ntrial++;
209 chi2 = bPi[i][2];
210 }
211 printf("pi) %f +/- %f\n",bPi[i][0],bPi[i][1]);
212
213 hh=GetHistoKaUser(pos,ptmin,ptmax,etaminkp,etamaxkp);
214 sprintf(name,"ka%i",i);
215 h = hh->ProjectionX(name,cmin,cmax);
216 ntrial = 0;
217 chi2 = 10000;
218 while(ntrial < 3 && (chi2 > 20 + 1000*selectTrue)){
219 fit(h,bKa[i],"WW","",xx[i]);
220 fit(h,bKa[i],"","",xx[i]);
221 ntrial++;
222 chi2 = bKa[i][2];
223 }
224 printf("ka) %f +/- %f\n",bKa[i][0],bKa[i][1]);
225
226 hh=GetHistoPrUser(pos,ptmin,ptmax,etaminkp,etamaxkp);
227 sprintf(name,"pr%i",i);
228 h = hh->ProjectionX(name,cmin,cmax);
229 ntrial = 0;
230 chi2 = 10000;
231 while(ntrial < 3 && (chi2 > 20 + 1000*selectTrue)){
232 fit(h,bPr[i],"WW","",xx[i]);
233 fit(h,bPr[i],"","",xx[i]);
234 ntrial++;
235 chi2 = bPr[i][2];
236 }
237 printf("pr) %f +/- %f\n",bPr[i][0],bPr[i][1]);
238
239 yyPi[i] = bPi[i][0] / b[i][0];
240 yyKa[i] = bKa[i][0] / b[i][0];
241 yyPr[i] = bPr[i][0] / b[i][0];
242
243 eyyPi[i] = bPi[i][1]/bPi[i][0]*yyPi[i];
244 eyyKa[i] = bKa[i][1]/bKa[i][0]*yyKa[i];
245 eyyPr[i] = bPr[i][1]/bPr[i][0]*yyPr[i];
246 }
247
248 /*TCanvas *c2 =*/ new TCanvas();
249 TGraphErrors *gPi = new TGraphErrors(nptbin,xx,yyPi,exx,eyyPi);
250 gPi->Draw("AP");
251 gPi->SetLineColor(4);
252 gPi->SetMarkerColor(4);
253 gPi->SetMarkerStyle(20);
254
255 TGraphErrors *gKa = new TGraphErrors(nptbin,xx,yyKa,exx,eyyKa);
256 gKa->Draw("P");
257 gKa->SetLineColor(1);
258 gKa->SetMarkerColor(1);
259 gKa->SetMarkerStyle(21);
260
261 TGraphErrors *gPr = new TGraphErrors(nptbin,xx,yyPr,exx,eyyPr);
262 gPr->Draw("P");
263 gPr->SetLineColor(2);
264 gPr->SetMarkerColor(2);
265 gPr->SetMarkerStyle(22);
266
267 if(pos) sprintf(name,"k0sUserAnalPos_%3.1f-%3.1f_%i-%i.root",etaminkp,etamaxkp,cmin,cmax);
268 else sprintf(name,"k0sUserAnalNeg_%3.1f-%3.1f_%i-%i.root",etaminkp,etamaxkp,cmin,cmax);
269
270 gPi->SetName("piSelected");
271 gKa->SetName("kaSelected");
272 gPr->SetName("prSelected");
273
274 TFile *fout = new TFile(name,"RECREATE");
275 gPi->Write();
276 gKa->Write();
277 gPr->Write();
278 fout->Close();
279}
280
281TH2F *GetHistoUser(Int_t pos,Float_t pt,Float_t ptM,Float_t etaminkp,Float_t etamaxkp){
282 // Int_t binUser[nBinUser] = {8/*Eta*/,20/*pt*/,2/*istrue*/,4/*whatSelection*/,1/*DeltaPhi*/,1/*Psi*/};
283 Float_t x[] = {etaminkp+0.0001,pt+0.0001,isMC,0.0001,-TMath::Pi(),TMath::Pi(),-TMath::Pi()/2,TMath::Pi()/2};
284 Float_t x2[] = {etamaxkp-0.0001,ptM-0.0001,isMC,2.9999,-TMath::Pi(),TMath::Pi(),-TMath::Pi()/2,TMath::Pi()/2};
285
286 AliPIDperfContainer *tmp = (AliPIDperfContainer *) fContUser1;
287 if(!pos) tmp = (AliPIDperfContainer *) fContUser2;
288
289 TH2F *h = tmp->GetQA(0, x, x2);
290
291 h->GetXaxis()->SetTitle("M_{K^{0}_{s}} (GeV/#it{c}^{2})");
292 h->GetYaxis()->SetTitle("centrality [%]");
293
294 return h;
295
296}
297
298
299TH2F *GetHistoPiUser(Int_t pos,Float_t pt,Float_t ptM,Float_t etaminkp,Float_t etamaxkp){
300 // Int_t binUser[nBinUser] = {8/*Eta*/,20/*pt*/,2/*istrue*/,4/*whatSelection*/,1/*DeltaPhi*/,1/*Psi*/};
301 Float_t x[] = {etaminkp+0.0001,pt+0.0001,isMC,1.0001,-TMath::Pi(),TMath::Pi(),-TMath::Pi()/2,TMath::Pi()/2};
302 Float_t x2[] = {etamaxkp-0.0001,ptM-0.0001,isMC,1.0002,-TMath::Pi(),TMath::Pi(),-TMath::Pi()/2,TMath::Pi()/2};
303
304 AliPIDperfContainer *tmp = (AliPIDperfContainer *) fContUser1;
305 if(!pos) tmp = (AliPIDperfContainer *) fContUser2;
306
307 TH2F *h = tmp->GetQA(0, x, x2);
308
309 h->GetXaxis()->SetTitle("M_{K^{0}_{s}} (GeV/#it{c}^{2})");
310 h->GetYaxis()->SetTitle("centrality [%]");
311
312 return h;
313
314}
315
316
317TH2F *GetHistoKaUser(Int_t pos,Float_t pt,Float_t ptM,Float_t etaminkp,Float_t etamaxkp){
318 // Int_t binUser[nBinUser] = {8/*Eta*/,20/*pt*/,2/*istrue*/,4/*whatSelection*/,1/*DeltaPhi*/,1/*Psi*/};
319 Float_t x[] = {etaminkp+0.0001,pt+0.0001,isMC,2.0001,-TMath::Pi(),TMath::Pi(),-TMath::Pi()/2,TMath::Pi()/2};
320 Float_t x2[] = {etamaxkp-0.0001,ptM-0.0001,isMC,2.0002,-TMath::Pi(),TMath::Pi(),-TMath::Pi()/2,TMath::Pi()/2};
321
322 AliPIDperfContainer *tmp = (AliPIDperfContainer *) fContUser1;
323 if(!pos) tmp = (AliPIDperfContainer *) fContUser2;
324
325 TH2F *h = tmp->GetQA(0, x, x2);
326
327 h->GetXaxis()->SetTitle("M_{K^{0}_{s}} (GeV/#it{c}^{2})");
328 h->GetYaxis()->SetTitle("centrality [%]");
329
330 return h;
331
332}
333
334
335TH2F *GetHistoPrUser(Int_t pos,Float_t pt,Float_t ptM,Float_t etaminkp,Float_t etamaxkp){
336 // Int_t binUser[nBinUser] = {8/*Eta*/,20/*pt*/,2/*istrue*/,4/*whatSelection*/,1/*DeltaPhi*/,1/*Psi*/};
337 Float_t x[] = {etaminkp+0.0001,pt+0.0001,isMC,3.0001,-TMath::Pi(),TMath::Pi(),-TMath::Pi()/2,TMath::Pi()/2};
338 Float_t x2[] = {etamaxkp-0.0001,ptM-0.0001,isMC,3.0002,-TMath::Pi(),TMath::Pi(),-TMath::Pi()/2,TMath::Pi()/2};
339
340 AliPIDperfContainer *tmp = (AliPIDperfContainer *) fContUser1;
341 if(!pos) tmp = (AliPIDperfContainer *) fContUser2;
342
343 TH2F *h = tmp->GetQA(0, x, x2);
344
345 h->GetXaxis()->SetTitle("M_{K^{0}_{s}} (GeV/#it{c}^{2})");
346 h->GetYaxis()->SetTitle("centrality [%]");
347
348 return h;
349
350}
351
7484404e 352void doeffPi(Int_t pos,Float_t prob,Float_t etaminkp,Float_t etamaxkp){
a8ad4709 353 LoadLib();
08b5b548 354 TH1D *hm = hmatched->ProjectionX("matchingPiEff",cmin,cmax);
355 TH1D *ht = htracked->ProjectionX("tracking",cmin,cmax);
356
357 hm->GetYaxis()->SetTitle("TOF matching eff.");
358 hm->SetTitle("Using probability as weights");
359
360 hm->Sumw2();
361 ht->Sumw2();
362
363 hm->Divide(hm,ht,1,1,"B");
a8ad4709 364
365 Int_t nptbin = binPid[2];
366 Float_t minptbin = xmin[2];
367 Float_t maxptbin = xmax[2];
368
369 if(pos == 0){
370 nptbin = binPid[3];
371 minptbin = xmin[3];
372 maxptbin = xmax[3];
373 }
374
375 if(prob > 0.1999){
376 kGoodMatch = kFALSE;
377 kSigma2vs3 = kFALSE;
754154d5 378 kSigma2vs3TPC = kFALSE;
a8ad4709 379 if(! kOverAll) require5sigma = kTRUE;
754154d5 380 if(!isMC) weightS = -0.9;
a8ad4709 381 }
382
7484404e 383 TCanvas *c1 = new TCanvas();
384 c1->Divide((nptbin+1)/2,2);
385 TH2F *hh,*hh2;
386 TH1D *h;
a8ad4709 387 char name[100];
388 Float_t b[50][3];
389
390 Double_t xx[50],yy[50];
391 Double_t exx[50],eyy[50];
392
393 for(Int_t i=0;i < nptbin;i++){
7484404e 394 c1->cd(i+1);//->SetLogy();
a8ad4709 395 Float_t ptmin = minptbin+(maxptbin-minptbin)/nptbin*(i);
396 Float_t ptmax = minptbin+(maxptbin-minptbin)/nptbin*(i+1);
397
398 xx[i] = (ptmin+ptmax)/2;
399 exx[i] = (-ptmin+ptmax)/2;
400
401 Float_t pp=0.1;
402 if(prob < 0.2) pp = 0.;
403 if(pos) hh=GetHistoPip(ptmin,ptmax,pp,0.0,etaminkp,etamaxkp);
404 else hh=GetHistoPin(ptmin,ptmax,pp,0.0);
405 sprintf(name,"TOF matched: %f < p_{T} < %f GeV/#it{c}",ptmin,ptmax);
406 hh->SetTitle(name);
407 sprintf(name,"hNoPid%i",i);
408
409 pp=prob;
410 if(prob < 0.2) pp = 0.1;
411 if(pos) hh2=GetHistoPip(ptmin,ptmax,pp,0.0,etaminkp,etamaxkp);
412 else hh2=GetHistoPin(ptmin,ptmax,pp,0.0);
413 AddHisto(hh,hh2,weightS);
414
415 h = hh->ProjectionX(name,cmin,cmax);
416 h->RebinX(rebinsize);
417 h->Draw("ERR");
418 h->SetMarkerStyle(24);
419 b[i][0]=-1;
420 Int_t ntrial = 0;
421 Float_t chi2 = 10000;
08b5b548 422 while(ntrial < 3 && (chi2 > 20 + 1000*selectTrue)){
bf2e05d5 423 fit(h,b[i],"WW","",xx[i]);
a8ad4709 424 c1->Update();
425// getchar();
bf2e05d5 426 fit(h,b[i],"","",xx[i]);
a8ad4709 427 ntrial++;
428 chi2 = b[i][2];
429 printf("chi2 = %f\n",chi2);
430 c1->Update();
431// getchar();
432
433 }
434
435 yy[i] = fall->GetParameter(parplotted);
436 eyy[i] = fall->GetParError(parplotted);
437 }
438
439 TGraphErrors *gpar = new TGraphErrors(nptbin,xx,yy,exx,eyy);
7484404e 440 c1->cd(8);
08b5b548 441// gpar->Draw("AP");
a8ad4709 442 gpar->SetMarkerStyle(20);
443
444 TCanvas *c2 = new TCanvas();
445 c2->Divide((nptbin+1)/2,2);
446 Float_t b2[50][3];
447
448 for(Int_t i=0;i < nptbin;i++){
449 c2->cd(i+1);
450 Float_t ptmin = minptbin+(maxptbin-minptbin)/nptbin*(i);
451 Float_t ptmax = minptbin+(maxptbin-minptbin)/nptbin*(i+1);
452
453 Float_t pp=prob;
454 if(prob < 0.2) pp = 0.1;
455 if(pos) hh=GetHistoPip(ptmin,ptmax,pp,0.0,etaminkp,etamaxkp);
456 else hh=GetHistoPin(ptmin,ptmax,pp,0.0);
457 sprintf(name,"P_{TOF} > 0.8: %f < p_{T} < %f GeV/#it{c}",ptmin,ptmax);
458 hh->SetTitle(name);
459 sprintf(name,"hPid60_%i",i);
460 h = hh->ProjectionX(name,cmin,cmax);
461 h->RebinX(rebinsize);
462 h->Draw("ERR");
463 h->SetMarkerStyle(24);
464 b2[i][0]=-1;
465 Int_t ntrial = 0;
466 Float_t chi2 = 10000;
08b5b548 467 while(ntrial < 3 && (chi2 > 20 + 1000*selectTrue)){
bf2e05d5 468 fit(h,b2[i],"WW","",xx[i]);
469 fit(h,b2[i],"","",xx[i]);
a8ad4709 470 ntrial++;
471 chi2 = b2[i][2];
472 printf("chi2 = %f\n",chi2);
473 }
474 yy[i] = fall->GetParameter(parplotted);
475 eyy[i] = fall->GetParError(parplotted);
476
477 }
478
479 TGraphErrors *gpar2 = new TGraphErrors(nptbin,xx,yy,exx,eyy);
480 c2->cd(8);
08b5b548 481// gpar2->Draw("AP");
a8ad4709 482 gpar2->SetMarkerStyle(20);
483
484 Double_t xpt[50],expt[50],eff[50],efferr[50];
485 for(Int_t i=0;i<nptbin;i++){
486 printf("%f +/- %f - %f +/- %f\n",b[i][0],b[i][1],b2[i][0],b2[i][1]);
487
488 Float_t ptmin = minptbin+(maxptbin-minptbin)/nptbin*(i);
489 Float_t ptmax = minptbin+(maxptbin-minptbin)/nptbin*(i+1);
490
491 xpt[i] = (ptmin+ptmax)/2;
492 expt[i] = (-ptmin+ptmax)/2;
493 eff[i] = b2[i][0]/(b[i][0]-b2[i][0]*weightS);
494
495 b[i][0] = b[i][0]-b2[i][0]*weightS;
496
497 efferr[i] = TMath::Sqrt(b[i][1]*b[i][1]/b[i][0]/b[i][0] + b2[i][1]*b2[i][1]/b2[i][0]/b2[i][0])*(b2[i][0]+b2[i][1])*(1+weightS*(b2[i][0]-b2[i][1])/b[i][0])/b[i][0];//*(1-eff[i]);//der2*der2*(b[i][1]*b[i][1] - b2[i][1]*b2[i][1]));
498
499 if(TMath::Abs(efferr[i]) > 1)efferr[i]=1;
500 }
501 new TCanvas();
502 TGraphErrors *geff = new TGraphErrors(nptbin,xpt,eff,expt,efferr);
503 geff->Draw("AP");
504
505 char flag[100];
7484404e 506 flag[0] = '\0';
a8ad4709 507
508 if(isMC){
509 if(selectTrue) sprintf(flag,"true");
510 else if(!keepTrue) sprintf(flag,"back");
511 }
512
08b5b548 513 Bool_t kWriteME = kFALSE;
514
a8ad4709 515 char flag2[100];
7484404e 516 flag2[0] = '\0';
a8ad4709 517
518 char etarange[100];
519 sprintf(etarange,"_%.1f-%.1f_",etaminkp,etamaxkp);
520
521 if(kGoodMatch)
522 sprintf(flag2,"GM");
523
524 if(bayesVsigma)
525 sprintf(flag2,"BayesVsSigma");
526
527 if(kSigma2vs3)
528 sprintf(flag2,"Sigma2vs3");
529
754154d5 530 if(kSigma2vs3TPC)
531 sprintf(flag2,"Sigma2vs3TPC");
532
a8ad4709 533 if(kOverAll)
534 sprintf(flag2,"OverAll");
e34b28fe 535 if(kOverAllTOFmatch)
536 sprintf(flag2,"OverAllTOF");
537 if(kOverAll2Sigma)
538 sprintf(flag2,"OverAll2sigma");
a8ad4709 539
540 if(pos){
541 if(prob >=0.2) sprintf(name,"pionPos%sP%iEff%i_%i%s%s.root",etarange,Int_t(prob*100),(cmin-1)*10,cmax*10,flag,flag2);
08b5b548 542 else{
543 sprintf(name,"pionPos%sMatchEff%i_%i%s%s.root",etarange,(cmin-1)*10,cmax*10,flag,flag2);
754154d5 544 if(!(kOverAll || bayesVsigma || kGoodMatch || kSigma2vs3 || kSigma2vs3TPC)) kWriteME = kTRUE;
08b5b548 545 }
a8ad4709 546 }
547 else{
548 if(prob >=0.2) sprintf(name,"pionNeg%sP%iEff%i_%i%s%s.root",etarange,Int_t(prob*100),(cmin-1)*10,cmax*10,flag,flag2);
549 else sprintf(name,"pionNeg%sMatchEff%i_%i%s%s.root",etarange,(cmin-1)*10,cmax*10,flag,flag2);
550 }
551
08b5b548 552 geff->SetTitle("#pi efficiency (from K^{0}_{s});p_{T} (GeV/#it{c};efficiency");
a8ad4709 553 TFile *fout = new TFile(name,"RECREATE");
554 geff->Write();
08b5b548 555 if(kWriteME) hm->Write();
a8ad4709 556 fout->Close();
557
08b5b548 558 if(kWriteME) hm->Draw("SAME");
a8ad4709 559}
560
7484404e 561TH2F *GetHistoPip(Float_t pt,Float_t ptM,Float_t pMinkp,Float_t pMinkn,Float_t etaminkp,Float_t etamaxkp){
a8ad4709 562
563 Float_t x[] = {xmin[0]+0.001,etaminkp+0.001,pt+0.001,xmin[3]+0.001,pMinkp+0.001,pMinkn+0.001,(pMinkp>0.09)+0.001,kTOFmatch+0.001,selectTrue,xmin[9],xmin[10],xmin[11],xmin[12],xmin[13]};
564 Float_t x2[] = {xmax[0],etamaxkp-0.001,ptM-0.001,xmax[3],xmax[4],xmax[5],xmax[6],xmax[7],keepTrue,xmax[9],xmax[10],xmax[11],xmax[12],xmax[13]};
565
e34b28fe 566 if(kOverAll){
567 x[6] = 0.0001;
568 x2[9] = 5.9;
569 if(pMinkp > 0.19) x2[9] = 4.9;
570 }
571
572 if(kOverAllTOFmatch && pMinkp > 0.19){
573 x[6] = 1.0001;
574 x2[9] = 4.9;
575 }
576
577 if(kOverAll2Sigma && pMinkp > 0.09){
08b5b548 578 x2[9] = 2.;
e34b28fe 579 x[6] = 1.0001;
580 }
581
a8ad4709 582 if(kGoodMatch){
583 x[6] = 1.0001;
584 if(pMinkp > 0)
585 x2[9] = 4.9;
586
587 }
588
589 if(kTOFmatch){
590 x[6] = 1.0001;
591 }
592
593 if(kSigma2vs3){
594 x[6] = 1.0001;
595 x2[9] = 3;
596 if(pMinkp > 0)
597 x2[9] = 2;
598 }
599
754154d5 600 if(kSigma2vs3TPC){
601 x[6] = 0.0001;
602 x2[6] = 0.0002;
603 x2[9] = 3;
604 if(pMinkp > 0)
605 x2[9] = 2;
606 }
607
a8ad4709 608 if(bayesVsigma){
609 if(pMinkp > 0){
610 x[4] = 0.2001;
611 x2[9] = 5;
612 }
613 else{
614 x2[9] = 3;
615 }
616
617
618 }
619
620 if(require5sigma) x2[9] = 4.9;
621
7484404e 622 AliPIDperfContainer *tmp = (AliPIDperfContainer *) fContPid1;
a8ad4709 623
624 TH2F *h = tmp->GetQA(0, x, x2);
625
626 h->GetXaxis()->SetTitle("M_{K^{0}_{s}} (GeV/#it{c}^{2})");
627 h->GetYaxis()->SetTitle("centrality [%]");
628
629 return h;
630}
631
7484404e 632TH2F *GetHistoPin(Float_t pt,Float_t ptM,Float_t pMinkn,Float_t pMinkp,Float_t etaminkp,Float_t etamaxkp){
a8ad4709 633
634 Float_t x[] = {xmin[0]+0.001,etaminkp+0.001,xmin[2]+0.001,pt+0.001,pMinkp+0.001,pMinkn+0.001,kTOFmatch+0.001,(pMinkn>0.09)+0.001,selectTrue,xmin[9],xmin[10],xmin[11],xmin[12],xmin[13]};
635 Float_t x2[] = {xmax[0],etamaxkp-0.001,xmax[2],ptM-0.001,xmax[4],xmax[5],xmax[6],xmax[7],keepTrue,xmax[9],xmax[10],xmax[11],xmax[12],xmax[13]};
636
e34b28fe 637 if(kOverAll){
638 x[7] = 0.0001;
639 x2[10] = 5.9;
640 if(pMinkn > 0.19) x2[10] = 4.9;
641 }
642
643 if(kOverAllTOFmatch && pMinkn > 0.19){
644 x[7] = 1.0001;
645 x2[10] = 4.9;
646 }
647
648 if(kOverAll2Sigma && pMinkn > 0.09){
649 x2[10] = 2;
650 x[7] = 1.0001;
651 }
652
a8ad4709 653 if(kGoodMatch){
654 x[7] = 1.0001;
655 if(pMinkn > 0)
656 x2[10] = 4.9;
657
658 }
659
660 if(kTOFmatch){
661 x[7] = 1.0001;
662 }
663
664 if(kSigma2vs3){
665 x[7] = 1.0001;
666 x2[10] = 3;
667 if(pMinkn > 0)
668 x2[10] = 2;
669 }
670
754154d5 671 if(kSigma2vs3TPC){
672 x[7] = 0.0001;
673 x2[7] = 0.0002;
674 x2[10] = 3;
675 if(pMinkn > 0)
676 x2[10] = 2;
677 }
678
a8ad4709 679 if(bayesVsigma){
680 if(pMinkn > 0){
681 x[5] = 0.2001;
682 x2[10] = 5;
683 }
684 else{
685 x2[10] = 3;
686 }
687 }
688
689 if(require5sigma) x2[10] = 4.9;
690
7484404e 691 AliPIDperfContainer *tmp = (AliPIDperfContainer *) fContPid2;
a8ad4709 692
693 TH2F *h = tmp->GetQA(0, x, x2);
694
695 h->GetXaxis()->SetTitle("M_{K^{0}_{s}} (GeV/#it{c}^{2})");
696 h->GetYaxis()->SetTitle("centrality [%]");
697
698 return h;
699}
700
bf2e05d5 701void fit(TH1D *h,Float_t *a,char *opt,char *opt2,Float_t pt){
a8ad4709 702 if(h->Integral(1,h->GetNbinsX()) < 1){
703 if(a){
704 a[0]=0.001;
705 a[1]=1;
706 }
707 return;
708 }
709
710
711 fall->SetParameter(0,100);
712 fall->SetParameter(1,0.4971);
bf2e05d5 713 fall->SetParameter(2,2.89748e-03);
714 fall->FixParameter(3,230+30/pt);
a8ad4709 715
bf2e05d5 716 fall->SetParLimits(0,0.00001,1000000);
a8ad4709 717 fall->SetParLimits(1,0.4965,0.4985);
bf2e05d5 718 fall->SetParLimits(2,0.0025,0.005);
719 //fall->SetParLimits(3,200,350);
a8ad4709 720
721 fall->ReleaseParameter(4);
722 fall->ReleaseParameter(5);
723
724 if(selectTrue){
725 fall->FixParameter(4,0);
726 fall->FixParameter(5,0);
bf2e05d5 727 fall->FixParameter(6,0);
a8ad4709 728 }
729
cd9f95f6 730 char namenew[100];
731 sprintf(namenew,"%s_%i",h->GetName(),Int_t(gRandom->Rndm()*10000));
732 TH1D *h2 = new TH1D(*h);
733 h2->SetName(namenew);
734
9c5bbe8a 735 // Float_t entries = h2->GetBinContent(h2->FindBin(0.497));
cd9f95f6 736// printf("entries under the peak = %f, pt = %f\n",entries,pt);
737// getchar();
738
739 if(pt > 2.5){
740 if(pt < 2.8) h2->RebinX(2);
741 else if(pt < 3) h2->RebinX(4);
742 else h2->RebinX(10);
743 }
744
745 h=h2;
746
a8ad4709 747 char name[100];
748 TF1 *ftmp=fall;
749
750 TF1 *ftmp2=new TF1(*fsign);
751 sprintf(name,"fsign%i",ifunc);
752 ftmp2->SetName(name);
753
754 TF1 *ftmp3=new TF1(*fback);
755 sprintf(name,"ftmp3%i",ifunc);
756 ftmp3->SetName(name);
757
758 ifunc++;
759
760 h->Fit(ftmp,opt,opt2,fitmin,fitmax);
761 h->Draw("ERR");
762
763 ftmp2->SetParameter(0,ftmp->GetParameter(0));
764 ftmp2->SetParameter(1,ftmp->GetParameter(1));
765 ftmp2->SetParameter(2,ftmp->GetParameter(2));
766 ftmp2->SetParameter(3,ftmp->GetParameter(3));
767 ftmp2->Draw("SAME");
768 ftmp3->SetParameter(0,ftmp->GetParameter(4));
769 ftmp3->SetParameter(1,ftmp->GetParameter(5));
bf2e05d5 770 ftmp3->SetParameter(2,ftmp->GetParameter(6));
a8ad4709 771 ftmp3->Draw("SAME");
772
773 Float_t mean = ftmp->GetParameter(1);
bf2e05d5 774 Float_t sigma = TMath::Abs(ftmp->GetParameter(2));
a8ad4709 775
776 Float_t signI = ftmp2->Integral(mean-10*sigma,mean+10*sigma)/h->GetBinWidth(1);
777 Float_t backI = ftmp3->Integral(mean-3*sigma,mean+3*sigma)/h->GetBinWidth(1);
778
779 Float_t errI = TMath::Sqrt(ftmp->GetParError(0)*ftmp->GetParError(0)/(0.001+ftmp->GetParameter(0))/(0.001+ftmp->GetParameter(0)));
780
781 printf("signal(5 sigma) = %f +/- %f(fit) +/- %f(stat)\n",signI,errI*signI,TMath::Sqrt(signI));
782 printf("backgr(3sigma) = %f\n",backI);
783 printf("significance(3 sigma) = %f\n",signI/sqrt(signI+backI));
784
785 if(a){
786 a[0]=signI;
787 a[1]=signI*errI*signI*errI + signI;
788 a[1] = TMath::Sqrt(a[1]);
789 if(ftmp->GetNDF()) a[2] = ftmp->GetChisquare()/ftmp->GetNDF();
790
791
792 if(selectTrue){
793 a[0] = h->Integral(1,h->GetNbinsX());
794 a[1] = TMath::Sqrt(a[0]);
795 }
796 }
797}
798
7484404e 799void AddHisto(TH2F *h1,TH2F *h2,Float_t w){
a8ad4709 800 Int_t nbinx = h1->GetNbinsX();
801 Int_t nbiny = h1->GetNbinsY();
802
803 for(Int_t i=1;i<=nbinx;i++){
804 for(Int_t j=1;j<=nbiny;j++){
7484404e 805 Double_t val = h1->GetBinContent(i,j) + h2->GetBinContent(i,j)*w;
a8ad4709 806 Float_t err = TMath::Min(TMath::Sqrt(val),val);
807 h1->SetBinContent(i,j,val);
808 h1->SetBinError(i,j,err);
809 }
810 }
811}