PhotonIsolationCalculator.h

Go to the documentation of this file.

#ifndef PhotonIsolationCalculator_H
#define PhotonIsolationCalculator_H

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/EgammaCandidates/interface/Photon.h"

#include "CondFormats/EcalObjects/interface/EcalChannelStatus.h"
#include "RecoLocalCalo/EcalRecAlgos/interface/EcalSeverityLevelAlgo.h"

#include <string>

#include "FWCore/Utilities/interface/GCC11Compatibility.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

class PhotonIsolationCalculator {
public:
  PhotonIsolationCalculator() {}

  ~PhotonIsolationCalculator() {}

  void setup(const edm::ParameterSet& conf,
             std::vector<int> const& flagsEB_,
             std::vector<int> const& flagsEE_,
             std::vector<int> const& severitiesEB_,
             std::vector<int> const& severitiesEE_,
             edm::ConsumesCollector&& iC);

  void calculate(const reco::Photon*,
                 const edm::Event&,
                 const edm::EventSetup& es,
                 reco::Photon::FiducialFlags& phofid,
                 reco::Photon::IsolationVariables& phoisolR03,
                 reco::Photon::IsolationVariables& phoisolR04) const;

private:
  static void classify(const reco::Photon* photon,
                       bool& isEBPho,
                       bool& isEEPho,
                       bool& isEBEtaGap,
                       bool& isEBPhiGap,
                       bool& isEERingGap,
                       bool& isEEDeeGap,
                       bool& isEBEEGap) dso_internal;

  void calculateTrackIso(const reco::Photon* photon,
                         const edm::Event& e,
                         double& trkCone,
                         int& ntrkCone,
                         double pTThresh = 0,
                         double RCone = .4,
                         double RinnerCone = .1,
                         double etaSlice = 0.015,
                         double lip = 0.2,
                         double d0 = 0.1) const dso_internal;

  double calculateEcalRecHitIso(const reco::Photon* photon,
                                const edm::Event& iEvent,
                                const edm::EventSetup& iSetup,
                                double RCone,
                                double RConeInner,
                                double etaSlice,
                                double eMin,
                                double etMin,
                                bool vetoClusteredHits,
                                bool useNumCrystals) const dso_internal;

  double calculateHcalTowerIso(const reco::Photon* photon,
                               const edm::Event& iEvent,
                               const edm::EventSetup& iSetup,
                               double RCone,
                               double RConeInner,
                               double eMin,
                               signed int depth) const dso_internal;

  double calculateHcalTowerIso(const reco::Photon* photon,
                               const edm::Event& iEvent,
                               const edm::EventSetup& iSetup,
                               double RCone,
                               double eMin,
                               signed int depth) const dso_internal;

private:
  edm::EDGetToken barrelecalCollection_;
  edm::EDGetToken endcapecalCollection_;
  edm::EDGetToken hcalCollection_;

  edm::EDGetToken trackInputTag_;
  edm::EDGetToken beamSpotProducerTag_;
  double modulePhiBoundary_;
  std::vector<double> moduleEtaBoundary_;
  bool vetoClusteredEcalHits_;
  bool useNumCrystals_;

  double trkIsoBarrelRadiusA_[6];
  double ecalIsoBarrelRadiusA_[5];
  double hcalIsoBarrelRadiusA_[9];
  double trkIsoBarrelRadiusB_[6];
  double ecalIsoBarrelRadiusB_[5];
  double hcalIsoBarrelRadiusB_[9];

  double trkIsoEndcapRadiusA_[6];
  double ecalIsoEndcapRadiusA_[5];
  double hcalIsoEndcapRadiusA_[9];
  double trkIsoEndcapRadiusB_[6];
  double ecalIsoEndcapRadiusB_[5];
  double hcalIsoEndcapRadiusB_[9];

  std::vector<int> flagsEB_;
  std::vector<int> flagsEE_;
  std::vector<int> severityExclEB_;
  std::vector<int> severityExclEE_;
};

#endif  // PhotonIsolationCalculator_H