BFieldCorFunc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
//+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
// 19.07.2006, AUTHOR: OLIVER KORTNER
// Modified: 16.01.2008 by O. Kortner, RtSpline allowed as input; faster, but
//                      less accuracte implementation of correction funtion can
//                      be requested; bug fix in integral calculation.
//+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 
#ifndef MuonCalib_BFieldCorFuncH
#define MuonCalib_BFieldCorFuncH
 
//:::::::::::::::::::::::::
//:: CLASS BFieldCorFunc ::
//:::::::::::::::::::::::::
 
namespace MuonCalib {
}
 
//::::::::::::::::::
//:: HEADER FILES ::
//::::::::::::::::::
 
// STL //
#include <string>
#include <vector>
 
// CLHEP //
#include "EventPrimitives/EventPrimitives.h"
 
// MuonCalib //
#include "MdtCalibData/IMdtBFieldCorFunc.h"
#include "MdtCalibData/IRtRelation.h"
#include "MuonCalibMath/Legendre_polynomial.h"
 
namespace MuonCalib {
 
    class IRtRelation;
 
    class BFieldCorFunc : public IMdtBFieldCorFunc {
    public:
        // Constructors //
        explicit BFieldCorFunc(const std::string &quality, const CalibFunc::ParVec &parameters, const IRtRelation *rt) :
            IMdtBFieldCorFunc(parameters) {
            init(quality, parameters, rt);
        }
        explicit BFieldCorFunc(const CalibFunc::ParVec &parameters, const IRtRelation *rt) : IMdtBFieldCorFunc(parameters) {
            init(std::string("medium"), parameters, rt);
        }
 
        // Methods //
        // get-methods //
 
        double epsilon() const;
 
        // set-methods //
        //  void setEpsilon(const double & eps);
 
        void setRtRelationship(const IRtRelation &rt);
 
        // methods required by the base class "IMdtBFieldCorFunc" //
        std::string name() const;  
 
        double correction(double t, double B_wire, double B_mu) const;
 
        double correction_to_B(double t, double B_wire, double B_mu, double B_factor = -1.0) const;
 
    private:
        // data //
 
        // parameters (copy of m_parameter in the base class, needed to enable overwriting)
        std::vector<double> m_param;
 
        // quality setting //
        std::string m_quality;  // quality string ("high", "medium", "low")
        double m_step_size;     // integration step size steering the quality
 
        // correction function //
        Amg::VectorX m_alpha;  // parameter of the fit function describing the
                               // integral part of the correction function
 
        // pointer to the Legendre polynomial singleton (chosen in order to limit the amount of memory used)
        const Legendre_polynomial *m_Legendre;
 
        // information about r(t) used to calculate the correction function //
 
        // minimum and maximum drift times allowed by the r-t relationship
        double m_t_min, m_t_max;
        // minimum and maximum drift radius
        double m_r_min, m_r_max;
 
        // private methods //
        // initialization method, parameters are the parameters passed to constructor of this class
        void init(const std::string &quality, const CalibFunc::ParVec &params, const IRtRelation *rt);
 
        // get t(r) for the relationship "rt", the method is auxiliary and not optimized;
        // it will disappear when the t(r) will be available in the MuonCalib framework
        double t_from_r(const double &r, const IRtRelation *rt) const;
 
        // get the integral in the correction function from r_min to r_max; rt contains the drift
        // velocity needed to calculate the integral
        double integral(const double &r_min, const double &r_max, const IRtRelation *rt) const;
    };
 
}  // namespace MuonCalib
 
#endif