NoPivotLU.hpp

Go to the documentation of this file.

#pragma once
 
#include "Matrix.hpp"
 
class NoPivotLU {
  public:
 
    NoPivotLU() = default;
    NoPivotLU(const SquareMatrix &matrix) { compute(matrix); }
    NoPivotLU(SquareMatrix &&matrix) { compute(std::move(matrix)); }
 
 
  public:
 
    void compute(SquareMatrix &&matrix);
    void compute(const SquareMatrix &matrix);
 
 
  public:
 
    SquareMatrix &&steal_L();
 
    void get_L_inplace(Matrix &L) const;
    SquareMatrix get_L() const &;
    SquareMatrix &&get_L() && { return steal_L(); }
 
 
  public:
 
    SquareMatrix &&steal_U();
 
    void get_U_inplace(Matrix &U) const;
    SquareMatrix get_U() const &;
    SquareMatrix &&get_U() && { return steal_U(); }
 
 
  public:
 
    void solve_inplace(Matrix &B) const;
    Matrix solve(const Matrix &B) const;
    Matrix &&solve(Matrix &&B) const;
    Vector solve(const Vector &B) const;
    Vector &&solve(Vector &&B) const;
 
 
  public:
 
    bool is_factored() const { return state == Factored; }
 
    bool has_LU() const { return is_factored(); }
 
    SquareMatrix &&steal_LU();
 
    const SquareMatrix &get_LU() const & { return LU; }
    SquareMatrix &&get_LU() && { return steal_LU(); }
 
 
  private:
    void compute_factorization();
    void back_subs(const Matrix &B, Matrix &X) const;
    void forward_subs(const Matrix &B, Matrix &X) const;
 
  private:
    SquareMatrix LU;
 
    enum State {
        NotFactored = 0,
        Factored = 1,
    } state = NotFactored;
};
 
std::ostream &operator<<(std::ostream &os, const NoPivotLU &lu);