Inheritance diagram for JetPartonMatcher:

Public Member Functions
	JetPartonMatcher (const edm::ParameterSet &)

	~JetPartonMatcher ()

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
	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

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Member Functions
int	fillAlgoritDefinition (const Jet &)

int	fillPhysicsDefinition (const Jet &)

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

Private Attributes
double	coneSizeToAssociate

bool	doPriority

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

edm::EDGetTokenT < GenParticleRefVector >	m_ParticleSrcToken

Handle< GenParticleRefVector >	particles

bool	physDefinition

vector< int >	priorityList

int	theHardest

int	theHeaviest

int	theNearest2

int	theNearest3

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

Constructor & Destructor Documentation

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

Definition at line 123 of file JetPartonMatcher.cc.

References coneSizeToAssociate, doPriority, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), m_jetsSrcToken, m_ParticleSrcToken, and priorityList.

                                                                    :
   theHeaviest(0),
   theNearest2(0),
   theNearest3(0),
   theHardest(0)
 {
     produces<JetMatchedPartonsCollection>();
     m_jetsSrcToken           = consumes<edm::View <reco::Jet> >(iConfig.getParameter<edm::InputTag>("jets"));
     m_ParticleSrcToken       = consumes<GenParticleRefVector>(iConfig.getParameter<edm::InputTag>("partons"));
     coneSizeToAssociate = iConfig.getParameter<double>("coneSizeToAssociate");
     if ( iConfig.exists("doPriority") ) {
       doPriority = iConfig.getParameter<bool>("doPriority");
       priorityList = iConfig.getParameter<vector<int> >("priorityList");
     } else {
       doPriority = false;
       priorityList.clear();
     }
 }

JetPartonMatcher::~JetPartonMatcher ( )

Definition at line 144 of file JetPartonMatcher.cc.

145 {

146 }

Member Function Documentation

int JetPartonMatcher::fillAlgoritDefinition ( const Jet & theJet )

private

Definition at line 216 of file JetPartonMatcher.cc.

References funct::abs(), coneSizeToAssociate, doPriority, spr::find(), m, reco::Candidate::p4(), reco::LeafCandidate::p4(), particles, reco::Candidate::pdgId(), priorityList, reco::Candidate::pt(), reco::Candidate::status(), theHardest, theHeaviest, and theNearest2.

Referenced by produce().

                                                                {
 
   int tempParticle = -1;
   int tempPartonHighestPt = -1;
   int tempNearest = -1;
   float maxPt = 0;
   float minDr = 1000;
   bool foundPriority = false;
 
   // Loop over the particles in question until we find a priority
   // "hit", or if we find none in the priority list (or we don't want
   // to consider priority), then loop through all particles and fill
   // standard definition.
 
   //
   // Matching:
   //
   // 1) First try to match by hand. The "priority list" is given
   //    by the user. The algorithm finds any particles in that
   //    "priority list" that are within the specified cone size.
   //    If it finds one, it counts the object as associating to that
   //    particle.
   //    NOTE! The objects in the priority list are given in order
   //    of priority. So if the user specifies:
   //    6, 24, 21,
   //    then it will first look for top quarks, then W bosons, then gluons.
   // 2) If no priority items are found, do the default "standard"
   //    matching.
   for( size_t m = 0; m != particles->size() && !foundPriority; ++ m ) {
     const Candidate & aParton = *(particles->at(m).get());
 
     // "Priority" behavoir:
     // Associate to the first particle found in the priority list, regardless
     // of delta R.
     if ( doPriority ) {
       int ipdgId = aParton.pdgId();
       vector<int>::const_iterator ipriority = find( priorityList.begin(), priorityList.end(), abs(ipdgId) );
       // Check to see if this particle is in our priority list
       if ( ipriority != priorityList.end() ) {
     // This particle is on our priority list. If it matches,
     // we break, since we take in order of priority, not deltaR
     double dist = DeltaR( theJet.p4(), aParton.p4() );
     if( dist <= coneSizeToAssociate ) {
       tempParticle = m;
       foundPriority = true;
     }
       }
     }
     // Here we do not want to do priority matching. Ensure the "foundPriority" swtich
     // is turned off.
     else {
       foundPriority = false;
     }
 
     // Default behavior:
     // Look for partons before the color string to associate.
     // Do this if we don't want to do priority matching, or if
     // we didn't find a priority object.
 
     if( !foundPriority && aParton.status() != 3 ) {
       double dist = DeltaR( theJet.p4(), aParton.p4() );
       if( dist <= coneSizeToAssociate ) {
     if( dist < minDr ) {
       minDr = dist;
       tempNearest = m;
     }
     if( tempParticle == -1 && ( abs( aParton.pdgId() ) == 4 )  ) tempParticle = m;
     if(                         abs( aParton.pdgId() ) == 5    ) tempParticle = m;
     if( aParton.pt() > maxPt ) {
       maxPt = aParton.pt();
       tempPartonHighestPt = m;
     }
       }
     }
   }
 
   if ( foundPriority ) {
     theHeaviest = tempParticle; // The priority-matched particle
     theHardest  = -1;  //  set the rest to -1
     theNearest2 = -1;  // "                  "
   } else {
     theHeaviest = tempParticle;
     theHardest  = tempPartonHighestPt;
     theNearest2 = tempNearest;
     if ( tempParticle == -1 ) tempParticle = tempPartonHighestPt;
   }
   return tempParticle;
 }

int JetPartonMatcher::fillPhysicsDefinition ( const Jet & theJet )

private

Definition at line 327 of file JetPartonMatcher.cc.

References funct::abs(), coneSizeToAssociate, doPriority, spr::find(), reco::flavour(), m, reco::Candidate::numberOfDaughters(), reco::Candidate::p4(), reco::LeafCandidate::p4(), particles, benchmark_cfg::pdgId, reco::Candidate::pdgId(), priorityList, reco::Candidate::status(), theHardest, theHeaviest, theNearest2, and theNearest3.

Referenced by produce().

                                                                {
 
   float TheBiggerConeSize = 0.7; // In HepMC it's 0.3 --> it's a mistake: value has to be 0.7
   int tempParticle = -1;
   int nInTheCone = 0;
   int tempNearest = -1;
   float minDr = 1000;
   bool foundPriority = false;
 
   vector<const reco::Candidate *> theContaminations;
   theContaminations.clear();
 
   // Loop over the particles in question until we find a priority
   // "hit", or if we find none in the priority list (or we don't want
   // to consider priority), then loop through all particles and fill
   // standard definition.
 
   //
   // Matching:
   //
   // 1) First try to match by hand. The "priority list" is given
   //    by the user. The algorithm finds any particles in that
   //    "priority list" that are within the specified cone size.
   //    If it finds one, it counts the object as associating to that
   //    particle.
   //    NOTE! The objects in the priority list are given in order
   //    of priority. So if the user specifies:
   //    6, 24, 21,
   //    then it will first look for top quarks, then W bosons, then gluons.
   // 2) If no priority items are found, do the default "standard"
   //    matching.
   for( size_t m = 0; m != particles->size() && !foundPriority; ++ m ) {
 
     const Candidate & aParticle = *(particles->at(m).get());
 
     // "Priority" behavoir:
     // Associate to the first particle found in the priority list, regardless
     // of delta R.
     if ( doPriority ) {
       int ipdgId = aParticle.pdgId();
       vector<int>::const_iterator ipriority = find( priorityList.begin(), priorityList.end(), abs(ipdgId) );
       // Check to see if this particle is in our priority list
       if ( ipriority != priorityList.end() ) {
     // This particle is on our priority list. If it matches,
     // we break, since we take in order of priority, not deltaR
     double dist = DeltaR( theJet.p4(), aParticle.p4() );
     if( dist <= coneSizeToAssociate ) {
       tempParticle = m;
       foundPriority = true;
     }
       }
     }
     // Here we do not want to do priority matching. Ensure the "foundPriority" swtich
     // is turned off.
     else {
       foundPriority = false;
     }
 
     // Default behavior:
     if ( !foundPriority ) {
 
       // skipping all particle but udscbg (is this correct/enough?!?!)
       bool isAParton = false;
       int flavour = abs(aParticle.pdgId());
       if(flavour == 1 ||
      flavour == 2 ||
      flavour == 3 ||
      flavour == 4 ||
      flavour == 5 ||
      flavour == 21 ) isAParton = true;
       if(!isAParton) continue;
       double dist = DeltaR( theJet.p4(), aParticle.p4() );
       if( aParticle.status() == 3 && dist < minDr ) {
     minDr = dist;
     tempNearest = m;
       }
       if( aParticle.status() == 3 && dist <= coneSizeToAssociate ) {
     //cout << "particle in small cone=" << aParticle.pdgId() << endl;
     tempParticle = m;
     nInTheCone++;
       }
       // Look for heavy partons in TheBiggerConeSize now
       if( aParticle.numberOfDaughters() > 0 && aParticle.status() != 3 ) {
     if( flavour ==  1 ||
         flavour ==  2 ||
         flavour ==  3 ||
         flavour == 21 ) continue;
     if( dist < TheBiggerConeSize ) theContaminations.push_back( &aParticle );
       }
     }
   }
 
 
   // Here's the default behavior for assignment if there is no priority.
   if ( !foundPriority ) {
     theNearest3 = tempNearest;
 
     if(nInTheCone != 1) return -1; // rejected --> only one initialParton requested
     if(theContaminations.size() == 0 ) return tempParticle; //no contamination
     int initialPartonFlavour = abs( (particles->at(tempParticle).get()) ->pdgId() );
 
     vector<const Candidate *>::const_iterator itCont = theContaminations.begin();
     for( ; itCont != theContaminations.end(); itCont++ ) {
       int contaminatingFlavour = abs( (*itCont)->pdgId() );
       if( (*itCont)->numberOfMothers()>0 && (*itCont)->mother(0) == particles->at(tempParticle).get() ) continue; // mother is the initialParton --> OK
       if( initialPartonFlavour == 4 ) {
     if( contaminatingFlavour == 4 ) continue; // keep association --> the initialParton is a c --> the contaminated parton is a c
     tempParticle = -1; // all the other cases reject!
     return tempParticle;
       }
     }
   }
   // If there is priority, then just set the heaviest to priority, the rest are -1.
   else {
     theHeaviest = tempParticle; // Set the heaviest to tempParticle
     theNearest2 = -1; //  Set the rest to -1
     theNearest3 = -1; // "                  "
     theHardest = -1;  // "                  "
   }
 
   return tempParticle;
 }

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

overrideprivatevirtual

Implements edm::EDProducer.

Definition at line 150 of file JetPartonMatcher.cc.

References reco::LeafCandidate::eta(), fillAlgoritDefinition(), fillPhysicsDefinition(), edm::Event::getByToken(), j, m, m_jetsSrcToken, m_ParticleSrcToken, particles, reco::LeafCandidate::pdgId(), reco::LeafCandidate::phi(), reco::LeafCandidate::pt(), edm::Event::put(), reco::LeafCandidate::status(), theHeaviest, theNearest2, and theNearest3.

 {
   edm::Handle <edm::View <reco::Jet> > jets_h;
   iEvent.getByToken(m_jetsSrcToken,     jets_h    );
   iEvent.getByToken(m_ParticleSrcToken, particles );
 
   edm::LogVerbatim("JetPartonMatcher") << "=== Partons size:" << particles->size();
 
   for( size_t m = 0; m != particles->size(); ++ m ) {
     const GenParticle & aParton = *(particles->at(m).get());
     edm::LogVerbatim("JetPartonMatcher") <<  aParton.status() << " " <<
                                              aParton.pdgId()  << " " <<
                                              aParton.pt()     << " " <<
                                              aParton.eta()    << " " <<
                                              aParton.phi()    << endl;
   }
 
   auto_ptr<reco::JetMatchedPartonsCollection> jetMatchedPartons( new JetMatchedPartonsCollection(reco::JetRefBaseProd(jets_h)));
 
   for (size_t j = 0; j < jets_h->size(); j++) {
 
     const int theMappedPartonAlg = fillAlgoritDefinition( (*jets_h)[j] );
     const int theMappedPartonPhy = fillPhysicsDefinition( (*jets_h)[j] );
 
     GenParticleRef pHV;
     GenParticleRef pN2;
     GenParticleRef pN3;
     GenParticleRef pPH;
     GenParticleRef pAL;
 
     if(theHeaviest>=0)        pHV = particles->at( theHeaviest        );
     if(theNearest2>=0)        pN2 = particles->at( theNearest2        );
     if(theNearest3>=0)        pN3 = particles->at( theNearest3        );
     if(theMappedPartonPhy>=0) pPH = particles->at( theMappedPartonPhy );
     if(theMappedPartonAlg>=0) pAL = particles->at( theMappedPartonAlg );
 
     (*jetMatchedPartons)[jets_h->refAt(j)]=MatchedPartons(pHV,pN2,pN3,pPH,pAL);
   }
 
   iEvent.put(  jetMatchedPartons );
 
 }