Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-02-14 13:08:57

0001 #include <cmath>
0002 #include "TH1D.h"
0003 #include "TProfile2D.h"
0004 
0005 TH1D* projectProfile2DAlongX(TProfile2D* prof2d) {
0006 
0007   TH1D* res=nullptr;
0008 
0009   if(prof2d) {
0010     char name[200];
0011     sprintf(name,"%s_proj",prof2d->GetName());
0012     res = new TH1D(name,prof2d->GetTitle(),prof2d->GetNbinsY(),prof2d->GetYaxis()->GetXmin(),prof2d->GetYaxis()->GetXmax());
0013     res->SetDirectory(nullptr);
0014     res->Sumw2();
0015     for(int iy=1;iy<prof2d->GetNbinsY()+1;++iy) {
0016       double sum=0.;
0017       double sumsq=0.;
0018       double nevt=0.;
0019       for(int ix=1;ix<prof2d->GetNbinsX()+1;++ix) {
0020     const int ibin = prof2d->GetBin(ix,iy);
0021     sum += prof2d->GetBinContent(ibin)*prof2d->GetBinEntries(ibin);
0022     sumsq += prof2d->GetBinError(ibin)*prof2d->GetBinError(ibin)*prof2d->GetBinEntries(ibin)*prof2d->GetBinEntries(ibin)+
0023       prof2d->GetBinContent(ibin)*prof2d->GetBinContent(ibin)*prof2d->GetBinEntries(ibin);
0024     nevt += prof2d->GetBinEntries(ibin);
0025       }
0026       double mean = nevt==0 ? 0: sum/nevt;
0027       double meansq = nevt==0 ? 0: sumsq/nevt;
0028       double err = meansq >= mean*mean ? sqrt(meansq-mean*mean) : 0;
0029       err = nevt==0 ? 0 : err/sqrt(nevt);
0030       res->SetBinContent(iy,mean);
0031       res->SetBinError(iy,err);
0032     }
0033   }
0034   
0035   return res;
0036 }
0037 
0038 TH1D* projectProfile2DAlongY(TProfile2D* prof2d) {
0039 
0040   TH1D* res=nullptr;
0041 
0042   if(prof2d) {
0043     char name[200];
0044     sprintf(name,"%s_proj",prof2d->GetName());
0045     res = new TH1D(name,prof2d->GetTitle(),prof2d->GetNbinsX(),prof2d->GetXaxis()->GetXmin(),prof2d->GetXaxis()->GetXmax());
0046     res->SetDirectory(nullptr);
0047     res->Sumw2();
0048     for(int ix=1;ix<prof2d->GetNbinsX()+1;++ix) {
0049       double sum=0.;
0050       double sumsq=0.;
0051       double nevt=0.;
0052       for(int iy=1;iy<prof2d->GetNbinsY()+1;++iy) {
0053     const int ibin = prof2d->GetBin(ix,iy);
0054     sum += prof2d->GetBinContent(ibin)*prof2d->GetBinEntries(ibin);
0055     sumsq += prof2d->GetBinError(ibin)*prof2d->GetBinError(ibin)*prof2d->GetBinEntries(ibin)*prof2d->GetBinEntries(ibin)+
0056       prof2d->GetBinContent(ibin)*prof2d->GetBinContent(ibin)*prof2d->GetBinEntries(ibin);
0057     nevt += prof2d->GetBinEntries(ibin);
0058       }
0059       double mean = nevt==0 ? 0: sum/nevt;
0060       double meansq = nevt==0 ? 0: sumsq/nevt;
0061       double err = meansq >= mean*mean ? sqrt(meansq-mean*mean) : 0;
0062       err = nevt==0 ? 0 : err/sqrt(nevt);
0063       res->SetBinContent(ix,mean);
0064       res->SetBinError(ix,err);
0065     }
0066   }
0067   
0068   return res;
0069 }
0070