SegmentAmbiSolver.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONR4_MUONPATTERNHELPERS_SEGMENTAMBISOLVER_H
#define MUONR4_MUONPATTERNHELPERS_SEGMENTAMBISOLVER_H
 
#include <AthenaBaseComps/AthMessaging.h>
#include <MuonPatternEvent/Segment.h>
 
#include <vector>
#include <unordered_map>
#include <array>
 
namespace MuonR4::SegmentFit {
    class SegmentAmbiSolver : public AthMessaging {
        public:
            struct Config{
                double selectByNDoFChi2{5.};
                unsigned int sharedPrecHits{3};
                bool remLeftRightAmbi{false};
            };

            SegmentAmbiSolver(const std::string&name,
                              Config&& config);

            using SegmentVec = std::vector<std::unique_ptr<Segment>>; 
            SegmentVec resolveAmbiguity(const ActsTrk::GeometryContext& gctx,
                                        SegmentVec&& toResolve) const;
        private:
            const Config m_cfg{};
            enum class Resolution: std::uint8_t{
                noOverlap, 
                superSet,  
                subSet     
            };

            std::vector<int> driftSigns(const ActsTrk::GeometryContext& gctx,
                                        const Segment& segment,
                                        const Segment::MeasVec& measurements) const;
            using MeasurementSet = std::unordered_set<const xAOD::UncalibratedMeasurement*>;
            MeasurementSet extractPrds(const Segment& segment) const;
            
            unsigned int countShared(const MeasurementSet& measSet1, 
                                     const MeasurementSet& measSet2) const;

            double redChi2(const Segment& segment) const;
 
 
 
 
    };
}
 
 
#endif