gsGradientMethod.hpp

Go to the documentation of this file.

namespace gismo
{

template<class T>
bool gsGradientMethod<T>::initIteration( const typename gsGradientMethod<T>::VectorType& rhs,
                                            typename gsGradientMethod<T>::VectorType& x )
{
    if (Base::initIteration(rhs,x))
        return true;

    m_mat->apply(x,m_tmp);
    m_res = rhs - m_tmp;

    m_error = m_res.norm() / m_rhs_norm;
    return m_error < m_tol;

}

template<class T>
bool gsGradientMethod<T>::step( typename gsGradientMethod<T>::VectorType& x )
{
    m_precond->apply(m_res,m_update);
    m_mat->apply(m_update,m_tmp);

    T step_size;
    if (m_adapt_step_size)
        step_size = m_tmp.col(0).dot(m_res.col(0)) / m_tmp.col(0).dot(m_tmp.col(0));
    else
        step_size = m_step_size;

    x += step_size * m_update;
    m_res -= step_size * m_tmp;
    m_error = m_res.norm() / m_rhs_norm;
    return m_error < m_tol;
}


} // end namespace gismo