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 "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.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 theUseRecHitsFlag;

    std::string theDepositLabel;

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

    std::string thePropagatorName;

    double theThreshold_E;
    double theThreshold_H;
    double theThreshold_HO;

    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_;

    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