PATLeptonTimeLifeInfoProducer< T > Class Template Reference

Produces lepton life-time information. More...

Inheritance diagram for PATLeptonTimeLifeInfoProducer< T >:

Public Member Functions
	PATLeptonTimeLifeInfoProducer (const edm::ParameterSet &)
void	produce (edm::Event &, const edm::EventSetup &) override
	~PATLeptonTimeLifeInfoProducer () 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 &descriptions)

Private Types
enum	PVChoice { useFront = 0, useClosestInDz }
using	TrackTimeLifeInfoMap = edm::ValueMap< TrackTimeLifeInfo >

Private Member Functions
const reco::Track *	getTrack (const T &)
template<>
const reco::Track *	getTrack (const pat::Electron &electron)
template<>
const reco::Track *	getTrack (const pat::Muon &muon)
template<>
const reco::Track *	getTrack (const pat::Tau &tau)
void	produceAndFillIPInfo (const T &, const TransientTrackBuilder &, const reco::Vertex &, TrackTimeLifeInfo &)
void	produceAndFillSVInfo (const T &, const TransientTrackBuilder &, const reco::Vertex &, TrackTimeLifeInfo &)
template<>
void	produceAndFillSVInfo (const pat::Tau &tau, const TransientTrackBuilder &transTrackBuilder, const reco::Vertex &pv, TrackTimeLifeInfo &info)

Static Private Member Functions
static bool	fitVertex (const std::vector< reco::TransientTrack > &transTrk, TransientVertex &transVtx)

Private Attributes
edm::EDGetTokenT< std::vector< T > >	leptonsToken_
int	pvChoice_
edm::EDGetTokenT< reco::VertexCollection >	pvToken_
const StringCutObjectSelector< T >	selector_
edm::ESGetToken< TransientTrackBuilder, TransientTrackRecord >	transTrackBuilderToken_

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

template<typename T>
class PATLeptonTimeLifeInfoProducer< T >

Produces lepton life-time information.

Author

Michal Bluj, NCBJ, Warsaw

Definition at line 40 of file PATLeptonTimeLifeInfoProducer.cc.

Member Typedef Documentation

TrackTimeLifeInfoMap

template<typename T >

using PATLeptonTimeLifeInfoProducer< T >::TrackTimeLifeInfoMap = edm::ValueMap<TrackTimeLifeInfo>

private

Definition at line 75 of file PATLeptonTimeLifeInfoProducer.cc.

Member Enumeration Documentation

PVChoice

template<typename T >

enum PATLeptonTimeLifeInfoProducer::PVChoice

private

Enumerator
useFront
useClosestInDz

Definition at line 73 of file PATLeptonTimeLifeInfoProducer.cc.

{ useFront = 0, useClosestInDz };

Constructor & Destructor Documentation

PATLeptonTimeLifeInfoProducer()

template<typename T >

PATLeptonTimeLifeInfoProducer< T >::PATLeptonTimeLifeInfoProducer ( const edm::ParameterSet &  cfg )

explicit

Definition at line 79 of file PATLeptonTimeLifeInfoProducer.cc.

    : leptonsToken_(consumes<std::vector<T>>(cfg.getParameter<edm::InputTag>("src"))),
      pvToken_(consumes<reco::VertexCollection>(cfg.getParameter<edm::InputTag>("pvSource"))),
      transTrackBuilderToken_(esConsumes(edm::ESInputTag("", "TransientTrackBuilder"))),
      selector_(cfg.getParameter<std::string>("selection")),
      pvChoice_(cfg.getParameter<int>("pvChoice")) {
  produces<TrackTimeLifeInfoMap>();
}

~PATLeptonTimeLifeInfoProducer()

template<typename T >

PATLeptonTimeLifeInfoProducer< T >::~PATLeptonTimeLifeInfoProducer ( )

inlineoverride

Definition at line 43 of file PATLeptonTimeLifeInfoProducer.cc.

{}

Member Function Documentation

fillDescriptions()

template<typename T >

void PATLeptonTimeLifeInfoProducer< T >::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 255 of file PATLeptonTimeLifeInfoProducer.cc.

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

                                                                                                  {
  // pat{Electron,Muon,Tau}TimeLifeInfoProducer
  edm::ParameterSetDescription desc;

  std::string lepCollName;
  if (typeid(T) == typeid(pat::Electron))
    lepCollName = "slimmedElectrons";
  else if (typeid(T) == typeid(pat::Muon))
    lepCollName = "slimmedMuons";
  else if (typeid(T) == typeid(pat::Tau))
    lepCollName = "slimmedTaus";
  desc.add<edm::InputTag>("src", edm::InputTag(lepCollName));
  desc.add<edm::InputTag>("pvSource", edm::InputTag("offlineSlimmedPrimaryVertices"));
  desc.add<std::string>("selection", "")->setComment("Selection required to produce and store time-life information");
  desc.add<int>("pvChoice", useFront)
      ->setComment(
          "Define PV to compute IP: 0: first PV, 1: PV with the smallest dz of the tau leading track (default: " +
          std::to_string(useFront) + ")");

  descriptions.addWithDefaultLabel(desc);
}

fitVertex()

template<typename T >

static bool PATLeptonTimeLifeInfoProducer< T >::fitVertex ( const std::vector< reco::TransientTrack > &  transTrk, TransientVertex &  transVtx  )

inlinestaticprivate

Definition at line 53 of file PATLeptonTimeLifeInfoProducer.cc.

References MillePedeFileConverter_cfg::e, TransientVertex::hasRefittedTracks(), TransientVertex::refittedTracks(), and KalmanVertexFitter::vertex().

                                                                                                  {
    if (transTrk.size() < 2)
      return false;
    try {
      KalmanVertexFitter kvf(true);
      transVtx = kvf.vertex(transTrk);
      return transVtx.hasRefittedTracks() && transVtx.refittedTracks().size() == transTrk.size();
    } catch (VertexException& e) {
      edm::LogWarning("PATLeptonTimeLifeInfoProducer") << " fitVertex failed: " << e.what();
      return false;
    }
  }

getTrack() [1/4]

template<typename T >

const reco::Track* PATLeptonTimeLifeInfoProducer< T >::getTrack ( const T &  )

private

getTrack() [2/4]

template<>

const reco::Track * PATLeptonTimeLifeInfoProducer< pat::Electron >::getTrack ( const pat::Electron &  electron )

private

Definition at line 146 of file PATLeptonTimeLifeInfoProducer.cc.

References HPSPFTauProducerPuppi_cfi::electron.

                                                                                                 {
  return electron.gsfTrack().isNonnull() ? electron.gsfTrack().get() : nullptr;
}

getTrack() [3/4]

template<>

const reco::Track * PATLeptonTimeLifeInfoProducer< pat::Muon >::getTrack ( const pat::Muon &  muon )

private

Definition at line 151 of file PATLeptonTimeLifeInfoProducer.cc.

                                                                                     {
  return muon.innerTrack().isNonnull() ? muon.innerTrack().get() : nullptr;
}

getTrack() [4/4]

template<>

const reco::Track * PATLeptonTimeLifeInfoProducer< pat::Tau >::getTrack ( const pat::Tau &  tau )

private

Definition at line 156 of file PATLeptonTimeLifeInfoProducer.cc.

References HLT_2024v14_cff::track.

                                                                                  {
  const reco::Track* track = nullptr;
  if (tau.leadChargedHadrCand().isNonnull())
    track = tau.leadChargedHadrCand()->bestTrack();
  return track;
}

produce()

template<typename T >

void PATLeptonTimeLifeInfoProducer< T >::produce ( edm::Event &  evt, const edm::EventSetup &  es  )

override

Definition at line 89 of file PATLeptonTimeLifeInfoProducer.cc.

References funct::abs(), PVValHelper::dz, trigObjTnPSource_cfi::filler, edm::Event::getByToken(), edm::EventSetup::getData(), getTrack(), info(), submitPVValidationJobs::infos, HLT_2024v14_cff::leptons, eostools::move(), edm::Event::put(), HltBtagValidation_cff::Vertex, AlignmentTracksFromVertexSelector_cfi::vertices, and L1BJetProducer_cff::vtx.

                                                                                     {
  // Get leptons
  edm::Handle<std::vector<T>> leptons;
  evt.getByToken(leptonsToken_, leptons);

  // Get the vertices
  edm::Handle<reco::VertexCollection> vertices;
  evt.getByToken(pvToken_, vertices);

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

  std::vector<TrackTimeLifeInfo> infos;
  infos.reserve(leptons->size());

  for (const auto& lepton : *leptons) {
    TrackTimeLifeInfo info;

    // Do nothing for lepton not passing selection
    if (!selector_(lepton)) {
      infos.push_back(info);
      continue;
    }
    size_t pv_idx = 0;
    if (pvChoice_ == useClosestInDz && getTrack(lepton) != nullptr) {
      float dz_min = 999;
      size_t vtx_idx = 0;
      for (const auto& vtx : *vertices) {
        float dz_tmp = std::abs(getTrack(lepton)->dz(vtx.position()));
        if (dz_tmp < dz_min) {
          dz_min = dz_tmp;
          pv_idx = vtx_idx;
        }
        vtx_idx++;
      }
    }
    const reco::Vertex& pv = !vertices->empty() ? (*vertices)[pv_idx] : reco::Vertex();

    // Obtain IP vector and set related info into lepton
    produceAndFillIPInfo(lepton, transTrackBuilder, pv, info);

    // Fit SV and set related info for taus or do nothing for other lepton types
    produceAndFillSVInfo(lepton, transTrackBuilder, pv, info);
    infos.push_back(info);
  }  // end of lepton loop

  // Build the valuemap
  auto infoMap = std::make_unique<TrackTimeLifeInfoMap>();
  TrackTimeLifeInfoMap::Filler filler(*infoMap);
  filler.insert(leptons, infos.begin(), infos.end());
  filler.fill();

  // Store output into the event
  evt.put(std::move(infoMap));
}

produceAndFillIPInfo()

template<typename T >

void PATLeptonTimeLifeInfoProducer< T >::produceAndFillIPInfo ( const T &  lepton, const TransientTrackBuilder &  transTrackBuilder, const reco::Vertex &  pv, TrackTimeLifeInfo &  info  )

private

Definition at line 164 of file PATLeptonTimeLifeInfoProducer.cc.

References TransientTrackBuilder::build(), RecoVertex::convertPos(), Vector3DBase< T, FrameTag >::dot(), Measurement1D::error(), TransientTrackBuilder::field(), reco::TransientTrack::field(), getTrack(), reco::TransientTrack::impactPointState(), info(), MagneticField::inInverseGeV(), HLT_2024v14_cff::track, Measurement1D::value(), PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

                                                                                     {
  const reco::Track* track = getTrack(lepton);
  if (track != nullptr) {
    // Extrapolate track to the point closest to PV
    reco::TransientTrack transTrack = transTrackBuilder.build(track);
    AnalyticalImpactPointExtrapolator extrapolator(transTrack.field());
    TrajectoryStateOnSurface closestState =
        extrapolator.extrapolate(transTrack.impactPointState(), RecoVertex::convertPos(pv.position()));
    if (!closestState.isValid()) {
      edm::LogError("PATLeptonTimeLifeInfoProducer")
          << "closestState not valid! From:\n"
          << "transTrack.impactPointState():\n"
          << transTrack.impactPointState() << "RecoVertex::convertPos(pv.position()):\n"
          << RecoVertex::convertPos(pv.position());
      return;
    }
    GlobalPoint pca = closestState.globalPosition();
    GlobalError pca_cov = closestState.cartesianError().position();
    GlobalVector ip_vec = GlobalVector(pca.x() - pv.x(), pca.y() - pv.y(), pca.z() - pv.z());
    GlobalError ip_cov = pca_cov + GlobalError(pv.covariance());
    VertexDistance3D pca_dist;
    Measurement1D ip_mes = pca_dist.distance(pv, VertexState(pca, pca_cov));
    if (ip_vec.dot(GlobalVector(lepton.px(), lepton.py(), lepton.pz())) < 0)
      ip_mes = Measurement1D(-1. * ip_mes.value(), ip_mes.error());

    // Store Track and PCA info
    info.setTrack(track);
    info.setBField_z(transTrackBuilder.field()->inInverseGeV(GlobalPoint(track->vx(), track->vy(), track->vz())).z());
    info.setPCA(pca, pca_cov);
    info.setIP(ip_vec, ip_cov);
    info.setIPLength(ip_mes);
  }
}

produceAndFillSVInfo() [1/2]

template<typename T >

void PATLeptonTimeLifeInfoProducer< T >::produceAndFillSVInfo ( const T &  lepton, const TransientTrackBuilder &  transTrackBuilder, const reco::Vertex &  pv, TrackTimeLifeInfo &  info  )

private

Definition at line 202 of file PATLeptonTimeLifeInfoProducer.cc.

                                                                                     {}

produceAndFillSVInfo() [2/2]

template<>

void PATLeptonTimeLifeInfoProducer< pat::Tau >::produceAndFillSVInfo ( const pat::Tau &  tau, const TransientTrackBuilder &  transTrackBuilder, const reco::Vertex &  pv, TrackTimeLifeInfo &  info  )

private

Definition at line 208 of file PATLeptonTimeLifeInfoProducer.cc.

References TransientTrackBuilder::build(), info(), TransientVertex::positionError(), pfDeepBoostedJetPreprocessParams_cfi::sv, and HLT_2024v14_cff::track.

                                                                                            {
  // Fit SV with tracks of charged tau decay products
  int fitOK = 0;
  if (tau.signalChargedHadrCands().size() + tau.signalLostTracks().size() > 1) {
    // Get tracks from tau signal charged candidates
    std::vector<reco::TransientTrack> transTrks;
    TransientVertex transVtx;
    for (const auto& cand : tau.signalChargedHadrCands()) {
      if (cand.isNull())
        continue;
      const reco::Track* track = cand->bestTrack();
      if (track != nullptr)
        transTrks.push_back(transTrackBuilder.build(track));
    }
    for (const auto& cand : tau.signalLostTracks()) {
      if (cand.isNull())
        continue;
      const reco::Track* track = cand->bestTrack();
      if (track != nullptr)
        transTrks.push_back(transTrackBuilder.build(track));
    }
    // Fit SV with KalmanVertexFitter
    fitOK = fitVertex(transTrks, transVtx) ? 1 : -1;
    if (fitOK > 0) {
      reco::Vertex sv = transVtx;
      // Get flight-length
      // Full PV->SV flight vector with its covariance
      GlobalVector flight_vec = GlobalVector(sv.x() - pv.x(), sv.y() - pv.y(), sv.z() - pv.z());
      GlobalError flight_cov = transVtx.positionError() + GlobalError(pv.covariance());
      //MB: can be taken from tau itself (but with different fit of PV and SV) as follows:
      //tau.flightLength().mag2());
      //tau.flightLengthSig();
      VertexDistance3D sv_dist;
      Measurement1D flightLength_mes = sv_dist.signedDistance(pv, sv, GlobalVector(tau.px(), tau.py(), tau.pz()));

      // Store SV info
      info.setSV(sv);
      info.setFlightVector(flight_vec, flight_cov);
      info.setFlightLength(flightLength_mes);
    }
  }
}

Member Data Documentation

leptonsToken_

template<typename T >

edm::EDGetTokenT<std::vector<T> > PATLeptonTimeLifeInfoProducer< T >::leptonsToken_

private

Definition at line 67 of file PATLeptonTimeLifeInfoProducer.cc.

pvChoice_

template<typename T >

int PATLeptonTimeLifeInfoProducer< T >::pvChoice_

private

Definition at line 71 of file PATLeptonTimeLifeInfoProducer.cc.

pvToken_

template<typename T >

edm::EDGetTokenT<reco::VertexCollection> PATLeptonTimeLifeInfoProducer< T >::pvToken_

private

Definition at line 68 of file PATLeptonTimeLifeInfoProducer.cc.

selector_

template<typename T >

const StringCutObjectSelector<T> PATLeptonTimeLifeInfoProducer< T >::selector_

private

Definition at line 70 of file PATLeptonTimeLifeInfoProducer.cc.

transTrackBuilderToken_

template<typename T >

edm::ESGetToken<TransientTrackBuilder, TransientTrackRecord> PATLeptonTimeLifeInfoProducer< T >::transTrackBuilderToken_

private

Definition at line 69 of file PATLeptonTimeLifeInfoProducer.cc.