PATRefitVertexProducer Class Reference

Inheritance diagram for PATRefitVertexProducer:

Public Member Functions
	PATRefitVertexProducer (const edm::ParameterSet &)
void	produce (edm::Event &, const edm::EventSetup &) override
	~PATRefitVertexProducer () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &)

Private Attributes
edm::EDGetTokenT< reco::BeamSpot >	srcBeamSpot_
edm::EDGetTokenT< std::vector< pat::PackedCandidate > >	srcCands_
edm::EDGetTokenT< std::vector< pat::PackedCandidate > >	srcEleKfTracks_
edm::EDGetTokenT< std::vector< pat::PackedCandidate > >	srcLostTracks_
edm::EDGetTokenT< reco::VertexCollection >	srcVertices_
edm::ESGetToken< TransientTrackBuilder, TransientTrackRecord >	transTrackBuilderToken_
bool	useBeamSpot_
bool	useEleKfTracks_
bool	useLostTracks_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

This producer is intended to take packedCandidates with tracks associated to the PV and refit the PV (applying or not) BeamSpot constraint

Michal Bluj, NCBJ Warsaw (and then others)

Definition at line 29 of file PATRefitVertexProducer.cc.

Constructor & Destructor Documentation

PATRefitVertexProducer()

PATRefitVertexProducer::PATRefitVertexProducer ( const edm::ParameterSet &  cfg )

explicit

Definition at line 50 of file PATRefitVertexProducer.cc.

    : srcCands_(consumes<std::vector<pat::PackedCandidate> >(cfg.getParameter<edm::InputTag>("srcCands"))),
      srcLostTracks_(consumes<std::vector<pat::PackedCandidate> >(cfg.getParameter<edm::InputTag>("srcLostTracks"))),
      srcEleKfTracks_(consumes<std::vector<pat::PackedCandidate> >(cfg.getParameter<edm::InputTag>("srcEleKfTracks"))),
      srcVertices_(consumes<reco::VertexCollection>(cfg.getParameter<edm::InputTag>("srcVertices"))),
      srcBeamSpot_(consumes<reco::BeamSpot>(cfg.getParameter<edm::InputTag>("srcBeamSpot"))),
      transTrackBuilderToken_(esConsumes(edm::ESInputTag("", "TransientTrackBuilder"))),
      useBeamSpot_(cfg.getParameter<bool>("useBeamSpot")),
      useLostTracks_(cfg.getParameter<bool>("useLostTracks")),
      useEleKfTracks_(cfg.getParameter<bool>("useEleKfTracks")) {
  produces<reco::VertexCollection>();
}

~PATRefitVertexProducer()

PATRefitVertexProducer::~PATRefitVertexProducer ( )

inlineoverride

Definition at line 32 of file PATRefitVertexProducer.cc.

{}

Member Function Documentation

fillDescriptions()

void PATRefitVertexProducer::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 174 of file PATRefitVertexProducer.cc.

References edm::ConfigurationDescriptions::addWithDefaultLabel(), submitPVResolutionJobs::desc, and ProducerED_cfi::InputTag.

                                                                                        {
  // patRefitVertexProducer
  edm::ParameterSetDescription desc;

  desc.add<edm::InputTag>("srcVertices", edm::InputTag("offlineSlimmedPrimaryVertices"));
  desc.add<edm::InputTag>("srcCands", edm::InputTag("packedPFCandidates"));
  desc.add<edm::InputTag>("srcLostTracks", edm::InputTag("lostTracks"));
  desc.add<edm::InputTag>("srcEleKfTracks", edm::InputTag("lostTracks:eleTracks"));
  desc.add<edm::InputTag>("srcBeamSpot", edm::InputTag("offlineBeamSpot"));
  desc.add<bool>("useBeamSpot", true)->setComment("Refit PV with beam-spot constraint");
  desc.add<bool>("useLostTracks", true)
      ->setComment("Use collection of tracks not used by PF-candidates, aka lost-tracks");
  desc.add<bool>("useEleKfTracks", true)
      ->setComment("Use collection of electron KF-tracks instead of GSF-tracks of electron PF-candidates");

  descriptions.addWithDefaultLabel(desc);
}

produce()

void PATRefitVertexProducer::produce ( edm::Event &  evt, const edm::EventSetup &  es  )

override

Definition at line 63 of file PATRefitVertexProducer.cc.

References funct::abs(), isoTrack_cff::beamSpot, TransientTrackBuilder::build(), HLT_2024v14_cff::cands, edm::Event::getByToken(), edm::EventSetup::getData(), TransientVertex::isValid(), submitPVResolutionJobs::key, eostools::move(), TransientVertex::position(), edm::Event::put(), quality, AdaptiveVertexFitter::setWeightThreshold(), srcBeamSpot_, srcCands_, srcEleKfTracks_, srcLostTracks_, srcVertices_, transTrackBuilderToken_, useBeamSpot_, pat::PackedCandidate::UsedInFitLoose, pat::PackedCandidate::UsedInFitTight, useEleKfTracks_, useLostTracks_, AdaptiveVertexFitter::vertex(), AlignmentTracksFromVertexSelector_cfi::vertices, and PV3DBase< T, PVType, FrameType >::z().

                                                                           {
  // Obtain collections
  edm::Handle<std::vector<pat::PackedCandidate> > cands;
  evt.getByToken(srcCands_, cands);

  edm::Handle<std::vector<pat::PackedCandidate> > lostTrackCands;
  if (useLostTracks_)
    evt.getByToken(srcLostTracks_, lostTrackCands);

  edm::Handle<std::vector<pat::PackedCandidate> > eleKfTrackCands;
  if (useEleKfTracks_)
    evt.getByToken(srcEleKfTracks_, eleKfTrackCands);

  edm::Handle<reco::VertexCollection> vertices;
  evt.getByToken(srcVertices_, vertices);
  const reco::Vertex& pv = vertices->front();
  size_t vtxIdx = 0;

  edm::Handle<reco::BeamSpot> beamSpot;
  if (useBeamSpot_)
    evt.getByToken(srcBeamSpot_, beamSpot);

  // Get transient track builder
  const TransientTrackBuilder& transTrackBuilder = es.getData(transTrackBuilderToken_);

  // Output collection
  auto outputVertices = std::make_unique<reco::VertexCollection>();
  outputVertices->reserve(1);

  // Create a new track collection for vertex refit
  std::vector<reco::TransientTrack> transTracks;

  // loop over the PFCandidates
  for (const auto& cand : (*cands)) {
    if (cand.charge() == 0 || cand.vertexRef().isNull())
      continue;
    if (cand.bestTrack() == nullptr)
      continue;
    auto key = cand.vertexRef().key();
    auto quality = cand.pvAssociationQuality();
    if (key != vtxIdx ||
        (quality != pat::PackedCandidate::UsedInFitTight && quality != pat::PackedCandidate::UsedInFitLoose))
      continue;
    if (useEleKfTracks_ && std::abs(cand.pdgId()) == 11)
      continue;
    transTracks.push_back(transTrackBuilder.build(cand.bestTrack()));
  }

  // loop over the lostTracks
  if (useLostTracks_) {
    for (const auto& cand : (*lostTrackCands)) {
      if (cand.charge() == 0 || cand.vertexRef().isNull())
        continue;
      if (cand.bestTrack() == nullptr)
        continue;
      auto key = cand.vertexRef().key();
      auto quality = cand.pvAssociationQuality();
      if (key != vtxIdx ||
          (quality != pat::PackedCandidate::UsedInFitTight && quality != pat::PackedCandidate::UsedInFitLoose))
        continue;
      transTracks.push_back(transTrackBuilder.build(cand.bestTrack()));
    }
  }

  // loop over the electronKfTracks
  if (useEleKfTracks_) {
    for (const auto& cand : (*eleKfTrackCands)) {
      if (cand.charge() == 0 || cand.vertexRef().isNull())
        continue;
      if (cand.bestTrack() == nullptr)
        continue;
      auto key = cand.vertexRef().key();
      auto quality = cand.pvAssociationQuality();
      if (key != vtxIdx ||
          (quality != pat::PackedCandidate::UsedInFitTight && quality != pat::PackedCandidate::UsedInFitLoose))
        continue;
      transTracks.push_back(transTrackBuilder.build(cand.bestTrack()));
    }
  }

  // Refit the vertex
  TransientVertex transVtx;
  reco::Vertex refitPV(pv);  // initialized to the original PV

  bool fitOK = true;
  if (transTracks.size() >= 3) {
    AdaptiveVertexFitter avf;
    avf.setWeightThreshold(0.1);  // weight per track, allow almost every fit, else --> exception
    if (!useBeamSpot_) {
      transVtx = avf.vertex(transTracks);
    } else {
      transVtx = avf.vertex(transTracks, *beamSpot);
    }
    if (!transVtx.isValid()) {
      fitOK = false;
    } else {
      //MB: protect against rare cases when transVtx is valid but its postion is ill-defined
      if (!std::isfinite(transVtx.position().z()))  //MB: it is enough to check one coordinate (?)
        fitOK = false;
    }
  } else
    fitOK = false;
  if (fitOK) {
    refitPV = transVtx;
  }

  outputVertices->push_back(refitPV);

  evt.put(std::move(outputVertices));
}

Member Data Documentation

srcBeamSpot_

edm::EDGetTokenT<reco::BeamSpot> PATRefitVertexProducer::srcBeamSpot_

private

Definition at line 43 of file PATRefitVertexProducer.cc.

Referenced by produce().

srcCands_

edm::EDGetTokenT<std::vector<pat::PackedCandidate> > PATRefitVertexProducer::srcCands_

private

Definition at line 41 of file PATRefitVertexProducer.cc.

Referenced by produce().

srcEleKfTracks_

edm::EDGetTokenT<std::vector<pat::PackedCandidate> > PATRefitVertexProducer::srcEleKfTracks_

private

Definition at line 41 of file PATRefitVertexProducer.cc.

Referenced by produce().

srcLostTracks_

edm::EDGetTokenT<std::vector<pat::PackedCandidate> > PATRefitVertexProducer::srcLostTracks_

private

Definition at line 41 of file PATRefitVertexProducer.cc.

Referenced by produce().

srcVertices_

edm::EDGetTokenT<reco::VertexCollection> PATRefitVertexProducer::srcVertices_

private

Definition at line 42 of file PATRefitVertexProducer.cc.

Referenced by produce().

transTrackBuilderToken_

edm::ESGetToken<TransientTrackBuilder, TransientTrackRecord> PATRefitVertexProducer::transTrackBuilderToken_

private

Definition at line 44 of file PATRefitVertexProducer.cc.

Referenced by produce().

useBeamSpot_

bool PATRefitVertexProducer::useBeamSpot_

private

Definition at line 45 of file PATRefitVertexProducer.cc.

Referenced by produce().

useEleKfTracks_

bool PATRefitVertexProducer::useEleKfTracks_

private

Definition at line 47 of file PATRefitVertexProducer.cc.

Referenced by produce().

useLostTracks_

bool PATRefitVertexProducer::useLostTracks_

private

Definition at line 46 of file PATRefitVertexProducer.cc.

Referenced by produce().