imstkDirectLinearSolver.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/

#pragma once

#include "imstkLinearSolver.h"
#include "imstkMath.h"

#ifdef WIN32
#pragma warning( push )
#pragma warning( disable : 4127 )
#endif
#include <Eigen/Sparse>
#include <Eigen/SparseLU>
#ifdef WIN32
#pragma warning( pop )
#endif

namespace imstk
{
template<typename MatrixType> class DirectLinearSolver;

template<>
class DirectLinearSolver<Matrixd>: public LinearSolver<Matrixd>
{
public:
    DirectLinearSolver() { }
    DirectLinearSolver(const Matrixd& A, const Vectord& b);
    ~DirectLinearSolver() override { }

    void solve(Vectord& x) override;

    void solve(const Vectord& rhs, Vectord& x);

    void setSystem(std::shared_ptr<LinearSystemType> newSystem) override;

    void setSystem(std::shared_ptr<Matrixd> matrix);

    bool isIterative() const override
    {
        return false;
    };

private:
    Eigen::LDLT<Matrixd> m_solver;
};

template<>
class DirectLinearSolver<SparseMatrixd>: public LinearSolver<SparseMatrixd>
{
public:
    DirectLinearSolver() = default;
    ~DirectLinearSolver() override = default;

    DirectLinearSolver(const SparseMatrixd& matrix, const Vectord& b);

    void setSystem(std::shared_ptr<LinearSystemType> newSystem) override;

    void solve(Vectord& x) override;

    void solve(const Vectord& rhs, Vectord& x);

private:
    Eigen::SparseLU<SparseMatrixd, Eigen::COLAMDOrdering<MatrixType::StorageIndex>> m_solver;
};
} // namespace imstk