KalmanVertexOnJetAxisUpdator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TrkVertexFitterUtils_KalmanVertexOnJetAxisUpdator_H
#define TrkVertexFitterUtils_KalmanVertexOnJetAxisUpdator_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "EventPrimitives/EventPrimitives.h"
#include "GeoPrimitives/GeoPrimitives.h"
 
namespace Trk
{
  class VxJetCandidate;
  class VxTrackAtVertex;
  class LinearizedTrack;
  class RecVertexPositions;
  class VxVertexOnJetAxis;
 
  typedef Eigen::Matrix<double, 3, Eigen::Dynamic> AmgMatrix3X;
 
  static const InterfaceID IID_KalmanVertexOnJetAxisUpdator("Trk::KalmanVertexOnJetAxisUpdator", 1, 0);
 
 
 class KalmanVertexOnJetAxisUpdator : public AthAlgTool
 {
  public:
     
   static const InterfaceID& interfaceID()
     {
       return IID_KalmanVertexOnJetAxisUpdator;
     };
 
 
  virtual StatusCode initialize() override;
 
  KalmanVertexOnJetAxisUpdator(const std::string& t, const std::string& n, const IInterface*  p);
 
  void add(VxTrackAtVertex* trackToAdd,const VxVertexOnJetAxis* vertexToUpdate,VxJetCandidate* candidateToUpdate) const;
  void addWithFastUpdate(VxTrackAtVertex* trackToAdd,const VxVertexOnJetAxis* vertexToUpdate,VxJetCandidate* candidateToUpdate) const;
 
  void remove(VxTrackAtVertex* trackToRemove,const VxVertexOnJetAxis* vertexToUpdate,VxJetCandidate* candidateToUpdate) const;
  void removeWithFastUpdate (VxTrackAtVertex* trackToRemove,const VxVertexOnJetAxis* vertexToUpdate,VxJetCandidate* candidateToUpdate) const;
 
  void updateChi2NdfInfo(VxTrackAtVertex* trackToUpdate,
             const VxVertexOnJetAxis* vertexToUpdate,
             VxJetCandidate* vertexCandidate) const;
 
  
  void updateVertexChi2(VxJetCandidate* vertexCandidate) const;
 
 
  Trk::RecVertexPositions positionUpdate (VxJetCandidate& vertexCandidate,
                      const LinearizedTrack * trk, 
                      double trackWeight, int sign, 
                      int numVertex,bool isPrimary,
                                          bool doFastUpdate) const;
 
 
  std::pair<Amg::Vector3D,Eigen::Matrix3Xd> createTransformJacobian(const Amg::VectorX & initialjetdir,
                                                                    int numVertex, bool isPrimary,
                                                                    bool truncate) const;
 
  void smartInvert(Amg::MatrixX & new_vrt_cov) const;
   
 
 private:
 
  void update(VxTrackAtVertex* trackToUpdate,const VxVertexOnJetAxis* vertexToUpdate,VxJetCandidate* candidateToUpdate,int sign, bool doFastUpdate) const ;
  
  double calculateTrackChi2 (const VxJetCandidate& vertexCandidate,
                const LinearizedTrack * trk, 
                double trackWeight, 
                int numVertex,bool isPrimary=false) const;
 
 
 
  
 private:
 
  double m_initialMomentumError;
 
 };//end of class definition
}//end of namespace definition
 
#endif