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<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

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

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 edm::View< T >::at(), edm::View< T >::begin(), reco::Candidate::charge(), DEFINE_FWK_MODULE, edm::View< T >::end(), edm::helper::Filler< Map >::fill(), edm::Event::getByToken(), mps_fire::i, triggerObjects_cff::id, edm::helper::Filler< Map >::insert(), dqmiolumiharvest::j, eostools::move(), reco::Candidate::p4(), reco::Candidate::pdgId(), zmumugammaAnalyzer_cfi::pfCandidates, edm::Event::put(), Rho_EtaMax_, rParam_, reco::LeafCandidate::setP4(), tokenPFCandidates_, reco::PFCandidate::translatePdgIdToType(), and heppy_batch::val.

Referenced by JSONExport.JsonExport::export(), HTMLExport.HTMLExport::export(), and HTMLExport.HTMLExportStatic::export().

                                                                              {
  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().