DerivationFramework::TrackParametersKVU Class Reference

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

#include <TrackParametersKVU.h>

Inheritance diagram for DerivationFramework::TrackParametersKVU:

Collaboration diagram for DerivationFramework::TrackParametersKVU:

Public Member Functions
virtual StatusCode	initialize () override
virtual StatusCode	addBranches (const EventContext &ctx) const override
	Check that the current event passes this filter.

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

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 53 of file TrackParametersKVU.h.

Member Function Documentation

addBranches()

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

overridevirtual

Check that the current event passes this filter.

Definition at line 43 of file TrackParametersKVU.cxx.

{
  // --- Get the tracks
  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 12 of file TrackParametersKVU.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());
 
  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 89 of file TrackParametersKVU.h.

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

m_IPEstimator

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

private

Initial value:

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

Definition at line 93 of file TrackParametersKVU.h.

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

m_KVUChi2Key

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

private

Initial value:

{
      this, "KVUChi2Key", m_trackContainerKey, "KVUChi2"}

Definition at line 80 of file TrackParametersKVU.h.

                                                                    {
      this, "KVUChi2Key", m_trackContainerKey, "KVUChi2"};

m_KVUCovMatKey

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

private

Initial value:

{
      this, "KVUCovMatKey", m_trackContainerKey, "KVUCovMat"}

Definition at line 84 of file TrackParametersKVU.h.

                                                                      {
      this, "KVUCovMatKey", m_trackContainerKey, "KVUCovMat"};

m_KVUd0Key

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

private

Initial value:

{
      this, "KVUd0Key", m_trackContainerKey, "KVUd0"}

Definition at line 74 of file TrackParametersKVU.h.

                                                                  {
      this, "KVUd0Key", m_trackContainerKey, "KVUd0"};

m_KVUphiKey

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

private

Initial value:

{
      this, "KVUphiKey", m_trackContainerKey, "KVUphi"}

Definition at line 70 of file TrackParametersKVU.h.

                                                                   {
      this, "KVUphiKey", m_trackContainerKey, "KVUphi"};

m_KVUqOverPKey

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

private

Initial value:

{
      this, "KVUqOverPKey", m_trackContainerKey, "KVUqOverP"}

Definition at line 78 of file TrackParametersKVU.h.

                                                                      {
      this, "KVUqOverPKey", m_trackContainerKey, "KVUqOverP"};

m_KVUthetaKey

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

private

Initial value:

{
      this, "KVUthetaKey", m_trackContainerKey, "KVUtheta"}

Definition at line 72 of file TrackParametersKVU.h.

                                                                     {
      this, "KVUthetaKey", m_trackContainerKey, "KVUtheta"};

m_KVUusedPVKey

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

private

Initial value:

{
      this, "KVUusedPVKey", m_trackContainerKey, "KVUusedPV"}

Definition at line 82 of file TrackParametersKVU.h.

                                                                      {
      this, "KVUusedPVKey", m_trackContainerKey, "KVUusedPV"};

m_KVUz0Key

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

private

Initial value:

{
      this, "KVUz0Key", m_trackContainerKey, "KVUz0"}

Definition at line 76 of file TrackParametersKVU.h.

                                                                  {
      this, "KVUz0Key", m_trackContainerKey, "KVUz0"};

m_LinearizedTrackFactory

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

private

Initial value:

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

Definition at line 91 of file TrackParametersKVU.h.

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

m_trackContainerKey

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

private

Initial value:

{
      this, "TrackParticleContainerName", "InDetDisappearingTrackParticles"}

Definition at line 66 of file TrackParametersKVU.h.

                                                                     {
      this, "TrackParticleContainerName", "InDetDisappearingTrackParticles"};

m_vertexContainerKey

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

private

Initial value:

{
      this, "VertexContainerName", "PrimaryVertices"}

Definition at line 68 of file TrackParametersKVU.h.

                                                               {
      this, "VertexContainerName", "PrimaryVertices"};

m_vertexTrackUpdator

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

private

Initial value:

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

Definition at line 87 of file TrackParametersKVU.h.

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

---

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

• TrackParametersKVU.h
• TrackParametersKVU.cxx