CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
List of all members | Public Member Functions | Static Public Member Functions | Private Member Functions | Private Attributes
HadronAndPartonSelector Class Reference

Selects hadrons and partons from a collection of GenParticles. More...

#include <PhysicsTools/JetMCAlgos/plugins/HadronAndPartonSelector.cc>

Inheritance diagram for HadronAndPartonSelector:
edm::stream::EDProducer<> edm::stream::EDProducerBase edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 HadronAndPartonSelector (const edm::ParameterSet &)
 
 ~HadronAndPartonSelector ()
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
- Public Member Functions inherited from edm::stream::EDProducerBase
 EDProducerBase ()
 
ModuleDescription const & moduleDescription () const
 
virtual ~EDProducerBase ()
 
- Public Member Functions inherited from edm::ProducerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
 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< ConsumesInfoconsumesInfo () 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 (std::string const &iProcessName, std::string const &iModuleLabel, bool iPrint, std::vector< char const * > &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 ()
 

Static Public Member Functions

static void fillDescriptions (edm::ConfigurationDescriptions &descriptions)
 
- Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 

Private Member Functions

virtual void produce (edm::Event &, const edm::EventSetup &)
 

Private Attributes

const edm::EDGetTokenT
< reco::GenParticleCollection
particlesToken_
 
std::string partonMode_
 
PartonSelectorPtr partonSelector_
 
bool partonSelectorSet_
 
const edm::EDGetTokenT
< GenEventInfoProduct
srcToken_
 

Additional Inherited Members

- Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T...> CacheTypes
 
typedef CacheTypes::GlobalCache GlobalCache
 
typedef AbilityChecker< T...> HasAbility
 
typedef
CacheTypes::LuminosityBlockCache 
LuminosityBlockCache
 
typedef
LuminosityBlockContextT
< LuminosityBlockCache,
RunCache, GlobalCache
LuminosityBlockContext
 
typedef
CacheTypes::LuminosityBlockSummaryCache 
LuminosityBlockSummaryCache
 
typedef CacheTypes::RunCache RunCache
 
typedef RunContextT< RunCache,
GlobalCache
RunContext
 
typedef CacheTypes::RunSummaryCache RunSummaryCache
 
- Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase ModuleType
 
- Public Types inherited from edm::ProducerBase
typedef
ProductRegistryHelper::TypeLabelList 
TypeLabelList
 
- Public Types inherited from edm::EDConsumerBase
typedef ProductLabels Labels
 
- 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

Selects hadrons and partons from a collection of GenParticles.

This producer selects hadrons, partons, and leptons from a collection of GenParticles and stores vectors of EDM references to these particles in the event. The following hadrons are selected:

Older Fortran Monte Carlo generators (Pythia6 and Herwig6) follow the HEPEVT [1] particle status code convention while newer C++ Monte Carlo generators (Pythia8, Herwig++, and Sherpa) follow the HepMC [2] particle status code convention. However, both conventions give considerable freedom in defining the status codes of intermediate particle states. Hence, the parton selection is generator-dependent and is described in each of the parton selectors separately.

Using the provenance information of the GenEventInfoProduct, the producer attempts to automatically determine what generator was used to hadronize events and based on that information decides what parton selection mode to use. It is also possible to enforce any of the supported parton selection modes.

The selected hadrons and partons are finally used by the JetFlavourClustering producer to determine the jet flavour.

The following leptons are selected:

[1] http://cepa.fnal.gov/psm/stdhep/ [2] http://lcgapp.cern.ch/project/simu/HepMC/

Definition at line 79 of file HadronAndPartonSelector.cc.

Constructor & Destructor Documentation

HadronAndPartonSelector::HadronAndPartonSelector ( const edm::ParameterSet iConfig)
explicit

Definition at line 105 of file HadronAndPartonSelector.cc.

References partonSelector_, and partonSelectorSet_.

105  :
106 
107  srcToken_(mayConsume<GenEventInfoProduct>( iConfig.getParameter<edm::InputTag>("src") )),
108  particlesToken_(consumes<reco::GenParticleCollection>( iConfig.getParameter<edm::InputTag>("particles") )),
109  partonMode_(iConfig.getParameter<std::string>("partonMode"))
110 
111 {
112  //register your products
113  produces<reco::GenParticleRefVector>( "bHadrons" );
114  produces<reco::GenParticleRefVector>( "cHadrons" );
115  produces<reco::GenParticleRefVector>( "algorithmicPartons" );
116  produces<reco::GenParticleRefVector>( "physicsPartons" );
117  produces<reco::GenParticleRefVector>( "leptons" );
118 
119  partonSelectorSet_=false;
120  partonSelector_=nullptr;
121 }
T getParameter(std::string const &) const
const edm::EDGetTokenT< reco::GenParticleCollection > particlesToken_
const edm::EDGetTokenT< GenEventInfoProduct > srcToken_
HadronAndPartonSelector::~HadronAndPartonSelector ( )

Definition at line 124 of file HadronAndPartonSelector.cc.

125 {
126 
127  // do anything here that needs to be done at desctruction time
128  // (e.g. close files, deallocate resources etc.)
129 
130 }

Member Function Documentation

void HadronAndPartonSelector::fillDescriptions ( edm::ConfigurationDescriptions descriptions)
static

Definition at line 272 of file HadronAndPartonSelector.cc.

References edm::ConfigurationDescriptions::addDefault(), and edm::ParameterSetDescription::setUnknown().

272  {
273  //The following says we do not know what parameters are allowed so do no validation
274  // Please change this to state exactly what you do use, even if it is no parameters
276  desc.setUnknown();
277  descriptions.addDefault(desc);
278 }
void addDefault(ParameterSetDescription const &psetDescription)
void HadronAndPartonSelector::produce ( edm::Event iEvent,
const edm::EventSetup iSetup 
)
privatevirtual

Implements edm::stream::EDProducerBase.

Definition at line 139 of file HadronAndPartonSelector.cc.

References Exception, edm::Event::getByToken(), edm::Event::getProvenance(), CandMCTagUtils::hasBottom(), CandMCTagUtils::hasCharm(), i, edm::HandleBase::id(), reco::isElectron(), reco::isMuon(), CandMCTagUtils::isParton(), reco::isTau(), edm::HandleBase::isValid(), HLT_FULL_cff::leptons, edm::moduleName(), particlesToken_, partonMode_, partonSelector_, partonSelectorSet_, edm::Event::put(), srcToken_, and AlCaHLTBitMon_QueryRunRegistry::string.

140 {
141  // determine hadronizer type (done only once per job)
142  if( partonMode_=="Auto" )
143  {
144  edm::Handle<GenEventInfoProduct> genEvtInfoProduct;
145  iEvent.getByToken(srcToken_, genEvtInfoProduct);
146 
147  std::string moduleName = "";
148  if( genEvtInfoProduct.isValid() ) {
149  const edm::Provenance& prov = iEvent.getProvenance(genEvtInfoProduct.id());
150  moduleName = edm::moduleName(prov);
151  }
152 
153  if( moduleName.find("Pythia6")!=std::string::npos )
154  partonMode_="Pythia6";
155  else if( moduleName.find("Pythia8")!=std::string::npos )
156  partonMode_="Pythia8";
157  else if( moduleName.find("Herwig6")!=std::string::npos )
158  partonMode_="Herwig6";
159  else if( moduleName.find("ThePEG")!=std::string::npos )
160  partonMode_="Herwig++";
161  else if( moduleName.find("Sherpa")!=std::string::npos )
162  partonMode_="Sherpa";
163  else
164  partonMode_="Undefined";
165  }
166 
167  // set the parton selection mode (done only once per job)
168  if( !partonSelectorSet_ )
169  {
170  if ( partonMode_=="Undefined" )
171  edm::LogWarning("UndefinedPartonMode") << "Could not automatically determine the hadronizer type and set the correct parton selection mode. Parton-based jet flavour will not be defined.";
172  else if ( partonMode_=="Pythia6" )
173  {
175  edm::LogInfo("PartonModeDefined") << "Using Pythia6 parton selection mode.";
176  }
177  else if ( partonMode_=="Pythia8" )
178  {
180  edm::LogInfo("PartonModeDefined") << "Using Pythia8 parton selection mode.";
181  }
182  else if ( partonMode_=="Herwig6" )
183  {
185  edm::LogInfo("PartonModeDefined") << "Using Herwig6 parton selection mode.";
186  }
187  else if ( partonMode_=="Herwig++" )
188  {
190  edm::LogInfo("PartonModeDefined") << "Using Herwig++ parton selection mode.";
191  }
192  else if ( partonMode_=="Sherpa" )
193  {
195  edm::LogInfo("PartonModeDefined") << "Using Sherpa parton selection mode.";
196  }
197  else
198  throw cms::Exception("InvalidPartonMode") <<"Parton selection mode is invalid: " << partonMode_ << ", use Auto | Pythia6 | Pythia8 | Herwig6 | Herwig++ | Sherpa" << std::endl;
199 
200  partonSelectorSet_=true;
201  }
202 
204  iEvent.getByToken(particlesToken_, particles);
205 
206  std::auto_ptr<reco::GenParticleRefVector> bHadrons ( new reco::GenParticleRefVector );
207  std::auto_ptr<reco::GenParticleRefVector> cHadrons ( new reco::GenParticleRefVector );
208  std::auto_ptr<reco::GenParticleRefVector> partons ( new reco::GenParticleRefVector );
209  std::auto_ptr<reco::GenParticleRefVector> physicsPartons ( new reco::GenParticleRefVector );
210  std::auto_ptr<reco::GenParticleRefVector> leptons ( new reco::GenParticleRefVector );
211 
212  // loop over particles and select b and c hadrons and leptons
213  for(reco::GenParticleCollection::const_iterator it = particles->begin(); it != particles->end(); ++it)
214  {
215  // if b hadron
216  if( CandMCTagUtils::hasBottom( *it ) )
217  {
218  // check if any of the daughters is also a b hadron
219  bool hasbHadronDaughter = false;
220  for(size_t i=0; i < it->numberOfDaughters(); ++i)
221  {
222  if( CandMCTagUtils::hasBottom( *(it->daughter(i)) ) ) { hasbHadronDaughter = true; break; }
223  }
224  if( hasbHadronDaughter ) continue; // skip excited b hadrons that have other b hadrons as daughters
225 
226  bHadrons->push_back( reco::GenParticleRef( particles, it - particles->begin() ) );
227  }
228 
229  // if c hadron
230  if( CandMCTagUtils::hasCharm( *it ) )
231  {
232  // check if any of the daughters is also a c hadron
233  bool hascHadronDaughter = false;
234  for(size_t i=0; i < it->numberOfDaughters(); ++i)
235  {
236  if( CandMCTagUtils::hasCharm( *(it->daughter(i)) ) ) { hascHadronDaughter = true; break; }
237  }
238  if( hascHadronDaughter ) continue; // skip excited c hadrons that have other c hadrons as daughters
239 
240  cHadrons->push_back( reco::GenParticleRef( particles, it - particles->begin() ) );
241  }
242 
243  // status==1 electrons and muons
244  if( ( reco::isElectron( *it ) || reco::isMuon( *it ) ) && it->status()==1 )
245  leptons->push_back( reco::GenParticleRef( particles, it - particles->begin() ) );
246 
247  // status==2 taus
248  if( reco::isTau( *it ) && it->status()==2 )
249  leptons->push_back( reco::GenParticleRef( particles, it - particles->begin() ) );
250  }
251 
252  // select partons
253  if ( partonMode_!="Undefined" ) {
254  partonSelector_->run(particles,partons);
255  for(reco::GenParticleCollection::const_iterator it = particles->begin(); it != particles->end(); ++it)
256  {
257  if( !(it->status()==3 || (( partonMode_=="Pythia8" ) && (it->status()==23)))) continue;
258  if( !CandMCTagUtils::isParton( *it ) ) continue; // skip particle if not a parton
259  physicsPartons->push_back( reco::GenParticleRef( particles, it - particles->begin() ) );
260  }
261  }
262 
263  iEvent.put( bHadrons, "bHadrons" );
264  iEvent.put( cHadrons, "cHadrons" );
265  iEvent.put( partons, "algorithmicPartons" );
266  iEvent.put( physicsPartons, "physicsPartons" );
267  iEvent.put( leptons, "leptons" );
268 }
int i
Definition: DBlmapReader.cc:9
bool isMuon(const Candidate &part)
Definition: pdgIdUtils.h:11
ProductID id() const
Definition: HandleBase.cc:15
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:462
bool hasCharm(const reco::Candidate &c)
Definition: CandMCTag.cc:37
Herwig++ parton selector derived from the base parton selector.
const edm::EDGetTokenT< reco::GenParticleCollection > particlesToken_
Herwig6 parton selector derived from the base parton selector.
std::string moduleName(Provenance const &provenance)
Definition: Provenance.cc:27
Sherpa parton selector derived from the base parton selector.
bool isElectron(const Candidate &part)
Definition: pdgIdUtils.h:7
OrphanHandle< PROD > put(std::auto_ptr< PROD > product)
Put a new product.
Definition: Event.h:121
bool isValid() const
Definition: HandleBase.h:75
boost::shared_ptr< BasePartonSelector > PartonSelectorPtr
Pythia6 parton selector derived from the base parton selector.
Pythia8 parton selector derived from the base parton selector.
bool isParton(const reco::Candidate &c)
Definition: CandMCTag.cc:48
Provenance getProvenance(BranchID const &theID) const
Definition: Event.cc:80
bool hasBottom(const reco::Candidate &c)
Definition: CandMCTag.cc:26
const edm::EDGetTokenT< GenEventInfoProduct > srcToken_
bool isTau(const Candidate &part)
Definition: pdgIdUtils.h:15

Member Data Documentation

const edm::EDGetTokenT<reco::GenParticleCollection> HadronAndPartonSelector::particlesToken_
private

Definition at line 91 of file HadronAndPartonSelector.cc.

Referenced by produce().

std::string HadronAndPartonSelector::partonMode_
private

Definition at line 93 of file HadronAndPartonSelector.cc.

Referenced by produce().

PartonSelectorPtr HadronAndPartonSelector::partonSelector_
private

Definition at line 95 of file HadronAndPartonSelector.cc.

Referenced by HadronAndPartonSelector(), and produce().

bool HadronAndPartonSelector::partonSelectorSet_
private

Definition at line 94 of file HadronAndPartonSelector.cc.

Referenced by HadronAndPartonSelector(), and produce().

const edm::EDGetTokenT<GenEventInfoProduct> HadronAndPartonSelector::srcToken_
private

Definition at line 90 of file HadronAndPartonSelector.cc.

Referenced by produce().