DerivationFramework::TrackParametersKVU Class Reference

#include <TrackParametersKVU.h>

Inheritance diagram for DerivationFramework::TrackParametersKVU:

Collaboration diagram for DerivationFramework::TrackParametersKVU:

Public Member Functions
	TrackParametersKVU (const std::string &t, const std::string &n, const IInterface *p)
	Constructor with parameters. More...
virtual	~TrackParametersKVU ()=default
	Destructor. More...
virtual StatusCode	initialize () override
virtual StatusCode	addBranches (const EventContext &ctx) const override
	Check that the current event passes this filter. More...

Private Attributes
SG::ReadHandleKey< xAOD::TrackParticleContainer >	m_trackContainerKey
SG::ReadHandleKey< xAOD::VertexContainer >	m_vertexContainerKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer >	m_KVUphiKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer >	m_KVUthetaKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer >	m_KVUd0Key
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer >	m_KVUz0Key
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer >	m_KVUqOverPKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer >	m_KVUChi2Key
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer >	m_KVUusedPVKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer >	m_KVUCovMatKey
ToolHandle< Trk::IVertexTrackUpdator >	m_vertexTrackUpdator
ToolHandle< Trk::IExtrapolator >	m_extrapolator
ToolHandle< Trk::IVertexLinearizedTrackFactory >	m_LinearizedTrackFactory
ToolHandle< Trk::ITrackToVertexIPEstimator >	m_IPEstimator
StringProperty	m_sgName {this, "DecorationPrefix", "", "decoration prefix"}

Detailed Description

the code used in this implementation is kindly stolen from: atlasoff:: ISF/ISF_Core/ISF_Tools

Author

James Catmore -at- cern.ch

Definition at line 57 of file TrackParametersKVU.h.

Constructor & Destructor Documentation

TrackParametersKVU()

DerivationFramework::TrackParametersKVU::TrackParametersKVU	(	const std::string &	t,
		const std::string &	n,
		const IInterface *	p
	)

Constructor with parameters.

Definition at line 16 of file TrackParameterKVU.cxx.

                                                       :
  base_class(t, n, p){}

~TrackParametersKVU()

virtual DerivationFramework::TrackParametersKVU::~TrackParametersKVU ( )

virtualdefault

Destructor.

Member Function Documentation

addBranches()

StatusCode DerivationFramework::TrackParametersKVU::addBranches ( const EventContext &  ctx ) const

overridevirtual

Check that the current event passes this filter.

Definition at line 62 of file TrackParameterKVU.cxx.

{
  // --- Get the tracks
  //const xAOD::TrackParticleContainer* tracks = nullptr;
  SG::ReadHandle<xAOD::TrackParticleContainer> tracks(m_trackContainerKey,ctx);
  ATH_CHECK( tracks.isValid() );
 
  //-- for each track, update track params with vtx considered as extra measurement (choose the closest vtx)
  if(tracks->size() !=0) {
    SG::WriteDecorHandle<xAOD::TrackParticleContainer, float> decoratorKVUphi(m_KVUphiKey, ctx);
    SG::WriteDecorHandle<xAOD::TrackParticleContainer, float> decoratorKVUtheta(m_KVUthetaKey, ctx);
    SG::WriteDecorHandle<xAOD::TrackParticleContainer, float> decoratorKVUd0(m_KVUd0Key, ctx);
    SG::WriteDecorHandle<xAOD::TrackParticleContainer, float> decoratorKVUz0(m_KVUz0Key, ctx);
    SG::WriteDecorHandle<xAOD::TrackParticleContainer, float> decoratorKVUqOverP(m_KVUqOverPKey, ctx);
    SG::WriteDecorHandle<xAOD::TrackParticleContainer, float> decoratorKVUChi2(m_KVUChi2Key, ctx);
    SG::WriteDecorHandle<xAOD::TrackParticleContainer, float> decoratorKVUusedPV(m_KVUusedPVKey, ctx);
    SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> decoratorKVUCovMat(m_KVUCovMatKey, ctx);
 
    for (const auto track : *tracks) {
      if(track){
    const Trk::TrackParameters *tPerigee = &(track->perigeeParameters());
    // --- list of new variables that will decorate the track
    std::unique_ptr<AmgSymMatrix(5)> updateTrackCov = nullptr;
    float updatephi        = -999;
    float updatetheta      = -999;
    float updated0         = -999;
    float updatez0         = -999;
    float updateqOverP     = -999;
    float updateChi2       = -999; 
        bool usedPrimaryVertex = false;
    std::vector<float> vec;
 
    std::unique_ptr<const Trk::TrackParameters> trackParams = nullptr;
    float minIP = 1000.;
    //--- retrieve closest vertex to track
    const xAOD::Vertex* closestVertex = nullptr;
    int nVtx = 0;
    SG::ReadHandle<xAOD::VertexContainer> vxContainer(m_vertexContainerKey,ctx);
    if (vxContainer.isValid()) {
      for (const xAOD::Vertex* vtx: *vxContainer) {
        if ( (vtx->vertexType() == xAOD::VxType::PriVtx) ||
         (vtx->vertexType() == xAOD::VxType::PileUp) ) {
          Amg::Vector3D  vtxPos(vtx->position());
          auto vtxSurface = std::make_unique<Trk::PerigeeSurface>(vtxPos);
          trackParams = m_extrapolator->extrapolate(ctx,*tPerigee,*vtxSurface);
          std::unique_ptr<const Trk::ImpactParametersAndSigma> iPandSigma = nullptr;
          iPandSigma = m_IPEstimator->estimate(trackParams.get(), vtx);
          if(sqrt(iPandSigma->IPd0*iPandSigma->IPd0+iPandSigma->IPz0*iPandSigma->IPz0) < minIP){
        minIP = sqrt(iPandSigma->IPd0*iPandSigma->IPd0+iPandSigma->IPz0*iPandSigma->IPz0);
        closestVertex = vtx;
          }
          ATH_MSG_VERBOSE("n vtx pos(x, y, z) IPd0 IPz0 : " << nVtx << " " << vtx->position().x() << " "<<  vtx->position().y() << " "<<  vtx->position().z() << " " << iPandSigma->IPd0 << " " << iPandSigma->IPz0);
          ATH_MSG_VERBOSE("                   d0,Delta_z0-TrackParticle:" << track->d0() <<", "<< track->z0() - vtx->z() + track->vz());
          nVtx++;
        }
      }
    }else{
      ATH_MSG_ERROR ("Couldn't retrieve vxContainer  with key: " << m_vertexContainerKey.key() );
      return StatusCode::FAILURE;
    } // --- end retrieve closest vertex to track
 
    
    // update the track params with vtx info after linearization of track around it
    if(closestVertex){
      ATH_MSG_VERBOSE("Vertex selected position (x y z): " << closestVertex->position().x()<<" "<<closestVertex->position().y()<<" "<<closestVertex->position().z());
      ATH_MSG_VERBOSE("                type: " << closestVertex->vertexType());
      ATH_MSG_VERBOSE("                nParticles: " << closestVertex->nTrackParticles());  
      ATH_MSG_VERBOSE("                quality chi2: " << closestVertex->chiSquared() << " " << closestVertex->numberDoF());
      Amg::Vector3D  globPos(closestVertex->position());
      auto recVtx = std::make_unique<xAOD::Vertex>();
      recVtx->makePrivateStore(*closestVertex);
      auto surface = std::make_unique<const Trk::PerigeeSurface>(globPos);
      trackParams = m_extrapolator->extrapolate(ctx,*tPerigee,*surface);
      auto linearTrack = std::make_unique<Trk::VxTrackAtVertex>(0., nullptr, nullptr, trackParams.get(), nullptr);
      if(linearTrack){
        ATH_MSG_VERBOSE("Linearizing track");
        m_LinearizedTrackFactory->linearize(*linearTrack,globPos);
        ATH_MSG_VERBOSE("Updating linearized track parameters after vertex fit. Track weight = " << linearTrack->weight());
        m_vertexTrackUpdator->update((*linearTrack), (*recVtx));
        ATH_MSG_VERBOSE ("track info after vertex track updator !"<<*linearTrack );
        if(linearTrack->perigeeAtVertex()) {
          //retrieve & store updated track param qOverP,d0, z0,theta,phi after KVU
          updateqOverP = linearTrack->perigeeAtVertex()->parameters()[Trk::qOverP];
          updated0 = linearTrack->perigeeAtVertex()->parameters()[Trk::d0];
          updatez0 = linearTrack->perigeeAtVertex()->parameters()[Trk::z0];
          updatephi = linearTrack->perigeeAtVertex()->parameters()[Trk::phi0];
          updatetheta = linearTrack->perigeeAtVertex()->parameters()[Trk::theta];
          //retrieve & store updated track cov matrix plus save the KVU Chi2 
          updateTrackCov = std::make_unique<AmgSymMatrix(5)>(*linearTrack->perigeeAtVertex()->covariance());
          updateChi2 = linearTrack->trackQuality().chiSquared();
          // store whether this used the PV or not
          usedPrimaryVertex = closestVertex->vertexType()==xAOD::VxType::PriVtx;
        }
      }
    }// --- end if closest vertex
    
    //decorate tracks with new updated track parameters:
    decoratorKVUqOverP(*track) = updateqOverP;
    decoratorKVUd0(*track)     = updated0;
    decoratorKVUz0(*track)     = updatez0;
    decoratorKVUphi(*track)    = updatephi;
    decoratorKVUtheta(*track)  = updatetheta;
    decoratorKVUChi2(*track)   = updateChi2;
        decoratorKVUusedPV(*track) = usedPrimaryVertex;
    if (updateTrackCov){
      Amg::compress(*updateTrackCov, vec);
    }else{
      vec.assign(5, 0.0);
    }
    decoratorKVUCovMat(*track) =  vec;
    ATH_MSG_VERBOSE("track updated.");
      } // --- end if(track)
    } // --- end loop tracks
    ATH_MSG_VERBOSE("All tracks updated.");
  }
  
  return StatusCode::SUCCESS;
}

initialize()

StatusCode DerivationFramework::TrackParametersKVU::initialize ( )

overridevirtual

Definition at line 22 of file TrackParameterKVU.cxx.

{
 
  if (m_trackContainerKey.empty() || m_vertexContainerKey.empty()) {
    ATH_MSG_ERROR("No selection variables for the TrackParametersKVU tool!");
    return StatusCode::FAILURE;
  }
 
  ATH_CHECK(m_trackContainerKey.initialize());
  ATH_CHECK(m_vertexContainerKey.initialize());
 
  m_KVUphiKey = m_trackContainerKey.key() + m_KVUphiKey.key();
  m_KVUthetaKey = m_trackContainerKey.key() + m_KVUthetaKey.key();
  m_KVUd0Key = m_trackContainerKey.key() + m_KVUd0Key.key();
  m_KVUz0Key = m_trackContainerKey.key() + m_KVUz0Key.key();
  m_KVUqOverPKey = m_trackContainerKey.key() + m_KVUqOverPKey.key();
  m_KVUChi2Key = m_trackContainerKey.key() + m_KVUChi2Key.key();
  m_KVUusedPVKey = m_trackContainerKey.key() + m_KVUusedPVKey.key();
  m_KVUCovMatKey = m_trackContainerKey.key() + m_KVUCovMatKey.key();
 
  ATH_CHECK(m_KVUphiKey.initialize());
  ATH_CHECK(m_KVUthetaKey.initialize());
  ATH_CHECK(m_KVUd0Key.initialize());
  ATH_CHECK(m_KVUz0Key.initialize());
  ATH_CHECK(m_KVUqOverPKey.initialize());
  ATH_CHECK(m_KVUChi2Key.initialize());
  ATH_CHECK(m_KVUusedPVKey.initialize());
  ATH_CHECK(m_KVUCovMatKey.initialize());
 
  ATH_CHECK(m_vertexTrackUpdator.retrieve());
  ATH_CHECK(m_extrapolator.retrieve());
  ATH_CHECK(m_LinearizedTrackFactory.retrieve());
  ATH_CHECK(m_IPEstimator.retrieve());
 
  ATH_MSG_DEBUG("initialize() ...");
  ATH_MSG_DEBUG("Successfully retrieved the TrackParametersKVU tool" );
  return StatusCode::SUCCESS;
}

Member Data Documentation

m_extrapolator

ToolHandle< Trk::IExtrapolator > DerivationFramework::TrackParametersKVU::m_extrapolator

private

Initial value:

{
      this, "TrackExtrapolator", "Trk::Extrapolator/AtlasExtrapolator"}

Definition at line 96 of file TrackParametersKVU.h.

m_IPEstimator

ToolHandle< Trk::ITrackToVertexIPEstimator> DerivationFramework::TrackParametersKVU::m_IPEstimator

private

Initial value:

{
      this, "IPEstimator", "Trk::TrackToVertexIPEstimator/TrackToVertexIPEstimator"}

Definition at line 100 of file TrackParametersKVU.h.

m_KVUChi2Key

SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > DerivationFramework::TrackParametersKVU::m_KVUChi2Key

private

Initial value:

{
      this, "KVUChi2Key", ".KVUChi2"}

Definition at line 87 of file TrackParametersKVU.h.

m_KVUCovMatKey

SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > DerivationFramework::TrackParametersKVU::m_KVUCovMatKey

private

Initial value:

{
      this, "KVUCovMatKey", ".KVUCovMat"}

Definition at line 91 of file TrackParametersKVU.h.

m_KVUd0Key

SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > DerivationFramework::TrackParametersKVU::m_KVUd0Key

private

Initial value:

{
      this, "KVUd0Key", ".KVUd0"}

Definition at line 81 of file TrackParametersKVU.h.

m_KVUphiKey

SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > DerivationFramework::TrackParametersKVU::m_KVUphiKey

private

Initial value:

{
      this, "KVUphiKey", ".KVUphi"}

Definition at line 77 of file TrackParametersKVU.h.

m_KVUqOverPKey

SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > DerivationFramework::TrackParametersKVU::m_KVUqOverPKey

private

Initial value:

{
      this, "KVUqOverPKey", ".KVUqOverP"}

Definition at line 85 of file TrackParametersKVU.h.

m_KVUthetaKey

SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > DerivationFramework::TrackParametersKVU::m_KVUthetaKey

private

Initial value:

{
      this, "KVUthetaKey", ".KVUtheta"}

Definition at line 79 of file TrackParametersKVU.h.

m_KVUusedPVKey

SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > DerivationFramework::TrackParametersKVU::m_KVUusedPVKey

private

Initial value:

{
      this, "KVUusedPVKey", ".KVUusedPV"}

Definition at line 89 of file TrackParametersKVU.h.

m_KVUz0Key

SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > DerivationFramework::TrackParametersKVU::m_KVUz0Key

private

Initial value:

{
      this, "KVUz0Key", ".KVUz0"}

Definition at line 83 of file TrackParametersKVU.h.

m_LinearizedTrackFactory

ToolHandle< Trk::IVertexLinearizedTrackFactory > DerivationFramework::TrackParametersKVU::m_LinearizedTrackFactory

private

Initial value:

{
      this, "LinearizedTrackFactory", "Trk::FullLinearizedTrackFactory/FullLinearizedTrackFactory"}

Definition at line 98 of file TrackParametersKVU.h.

m_sgName

StringProperty DerivationFramework::TrackParametersKVU::m_sgName {this, "DecorationPrefix", "", "decoration prefix"}

private

Definition at line 103 of file TrackParametersKVU.h.

m_trackContainerKey

SG::ReadHandleKey< xAOD::TrackParticleContainer > DerivationFramework::TrackParametersKVU::m_trackContainerKey

private

Initial value:

{
      this, "TrackParticleContainerName", "InDetDisappearingTrackParticles"}

Definition at line 73 of file TrackParametersKVU.h.

m_vertexContainerKey

SG::ReadHandleKey< xAOD::VertexContainer > DerivationFramework::TrackParametersKVU::m_vertexContainerKey

private

Initial value:

{
      this, "VertexContainerName", "PrimaryVertices"}

Definition at line 75 of file TrackParametersKVU.h.

m_vertexTrackUpdator

ToolHandle< Trk::IVertexTrackUpdator > DerivationFramework::TrackParametersKVU::m_vertexTrackUpdator

private

Initial value:

{
      this, "VertexTrackUpdator", "Trk::KalmanVertexTrackUpdator"}

Definition at line 94 of file TrackParametersKVU.h.

---

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

• TrackParametersKVU.h
• TrackParameterKVU.cxx