JetExtractor.h

Go to the documentation of this file.

#ifndef MuonIsolation_JetExtractor_H
#define MuonIsolation_JetExtractor_H

#include <string>

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

#include "PhysicsTools/IsolationAlgos/interface/IsoDepositExtractor.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/JetReco/interface/CaloJetCollection.h"

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

#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"

class TrackAssociatorParameters;
class TrackDetectorAssociator;
class MuonServiceProxy;

namespace muonisolation {

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

    ~JetExtractor() 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;

  private:
    edm::EDGetTokenT<reco::CaloJetCollection> theJetCollectionToken;

    std::string thePropagatorName;

    edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> theFieldToken;

    // Cone cuts and thresholds
    double theThreshold;
    double theDR_Veto;
    double theDR_Max;

    //excludes sumEt of towers that are inside muon veto cone
    bool theExcludeMuonVeto;

    std::unique_ptr<MuonServiceProxy> theService;

    std::unique_ptr<TrackAssociatorParameters> theAssociatorParameters;
    std::unique_ptr<TrackDetectorAssociator> theAssociator;

    bool thePrintTimeReport;
  };

}  // namespace muonisolation

#endif