Inheritance diagram for GenParticlePruner:

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

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () 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 95 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 104 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 110 of file GenParticlePruner.cc.

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

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 275 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, 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 296 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, 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 49 of file GenParticlePruner.cc.

References helper::SelectCode::all_, EnergyCorrector::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_, gen::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 154 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_, keep, helper::SelectCode::keepOrDrop_, keepOrDropAll_, WDecay::kNone, reco::CompositeRefCandidateT< D >::motherRefVector(), eostools::move(), gen::n, MillePedeFileConverter_cfg::out, AlCaHLTBitMon_ParallelJobs::p, parse(), edm::Event::put(), recursiveFlagDaughters(), recursiveFlagMothers(), cuy::rep, PdgEntryReplacer::replace(), triggerObjects_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 116 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 135 of file GenParticlePruner.cc.

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

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 27 of file GenParticlePruner.cc.

Referenced by produce().

std::vector<int> GenParticlePruner::flags_

private

Definition at line 32 of file GenParticlePruner.cc.

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

std::vector<size_t> GenParticlePruner::indices_

private

Definition at line 33 of file GenParticlePruner.cc.

Referenced by produce().

int GenParticlePruner::keepOrDropAll_

private

Definition at line 29 of file GenParticlePruner.cc.

Referenced by produce().

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

private

Definition at line 31 of file GenParticlePruner.cc.

Referenced by produce().

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

private

Definition at line 30 of file GenParticlePruner.cc.

Referenced by produce().

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

private

Definition at line 28 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