KalmanVertexUpdator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TrkVertexFitterUtils_KalmanVertexUpdator_H
#define TrkVertexFitterUtils_KalmanVertexUpdator_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "TrkVertexFitterInterfaces/IVertexUpdator.h"

 
namespace Trk
{
 class VxTrackAtVertex;
 class LinearizedTrack;
 
 class KalmanVertexUpdator : public AthAlgTool, virtual public IVertexUpdator
 {
  public:
  
  virtual StatusCode initialize() override;
  KalmanVertexUpdator(const std::string& t, const std::string& n, const IInterface*  p);
 
  ~KalmanVertexUpdator();

   virtual xAOD::Vertex * add(xAOD::Vertex& vtx, VxTrackAtVertex& trk) const override;

   virtual xAOD::Vertex * remove(xAOD::Vertex& vtx,  VxTrackAtVertex& trk) const override;

  virtual IVertexUpdator::positionUpdateOutcome positionUpdate (const xAOD::Vertex& vtx, const LinearizedTrack * trk, double trackWeight, IVertexUpdator::updateMode mode) const override;
 
  virtual float trackParametersChi2(const xAOD::Vertex& new_vtx, const LinearizedTrack * trk) const override;
  virtual float trackParametersChi2(const IVertexUpdator::positionUpdateOutcome& new_vtx, const LinearizedTrack * trk) const override;

  virtual float vertexPositionChi2(const xAOD::Vertex& old_vtx, const xAOD::Vertex& new_vtx) const override;
  virtual float vertexPositionChi2(const xAOD::Vertex& old_vtx, const IVertexUpdator::positionUpdateOutcome& new_vtx) const override;
 
  private:
 
  // internal methods performing the actual calculations
  static float trackParametersChi2(const Amg::Vector3D & new_position, const LinearizedTrack * trk, const AmgSymMatrix(5) & trkParametersWeight) ;
  static float vertexPositionChi2(const xAOD::Vertex& old_vtx, const Amg::Vector3D & new_position) ;

  xAOD::Vertex* update(xAOD::Vertex& vtx, const VxTrackAtVertex& trk, IVertexUpdator::updateMode mode) const;
   
 };//end of class definition
}//end of namespace definition
 
#endif