PFClient.cc

Go to the documentation of this file.

#include "DQMOffline/PFTau/plugins/PFClient.h"

#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"


#include "DQMServices/Core/interface/DQMStore.h"
#include "DQMServices/Core/interface/MonitorElement.h"
#include "FWCore/ServiceRegistry/interface/Service.h"


//
// -- Constructor
//
PFClient::PFClient(const edm::ParameterSet& parameterSet)  
{
  folderNames_ = parameterSet.getParameter< std::vector<std::string> >( "FolderNames" );
  histogramNames_  = parameterSet.getParameter< std::vector<std::string> >( "HistogramNames" );
  efficiencyFlag_ =  parameterSet.getParameter< bool> ("CreateEfficiencyPlots" );
  effHistogramNames_  = parameterSet.getParameter< std::vector<std::string> >( "HistogramNamesForEfficiencyPlots" );
  projectionHistogramNames_  = parameterSet.getParameter< std::vector<std::string> >( "HistogramNamesForProjectionPlots" );
  profileFlag_ =  parameterSet.getParameter< bool> ("CreateProfilePlots" );
  profileHistogramNames_  = parameterSet.getParameter< std::vector<std::string> >( "HistogramNamesForProfilePlots" );
}


//
// -- EndJobBegin Run
// 
void PFClient::dqmEndJob(DQMStore::IBooker& ibooker, DQMStore::IGetter& igetter)
{
  doSummaries(ibooker, igetter); 
  doProjection(ibooker, igetter);
  if (efficiencyFlag_) doEfficiency(ibooker, igetter);
  if (profileFlag_) doProfiles(ibooker, igetter);
}


//
// -- Create Summaries
//
void PFClient::doSummaries(DQMStore::IBooker& ibooker, DQMStore::IGetter& igetter) {

  for (std::vector<std::string>::const_iterator ifolder = folderNames_.begin(); ifolder != folderNames_.end(); ifolder++) {
    std::string path  = "ParticleFlow/"+(*ifolder);

    for (std::vector<std::string>::const_iterator ihist = histogramNames_.begin(); ihist != histogramNames_.end(); ihist++) {
      std::string hname = (*ihist);
      createResolutionPlots(ibooker, igetter, path, hname);
    }
  }
}


//
// -- Create Projection
//
void PFClient::doProjection(DQMStore::IBooker& ibooker, DQMStore::IGetter& igetter) {

  for (std::vector<std::string>::const_iterator ifolder = folderNames_.begin();
                                    ifolder != folderNames_.end(); ifolder++) {
    std::string path  = "ParticleFlow/"+(*ifolder);

    for (std::vector<std::string>::const_iterator ihist = projectionHistogramNames_.begin();
       ihist != projectionHistogramNames_.end(); ihist++) {
      std::string hname = (*ihist);
      createProjectionPlots(ibooker, igetter, path, hname);
    }
  }
}


//
// -- Create Profile
//
void PFClient::doProfiles(DQMStore::IBooker& ibooker, DQMStore::IGetter& igetter) {

  for (std::vector<std::string>::const_iterator ifolder = folderNames_.begin();
                                    ifolder != folderNames_.end(); ifolder++) {
    std::string path  = "ParticleFlow/"+(*ifolder);

    for (std::vector<std::string>::const_iterator ihist = profileHistogramNames_.begin();
       ihist != profileHistogramNames_.end(); ihist++) {
      std::string hname = (*ihist);
      createProfilePlots(ibooker, igetter, path, hname);
    }
  }
}


//
// -- Create Efficiency
//
void PFClient::doEfficiency(DQMStore::IBooker& ibooker, DQMStore::IGetter& igetter) {
  for (std::vector<std::string>::const_iterator ifolder = folderNames_.begin();
                                    ifolder != folderNames_.end(); ifolder++) {
    std::string path  = "ParticleFlow/"+(*ifolder);

    for (std::vector<std::string>::const_iterator ihist = effHistogramNames_.begin();
     ihist != effHistogramNames_.end(); ihist++) {
      std::string hname = (*ihist);
      createEfficiencyPlots(ibooker, igetter, path, hname);
    }
  }
}


//
// -- Create Resolution Plots
//
void PFClient::createResolutionPlots(DQMStore::IBooker& ibooker, DQMStore::IGetter& igetter, std::string& folder, std::string& name) {     
  MonitorElement* me = igetter.get(folder+"/"+name);

  if (!me) return;

  MonitorElement* me_average; 
  MonitorElement* me_rms; 
  MonitorElement* me_mean; 
  MonitorElement* me_sigma; 

  if ( (me->kind() == MonitorElement::DQM_KIND_TH2F) ||
       (me->kind() == MonitorElement::DQM_KIND_TH2S) ||
       (me->kind() == MonitorElement::DQM_KIND_TH2D) ) {
    TH2* th = me->getTH2F();
    size_t nbinx = me->getNbinsX();
    size_t nbiny = me->getNbinsY();
    
    float ymin = th->GetYaxis()->GetXmin();
    float ymax = th->GetYaxis()->GetXmax();
    std::string xtit = th->GetXaxis()->GetTitle();
    std::string ytit = th->GetYaxis()->GetTitle();
    float* xbins = new float[nbinx+1];
    for (size_t ix = 1; ix < nbinx+1; ++ix) {
       xbins[ix-1] = th->GetXaxis()->GetBinLowEdge(ix);
       if (ix == nbinx) xbins[ix] = th->GetXaxis()->GetBinUpEdge(ix);
    }
    std::string tit_new = ";"+xtit+";"+ytit;
    ibooker.setCurrentFolder(folder);
    MonitorElement* me_slice = ibooker.book1D("PFlowSlice","PFlowSlice",nbiny,ymin,ymax); 

    tit_new = ";"+xtit+";Average_"+ytit; 
    me_average = ibooker.book1D("average_"+name,tit_new, nbinx, xbins); 
    me_average->setEfficiencyFlag();
    tit_new = ";"+xtit+";RMS_"+ytit; 
    me_rms     = ibooker.book1D("rms_"+name,tit_new, nbinx, xbins); 
    me_rms->setEfficiencyFlag();
    tit_new = ";"+xtit+";Mean_"+ytit; 
    me_mean    = ibooker.book1D("mean_"+name,tit_new, nbinx, xbins); 
    me_mean->setEfficiencyFlag();
    tit_new = ";"+xtit+";Sigma_"+ytit;               
    me_sigma   = ibooker.book1D("sigma_"+name,tit_new, nbinx, xbins); 
    me_sigma->setEfficiencyFlag();

    double  average, rms, mean, sigma;

    for (size_t ix = 1; ix < nbinx+1; ++ix) {
      me_slice->Reset();
      for (size_t iy = 1; iy < nbiny+1; ++iy) {
    me_slice->setBinContent(iy,th->GetBinContent(ix,iy)); 
      }
      getHistogramParameters(me_slice, average, rms, mean, sigma);
      me_average->setBinContent(ix,average);
      me_rms->setBinContent(ix,rms);
      me_mean->setBinContent(ix,mean);
      me_sigma->setBinContent(ix,sigma);
    }
    if (me_slice) igetter.removeElement(me_slice->getName()); //?
    delete [] xbins;
  }
}


//
// -- Create Projection Plots
//
void PFClient::createProjectionPlots(DQMStore::IBooker& ibooker, DQMStore::IGetter& igetter, std::string& folder, std::string& name) {     
  MonitorElement* me = igetter.get(folder+"/"+name);
  if (!me) return;

  MonitorElement* projection = 0;

  if ( (me->kind() == MonitorElement::DQM_KIND_TH2F) ||
       (me->kind() == MonitorElement::DQM_KIND_TH2S) ||
       (me->kind() == MonitorElement::DQM_KIND_TH2D) ) {
    TH2* th = me->getTH2F();
    size_t nbinx = me->getNbinsX();
    size_t nbiny = me->getNbinsY();
    
    float ymin = th->GetYaxis()->GetXmin();
    float ymax = th->GetYaxis()->GetXmax();
    std::string xtit = th->GetXaxis()->GetTitle();
    std::string ytit = th->GetYaxis()->GetTitle();
    float* xbins = new float[nbinx+1];
    for (size_t ix = 1; ix < nbinx+1; ++ix) {
       xbins[ix-1] = th->GetXaxis()->GetBinLowEdge(ix);
       if (ix == nbinx) xbins[ix] = th->GetXaxis()->GetBinUpEdge(ix);
    }    

    std::string tit_new;
    ibooker.setCurrentFolder(folder);

    if (folder == "ParticleFlow/PFElectronValidation/CompWithGenElectron") {
      if (name == "delta_et_Over_et_VS_et_") projection = ibooker.book1D("delta_et_Over_et","E_{T} resolution;#DeltaE_{T}/E_{T}",nbiny,ymin,ymax) ;
      if (name == "delta_et_VS_et_") projection = ibooker.book1D("delta_et_","#DeltaE_{T};#DeltaE_{T}",nbiny,ymin,ymax) ;
      if (name == "delta_eta_VS_et_") projection = ibooker.book1D("delta_eta_","#Delta#eta;#Delta#eta",nbiny,ymin,ymax) ;
      if (name == "delta_phi_VS_et_") projection = ibooker.book1D("delta_phi_","#Delta#phi;#Delta#phi",nbiny,ymin,ymax) ;
    }

    if (projection) {
      for (size_t iy = 1; iy < nbiny+1; ++iy) {
    projection->setBinContent(iy, th->ProjectionY("e")->GetBinContent(iy) ); }
      projection->setEntries( me->getEntries() ) ;
    }
    
    delete [] xbins;
  }
}


//
// -- Create Profile Plots
//
void PFClient::createProfilePlots(DQMStore::IBooker& ibooker, DQMStore::IGetter& igetter, std::string& folder, std::string& name) {     
  MonitorElement* me = igetter.get(folder+"/"+name);
  if (!me) return;

  if ( (me->kind() == MonitorElement::DQM_KIND_TH2F) ||
       (me->kind() == MonitorElement::DQM_KIND_TH2S) ||
       (me->kind() == MonitorElement::DQM_KIND_TH2D) ) {
    TH2* th = me->getTH2F();
    size_t nbinx = me->getNbinsX();
    
    float ymin = th->GetYaxis()->GetXmin();
    float ymax = th->GetYaxis()->GetXmax();
    std::string xtit = th->GetXaxis()->GetTitle();
    std::string ytit = th->GetYaxis()->GetTitle();
    double* xbins = new double[nbinx+1];
    for (size_t ix = 1; ix < nbinx+1; ++ix) {
       xbins[ix-1] = th->GetXaxis()->GetBinLowEdge(ix);
       if (ix == nbinx) xbins[ix] = th->GetXaxis()->GetBinUpEdge(ix);
    }    

    std::string tit_new;
    ibooker.setCurrentFolder(folder);
    //TProfiles
    MonitorElement* me_profile[2]; 
    me_profile[0] = ibooker.bookProfile("profile_"+name, tit_new,
                       nbinx, xbins, ymin, ymax, "" ); 
    me_profile[1] = ibooker.bookProfile("profileRMS_"+name, tit_new,
                       nbinx, xbins, ymin, ymax, "s" ); 
    TProfile* profileX = th->ProfileX();
    //size_t nbiny = me->getNbinsY();
    //TProfile* profileX = th->ProfileX("",0,nbiny+1); add underflow and overflow
    static const Int_t NUM_STAT = 7;
    Double_t stats[NUM_STAT] = {0}; th->GetStats(stats);

    for (Int_t i = 0; i<2; i++) {
      if (me_profile[i]) {
    for (size_t ix = 0; ix <= nbinx+1; ++ix) {
      me_profile[i]->setBinContent( ix, profileX->GetBinContent(ix)*profileX->GetBinEntries(ix) ) ;
      //me_profile[i]->Fill( profileX->GetBinCenter(ix), profileX->GetBinContent(ix)*profileX->GetBinEntries(ix) ) ;
      me_profile[i]->setBinEntries( ix, profileX->GetBinEntries(ix) );
      me_profile[i]->getTProfile()->GetSumw2()->fArray[ix] = profileX->GetSumw2()->fArray[ix];
      //me_profile[i]->getTProfile()->GetBinSumw2()->fArray[ix] = profileX->GetBinSumw2()->fArray[ix]; // segmentation violation
    }
      }
      me_profile[i]->getTProfile()->PutStats(stats);
      me_profile[i]->setEntries( profileX->GetEntries() ) ;
    }

    delete [] xbins;
  }
}


//
// -- Get Histogram Parameters
//
void PFClient::getHistogramParameters(MonitorElement* me_slice, double& average, 
                         double& rms,double& mean, double& sigma) {
  average = 0.0;
  rms     = 0.0;
  mean    = 0.0;
  sigma   = 0.0;

  if (!me_slice) return;
  if  (me_slice->kind() == MonitorElement::DQM_KIND_TH1F) {
    average = me_slice->getMean();
    rms     = me_slice->getRMS();  
    TH1F* th_slice = me_slice->getTH1F();
    if (th_slice && th_slice->GetEntries() > 0) {
      //need our own copy for thread safety
      TF1 gaus("mygaus","gaus");
      th_slice->Fit( &gaus,"Q0");
      sigma = gaus.GetParameter(2);
      mean  = gaus.GetParameter(1);
    }
  }
}


//
// -- Create Efficiency Plots
//
void PFClient::createEfficiencyPlots(DQMStore::IBooker& ibooker, DQMStore::IGetter& igetter, std::string& folder, std::string& name) {     
  MonitorElement* me1 = igetter.get(folder+"/"+name+"ref_");
  MonitorElement* me2 = igetter.get(folder+"/"+name+"gen_");
  if (!me1 || !me2) return;

  TH1F* me1_forEff = (TH1F*)me1->getTH1F()->Rebin(2,"me1_forEff");
  TH1F* me2_forEff = (TH1F*)me2->getTH1F()->Rebin(2,"me2_forEff");
    
  MonitorElement* me_eff;
  if ( (me1->kind() == MonitorElement::DQM_KIND_TH1F) &&
       (me1->kind() == MonitorElement::DQM_KIND_TH1F) ) {
    //TH1* th1 = me1->getTH1F();
    //size_t nbinx = me1->getNbinsX();
    size_t nbinx = me1_forEff->GetNbinsX();

    //float xmin = th1->GetXaxis()->GetXmin();
    //float xmax = th1->GetXaxis()->GetXmax();
    float xmin = me1_forEff->GetXaxis()->GetXmin();
    float xmax = me1_forEff->GetXaxis()->GetXmax();
    std::string xtit = me1->getAxisTitle(1);
    std::string tit_new;
    tit_new = ";"+xtit+";"+xtit+" efficiency"; 

    ibooker.setCurrentFolder(folder);
    me_eff = ibooker.book1D("efficiency_"+name,tit_new, nbinx, xmin, xmax); 
        
    me_eff->Reset(); me_eff->setEfficiencyFlag();
    /*
    double  efficiency;
    for (size_t ix = 1; ix < nbinx+1; ++ix) {
      float val1 = me1->getBinContent(ix);
      float val2 = me2->getBinContent(ix);
      if (val2 > 0.0) efficiency = val1/val2;
      else efficiency = 0;   
      me_eff->setBinContent(ix,efficiency);
    }
    */
    // Binomial errors "B" asked by Florian
    /*me1_forEff->Sumw2(); me2_forEff->Sumw2();*/ me_eff->getTH1F()->Sumw2();
    me_eff->getTH1F()->Divide( me1_forEff, me2_forEff, 1, 1, "B");
  }
}

#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE (PFClient) ;