ccstruct/lmedsq.h

Go to the documentation of this file.

#ifndef           LMEDSQ_H
#define           LMEDSQ_H

#include          "points.h"
#include          "varable.h"
#include          "grphics.h"
#include          "notdll.h"

class LMS
{
  public:
    LMS(              //constructor
        INT32 size);  //no of samples
    ~LMS ();                     //destructor
    void clear();  //clear samples
    void add(                 //add sample
             FCOORD sample);  //sample coords
    void fit(           //generate fit
             float &m,  //output line
             float &c);
    void constrained_fit(                //fixed gradient
                         float fixed_m,  //forced gradient
                         float &out_c);  //output line
    void fit_quadratic(                          //easy quadratic
                       float outlier_threshold,  //min outlier
                       double &a,                //x squared
                       float &b,                 //x
                       float &c);                //constant
    void plot(                 //plot fit
              WINDOW win,      //window
              COLOUR colour);  //colour to draw in
    float error() {  //get error
      return fitted ? line_error : -1;
    }

  private:

    void pick_line(           //random choice
                   float &m,  //output line
                   float &c);
    void pick_quadratic(            //random choice
                        double &a,  //output curve
                        float &b,
                        float &c);
    void compute_errors(          //find errors
                        float m,  //from line
                        float c);
                                 //find errors
    float compute_quadratic_errors(float outlier_threshold,  //min outlier
                                   double a,                 //from curve
                                   float m,
                                   float c);

    BOOL8 fitted;                //line parts valid
    INT32 samplesize;            //max samples
    INT32 samplecount;           //current sample size
    FCOORD *samples;             //array of samples
    float *errors;               //error distances
    double a;                    //x squared
    float m;                     //line gradient
    float c;
    float line_error;            //error of fit
};

extern INT_VAR_H (lms_line_trials, 12, "Number of linew fits to do");
#endif

---