#include <PFCand_AssoMapAlgos.h>

Inheritance diagram for PFCand_AssoMapAlgos:

Public Member Functions
std::pair< std::unique_ptr< PFCandToVertexAssMap >, std::unique_ptr< VertexToPFCandAssMap > >	createMappings (edm::Handle< reco::PFCandidateCollection > pfCandH, const edm::EventSetup &iSetup)

std::unique_ptr< PFCandToVertexAssMap >	CreatePFCandToVertexMap (edm::Handle< reco::PFCandidateCollection >, const edm::EventSetup &)

std::unique_ptr< VertexToPFCandAssMap >	CreateVertexToPFCandMap (edm::Handle< reco::PFCandidateCollection >, const edm::EventSetup &)

void	GetInputCollections (edm::Event &, const edm::EventSetup &) override

	PFCand_AssoMapAlgos (const edm::ParameterSet &, edm::ConsumesCollector &&)

std::unique_ptr< PFCandToVertexAssMap >	SortPFCandAssociationMap (PFCandToVertexAssMap , edm::EDProductGetter const getter)

Public Member Functions inherited from PF_PU_AssoMapAlgos
std::pair< std::unique_ptr< TrackToVertexAssMap >, std::unique_ptr< VertexToTrackAssMap > >	createMappings (edm::Handle< reco::TrackCollection > trkcollH, const edm::EventSetup &iSetup)

std::unique_ptr< TrackToVertexAssMap >	CreateTrackToVertexMap (edm::Handle< reco::TrackCollection >, const edm::EventSetup &)

std::unique_ptr< VertexToTrackAssMap >	CreateVertexToTrackMap (edm::Handle< reco::TrackCollection >, const edm::EventSetup &)

	PF_PU_AssoMapAlgos (const edm::ParameterSet &, edm::ConsumesCollector &)

	PF_PU_AssoMapAlgos (const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC)

std::unique_ptr< TrackToVertexAssMap >	SortAssociationMap (TrackToVertexAssMap *, edm::Handle< reco::TrackCollection >)

virtual	~PF_PU_AssoMapAlgos () noexcept(false)

Private Attributes
edm::Handle< reco::BeamSpot >	beamspotH

edm::ESHandle< MagneticField >	bFieldH

int	input_MaxNumAssociations_

edm::EDGetTokenT< reco::BeamSpot >	token_BeamSpot_

edm::EDGetTokenT< reco::VertexCollection >	token_VertexCollection_

edm::Handle< reco::VertexCollection >	vtxcollH

Additional Inherited Members
Protected Member Functions inherited from PF_PU_AssoMapAlgos
std::vector< reco::VertexRef >	CreateVertexVector (edm::Handle< reco::VertexCollection >)

int	DefineQuality (int, int, double)

void	EraseVertex (std::vector< reco::VertexRef > &, reco::VertexRef)

VertexStepPair	FindAssociation (const reco::TrackRef &, const std::vector< reco::VertexRef > &, edm::ESHandle< MagneticField >, const edm::EventSetup &, edm::Handle< reco::BeamSpot >, int)

Detailed Description

Definition at line 41 of file PFCand_AssoMapAlgos.h.

Constructor & Destructor Documentation

◆ PFCand_AssoMapAlgos()

PFCand_AssoMapAlgos::PFCand_AssoMapAlgos	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector &&	iC
	)

Definition at line 14 of file PFCand_AssoMapAlgos.cc.

     : PF_PU_AssoMapAlgos(iConfig, iC) {
   input_MaxNumAssociations_ = iConfig.getParameter<int>("MaxNumberOfAssociations");
  
   token_VertexCollection_ = iC.consumes<VertexCollection>(iConfig.getParameter<InputTag>("VertexCollection"));
  
   token_BeamSpot_ = iC.consumes<BeamSpot>(iConfig.getParameter<InputTag>("BeamSpot"));
 }

References edm::ParameterSet::getParameter(), input_MaxNumAssociations_, token_BeamSpot_, and token_VertexCollection_.

Member Function Documentation

◆ createMappings()

std::pair< std::unique_ptr< PFCandToVertexAssMap >, std::unique_ptr< VertexToPFCandAssMap > > PFCand_AssoMapAlgos::createMappings	(	edm::Handle< reco::PFCandidateCollection >	pfCandH,
		const edm::EventSetup &	iSetup
	)

Definition at line 43 of file PFCand_AssoMapAlgos.cc.

                                                                                                              {
   unique_ptr<PFCandToVertexAssMap> pfcand2vertex(new PFCandToVertexAssMap(vtxcollH, pfCandH));
   unique_ptr<VertexToPFCandAssMap> vertex2pfcand(new VertexToPFCandAssMap(pfCandH, vtxcollH));
  
   int num_vertices = vtxcollH->size();
   if (num_vertices < input_MaxNumAssociations_)
     input_MaxNumAssociations_ = num_vertices;
   vector<VertexRef> vtxColl_help;
   if (input_MaxNumAssociations_ == 1)
     vtxColl_help = CreateVertexVector(vtxcollH);
  
   for (unsigned i = 0; i < pfCandH->size(); i++) {
     PFCandidateRef candref(pfCandH, i);
  
     if (input_MaxNumAssociations_ > 1)
       vtxColl_help = CreateVertexVector(vtxcollH);
  
     VertexPfcQuality VtxPfcQual;
  
     TrackRef PFCtrackref = candref->trackRef();
  
     if (PFCtrackref.isNull()) {
       for (int assoc_ite = 0; assoc_ite < input_MaxNumAssociations_; ++assoc_ite) {
         int quality = -1 - assoc_ite;
  
         // Insert the best vertex and the pair of track and the quality of this association in the map
         pfcand2vertex->insert(vtxColl_help.at(0), make_pair(candref, quality));
         vertex2pfcand->insert(candref, make_pair(vtxColl_help.at(0), quality));
  
         //cleanup only if multiple iterations are made
         if (input_MaxNumAssociations_ > 1)
           PF_PU_AssoMapAlgos::EraseVertex(vtxColl_help, vtxColl_help.at(0));
       }
  
     } else {
       TransientTrack transtrk(PFCtrackref, &(*bFieldH));
       transtrk.setBeamSpot(*beamspotH);
       transtrk.setES(iSetup);
  
       for (int assoc_ite = 0; assoc_ite < input_MaxNumAssociations_; ++assoc_ite) {
         VertexStepPair assocVtx = FindAssociation(PFCtrackref, vtxColl_help, bFieldH, iSetup, beamspotH, assoc_ite);
         int step = assocVtx.second;
         double distance = (IPTools::absoluteImpactParameter3D(transtrk, *(assocVtx.first))).second.value();
  
         int quality = DefineQuality(assoc_ite, step, distance);
  
         // Insert the best vertex and the pair of track and the quality of this association in the map
         pfcand2vertex->insert(assocVtx.first, make_pair(candref, quality));
         vertex2pfcand->insert(candref, make_pair(assocVtx.first, quality));
  
         //cleanup only if multiple iterations are made
         if (input_MaxNumAssociations_ > 2)
           PF_PU_AssoMapAlgos::EraseVertex(vtxColl_help, assocVtx.first);
       }
  
     }  //check PFCtrackref.isNull
  
   }  //i on pfCandH
  
   return {std::move(pfcand2vertex), std::move(vertex2pfcand)};
 }

References IPTools::absoluteImpactParameter3D(), beamspotH, bFieldH, PF_PU_AssoMapAlgos::CreateVertexVector(), PF_PU_AssoMapAlgos::DefineQuality(), HLT_FULL_cff::distance, PF_PU_AssoMapAlgos::EraseVertex(), PF_PU_AssoMapAlgos::FindAssociation(), mps_fire::i, input_MaxNumAssociations_, edm::Ref< C, T, F >::isNull(), eostools::move(), quality, edm::second(), reco::TransientTrack::setBeamSpot(), reco::TransientTrack::setES(), and vtxcollH.

Referenced by CreatePFCandToVertexMap(), CreateVertexToPFCandMap(), and PFCand_AssoMap::produce().

◆ CreatePFCandToVertexMap()

std::unique_ptr< PFCandToVertexAssMap > PFCand_AssoMapAlgos::CreatePFCandToVertexMap	(	edm::Handle< reco::PFCandidateCollection >	pfCandH,
		const edm::EventSetup &	iSetup
	)

Definition at line 109 of file PFCand_AssoMapAlgos.cc.

                                                                                {
   return createMappings(pfCandH, iSetup).first;
 }

References createMappings().

◆ CreateVertexToPFCandMap()

std::unique_ptr< VertexToPFCandAssMap > PFCand_AssoMapAlgos::CreateVertexToPFCandMap	(	edm::Handle< reco::PFCandidateCollection >	pfCandH,
		const edm::EventSetup &	iSetup
	)

Definition at line 118 of file PFCand_AssoMapAlgos.cc.

                                                                                {
   return createMappings(pfCandH, iSetup).second;
 }

References createMappings().

◆ GetInputCollections()

void PFCand_AssoMapAlgos::GetInputCollections	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overridevirtual

Reimplemented from PF_PU_AssoMapAlgos.

Definition at line 27 of file PFCand_AssoMapAlgos.cc.

                                                                                            {
   PF_PU_AssoMapAlgos::GetInputCollections(iEvent, iSetup);
  
   //get the offline beam spot
   iEvent.getByToken(token_BeamSpot_, beamspotH);
  
   //get the input vertex collection
   iEvent.getByToken(token_VertexCollection_, vtxcollH);
  
   iSetup.get<IdealMagneticFieldRecord>().get(bFieldH);
 }

References beamspotH, bFieldH, edm::EventSetup::get(), get, PF_PU_AssoMapAlgos::GetInputCollections(), iEvent, token_BeamSpot_, token_VertexCollection_, and vtxcollH.

Referenced by PFCand_AssoMap::produce().

◆ SortPFCandAssociationMap()

std::unique_ptr< PFCandToVertexAssMap > PFCand_AssoMapAlgos::SortPFCandAssociationMap	(	PFCandToVertexAssMap *	pfcvertexassInput,
		edm::EDProductGetter const *	getter
	)

Definition at line 127 of file PFCand_AssoMapAlgos.cc.

                                                                                {
   //create a new PFCandVertexAssMap for the Output which will be sorted
   unique_ptr<PFCandToVertexAssMap> pfcvertexassOutput(new PFCandToVertexAssMap(getter));
  
   //Create and fill a vector of pairs of vertex and the summed (pT)**2 of the pfcandidates associated to the vertex
   VertexPtsumVector vertexptsumvector;
  
   //loop over all vertices in the association map
   for (PFCandToVertexAssMap::const_iterator assomap_ite = pfcvertexassInput->begin();
        assomap_ite != pfcvertexassInput->end();
        assomap_ite++) {
     const VertexRef assomap_vertexref = assomap_ite->key;
     const PFCandQualityPairVector pfccoll = assomap_ite->val;
  
     float ptsum = 0;
  
     PFCandidateRef pfcandref;
  
     //get the pfcandidates associated to the vertex and calculate the pT**2
     for (unsigned int pfccoll_ite = 0; pfccoll_ite < pfccoll.size(); pfccoll_ite++) {
       pfcandref = pfccoll[pfccoll_ite].first;
       int quality = pfccoll[pfccoll_ite].second;
  
       if ((quality <= 2) && (quality != -1))
         continue;
  
       double man_pT = pfcandref->pt();
       if (man_pT > 0.)
         ptsum += man_pT * man_pT;
     }
  
     vertexptsumvector.push_back(make_pair(assomap_vertexref, ptsum));
   }
  
   while (!vertexptsumvector.empty()) {
     VertexRef vertexref_highestpT;
     float highestpT = 0.;
     int highestpT_index = 0;
  
     for (unsigned int vtxptsumvec_ite = 0; vtxptsumvec_ite < vertexptsumvector.size(); vtxptsumvec_ite++) {
       if (vertexptsumvector[vtxptsumvec_ite].second > highestpT) {
         vertexref_highestpT = vertexptsumvector[vtxptsumvec_ite].first;
         highestpT = vertexptsumvector[vtxptsumvec_ite].second;
         highestpT_index = vtxptsumvec_ite;
       }
     }
  
     //loop over all vertices in the association map
     for (PFCandToVertexAssMap::const_iterator assomap_ite = pfcvertexassInput->begin();
          assomap_ite != pfcvertexassInput->end();
          assomap_ite++) {
       const VertexRef assomap_vertexref = assomap_ite->key;
       const PFCandQualityPairVector pfccoll = assomap_ite->val;
  
       //if the vertex from the association map the vertex with the highest pT
       //insert all associated pfcandidates in the output Association Map
       if (assomap_vertexref == vertexref_highestpT)
         for (unsigned int pfccoll_ite = 0; pfccoll_ite < pfccoll.size(); pfccoll_ite++)
           pfcvertexassOutput->insert(assomap_vertexref, pfccoll[pfccoll_ite]);
     }
  
     vertexptsumvector.erase(vertexptsumvector.begin() + highestpT_index);
   }
  
   return pfcvertexassOutput;
 }