imstkGaussSeidel.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 "imstkIterativeLinearSolver.h"

namespace imstk
{
class LinearProjectionConstraint;

class GaussSeidel : public IterativeLinearSolver
{
public:
    GaussSeidel() { m_type = Type::GaussSeidel; };
    GaussSeidel(const SparseMatrixd& A, const Vectord& rhs);
    ~GaussSeidel() override = default;

    GaussSeidel(const GaussSeidel&)  = delete;
    GaussSeidel(const GaussSeidel&&) = delete;
    GaussSeidel& operator=(const GaussSeidel&)  = delete;
    GaussSeidel& operator=(const GaussSeidel&&) = delete;

    //void iterate(Vectord& x, bool updateResidual = true) override {};

    void gaussSeidelSolve(Vectord& x);

    void solve(Vectord& x) override;

    void solve(Vectord& x, const double tolerance);

    double getResidual(const Vectord& x) override;

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

    virtual void setMaxNumIterations(const size_t maxIter) override;

    void setTolerance(const double tolerance);

    void print() const override;

    void setLinearProjectors(std::vector<LinearProjectionConstraint>* f)
    {
        m_FixedLinearProjConstraints = f;
    }

    std::vector<LinearProjectionConstraint>& getLinearProjectors()
    {
        return *m_FixedLinearProjConstraints;
    }

    void setDynamicLinearProjectors(std::vector<LinearProjectionConstraint>* f)
    {
        m_DynamicLinearProjConstraints = f;
    }

    std::vector<LinearProjectionConstraint>& getDynamicLinearProjectors()
    {
        return *m_DynamicLinearProjConstraints;
    }

private:
    std::vector<LinearProjectionConstraint>* m_FixedLinearProjConstraints   = nullptr;
    std::vector<LinearProjectionConstraint>* m_DynamicLinearProjConstraints = nullptr;
};
} // namespace imstk