EgammaTrackExtractor.h

Go to the documentation of this file.

#ifndef EgammaIsolationProducers_EgammaTrackExtractor_H
#define EgammaIsolationProducers_EgammaTrackExtractor_H
 
#include <string>
#include <vector>
 
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "DataFormats/Common/interface/View.h"
#include "DataFormats/RecoCandidate/interface/IsoDeposit.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "PhysicsTools/IsolationAlgos/interface/IsoDepositExtractor.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
 
namespace egammaisolation {
 
  class EgammaTrackExtractor : public reco::isodeposit::IsoDepositExtractor {
  public:
    EgammaTrackExtractor(){};
    EgammaTrackExtractor(const edm::ParameterSet& par, edm::ConsumesCollector&& iC) : EgammaTrackExtractor(par, iC) {}
    EgammaTrackExtractor(const edm::ParameterSet& par, edm::ConsumesCollector& iC);
 
    ~EgammaTrackExtractor() override {}
 
    void fillVetos(const edm::Event& ev, const edm::EventSetup& evSetup, const reco::TrackCollection& track) override {}
 
    virtual reco::IsoDeposit::Vetos vetos(const edm::Event& ev,
                                          const edm::EventSetup& evSetup,
                                          const reco::Track& track) const;
 
    reco::IsoDeposit deposit(const edm::Event& ev,
                             const edm::EventSetup& evSetup,
                             const reco::Track& muon) const override {
      edm::LogWarning("EgammaIsolationAlgos|EgammaTrackExtractor")
          << "This Function is not implemented, bad IsoDeposit Returned";
      return reco::IsoDeposit(reco::isodeposit::Direction(1, 1));
    }
 
    reco::IsoDeposit deposit(const edm::Event& ev,
                             const edm::EventSetup& evSetup,
                             const reco::Candidate& muon) const override;
 
  private:
    reco::IsoDeposit::Veto veto(const reco::IsoDeposit::Direction& dir) const;
 
  private:
    // Parameter set
    edm::EDGetTokenT<edm::View<reco::Track> > theTrackCollectionToken;  
    std::string theDepositLabel;                                        
    double minCandEt_;                                                  
    double theDiff_r;                                                   
    double theDiff_z;                                                   
    double theDR_Max;                                                   
    double theDR_Veto;                                                  
    std::string theBeamlineOption;                                      
    edm::InputTag barrelEcalHitsTag_;
    edm::InputTag endcapEcalHitsTag_;
    edm::EDGetTokenT<reco::BeamSpot> theBeamSpotToken;  
    unsigned int theNHits_Min;                          
    double theChi2Ndof_Max;                             
    double theChi2Prob_Min;                             
    double thePt_Min;                                   
    std::vector<double> paramForIsolBarrel_;  
    std::vector<double> paramForIsolEndcap_;  
    std::string dzOptionString;
    int dzOption;
  };
 
}  // namespace egammaisolation
 
#endif