CombinedFitTag.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCOMBINEDEVENT_COMBINEDFITTAG_H
#define MUONCOMBINEDEVENT_COMBINEDFITTAG_H
 
#include <vector>
 
#include "MuidEvent/FieldIntegral.h"
#include "MuonCombinedEvent/TagBase.h"
#include "TrkEventPrimitives/TrackScore.h"
#include "TrkTrack/TrackCollection.h"
 
namespace Trk {
    class Track;
}
 
namespace MuonCombined {
 
    class MuonCandidate;

    class CombinedFitTag : public TagBase {
    public:
        CombinedFitTag(xAOD::Muon::Author author, const MuonCandidate& muonCandidate, const Trk::TrackScore& score);

        ~CombinedFitTag();

        const Trk::Track* combinedTrack() const;
 
        /*ElementLink to combined track*/
        ElementLink<TrackCollection> combinedTrackLink() const;

        void setCombinedTrackLink(const ElementLink<TrackCollection>& combLink);

        const MuonCandidate& muonCandidate() const;

        const Trk::Track* updatedExtrapolatedTrack() const;
 
        /* ElementLink to ME track*/
        ElementLink<TrackCollection> updatedExtrapolatedTrackLink() const;

        void setUpdatedExtrapolatedTrackLink(const ElementLink<TrackCollection>& MELink);

        void innerMatch(double chi2, int dof, double prob);

        double matchChi2() const;

        int matchDoF() const;

        double matchProb() const;

        void fieldIntegral(const Rec::FieldIntegral& fieldI);

        const Rec::FieldIntegral& fieldIntegral() const;

        void momentumBalanceSignificance(double m);

        double momentumBalanceSignificance() const;

        const Trk::TrackScore& trackScore() const;

        std::string name() const override;

        std::string toString() const override;

        const Trk::Track* primaryTrack() const override;

        std::vector<const Muon::MuonSegment*> associatedSegments() const override;

        bool isCommissioning() const override;
 
    private:
        CombinedFitTag(const CombinedFitTag&) = delete;
        CombinedFitTag& operator=(const CombinedFitTag&) = delete;

        const MuonCandidate* m_muonCandidate{nullptr};  
        ElementLink<TrackCollection> m_combLink{};      // link to combined track
        Trk::TrackScore m_trackScore;
        ElementLink<TrackCollection> m_MELink{};  // link to ME track
 
        Rec::FieldIntegral m_fieldIntegral;
 
        double m_matchChi2{0.};
        int m_matchDoF{0};
        double m_matchProb{0.};
        double m_momentumBalanceSignificance{0.};
    };
 
    inline bool operator<(const CombinedFitTag& t1, const CombinedFitTag& t2) { return t1.trackScore() < t2.trackScore(); }
}  // namespace MuonCombined
 
#endif