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