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
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)
virtual	~ProducerBase () noexcept(false)
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
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().