FullVertexFitter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRKVERTEXFITTERBILLOIR_FULLVERTEXFITTER_H
#define TRKVERTEXFITTERBILLOIR_FULLVERTEXFITTER_H
 
#include "GaudiKernel/ToolHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "TrkVertexFitterInterfaces/IVertexFitter.h"
#include "TrkParameters/TrackParameters.h"
#include "xAODTracking/VertexFwd.h" 
#include "xAODTracking/TrackParticleFwd.h" 
#include <memory>
#include <vector>
 
namespace Trk
{
  class Track;
  class TrackParticleBase;
  class IExtrapolator;
  class IVertexLinearizedTrackFactory;
  
 
  class FullVertexFitter : 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;
 
    enum FitError
    {
      FITOK,
      MATINV,
      NEGTRCHI2,
      MAXCHI2,
      MAXTRCHI2,
      NOTRKS,
      NOFIT
    };
    FullVertexFitter(const std::string& t,
                     const std::string& n,
                     const IInterface* p);
    virtual ~FullVertexFitter(); 

    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)
        << "FullVertexFitter::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)
        << "FullVertexFitter::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) << "FullVertexFitter::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)
        << "FullVertexFitter::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)
        << "FullVertexFitter::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::IVertexLinearizedTrackFactory> m_linFactory;
  };
  }
#endif