MuGirlLowBetaTag.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCOMBINEDEVENT_MUGIRLLOWBETATAG_H
#define MUONCOMBINEDEVENT_MUGIRLLOWBETATAG_H
 
#include <vector>
#include <memory>
 
#include "MuonCombinedEvent/CandidateSummary.h"
#include "MuonCombinedEvent/TagBase.h"
#include "TrkSegment/SegmentCollection.h"
#include "TrkTrack/TrackCollection.h"
 
namespace Trk {
    class Track;
}
 
namespace Muon {
    class MuonSegment;
}
 
namespace MuonCombined {
 
    class MuGirlLowBetaTag : public TagBase {
    public:
        MuGirlLowBetaTag(const ElementLink<TrackCollection>& combLink, 
                         const std::vector<ElementLink<Trk::SegmentCollection> >& segments);
 
        MuGirlLowBetaTag(const std::vector<ElementLink<Trk::SegmentCollection> >& segments);
 
        ~MuGirlLowBetaTag() = default;
 
        const Trk::Track* combinedTrack() const;
 
        /*ElementLink to the combined track*/
        ElementLink<TrackCollection> combinedTrackLink() const { return m_combLink; }
 
        const Trk::Track* updatedExtrapolatedTrack() const;
 
        void setUpdatedExtrapolatedTrack(const ElementLink<TrackCollection>& meLink);
 
        const std::vector<ElementLink<Trk::SegmentCollection> >& segments() const;
        std::vector<const Muon::MuonSegment*> associatedSegments() const override;
 
        /* stau Dressing */
        void setMuBeta(float muBeta);
        float getMuBeta() const;
 
    
 
        void setStauExtras(std::unique_ptr<MuGirlNS::StauExtras> stauExtras);
        MuGirlNS::StauExtras* getStauExtras();
        const MuGirlNS::StauExtras* getStauExtras() const;
 
 
        std::string name() const override { return "MuGirlLowBetaTag"; }
 
        std::string toString() const override { return name(); }
 
        const Trk::Track* primaryTrack() const override;
 
    private:
        MuGirlLowBetaTag(const MuGirlLowBetaTag&) = delete;
        MuGirlLowBetaTag& operator=(const MuGirlLowBetaTag&) = delete;
 
        ElementLink<TrackCollection> m_combLink{};
        ElementLink<TrackCollection> m_meLink{};
        std::vector<ElementLink<Trk::SegmentCollection> > m_segments{};  
 
        // dressing
        float m_muBeta{-9999.};
        std::unique_ptr<MuGirlNS::StauExtras> m_stauExtras{};
    };
    inline bool operator<(const MuGirlLowBetaTag& t1, const MuGirlLowBetaTag& t2) {
        const Trk::FitQuality* t1FQ = t1.combinedTrack() ? t1.combinedTrack()->fitQuality() : nullptr;
        const Trk::FitQuality* t2FQ = t2.combinedTrack() ? t2.combinedTrack()->fitQuality() : nullptr;
        if (t1FQ && t2FQ) { return t1FQ->chiSquared() < t2FQ->chiSquared(); }
        return t1.segments().size() < t2.segments().size();
    }
 
}  // namespace MuonCombined
 
#endif