SequentialVertexFitter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TrkVertexFitters_SequentialVertexFitter_H
#define TrkVertexFitters_SequentialVertexFitter_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "TrkVertexFitterInterfaces/IVertexFitter.h"
#include "TrkParameters/TrackParameters.h"
#include "TrkParticleBase/TrackParticleBase.h"
 
//xAOD includes
#include "xAODTracking/VertexFwd.h"
#include "xAODTracking/TrackParticleFwd.h"
#include "xAODTracking/NeutralParticle.h"
 
 
namespace Trk
{
 class Track;
 class VxTrackAtVertex;
 class IVertexUpdator;
 class IVertexSmoother;
 class IVertexLinearizedTrackFactory;
 
 class SequentialVertexFitter : public extends<AthAlgTool, IVertexFitter>
 {
  public:
    // The following 'using' can be removed when IVertexFitter::fit has been fully migrated to the one with the EventContext
    using Trk::IVertexFitter::fit;
  
    virtual StatusCode initialize() override;
    virtual StatusCode finalize() override;
 
    SequentialVertexFitter(const std::string& t, const std::string& n, const IInterface*  p);
 
    virtual ~SequentialVertexFitter();
    
    virtual xAOD::Vertex * fit(const std::vector<const xAOD::TrackParticle*>& vectorTrk,
                   const std::vector<const xAOD::NeutralParticle*>& vectorNeut,
                   const Amg::Vector3D& startingPoint) const override;
    virtual xAOD::Vertex * fit(const std::vector<const xAOD::TrackParticle*>& vectorTrk,
                   const Amg::Vector3D& startingPoint) const override
   {return fit(vectorTrk, std::vector<const xAOD::NeutralParticle*>(), startingPoint);};
 
    virtual xAOD::Vertex * fit(const std::vector<const xAOD::TrackParticle*>& vectorTrk,
                   const std::vector<const xAOD::NeutralParticle*>& vectorNeut,
                   const xAOD::Vertex& constraint) const override;
    virtual xAOD::Vertex * fit(const std::vector<const xAOD::TrackParticle*>& vectorTrk,
                                 const xAOD::Vertex& constraint)  const override
   {return fit(vectorTrk, std::vector<const xAOD::NeutralParticle*>(), constraint);};
 
    
    
    virtual xAOD::Vertex * fit(const std::vector<const Trk::TrackParameters*> & perigeeList, 
                   const std::vector<const Trk::NeutralParameters*> & neutralPerigeeList,
                               const Amg::Vector3D& startingPoint) const override;
    virtual xAOD::Vertex * fit(const std::vector<const Trk::TrackParameters*> & paramList,
                               const Amg::Vector3D& startingPoint) const override
   {return fit(paramList, std::vector<const Trk::NeutralParameters*>(), startingPoint);};
    
    virtual xAOD::Vertex * fit(const std::vector<const Trk::TrackParameters*> & perigeeList, 
                   const std::vector<const Trk::NeutralParameters*> & neutralPerigeeList,
                               const xAOD::Vertex& constraint) const override;
    virtual xAOD::Vertex * fit(const std::vector<const Trk::TrackParameters*> & perigeeList,
                   const xAOD::Vertex& constraint) const override
   {return fit(perigeeList, std::vector<const Trk::NeutralParameters*>(), constraint);};
 
   virtual xAOD::Vertex* fit(
     const std::vector<const Trk::TrackParameters*>& perigeeList,
     const std::vector<const Trk::NeutralParameters*>& neutralPerigeeList)
     const override;
   virtual xAOD::Vertex* fit(
     const std::vector<const Trk::TrackParameters*>& perigeeList) const override
   {return fit(perigeeList, std::vector<const Trk::NeutralParameters*>());};
    
    
 private:
    
   std::vector<Trk::VxTrackAtVertex> linearizeTracks(
     const std::vector<const Trk::TrackParameters*>& perigeeList,
     const std::vector<const Trk::NeutralParameters*>& neutralPerigeeList,
     const xAOD::Vertex& vrt) const;
 
   void reLinearizeTracks(std::vector<Trk::VxTrackAtVertex>& tracks, const Amg::Vector3D & vrt)const;
 
   ToolHandle< IVertexUpdator > m_Updator;
   ToolHandle< IVertexSmoother > m_Smoother;
   ToolHandle< Trk::IVertexLinearizedTrackFactory > m_LinTrkFactory;  
   
   bool m_doSmoothing; 
   
   unsigned int m_maxStep;
   
   float m_maxShift;
 
   bool m_useLooseConvergence;
   
   float m_maxDeltaChi2;
 
    /*
     * Avoid vertex to be initialize outside ID envelope given bu m_maxR and m_maxZ
     */
 
    double m_maxR;
    double m_maxZ;
 
 }; //end of class description
}//end of namespace definition
 
#endif