#include <InputGenJetsParticleSelector.h>

Inheritance diagram for InputGenJetsParticleSelector:

Public Types
typedef std::vector< bool >	ParticleBitmap

typedef std::vector< const reco::Candidate * >	ParticleVector

enum	ResonanceState { kNo = 0, kDirect, kIndirect }

Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase	ModuleType

Public Types inherited from edm::ProducerBase
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const , const char , edm::ProductResolverIndex > >

typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Public Member Functions
ResonanceState	fromResonance (ParticleBitmap &invalid, const ParticleVector &p, const reco::Candidate *particle) const

bool	getExcludeResonances () const

const std::vector< unsigned int > &	getIgnoredParticles () const

bool	getPartonicFinalState () const

double	getPtMin () const

bool	getTausAndJets () const

bool	hasPartonChildren (ParticleBitmap &invalid, const ParticleVector &p, const reco::Candidate *particle) const

	InputGenJetsParticleSelector (const edm::ParameterSet &)

bool	isIgnored (int pdgId) const

bool	isParton (int pdgId) const

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

void	setExcludeResonances (bool flag=true)

void	setPartonicFinalState (bool flag=true)

void	setPtMin (double ptMin)

void	setTausAsJets (bool flag=true)

	~InputGenJetsParticleSelector () override

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

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndRuns () const final

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

bool	wantsStreamLuminosityBlocks () const final

bool	wantsStreamRuns () const final

Public Member Functions inherited from edm::global::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

	~EDProducerBase () override

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const

std::vector< bool > const &	recordProvenanceList () const

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)

TypeLabelList const &	typeLabelList () const
	used by the fwk to register the list of products of this module More...

	~ProducerBase () noexcept(false) override

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase &&)=default

	EDConsumerBase (EDConsumerBase const &)=delete

ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase &	operator= (EDConsumerBase &&)=default

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Static Public Member Functions
static bool	isHadron (int pdgId)

static bool	isResonance (int pdgId)

Static Public Member Functions inherited from edm::global::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Private Member Functions
	InputGenJetsParticleSelector ()

bool	isExcludedFromResonance (int pdgId) const

void	setExcludeFromResonancePids (const std::vector< unsigned int > &particleIDs)

void	setIgnoredParticles (const std::vector< unsigned int > &particleIDs)

int	testPartonChildren (ParticleBitmap &invalid, const ParticleVector &p, const reco::Candidate *particle) const

Private Attributes
std::vector< unsigned int >	excludeFromResonancePids

bool	excludeResonances

std::vector< unsigned int >	ignoreParticleIDs

edm::EDGetTokenT< reco::CandidateView >	input_genpartcoll_token_

edm::EDGetTokenT< reco::CandidateView >	input_prunedgenpartcoll_token_

edm::InputTag	inTag

bool	isMiniAOD

bool	partonicFinalState

edm::InputTag	prunedInTag

double	ptMin

bool	tausAsJets

Additional Inherited Members
Protected Member Functions inherited from edm::ProducerBase
template<class ProductType >
BranchAliasSetterT< ProductType >	produces ()
	declare what type of product will make and with which optional label More...

template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType >	produces ()

template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType >	produces ()

BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)

template<BranchType B>
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)

template<Transition B>
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)

template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType >	produces (std::string instanceName)

template<class ProductType >
BranchAliasSetterT< ProductType >	produces (std::string instanceName)

template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType >	produces (std::string instanceName)

ProducesCollector	producesCollector ()

Protected Member Functions inherited from edm::EDConsumerBase
EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

Detailed Description

Definition at line 27 of file InputGenJetsParticleSelector.h.

Member Typedef Documentation

◆ ParticleBitmap

typedef std::vector<bool> InputGenJetsParticleSelector::ParticleBitmap

Definition at line 30 of file InputGenJetsParticleSelector.h.

◆ ParticleVector

typedef std::vector<const reco::Candidate *> InputGenJetsParticleSelector::ParticleVector

Definition at line 31 of file InputGenJetsParticleSelector.h.

Member Enumeration Documentation

◆ ResonanceState

enum InputGenJetsParticleSelector::ResonanceState

Enumerator
kNo
kDirect
kIndirect

Definition at line 56 of file InputGenJetsParticleSelector.h.

56 { kNo = 0, kDirect, kIndirect };

Constructor & Destructor Documentation

◆ InputGenJetsParticleSelector() [1/2]

InputGenJetsParticleSelector::InputGenJetsParticleSelector ( const edm::ParameterSet & params )

Definition at line 49 of file InputGenJetsParticleSelector.cc.

     : inTag(params.getParameter<edm::InputTag>("src")),
       prunedInTag(params.exists("prunedGenParticles") ? params.getParameter<edm::InputTag>("prunedGenParticles")
                                                       : edm::InputTag("prunedGenParticles")),
       partonicFinalState(params.getParameter<bool>("partonicFinalState")),
       excludeResonances(params.getParameter<bool>("excludeResonances")),
       tausAsJets(params.getParameter<bool>("tausAsJets")),
       ptMin(0.0) {
   if (params.exists("ignoreParticleIDs"))
     setIgnoredParticles(params.getParameter<std::vector<unsigned int> >("ignoreParticleIDs"));
   setExcludeFromResonancePids(params.getParameter<std::vector<unsigned int> >("excludeFromResonancePids"));
   isMiniAOD =
       (params.exists("isMiniAOD") ? params.getParameter<bool>("isMiniAOD") : (inTag.label() == "packedGenParticles"));
  
   if (isMiniAOD && partonicFinalState) {
     edm::LogError("PartonicFinalStateFromMiniAOD")
         << "Partonic final state not supported for MiniAOD. Falling back to the stable particle selection.";
     partonicFinalState = false;
   }
  
   produces<reco::CandidatePtrVector>();
  
   input_genpartcoll_token_ = consumes<reco::CandidateView>(inTag);
   if (isMiniAOD)
     input_prunedgenpartcoll_token_ = consumes<reco::CandidateView>(prunedInTag);
 }

References input_genpartcoll_token_, input_prunedgenpartcoll_token_, inTag, isMiniAOD, edm::InputTag::label(), CalibrationSummaryClient_cfi::params, partonicFinalState, prunedInTag, setExcludeFromResonancePids(), and setIgnoredParticles().

◆ ~InputGenJetsParticleSelector()

InputGenJetsParticleSelector::~InputGenJetsParticleSelector ( )

override

Definition at line 76 of file InputGenJetsParticleSelector.cc.

76 {}

◆ InputGenJetsParticleSelector() [2/2]

InputGenJetsParticleSelector::InputGenJetsParticleSelector ( )

inlineprivate

Definition at line 63 of file InputGenJetsParticleSelector.h.

63 {} //should not be used!

Member Function Documentation

◆ fromResonance()

InputGenJetsParticleSelector::ResonanceState InputGenJetsParticleSelector::fromResonance	(	ParticleBitmap &	invalid,
		const ParticleVector &	p,
		const reco::Candidate *	particle
	)		const

Definition at line 172 of file InputGenJetsParticleSelector.cc.

                                                                                            {
   unsigned int idx = partIdx(p, particle);
   int id = particle->pdgId();
  
   if (invalid[idx])
     return kIndirect;
  
   if (isResonance(id) && (particle->status() == 3 || particle->status() == 22)) {
     return kDirect;
   }
  
   if (!isIgnored(id) && (isParton(id)))
     return kNo;
  
   unsigned int nMo = particle->numberOfMothers();
   if (!nMo)
     return kNo;
  
   for (unsigned int i = 0; i < nMo; ++i) {
     ResonanceState result = fromResonance(invalid, p, particle->mother(i));
     switch (result) {
       case kNo:
         break;
       case kDirect:
         if (particle->mother(i)->pdgId() == id || isResonance(id))
           return kDirect;
         if (!isExcludedFromResonance(id))
           break;
         [[fallthrough]];
       case kIndirect:
         return kIndirect;
     }
   }
   return kNo;
 }

References mps_fire::i, training_settings::idx, invalid, isExcludedFromResonance(), isIgnored(), isParton(), isResonance(), kDirect, kIndirect, kNo, reco::Candidate::mother(), reco::Candidate::numberOfMothers(), AlCaHLTBitMon_ParallelJobs::p, partIdx(), reco::Candidate::pdgId(), mps_fire::result, and reco::Candidate::status().

Referenced by produce().

◆ getExcludeResonances()

bool InputGenJetsParticleSelector::getExcludeResonances ( ) const

inline

Definition at line 40 of file InputGenJetsParticleSelector.h.

40 { return excludeResonances; }

References excludeResonances.

◆ getIgnoredParticles()

const std::vector<unsigned int>& InputGenJetsParticleSelector::getIgnoredParticles ( ) const

inline

Definition at line 43 of file InputGenJetsParticleSelector.h.

43 { return ignoreParticleIDs; }

References ignoreParticleIDs.

◆ getPartonicFinalState()

bool InputGenJetsParticleSelector::getPartonicFinalState ( ) const

inline

Definition at line 39 of file InputGenJetsParticleSelector.h.

39 { return partonicFinalState; }

References partonicFinalState.

◆ getPtMin()

double InputGenJetsParticleSelector::getPtMin ( ) const

inline

Definition at line 42 of file InputGenJetsParticleSelector.h.

42 { return ptMin; }

References ptMin.

◆ getTausAndJets()

bool InputGenJetsParticleSelector::getTausAndJets ( ) const

inline

Definition at line 41 of file InputGenJetsParticleSelector.h.

41 { return tausAsJets; }

References tausAsJets.

◆ hasPartonChildren()

bool InputGenJetsParticleSelector::hasPartonChildren	(	ParticleBitmap &	invalid,
		const ParticleVector &	p,
		const reco::Candidate *	particle
	)		const

Definition at line 209 of file InputGenJetsParticleSelector.cc.

                                                                                           {
   return testPartonChildren(invalid, p, particle) > 0;
 }

References invalid, AlCaHLTBitMon_ParallelJobs::p, and testPartonChildren().

Referenced by produce().

◆ isExcludedFromResonance()

bool InputGenJetsParticleSelector::isExcludedFromResonance ( int pdgId ) const

private

Definition at line 113 of file InputGenJetsParticleSelector.cc.

                                                                           {
   pdgId = pdgId > 0 ? pdgId : -pdgId;
   std::vector<unsigned int>::const_iterator pos =
       std::lower_bound(excludeFromResonancePids.begin(), excludeFromResonancePids.end(), (unsigned int)pdgId);
   return pos != excludeFromResonancePids.end() && *pos == (unsigned int)pdgId;
 }

References excludeFromResonancePids, createfilelist::int, cuda_std::lower_bound(), and EgammaValidation_cff::pdgId.

Referenced by fromResonance().

◆ isHadron()

bool InputGenJetsParticleSelector::isHadron ( int pdgId )

static

Definition at line 95 of file InputGenJetsParticleSelector.cc.

                                                      {
   pdgId = (pdgId > 0 ? pdgId : -pdgId) % 10000;
   return (pdgId > 100 && pdgId < 900) || (pdgId > 1000 && pdgId < 9000);
 }

References EgammaValidation_cff::pdgId.

Referenced by testPartonChildren().

◆ isIgnored()

bool InputGenJetsParticleSelector::isIgnored ( int pdgId ) const

Definition at line 106 of file InputGenJetsParticleSelector.cc.

                                                             {
   pdgId = pdgId > 0 ? pdgId : -pdgId;
   std::vector<unsigned int>::const_iterator pos =
       std::lower_bound(ignoreParticleIDs.begin(), ignoreParticleIDs.end(), (unsigned int)pdgId);
   return pos != ignoreParticleIDs.end() && *pos == (unsigned int)pdgId;
 }

References ignoreParticleIDs, createfilelist::int, cuda_std::lower_bound(), and EgammaValidation_cff::pdgId.

Referenced by fromResonance(), and produce().

◆ isParton()

bool InputGenJetsParticleSelector::isParton ( int pdgId ) const

Definition at line 88 of file InputGenJetsParticleSelector.cc.

                                                            {
   pdgId = (pdgId > 0 ? pdgId : -pdgId) % 10000;
   return (pdgId > 0 && pdgId < 6) || pdgId == 9 || (tausAsJets && pdgId == 15) || pdgId == 21;
   // tops are not considered "regular" partons
   // but taus eventually are (since they may hadronize later)
 }

References EgammaValidation_cff::pdgId, and tausAsJets.

Referenced by fromResonance(), produce(), and testPartonChildren().

◆ isResonance()

bool InputGenJetsParticleSelector::isResonance ( int pdgId )

static

Definition at line 100 of file InputGenJetsParticleSelector.cc.

                                                         {
   // gauge bosons and tops
   pdgId = (pdgId > 0 ? pdgId : -pdgId) % 10000;
   return (pdgId > 21 && pdgId <= 42) || pdgId == 6 || pdgId == 7 || pdgId == 8;  //BUG! was 21. 22=gamma..
 }

References EgammaValidation_cff::pdgId.

Referenced by fromResonance().

◆ produce()

void InputGenJetsParticleSelector::produce	(	edm::StreamID	,
		edm::Event &	evt,
		const edm::EventSetup &	evtSetup
	)		const

overridevirtual

Implements edm::global::EDProducerBase.

Definition at line 217 of file InputGenJetsParticleSelector.cc.

                                                                                                           {
   auto selected_ = std::make_unique<reco::CandidatePtrVector>();
  
   ParticleVector particles;
  
   edm::Handle<reco::CandidateView> prunedGenParticles;
   if (isMiniAOD) {
     evt.getByToken(input_prunedgenpartcoll_token_, prunedGenParticles);
  
     for (edm::View<reco::Candidate>::const_iterator iter = prunedGenParticles->begin();
          iter != prunedGenParticles->end();
          ++iter) {
       if (iter->status() !=
           1)  // to avoid double-counting, skipping stable particles already contained in the collection of PackedGenParticles
         particles.push_back(&*iter);
     }
   }
  
   edm::Handle<reco::CandidateView> genParticles;
   evt.getByToken(input_genpartcoll_token_, genParticles);
  
   std::map<const reco::Candidate *, size_t> particlePtrIdxMap;
  
   for (edm::View<reco::Candidate>::const_iterator iter = genParticles->begin(); iter != genParticles->end(); ++iter) {
     particles.push_back(&*iter);
     particlePtrIdxMap[&*iter] = (iter - genParticles->begin());
   }
  
   std::sort(particles.begin(), particles.end());
   unsigned int size = particles.size();
  
   ParticleBitmap selected(size, false);
   ParticleBitmap invalid(size, false);
  
   for (unsigned int i = 0; i < size; i++) {
     const reco::Candidate *particle = particles[i];
     if (invalid[i])
       continue;
     if (particle->status() == 1)  // selecting stable particles
       selected[i] = true;
     if (partonicFinalState && isParton(particle->pdgId())) {
       if (particle->numberOfDaughters() == 0 && particle->status() != 1) {
         // some brokenness in event...
         invalid[i] = true;
       } else if (!hasPartonChildren(invalid, particles, particle)) {
         selected[i] = true;
         invalidateTree(invalid, particles, particle);  //this?!?
       }
     }
   }
  
   for (size_t idx = 0; idx < size; ++idx) {
     const reco::Candidate *particle = particles[idx];
     if (!selected[idx] || invalid[idx]) {
       continue;
     }
  
     if (excludeResonances && fromResonance(invalid, particles, particle)) {
       invalid[idx] = true;
       //cout<<"[INPUTSELECTOR] Invalidates FROM RESONANCE!: ["<<setw(4)<<idx<<"] "<<particle->pdgId()<<" "<<particle->pt()<<endl;
       continue;
     }
  
     if (isIgnored(particle->pdgId())) {
       continue;
     }
  
     if (particle->pt() >= ptMin) {
       selected_->push_back(genParticles->ptrAt(particlePtrIdxMap[particle]));
       //cout<<"Finally we have: ["<<setw(4)<<idx<<"] "<<setw(4)<<particle->pdgId()<<" "<<particle->pt()<<endl;
     }
   }
   evt.put(std::move(selected_));
 }

References excludeResonances, fromResonance(), genParticles2HepMC_cfi::genParticles, edm::Event::getByToken(), hasPartonChildren(), mps_fire::i, training_settings::idx, input_genpartcoll_token_, input_prunedgenpartcoll_token_, invalid, invalidateTree(), isIgnored(), isMiniAOD, isParton(), eostools::move(), reco::Candidate::numberOfDaughters(), ecalTrigSettings_cff::particles, partonicFinalState, reco::Candidate::pdgId(), prunedGenParticles_cfi::prunedGenParticles, reco::Candidate::pt(), ptMin, edm::Event::put(), findQualityFiles::size, and reco::Candidate::status().

◆ setExcludeFromResonancePids()

void InputGenJetsParticleSelector::setExcludeFromResonancePids ( const std::vector< unsigned int > & particleIDs )

private

Definition at line 83 of file InputGenJetsParticleSelector.cc.

                                                                                                          {
   excludeFromResonancePids = particleIDs;
   std::sort(excludeFromResonancePids.begin(), excludeFromResonancePids.end());
 }

References excludeFromResonancePids.

Referenced by InputGenJetsParticleSelector().

◆ setExcludeResonances()

void InputGenJetsParticleSelector::setExcludeResonances ( bool flag = true )

inline

Definition at line 46 of file InputGenJetsParticleSelector.h.

46 { excludeResonances = flag; }

References excludeResonances, and RemoveAddSevLevel::flag.

◆ setIgnoredParticles()

void InputGenJetsParticleSelector::setIgnoredParticles ( const std::vector< unsigned int > & particleIDs )

private

Definition at line 78 of file InputGenJetsParticleSelector.cc.

                                                                                                  {
   ignoreParticleIDs = particleIDs;
   std::sort(ignoreParticleIDs.begin(), ignoreParticleIDs.end());
 }

References ignoreParticleIDs.

Referenced by InputGenJetsParticleSelector().

◆ setPartonicFinalState()

void InputGenJetsParticleSelector::setPartonicFinalState ( bool flag = true )

inline

Definition at line 45 of file InputGenJetsParticleSelector.h.

45 { partonicFinalState = flag; }

References RemoveAddSevLevel::flag, and partonicFinalState.

◆ setPtMin()

void InputGenJetsParticleSelector::setPtMin ( double ptMin )

inline

Definition at line 48 of file InputGenJetsParticleSelector.h.

48 { this->ptMin = ptMin; }

References ptMin.

◆ setTausAsJets()

void InputGenJetsParticleSelector::setTausAsJets ( bool flag = true )

inline

Definition at line 47 of file InputGenJetsParticleSelector.h.

47 { tausAsJets = flag; }

References RemoveAddSevLevel::flag, and tausAsJets.

◆ testPartonChildren()

int InputGenJetsParticleSelector::testPartonChildren	(	InputGenJetsParticleSelector::ParticleBitmap &	invalid,
		const ParticleVector &	p,
		const reco::Candidate *	particle
	)		const

private

Definition at line 147 of file InputGenJetsParticleSelector.cc.

                                                                                           {
   unsigned int npart = particle->numberOfDaughters();
   if (!npart) {
     return 0;
   }
  
   for (unsigned int i = 0; i < npart; ++i) {
     unsigned int idx = partIdx(p, particle->daughter(i));
     if (invalid[idx])
       continue;
     if (isParton((particle->daughter(i)->pdgId()))) {
       return 1;
     }
     if (isHadron((particle->daughter(i)->pdgId()))) {
       return -1;
     }
     int result = testPartonChildren(invalid, p, particle->daughter(i));
     if (result)
       return result;
   }
   return 0;
 }