NswErrorCalibData.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCONDDATA_NSWERRORCALIBDATA_H
#define MUONCONDDATA_NSWERRORCALIBDATA_H
 
 
// Athena includes
#include "AthenaKernel/CondCont.h" 
#include "AthenaKernel/BaseInfo.h" 
#include "AthenaBaseComps/AthMessaging.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "GeoPrimitives/GeoPrimitives.h"
#include "GeoModelUtilities/TransientConstSharedPtr.h"
#include <functional>
#include <map>
 
class NswErrorCalibData: public AthMessaging {
    public:
    struct Input {
        Identifier stripId{};
        uint8_t clusterAuthor{0};
        double locTheta{0.};
        double locPhi{0.};
        Amg::Vector2D localPos{Amg::Vector2D::Zero()};
        unsigned int clusterSize{0};
        double clusterError{0};
    };
    
    using errorParametrizer = std::function<double(const Input& input, 
                                                   const std::vector<double>& pars)>;
    
    static errorParametrizer getParametrizer(const std::string& funcName);
    class ErrorConstants {
        public:
            ErrorConstants(const std::string& funcName,
                           uint8_t author,
                           uint16_t minStrip,
                           uint16_t maxStrip,
                           std::vector<double>&& parameters);
            uint16_t minStrip() const;
            uint16_t maxStrip() const;
            uint8_t author() const;
            double clusterUncertainty(const Input& clustInfo) const;
 
        bool operator<(const ErrorConstants& other) const;
        const std::vector<double>& pars() const { return m_pars; }
    private:       
        errorParametrizer m_evalFunc;
        uint8_t m_clusAlgAuthor{0};
        uint16_t m_stripMin{0};
        uint16_t m_stripMax{0};
        std::vector<double> m_pars{};
    };
 
    struct ErrorIdentifier{
        uint16_t strip{0};
        uint8_t clusAlgAuthor{0};
    };

    using ErrorConstantsSet = std::set<ErrorConstants, std::less<>>;
    
    public:
        NswErrorCalibData(const Muon::IMuonIdHelperSvc* idHelperSvc);
         ~NswErrorCalibData() = default;
 
        StatusCode storeConstants(const Identifier& gasGapId,
                                  ErrorConstants&& newConstants);
    
        double clusterUncertainty(const Input& clustInfo) const;
        
    private:
        const Muon::IMuonIdHelperSvc* m_idHelperSvc{nullptr};
        using ErrorMap = std::unordered_map<Identifier, ErrorConstantsSet>; 
        ErrorMap m_database{};
};
 
bool operator<(const NswErrorCalibData::ErrorConstants& a, const NswErrorCalibData::ErrorIdentifier& b);
bool operator<(const NswErrorCalibData::ErrorIdentifier& a, const NswErrorCalibData::ErrorConstants& b);
 
CLASS_DEF( NswErrorCalibData , 118696870 , 1 );
CONDCONT_DEF( NswErrorCalibData , 134103948 );
 
#endif