noPileUpMEtAuxFunctions.cc File Reference

#include "RecoMET/METPUSubtraction/interface/noPileUpMEtAuxFunctions.h"
#include "DataFormats/Math/interface/deltaR.h"
#include <math.h>

Go to the source code of this file.

Functions
int	isVertexAssociated (const reco::PFCandidate &pfCandidate, const PFCandToVertexAssMap &pfCandToVertexAssociations, const reco::VertexCollection &vertices, double dZ)
int	isVertexAssociated_fast (const reco::PFCandidateRef &pfCandidate, const reversedPFCandidateToVertexAssociationMap &pfCandToVertexAssociations, const reco::VertexCollection &vertices, double dZ, int &numWarnings, int maxWarnings)
reversedPFCandidateToVertexAssociationMap	reversePFCandToVertexAssociation (const PFCandToVertexAssMap &pfCandToVertexAssociations)

Variables
const double	dR2Min =0.01*0.01
const int	minPFCandToVertexAssocQuality = noPuUtils::kChHSAssoc

Function Documentation

int isVertexAssociated	(	const reco::PFCandidate &	pfCandidate,
		const PFCandToVertexAssMap &	pfCandToVertexAssociations,
		const reco::VertexCollection &	vertices,
		double	dZ
	)

Definition at line 11 of file noPileUpMEtAuxFunctions.cc.

References funct::abs(), edm::AssociationMap< Tag >::begin(), reco::LeafCandidate::charge(), reco::deltaR2(), dR2Min, edm::AssociationMap< Tag >::end(), noPuUtils::kChHSAssoc, noPuUtils::kChNoAssoc, noPuUtils::kChPUAssoc, noPuUtils::kNeutral, minPFCandToVertexAssocQuality, and reco::LeafCandidate::p4().

{
  int vtxAssociationType = noPuUtils::kNeutral;

  if ( pfCandidate.charge() != 0 ) {
    vtxAssociationType = noPuUtils::kChNoAssoc;
    for ( PFCandToVertexAssMap::const_iterator pfCandToVertexAssociation = pfCandToVertexAssociations.begin();
          pfCandToVertexAssociation != pfCandToVertexAssociations.end(); ++pfCandToVertexAssociation ) {
      typedef std::vector<std::pair<reco::PFCandidateRef, int> > PFCandidateQualityPairVector;
      const PFCandidateQualityPairVector& pfCandidates_vertex = pfCandToVertexAssociation->val;
      for ( PFCandidateQualityPairVector::const_iterator pfCandidate_vertex = pfCandidates_vertex.begin();
            pfCandidate_vertex != pfCandidates_vertex.end(); ++pfCandidate_vertex ) {
    int pfCandToVertexAssocQuality = pfCandidate_vertex->second;
    if ( pfCandToVertexAssocQuality >= minPFCandToVertexAssocQuality && deltaR2(pfCandidate.p4(), pfCandidate_vertex->first->p4()) < dR2Min ) {
          if ( vtxAssociationType < noPuUtils::kChPUAssoc ) vtxAssociationType = noPuUtils::kChPUAssoc;
          for ( reco::VertexCollection::const_iterator vertex = vertices.begin();
                vertex != vertices.end(); ++vertex ) {
            if ( std::abs(pfCandToVertexAssociation->key->position().z() - vertex->position().z()) < dZ ) {
              if ( vtxAssociationType < noPuUtils::kChHSAssoc ) vtxAssociationType = noPuUtils::kChHSAssoc;
            }
          }
        }
      }
    }
  }

  return vtxAssociationType;
}

int isVertexAssociated_fast	(	const reco::PFCandidateRef &	pfCandidate,
		const reversedPFCandidateToVertexAssociationMap &	pfCandToVertexAssociations,
		const reco::VertexCollection &	vertices,
		double	dZ,
		int &	numWarnings,
		int	maxWarnings
	)

Definition at line 60 of file noPileUpMEtAuxFunctions.cc.

References funct::abs(), edm::AssociationMap< Tag >::begin(), reco::deltaR2(), dR2Min, edm::AssociationMap< Tag >::end(), edm::AssociationMap< Tag >::find(), noPuUtils::kChHSAssoc, noPuUtils::kChPUAssoc, edm::Ref< C, T, F >::key(), minPFCandToVertexAssocQuality, and edm::helpers::KeyVal< K, V >::val.

Referenced by NoPileUpPFMEtDataProducer::produce().

{
  int vtxAssociationType = 0; // 0 = neutral particle, 
                              // 1 = charged particle not associated to any vertex
                              // 2 = charged particle associated to pile-up vertex
                              // 3 = charged particle associated to vertex of hard-scatter event

  if ( pfCandidate->charge() != 0 ) {
    vtxAssociationType = 1;
    typedef std::vector<std::pair<reco::VertexRef, int> > VertexQualityPairVector;
    const VertexQualityPairVector* pfCandidate_associated_vertices = nullptr;
    reversedPFCandidateToVertexAssociationMap::const_iterator pfCandToVertexAssociation_iter = pfCandToVertexAssociations.find(pfCandidate);
    if ( pfCandToVertexAssociation_iter != pfCandToVertexAssociations.end() ) {
      pfCandidate_associated_vertices = &pfCandToVertexAssociation_iter->val;
    } else {
      for ( reversedPFCandidateToVertexAssociationMap::const_iterator pfCandToVertexAssociation = pfCandToVertexAssociations.begin();
        pfCandToVertexAssociation != pfCandToVertexAssociations.end(); ++pfCandToVertexAssociation ) {
    if ( deltaR2(pfCandidate->p4(), pfCandToVertexAssociation->key->p4()) < dR2Min ) {
          pfCandidate_associated_vertices = &pfCandToVertexAssociation->val;
          break;
        }
      }
      if ( numWarnings < maxWarnings ) {
        edm::LogWarning ("isVertexAssociated") 
          << " The productIDs of PFCandidate and PFCandToVertexAssociationMap passed as function arguments don't match.\n" 
          << "NOTE: The return value will be unaffected, but the code will run MUCH slower !!";
        ++numWarnings;
      }
    }
    if ( pfCandidate_associated_vertices ) {
      for ( VertexQualityPairVector::const_iterator pfCandidate_associated_vertex = pfCandidate_associated_vertices->begin();
        pfCandidate_associated_vertex != pfCandidate_associated_vertices->end(); ++pfCandidate_associated_vertex ) {    
    if ( pfCandidate_associated_vertex->second >= minPFCandToVertexAssocQuality ) {
      if ( vtxAssociationType <  noPuUtils::kChPUAssoc ) vtxAssociationType =  noPuUtils::kChPUAssoc;
      for ( reco::VertexCollection::const_iterator vertex = vertices.begin();
        vertex != vertices.end(); ++vertex ) {
        if ( std::abs(pfCandidate_associated_vertex->first->position().z() - vertex->position().z()) < dZ ) {
          if ( vtxAssociationType < noPuUtils::kChHSAssoc ) vtxAssociationType = noPuUtils::kChHSAssoc;
        }
      }
    }
      }
    }
  }

  return vtxAssociationType;
}

reversedPFCandidateToVertexAssociationMap reversePFCandToVertexAssociation

(

const PFCandToVertexAssMap &

pfCandToVertexAssociations

)

Definition at line 42 of file noPileUpMEtAuxFunctions.cc.

References edm::AssociationMap< Tag >::begin(), edm::AssociationMap< Tag >::end(), edm::AssociationMap< Tag >::insert(), and edm::Ref< C, T, F >::key().

Referenced by NoPileUpPFMEtDataProducer::produce().

{
  reversedPFCandidateToVertexAssociationMap pfCandToVertexAssociations_reversed;

  for ( PFCandToVertexAssMap::const_iterator pfCandToVertexAssociation = pfCandToVertexAssociations.begin();
    pfCandToVertexAssociation != pfCandToVertexAssociations.end(); ++pfCandToVertexAssociation ) {
    const reco::VertexRef& vertex = pfCandToVertexAssociation->key;
    typedef std::vector<std::pair<reco::PFCandidateRef, int> > PFCandidateQualityPairVector;
    const PFCandidateQualityPairVector& pfCandidates_vertex = pfCandToVertexAssociation->val;
    for ( PFCandidateQualityPairVector::const_iterator pfCandidate_vertex = pfCandidates_vertex.begin();
      pfCandidate_vertex != pfCandidates_vertex.end(); ++pfCandidate_vertex ) {
      pfCandToVertexAssociations_reversed.insert(pfCandidate_vertex->first, std::make_pair(vertex,  pfCandidate_vertex->second));
    }
  }

  return pfCandToVertexAssociations_reversed;
}

Variable Documentation

const double dR2Min =0.01*0.01

Definition at line 9 of file noPileUpMEtAuxFunctions.cc.

const int minPFCandToVertexAssocQuality = noPuUtils::kChHSAssoc

Definition at line 7 of file noPileUpMEtAuxFunctions.cc.

Referenced by isVertexAssociated(), and isVertexAssociated_fast().