EgammaIsoHcalDetIdCollectionProducer.cc

Go to the documentation of this file.

#include "RecoEgamma/EgammaIsolationAlgos/plugins/EgammaIsoHcalDetIdCollectionProducer.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Framework/interface/EventSetup.h"

#include "DataFormats/DetId/interface/DetIdCollection.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"

EgammaIsoHcalDetIdCollectionProducer::EgammaIsoHcalDetIdCollectionProducer(const edm::ParameterSet& iConfig)
    : hcalHitSelector_(iConfig.getParameter<edm::ParameterSet>("hitSelection")) {
  recHitsToken_ = consumes<HBHERecHitCollection>(iConfig.getParameter<edm::InputTag>("recHitsLabel"));
  elesToken_ = consumes<reco::GsfElectronCollection>(iConfig.getParameter<edm::InputTag>("elesLabel"));

  phosToken_ = consumes<reco::PhotonCollection>(iConfig.getParameter<edm::InputTag>("phosLabel"));

  superClustersToken_ =
      consumes<reco::SuperClusterCollection>(iConfig.getParameter<edm::InputTag>("superClustersLabel"));

  minSCEt_ = iConfig.getParameter<double>("minSCEt");
  minEleEt_ = iConfig.getParameter<double>("minEleEt");
  minPhoEt_ = iConfig.getParameter<double>("minPhoEt");

  interestingDetIdCollection_ = iConfig.getParameter<std::string>("interestingDetIdCollection");

  //register your products
  produces<DetIdCollection>(interestingDetIdCollection_);
}

void EgammaIsoHcalDetIdCollectionProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("recHitsLabel", edm::InputTag("hbhereco"));
  desc.add<edm::InputTag>("elesLabel", edm::InputTag("gedGsfElectrons"));
  desc.add<edm::InputTag>("phosLabel", edm::InputTag("gedPhotons"));
  desc.add<edm::InputTag>("superClustersLabel", edm::InputTag("particleFlowEGamma"));
  desc.add<double>("minSCEt", 20);
  desc.add<double>("minEleEt", 20);
  desc.add<double>("minPhoEt", 20);
  desc.add<std::string>("interestingDetIdCollection", "");
  desc.add<edm::ParameterSetDescription>("hitSelection", EGHcalRecHitSelector::makePSetDescription());
  descriptions.add(("interestingGedEgammaIsoHCALDetId"), desc);
}

void EgammaIsoHcalDetIdCollectionProducer::beginRun(edm::Run const& run, const edm::EventSetup& iSetup) {
  hcalHitSelector_.setup(iSetup);
}

// ------------ method called to produce the data  ------------
void EgammaIsoHcalDetIdCollectionProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  edm::Handle<reco::SuperClusterCollection> superClusters;
  iEvent.getByToken(superClustersToken_, superClusters);

  edm::Handle<reco::GsfElectronCollection> eles;
  iEvent.getByToken(elesToken_, eles);

  edm::Handle<reco::PhotonCollection> phos;
  iEvent.getByToken(phosToken_, phos);

  edm::Handle<HBHERecHitCollection> recHits;
  iEvent.getByToken(recHitsToken_, recHits);

  std::vector<DetId> indexToStore;
  indexToStore.reserve(100);

  if (eles.isValid() && recHits.isValid()) {
    for (auto& ele : *eles) {
      float scEt = ele.superCluster()->energy() * std::sin(ele.superCluster()->position().theta());
      if (scEt > minEleEt_ || ele.et() > minEleEt_) {
        hcalHitSelector_.addDetIds(*ele.superCluster(), *recHits, indexToStore);
      }
    }
  }
  if (phos.isValid() && recHits.isValid()) {
    for (auto& pho : *phos) {
      float scEt = pho.superCluster()->energy() * std::sin(pho.superCluster()->position().theta());
      if (scEt > minPhoEt_ || pho.et() > minPhoEt_) {
        hcalHitSelector_.addDetIds(*pho.superCluster(), *recHits, indexToStore);
      }
    }
  }
  if (superClusters.isValid() && recHits.isValid()) {
    for (auto& sc : *superClusters) {
      float scEt = sc.energy() * std::sin(sc.position().theta());
      if (scEt > minSCEt_) {
        hcalHitSelector_.addDetIds(sc, *recHits, indexToStore);
      }
    }
  }

  //unify the vector
  std::sort(indexToStore.begin(), indexToStore.end());
  std::unique(indexToStore.begin(), indexToStore.end());

  auto detIdCollection = std::make_unique<DetIdCollection>(indexToStore);

  iEvent.put(std::move(detIdCollection), interestingDetIdCollection_);
}