CandViewExtractor.h

Go to the documentation of this file.

#ifndef MuonIsolation_CandViewExtractor_H
#define MuonIsolation_CandViewExtractor_H

#include <string>
#include <vector>

#include "FWCore/Framework/interface/ConsumesCollector.h"

#include "DataFormats/RecoCandidate/interface/IsoDeposit.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 "PhysicsTools/IsolationAlgos/interface/IsoDepositExtractor.h"

namespace muonisolation {

  class CandViewExtractor : public reco::isodeposit::IsoDepositExtractor {
  public:
    CandViewExtractor(){};
    CandViewExtractor(const edm::ParameterSet& par, edm::ConsumesCollector&& iC);

    ~CandViewExtractor() override {}

    void fillVetos(const edm::Event& ev, const edm::EventSetup& evSetup, const reco::TrackCollection& cand) override {}

    /*  virtual reco::IsoDeposit::Vetos vetos(const edm::Event & ev,
      const edm::EventSetup & evSetup, const reco::Candidate & cand) const;

  virtual reco::IsoDeposit::Vetos vetos(const edm::Event & ev,
      const edm::EventSetup & evSetup, const reco::Track & cand) const;
*/

    void initEvent(const edm::Event& ev, const edm::EventSetup& evSetup) override;

    reco::IsoDeposit deposit(const edm::Event& ev,
                             const edm::EventSetup& evSetup,
                             const reco::Track& muon) const override {
      return depositFromObject(ev, evSetup, muon);
    }

    reco::IsoDeposit deposit(const edm::Event& ev,
                             const edm::EventSetup& evSetup,
                             const reco::Candidate& muon) const override {
      return depositFromObject(ev, evSetup, muon);
    }

  private:
    reco::IsoDeposit::Veto veto(const reco::IsoDeposit::Direction& dir) const;

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

    // Parameter set
    edm::EDGetTokenT<edm::View<reco::Candidate> > theCandViewToken;  // Track Collection Label
    std::string theDepositLabel;                                     // name for deposit
    edm::Handle<edm::View<reco::Candidate> > theCandViewH;           //cached handle
    edm::Event::CacheIdentifier_t theCacheID;                        //event cacheID
    double theDiff_r;                                                // transverse distance to vertex
    double theDiff_z;                                                // z distance to vertex
    double theDR_Max;                                                // Maximum cone angle for deposits
    double theDR_Veto;                                               // Veto cone angle
  };

}  // namespace muonisolation

#endif