SoftKillerProducer Class Reference

#include <CommonTools/PileupAlgos/plugins/SoftKillerProducer.cc>

Inheritance diagram for SoftKillerProducer:

Public Types
typedef math::XYZTLorentzVector	LorentzVector
typedef std::vector < LorentzVector >	LorentzVectorCollection
typedef std::vector < reco::PFCandidate >	PFOutputCollection
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T...>
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T...>
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Public Member Functions
	SoftKillerProducer (const edm::ParameterSet &)
	~SoftKillerProducer () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Private Member Functions
void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
double	Rho_EtaMax_
double	rParam_
edm::EDGetTokenT < reco::CandidateView >	tokenPFCandidates_

Detailed Description

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 42 of file SoftKillerProducer.cc.

Member Typedef Documentation

typedef math::XYZTLorentzVector SoftKillerProducer::LorentzVector

Definition at line 44 of file SoftKillerProducer.cc.

typedef std::vector<LorentzVector> SoftKillerProducer::LorentzVectorCollection

Definition at line 45 of file SoftKillerProducer.cc.

typedef std::vector<reco::PFCandidate> SoftKillerProducer::PFOutputCollection

Definition at line 46 of file SoftKillerProducer.cc.

Constructor & Destructor Documentation

SoftKillerProducer::SoftKillerProducer ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 71 of file SoftKillerProducer.cc.

References edm::ParameterSet::getParameter(), and tokenPFCandidates_.

    : Rho_EtaMax_(iConfig.getParameter<double>("Rho_EtaMax")), rParam_(iConfig.getParameter<double>("rParam")) {
  tokenPFCandidates_ = consumes<reco::CandidateView>(iConfig.getParameter<edm::InputTag>("PFCandidates"));

  produces<edm::ValueMap<LorentzVector> >("SoftKillerP4s");
  produces<PFOutputCollection>();
}

SoftKillerProducer::~SoftKillerProducer ( )

override

Definition at line 79 of file SoftKillerProducer.cc.

{}

Member Function Documentation

void SoftKillerProducer::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprivate

Definition at line 86 of file SoftKillerProducer.cc.

References reco::Candidate::charge(), edm::helper::Filler< Map >::fill(), edm::Event::getByToken(), mps_fire::i, gpuClustering::id, edm::helper::Filler< Map >::insert(), dqmiolumiharvest::j, eostools::move(), reco::Candidate::p4(), reco::Candidate::pdgId(), slimmedMuons_cfi::pfCandidates, edm::Event::put(), Rho_EtaMax_, rParam_, reco::LeafCandidate::setP4(), tokenPFCandidates_, reco::PFCandidate::translatePdgIdToType(), and hgcalPerformanceValidation::val.

                                                                              {
  std::unique_ptr<PFOutputCollection> pOutput(new PFOutputCollection);

  // get PF Candidates
  edm::Handle<reco::CandidateView> pfCandidates;
  iEvent.getByToken(tokenPFCandidates_, pfCandidates);

  std::vector<fastjet::PseudoJet> fjInputs;
  for (auto i = pfCandidates->begin(), ibegin = pfCandidates->begin(), iend = pfCandidates->end(); i != iend; ++i) {
    fjInputs.push_back(fastjet::PseudoJet(i->px(), i->py(), i->pz(), i->energy()));
    fjInputs.back().set_user_index(i - ibegin);
  }

  // soft killer:
  fastjet::contrib::SoftKiller soft_killer(Rho_EtaMax_, rParam_);

  double pt_threshold = 0.;
  std::vector<fastjet::PseudoJet> soft_killed_event;
  soft_killer.apply(fjInputs, soft_killed_event, pt_threshold);

  std::unique_ptr<edm::ValueMap<LorentzVector> > p4SKOut(new edm::ValueMap<LorentzVector>());
  LorentzVectorCollection skP4s;

  static const reco::PFCandidate dummySinceTranslateIsNotStatic;

  // To satisfy the value map, the size of the "killed" collection needs to be the
  // same size as the input collection, so if the constituent is killed, just set E = 0
  for (auto j = fjInputs.begin(), jend = fjInputs.end(); j != jend; ++j) {
    const reco::Candidate& cand = pfCandidates->at(j->user_index());
    auto id = dummySinceTranslateIsNotStatic.translatePdgIdToType(cand.pdgId());
    const reco::PFCandidate* pPF = dynamic_cast<const reco::PFCandidate*>(&cand);
    reco::PFCandidate pCand(pPF ? *pPF : reco::PFCandidate(cand.charge(), cand.p4(), id));
    auto val = j->user_index();
    auto skmatch = find_if(soft_killed_event.begin(), soft_killed_event.end(), [&val](fastjet::PseudoJet const& i) {
      return i.user_index() == val;
    });
    LorentzVector pVec;
    if (skmatch != soft_killed_event.end()) {
      pVec.SetPxPyPzE(skmatch->px(), skmatch->py(), skmatch->pz(), skmatch->E());
    } else {
      pVec.SetPxPyPzE(0., 0., 0., 0.);
    }
    pCand.setP4(pVec);
    skP4s.push_back(pVec);
    pOutput->push_back(pCand);
  }

  //Compute the modified p4s
  edm::ValueMap<LorentzVector>::Filler p4SKFiller(*p4SKOut);
  p4SKFiller.insert(pfCandidates, skP4s.begin(), skP4s.end());
  p4SKFiller.fill();

  iEvent.put(std::move(p4SKOut), "SoftKillerP4s");
  iEvent.put(std::move(pOutput));
}

Member Data Documentation

double SoftKillerProducer::Rho_EtaMax_

private

Definition at line 56 of file SoftKillerProducer.cc.

Referenced by produce().

double SoftKillerProducer::rParam_

private

Definition at line 57 of file SoftKillerProducer.cc.

Referenced by produce().

edm::EDGetTokenT<reco::CandidateView> SoftKillerProducer::tokenPFCandidates_

private

Definition at line 54 of file SoftKillerProducer.cc.

Referenced by produce(), and SoftKillerProducer().