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

    };
  }
}
#endif