regressionline.h

Go to the documentation of this file.

#ifndef regressionline_h
#define regressionline_h
 
#include "doublevector.h"
 
enum LineType { FREE = 1, FIXEDSLOPE, FIXEDINTERCEPT, FIXED };
 
class Regression {
public:
  Regression();
  Regression(LineType ltype);
  virtual ~Regression() {};
  virtual void storeVectors(const DoubleVector& modData, const DoubleVector& obsData) = 0;
  void calcFit();
  void setIntercept(double intercept) { a = intercept; };
  void setSlope(double slope) { b = slope; };
  void setWeights(const DoubleVector& weights);
  int getError() { return error; };
  double getSSE();
  double getIntercept() { return a; };
  double getSlope() { return b; };
  LineType getType() const { return linetype; };
protected:
  void calcSSE();
  void calcSSEWeights();
  void calcSlope();
  void calcIntercept();
  void calcSlopeIntercept();
  int error;
  int useweights;
  double sse;
  double a;
  double b;
  DoubleVector w;
  DoubleVector x;
  DoubleVector y;
  LineType linetype;
};
 
class LinearRegression : public Regression {
public:
  LinearRegression(LineType ltype);
  ~LinearRegression() {};
  virtual void storeVectors(const DoubleVector& modData, const DoubleVector& obsData);
};
 
class LogLinearRegression : public Regression {
public:
  LogLinearRegression(LineType ltype);
  ~LogLinearRegression() {};
  virtual void storeVectors(const DoubleVector& modData, const DoubleVector& obsData);
};
 
class WeightRegression : public LinearRegression {
public:
  WeightRegression(LineType ltype);
  ~WeightRegression() {};
  virtual void storeVectors(const DoubleVector& modData, const DoubleVector& obsData);
};
 
class LogWeightRegression : public LogLinearRegression {
public:
  LogWeightRegression(LineType ltype);
  ~LogWeightRegression() {};
  virtual void storeVectors(const DoubleVector& modData, const DoubleVector& obsData);
};
 
#endif