PFTauExtractor.h

Go to the documentation of this file.

#ifndef PhysicsTools_IsolationAlgos_PFTauExtractor_H
#define PhysicsTools_IsolationAlgos_PFTauExtractor_H

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

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

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/RecoCandidate/interface/IsoDeposit.h"
#include "DataFormats/TauReco/interface/PFTau.h"
#include "DataFormats/TauReco/interface/PFTauFwd.h"

class PFTauExtractor : public reco::isodeposit::IsoDepositExtractor
{
 public:

  explicit PFTauExtractor(const edm::ParameterSet&, edm::ConsumesCollector && iC);
  virtual ~PFTauExtractor(){}

  virtual void fillVetos(const edm::Event&, const edm::EventSetup&, const reco::TrackCollection&) { }
  virtual reco::IsoDeposit deposit(const edm::Event& evt, const edm::EventSetup& es, const reco::Track& track) const {
    return depositFromObject(evt, es, track);
  }
  virtual reco::IsoDeposit deposit(const edm::Event& evt, const edm::EventSetup& es, const reco::Candidate& candidate) const {
    return depositFromObject(evt, es, candidate);
  }

 private:

  edm::EDGetTokenT<reco::PFTauCollection> tauSourceToken_;
  edm::EDGetTokenT<edm::View<reco::Candidate> > candidateSourceToken_;
  double maxDxyTrack_;
  double maxDzTrack_;
  double dRmatchPFTau_;
  double dRVetoCone_;
  double dRIsoCone_;
  double dRvetoPFTauSignalConeConstituents_;

  template<typename T>
  reco::IsoDeposit depositFromObject(const edm::Event&, const edm::EventSetup&, const T&) const;

};

#endif