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

  virtual ~EgammaTrackExtractor(){}

  virtual void fillVetos (const edm::Event & ev,
      const edm::EventSetup & evSetup, const reco::TrackCollection & track) {}

  virtual reco::IsoDeposit::Vetos vetos(const edm::Event & ev,
      const edm::EventSetup & evSetup, const reco::Track & track)const;

  virtual reco::IsoDeposit deposit (const edm::Event & ev,
      const edm::EventSetup & evSetup, const reco::Track & muon) const {
        edm::LogWarning("EgammaIsolationAlgos|EgammaTrackExtractor")
           << "This Function is not implemented, bad IsoDeposit Returned";
        return reco::IsoDeposit( reco::isodeposit::Direction(1,1) );
      }

  virtual reco::IsoDeposit deposit (const edm::Event & ev,
      const edm::EventSetup & evSetup, const reco::Candidate & muon) const;

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

}

#endif