CMS 3D CMS Logo

List of all members | Public Types | Public Member Functions | Private Member Functions | Private Attributes
SoftKillerProducer Class Reference

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

Inheritance diagram for SoftKillerProducer:
edm::stream::EDProducer<>

Public Types

typedef math::XYZTLorentzVector LorentzVector
 
typedef std::vector< LorentzVectorLorentzVectorCollection
 
typedef std::vector< reco::PFCandidatePFOutputCollection
 
- 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, GlobalCacheLuminosityBlockContext
 
typedef CacheTypes::LuminosityBlockSummaryCache LuminosityBlockSummaryCache
 
typedef CacheTypes::RunCache RunCache
 
typedef RunContextT< RunCache, GlobalCacheRunContext
 
typedef CacheTypes::RunSummaryCache RunSummaryCache
 

Public Member Functions

 SoftKillerProducer (const edm::ParameterSet &)
 
 ~SoftKillerProducer () override
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
bool hasAbilityToProduceInLumis () const final
 
bool hasAbilityToProduceInRuns () const final
 

Private Member Functions

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

Private Attributes

double Rho_EtaMax_
 
double rParam_
 
edm::EDGetTokenT< reco::CandidateViewtokenPFCandidates_
 

Detailed Description

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 45 of file SoftKillerProducer.cc.

Member Typedef Documentation

Definition at line 47 of file SoftKillerProducer.cc.

Definition at line 48 of file SoftKillerProducer.cc.

Definition at line 49 of file SoftKillerProducer.cc.

Constructor & Destructor Documentation

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

Definition at line 76 of file SoftKillerProducer.cc.

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

76  :
77  Rho_EtaMax_( iConfig.getParameter<double>("Rho_EtaMax") ),
78  rParam_ ( iConfig.getParameter<double>("rParam") )
79 {
80 
82  = consumes<reco::CandidateView>(iConfig.getParameter<edm::InputTag>("PFCandidates"));
83 
84  produces<edm::ValueMap<LorentzVector> > ("SoftKillerP4s");
85  produces< PFOutputCollection >();
86 
87 }
T getParameter(std::string const &) const
edm::EDGetTokenT< reco::CandidateView > tokenPFCandidates_
SoftKillerProducer::~SoftKillerProducer ( )
override

Definition at line 90 of file SoftKillerProducer.cc.

91 {
92 
93 }

Member Function Documentation

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

Definition at line 102 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(), 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 heppy_batch::val.

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

103 {
104 
105  std::unique_ptr< PFOutputCollection > pOutput( new PFOutputCollection );
106 
107  // get PF Candidates
109  iEvent.getByToken( tokenPFCandidates_, pfCandidates);
110 
111  std::vector<fastjet::PseudoJet> fjInputs;
112  for ( auto i = pfCandidates->begin(),
113  ibegin = pfCandidates->begin(),
114  iend = pfCandidates->end(); i != iend; ++i ) {
115  fjInputs.push_back( fastjet::PseudoJet( i->px(), i->py(), i->pz(), i->energy() ) );
116  fjInputs.back().set_user_index( i - ibegin );
117  }
118 
119  // soft killer:
120  fastjet::contrib::SoftKiller soft_killer(Rho_EtaMax_, rParam_);
121 
122  double pt_threshold = 0.;
123  std::vector<fastjet::PseudoJet> soft_killed_event;
124  soft_killer.apply(fjInputs, soft_killed_event, pt_threshold);
125 
126  std::unique_ptr<edm::ValueMap<LorentzVector> > p4SKOut(new edm::ValueMap<LorentzVector>());
128 
129  static const reco::PFCandidate dummySinceTranslateIsNotStatic;
130 
131  // To satisfy the value map, the size of the "killed" collection needs to be the
132  // same size as the input collection, so if the constituent is killed, just set E = 0
133  for ( auto j = fjInputs.begin(),
134  jend = fjInputs.end(); j != jend; ++j ) {
135  const reco::Candidate& cand = pfCandidates->at(j->user_index());
136  auto id = dummySinceTranslateIsNotStatic.translatePdgIdToType(cand.pdgId());
137  const reco::PFCandidate *pPF = dynamic_cast<const reco::PFCandidate*>(&cand);
138  reco::PFCandidate pCand( pPF ? *pPF : reco::PFCandidate(cand.charge(), cand.p4(), id) );
139  auto val = j->user_index();
140  auto skmatch = find_if( soft_killed_event.begin(), soft_killed_event.end(), [&val](fastjet::PseudoJet const & i){return i.user_index() == val;} );
141  LorentzVector pVec;
142  if ( skmatch != soft_killed_event.end() ) {
143  pVec.SetPxPyPzE(skmatch->px(),skmatch->py(),skmatch->pz(),skmatch->E());
144  } else {
145  pVec.SetPxPyPzE( 0., 0., 0., 0.);
146  }
147  pCand.setP4(pVec);
148  skP4s.push_back( pVec );
149  pOutput->push_back(pCand);
150  }
151 
152  //Compute the modified p4s
153  edm::ValueMap<LorentzVector>::Filler p4SKFiller(*p4SKOut);
154  p4SKFiller.insert(pfCandidates,skP4s.begin(), skP4s.end() );
155  p4SKFiller.fill();
156 
157  iEvent.put(std::move(p4SKOut),"SoftKillerP4s");
158  iEvent.put(std::move(pOutput));
159 
160 }
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:137
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:579
const_iterator begin() const
virtual const LorentzVector & p4() const =0
four-momentum Lorentz vector
virtual int pdgId() const =0
PDG identifier.
math::XYZTLorentzVector LorentzVector
std::vector< LorentzVector > LorentzVectorCollection
std::vector< reco::PFCandidate > PFOutputCollection
ParticleType translatePdgIdToType(int pdgid) const
Definition: PFCandidate.cc:233
virtual int charge() const =0
electric charge
Particle reconstructed by the particle flow algorithm.
Definition: PFCandidate.h:40
const_reference at(size_type pos) const
const_iterator end() const
edm::EDGetTokenT< reco::CandidateView > tokenPFCandidates_
def move(src, dest)
Definition: eostools.py:511

Member Data Documentation

double SoftKillerProducer::Rho_EtaMax_
private

Definition at line 60 of file SoftKillerProducer.cc.

Referenced by produce().

double SoftKillerProducer::rParam_
private

Definition at line 61 of file SoftKillerProducer.cc.

Referenced by produce().

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

Definition at line 57 of file SoftKillerProducer.cc.

Referenced by produce(), and SoftKillerProducer().