CMS 3D CMS Logo

List of all members | Classes | Public Member Functions | Private Member Functions | Private Attributes
JetPartonMatcher Class Reference
Inheritance diagram for JetPartonMatcher:
edm::global::EDProducer<> edm::global::EDProducerBase edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Classes

struct  WorkingVariables
 

Public Member Functions

 JetPartonMatcher (const edm::ParameterSet &)
 
 ~JetPartonMatcher () override
 
- 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< ConsumesInfoconsumesInfo () 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
 
EDConsumerBaseoperator= (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

int fillAlgoritDefinition (const Jet &, WorkingVariables &) const
 
int fillPhysicsDefinition (const Jet &, WorkingVariables &) const
 
void produce (edm::StreamID, edm::Event &, const edm::EventSetup &) const override
 

Private Attributes

double coneSizeToAssociate
 
bool doPriority
 
edm::EDGetTokenT< edm::View< reco::Jet > > m_jetsSrcToken
 
edm::EDGetTokenT< GenParticleRefVectorm_ParticleSrcToken
 
bool physDefinition
 
vector< int > priorityList
 

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 94 of file JetPartonMatcher.cc.

Constructor & Destructor Documentation

JetPartonMatcher::JetPartonMatcher ( const edm::ParameterSet iConfig)

Definition at line 125 of file JetPartonMatcher.cc.

References AK4CaloJetsMCFlavour_cff::coneSizeToAssociate, edm::ParameterSet::exists(), and edm::ParameterSet::getParameter().

126 {
127  produces<JetMatchedPartonsCollection>();
128  m_jetsSrcToken = consumes<edm::View <reco::Jet> >(iConfig.getParameter<edm::InputTag>("jets"));
129  m_ParticleSrcToken = consumes<GenParticleRefVector>(iConfig.getParameter<edm::InputTag>("partons"));
130  coneSizeToAssociate = iConfig.getParameter<double>("coneSizeToAssociate");
131  if ( iConfig.exists("doPriority") ) {
132  doPriority = iConfig.getParameter<bool>("doPriority");
133  priorityList = iConfig.getParameter<vector<int> >("priorityList");
134  } else {
135  doPriority = false;
136  priorityList.clear();
137  }
138 }
T getParameter(std::string const &) const
bool exists(std::string const &parameterName) const
checks if a parameter exists
edm::EDGetTokenT< GenParticleRefVector > m_ParticleSrcToken
vector< int > priorityList
edm::EDGetTokenT< edm::View< reco::Jet > > m_jetsSrcToken
JetPartonMatcher::~JetPartonMatcher ( )
override

Definition at line 142 of file JetPartonMatcher.cc.

143 {
144 }

Member Function Documentation

int JetPartonMatcher::fillAlgoritDefinition ( const Jet theJet,
WorkingVariables wv 
) const
private

Definition at line 215 of file JetPartonMatcher.cc.

References funct::abs(), AK4CaloJetsMCFlavour_cff::coneSizeToAssociate, HiRegitMuonDetachedTripletStep_cff::DeltaR, spr::find(), GenHFHadronMatcher_cff::flavour, funct::m, MuonErrorMatrixAnalyzer_cfi::maxPt, reco::LeafCandidate::p4(), JetPartonMatcher::WorkingVariables::particles, reco::LeafCandidate::pdgId(), reco::LeafCandidate::pt(), reco::LeafCandidate::status(), JetPartonMatcher::WorkingVariables::theHardest, JetPartonMatcher::WorkingVariables::theHeaviest, and JetPartonMatcher::WorkingVariables::theNearest2.

215  {
216 
217  int tempParticle = -1;
218  int tempPartonHighestPt = -1;
219  int tempNearest = -1;
220  float maxPt = 0;
221  float minDr = 1000;
222  bool foundPriority = false;
223 
224  // Loop over the particles in question until we find a priority
225  // "hit", or if we find none in the priority list (or we don't want
226  // to consider priority), then loop through all particles and fill
227  // standard definition.
228 
229  //
230  // Matching:
231  //
232  // 1) First try to match by hand. The "priority list" is given
233  // by the user. The algorithm finds any particles in that
234  // "priority list" that are within the specified cone size.
235  // If it finds one, it counts the object as associating to that
236  // particle.
237  // NOTE! The objects in the priority list are given in order
238  // of priority. So if the user specifies:
239  // 6, 24, 21,
240  // then it will first look for top quarks, then W bosons, then gluons.
241  // 2) If no priority items are found, do the default "standard"
242  // matching.
243  for( size_t m = 0; m < wv.particles->size() && !foundPriority; ++ m ) {
244  // const Candidate & aParton = *(wv.particles->at(m).get());
245  const GenParticle & aParton = *(wv.particles->at(m).get());
246  int flavour = abs(aParton.pdgId());
247 
248  // "Priority" behavoir:
249  // Associate to the first particle found in the priority list, regardless
250  // of delta R.
251  if ( doPriority ) {
252  vector<int>::const_iterator ipriority = find( priorityList.begin(), priorityList.end(), flavour );
253  // Check to see if this particle is in our priority list
254  if ( ipriority != priorityList.end() ) {
255  // This particle is on our priority list. If it matches,
256  // we break, since we take in order of priority, not deltaR
257  double dist = DeltaR( theJet.p4(), aParton.p4() );
258  if( dist <= coneSizeToAssociate ) {
259  tempParticle = m;
260  foundPriority = true;
261  }
262  }
263  }
264  // Here we do not want to do priority matching. Ensure the "foundPriority" swtich
265  // is turned off.
266  else {
267  foundPriority = false;
268  }
269 
270  // Default behavior:
271  // Look for partons before the color string to associate.
272  // Do this if we don't want to do priority matching, or if
273  // we didn't find a priority object.
274 
275  if( !foundPriority && aParton.status() != 3 ) {
276  double dist = DeltaR( theJet.p4(), aParton.p4() );
277  if( dist <= coneSizeToAssociate ) {
278  if( dist < minDr ) {
279  minDr = dist;
280  tempNearest = m;
281  }
282  if( tempParticle == -1 && ( flavour == 4 ) ) tempParticle = m;
283  if( flavour == 5 ) tempParticle = m;
284  if( aParton.pt() > maxPt ) {
285  maxPt = aParton.pt();
286  tempPartonHighestPt = m;
287  }
288  }
289  }
290  }
291 
292  if ( foundPriority ) {
293  wv.theHeaviest = tempParticle; // The priority-matched particle
294  wv.theHardest = -1; // set the rest to -1
295  wv.theNearest2 = -1; // " "
296  } else {
297  wv.theHeaviest = tempParticle;
298  wv.theHardest = tempPartonHighestPt;
299  wv.theNearest2 = tempNearest;
300  if ( tempParticle == -1 ) tempParticle = tempPartonHighestPt;
301  }
302  return tempParticle;
303 }
virtual double pt() const final
transverse momentum
virtual int status() const final
status word
void find(edm::Handle< EcalRecHitCollection > &hits, DetId thisDet, std::vector< EcalRecHitCollection::const_iterator > &hit, bool debug=false)
Definition: FindCaloHit.cc:20
virtual int pdgId() const final
PDG identifier.
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
vector< int > priorityList
virtual const LorentzVector & p4() const final
four-momentum Lorentz vector
Definition: LeafCandidate.h:99
int JetPartonMatcher::fillPhysicsDefinition ( const Jet theJet,
WorkingVariables wv 
) const
private

Definition at line 327 of file JetPartonMatcher.cc.

References funct::abs(), AK4CaloJetsMCFlavour_cff::coneSizeToAssociate, DEFINE_FWK_MODULE, HiRegitMuonDetachedTripletStep_cff::DeltaR, spr::find(), GenHFHadronMatcher_cff::flavour, funct::m, reco::CompositeRefCandidateT< D >::numberOfDaughters(), reco::LeafCandidate::p4(), JetPartonMatcher::WorkingVariables::particles, cosmictrackingParticleSelector_cfi::pdgId, reco::LeafCandidate::pdgId(), reco::LeafCandidate::status(), JetPartonMatcher::WorkingVariables::theHardest, JetPartonMatcher::WorkingVariables::theHeaviest, JetPartonMatcher::WorkingVariables::theNearest2, and JetPartonMatcher::WorkingVariables::theNearest3.

327  {
328 
329  float TheBiggerConeSize = 0.7; // In HepMC it's 0.3 --> it's a mistake: value has to be 0.7
330  int tempParticle = -1;
331  int nInTheCone = 0;
332  int tempNearest = -1;
333  float minDr = 1000;
334  bool foundPriority = false;
335 
336  vector<const reco::Candidate *> theContaminations;
337  theContaminations.clear();
338 
339  // Loop over the particles in question until we find a priority
340  // "hit", or if we find none in the priority list (or we don't want
341  // to consider priority), then loop through all particles and fill
342  // standard definition.
343 
344  //
345  // Matching:
346  //
347  // 1) First try to match by hand. The "priority list" is given
348  // by the user. The algorithm finds any particles in that
349  // "priority list" that are within the specified cone size.
350  // If it finds one, it counts the object as associating to that
351  // particle.
352  // NOTE! The objects in the priority list are given in order
353  // of priority. So if the user specifies:
354  // 6, 24, 21,
355  // then it will first look for top quarks, then W bosons, then gluons.
356  // 2) If no priority items are found, do the default "standard"
357  // matching.
358  for( size_t m = 0; m < wv.particles->size() && !foundPriority; ++ m ) {
359 
360  // const Candidate & aParticle = *(wv.particles->at(m).get());
361  const GenParticle & aParticle = *(wv.particles->at(m).get());
362  int flavour = abs(aParticle.pdgId());
363 
364  // "Priority" behavoir:
365  // Associate to the first particle found in the priority list, regardless
366  // of delta R.
367  if ( doPriority ) {
368  vector<int>::const_iterator ipriority = find( priorityList.begin(), priorityList.end(), flavour );
369  // Check to see if this particle is in our priority list
370  if ( ipriority != priorityList.end() ) {
371  // This particle is on our priority list. If it matches,
372  // we break, since we take in order of priority, not deltaR
373  double dist = DeltaR( theJet.p4(), aParticle.p4() );
374  if( dist <= coneSizeToAssociate ) {
375  tempParticle = m;
376  foundPriority = true;
377  }
378  }
379  }
380  // Here we do not want to do priority matching. Ensure the "foundPriority" swtich
381  // is turned off.
382  else {
383  foundPriority = false;
384  }
385 
386  // Default behavior:
387  if ( !foundPriority ) {
388 
389  // skipping all particle but udscbg (is this correct/enough?!?!)
390  bool isAParton = false;
391  if(flavour == 1 ||
392  flavour == 2 ||
393  flavour == 3 ||
394  flavour == 4 ||
395  flavour == 5 ||
396  flavour == 21 ) isAParton = true;
397  if(!isAParton) continue;
398  double dist = DeltaR( theJet.p4(), aParticle.p4() );
399  if( aParticle.status() == 3 && dist < minDr ) {
400  minDr = dist;
401  tempNearest = m;
402  }
403  if( aParticle.status() == 3 && dist <= coneSizeToAssociate ) {
404  //cout << "particle in small cone=" << aParticle.pdgId() << endl;
405  tempParticle = m;
406  nInTheCone++;
407  }
408  // Look for heavy partons in TheBiggerConeSize now
409  if( aParticle.numberOfDaughters() > 0 && aParticle.status() != 3 ) {
410  if( flavour == 1 ||
411  flavour == 2 ||
412  flavour == 3 ||
413  flavour == 21 ) continue;
414  if( dist < TheBiggerConeSize ) theContaminations.push_back( &aParticle );
415  }
416  }
417  }
418 
419 
420  // Here's the default behavior for assignment if there is no priority.
421  if ( !foundPriority ) {
422  wv.theNearest3 = tempNearest;
423 
424  if(nInTheCone != 1) return -1; // rejected --> only one initialParton requested
425  if(theContaminations.empty() ) return tempParticle; //no contamination
426  int initialPartonFlavour = abs( (wv.particles->at(tempParticle).get()) ->pdgId() );
427 
428  vector<const Candidate *>::const_iterator itCont = theContaminations.begin();
429  for( ; itCont != theContaminations.end(); itCont++ ) {
430  int contaminatingFlavour = abs( (*itCont)->pdgId() );
431  if( (*itCont)->numberOfMothers()>0 && (*itCont)->mother(0) == wv.particles->at(tempParticle).get() ) continue; // mother is the initialParton --> OK
432  if( initialPartonFlavour == 4 ) {
433  if( contaminatingFlavour == 4 ) continue; // keep association --> the initialParton is a c --> the contaminated parton is a c
434  tempParticle = -1; // all the other cases reject!
435  return tempParticle;
436  }
437  }
438  }
439  // If there is priority, then just set the heaviest to priority, the rest are -1.
440  else {
441  wv.theHeaviest = tempParticle; // Set the heaviest to tempParticle
442  wv.theNearest2 = -1; // Set the rest to -1
443  wv.theNearest3 = -1; // " "
444  wv.theHardest = -1; // " "
445  }
446 
447  return tempParticle;
448 }
virtual int status() const final
status word
void find(edm::Handle< EcalRecHitCollection > &hits, DetId thisDet, std::vector< EcalRecHitCollection::const_iterator > &hit, bool debug=false)
Definition: FindCaloHit.cc:20
virtual size_t numberOfDaughters() const
number of daughters
virtual int pdgId() const final
PDG identifier.
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
vector< int > priorityList
virtual const LorentzVector & p4() const final
four-momentum Lorentz vector
Definition: LeafCandidate.h:99
void JetPartonMatcher::produce ( edm::StreamID  ,
edm::Event iEvent,
const edm::EventSetup iEs 
) const
overrideprivate

Definition at line 148 of file JetPartonMatcher.cc.

References reco::LeafCandidate::eta(), edm::Event::getByToken(), funct::m, eostools::move(), JetPartonMatcher::WorkingVariables::particles, reco::LeafCandidate::pdgId(), reco::LeafCandidate::phi(), reco::LeafCandidate::pt(), edm::Event::put(), reco::LeafCandidate::status(), JetPartonMatcher::WorkingVariables::theHeaviest, JetPartonMatcher::WorkingVariables::theNearest2, and JetPartonMatcher::WorkingVariables::theNearest3.

149 {
150  WorkingVariables wv;
152  iEvent.getByToken(m_jetsSrcToken, jets_h );
153  iEvent.getByToken(m_ParticleSrcToken, wv.particles );
154 
155  edm::LogVerbatim("JetPartonMatcher") << "=== Partons size:" << wv.particles->size();
156 
157  for( size_t m = 0; m < wv.particles->size(); ++ m ) {
158  const GenParticle & aParton = *(wv.particles->at(m).get());
159  edm::LogVerbatim("JetPartonMatcher") << aParton.status() << " " <<
160  aParton.pdgId() << " " <<
161  aParton.pt() << " " <<
162  aParton.eta() << " " <<
163  aParton.phi() << endl;
164  }
165 
166  auto jetMatchedPartons = std::make_unique<JetMatchedPartonsCollection>(reco::JetRefBaseProd(jets_h));
167 
168  for (size_t j = 0; j < jets_h->size(); j++) {
169 
170  const int theMappedPartonAlg = fillAlgoritDefinition( (*jets_h)[j], wv );
171  const int theMappedPartonPhy = fillPhysicsDefinition( (*jets_h)[j], wv );
172 
173  GenParticleRef pHV;
174  GenParticleRef pN2;
175  GenParticleRef pN3;
176  GenParticleRef pPH;
177  GenParticleRef pAL;
178 
179  if(wv.theHeaviest>=0) pHV = wv.particles->at( wv.theHeaviest );
180  if(wv.theNearest2>=0) pN2 = wv.particles->at( wv.theNearest2 );
181  if(wv.theNearest3>=0) pN3 = wv.particles->at( wv.theNearest3 );
182  if(theMappedPartonPhy>=0) pPH = wv.particles->at( theMappedPartonPhy );
183  if(theMappedPartonAlg>=0) pAL = wv.particles->at( theMappedPartonAlg );
184 
185  (*jetMatchedPartons)[jets_h->refAt(j)]=MatchedPartons(pHV,pN2,pN3,pPH,pAL);
186  }
187 
188  iEvent.put(std::move(jetMatchedPartons));
189 
190 }
virtual double pt() const final
transverse momentum
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:122
virtual double eta() const final
momentum pseudorapidity
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:460
int fillAlgoritDefinition(const Jet &, WorkingVariables &) const
virtual int status() const final
status word
virtual double phi() const final
momentum azimuthal angle
edm::RefToBaseProd< reco::Jet > JetRefBaseProd
Definition: JetCollection.h:14
edm::EDGetTokenT< GenParticleRefVector > m_ParticleSrcToken
virtual int pdgId() const final
PDG identifier.
int fillPhysicsDefinition(const Jet &, WorkingVariables &) const
edm::EDGetTokenT< edm::View< reco::Jet > > m_jetsSrcToken
def move(src, dest)
Definition: eostools.py:510

Member Data Documentation

double JetPartonMatcher::coneSizeToAssociate
private

Definition at line 116 of file JetPartonMatcher.cc.

bool JetPartonMatcher::doPriority
private

Definition at line 118 of file JetPartonMatcher.cc.

edm::EDGetTokenT<edm::View <reco::Jet> > JetPartonMatcher::m_jetsSrcToken
private

Definition at line 114 of file JetPartonMatcher.cc.

edm::EDGetTokenT<GenParticleRefVector> JetPartonMatcher::m_ParticleSrcToken
private

Definition at line 115 of file JetPartonMatcher.cc.

bool JetPartonMatcher::physDefinition
private

Definition at line 117 of file JetPartonMatcher.cc.

vector<int> JetPartonMatcher::priorityList
private

Definition at line 119 of file JetPartonMatcher.cc.