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 ()
 
- 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
 
virtual 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 ( )

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(), funct::m, MuonErrorMatrixAnalyzer_cfi::maxPt, reco::Candidate::p4(), reco::LeafCandidate::p4(), JetPartonMatcher::WorkingVariables::particles, reco::Candidate::pdgId(), reco::Candidate::pt(), reco::Candidate::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 
246  // "Priority" behavoir:
247  // Associate to the first particle found in the priority list, regardless
248  // of delta R.
249  if ( doPriority ) {
250  int ipdgId = aParton.pdgId();
251  vector<int>::const_iterator ipriority = find( priorityList.begin(), priorityList.end(), abs(ipdgId) );
252  // Check to see if this particle is in our priority list
253  if ( ipriority != priorityList.end() ) {
254  // This particle is on our priority list. If it matches,
255  // we break, since we take in order of priority, not deltaR
256  double dist = DeltaR( theJet.p4(), aParton.p4() );
257  if( dist <= coneSizeToAssociate ) {
258  tempParticle = m;
259  foundPriority = true;
260  }
261  }
262  }
263  // Here we do not want to do priority matching. Ensure the "foundPriority" swtich
264  // is turned off.
265  else {
266  foundPriority = false;
267  }
268 
269  // Default behavior:
270  // Look for partons before the color string to associate.
271  // Do this if we don't want to do priority matching, or if
272  // we didn't find a priority object.
273 
274  if( !foundPriority && aParton.status() != 3 ) {
275  double dist = DeltaR( theJet.p4(), aParton.p4() );
276  if( dist <= coneSizeToAssociate ) {
277  if( dist < minDr ) {
278  minDr = dist;
279  tempNearest = m;
280  }
281  if( tempParticle == -1 && ( abs( aParton.pdgId() ) == 4 ) ) tempParticle = m;
282  if( abs( aParton.pdgId() ) == 5 ) tempParticle = m;
283  if( aParton.pt() > maxPt ) {
284  maxPt = aParton.pt();
285  tempPartonHighestPt = m;
286  }
287  }
288  }
289  }
290 
291  if ( foundPriority ) {
292  wv.theHeaviest = tempParticle; // The priority-matched particle
293  wv.theHardest = -1; // set the rest to -1
294  wv.theNearest2 = -1; // " "
295  } else {
296  wv.theHeaviest = tempParticle;
297  wv.theHardest = tempPartonHighestPt;
298  wv.theNearest2 = tempNearest;
299  if ( tempParticle == -1 ) tempParticle = tempPartonHighestPt;
300  }
301  return tempParticle;
302 }
void find(edm::Handle< EcalRecHitCollection > &hits, DetId thisDet, std::vector< EcalRecHitCollection::const_iterator > &hit, bool debug=false)
Definition: FindCaloHit.cc:20
virtual int status() const =0
status word
virtual const LorentzVector & p4() const =0
four-momentum Lorentz vector
virtual int pdgId() const =0
PDG identifier.
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
vector< int > priorityList
virtual double pt() const =0
transverse momentum
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 326 of file JetPartonMatcher.cc.

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

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

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.