HFRecoEcalCandidateProducer.cc

Go to the documentation of this file.

#include <iostream>
#include <vector>
#include <memory>

// Framework
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Utilities/interface/Exception.h"
//
#include "DataFormats/EgammaReco/interface/HFEMClusterShape.h"
#include "DataFormats/EgammaReco/interface/HFEMClusterShapeAssociation.h"
#include "DataFormats/EgammaReco/interface/SuperCluster.h"
#include "DataFormats/RecoCandidate/interface/RecoEcalCandidate.h"
#include "DataFormats/RecoCandidate/interface/RecoEcalCandidateFwd.h"

#include "RecoEgamma/EgammaHFProducers/plugins/HFRecoEcalCandidateProducer.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"

HFRecoEcalCandidateProducer::HFRecoEcalCandidateProducer(edm::ParameterSet const& conf):
  defaultDB_(std::vector<double>()),
  hfclustersSC_(consumes<reco::SuperClusterCollection>(conf.getParameter<edm::InputTag>("hfclusters"))),
  hfclustersHFEM_(consumes<reco::HFEMClusterShapeAssociationCollection>(conf.getParameter<edm::InputTag>("hfclusters"))),
  
  HFDBversion_(conf.existsAs<int>("HFDBversion") ? conf.getParameter<int>("HFDBversion"):99),//do nothing
  HFDBvector_(conf.existsAs<std::vector<double> >("HFDBvector") ? conf.getParameter<std::vector<double> >("HFDBvector"):defaultDB_),
  doPU_(false),
  Cut2D_(conf.getParameter<double>("intercept2DCut")),
  defaultSlope2D_((Cut2D_<=0.83)?(0.475):((Cut2D_>0.83 && Cut2D_<=0.9)?(0.275):(0.2))),//fix for hlt unable to add slope variable now
  hfvars_(HFDBversion_,HFDBvector_),
  algo_(conf.existsAs<bool>("Correct") ? conf.getParameter<bool>("Correct") : true,
    conf.getParameter<double>("e9e25Cut"),
    conf.getParameter<double>("intercept2DCut"),
    conf.existsAs<double>("intercept2DSlope") ? conf.getParameter<double>("intercept2DSlope") : defaultSlope2D_,
    conf.getParameter<std::vector<double> >("e1e9Cut"),
    conf.getParameter<std::vector<double> >("eCOREe9Cut"),
    conf.getParameter<std::vector<double> >("eSeLCut"),
    hfvars_) 
{
  if(conf.existsAs<edm::InputTag>("VertexCollection")){
    vertices_ = consumes<reco::VertexCollection>(conf.getParameter<edm::InputTag>("VertexCollection"));
  }else vertices_ = consumes<reco::VertexCollection>(edm::InputTag("offlinePrimaryVertices"));

  produces<reco::RecoEcalCandidateCollection>();

} 

void HFRecoEcalCandidateProducer::produce(edm::Event & e, edm::EventSetup const& iSetup) {  
  
  
  edm::Handle<reco::SuperClusterCollection> super_clus;
  edm::Handle<reco::HFEMClusterShapeAssociationCollection> hf_assoc;
 
  e.getByToken(hfclustersSC_,super_clus);
  e.getByToken(hfclustersHFEM_,hf_assoc);
 
  int nvertex = 0;
  if(HFDBversion_!=99){
    edm:: Handle<reco::VertexCollection> pvHandle;
    e.getByToken(vertices_, pvHandle);
    const reco::VertexCollection & vertices = *pvHandle.product();
    static const int minNDOF = 4;
    static const double maxAbsZ = 15.0;
    static const double maxd0 = 2.0;
    
    //count verticies
  
    for(reco::VertexCollection::const_iterator vit = vertices.begin(); vit != vertices.end(); ++vit){
      if(vit->ndof() > minNDOF && ((maxAbsZ <= 0) || fabs(vit->z()) <= maxAbsZ) && ((maxd0 <= 0) || fabs(vit->position().rho()) <= maxd0)) 
    nvertex++;
    }
  }else{
    nvertex = 1;
  }

  

  // create return data
  auto retdata1 = std::make_unique<reco::RecoEcalCandidateCollection>();

  
  algo_.produce(super_clus,*hf_assoc,*retdata1,nvertex);
 
  e.put(std::move(retdata1));

}