FittingStatistics.h

Go to the documentation of this file.

/******************************************************************************************************
 * (C) 2016 markummitchell@github.com. This file is part of Engauge Digitizer, which is released      *
 * under GNU General Public License version 2 (GPLv2) or (at your option) any later version. See file *
 * LICENSE or go to gnu.org/licenses for details. Distribution requires prior written permission.     *
 ******************************************************************************************************/
 
#ifndef FITTING_STATISTICS_H
#define FITTING_STATISTICS_H
 
#include "FittingCurveCoefficients.h"
#include "FittingPointsConvenient.h"
#include <QVector>
 
const int MAX_POLYNOMIAL_ORDER = 9; // Check execution time if this is increased from, say, 6 or 7
 
class Matrix;

class FittingStatistics
{
  // For unit testing
  friend class TestFitting;
  
public:
  FittingStatistics ();
  virtual ~FittingStatistics ();

  void calculateCurveFitAndStatistics (unsigned int order,
                                       const FittingPointsConvenient &pointsConvenient,
                                       FittingCurveCoefficients &coefficients,
                                       double &mse,
                                       double &rms,
                                       double &rSquared,
                                       int significantDigits);
  
private:
  
  void calculateCurveFit (int orderReduced,
                          const FittingPointsConvenient &pointsConvenient,
                          FittingCurveCoefficients &coefficients,
                          int significantDigits);
  bool calculateCurveFitReducedFurther (int orderReducedFurther,
                                        const FittingPointsConvenient &pointsConvenient,
                                        int significantDigits,
                                        QVector<double> &a) const;
  void calculateStatistics (const FittingPointsConvenient &pointsConvenient,
                            const FittingCurveCoefficients &coefficients,
                            double &mse,
                            double &rms,
                            double &rSquared);
  void loadXAndYArrays (int orderReduced,
                        const FittingPointsConvenient &pointsConvenient,
                        Matrix &X,
                        QVector<double> &Y) const;
  double yFromXAndCoefficients (const FittingCurveCoefficients &coefficients,
                                double x) const;
};
 
#endif // FITTING_STATISTICS_H