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));
}