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	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::CandidateView >	tokenPFCandidates_

Detailed Description

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 45 of file SoftKillerProducer.cc.

Member Typedef Documentation

typedef math::XYZTLorentzVector SoftKillerProducer::LorentzVector

Definition at line 47 of file SoftKillerProducer.cc.

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

Definition at line 48 of file SoftKillerProducer.cc.

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

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_.

                                                                     : 
  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 90 of file SoftKillerProducer.cc.

{

}

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

{

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