PartonSelector Class Reference

Inheritance diagram for PartonSelector:

Public Member Functions
	PartonSelector (const edm::ParameterSet &)
	~PartonSelector ()
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
ModuleDescription const &	moduleDescription () const
virtual	~EDProducer ()
Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...
virtual	~ProducerBase ()
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const
	EDConsumerBase ()
ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const
void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)
virtual	~EDConsumerBase ()

Private Member Functions
virtual void	produce (edm::Event &, const edm::EventSetup &) 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::EDProducer
typedef EDProducer	ModuleType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)
Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
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(), and edm::ParameterSet::getParameter().

{
    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::Event &  iEvent, const edm::EventSetup &  iEs  )

overrideprivatevirtual

Implements edm::EDProducer.

Definition at line 83 of file PartonSelector.cc.

References funct::abs(), reco::CompositeRefCandidateT< D >::daughter(), reco::flavour(), edm::Event::getByToken(), i, contentValuesFiles::m, nPart(), reco::CompositeRefCandidateT< D >::numberOfDaughters(), reco::Candidate::pdgId(), reco::LeafCandidate::pdgId(), edm::Event::put(), and reco::LeafCandidate::status().

{

  //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_ptr<GenParticleRefVector> thePartons ( new 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( 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.