PartonSelector Class Reference

Inheritance diagram for PartonSelector:

Public Member Functions
	PartonSelector (const edm::ParameterSet &)
	~PartonSelector ()
Public Member Functions inherited from edm::global::EDProducer<>
	EDProducer ()=default
Public Member Functions inherited from edm::global::EDProducerBase
	EDProducerBase ()
ModuleDescription const &	moduleDescription () const
virtual	~EDProducerBase ()
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 ()
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, std::unordered_multimap< std::string, edm::ProductResolverIndex > 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)

Private Member Functions
virtual void	produce (edm::StreamID, edm::Event &, const edm::EventSetup &) const override

Private Attributes
bool	acceptNoDaughters
unsigned int	skipFirstN
edm::EDGetTokenT< reco::GenParticleCollection >	tokenGenParticles_
bool	withLeptons
bool	withTop

Additional Inherited Members
Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase	ModuleType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels
Static Public Member Functions inherited from edm::global::EDProducerBase
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)
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 36 of file PartonSelector.cc.

Constructor & Destructor Documentation

PartonSelector::PartonSelector

(

const edm::ParameterSet &

iConfig

)

Definition at line 53 of file PartonSelector.cc.

References edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), and SelectPartons_cff::withLeptons.

{
    produces<reco::GenParticleRefVector>();
    withLeptons           = iConfig.getParameter<bool>("withLeptons");
    tokenGenParticles_ = consumes<reco::GenParticleCollection>(iConfig.getParameter<edm::InputTag>("src"));
    if ( iConfig.exists("acceptNoDaughters") ) {
      acceptNoDaughters = iConfig.getParameter<bool>("acceptNoDaughters");
    } else {
    acceptNoDaughters=false;
    }
    if ( iConfig.exists("skipFirstN") ) {
      skipFirstN = iConfig.getParameter<unsigned int>("skipFirstN");
    } else {
    skipFirstN=6;
    }
    if ( iConfig.exists("withTop") ) {
      withTop = iConfig.getParameter<bool>("withTop");
    } else {
      withTop = false;
    }
}

PartonSelector::~PartonSelector

(

)

Definition at line 77 of file PartonSelector.cc.

{
}

Member Function Documentation

void PartonSelector::produce ( edm::StreamID  , edm::Event &  iEvent, const edm::EventSetup &  iEs  ) const

overrideprivatevirtual

Definition at line 83 of file PartonSelector.cc.

References funct::abs(), reco::CompositeRefCandidateT< D >::daughter(), DEFINE_FWK_MODULE, GenHFHadronMatcher_cff::flavour, edm::Event::getByToken(), mps_fire::i, funct::m, eostools::move(), nPart(), reco::CompositeRefCandidateT< D >::numberOfDaughters(), HadronAndPartonSelector_cfi::particles, reco::Candidate::pdgId(), reco::LeafCandidate::pdgId(), edm::Event::put(), reco::LeafCandidate::status(), and SelectPartons_cff::withLeptons.

{

  //edm::Handle <reco::CandidateView> particles;
  edm::Handle <reco::GenParticleCollection> particles;
  iEvent.getByToken (tokenGenParticles_, particles );
  edm::LogVerbatim("PartonSelector") << "=== GenParticle size:" << particles->size();
  int nPart=0;

  auto thePartons = std::make_unique<GenParticleRefVector>();

  for (size_t m = 0; m < particles->size(); m++) {

    // Don't take into account first 6 particles in generator list
    if (m<skipFirstN) continue;

    const GenParticle & aParticle = (*particles)[ m ];

    bool isAParton = false;
    bool isALepton = false;
    int flavour = abs(aParticle.pdgId());
    if(flavour == 1 ||
       flavour == 2 ||
       flavour == 3 ||
       flavour == 4 ||
       flavour == 5 ||
       (flavour == 6 && withTop) ||
       flavour == 21 ) isAParton = true;
    if(flavour == 11 ||
       flavour == 12 ||
       flavour == 13 ||
       flavour == 14 ||
       flavour == 15 ||
       flavour == 16 ) isALepton = true;


    //Add Partons status 3
    if( aParticle.status() == 3 && isAParton ) {
      thePartons->push_back( GenParticleRef( particles, m ) );
      nPart++;
    }

    //Add Partons status 2
    int nparton_daughters = 0;
    if( ( aParticle.numberOfDaughters() > 0 || acceptNoDaughters) && isAParton ) {

      for (unsigned int i=0; i < aParticle.numberOfDaughters(); i++){

    int daughterFlavour = abs(aParticle.daughter(i)->pdgId());
    if( (daughterFlavour == 1 || daughterFlavour == 2 || daughterFlavour == 3 ||
         daughterFlavour == 4 || daughterFlavour == 5 || daughterFlavour == 6 || daughterFlavour == 21)) {
          nparton_daughters++;
    }

      }
      if(nparton_daughters == 0){
      nPart++;
      thePartons->push_back( GenParticleRef( particles, m ) );
      }

    }

    //Add Leptons
    // Here you have to decide what to do with taus....
    // Now all leptons, including e and mu from leptonic tau decays, are added
    if( withLeptons && aParticle.status() == 3 && isALepton ) {
      thePartons->push_back( GenParticleRef( particles, m ) );
      nPart++;
    }
  }

  edm::LogVerbatim("PartonSelector") << "=== GenParticle selected:" << nPart;
  iEvent.put(std::move(thePartons) );

}

Member Data Documentation

bool PartonSelector::acceptNoDaughters

private

Definition at line 47 of file PartonSelector.cc.

unsigned int PartonSelector::skipFirstN

private

Definition at line 48 of file PartonSelector.cc.

edm::EDGetTokenT<reco::GenParticleCollection> PartonSelector::tokenGenParticles_

private

Definition at line 49 of file PartonSelector.cc.

bool PartonSelector::withLeptons

private

Definition at line 45 of file PartonSelector.cc.

bool PartonSelector::withTop

private

Definition at line 46 of file PartonSelector.cc.