noPileUpMEtAuxFunctions.h

Go to the documentation of this file.

#ifndef RecoMET_METPUSubtraction_noPileUpMEtAuxFunctions_h
#define RecoMET_METPUSubtraction_noPileUpMEtAuxFunctions_h

#include "DataFormats/Common/interface/AssociationMap.h"
#include "DataFormats/Common/interface/OneToManyWithQuality.h"
#include "DataFormats/Common/interface/Ref.h"

#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"

#include "CommonTools/RecoUtils/interface/PFCand_AssoMapAlgos.h"

 // 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
namespace noPuUtils {
  enum {kNeutral=0, kChNoAssoc, kChPUAssoc, kChHSAssoc};
}

typedef edm::AssociationMap<edm::OneToManyWithQuality<reco::PFCandidateCollection, reco::VertexCollection, int> >
  reversedPFCandidateToVertexAssociationMap;

int isVertexAssociated(const reco::PFCandidate&, const PFCandToVertexAssMap&, const reco::VertexCollection&, double);

reversedPFCandidateToVertexAssociationMap reversePFCandToVertexAssociation(const PFCandToVertexAssMap&);

int isVertexAssociated_fast(const reco::PFCandidateRef&, const reversedPFCandidateToVertexAssociationMap&, const reco::VertexCollection&, double, int&, int);

#endif