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 doeffPi(Int_t pos=1,Float_t prob=0.1,Float_t etaminkp=-0.8,Float_t etamaxkp=0.8);
  14 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);
  15 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);
  16 void fit(TH1D *h,Float_t *a=NULL,char *opt="",char *opt2="",Float_t pt=1.5);
  17 void AddHisto(TH2F *h1,TH2F *h2,Float_t w);
  18
  19 TObject* fContPid1;
  20 TObject* fContPid2;
  21 const Int_t nBinPid = 14; // pt,eta, ptPip, ptPin, PPip, PPin, TOF3sigmaPip, TOF3sigmaPin, isPhiTrue, nsigmaPip, nsigmaPin
  22 // 0.985 < mass < 1.045 (60) and 0 < centrality < 100 (10)
  23 Int_t binPid[nBinPid] = {1/*ptPhi*/,1/*EtaPi*/,20/*pt+*/,20/*pt-*/,5/*P+*/,5/*P-*/,2/*TOFmatch+*/,2/*TOFmatch-*/,2/*istrue*/,4/*Nsigma+*/,4/*Nsigma-*/,1/*DeltaPhi+*/,1/*DeltaPhi-*/,1/*Psi*/};
  24 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};
  25 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};
  26
  27 TF1 *fsign;
  28 TF1 *fall;
  29 TF1 *fback;
  30
  31 Int_t ifunc=0;
  32
  33 Float_t fitmin = 0.3;
  34 Float_t fitmax = 0.7;
  35
  36 Int_t cmin = 1; // min 1
  37 Int_t cmax = 10;//max 10
  38
  39 Float_t weightS = -1.;
  40
  41 Int_t rebinsize = 1;
  42
  43 Int_t parplotted = 2;
  44
  45 Bool_t isMC = kFALSE; // don't change this (is set automatically)
  46 Bool_t selectTrue = kTRUE; // put it to true to remove background (only for MC)
  47 Bool_t keepTrue = kTRUE; // put it to false to fit only background (only for MC)
  48
  49 Bool_t kGoodMatch = kFALSE; // to check good matching
  50
  51 Bool_t kSigma2vs3 = kFALSE; // to check good matching
  52
  53 Bool_t require5sigma = kFALSE; // don't touch this flag
  54
  55 Bool_t bayesVsigma = kFALSE; // only to do checks
  56
  57 Bool_t kTOFmatch = kFALSE; // for combined PID requires TOF matching
  58
  59 Bool_t kOverAll = kFALSE;
  60 Bool_t kOverAllTOFmatch = kFALSE;
  61 Bool_t kOverAll2Sigma = kFALSE;
  62
  63 Bool_t kPid2Sigma = kFALSE;
  64 Bool_t kPid3Sigma = kFALSE;
  65
  66 TH2F *hmatched;
  67 TH2F *htracked;
  68
  69 Bool_t kLoaded=kFALSE;
  70
  71 Int_t LoadLib(){
  72   weightS = -1.;
  73
  74   require5sigma = kFALSE;
  75
  76   if(! kLoaded){
  77     gSystem->Load("libVMC.so");
  78     gSystem->Load("libPhysics.so");
  79     gSystem->Load("libTree.so");
  80     gSystem->Load("libMinuit.so");
  81     gSystem->Load("libSTEERBase.so");
  82     gSystem->Load("libANALYSIS.so");
  83     gSystem->Load("libAOD.so");
  84     gSystem->Load("libESD.so");
  85     gSystem->Load("libANALYSIS.so");
  86     gSystem->Load("libANALYSISalice.so");
  87     gSystem->Load("libCORRFW.so");
  88     gSystem->Load("libNetx.so");
  89     gSystem->Load("libPWGPPpid.so");
  90
  91     TFile *f = new TFile("AnalysisResults.root");
  92     TList *l = (TList *) f->Get("contK0sBayes1");
  93     TList *l2 = (TList *) f->Get("contK0sBayes2");
  94
  95     if(! (l && l2)) return 0;
  96
  97     fContPid1 = (AliPIDperfContainer *) l->FindObject("contPID");
  98     fContPid2 = (AliPIDperfContainer *) l->FindObject("contPID2");
  99     hmatched = (TH2F *) l2->FindObject("hMatchPi");
 100     htracked = (TH2F *) l2->FindObject("hTrackingPi");
 101   }
 102   kLoaded = kTRUE;
 103
 104   // check if MC
 105   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],xmin[11],xmin[12],xmin[13]};
 106   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],xmax[11],xmax[12],xmax[13]};
 107
 108   AliPIDperfContainer *tmp = (AliPIDperfContainer *) fContPid1;
 109   TH1D *h = tmp->GetQA(0, x, x2)->ProjectionX("checkMC");
 110
 111   if(h->GetEntries()) isMC = kTRUE;
 112   else isMC=kFALSE;
 113
 114   if(!isMC){
 115     selectTrue = kFALSE;
 116     keepTrue = kTRUE;
 117   }
 118   else{
 119     printf("MC truth found!!!!!!\nIt is MC!!!!!!");
 120   }
 121
 122   fsign = new TF1("fsign","gaus(0) +0.5*[0]*TMath::Exp(-[3]*TMath::Abs(x-[1]))",fitmin,fitmax);
 123   fback = new TF1("fback","pol2",fitmin,fitmax);
 124   fall = new TF1("fall","gaus(0) +0.5*[0]*TMath::Exp(-[3]*TMath::Abs(x-[1])) + pol2(4)",fitmin,fitmax);
 125
 126   fsign->SetLineColor(2);
 127   fback->SetLineColor(4);
 128
 129   if(kSigma2vs3){
 130     kGoodMatch=kFALSE;
 131     kOverAll = 0;
 132   }
 133
 134   if(bayesVsigma){
 135     kOverAll = 0;
 136     kGoodMatch=kFALSE;
 137     kSigma2vs3=kFALSE;
 138     kTOFmatch=kTRUE;
 139     weightS = -0.7;
 140   }
 141
 142   return 1;
 143 }
 144
 145 void doeffPi(Int_t pos,Float_t prob,Float_t etaminkp,Float_t etamaxkp){
 146   LoadLib();
 147   TH1D *hm = hmatched->ProjectionX("matchingPiEff",cmin,cmax);
 148   TH1D *ht = htracked->ProjectionX("tracking",cmin,cmax);
 149
 150   hm->GetYaxis()->SetTitle("TOF matching eff.");
 151   hm->SetTitle("Using probability as weights");
 152
 153   hm->Sumw2();
 154   ht->Sumw2();
 155
 156   hm->Divide(hm,ht,1,1,"B");
 157
 158   Int_t nptbin = binPid[2];
 159   Float_t minptbin = xmin[2];
 160   Float_t maxptbin = xmax[2];
 161
 162   if(pos == 0){
 163     nptbin = binPid[3];
 164     minptbin = xmin[3];
 165     maxptbin = xmax[3];
 166   }
 167
 168   if(prob > 0.1999|| kPid3Sigma ||kPid2Sigma){
 169     kGoodMatch = kFALSE;
 170     kSigma2vs3 = kFALSE;
 171 //    if(! kOverAll) require5sigma = kTRUE;
 172     if(!isMC) weightS = -0.95;
 173   }
 174
 175   TCanvas *c1 = new TCanvas();
 176   c1->Divide((nptbin+1)/2,2);
 177   TH2F *hh,*hh2;
 178   TH1D *h;
 179   char name[100];
 180   Float_t b[50][3];
 181
 182   Double_t xx[50],yy[50];
 183   Double_t exx[50],eyy[50];
 184
 185   for(Int_t i=0;i < nptbin;i++){
 186     c1->cd(i+1);//->SetLogy();
 187     Float_t ptmin = minptbin+(maxptbin-minptbin)/nptbin*(i);
 188     Float_t ptmax = minptbin+(maxptbin-minptbin)/nptbin*(i+1);
 189
 190     xx[i] = (ptmin+ptmax)/2;
 191     exx[i] = (-ptmin+ptmax)/2;
 192
 193     Float_t pp=0.1;
 194     if(prob < 0.2) pp = 0.;
 195     if(pos) hh=GetHistoPip(ptmin,ptmax,pp,0.0,etaminkp,etamaxkp);
 196     else hh=GetHistoPin(ptmin,ptmax,pp,0.0);
 197     sprintf(name,"TOF matched: %f < p_{T} < %f GeV/#it{c}",ptmin,ptmax);
 198     hh->SetTitle(name);
 199     sprintf(name,"hNoPid%i",i);
 200
 201     pp=prob;
 202     if(prob < 0.2) pp = 0.1;
 203     if(pos) hh2=GetHistoPip(ptmin,ptmax,pp,0.0,etaminkp,etamaxkp);
 204     else hh2=GetHistoPin(ptmin,ptmax,pp,0.0);
 205     AddHisto(hh,hh2,weightS);
 206
 207     h = hh->ProjectionX(name,cmin,cmax);
 208     h->RebinX(rebinsize);
 209     h->Draw("ERR");
 210     h->SetMarkerStyle(24);
 211     b[i][0]=-1;
 212     Int_t ntrial = 0;
 213     Float_t chi2 = 10000;
 214     while(ntrial < 3 && (chi2 > 20 + 1000*selectTrue)){
 215       fit(h,b[i],"WW","",xx[i]);
 216       c1->Update();
 217 //       getchar();
 218       fit(h,b[i],"","",xx[i]);
 219       ntrial++;
 220       chi2 = b[i][2];
 221       printf("chi2 = %f\n",chi2);
 222       c1->Update();
 223 //       getchar();
 224
 225     }
 226
 227     yy[i] = fall->GetParameter(parplotted);
 228     eyy[i] = fall->GetParError(parplotted);
 229   }
 230
 231   TGraphErrors *gpar = new TGraphErrors(nptbin,xx,yy,exx,eyy);
 232   c1->cd(8);
 233 //   gpar->Draw("AP");
 234   gpar->SetMarkerStyle(20);
 235
 236   TCanvas *c2 = new TCanvas();
 237   c2->Divide((nptbin+1)/2,2);
 238   Float_t b2[50][3];
 239
 240   for(Int_t i=0;i < nptbin;i++){
 241     c2->cd(i+1);
 242     Float_t ptmin = minptbin+(maxptbin-minptbin)/nptbin*(i);
 243     Float_t ptmax = minptbin+(maxptbin-minptbin)/nptbin*(i+1);
 244
 245     Float_t pp=prob;
 246     if(prob < 0.2) pp = 0.1;
 247     if(pos) hh=GetHistoPip(ptmin,ptmax,pp,0.0,etaminkp,etamaxkp);
 248     else hh=GetHistoPin(ptmin,ptmax,pp,0.0);
 249     sprintf(name,"P_{TOF} > 0.8: %f < p_{T} < %f GeV/#it{c}",ptmin,ptmax);
 250     hh->SetTitle(name);
 251     sprintf(name,"hPid60_%i",i);
 252     h = hh->ProjectionX(name,cmin,cmax);
 253     h->RebinX(rebinsize);
 254     h->Draw("ERR");
 255     h->SetMarkerStyle(24);
 256     b2[i][0]=-1;
 257     Int_t ntrial = 0;
 258     Float_t chi2 = 10000;
 259     while(ntrial < 3 && (chi2 > 20 + 1000*selectTrue)){
 260       fit(h,b2[i],"WW","",xx[i]);
 261       fit(h,b2[i],"","",xx[i]);
 262       ntrial++;
 263       chi2 = b2[i][2];
 264       printf("chi2 = %f\n",chi2);
 265     }
 266     yy[i] = fall->GetParameter(parplotted);
 267     eyy[i] = fall->GetParError(parplotted);
 268
 269   }
 270
 271   TGraphErrors *gpar2 = new TGraphErrors(nptbin,xx,yy,exx,eyy);
 272   c2->cd(8);
 273 //   gpar2->Draw("AP");
 274   gpar2->SetMarkerStyle(20);
 275
 276   Double_t xpt[50],expt[50],eff[50],efferr[50];
 277   for(Int_t i=0;i<nptbin;i++){
 278     printf("%f +/- %f -  %f +/- %f\n",b[i][0],b[i][1],b2[i][0],b2[i][1]);
 279
 280     Float_t ptmin = minptbin+(maxptbin-minptbin)/nptbin*(i);
 281     Float_t ptmax = minptbin+(maxptbin-minptbin)/nptbin*(i+1);
 282
 283     xpt[i] = (ptmin+ptmax)/2;
 284     expt[i] = (-ptmin+ptmax)/2;
 285     eff[i] = b2[i][0]/(b[i][0]-b2[i][0]*weightS);
 286
 287     //    b[i][0] = b[i][0]-b2[i][0]*weightS;
 288
 289     //    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]));
 290
 291     efferr[i] = 1./(b[i][0]-b2[i][0]*weightS)/(b[i][0]-b2[i][0]*weightS)*TMath::Sqrt(b[i][0]*b[i][0]*b2[i][1]*b2[i][1] + b2[i][0]*b2[i][0]*b[i][1]*b[i][1]);
 292
 293     if(TMath::Abs(efferr[i]) > 1)efferr[i]=1;
 294   }
 295   new TCanvas();
 296   TGraphErrors *geff = new TGraphErrors(nptbin,xpt,eff,expt,efferr);
 297   geff->Draw("AP");
 298
 299   char flag[100];
 300   flag[0] = '\0';
 301
 302   if(isMC){
 303     if(selectTrue) sprintf(flag,"true");
 304     else if(!keepTrue) sprintf(flag,"back");
 305   }
 306
 307   Bool_t kWriteME = kFALSE;
 308
 309   char flag2[100];
 310   flag2[0] = '\0';
 311
 312   char etarange[100];
 313   sprintf(etarange,"_%.1f-%.1f_",etaminkp,etamaxkp);
 314
 315   if(kGoodMatch)
 316     sprintf(flag2,"GM");
 317
 318   if(bayesVsigma)
 319     sprintf(flag2,"BayesVsSigma");
 320
 321   if(kSigma2vs3)
 322     sprintf(flag2,"Sigma2vs3");
 323
 324   if(kOverAll)
 325     sprintf(flag2,"OverAll");
 326   if(kOverAllTOFmatch)
 327     sprintf(flag2,"OverAllTOF");
 328   if(kOverAll2Sigma)
 329     sprintf(flag2,"OverAll2sigma");
 330
 331   if(kPid3Sigma)
 332     sprintf(flag2,"pid3sigma");
 333   if(kPid2Sigma)
 334     sprintf(flag2,"pid2sigma");
 335
 336
 337   if(pos){
 338     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);
 339     else{
 340       sprintf(name,"pionPos%sMatchEff%i_%i%s%s.root",etarange,(cmin-1)*10,cmax*10,flag,flag2);
 341       if(!(kOverAll || bayesVsigma || kGoodMatch || kSigma2vs3)) kWriteME = kTRUE;
 342     }
 343   }
 344   else{
 345     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);
 346     else sprintf(name,"pionNeg%sMatchEff%i_%i%s%s.root",etarange,(cmin-1)*10,cmax*10,flag,flag2);
 347   }
 348
 349   geff->SetTitle("#pi efficiency (from K^{0}_{s});p_{T} (GeV/#it{c};efficiency");
 350   TFile *fout = new TFile(name,"RECREATE");
 351   geff->Write();
 352   if(kWriteME) hm->Write();
 353   fout->Close();
 354
 355   if(kWriteME) hm->Draw("SAME");
 356 }
 357
 358 TH2F *GetHistoPip(Float_t pt,Float_t ptM,Float_t pMinkp,Float_t pMinkn,Float_t etaminkp,Float_t etamaxkp){
 359
 360   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 || kPid3Sigma||kPid2Sigma)+0.001,kTOFmatch+0.001,selectTrue,xmin[9],xmin[10],xmin[11],xmin[12],xmin[13]};
 361   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]};
 362
 363   if(kOverAll){
 364     x[6] = 0.0001;
 365     x2[9] = 5.9;
 366     if(pMinkp > 0.19) x2[9] = 4.9;
 367   }
 368
 369   if(kOverAllTOFmatch && pMinkp > 0.19){
 370     x[6] = 1.0001;
 371     x2[9] = 4.9;
 372   }
 373
 374   if(kOverAll2Sigma && pMinkp > 0.09){
 375     x2[9] = 2.;
 376     x[6] = 1.0001;
 377   }
 378
 379   if(kGoodMatch){
 380     x[6] = 1.0001;
 381     if(pMinkp > 0)
 382       x2[9] = 4.9;
 383
 384   }
 385
 386   if(kTOFmatch){
 387     x[6] = 1.0001;
 388   }
 389
 390   if(kSigma2vs3){
 391     x[6] = 1.0001;
 392     x2[9] = 3;
 393     if(pMinkp > 0)
 394       x2[9] = 2;
 395   }
 396
 397   if(bayesVsigma){
 398     if(pMinkp > 0){
 399       x[4] = 0.2001;
 400       x2[9] = 5;
 401     }
 402     else{
 403       x2[9] = 3;
 404     }
 405
 406
 407   }
 408
 409   if(require5sigma) x2[9] = 4.9;
 410   if(kPid3Sigma && pMinkp>0.09) x2[9] = 2.9;
 411   if(kPid2Sigma && pMinkp>0.09) x2[9] = 1.9;
 412
 413   printf("max sigma = %f\n",x2[9]);
 414
 415   AliPIDperfContainer *tmp = (AliPIDperfContainer *) fContPid1;
 416
 417   TH2F *h = tmp->GetQA(0, x, x2);
 418
 419   h->GetXaxis()->SetTitle("M_{K^{0}_{s}} (GeV/#it{c}^{2})");
 420   h->GetYaxis()->SetTitle("centrality [%]");
 421
 422   return h;
 423 }
 424
 425 TH2F *GetHistoPin(Float_t pt,Float_t ptM,Float_t pMinkn,Float_t pMinkp,Float_t etaminkp,Float_t etamaxkp){
 426
 427   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 || kPid3Sigma|| kPid2Sigma)+0.001,selectTrue,xmin[9],xmin[10],xmin[11],xmin[12],xmin[13]};
 428   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]};
 429
 430   if(kOverAll){
 431     x[7] = 0.0001;
 432     x2[10] = 5.9;
 433     if(pMinkn > 0.19) x2[10] = 4.9;
 434   }
 435
 436   if(kOverAllTOFmatch && pMinkn > 0.19){
 437     x[7] = 1.0001;
 438     x2[10] = 4.9;
 439   }
 440
 441   if(kOverAll2Sigma && pMinkn > 0.09){
 442     x2[10] = 2;
 443     x[7] = 1.0001;
 444   }
 445
 446   if(kGoodMatch){
 447     x[7] = 1.0001;
 448     if(pMinkn > 0)
 449       x2[10] = 4.9;
 450
 451   }
 452
 453   if(kTOFmatch){
 454     x[7] = 1.0001;
 455   }
 456
 457   if(kSigma2vs3){
 458     x[7] = 1.0001;
 459     x2[10] = 3;
 460     if(pMinkn > 0)
 461       x2[10] = 2;
 462   }
 463
 464  if(bayesVsigma){
 465     if(pMinkn > 0){
 466       x[5] = 0.2001;
 467       x2[10] = 5;
 468     }
 469     else{
 470       x2[10] = 3;
 471     }
 472   }
 473
 474   if(require5sigma) x2[10] = 4.9;
 475
 476   if(kPid3Sigma && pMinkn>0.09) x2[10] = 2.9;
 477   if(kPid2Sigma && pMinkn>0.09) x2[10] = 1.9;
 478
 479   printf("max sigma = %f\n",x2[10]);
 480
 481   AliPIDperfContainer *tmp = (AliPIDperfContainer *) fContPid2;
 482
 483   TH2F *h = tmp->GetQA(0, x, x2);
 484
 485   h->GetXaxis()->SetTitle("M_{K^{0}_{s}} (GeV/#it{c}^{2})");
 486   h->GetYaxis()->SetTitle("centrality [%]");
 487
 488   return h;
 489 }
 490
 491 void fit(TH1D *h,Float_t *a,char *opt,char *opt2,Float_t pt){
 492   if(h->Integral(1,h->GetNbinsX()) < 1){
 493     if(a){
 494       a[0]=0.001;
 495       a[1]=1;
 496     }
 497     return;
 498   }
 499
 500
 501  fall->SetParameter(0,100);
 502  fall->SetParameter(1,0.4971);
 503  fall->SetParameter(2,2.89748e-03);
 504  fall->FixParameter(3,230+30/pt);
 505
 506  fall->SetParLimits(0,0.00001,1000000);
 507  fall->SetParLimits(1,0.4965,0.4985);
 508  fall->SetParLimits(2,0.0025,0.005);
 509  //fall->SetParLimits(3,200,350);
 510
 511  fall->ReleaseParameter(4);
 512  fall->ReleaseParameter(5);
 513
 514  if(selectTrue){
 515    fall->FixParameter(4,0);
 516    fall->FixParameter(5,0);
 517    fall->FixParameter(6,0);
 518  }
 519    fall->FixParameter(6,0);
 520
 521
 522  char namenew[100];
 523  sprintf(namenew,"%s_%i",h->GetName(),Int_t(gRandom->Rndm()*10000));
 524  TH1D *h2 = new TH1D(*h);
 525  h2->SetName(namenew);
 526
 527 // Float_t entries = h2->GetBinContent(h2->FindBin(0.497));
 528 //  printf("entries under the peak = %f, pt = %f\n",entries,pt);
 529 //  getchar();
 530
 531  if(pt > 2.5){
 532    if(pt < 2.8) h2->RebinX(2);
 533    else if(pt < 3) h2->RebinX(4);
 534    else h2->RebinX(4);
 535  }
 536
 537  h=h2;
 538
 539  char name[100];
 540  TF1 *ftmp=fall;
 541
 542  TF1 *ftmp2=new TF1(*fsign);
 543  sprintf(name,"fsign%i",ifunc);
 544  ftmp2->SetName(name);
 545
 546  TF1 *ftmp3=new TF1(*fback);
 547  sprintf(name,"ftmp3%i",ifunc);
 548  ftmp3->SetName(name);
 549
 550  ifunc++;
 551
 552  h->Fit(ftmp,opt,opt2,fitmin,fitmax);
 553  h->Draw("ERR");
 554
 555  ftmp2->SetParameter(0,ftmp->GetParameter(0));
 556  ftmp2->SetParameter(1,ftmp->GetParameter(1));
 557  ftmp2->SetParameter(2,ftmp->GetParameter(2));
 558  ftmp2->SetParameter(3,ftmp->GetParameter(3));
 559  ftmp2->Draw("SAME");
 560  ftmp3->SetParameter(0,ftmp->GetParameter(4));
 561  ftmp3->SetParameter(1,ftmp->GetParameter(5));
 562  ftmp3->SetParameter(2,ftmp->GetParameter(6));
 563  ftmp3->Draw("SAME");
 564
 565  Float_t mean = ftmp->GetParameter(1);
 566  Float_t sigma = TMath::Abs(ftmp->GetParameter(2));
 567
 568  Float_t signI = ftmp2->Integral(mean-10*sigma,mean+10*sigma)/h->GetBinWidth(1);
 569  Float_t backI = ftmp3->Integral(mean-3*sigma,mean+3*sigma)/h->GetBinWidth(1);
 570
 571  if(signI < 0) signI = 0;
 572  if(backI < 1) backI = 1;
 573
 574  Float_t errI = TMath::Sqrt(ftmp->GetParError(0)*ftmp->GetParError(0)/(0.001+ftmp->GetParameter(0))/(0.001+ftmp->GetParameter(0)));
 575
 576  printf("signal(5 sigma) = %f +/- %f(fit) +/- %f(stat)\n",signI,errI*signI,TMath::Sqrt(signI));
 577  printf("backgr(3sigma) = %f\n",backI);
 578  printf("significance(3 sigma) = %f\n",signI/sqrt(signI+backI));
 579
 580  if(a){
 581    a[0]=signI;
 582    a[1]=signI*errI*signI*errI + signI;
 583    a[1] = TMath::Sqrt(a[1]);
 584    if(ftmp->GetNDF()) a[2] = ftmp->GetChisquare()/ftmp->GetNDF();
 585
 586
 587    if(selectTrue){
 588      a[0] = h->Integral(1,h->GetNbinsX());
 589      a[1] = TMath::Sqrt(a[0]);
 590    }
 591  }
 592 }
 593
 594 void AddHisto(TH2F *h1,TH2F *h2,Float_t w){
 595   Int_t nbinx = h1->GetNbinsX();
 596   Int_t nbiny = h1->GetNbinsY();
 597
 598   for(Int_t i=1;i<=nbinx;i++){
 599     for(Int_t j=1;j<=nbiny;j++){
 600       Double_t val = h1->GetBinContent(i,j) + h2->GetBinContent(i,j)*w;
 601       Float_t err = TMath::Min(TMath::Sqrt(val),val);
 602       h1->SetBinContent(i,j,val);
 603       h1->SetBinError(i,j,err);
 604     }
 605   }
 606 }