CaloExtractorByAssociator.h

Go to the documentation of this file.

#ifndef MuonIsolation_CaloExtractorByAssociator_H
#define MuonIsolation_CaloExtractorByAssociator_H

#include <string>

#include "PhysicsTools/IsolationAlgos/interface/IsoDepositExtractor.h"

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

#include "DataFormats/RecoCandidate/interface/IsoDeposit.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"

#include "DataFormats/CaloTowers/interface/CaloTowerCollection.h"

#include "DataFormats/GeometryVector/interface/GlobalPoint.h"

#include "CondFormats/EcalObjects/interface/EcalPFRecHitThresholds.h"
#include "CondFormats/DataRecord/interface/EcalPFRecHitThresholdsRcd.h"
#include "CondFormats/DataRecord/interface/HcalPFCutsRcd.h"
#include "CondFormats/HcalObjects/interface/HcalChannelQuality.h"
#include "CondFormats/DataRecord/interface/HcalChannelQualityRcd.h"
#include "CondTools/Hcal/interface/HcalPFCutsHandler.h"

#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/CaloTopology/interface/HcalTopology.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "RecoLocalCalo/HcalRecAlgos/interface/HcalSeverityLevelComputer.h"
#include "RecoLocalCalo/HcalRecAlgos/interface/HcalSeverityLevelComputerRcd.h"

class TrackAssociatorParameters;
class TrackDetectorAssociator;
class MuonServiceProxy;

namespace muonisolation {

  class CaloExtractorByAssociator : public reco::isodeposit::IsoDepositExtractor {
  public:
    CaloExtractorByAssociator(){};
    CaloExtractorByAssociator(const edm::ParameterSet& par, edm::ConsumesCollector&& iC);

    ~CaloExtractorByAssociator() override;

    void fillVetos(const edm::Event& ev, const edm::EventSetup& evSetup, const reco::TrackCollection& tracks) override;
    reco::IsoDeposit deposit(const edm::Event& ev,
                             const edm::EventSetup& evSetup,
                             const reco::Track& track) const override;
    std::vector<reco::IsoDeposit> deposits(const edm::Event& ev,
                                           const edm::EventSetup& evSetup,
                                           const reco::Track& track) const override;

  private:
    bool theUseEcalRecHitsFlag;
    bool theUseHcalRecHitsFlag;
    bool theUseHORecHitsFlag;

    std::string theDepositLabel;

    std::vector<std::string> theDepositInstanceLabels;

    std::string thePropagatorName;

    double theThreshold_E;
    double theThreshold_H;
    double theThreshold_HO;
    int theMaxSeverityHB;
    int theMaxSeverityHE;

    double theDR_Veto_E;
    double theDR_Veto_H;
    double theDR_Veto_HO;
    bool theCenterConeOnCalIntersection;
    double theDR_Max;

    double theNoise_EB;
    double theNoise_EE;
    double theNoise_HB;
    double theNoise_HE;
    double theNoise_HO;
    double theNoiseTow_EB;
    double theNoiseTow_EE;

    std::vector<DetId> theVetoCollection;

    MuonServiceProxy* theService;

    TrackAssociatorParameters* theAssociatorParameters;
    TrackDetectorAssociator* theAssociator;

    edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> bFieldToken_;
    edm::ESGetToken<CaloGeometry, CaloGeometryRecord> caloGeomToken_;

    // Ecal noise thresholds
    edm::ESGetToken<EcalPFRecHitThresholds, EcalPFRecHitThresholdsRcd> ecalPFRechitThresholdsToken_;
    bool ecalRecHitThresh_;
    EcalPFRecHitThresholds* ecalThresholds = nullptr;

    // following are needed to grab HCal thresholds from GT
    edm::ESGetToken<HcalPFCuts, HcalPFCutsRcd> hcalCutsToken_;
    bool hcalCutsFromDB_;
    HcalPFCuts* hcalCuts = nullptr;

    edm::ESGetToken<HcalTopology, HcalRecNumberingRecord> hcalTopologyToken_;
    edm::ESGetToken<HcalChannelQuality, HcalChannelQualityRcd> hcalChannelQualityToken_;
    edm::ESGetToken<HcalSeverityLevelComputer, HcalSeverityLevelComputerRcd> hcalSevLvlComputerToken_;
    HcalTopology* hcalTopology_;
    HcalChannelQuality* hcalChStatus_;
    HcalSeverityLevelComputer* hcalSevLvlComputer_;

    bool thePrintTimeReport;

    double noiseEcal(const CaloTower& tower) const;
    double noiseHcal(const CaloTower& tower) const;
    double noiseHOcal(const CaloTower& tower) const;
    double noiseRecHit(const DetId& detId) const;
  };

}  // namespace muonisolation

#endif