MuPatSegment.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUPATSEGMENT_H
#define MUPATSEGMENT_H
 
#include <mutex>
#include <set>
 
#include "Identifier/Identifier.h"
#include "MuPatPrimitives/MuPatCandidateBase.h"
#include "MuonSegment/MuonSegment.h"
#include "MuonSegment/MuonSegmentQuality.h"
#include "MuonStationIndex/MuonStationIndex.h"
#include "TrkParameters/TrackParameters.h"
 
namespace Muon {
 
    class MuPatTrack;

 
    class MuPatSegment : public MuPatCandidateBase, public Trk::ObjectCounter<MuPatSegment> {
    public:
        using ChIdx = MuonStationIndex::ChIndex;
        using StIdx = MuonStationIndex::StIndex;
        MuPatSegment() = default;
        ~MuPatSegment() = default;
 
 
        int quality{0};
        const MuonSegmentQuality* segQuality{nullptr};
        const MuonSegment* segment{nullptr};
        std::shared_ptr<const Trk::AtaPlane> segPars{nullptr};
        int segmentIndex{-1};  
        int usedInFit{0};
        Identifier chid{0};
        std::string name{};
        ChIdx chIndex{ChIdx::ChUnknown};
        StIdx stIndex{StIdx::StUnknown};
        bool isEndcap{false};
        bool isMdt{false};  

        const Trk::TrackParameters& entryPars() const;

        void addTrack(MuPatTrack*);

        void removeTrack(MuPatTrack*);

        const std::set<MuPatTrack*>& tracks() const { return m_associatedTracks; }
 
    private:
        std::set<MuPatTrack*> m_associatedTracks;  //<! list of associated tracks
 
    };  // class MuPatSegment
    
    //
    // inline member functions implementations
    //
    inline const Trk::TrackParameters& MuPatSegment::entryPars() const { return *segPars; }
 
    struct SortSegInfoByR {
        bool operator()(const MuPatSegment* c1, const MuPatSegment* c2) { return operator()(*c1, *c2); }
        bool operator()(const MuPatSegment& c1, const MuPatSegment& c2) {
            return c1.segment->globalPosition().perp() < c2.segment->globalPosition().perp();
        }
    };
    struct SortSegInfoByZ {
        bool operator()(const MuPatSegment* c1, const MuPatSegment* c2) { return operator()(*c1, *c2); }
        bool operator()(const MuPatSegment& c1, const MuPatSegment& c2) {
            return fabs(c1.segment->globalPosition().z()) < fabs(c2.segment->globalPosition().z());
        }
    };
    struct SortSegInfoByQuality {
        bool operator()(const MuPatSegment* c1, const MuPatSegment* c2) { return operator()(*c1, *c2); }
        bool operator()(const MuPatSegment& c1, const MuPatSegment& c2) { return c1.quality > c2.quality; }
    };
 
}  // namespace Muon
 
#endif