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

}

#endif