InputGenJetsParticleSelector Class Reference

#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	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () 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
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)
	~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 const &)=delete
	EDConsumerBase (EDConsumerBase &&)=default
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 const &	operator= (EDConsumerBase const &)=delete
EDConsumerBase &	operator= (EDConsumerBase &&)=default
bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
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::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)
EDGetToken	consumes (const TypeToGet &id, 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 ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 28 of file InputGenJetsParticleSelector.h.

Member Typedef Documentation

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

Definition at line 31 of file InputGenJetsParticleSelector.h.

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

Definition at line 32 of file InputGenJetsParticleSelector.h.

Member Enumeration Documentation

enum InputGenJetsParticleSelector::ResonanceState

Enumerator
kNo
kDirect
kIndirect

Definition at line 62 of file InputGenJetsParticleSelector.h.

                      {
    kNo = 0,
    kDirect,
    kIndirect
  };

Constructor & Destructor Documentation

InputGenJetsParticleSelector::InputGenJetsParticleSelector

(

const edm::ParameterSet &

params

)

Definition at line 49 of file InputGenJetsParticleSelector.cc.

References edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), input_genpartcoll_token_, input_prunedgenpartcoll_token_, inTag, isMiniAOD, edm::InputTag::label(), partonicFinalState, prunedInTag, setExcludeFromResonancePids(), and setIgnoredParticles().

                                                                                        :
  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);
      
}

InputGenJetsParticleSelector::~InputGenJetsParticleSelector ( )

override

Definition at line 76 of file InputGenJetsParticleSelector.cc.

{}

InputGenJetsParticleSelector::InputGenJetsParticleSelector ( )

inlineprivate

Definition at line 76 of file InputGenJetsParticleSelector.h.

{} //should not be used!

Member Function Documentation

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

Definition at line 189 of file InputGenJetsParticleSelector.cc.

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

Referenced by produce().

{
  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;
    case kIndirect:
      return kIndirect;
    }
  }
return kNo;
}

bool InputGenJetsParticleSelector::getExcludeResonances ( ) const

inline

Definition at line 41 of file InputGenJetsParticleSelector.h.

References excludeResonances.

{ return excludeResonances; }

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

inline

Definition at line 44 of file InputGenJetsParticleSelector.h.

References ignoreParticleIDs.

    { return ignoreParticleIDs; }

bool InputGenJetsParticleSelector::getPartonicFinalState ( ) const

inline

Definition at line 40 of file InputGenJetsParticleSelector.h.

References partonicFinalState.

{ return partonicFinalState; }

double InputGenJetsParticleSelector::getPtMin ( ) const

inline

Definition at line 43 of file InputGenJetsParticleSelector.h.

References ptMin.

{ return ptMin; }

bool InputGenJetsParticleSelector::getTausAndJets ( ) const

inline

Definition at line 42 of file InputGenJetsParticleSelector.h.

References tausAsJets.

{ return tausAsJets; }

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

Definition at line 231 of file InputGenJetsParticleSelector.cc.

References testPartonChildren().

Referenced by produce(), and setPtMin().

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

bool InputGenJetsParticleSelector::isExcludedFromResonance ( int  pdgId ) const

private

Definition at line 122 of file InputGenJetsParticleSelector.cc.

References excludeFromResonancePids, createfilelist::int, and common_cff::pdgId.

Referenced by fromResonance().

{
  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;
 
}

bool InputGenJetsParticleSelector::isHadron ( int  pdgId )

static

Definition at line 98 of file InputGenJetsParticleSelector.cc.

References common_cff::pdgId.

Referenced by setPtMin(), and testPartonChildren().

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

bool InputGenJetsParticleSelector::isIgnored

(

int

pdgId

)

const

Definition at line 112 of file InputGenJetsParticleSelector.cc.

References ignoreParticleIDs, createfilelist::int, and common_cff::pdgId.

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

{
  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;
}

bool InputGenJetsParticleSelector::isParton

(

int

pdgId

)

const

Definition at line 90 of file InputGenJetsParticleSelector.cc.

References common_cff::pdgId, and tausAsJets.

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

                                                          {
  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)
}

bool InputGenJetsParticleSelector::isResonance ( int  pdgId )

static

Definition at line 105 of file InputGenJetsParticleSelector.cc.

References common_cff::pdgId.

Referenced by fromResonance(), and setPtMin().

{
  // 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..
}

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

override

Definition at line 239 of file InputGenJetsParticleSelector.cc.

References edm::View< T >::begin(), DEFINE_FWK_MODULE, edm::View< T >::end(), excludeResonances, fromResonance(), GenHFHadronMatcher_cfi::genParticles, edm::Event::getByToken(), hasPartonChildren(), mps_fire::i, training_settings::idx, input_genpartcoll_token_, input_prunedgenpartcoll_token_, align::invalid, invalidateTree(), isIgnored(), isMiniAOD, isParton(), eostools::move(), reco::Candidate::numberOfDaughters(), HadronAndPartonSelector_cfi::particles, partonicFinalState, reco::Candidate::pdgId(), prunedGenParticles_cfi::prunedGenParticles, reco::Candidate::pt(), ptMin, edm::View< T >::ptrAt(), edm::Event::put(), findQualityFiles::size, and reco::Candidate::status().

                                                                                                          {

 
  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_));
}

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

private

Definition at line 84 of file InputGenJetsParticleSelector.cc.

References excludeFromResonancePids.

Referenced by InputGenJetsParticleSelector().

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

void InputGenJetsParticleSelector::setExcludeResonances ( bool  flag = true )

inline

Definition at line 49 of file InputGenJetsParticleSelector.h.

References excludeResonances, and RemoveAddSevLevel::flag.

  { excludeResonances = flag; }

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

private

Definition at line 78 of file InputGenJetsParticleSelector.cc.

References ignoreParticleIDs.

Referenced by InputGenJetsParticleSelector().

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

void InputGenJetsParticleSelector::setPartonicFinalState ( bool  flag = true )

inline

Definition at line 47 of file InputGenJetsParticleSelector.h.

References RemoveAddSevLevel::flag, and partonicFinalState.

  { partonicFinalState = flag; }

void InputGenJetsParticleSelector::setPtMin ( double  ptMin )

inline

Definition at line 52 of file InputGenJetsParticleSelector.h.

References hasPartonChildren(), align::invalid, isHadron(), isIgnored(), isParton(), isResonance(), AlCaHLTBitMon_ParallelJobs::p, common_cff::pdgId, and ptMin.

{ this->ptMin = ptMin; }

void InputGenJetsParticleSelector::setTausAsJets ( bool  flag = true )

inline

Definition at line 51 of file InputGenJetsParticleSelector.h.

References RemoveAddSevLevel::flag, and tausAsJets.

{ tausAsJets = flag; }

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

private

Definition at line 165 of file InputGenJetsParticleSelector.cc.

References reco::Candidate::daughter(), mps_fire::i, training_settings::idx, isHadron(), isParton(), npart, reco::Candidate::numberOfDaughters(), partIdx(), reco::Candidate::pdgId(), and mps_fire::result.

Referenced by hasPartonChildren(), and invalidateTree().

{
  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;
}

Member Data Documentation

std::vector<unsigned int> InputGenJetsParticleSelector::excludeFromResonancePids

private

Definition at line 85 of file InputGenJetsParticleSelector.h.

Referenced by isExcludedFromResonance(), and setExcludeFromResonancePids().

bool InputGenJetsParticleSelector::excludeResonances

private

Definition at line 91 of file InputGenJetsParticleSelector.h.

Referenced by getExcludeResonances(), produce(), and setExcludeResonances().

std::vector<unsigned int> InputGenJetsParticleSelector::ignoreParticleIDs

private

Definition at line 84 of file InputGenJetsParticleSelector.h.

Referenced by getIgnoredParticles(), isIgnored(), and setIgnoredParticles().

edm::EDGetTokenT<reco::CandidateView> InputGenJetsParticleSelector::input_genpartcoll_token_

private

Definition at line 96 of file InputGenJetsParticleSelector.h.

Referenced by InputGenJetsParticleSelector(), and produce().

edm::EDGetTokenT<reco::CandidateView> InputGenJetsParticleSelector::input_prunedgenpartcoll_token_

private

Definition at line 97 of file InputGenJetsParticleSelector.h.

Referenced by InputGenJetsParticleSelector(), and produce().

edm::InputTag InputGenJetsParticleSelector::inTag

private

Definition at line 78 of file InputGenJetsParticleSelector.h.

Referenced by InputGenJetsParticleSelector().

bool InputGenJetsParticleSelector::isMiniAOD

private

Definition at line 93 of file InputGenJetsParticleSelector.h.

Referenced by InputGenJetsParticleSelector(), and produce().

bool InputGenJetsParticleSelector::partonicFinalState

private

Definition at line 90 of file InputGenJetsParticleSelector.h.

Referenced by getPartonicFinalState(), InputGenJetsParticleSelector(), produce(), and setPartonicFinalState().

edm::InputTag InputGenJetsParticleSelector::prunedInTag

private

Definition at line 79 of file InputGenJetsParticleSelector.h.

Referenced by InputGenJetsParticleSelector().

double InputGenJetsParticleSelector::ptMin

private

Definition at line 94 of file InputGenJetsParticleSelector.h.

Referenced by getPtMin(), produce(), and setPtMin().

bool InputGenJetsParticleSelector::tausAsJets

private

Definition at line 92 of file InputGenJetsParticleSelector.h.

Referenced by getTausAndJets(), isParton(), and setTausAsJets().