TrackParametersKVU.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// TrackParametersKVU.h  (c) ATLAS Detector software
 
/*
  Decorates the vertex constrained track parameters to the track container.
  The vertex fit is performed with a track and the nearest vertex.
  The updated the perigee paramters, chi2 of the vertex fit, and covariance matrix of the perigee parameters are decorated, but d0 and z0 should be 0.
  The nDoF of the vertex fit is always 2. (5 track parameters + 3 vertex parameters - 3 track parameters - 3 vertex parameters)
 
  user specifies...
 
  TrackParticleContainerName: track particle container.
  VertexContainerName:        vertex container.
*/
 
#ifndef DERIVATIONFRAMEWORK_TRACKPARAMETERSATPV_H
#define DERIVATIONFRAMEWORK_TRACKPARAMETERSATPV_H 
 
#include<string>
 
// Gaudi & Athena basics
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
 
 
#include "VxVertex/VxContainer.h"
#include "VxVertex/VxCandidate.h"
#include "VxVertex/VxTrackAtVertex.h"
#include "TrkTrack/TrackCollection.h"
#include "TrkTrack/Track.h"
 
// DerivationFramework includes
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
 
#include "xAODTracking/VertexContainer.h"
#include "TrkVertexFitterInterfaces/IVertexTrackUpdator.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "TrkExInterfaces/IExtrapolator.h"
#include "TrkVertexFitterInterfaces/IVertexLinearizedTrackFactory.h"
#include "TrkVertexFitterInterfaces/ITrackToVertexIPEstimator.h"
 
namespace DerivationFramework {
 
 
  class TrackParametersKVU : public extends<AthAlgTool, IAugmentationTool> {
   
  public: 
    TrackParametersKVU( const std::string& t, const std::string& n, const IInterface* p );
   
    virtual ~TrackParametersKVU() = default;
   
    // Athena algtool's Hooks
    virtual StatusCode initialize() override;
 
    virtual StatusCode addBranches() const override;
 
  private:
    SG::ReadHandleKey< xAOD::TrackParticleContainer > m_trackContainerKey{
      this, "TrackParticleContainerName", "InDetDisappearingTrackParticles"};
    SG::ReadHandleKey< xAOD::VertexContainer > m_vertexContainerKey{
      this, "VertexContainerName", "PrimaryVertices"}; 
    SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_KVUphiKey{
      this, "KVUphiKey", ".KVUphi"};
    SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_KVUthetaKey{
      this, "KVUthetaKey", ".KVUtheta"};
    SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_KVUd0Key{
      this, "KVUd0Key", ".KVUd0"};
    SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_KVUz0Key{
      this, "KVUz0Key", ".KVUz0"};
    SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_KVUqOverPKey{
      this, "KVUqOverPKey", ".KVUqOverP"};
    SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_KVUChi2Key{
      this, "KVUChi2Key", ".KVUChi2"};
    SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_KVUusedPVKey{
      this, "KVUusedPVKey", ".KVUusedPV"};
    SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_KVUCovMatKey{
      this, "KVUCovMatKey", ".KVUCovMat"};
 
    ToolHandle< Trk::IVertexTrackUpdator > m_vertexTrackUpdator {
      this, "VertexTrackUpdator", "Trk::KalmanVertexTrackUpdator"};
    ToolHandle< Trk::IExtrapolator >  m_extrapolator {
      this, "TrackExtrapolator", "Trk::Extrapolator/AtlasExtrapolator"};
    ToolHandle< Trk::IVertexLinearizedTrackFactory > m_LinearizedTrackFactory {
      this, "LinearizedTrackFactory", "Trk::FullLinearizedTrackFactory/FullLinearizedTrackFactory"};
    ToolHandle< Trk::ITrackToVertexIPEstimator> m_IPEstimator {
      this, "IPEstimator", "Trk::TrackToVertexIPEstimator/TrackToVertexIPEstimator"};
 
    StringProperty m_sgName {this, "DecorationPrefix", "", "decoration prefix"};
 
  }; 
 
}
 
#endif