btagbtvdeep::TrackInfoBuilder Class Reference

#include <TrackInfoBuilder.h>

Public Member Functions
void	buildTrackInfo (const reco::Candidate *candidate, const math::XYZVector &jetDir, GlobalVector refjetdirection, const reco::Vertex &pv)
const float	getTrackDeltaR () const
const float	getTrackEta () const
const float	getTrackEtaRel () const
const float	getTrackJetDistSig () const
const float	getTrackJetDistVal () const
const float	getTrackMomentum () const
const float	getTrackPPar () const
const float	getTrackPParRatio () const
const float	getTrackPtRatio () const
const float	getTrackPtRel () const
const float	getTrackSip2dSig () const
const float	getTrackSip2dVal () const
const float	getTrackSip3dSig () const
const float	getTrackSip3dVal () const
	TrackInfoBuilder (edm::ESHandle< TransientTrackBuilder > &build)

Private Attributes
edm::ESHandle< TransientTrackBuilder >	builder_
float	trackDeltaR_
float	trackEta_
float	trackEtaRel_
float	trackJetDistSig_
float	trackJetDistVal_
float	trackMomentum_
float	trackPPar_
float	trackPParRatio_
float	trackPtRatio_
float	trackPtRel_
float	trackSip2dSig_
float	trackSip2dVal_
float	trackSip3dSig_
float	trackSip3dVal_

Detailed Description

Definition at line 12 of file TrackInfoBuilder.h.

Constructor & Destructor Documentation

btagbtvdeep::TrackInfoBuilder::TrackInfoBuilder

(

edm::ESHandle< TransientTrackBuilder > &

build

)

Definition at line 15 of file TrackInfoBuilder.cc.

                                                                                 :
        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)
{

}

Member Function Documentation

void btagbtvdeep::TrackInfoBuilder::buildTrackInfo	(	const reco::Candidate *	candidate,
		const math::XYZVector &	jetDir,
		GlobalVector	refjetdirection,
		const reco::Vertex &	pv
	)

Definition at line 36 of file TrackInfoBuilder.cc.

References reco::PFCandidate::bestTrack(), TransientTrackBuilder::build(), builder_, reco::deltaR(), reco::Candidate::eta(), reco::btau::etaRel(), IPTools::jetTrackDistance(), reco::Candidate::momentum(), reco::Candidate::p(), IPTools::signedImpactParameter3D(), IPTools::signedTransverseImpactParameter(), Measurement1D::significance(), mathSSE::sqrt(), trackDeltaR_, trackEta_, trackEtaRel_, trackJetDistSig_, trackJetDistVal_, trackMomentum_, trackPPar_, trackPParRatio_, trackPtRatio_, trackPtRel_, trackSip2dSig_, trackSip2dVal_, trackSip3dSig_, trackSip3dVal_, and Measurement1D::value().

Referenced by DeepBoostedJetTagInfoProducer::fillParticleFeatures().

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

    }

const float btagbtvdeep::TrackInfoBuilder::getTrackDeltaR ( ) const

inline

Definition at line 17 of file TrackInfoBuilder.h.

References trackDeltaR_.

Referenced by btagbtvdeep::commonCandidateToFeatures().

{return trackDeltaR_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackEta ( ) const

inline

Definition at line 18 of file TrackInfoBuilder.h.

References trackEta_.

{return trackEta_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackEtaRel ( ) const

inline

Definition at line 19 of file TrackInfoBuilder.h.

References trackEtaRel_.

Referenced by btagbtvdeep::commonCandidateToFeatures(), and DeepBoostedJetTagInfoProducer::fillParticleFeatures().

{return trackEtaRel_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackJetDistSig ( ) const

inline

Definition at line 20 of file TrackInfoBuilder.h.

References trackJetDistSig_.

{return trackJetDistSig_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackJetDistVal ( ) const

inline

Definition at line 21 of file TrackInfoBuilder.h.

References trackJetDistVal_.

Referenced by btagbtvdeep::commonCandidateToFeatures(), and DeepBoostedJetTagInfoProducer::fillParticleFeatures().

{return trackJetDistVal_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackMomentum ( ) const

inline

Definition at line 22 of file TrackInfoBuilder.h.

References trackMomentum_.

{return trackMomentum_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackPPar ( ) const

inline

Definition at line 23 of file TrackInfoBuilder.h.

References trackPPar_.

Referenced by btagbtvdeep::commonCandidateToFeatures().

{return trackPPar_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackPParRatio ( ) const

inline

Definition at line 24 of file TrackInfoBuilder.h.

References trackPParRatio_.

Referenced by btagbtvdeep::commonCandidateToFeatures(), and DeepBoostedJetTagInfoProducer::fillParticleFeatures().

{return trackPParRatio_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackPtRatio ( ) const

inline

Definition at line 25 of file TrackInfoBuilder.h.

References trackPtRatio_.

Referenced by btagbtvdeep::commonCandidateToFeatures(), and DeepBoostedJetTagInfoProducer::fillParticleFeatures().

{return trackPtRatio_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackPtRel ( ) const

inline

Definition at line 26 of file TrackInfoBuilder.h.

References trackPtRel_.

Referenced by btagbtvdeep::commonCandidateToFeatures().

{return trackPtRel_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackSip2dSig ( ) const

inline

Definition at line 27 of file TrackInfoBuilder.h.

References trackSip2dSig_.

Referenced by btagbtvdeep::commonCandidateToFeatures(), and DeepBoostedJetTagInfoProducer::fillParticleFeatures().

{return trackSip2dSig_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackSip2dVal ( ) const

inline

Definition at line 28 of file TrackInfoBuilder.h.

References trackSip2dVal_.

Referenced by btagbtvdeep::commonCandidateToFeatures(), and DeepBoostedJetTagInfoProducer::fillParticleFeatures().

{return trackSip2dVal_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackSip3dSig ( ) const

inline

Definition at line 29 of file TrackInfoBuilder.h.

References trackSip3dSig_.

Referenced by btagbtvdeep::commonCandidateToFeatures(), and DeepBoostedJetTagInfoProducer::fillParticleFeatures().

{return trackSip3dSig_;}

const float btagbtvdeep::TrackInfoBuilder::getTrackSip3dVal ( ) const

inline

Definition at line 30 of file TrackInfoBuilder.h.

References trackSip3dVal_.

Referenced by btagbtvdeep::commonCandidateToFeatures(), and DeepBoostedJetTagInfoProducer::fillParticleFeatures().

{return trackSip3dVal_;}

Member Data Documentation

edm::ESHandle<TransientTrackBuilder> btagbtvdeep::TrackInfoBuilder::builder_

private

Definition at line 34 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo().

float btagbtvdeep::TrackInfoBuilder::trackDeltaR_

private

Definition at line 41 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackDeltaR().

float btagbtvdeep::TrackInfoBuilder::trackEta_

private

Definition at line 37 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackEta().

float btagbtvdeep::TrackInfoBuilder::trackEtaRel_

private

Definition at line 38 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackEtaRel().

float btagbtvdeep::TrackInfoBuilder::trackJetDistSig_

private

Definition at line 50 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackJetDistSig().

float btagbtvdeep::TrackInfoBuilder::trackJetDistVal_

private

Definition at line 49 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackJetDistVal().

float btagbtvdeep::TrackInfoBuilder::trackMomentum_

private

Definition at line 36 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackMomentum().

float btagbtvdeep::TrackInfoBuilder::trackPPar_

private

Definition at line 40 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackPPar().

float btagbtvdeep::TrackInfoBuilder::trackPParRatio_

private

Definition at line 43 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackPParRatio().

float btagbtvdeep::TrackInfoBuilder::trackPtRatio_

private

Definition at line 42 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackPtRatio().

float btagbtvdeep::TrackInfoBuilder::trackPtRel_

private

Definition at line 39 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackPtRel().

float btagbtvdeep::TrackInfoBuilder::trackSip2dSig_

private

Definition at line 45 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackSip2dSig().

float btagbtvdeep::TrackInfoBuilder::trackSip2dVal_

private

Definition at line 44 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackSip2dVal().

float btagbtvdeep::TrackInfoBuilder::trackSip3dSig_

private

Definition at line 47 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackSip3dSig().

float btagbtvdeep::TrackInfoBuilder::trackSip3dVal_

private

Definition at line 46 of file TrackInfoBuilder.h.

Referenced by buildTrackInfo(), and getTrackSip3dVal().