CandIsolatorFromDeposits.h

Go to the documentation of this file.

#ifndef MuonIsolationProducers_CandIsolatorFromDeposits_H
#define MuonIsolationProducers_CandIsolatorFromDeposits_H
 
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "DataFormats/RecoCandidate/interface/IsoDepositFwd.h"
#include "DataFormats/RecoCandidate/interface/IsoDeposit.h"
#include "PhysicsTools/IsolationAlgos/interface/EventDependentAbsVeto.h"
 
#include "DataFormats/Candidate/interface/Candidate.h"
#include "CommonTools/Utils/interface/StringObjectFunction.h"
#include "DataFormats/Common/interface/OwnVector.h"
 
#include <string>
 
namespace edm {
  class Event;
}
namespace edm {
  class EventSetup;
}
 
class CandIsolatorFromDeposits : public edm::stream::EDProducer<> {
public:
  typedef edm::ValueMap<double> CandDoubleMap;
 
  enum Mode { Sum, SumRelative, Sum2, Sum2Relative, Max, MaxRelative, Count, NearestDR, MeanDR, SumDR };
  CandIsolatorFromDeposits(const edm::ParameterSet &);
 
  ~CandIsolatorFromDeposits() override;
 
  void produce(edm::Event &, const edm::EventSetup &) override;
 
private:
  class SingleDeposit {
  public:
    SingleDeposit(const edm::ParameterSet &, edm::ConsumesCollector &&iC);
    void cleanup();
    void open(const edm::Event &iEvent, const edm::EventSetup &iSetup);
    double compute(const reco::CandidateBaseRef &cand);
    const reco::IsoDepositMap &map() { return *hDeps_; }
 
  private:
    Mode mode_;
    edm::EDGetTokenT<reco::IsoDepositMap> srcToken_;
    double deltaR_;
    bool usesFunction_;
    double weight_;
    StringObjectFunction<reco::Candidate> weightExpr_;
    reco::isodeposit::AbsVetos vetos_;
    reco::isodeposit::EventDependentAbsVetos evdepVetos_;  // note: these are a subset of the above. Don't delete twice!
    bool skipDefaultVeto_;
    edm::Handle<reco::IsoDepositMap> hDeps_;  // transient
  };
  // datamembers
  std::vector<SingleDeposit> sources_;
};
#endif