FastVertexFitter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRKVERTEXFITTERBILLOIR_FASTVERTEXFITTER_H
#define TRKVERTEXFITTERBILLOIR_FASTVERTEXFITTER_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "CxxUtils/checker_macros.h"
#include "GaudiKernel/ToolHandle.h"
#include "TrkParameters/TrackParameters.h"
#include "TrkVertexFitterInterfaces/IVertexFitter.h"
#include "TrkVertexFitterInterfaces/IVertexLinearizedTrackFactory.h"
#include "TrkExInterfaces/IExtrapolator.h"
#include "xAODTracking/TrackParticleFwd.h"
#include "xAODTracking/VertexFwd.h"
 
namespace Trk {
class Track;
class TrackParticleBase;
class IExtrapolator;
class IVertexLinearizedTrackFactory;

 
class FastVertexFitter : 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;
 
  FastVertexFitter(const std::string& t,
                   const std::string& n,
                   const IInterface* p);
  virtual ~FastVertexFitter(); 

  virtual std::unique_ptr<xAOD::Vertex> fit(
    const EventContext& ctx,
    const std::vector<const Trk::TrackParameters*>& perigeeList,
    const Amg::Vector3D& startingPoint) const override;

  virtual std::unique_ptr<xAOD::Vertex> fit(
    const EventContext& ctx,
    const std::vector<const TrackParameters*>& perigeeList,
    const std::vector<const Trk::NeutralParameters*>& /*neutralPerigeeList*/,
    const Amg::Vector3D& startingPoint) const override
  {
    msg(MSG::WARNING)
      << "FastVertexFitter::fit(fit(const std::vector<const "
         "TrackParameters*>&,const std::vector<const "
         "Trk::NeutralParameters*>&,const Amg::Vector3D&) ignoring neutrals"
      << endmsg;
    return fit(ctx, perigeeList, startingPoint);
  };

  virtual std::unique_ptr<xAOD::Vertex> fit(
    const EventContext& ctx,
    const std::vector<const Trk::TrackParameters*>& perigeeList,
    const xAOD::Vertex& constraint) const override;

  virtual std::unique_ptr<xAOD::Vertex> fit(
    const EventContext& ctx,
    const std::vector<const TrackParameters*>& perigeeList,
    const std::vector<const Trk::NeutralParameters*>& /*neutralPerigeeList*/,
    const xAOD::Vertex& constraint) const override
  {
    msg(MSG::WARNING)
      << "FastVertexFitter::fit(fit(const std::vector<const "
         "TrackParameters*>&,const std::vector<const "
         "Trk::NeutralParameters*>&,const xAOD::Vertex&) ignoring neutrals"
      << endmsg;
    return fit(ctx, perigeeList, constraint);
  };

  virtual std::unique_ptr<xAOD::Vertex> fit(
    const EventContext& ctx,
    const std::vector<const Trk::TrackParameters*>& perigeeList) const override;
 
  virtual std::unique_ptr<xAOD::Vertex> fit(
    const EventContext& ctx,
    const std::vector<const TrackParameters*>& perigeeList,
    const std::vector<const Trk::NeutralParameters*>& /*neutralPerigeeList*/)
    const override
  {
    msg(MSG::WARNING) << "FastVertexFitter::fit(fit(const std::vector<const "
                         "TrackParameters*>&,const std::vector<const "
                         "Trk::NeutralParameters*>&) ignoring neutrals"
                      << endmsg;
    return fit(ctx, perigeeList);
  };

  virtual std::unique_ptr<xAOD::Vertex> fit(
    const EventContext& ctx,
    const std::vector<const xAOD::TrackParticle*>& vectorTrk,
    const Amg::Vector3D& startingPoint) const override;

  virtual std::unique_ptr<xAOD::Vertex> fit(
    const EventContext& ctx,
    const std::vector<const xAOD::TrackParticle*>& vectorTrk,
    const std::vector<const xAOD::NeutralParticle*>& /*vectorNeut*/,
    const Amg::Vector3D& startingPoint) const override
  {
    msg(MSG::WARNING)
      << "FastVertexFitter::fit(fit(const std::vector<const "
         "TrackParticle*>&,const std::vector<const "
         "Trk::NeutralParticle*>&,const Amg::Vector3D&) ignoring neutrals"
      << endmsg;
    return fit(ctx, vectorTrk, startingPoint);
  };

  virtual std::unique_ptr<xAOD::Vertex> fit(
    const EventContext& ctx,
    const std::vector<const xAOD::TrackParticle*>& vectorTrk,
    const xAOD::Vertex& constraint) const override;

  virtual std::unique_ptr<xAOD::Vertex> fit(
    const EventContext& ctx,
    const std::vector<const xAOD::TrackParticle*>& vectorTrk,
    const std::vector<const xAOD::NeutralParticle*>& /*vectorNeut*/,
    const xAOD::Vertex& constraint) const override
  {
    msg(MSG::WARNING)
      << "FastVertexFitter::fit(fit(const std::vector<const "
         "TrackParticle*>&,const std::vector<const "
         "Trk::NeutralParticle*>&,const xAOD::Vertex&) ignoring neutrals"
      << endmsg;
    return fit(ctx, vectorTrk, constraint);
  };
 
private:
  unsigned int m_maxIterations;
  double m_maxDchi2PerNdf;
 
  ToolHandle<Trk::IExtrapolator> m_extrapolator{this, "Extrapolator", "Trk::Extrapolator"};
  ToolHandle<Trk::IVertexLinearizedTrackFactory> m_linFactory{this, "LinearizedTrackFactory", "Trk::FullLinearizedTrackFactory"};
};
}
#endif