ls_solver.h

Go to the documentation of this file.

#ifndef __LS_SOLVER_H__
#define __LS_SOLVER_H__

#include <cstddef>
#include <vector>



class LSSolver {
public:
    virtual ~LSSolver() {}
    
    virtual int solve(double* result, double* b) = 0; 

protected:  
    std::size_t m_iNumRow;
    std::size_t m_iNumCol;
    double *m_vA;
    double *m_vb;
};






class QRSolver : public LSSolver {
public: 
    QRSolver(std::size_t numRow, std::size_t numCol, double *A, double limitR=1e-3) {  
        m_iNumRow = numRow;
        m_iNumCol = numCol;
        m_vA = A;
        isDecomposed = false; // The matrix A is not decomposed into QR at initialization
        m_fLimitR = limitR;
    }

    virtual int solve(double* result, double* b);

private:
    bool isDecomposed;
    double m_fLimitR;
    std::vector<int> m_vJPVT;
    std::vector<double> m_vTAU;
};


#endif // __LS_SOLVER_H__