TrackInfoBuilder.cc

Go to the documentation of this file.

#include "RecoBTag/FeatureTools/interface/TrackInfoBuilder.h"
#include "DataFormats/PatCandidates/interface/PackedCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/BTauReco/interface/JetTagInfo.h"
#include "DataFormats/BTauReco/interface/IPTagInfo.h"
#include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
#include "TrackingTools/IPTools/interface/IPTools.h"
#include "TrackingTools/Records/interface/TransientTrackRecord.h"
#include "RecoVertex/VertexTools/interface/VertexDistance3D.h"
#include "TVector3.h"

namespace btagbtvdeep {

  // adapted from DeepNtuples
  TrackInfoBuilder::TrackInfoBuilder(edm::ESHandle<TransientTrackBuilder> &build)
      : builder_(build),
        trackMomentum_(0),
        trackEta_(0),
        trackEtaRel_(0),
        trackPtRel_(0),
        trackPPar_(0),
        trackDeltaR_(0),
        trackPtRatio_(0),
        trackPParRatio_(0),
        trackSip2dVal_(0),
        trackSip2dSig_(0),
        trackSip3dVal_(0),
        trackSip3dSig_(0),
        trackJetDistVal_(0),
        trackJetDistSig_(0) {}

  void TrackInfoBuilder::buildTrackInfo(const reco::Candidate *candidate,
                                        const math::XYZVector &jetDir,
                                        GlobalVector refjetdirection,
                                        const reco::Vertex &pv) {
    TVector3 jetDir3(jetDir.x(), jetDir.y(), jetDir.z());

    // deal with PAT/AOD polymorphism to get track
    const reco::Track *track_ptr = nullptr;
    auto packed_candidate = dynamic_cast<const pat::PackedCandidate *>(candidate);
    auto pf_candidate = dynamic_cast<const reco::PFCandidate *>(candidate);
    if (pf_candidate) {
      track_ptr = pf_candidate->bestTrack();  // trackRef was sometimes null
    } else if (packed_candidate && packed_candidate->hasTrackDetails()) {
      // if PackedCandidate does not have TrackDetails this gives an Exception
      // because unpackCovariance might be called for pseudoTrack/bestTrack
      track_ptr = &(packed_candidate->pseudoTrack());
    }

    if (!track_ptr) {
      TVector3 trackMom3(candidate->momentum().x(), candidate->momentum().y(), candidate->momentum().z());
      trackMomentum_ = candidate->p();
      trackEta_ = candidate->eta();
      trackEtaRel_ = reco::btau::etaRel(jetDir, candidate->momentum());
      trackPtRel_ = trackMom3.Perp(jetDir3);
      trackPPar_ = jetDir.Dot(candidate->momentum());
      trackDeltaR_ = reco::deltaR(candidate->momentum(), jetDir);
      trackPtRatio_ = trackMom3.Perp(jetDir3) / candidate->p();
      trackPParRatio_ = jetDir.Dot(candidate->momentum()) / candidate->p();
      trackSip2dVal_ = 0.;
      trackSip2dSig_ = 0.;
      trackSip3dVal_ = 0.;
      trackSip3dSig_ = 0.;
      trackJetDistVal_ = 0.;
      trackJetDistSig_ = 0.;
      return;
    }

    math::XYZVector trackMom = track_ptr->momentum();
    double trackMag = std::sqrt(trackMom.Mag2());
    TVector3 trackMom3(trackMom.x(), trackMom.y(), trackMom.z());

    trackMomentum_ = std::sqrt(trackMom.Mag2());
    trackEta_ = trackMom.Eta();
    trackEtaRel_ = reco::btau::etaRel(jetDir, trackMom);
    trackPtRel_ = trackMom3.Perp(jetDir3);
    trackPPar_ = jetDir.Dot(trackMom);
    trackDeltaR_ = reco::deltaR(trackMom, jetDir);
    trackPtRatio_ = trackMom3.Perp(jetDir3) / trackMag;
    trackPParRatio_ = jetDir.Dot(trackMom) / trackMag;

    reco::TransientTrack transientTrack;
    transientTrack = builder_->build(*track_ptr);
    Measurement1D meas_ip2d = IPTools::signedTransverseImpactParameter(transientTrack, refjetdirection, pv).second;
    Measurement1D meas_ip3d = IPTools::signedImpactParameter3D(transientTrack, refjetdirection, pv).second;
    Measurement1D jetdist = IPTools::jetTrackDistance(transientTrack, refjetdirection, pv).second;
    trackSip2dVal_ = static_cast<float>(meas_ip2d.value());
    trackSip2dSig_ = static_cast<float>(meas_ip2d.significance());
    trackSip3dVal_ = static_cast<float>(meas_ip3d.value());
    trackSip3dSig_ = static_cast<float>(meas_ip3d.significance());
    trackJetDistVal_ = static_cast<float>(jetdist.value());
    trackJetDistSig_ = static_cast<float>(jetdist.significance());
  }

}  // namespace btagbtvdeep