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