Inheritance diagram for GenParticlePruner:

Public Member Functions
	GenParticlePruner (const edm::ParameterSet &)

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	flagDaughters (const reco::GenParticle &, int)

void	flagMothers (const reco::GenParticle &, int)

void	getDaughterKeys (std::vector< size_t > &, std::vector< size_t > &, const reco::GenParticleRefVector &) const

void	getMotherKeys (std::vector< size_t > &, std::vector< size_t > &, const reco::GenParticleRefVector &) const

void	parse (const std::string &selection, helper::SelectCode &code, std::string &cut) const

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

void	recursiveFlagDaughters (size_t, const reco::GenParticleCollection &, int, std::vector< size_t > &)

void	recursiveFlagMothers (size_t, const reco::GenParticleCollection &, int, std::vector< size_t > &)

Private Attributes
bool	firstEvent_

std::vector< int >	flags_

std::vector< size_t >	indices_

int	keepOrDropAll_

std::vector< std::pair< StringCutObjectSelector< reco::GenParticle >, helper::SelectCode > >	select_

std::vector< std::string >	selection_

edm::EDGetTokenT< reco::GenParticleCollection >	srcToken_

Additional Inherited Members
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

Detailed Description

Definition at line 22 of file GenParticlePruner.cc.

Constructor & Destructor Documentation

GenParticlePruner::GenParticlePruner ( const edm::ParameterSet & cfg )

Definition at line 98 of file GenParticlePruner.cc.

     : firstEvent_(true),
       srcToken_(consumes<GenParticleCollection>(cfg.getParameter<InputTag>("src"))),
       keepOrDropAll_(drop),
       selection_(cfg.getParameter<vector<string>>("select")) {
   using namespace ::helper;
   produces<GenParticleCollection>();
   produces<edm::Association<reco::GenParticleCollection>>();
 }

Member Function Documentation

void GenParticlePruner::flagDaughters	(	const reco::GenParticle &	gen,
		int	keepOrDrop
	)

private

Definition at line 108 of file GenParticlePruner.cc.

References edm::RefVector< C, T, F >::begin(), reco::CompositeRefCandidateT< D >::daughterRefVector(), BPhysicsValidation_cfi::daughters, edm::RefVector< C, T, F >::end(), flags_, and mps_fire::i.

Referenced by produce().

                                                                                 {
   const GenParticleRefVector &daughters = gen.daughterRefVector();
   for (GenParticleRefVector::const_iterator i = daughters.begin(); i != daughters.end(); ++i)
     flags_[i->key()] = keepOrDrop;
 }

void GenParticlePruner::flagMothers	(	const reco::GenParticle &	gen,
		int	keepOrDrop
	)

private

Definition at line 114 of file GenParticlePruner.cc.

References edm::RefVector< C, T, F >::begin(), edm::RefVector< C, T, F >::end(), flags_, mps_fire::i, reco::CompositeRefCandidateT< D >::motherRefVector(), and HLTTauQualityTests_cfi::mothers.

Referenced by produce().

                                                                               {
   const GenParticleRefVector &mothers = gen.motherRefVector();
   for (GenParticleRefVector::const_iterator i = mothers.begin(); i != mothers.end(); ++i)
     flags_[i->key()] = keepOrDrop;
 }

void GenParticlePruner::getDaughterKeys	(	std::vector< size_t > &	daIndxs,
		std::vector< size_t > &	daNewIndxs,
		const reco::GenParticleRefVector &	daughters
	)		const

private

Definition at line 283 of file GenParticlePruner.cc.

References edm::RefVector< C, T, F >::begin(), edm::RefVector< C, T, F >::empty(), edm::RefVector< C, T, F >::end(), spr::find(), flags_, training_settings::idx, dqmiolumiharvest::j, and edm::Ref< C, T, F >::key().

Referenced by produce().

                                                                                      {
   for (GenParticleRefVector::const_iterator j = daughters.begin(); j != daughters.end(); ++j) {
     GenParticleRef dau = *j;
     if (find(daIndxs.begin(), daIndxs.end(), dau.key()) == daIndxs.end()) {
       daIndxs.push_back(dau.key());
       int idx = flags_[dau.key()];
       if (idx > 0) {
         daNewIndxs.push_back(idx);
       } else {
         const GenParticleRefVector &daus = dau->daughterRefVector();
         if (!daus.empty())
           getDaughterKeys(daIndxs, daNewIndxs, daus);
       }
     }
   }
 }

void GenParticlePruner::getMotherKeys	(	std::vector< size_t > &	moIndxs,
		std::vector< size_t > &	moNewIndxs,
		const reco::GenParticleRefVector &	mothers
	)		const

private

Definition at line 302 of file GenParticlePruner.cc.

References edm::RefVector< C, T, F >::begin(), DEFINE_FWK_MODULE, edm::RefVector< C, T, F >::empty(), edm::RefVector< C, T, F >::end(), spr::find(), flags_, training_settings::idx, dqmiolumiharvest::j, and edm::Ref< C, T, F >::key().

Referenced by produce().

                                                                                  {
   for (GenParticleRefVector::const_iterator j = mothers.begin(); j != mothers.end(); ++j) {
     GenParticleRef mom = *j;
     if (find(moIndxs.begin(), moIndxs.end(), mom.key()) == moIndxs.end()) {
       moIndxs.push_back(mom.key());
       int idx = flags_[mom.key()];
       if (idx >= 0) {
         moNewIndxs.push_back(idx);
       } else {
         const GenParticleRefVector &moms = mom->motherRefVector();
         if (!moms.empty())
           getMotherKeys(moIndxs, moNewIndxs, moms);
       }
     }
   }
 }

void GenParticlePruner::parse	(	const std::string &	selection,
		helper::SelectCode &	code,
		std::string &	cut
	)		const

private

Definition at line 50 of file GenParticlePruner.cc.

References helper::SelectCode::all_, HltBtagPostValidation_cff::c, mps_check::command, edm::errors::Configuration, helper::SelectCode::daughtersDepth_, Exception, f, helper::SelectCode::kAll, helper::SelectCode::kDrop, helper::SelectCode::keepOrDrop_, helper::SelectCode::kFirst, helper::SelectCode::kKeep, helper::SelectCode::kNone, helper::SelectCode::mothersDepth_, dqmiodumpmetadata::n, and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by produce().

                                                                                                       {
   using namespace ::helper;
   size_t f = selection.find_first_not_of(' ');
   size_t n = selection.size();
   string command;
   char c;
   for (; (c = selection[f]) != ' ' && f < n; ++f) {
     command.push_back(c);
   }
   if (command[0] == '+') {
     command.erase(0, 1);
     if (command[0] == '+') {
       command.erase(0, 1);
       code.mothersDepth_ = SelectCode::kAll;
     } else {
       code.mothersDepth_ = SelectCode::kFirst;
     }
   } else
     code.mothersDepth_ = SelectCode::kNone;
 
   if (command[command.size() - 1] == '+') {
     command.erase(command.size() - 1);
     if (command[command.size() - 1] == '+') {
       command.erase(command.size() - 1);
       code.daughtersDepth_ = SelectCode::kAll;
     } else {
       code.daughtersDepth_ = SelectCode::kFirst;
     }
   } else
     code.daughtersDepth_ = SelectCode::kNone;
 
   if (command == "keep")
     code.keepOrDrop_ = SelectCode::kKeep;
   else if (command == "drop")
     code.keepOrDrop_ = SelectCode::kDrop;
   else {
     throw Exception(errors::Configuration) << "invalid selection command: " << command << "\n" << endl;
   }
   for (; f < n; ++f) {
     if (selection[f] != ' ')
       break;
   }
   cut = string(selection, f);
   if (cut[0] == '*')
     cut = string(cut, 0, cut.find_first_of(' '));
   code.all_ = cut == "*";
 }

void GenParticlePruner::produce	(	edm::Event &	evt,
		const edm::EventSetup &	es
	)

overrideprivate

Definition at line 162 of file GenParticlePruner.cc.

References reco::CompositeRefCandidateT< D >::addDaughter(), reco::CompositeRefCandidateT< D >::addMother(), helper::SelectCode::all_, edm::errors::Configuration, TkAlMuonSelectors_cfi::cut, reco::CompositeRefCandidateT< D >::daughterRefVector(), drop, Exception, firstEvent_, flagDaughters(), flagMothers(), flags_, relval_steps::gen(), GenParticle::GenParticle, edm::Event::getByToken(), getDaughterKeys(), getMotherKeys(), edm::Event::getRefBeforePut(), mps_fire::i, indices_, dqmiolumiharvest::j, keep, helper::SelectCode::keepOrDrop_, keepOrDropAll_, WDecay::kNone, reco::CompositeRefCandidateT< D >::motherRefVector(), eostools::move(), dqmiodumpmetadata::n, MillePedeFileConverter_cfg::out, AlCaHLTBitMon_ParallelJobs::p, parse(), edm::Event::put(), recursiveFlagDaughters(), recursiveFlagMothers(), cuy::rep, PdgEntryReplacer::replace(), EgammaValidation_Wenu_cff::sel, select_, selection_, TrackRefitter_38T_cff::src, srcToken_, and reco::GenParticle::statusFlags().

                                                                 {
   if (firstEvent_) {
     PdgEntryReplacer rep(es);
     for (vector<string>::const_iterator i = selection_.begin(); i != selection_.end(); ++i) {
       string cut;
       ::helper::SelectCode code;
       parse(*i, code, cut);
       if (code.all_) {
         if (i != selection_.begin())
           throw Exception(errors::Configuration)
               << "selections \"keep *\" and \"drop *\" can be used only as first options. Here used in position # "
               << (i - selection_.begin()) + 1 << "\n"
               << endl;
         switch (code.keepOrDrop_) {
           case ::helper::SelectCode::kDrop:
             keepOrDropAll_ = drop;
             break;
           case ::helper::SelectCode::kKeep:
             keepOrDropAll_ = keep;
         };
       } else {
         cut = rep.replace(cut);
         select_.push_back(make_pair(StringCutObjectSelector<GenParticle>(cut), code));
       }
     }
     firstEvent_ = false;
   }
 
   using namespace ::helper;
   Handle<GenParticleCollection> src;
   evt.getByToken(srcToken_, src);
   const size_t n = src->size();
   flags_.clear();
   flags_.resize(n, keepOrDropAll_);
   for (size_t j = 0; j < select_.size(); ++j) {
     const pair<StringCutObjectSelector<GenParticle>, SelectCode> &sel = select_[j];
     SelectCode code = sel.second;
     const StringCutObjectSelector<GenParticle> &cut = sel.first;
     for (size_t i = 0; i < n; ++i) {
       const GenParticle &p = (*src)[i];
       if (cut(p)) {
         int keepOrDrop = keep;
         switch (code.keepOrDrop_) {
           case SelectCode::kKeep:
             keepOrDrop = keep;
             break;
           case SelectCode::kDrop:
             keepOrDrop = drop;
         };
         flags_[i] = keepOrDrop;
         std::vector<size_t> allIndicesDa;
         std::vector<size_t> allIndicesMo;
         switch (code.daughtersDepth_) {
           case SelectCode::kAll:
             recursiveFlagDaughters(i, *src, keepOrDrop, allIndicesDa);
             break;
           case SelectCode::kFirst:
             flagDaughters(p, keepOrDrop);
             break;
           case SelectCode::kNone:;
         };
         switch (code.mothersDepth_) {
           case SelectCode::kAll:
             recursiveFlagMothers(i, *src, keepOrDrop, allIndicesMo);
             break;
           case SelectCode::kFirst:
             flagMothers(p, keepOrDrop);
             break;
           case SelectCode::kNone:;
         };
       }
     }
   }
   indices_.clear();
   int counter = 0;
   for (size_t i = 0; i < n; ++i) {
     if (flags_[i] == keep) {
       indices_.push_back(i);
       flags_[i] = counter++;
     } else {
       flags_[i] = -1;  //set to invalid ref
     }
   }
 
   auto out = std::make_unique<GenParticleCollection>();
   GenParticleRefProd outRef = evt.getRefBeforePut<GenParticleCollection>();
   out->reserve(counter);
 
   for (vector<size_t>::const_iterator i = indices_.begin(); i != indices_.end(); ++i) {
     size_t index = *i;
     const GenParticle &gen = (*src)[index];
     const LeafCandidate &part = gen;
     out->push_back(GenParticle(part));
     GenParticle &newGen = out->back();
     //fill status flags
     newGen.statusFlags() = gen.statusFlags();
     // The "daIndxs" and "moIndxs" keep a list of the keys for the mother/daughter
     // parentage/descendency. In some cases, a circular referencing is encountered,
     // which would result in an infinite loop. The list is checked to
     // avoid this.
     vector<size_t> daIndxs, daNewIndxs;
     getDaughterKeys(daIndxs, daNewIndxs, gen.daughterRefVector());
     std::sort(daNewIndxs.begin(), daNewIndxs.end());
     for (size_t i = 0; i < daNewIndxs.size(); ++i)
       newGen.addDaughter(GenParticleRef(outRef, daNewIndxs[i]));
 
     vector<size_t> moIndxs, moNewIndxs;
     getMotherKeys(moIndxs, moNewIndxs, gen.motherRefVector());
     std::sort(moNewIndxs.begin(), moNewIndxs.end());
     for (size_t i = 0; i < moNewIndxs.size(); ++i)
       newGen.addMother(GenParticleRef(outRef, moNewIndxs[i]));
   }
 
   edm::OrphanHandle<reco::GenParticleCollection> oh = evt.put(std::move(out));
   auto orig2new = std::make_unique<edm::Association<reco::GenParticleCollection>>(oh);
   edm::Association<reco::GenParticleCollection>::Filler orig2newFiller(*orig2new);
   orig2newFiller.insert(src, flags_.begin(), flags_.end());
   orig2newFiller.fill();
   evt.put(std::move(orig2new));
 }

void GenParticlePruner::recursiveFlagDaughters	(	size_t	index,
		const reco::GenParticleCollection &	src,
		int	keepOrDrop,
		std::vector< size_t > &	allIndices
	)

private

Definition at line 120 of file GenParticlePruner.cc.

References edm::RefVector< C, T, F >::begin(), BPhysicsValidation_cfi::daughters, edm::RefVector< C, T, F >::end(), spr::find(), flags_, and mps_fire::i.

Referenced by produce().

                                                                               {
   GenParticleRefVector daughters = src[index].daughterRefVector();
   // avoid infinite recursion if the daughters are set to "this" particle.
   size_t cachedIndex = index;
   for (GenParticleRefVector::const_iterator i = daughters.begin(); i != daughters.end(); ++i) {
     index = i->key();
     // To also avoid infinite recursion if a "loop" is found in the daughter list,
     // check to make sure the index hasn't already been added.
     if (find(allIndices.begin(), allIndices.end(), index) == allIndices.end()) {
       allIndices.push_back(index);
       if (cachedIndex != index) {
         flags_[index] = keepOrDrop;
         recursiveFlagDaughters(index, src, keepOrDrop, allIndices);
       }
     }
   }
 }

void GenParticlePruner::recursiveFlagMothers	(	size_t	index,
		const reco::GenParticleCollection &	src,
		int	keepOrDrop,
		std::vector< size_t > &	allIndices
	)

private

Definition at line 141 of file GenParticlePruner.cc.

References edm::RefVector< C, T, F >::begin(), edm::RefVector< C, T, F >::end(), spr::find(), flags_, mps_fire::i, and HLTTauQualityTests_cfi::mothers.

Referenced by produce().

                                                                             {
   GenParticleRefVector mothers = src[index].motherRefVector();
   // avoid infinite recursion if the mothers are set to "this" particle.
   size_t cachedIndex = index;
   for (GenParticleRefVector::const_iterator i = mothers.begin(); i != mothers.end(); ++i) {
     index = i->key();
     // To also avoid infinite recursion if a "loop" is found in the daughter list,
     // check to make sure the index hasn't already been added.
     if (find(allIndices.begin(), allIndices.end(), index) == allIndices.end()) {
       allIndices.push_back(index);
       if (cachedIndex != index) {
         flags_[index] = keepOrDrop;
         recursiveFlagMothers(index, src, keepOrDrop, allIndices);
       }
     }
   }
 }

Member Data Documentation

bool GenParticlePruner::firstEvent_

private

Definition at line 28 of file GenParticlePruner.cc.

Referenced by produce().

std::vector<int> GenParticlePruner::flags_

private

Definition at line 33 of file GenParticlePruner.cc.

Referenced by flagDaughters(), flagMothers(), getDaughterKeys(), getMotherKeys(), produce(), recursiveFlagDaughters(), and recursiveFlagMothers().

std::vector<size_t> GenParticlePruner::indices_

private

Definition at line 34 of file GenParticlePruner.cc.

Referenced by produce().

int GenParticlePruner::keepOrDropAll_

private

Definition at line 30 of file GenParticlePruner.cc.

Referenced by produce().

std::vector<std::pair<StringCutObjectSelector<reco::GenParticle>, helper::SelectCode> > GenParticlePruner::select_

private

Definition at line 32 of file GenParticlePruner.cc.

Referenced by produce().

std::vector<std::string> GenParticlePruner::selection_

private

Definition at line 31 of file GenParticlePruner.cc.

Referenced by produce().

edm::EDGetTokenT<reco::GenParticleCollection> GenParticlePruner::srcToken_

private

Definition at line 29 of file GenParticlePruner.cc.

Referenced by produce().

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation