#include <PFCand_AssoMapAlgos.h>

Inheritance diagram for PFCand_AssoMapAlgos:

Public Member Functions
std::auto_ptr< PFCandToVertexAssMap >	CreatePFCandToVertexMap (edm::Handle< reco::PFCandidateCollection >, const edm::EventSetup &)

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

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

	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::auto_ptr< TrackToVertexAssMap >	CreateTrackToVertexMap (edm::Handle< reco::TrackCollection >, const edm::EventSetup &)

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

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

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

std::unique_ptr< TrackToVertexAssMap >	SortAssociationMap (TrackToVertexAssMap *)

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 &, std::vector< reco::VertexRef > *, edm::ESHandle< MagneticField >, const edm::EventSetup &, edm::Handle< reco::BeamSpot >, int)

Detailed Description

Definition at line 40 of file PFCand_AssoMapAlgos.h.

Constructor & Destructor Documentation

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

Definition at line 14 of file PFCand_AssoMapAlgos.cc.

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

                                                                                                   :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"));
 
 }

Member Function Documentation

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

Definition at line 50 of file PFCand_AssoMapAlgos.cc.

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

Referenced by PFCand_AssoMap::produce().

 {
 
         auto_ptr<PFCandToVertexAssMap> pfcand2vertex(new PFCandToVertexAssMap(vtxcollH, pfCandH));
 
     int num_vertices = vtxcollH->size();
     if ( num_vertices < input_MaxNumAssociations_) input_MaxNumAssociations_ = num_vertices;
 
     for( unsigned i=0; i<pfCandH->size(); i++ ) {
 
           PFCandidateRef candref(pfCandH, i);
 
       vector<VertexRef>* 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) );
 
           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) );
 
           PF_PU_AssoMapAlgos::EraseVertex(vtxColl_help, assocVtx.first);
 
         }
 
       }
 
       delete vtxColl_help;
         }
 
     return pfcand2vertex;
 
 }

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

Definition at line 116 of file PFCand_AssoMapAlgos.cc.

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

Referenced by PFCand_AssoMap::produce().

 {
 
     auto_ptr<VertexToPFCandAssMap> vertex2pfcand(new VertexToPFCandAssMap(pfCandH, vtxcollH));
 
     int num_vertices = vtxcollH->size();
     if ( num_vertices < input_MaxNumAssociations_) input_MaxNumAssociations_ = num_vertices;
 
     for( unsigned i=0; i<pfCandH->size(); i++ ) {
 
           PFCandidateRef candref(pfCandH, i);
 
       vector<VertexRef>* 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
               vertex2pfcand->insert( candref, make_pair(vtxColl_help->at(0), quality) );
 
           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
               vertex2pfcand->insert( candref, make_pair(assocVtx.first, quality) );
 
           PF_PU_AssoMapAlgos::EraseVertex(vtxColl_help, assocVtx.first);
 
         }
 
       }
 
       delete vtxColl_help;
         }
 
     return vertex2pfcand;
 }

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

virtual

Reimplemented from PF_PU_AssoMapAlgos.

Definition at line 30 of file PFCand_AssoMapAlgos.cc.

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

Referenced by PFCand_AssoMap::produce().

 {
 
     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);
 
 }

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

Definition at line 181 of file PFCand_AssoMapAlgos.cc.

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

Referenced by PFCand_AssoMap::produce().

 {
     //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.size()!=0){
 
       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;
 }