GXFTrajectory.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRK_GXFTRAJECTORY_H
#define TRK_GXFTRAJECTORY_H
 
#include "TrkTrack/GXFTrackState.h"
#include "TrkGeometry/MagneticFieldProperties.h"
 
namespace Trk {
  class MeasurementBase;
  class MaterialEffectsBase;
  class Layer;
  class CylinderLayer;
  class DiscLayer;
  class MagneticFieldProperties;
 
  class GXFTrajectory {
 
  public:
    GXFTrajectory();
    GXFTrajectory(const GXFTrajectory & rhs);
    GXFTrajectory &operator=(const GXFTrajectory &rhs);
    GXFTrajectory(GXFTrajectory &&rhs) = default;
    GXFTrajectory &operator=(GXFTrajectory &&rhs) = default;
    ~GXFTrajectory() = default;
 
    bool addMeasurementState(std::unique_ptr<GXFTrackState>, int index = -1);
    void addMaterialState(std::unique_ptr<GXFTrackState>, int index = -1);
    void addBasicState(std::unique_ptr<GXFTrackState>, int index = -1);
 
    void setReferenceParameters(std::unique_ptr<const TrackParameters>);
    void setScatteringAngles(std::vector < std::pair < double, double > >&);
    void setBrems(std::vector<double> &);
    void setNumberOfPerigeeParameters(int);
    void setConverged(bool);
    void reset();
    void setPrefit(int);
    void setOutlier(int, bool isoutlier = true);
    void setPrevChi2(double);
    void setChi2(double);
    void setMass(double);
 
    void conditionalSetCalorimeterEnergyLossState(GXFTrackState *);
 
    std::pair<GXFTrackState *, GXFTrackState *> findFirstLastMeasurement(void);
    bool hasKink(void);
 
    int numberOfScatterers() const;
    void setNumberOfScatterers(int);
    int numberOfBrems() const;
    void setNumberOfBrems(int);
    int numberOfUpstreamStates() const;
    int numberOfUpstreamScatterers() const;
    int numberOfUpstreamBrems() const;
    int numberOfPerigeeParameters() const;
    int numberOfFitParameters() const;
    int numberOfSiliconHits() const;
    int numberOfTRTPrecHits() const;
    int numberOfTRTTubeHits() const;
    int numberOfTRTHits() const;
    int numberOfHits() const;
    int numberOfPseudoMeasurements() const;
    int numberOfOutliers() const;
 
    void updateTRTHitCount(int index, float oldError);
 
    const std::vector<std::unique_ptr<GXFTrackState>> & trackStates() const;
    std::vector<std::unique_ptr<GXFTrackState>> & trackStates();
    std::vector < std::pair < double, double >>&scatteringAngles();
    std::vector < std::pair < double, double >>&scatteringSigmas();
    std::vector<double> & brems();
 
    const TrackParameters * referenceParameters();
    bool converged() const;
    int prefit() const;
    void resetReferenceParameters();
    double chi2() const;
    double prevchi2() const;
    int nDOF() const;
 
    Amg::VectorX & residuals();
    Amg::VectorX & errors();
    Amg::MatrixX & weightedResidualDerivatives();
 
    double totalX0() const;
    double totalEnergyLoss() const;
 
    double mass() const;
    GXFTrackState *caloElossState();
 
    std::vector < std::pair < const Layer *,
    const Layer *>>&upstreamMaterialLayers();
 
    void resetCovariances(void);
    std::unique_ptr<const FitQuality> quality(void) const;
 
    bool m_straightline;
    MagneticFieldProperties m_fieldprop = Trk::FullField;
 
  private:
    std::vector<std::unique_ptr<GXFTrackState>> m_states;  
    int m_ndof;
    double m_chi2;
    double m_prevchi2;
    int m_nperpars;
    int m_nscatterers;
    int m_ncaloscatterers;
    int m_nbrems;
    int m_nupstreamstates;
    int m_nupstreamscatterers;
    int m_nupstreamcaloscatterers;
    int m_nupstreambrems;
    int m_nhits;
    int m_noutl;
    int m_nsihits;
    int m_ntrthits;
    int m_ntrtprechits;
    int m_ntrttubehits;
    int m_npseudo;
    int m_nmeasoutl;
    std::unique_ptr<const TrackParameters> m_refpar;
    bool m_converged;
    std::vector < std::pair < double, double >>m_scatteringangles;
    std::vector < std::pair < double, double >>m_scatteringsigmas;
    std::vector<double> m_brems;
    Amg::VectorX m_res;
    Amg::VectorX m_errors;
    Amg::MatrixX m_weightresderiv;
    double m_totx0;
    double m_toteloss;
    double m_mass;
    int m_prefit;
    GXFTrackState *m_caloelossstate;
    std::vector < std::pair < const Layer *, const Layer *>>m_upstreammat;
  };
}
#endif