PFCand_AssoMapAlgos Class Reference

#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)
std::unique_ptr < PFCandToVertexAssMap >	CreatePFCandToVertexMap (edm::Handle< reco::PFCandidateCollection >)
std::unique_ptr < VertexToPFCandAssMap >	CreateVertexToPFCandMap (edm::Handle< reco::PFCandidateCollection >)
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)
std::unique_ptr < TrackToVertexAssMap >	CreateTrackToVertexMap (edm::Handle< reco::TrackCollection >)
std::unique_ptr < VertexToTrackAssMap >	CreateVertexToTrackMap (edm::Handle< reco::TrackCollection >)
	PF_PU_AssoMapAlgos (const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC)
	PF_PU_AssoMapAlgos (const edm::ParameterSet &, edm::ConsumesCollector &)
std::unique_ptr < TrackToVertexAssMap >	SortAssociationMap (TrackToVertexAssMap *, edm::Handle< reco::TrackCollection >)
template<typename TREF >
VertexRef	TrackWeightAssociation (const TREF &trackRef, const std::vector< reco::VertexRef > &vtxcollV)
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_
const edm::ESGetToken < MagneticField, IdealMagneticFieldRecord >	token_bField_
const edm::ESGetToken < GlobalTrackingGeometry, GlobalTrackingGeometryRecord >	token_TrackingGeometry_
edm::EDGetTokenT < reco::VertexCollection >	token_VertexCollection_
edm::ESHandle < GlobalTrackingGeometry >	trackingGeometryH
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 >, edm::ESHandle< GlobalTrackingGeometry >, edm::Handle< reco::BeamSpot >, int)

Detailed Description

Definition at line 42 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), token_bField_(iC.esConsumes()), token_TrackingGeometry_(iC.esConsumes()) {
  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::pair< std::unique_ptr< PFCandToVertexAssMap >, std::unique_ptr< VertexToPFCandAssMap > > PFCand_AssoMapAlgos::createMappings

(

edm::Handle< reco::PFCandidateCollection >

pfCandH

)

Definition at line 44 of file PFCand_AssoMapAlgos.cc.

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::setTrackingGeometry(), trackingGeometryH, and vtxcollH.

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

                                                                                {
  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.setTrackingGeometry(trackingGeometryH);

      for (int assoc_ite = 0; assoc_ite < input_MaxNumAssociations_; ++assoc_ite) {
        VertexStepPair assocVtx =
            FindAssociation(PFCtrackref, vtxColl_help, bFieldH, trackingGeometryH, 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)};
}

std::unique_ptr< PFCandToVertexAssMap > PFCand_AssoMapAlgos::CreatePFCandToVertexMap

(

edm::Handle< reco::PFCandidateCollection >

pfCandH

)

Definition at line 111 of file PFCand_AssoMapAlgos.cc.

References createMappings().

                                                  {
  return createMappings(pfCandH).first;
}

std::unique_ptr< VertexToPFCandAssMap > PFCand_AssoMapAlgos::CreateVertexToPFCandMap

(

edm::Handle< reco::PFCandidateCollection >

pfCandH

)

Definition at line 120 of file PFCand_AssoMapAlgos.cc.

References createMappings().

                                                  {
  return createMappings(pfCandH).second;
}

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.

References beamspotH, bFieldH, edm::Event::getByToken(), edm::EventSetup::getHandle(), PF_PU_AssoMapAlgos::GetInputCollections(), token_BeamSpot_, token_bField_, token_TrackingGeometry_, token_VertexCollection_, trackingGeometryH, 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);

  bFieldH = iSetup.getHandle(token_bField_);
  trackingGeometryH = iSetup.getHandle(token_TrackingGeometry_);
}

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

Definition at line 129 of file PFCand_AssoMapAlgos.cc.

References edm::AssociationMap< Tag >::begin(), edm::AssociationMap< Tag >::end(), edm::Ref< C, T, F >::key(), 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.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;
}

Member Data Documentation

edm::Handle<reco::BeamSpot> PFCand_AssoMapAlgos::beamspotH

private

Definition at line 76 of file PFCand_AssoMapAlgos.h.

Referenced by createMappings(), and GetInputCollections().

edm::ESHandle<MagneticField> PFCand_AssoMapAlgos::bFieldH

private

Definition at line 79 of file PFCand_AssoMapAlgos.h.

Referenced by createMappings(), and GetInputCollections().

int PFCand_AssoMapAlgos::input_MaxNumAssociations_

private

Definition at line 70 of file PFCand_AssoMapAlgos.h.

Referenced by createMappings(), and PFCand_AssoMapAlgos().

edm::EDGetTokenT<reco::BeamSpot> PFCand_AssoMapAlgos::token_BeamSpot_

private

Definition at line 75 of file PFCand_AssoMapAlgos.h.

Referenced by GetInputCollections(), and PFCand_AssoMapAlgos().

const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> PFCand_AssoMapAlgos::token_bField_

private

Definition at line 78 of file PFCand_AssoMapAlgos.h.

Referenced by GetInputCollections().

const edm::ESGetToken<GlobalTrackingGeometry, GlobalTrackingGeometryRecord> PFCand_AssoMapAlgos::token_TrackingGeometry_

private

Definition at line 80 of file PFCand_AssoMapAlgos.h.

Referenced by GetInputCollections().

edm::EDGetTokenT<reco::VertexCollection> PFCand_AssoMapAlgos::token_VertexCollection_

private

Definition at line 72 of file PFCand_AssoMapAlgos.h.

Referenced by GetInputCollections(), and PFCand_AssoMapAlgos().

edm::ESHandle<GlobalTrackingGeometry> PFCand_AssoMapAlgos::trackingGeometryH

private

Definition at line 81 of file PFCand_AssoMapAlgos.h.

Referenced by createMappings(), and GetInputCollections().

edm::Handle<reco::VertexCollection> PFCand_AssoMapAlgos::vtxcollH

private

Definition at line 73 of file PFCand_AssoMapAlgos.h.

Referenced by createMappings(), and GetInputCollections().