Fit2D.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONLINEARSEGMENTMAKERUTILITIES_FIT2D_H
#define MUONLINEARSEGMENTMAKERUTILITIES_FIT2D_H
 
#include "GaudiKernel/MsgStream.h"
#include <vector>
#include <cstring>
#include <string>
 
namespace Muon
{

    class Fit2D
    {
    public:
        struct Point
        {
            int    nIdx;        
            double fX;          
            double fY;          
            double fW;          
            double fChi2;       
            bool   bExclude;    
            const void*   pData;       

            Point();

            Point(int nIdx, double fX, double fY, double fW = 1.0, const void* pData = nullptr);
        };
 
        typedef std::vector<Point*> PointArray;     
 
        struct SimpleStats
        {
            int n;
            double fMean;
            double fStd;
            double fChi2;
 
            SimpleStats();
            void clear();
            std::string toString() const;
        };

        struct LinStats
        {
            int    n;           
            double fIntercept;  
            double fSlope;      
            double fCov[2][2];   
            double fChi2;       

            LinStats();

            void clear();

            void eval(double fX, double& fY, double& fYerr) const;

            std::string toString() const;
        };

        Fit2D();

        static void SimpleStatistics(const PointArray& points, SimpleStats& stats);

        static void fitLine(PointArray& points, double fExclChi2, bool bDump, LinStats& stats);

        static void fitPoint(PointArray& points, double fExclChi2, bool bDump, SimpleStats& stats);
    };
 
    inline Fit2D::Point::Point() :
    nIdx(0), fX(0.0), fY(0.0), fW(1.0), fChi2(0.0), bExclude(false), pData(NULL)
    {
    }
 
    inline Fit2D::Point::Point(int nIdx, double fX, double fY, double fW, const void*  pData) :
    nIdx(nIdx), fX(fX), fY(fY), fW(fW), fChi2(0.0), bExclude(false), pData(pData)
    {
    }
 
    inline Fit2D::SimpleStats::SimpleStats() :
    n(0),
    fMean(0.0),
    fStd(0.0),
    fChi2(0.0)
    {
    }
 
    inline void Fit2D::SimpleStats::clear()
    {
        n = 0;
        fMean = fStd = fChi2 = 0.0;
    }
 
    inline Fit2D::LinStats::LinStats() :
    n(0),
    fIntercept(0.0),
    fSlope(0.0),
    fChi2(0.0)
    {
        memset(fCov, 0, sizeof(fCov));
    }
 
    inline void Fit2D::LinStats::clear()
    {
        n = 0;
        fIntercept = fSlope = fChi2 = 0.0;
        memset(fCov, 0, sizeof(fCov));
    }
 
    inline Fit2D::Fit2D()
    {
    }
 
}
 
#endif //MUONLINEARSEGMENTMAKERUTILITIES_FIT2D_H