#include <BTagTrackIpAccessor.h>

Collaboration diagram for BTagTrackIpAccessor:

Public Member Functions
	BTagTrackIpAccessor (const std::string &prefix)
void	augment (const xAOD::TrackParticle &track, const xAOD::IParticle &jet) const
void	augment_with_grades (const xAOD::TrackParticle &track, const xAOD::IParticle &jet) const
void	augment_with_ip (const xAOD::TrackParticle &track, const xAOD::IParticle &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::IParticle &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_sin_theta
AE::Decorator< float >	m_ip3d_signed_d0_significance
AE::Decorator< float >	m_ip3d_signed_z0_sin_theta_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_sin_theta("IP3D_signed_z0"),
  m_ip3d_signed_d0_significance("IP3D_signed_d0_significance"),
  m_ip3d_signed_z0_sin_theta_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::IParticle &	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::IParticle &	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::IParticle &	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_sin_theta(track) = ip.ip3d_signed_z0_sin_theta;
  m_ip3d_signed_z0_sin_theta_significance(track) = ip.ip3d_signed_z0_sin_theta_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::IParticle &	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_sin_theta = std::copysign(ip_z0, (jet_threeVector.eta() - track_momentum.eta()) * ip_z0);
  ip.ip3d_signed_z0_sin_theta = signed_z0_sin_theta;
  ip.ip3d_signed_z0_sin_theta_significance = signed_z0_sin_theta / 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_sin_theta

AE::Decorator<float> BTagTrackIpAccessor::m_ip3d_signed_z0_sin_theta

private

Definition at line 70 of file BTagTrackIpAccessor.h.

◆ m_ip3d_signed_z0_sin_theta_significance

AE::Decorator<float> BTagTrackIpAccessor::m_ip3d_signed_z0_sin_theta_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_sin_theta

◆ m_ip3d_signed_z0_sin_theta_significance

◆ m_ip3d_trackParticleLinks

◆ m_ip_d0_sigma

◆ m_ip_z0_sigma

◆ m_prefix

◆ m_track_displacement

◆ m_track_momentum