EGPhotonImporter.cc

Go to the documentation of this file.

#include "RecoParticleFlow/PFProducer/interface/BlockElementImporterBase.h"
#include "RecoParticleFlow/PFProducer/interface/PhotonSelectorAlgo.h"
#include "DataFormats/ParticleFlowReco/interface/PFClusterFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFCluster.h"
#include "DataFormats/ParticleFlowReco/interface/PFBlockElementSuperCluster.h"
#include "DataFormats/EgammaCandidates/interface/PhotonFwd.h"
#include "DataFormats/EgammaCandidates/interface/Photon.h"
#include "DataFormats/EgammaReco/interface/SuperCluster.h"
#include "RecoParticleFlow/PFProducer/interface/PFBlockElementSCEqual.h"

#include <unordered_map>

class EGPhotonImporter : public BlockElementImporterBase {
public:
  enum SelectionChoices { SeparateDetectorIso, CombinedDetectorIso };

  EGPhotonImporter(const edm::ParameterSet&, edm::ConsumesCollector&);

  void importToBlock(const edm::Event&, ElementList&) const override;

private:
  edm::EDGetTokenT<reco::PhotonCollection> _src;
  const std::unordered_map<std::string, SelectionChoices> _selectionTypes;
  SelectionChoices _selectionChoice;
  std::unique_ptr<const PhotonSelectorAlgo> _selector;
  bool _superClustersArePF;
};

DEFINE_EDM_PLUGIN(BlockElementImporterFactory, EGPhotonImporter, "EGPhotonImporter");

EGPhotonImporter::EGPhotonImporter(const edm::ParameterSet& conf, edm::ConsumesCollector& sumes)
    : BlockElementImporterBase(conf, sumes),
      _src(sumes.consumes<reco::PhotonCollection>(conf.getParameter<edm::InputTag>("source"))),
      _selectionTypes({{"SeparateDetectorIso", EGPhotonImporter::SeparateDetectorIso},
                       {"CombinedDetectorIso", EGPhotonImporter::CombinedDetectorIso}}),
      _superClustersArePF(conf.getParameter<bool>("superClustersArePF")) {
  const std::string& selChoice = conf.getParameter<std::string>("SelectionChoice");
  _selectionChoice = _selectionTypes.at(selChoice);
  const edm::ParameterSet& selDef = conf.getParameterSet("SelectionDefinition");
  const float minEt = selDef.getParameter<double>("minEt");
  const float trackIso_const = selDef.getParameter<double>("trackIsoConstTerm");
  const float trackIso_slope = selDef.getParameter<double>("trackIsoSlopeTerm");
  const float ecalIso_const = selDef.getParameter<double>("ecalIsoConstTerm");
  const float ecalIso_slope = selDef.getParameter<double>("ecalIsoSlopeTerm");
  const float hcalIso_const = selDef.getParameter<double>("hcalIsoConstTerm");
  const float hcalIso_slope = selDef.getParameter<double>("hcalIsoSlopeTerm");
  const float hoe = selDef.getParameter<double>("HoverE");
  const float loose_hoe = selDef.getParameter<double>("LooseHoverE");
  const float combIso = selDef.getParameter<double>("combIsoConstTerm");
  _selector.reset(new PhotonSelectorAlgo((float)_selectionChoice,
                                         minEt,
                                         trackIso_const,
                                         trackIso_slope,
                                         ecalIso_const,
                                         ecalIso_slope,
                                         hcalIso_const,
                                         hcalIso_slope,
                                         hoe,
                                         combIso,
                                         loose_hoe));
}

void EGPhotonImporter::importToBlock(const edm::Event& e, BlockElementImporterBase::ElementList& elems) const {
  typedef BlockElementImporterBase::ElementList::value_type ElementType;
  auto photons = e.getHandle(_src);
  elems.reserve(elems.size() + photons->size());
  // setup our elements so that all the SCs are grouped together
  auto SCs_end = std::partition(
      elems.begin(), elems.end(), [](const ElementType& a) { return a->type() == reco::PFBlockElement::SC; });
  //now add the photons
  auto bphoton = photons->cbegin();
  auto ephoton = photons->cend();
  reco::PFBlockElementSuperCluster* scbe = nullptr;
  reco::PhotonRef phoref;
  for (auto photon = bphoton; photon != ephoton; ++photon) {
    if (_selector->passPhotonSelection(*photon)) {
      phoref = reco::PhotonRef(photons, std::distance(bphoton, photon));
      const reco::SuperClusterRef& scref = photon->superCluster();
      PFBlockElementSCEqual myEqual(scref);
      auto sc_elem = std::find_if(elems.begin(), SCs_end, myEqual);
      if (sc_elem != SCs_end) {
        scbe = static_cast<reco::PFBlockElementSuperCluster*>(sc_elem->get());
        scbe->setFromPhoton(true);
        scbe->setPhotonRef(phoref);
        scbe->setTrackIso(photon->trkSumPtHollowConeDR04());
        scbe->setEcalIso(photon->ecalRecHitSumEtConeDR04());
        scbe->setHcalIso(photon->hcalTowerSumEtConeDR04());
        scbe->setHoE(photon->hadronicOverEm());
      } else {
        scbe = new reco::PFBlockElementSuperCluster(scref);
        scbe->setFromPhoton(true);
        scbe->setFromPFSuperCluster(_superClustersArePF);
        scbe->setPhotonRef(phoref);
        scbe->setTrackIso(photon->trkSumPtHollowConeDR04());
        scbe->setEcalIso(photon->ecalRecHitSumEtConeDR04());
        scbe->setHcalIso(photon->hcalTowerSumEtConeDR04());
        scbe->setHoE(photon->hadronicOverEm());
        SCs_end = elems.insert(SCs_end, ElementType(scbe));
        ++SCs_end;  // point to element *after* the new one
      }
    }
  }  // loop on photons
  elems.shrink_to_fit();
}