IsoDepositExtractor.h

Go to the documentation of this file.

#ifndef IsolationAlgos_IsoDepositExtractor_H
#define IsolationAlgos_IsoDepositExtractor_H
 
//
//
//
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.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"
 
namespace reco {
  namespace isodeposit {
 
    class IsoDepositExtractor {
    public:
      virtual ~IsoDepositExtractor(){};
 
      virtual void fillVetos(const edm::Event& ev,
                             const edm::EventSetup& evSetup,
                             const reco::TrackCollection& tracks) = 0;
 
      virtual void initEvent(const edm::Event&, const edm::EventSetup&) {}
 
      virtual reco::IsoDeposit deposit(const edm::Event& ev,
                                       const edm::EventSetup& evSetup,
                                       const reco::Track& track) const = 0;
 
      virtual reco::IsoDeposit deposit(const edm::Event& ev,
                                       const edm::EventSetup& evSetup,
                                       const reco::TrackBaseRef& track) const {
        return deposit(ev, evSetup, *track);
      }
 
      virtual reco::IsoDeposit deposit(const edm::Event& ev,
                                       const edm::EventSetup& evSetup,
                                       const reco::Candidate& track) const {
        //track-based implementation as default <-- do I want this pure virtual?
        return deposit(
            ev,
            evSetup,
            reco::Track(10, 10, track.vertex(), track.momentum(), track.charge(), reco::Track::CovarianceMatrix()));
      }
 
      virtual reco::IsoDeposit deposit(const edm::Event& ev,
                                       const edm::EventSetup& evSetup,
                                       const reco::CandidateBaseRef& track) const {
        return deposit(ev, evSetup, *track);
      }
 
      virtual std::vector<reco::IsoDeposit> deposits(const edm::Event& ev,
                                                     const edm::EventSetup& evSetup,
                                                     const reco::Track& track) const {
        return std::vector<reco::IsoDeposit>(1, deposit(ev, evSetup, track));
      }
 
      virtual std::vector<reco::IsoDeposit> deposits(const edm::Event& ev,
                                                     const edm::EventSetup& evSetup,
                                                     const reco::TrackBaseRef& track) const {
        return deposits(ev, evSetup, *track);
      }
 
      virtual std::vector<reco::IsoDeposit> deposits(const edm::Event& ev,
                                                     const edm::EventSetup& evSetup,
                                                     const reco::Candidate& track) const {
        return deposits(
            ev,
            evSetup,
            reco::Track(10, 10, track.vertex(), track.momentum(), track.charge(), reco::Track::CovarianceMatrix()));
      }
 
      virtual std::vector<reco::IsoDeposit> deposits(const edm::Event& ev,
                                                     const edm::EventSetup& evSetup,
                                                     const reco::CandidateBaseRef& track) const {
        return deposits(ev, evSetup, *track);
      }
    };
  }  // namespace isodeposit
}  // namespace reco
#endif