AlignResidualCalculator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRKALIGNGENTOOLS_ALIGNRESIDUALCALCULATOR_H
#define TRKALIGNGENTOOLS_ALIGNRESIDUALCALCULATOR_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "AthContainers/DataVector.h"
 
#include "TrkAlignInterfaces/IAlignResidualCalculator.h"
 
#include "TrkToolInterfaces/IResidualPullCalculator.h"
#include "TrkToolInterfaces/IUpdator.h"
#include "TrkAlignEvent/AlignResidualType.h"

 
class AtlasDetectorID;
 
namespace Trk {
 
  class AlignTrack;
  class ResidualPull;
  class Track;
  class AlignTSOS;
  class TrackStateOnSurface;
 
  class AlignResidualCalculator :  public AthAlgTool, virtual public IAlignResidualCalculator {
 
  public:
 
    AlignResidualCalculator(const std::string& type, const std::string& name,
            const IInterface* parent);
    virtual ~AlignResidualCalculator();
    
    virtual StatusCode initialize() override;
    virtual StatusCode finalize() override;

    virtual double setResiduals(DataVector<AlignTSOS>* alignTSOSColl, 
                                const Track* track) override;
    
    virtual double setResiduals(AlignTrack* alignTrack,
                                const Track* track) override;
    
    virtual double setResiduals(DataVector<AlignTSOS>::iterator firstAtsos,
                                DataVector<AlignTSOS>::iterator lastAtsos,
                                const Track* track, bool newTrack) override;

    virtual double chi2ForMeasType(int measType) const override
    { return m_chi2ForMeasType[measType]; }
 
  private:
 
    double setResidualsOnATSOS(AlignTSOS* atsos, const TrackStateOnSurface* tsos);
    void   accumulateScattering(const TrackStateOnSurface* tsos);
    const TrackStateOnSurface* getMatchingTSOS(const AlignTSOS* atsos, const Track* track);
 
    ToolHandle <IResidualPullCalculator> m_pullCalculator{
      this, "ResidualPullCalculator", "Trk::ResidualPullCalculator/ResidualPullCalculator"};
    ToolHandle <IUpdator>                m_updator{
      this, "UpdatorTool", "Trk::KalmanUpdator/TrkKalmanUpdator"};
 
    IntegerProperty m_resType{
      this, "ResidualType", Trk::AlignResidualType::HitOnly,
      "residual type as defined in the Trk::AlignResidualType enum"};
 
    BooleanProperty m_includeScatterers{this, "IncludeScatterers", false};
    double m_qOverP{};
    double m_previousQOverP{};
   
    int    m_nDoF{};
 
    double* m_chi2ForMeasType = nullptr;
 
    std::vector<const TrackStateOnSurface*> m_matchedTSOS;
 
  };
 
  
}
 
#endif // TRKALIGNGENTOOLS_ALIGNRESIDUALCALCULATOR_H