#include <BTagTrackIpAccessor.h>

Collaboration diagram for BTagTrackIpAccessor:

Public Member Functions
	BTagTrackIpAccessor (const std::string &prefix)

void	augment (const xAOD::TrackParticle &track, const xAOD::Jet &jet) const

void	augment_with_grades (const xAOD::TrackParticle &track, const xAOD::Jet &jet) const

void	augment_with_ip (const xAOD::TrackParticle &track, const xAOD::Jet &jet) const

double	d0 (const xAOD::TrackParticle &track) const

double	d0Uncertainty (const xAOD::TrackParticle &track) const

double	z0SinTheta (const xAOD::TrackParticle &track) const

double	z0SinThetaUncertainty (const xAOD::TrackParticle &track) const

BTagSignedIP	getSignedIp (const xAOD::TrackParticle &track, const xAOD::Jet &jet) const

std::set< std::string >	getTrackIpDataDependencyNames () const

Private Types
typedef SG::AuxElement	AE

Private Attributes
AE::ConstAccessor< float >	m_ip_d0_sigma

AE::ConstAccessor< float >	m_ip_z0_sigma

AE::ConstAccessor< std::vector< float > >	m_track_displacement

AE::ConstAccessor< std::vector< float > >	m_track_momentum

AE::ConstAccessor< std::vector< ElementLink< xAOD::TrackParticleContainer > > >	m_ip2d_trackParticleLinks

AE::ConstAccessor< std::vector< ElementLink< xAOD::TrackParticleContainer > > >	m_ip3d_trackParticleLinks

AE::ConstAccessor< std::vector< int > >	m_ip2d_gradeOfTracks

AE::ConstAccessor< std::vector< int > >	m_ip3d_gradeOfTracks

AE::Decorator< float >	m_ip2d_signed_d0

AE::Decorator< float >	m_ip3d_signed_d0

AE::Decorator< float >	m_ip3d_signed_z0

AE::Decorator< float >	m_ip3d_signed_d0_significance

AE::Decorator< float >	m_ip3d_signed_z0_significance

AE::Decorator< int >	m_ip2d_grade

AE::Decorator< int >	m_ip3d_grade

std::string	m_prefix

Detailed Description

Definition at line 29 of file BTagTrackIpAccessor.h.

Member Typedef Documentation

◆ AE

typedef SG::AuxElement BTagTrackIpAccessor::AE

private

Definition at line 58 of file BTagTrackIpAccessor.h.

Constructor & Destructor Documentation

◆ BTagTrackIpAccessor()

BTagTrackIpAccessor::BTagTrackIpAccessor ( const std::string & prefix )

Definition at line 18 of file BTagTrackIpAccessor.cxx.

                                                                :
   m_ip_d0_sigma(prefix + str::d0Uncertainty),
   m_ip_z0_sigma(prefix + str::z0SinThetaUncertainty),
   m_track_displacement(prefix + str::trackDisplacement),
   m_track_momentum(prefix + str::trackMomentum),
   m_ip2d_trackParticleLinks("IP2D_TrackParticleLinks"),
   m_ip3d_trackParticleLinks("IP3D_TrackParticleLinks"),
   m_ip2d_gradeOfTracks("IP2D_gradeOfTracks"),
   m_ip3d_gradeOfTracks("IP3D_gradeOfTracks"),
   m_ip2d_signed_d0("IP2D_signed_d0"),
   m_ip3d_signed_d0("IP3D_signed_d0"),
   m_ip3d_signed_z0("IP3D_signed_z0"),
   m_ip3d_signed_d0_significance("IP3D_signed_d0_significance"),
   m_ip3d_signed_z0_significance("IP3D_signed_z0_significance"),
   m_ip2d_grade("IP2D_grade"),
   m_ip3d_grade("IP3D_grade"),
   m_prefix(prefix)
 {
 }

Member Function Documentation

◆ augment()

void BTagTrackIpAccessor::augment	(	const xAOD::TrackParticle &	track,
		const xAOD::Jet &	jet
	)		const

Definition at line 49 of file BTagTrackIpAccessor.cxx.

                                                                                           {
   augment_with_grades(track, jet);
   augment_with_ip(track, jet);
 }

◆ augment_with_grades()

void BTagTrackIpAccessor::augment_with_grades	(	const xAOD::TrackParticle &	track,
		const xAOD::Jet &	jet
	)		const

Definition at line 105 of file BTagTrackIpAccessor.cxx.

                                                                                                       {
   int ip3d_grade = -1;
   const xAOD::BTagging &btagging = *xAOD::BTaggingUtilities::getBTagging( jet );
   const std::vector<ElementLink<xAOD::TrackParticleContainer> > ip3d_tracks = m_ip3d_trackParticleLinks(btagging);
   for (std::size_t ip3d_track_index = 0; ip3d_track_index < ip3d_tracks.size(); ++ip3d_track_index) {
     if (&track == *(ip3d_tracks.at(ip3d_track_index))) {
       ip3d_grade = m_ip3d_gradeOfTracks(btagging).at(ip3d_track_index);
       break;
     }
   }
   m_ip3d_grade(track) = ip3d_grade;
  
   int ip2d_grade = -1;
   const std::vector<ElementLink<xAOD::TrackParticleContainer> > ip2d_tracks = m_ip2d_trackParticleLinks(btagging);
   for (std::size_t ip2d_track_index = 0; ip2d_track_index < ip2d_tracks.size(); ++ip2d_track_index) {
     if (&track == *(ip2d_tracks.at(ip2d_track_index))) {
       ip2d_grade = m_ip2d_gradeOfTracks(btagging).at(ip2d_track_index);
       break;
     }
   }
   m_ip2d_grade(track) = ip2d_grade;
 }

◆ augment_with_ip()

void BTagTrackIpAccessor::augment_with_ip	(	const xAOD::TrackParticle &	track,
		const xAOD::Jet &	jet
	)		const

Definition at line 97 of file BTagTrackIpAccessor.cxx.

                                                                                                   {
   BTagSignedIP ip = getSignedIp(track, jet);
   m_ip2d_signed_d0(track) = ip.ip2d_signed_d0;
   m_ip3d_signed_d0(track) = ip.ip3d_signed_d0;
   m_ip3d_signed_d0_significance(track) = ip.ip3d_signed_d0_significance;
   m_ip3d_signed_z0(track) = ip.ip3d_signed_z0;
   m_ip3d_signed_z0_significance(track) = ip.ip3d_signed_z0_significance;
 }

◆ d0()

double BTagTrackIpAccessor::d0 ( const xAOD::TrackParticle & track ) const

Definition at line 76 of file BTagTrackIpAccessor.cxx.

                                                                    {
   const Amg::Vector3D track_displacement = get_vector3d(m_track_displacement(track));
   const Amg::Vector3D track_momentum = get_vector3d(m_track_momentum(track));
   return std::copysign(
     std::hypot(track_displacement[Amg::x], track_displacement[Amg::y]),
     track_momentum.cross(track_displacement)[Amg::z]);
 }

◆ d0Uncertainty()

double BTagTrackIpAccessor::d0Uncertainty ( const xAOD::TrackParticle & track ) const

Definition at line 83 of file BTagTrackIpAccessor.cxx.

         {
   return m_ip_d0_sigma(track);
 }

◆ getSignedIp()

BTagSignedIP BTagTrackIpAccessor::getSignedIp	(	const xAOD::TrackParticle &	track,
		const xAOD::Jet &	jet
	)		const

Definition at line 54 of file BTagTrackIpAccessor.cxx.

                                                                                                       {
   const TLorentzVector jet_fourVector = jet.p4();
   const Amg::Vector3D jet_threeVector(jet_fourVector.X(),jet_fourVector.Y(),jet_fourVector.Z());
   const Amg::Vector3D track_displacement = get_vector3d(m_track_displacement(track));
   const Amg::Vector3D track_momentum = get_vector3d(m_track_momentum(track));
  
   BTagSignedIP ip;
   const double ip_d0 = d0(track);
   ip.ip2d_signed_d0 = std::copysign(ip_d0, std::sin(jet_threeVector.phi() - track_momentum.phi()) * ip_d0);
   const double ip3d_signed_d0 = std::copysign(ip_d0, jet_threeVector.cross(track_momentum).dot(track_momentum.cross(-track_displacement)));
   ip.ip3d_signed_d0 = ip3d_signed_d0;
   ip.ip3d_signed_d0_significance = ip3d_signed_d0 / m_ip_d0_sigma(track);
  
   const double ip_z0 = z0SinTheta(track);
   const double signed_z0 = std::copysign(ip_z0, (jet_threeVector.eta() - track_momentum.eta()) * ip_z0);
   ip.ip3d_signed_z0 = signed_z0;
   ip.ip3d_signed_z0_significance = signed_z0 / m_ip_z0_sigma(track);
   ip.ip2d_grade = 0;
   ip.ip3d_grade = 0;
   return ip;
 }

◆ getTrackIpDataDependencyNames()

std::set< std::string > BTagTrackIpAccessor::getTrackIpDataDependencyNames ( ) const

Definition at line 128 of file BTagTrackIpAccessor.cxx.

 {
   return {
     m_prefix + str::d0Uncertainty,
     m_prefix + str::z0SinThetaUncertainty,
     m_prefix + str::trackDisplacement,
     m_prefix + str::trackMomentum};
 }

◆ z0SinTheta()

double BTagTrackIpAccessor::z0SinTheta ( const xAOD::TrackParticle & track ) const

Definition at line 87 of file BTagTrackIpAccessor.cxx.

                                                                            {
   const Amg::Vector3D track_displacement = get_vector3d(m_track_displacement(track));
   const Amg::Vector3D track_momentum = get_vector3d(m_track_momentum(track));
   return track_displacement[Amg::z] * std::sqrt(
     1 - std::pow(track_momentum[Amg::z],2) / track_momentum.squaredNorm());
 }

◆ z0SinThetaUncertainty()

double BTagTrackIpAccessor::z0SinThetaUncertainty ( const xAOD::TrackParticle & track ) const

Definition at line 93 of file BTagTrackIpAccessor.cxx.

                                                                                       {
   return m_ip_z0_sigma(track);
 }

Member Data Documentation

◆ m_ip2d_grade

AE::Decorator<int> BTagTrackIpAccessor::m_ip2d_grade

private

Definition at line 73 of file BTagTrackIpAccessor.h.

◆ m_ip2d_gradeOfTracks

AE::ConstAccessor<std::vector<int> > BTagTrackIpAccessor::m_ip2d_gradeOfTracks

private

Definition at line 66 of file BTagTrackIpAccessor.h.

◆ m_ip2d_signed_d0

AE::Decorator<float> BTagTrackIpAccessor::m_ip2d_signed_d0

private

Definition at line 68 of file BTagTrackIpAccessor.h.

◆ m_ip2d_trackParticleLinks

AE::ConstAccessor<std::vector<ElementLink<xAOD::TrackParticleContainer> > > BTagTrackIpAccessor::m_ip2d_trackParticleLinks

private

Definition at line 64 of file BTagTrackIpAccessor.h.

◆ m_ip3d_grade

AE::Decorator<int> BTagTrackIpAccessor::m_ip3d_grade

private

Definition at line 74 of file BTagTrackIpAccessor.h.

◆ m_ip3d_gradeOfTracks

AE::ConstAccessor<std::vector<int> > BTagTrackIpAccessor::m_ip3d_gradeOfTracks

private

Definition at line 67 of file BTagTrackIpAccessor.h.

◆ m_ip3d_signed_d0

AE::Decorator<float> BTagTrackIpAccessor::m_ip3d_signed_d0

private

Definition at line 69 of file BTagTrackIpAccessor.h.

◆ m_ip3d_signed_d0_significance

AE::Decorator<float> BTagTrackIpAccessor::m_ip3d_signed_d0_significance

private

Definition at line 71 of file BTagTrackIpAccessor.h.

◆ m_ip3d_signed_z0

AE::Decorator<float> BTagTrackIpAccessor::m_ip3d_signed_z0

private

Definition at line 70 of file BTagTrackIpAccessor.h.

◆ m_ip3d_signed_z0_significance

AE::Decorator<float> BTagTrackIpAccessor::m_ip3d_signed_z0_significance

private

Definition at line 72 of file BTagTrackIpAccessor.h.

◆ m_ip3d_trackParticleLinks

AE::ConstAccessor<std::vector<ElementLink<xAOD::TrackParticleContainer> > > BTagTrackIpAccessor::m_ip3d_trackParticleLinks

private

Definition at line 65 of file BTagTrackIpAccessor.h.

◆ m_ip_d0_sigma

AE::ConstAccessor<float> BTagTrackIpAccessor::m_ip_d0_sigma

private

Definition at line 60 of file BTagTrackIpAccessor.h.

◆ m_ip_z0_sigma

AE::ConstAccessor<float> BTagTrackIpAccessor::m_ip_z0_sigma

private

Definition at line 61 of file BTagTrackIpAccessor.h.

◆ m_prefix

std::string BTagTrackIpAccessor::m_prefix

private

Definition at line 76 of file BTagTrackIpAccessor.h.

◆ m_track_displacement

AE::ConstAccessor<std::vector<float> > BTagTrackIpAccessor::m_track_displacement

private

Definition at line 62 of file BTagTrackIpAccessor.h.

◆ m_track_momentum

AE::ConstAccessor<std::vector<float> > BTagTrackIpAccessor::m_track_momentum

private

Definition at line 63 of file BTagTrackIpAccessor.h.

The documentation for this class was generated from the following files:

Public Member Functions

Private Types

Private Attributes

Detailed Description

Member Typedef Documentation

◆ AE

Constructor & Destructor Documentation

◆ BTagTrackIpAccessor()

Member Function Documentation

◆ augment()

◆ augment_with_grades()

◆ augment_with_ip()

◆ d0()

◆ d0Uncertainty()

◆ getSignedIp()

◆ getTrackIpDataDependencyNames()

◆ z0SinTheta()

◆ z0SinThetaUncertainty()

Member Data Documentation

◆ m_ip2d_grade

◆ m_ip2d_gradeOfTracks

◆ m_ip2d_signed_d0

◆ m_ip2d_trackParticleLinks

◆ m_ip3d_grade

◆ m_ip3d_gradeOfTracks

◆ m_ip3d_signed_d0

◆ m_ip3d_signed_d0_significance

◆ m_ip3d_signed_z0

◆ m_ip3d_signed_z0_significance

◆ m_ip3d_trackParticleLinks

◆ m_ip_d0_sigma

◆ m_ip_z0_sigma

◆ m_prefix

◆ m_track_displacement

◆ m_track_momentum