PWGPP/pid/doeffPi.C

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