MuonCalibSelector.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCALIBEXTRANTUPLEANALYSIS_MUONCALIBSELECTOR_H
#define MUONCALIBEXTRANTUPLEANALYSIS_MUONCALIBSELECTOR_H
 
#include <float.h>
 
#include <vector>
 
#include "CxxUtils/checker_macros.h"
#include "MuonCalibExtraTreeEvent/EventHandler.h"
namespace MuonCalib {
 
    class MuonCalibExtendedSegment;
    class MuonCalibExtendedTrack;
    class MuonCalibTrackSummary;
 
    class MuonCalibSelector {
    public:
        using SegVec = EventHandler::SegmentVec;
        using TrackVec = EventHandler::TrackVec;
 
        struct Selection {
            Selection() = default;
            virtual ~Selection() = default;
 
            unsigned int nminMdtHits{3};        
            unsigned int nminMdtMl1{0};         
            unsigned int nminMdtMl2{0};         
            unsigned int nminMdtCh{0};          
            unsigned int nmaxMdtCh{1000};       
            unsigned int nminPhiTrigHits{0};    
            unsigned int nminEtaTrigHits{0};    
            unsigned int nmaxOutliers{100000};  
            unsigned int nmaxPseudo{1000};      
            double maxChi2Ndof{DBL_MAX};        
        };
 
        struct SegmentSelection : public Selection {
            SegmentSelection() = default;
 
            bool hasFittedT0{false};
            bool hasAssociatedTrack{false};
            bool singleML{false};
            bool twoML{false};
            double minAdcGoodFraction{0.};
        };
 
        struct TrackSelection : public Selection {
            TrackSelection() = default;
            bool barrelTrack{false};
            bool endcapTrackA{false};
            bool endcapTrackC{false};
            double maxr0{DBL_MAX};  
            double maxz0{DBL_MAX};  
        };
 
    public:
        // default constructor
        MuonCalibSelector();

        SegVec select(const SegVec& segments, const Selection* selection = nullptr) const;

        TrackVec select(const TrackVec& tracks, const Selection* selection = nullptr) const;

        bool select(const MuonCalibExtendedSegment& segment, const Selection* selection = nullptr) const;

        bool select(const MuonCalibExtendedTrack& track, const Selection* selection = nullptr) const;

        bool select(const MuonCalibTrackSummary& summary, const Selection* selection = nullptr) const;
 
    private:
        Selection m_selection{};
    };
 
}  // namespace MuonCalib
#endif