#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	GetPFCandInputCollections (edm::Event &, const edm::EventSetup &)

	PFCand_AssoMapAlgos (const edm::ParameterSet &)

void	SetMotherPointer (edm::Handle< reco::PFCandidateCollection > &)

std::auto_ptr < PFCandToVertexAssMap >	SortPFCandAssociationMap (PFCandToVertexAssMap *)

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

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

	PF_PU_AssoMapAlgos (const edm::ParameterSet &)

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

virtual	~PF_PU_AssoMapAlgos ()

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

edm::ESHandle< MagneticField >	bFieldH

edm::InputTag	input_BeamSpot_

int	input_MaxNumAssociations_

edm::InputTag	input_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 (StepDistancePairVector, 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 34 of file PFCand_AssoMapAlgos.h.

Constructor & Destructor Documentation

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

Definition at line 14 of file PFCand_AssoMapAlgos.cc.

References edm::ParameterSet::getParameter(), input_BeamSpot_, input_MaxNumAssociations_, and input_VertexCollection_.

                                                                       :PF_PU_AssoMapAlgos(iConfig)
 {
 
   input_MaxNumAssociations_ = iConfig.getParameter<int>("MaxNumberOfAssociations");
 
   input_VertexCollection_= iConfig.getParameter<InputTag>("VertexCollection");
 
   input_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 69 of file PFCand_AssoMapAlgos.cc.

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

Referenced by PFCand_AssoMap::produce().

 {
 
   auto_ptr<PFCandToVertexAssMap> pfcand2vertex(new PFCandToVertexAssMap());
 
   int num_associations = input_MaxNumAssociations_;
   int num_vertices = vtxcollH->size();
   if ( num_vertices < num_associations) num_associations = num_vertices;
    
   for( unsigned i=0; i<pfCandH->size(); i++ ) {
      
     PFCandidateRef candref(pfCandH, i);
 
     vector<VertexRef>* vtxColl_help = CreateVertexVector(vtxcollH);
     StepDistancePairVector distances;
 
     VertexPfcQuality VtxPfcQual;
 
     TrackRef PFCtrackref = candref->trackRef();
 
     if ( PFCtrackref.isNull() ){
 
       for ( int assoc_ite = 0; assoc_ite < num_associations; ++assoc_ite ) {
 
         int quality = -1 - assoc_ite;
 
         // edm::LogInfo("Prtcl2VtxAssociation") << "associating track with PdgId " << candref->pdgId() << ": Pt = " << candref->pt() << ","
         // << " eta = " << candref->eta() << ", phi = " << candref->phi()
         // << " to vertex: z = " << vtxColl_help->at(0)->position().z() << " with quality q = " << quality << std::endl;
 
         // 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 < num_associations; ++assoc_ite ) {
 
         VertexStepPair assocVtx = FindAssociation(PFCtrackref, vtxColl_help, bFieldH, iSetup, beamspotH, assoc_ite);
         VertexRef associatedVertex = assocVtx.first;
         int step = assocVtx.second;
     double distance = ( IPTools::absoluteImpactParameter3D( transtrk, *associatedVertex ) ).second.value();
        
     int quality = DefineQuality(distances, step, distance);
     distances.push_back( make_pair(step, distance) );
 
         // edm::LogInfo("Prtcl2VtxAssociation") << "associating track with PdgId " << candref->pdgId() << ": Pt = " << candref->pt() << ","
         // << " eta = " << candref->eta() << ", phi = " << candref->phi()
         // << " to vertex: z = " << vtxColl_help->at(0)->position().z() << " with quality q = " << quality << std::endl;
 
         // Insert the best vertex and the pair of track and the quality of this association in the map
         pfcand2vertex->insert( associatedVertex, make_pair(candref, quality) );
 
     PF_PU_AssoMapAlgos::EraseVertex(vtxColl_help, associatedVertex);
 
       }
 
     }
 
     delete vtxColl_help;
 
   }
 
   return pfcand2vertex;
 
 }

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

Definition at line 148 of file PFCand_AssoMapAlgos.cc.

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

Referenced by PFCand_AssoMap::produce().

 {
 
   auto_ptr<VertexToPFCandAssMap> vertex2pfcand(new VertexToPFCandAssMap());
 
   int num_associations = input_MaxNumAssociations_;
   int num_vertices = vtxcollH->size();
   if ( num_vertices < num_associations) num_associations = num_vertices;
    
   for( unsigned i=0; i<pfCandH->size(); i++ ) {
      
     PFCandidateRef candref(pfCandH, i);
 
     vector<VertexRef>* vtxColl_help = CreateVertexVector(vtxcollH);
     StepDistancePairVector distances;
 
     VertexPfcQuality VtxPfcQual;
 
     TrackRef PFCtrackref = candref->trackRef();
 
     if ( PFCtrackref.isNull() ) {
 
       for ( int assoc_ite = 0; assoc_ite < num_associations; ++assoc_ite ) {
 
     int quality = -1 - assoc_ite;
 
         // edm::LogInfo("Vtx2PrtclAssociation") << "associating track with PdgId " << candref->pdgId() << ": Pt = " << candref->pt() << ","
         // << " eta = " << candref->eta() << ", phi = " << candref->phi()
         // << " to vertex: z = " << vtxColl_help->at(0)->position().z() << " with quality q = " << quality << std::endl;
 
         // 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 < num_associations; ++assoc_ite ) {
 
          VertexStepPair assocVtx = FindAssociation(PFCtrackref, vtxColl_help, bFieldH, iSetup, beamspotH, assoc_ite);
          VertexRef associatedVertex = assocVtx.first;
          int step = assocVtx.second;
      double distance = ( IPTools::absoluteImpactParameter3D( transtrk, *associatedVertex ) ).second.value();
        
      int quality = DefineQuality(distances, step, distance);
      distances.push_back( make_pair(step, distance) );
 
      // edm::LogInfo("Vtx2PrtclAssociation") << "associating track with PdgId " << candref->pdgId() << ": Pt = " << candref->pt() << ","
      // << " eta = " << candref->eta() << ", phi = " << candref->phi()
      // << " to vertex: z = " << vtxColl_help->at(0)->position().z() << " with quality q = " << quality << std::endl;
 
      // Insert the best vertex and the pair of track and the quality of this association in the map
          vertex2pfcand->insert( candref, make_pair(associatedVertex, quality) );
 
      PF_PU_AssoMapAlgos::EraseVertex(vtxColl_help, associatedVertex);
 
        }
 
      }
 
      delete vtxColl_help;
 
   }
 
   return vertex2pfcand;
 
 }

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

Definition at line 30 of file PFCand_AssoMapAlgos.cc.

References beamspotH, bFieldH, edm::EventSetup::get(), edm::Event::getByLabel(), PF_PU_AssoMapAlgos::GetInputCollections(), input_BeamSpot_, input_VertexCollection_, and vtxcollH.

Referenced by PFCand_AssoMap::produce().

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

void PFCand_AssoMapAlgos::SetMotherPointer ( edm::Handle< reco::PFCandidateCollection > & pfCandH )

Definition at line 50 of file PFCand_AssoMapAlgos.cc.

References i.

Referenced by PFCand_AssoMap::produce().

 {
   
   PFCandidateCollection& inCands = (PFCandidateCollection&) *pfCandH;
      
   for(unsigned i=0; i<inCands.size(); i++) {
   
     const PFCandidatePtr motherPtr( pfCandH, i );
     inCands[i].setSourceCandidatePtr( motherPtr );
   
   }
 
 }

std::auto_ptr< PFCandToVertexAssMap > PFCand_AssoMapAlgos::SortPFCandAssociationMap ( PFCandToVertexAssMap * pfcvertexassInput )

Definition at line 227 of file PFCand_AssoMapAlgos.cc.

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

Referenced by PFCand_AssoMap::produce().

 {
 
   //create a new PFCandVertexAssMap for the Output which will be sorted
   auto_ptr<PFCandToVertexAssMap> pfcvertexassOutput(new PFCandToVertexAssMap() );
 
   //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;
 
 }