PolynomialLeastSquaresFitting.h

Go to the documentation of this file.

#pragma once

#include <vector>
#include "SupportPoints.h"

namespace NuTo
{
class PolynomialLeastSquaresFitting
{
public:
    void AddBoundaryCondition(std::pair<double, double> boundaryCondition);

    void AddBoundaryCondition(double x, double y);

    void BuildDerived();

    Eigen::VectorXd GetPolynomialCoefficients() const;

    void SetDegree(int degree);

    void SolveTransformed(const Eigen::MatrixXd& inputCoordinates, Eigen::MatrixXd& outputCoordinates) const;

    void SetSupportPoints(int dimInput, int dimOutput, Eigen::MatrixXd inputCoordinates,
                          Eigen::MatrixXd outputCoordinates);

private:
    SupportPoints mSupportPoints;
    Eigen::VectorXd mPolynomialCoeffs;
    std::vector<std::pair<double, double>> mBoundaryConditions;
    int mDegree = -1;
};

} // namespace NuTo