#pragma  once

#include <QObject>
#include <QColor>
#include <QVector>
#include <QString>
#include <QMap>
#include <QMapIterator>
#include <QMatrix>

#include "iAlgRegress_global.h"
#include "zxAlgMatrix.h"

// 拟合算法
class I_ALGREGRESS_EXPORT zxAlgRegress
{

public:
    zxAlgRegress();
    virtual ~zxAlgRegress();

    // 组织表达式
    virtual QString getExpStr() = 0;

public:

	// 过滤掉无效数据。
	virtual bool setFiltering() = 0;
    virtual bool cal() = 0;
	// 将参数约束、数据源等转换为线性回归方程可以处理的形式。
	virtual bool setTransformation() = 0;
	// 将线性回归方程计算完成的参数反向变换成当前回归类型的参数。
    virtual bool setReciprocalTransformation() = 0;
	// 根据回归线进行计算
	virtual double getValue(VecDouble& vMatrixX) = 0;
    double getValue(double d)
	{
		VecDouble vMatrixX;
        vMatrixX.push_back(d);
		return getValue(vMatrixX);
	}

public:

    void setCoeff(QMap<uint, double>& mapCoeff);
    void setRestrain(bool bRestrain);
    void setSourceData(VVecDouble& vSourceData);
    void setRestrainMatrix(VVecDouble& vRestrainMatrix);
    void setRestrainType(uint nType);
    void setRestrainPoint(VecDouble vRestrainPoint);

    QMap<uint, double>& getCoeff();
    bool isRestrain();
    VVecDouble getSourceData();
    VVecDouble getRestrainMatrix();
    double getDeterCoff();
    uint getValidDataNum();
    uint getDataNum();

protected:

    double m_dDeterCoff;	// 决定系数R2
    bool m_bModeLimit;      // 是否为约束回归
    uint m_nDataNum;		// 数据个数
    uint m_nValidDataNum;   // 有效数据个数

    uint m_nTypeLimit;// 约束类型，0：定点；1：定参数；2：矩阵约束。

    // 所有项的系数。map<参数编号， 对应项系数>
    // 线性下   0：乘积项系数		1：常数项系数
    // 指数下   0：乘积项系数		1：指数项系数
    // 对数下	0：乘积项系数		1：常数项系数
    // 乘幂下	0：乘积项系数		1：指数项系数
    // 特殊回归模型的参数编号见具体的类。
    QMap<uint, double> m_mapCoeff;

    // 约束点。第一个为Y0，再依次为X10、X20、X30...
    VecDouble m_vRestrainPoint;

    // 源数据。
    // 外部的vector的每一项，指的是一组观察变量。
    // 内部的每一个vector指的是每一组观察变量的每一个子项：	Yi、Xi1、Xi2、Xi3......
    VVecDouble m_vSourceData;

    // 约束矩阵
    // 假定回归方程的待求参数向量为 C 。C的排序为：由低次向高次参数。
    // C受矩阵约束： AC = b 。 A的排序与C相同。
    // A 为 k * p 矩阵，并且秩为 k ， C 为 p * 1 矩阵，b 为 k * 1 矩阵，即C、b为列向量。
    // 外部的vector的每一项，指的是一组约束条件,即约束方程组中的一个方程。
    // 内部的vector的每一项，指的是一组约束条件中的每一个参数。
    // 内部的vector的排序方式为 : bi 、Ai1、Ai2、Ai3......
    VVecDouble m_vRestrainMatrix;
};