source: src/smoother/gs.cpp@ dfed1c

/**
 * @file   gs.cpp
 * @author Julian Iseringhausen <isering@ins.uni-bonn.de>
 * @date   Mon Apr 18 13:07:44 2011
 *
 * @brief  Gauss-Seidel method
 *
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <cassert>

#include "base/discretization.hpp"
#include "base/stencil.hpp"
#include "grid/multigrid.hpp"
#include "smoother/gs.hpp"
#include "mg.hpp"

using namespace VMG;

void GaussSeidel::Compute(Grid& sol, Grid& rhs)
{
#ifdef DEBUG_MATRIX_CHECKS
  sol.IsConsistent();
  rhs.IsConsistent();
  sol.IsCompatible(rhs);
#endif

  Grid::iterator grid_iter;
  Stencil::iterator stencil_iter;
  vmg_float temp;

  const Stencil& A = MG::GetDiscretization()->GetStencil();
  const vmg_float prefactor_inv = 1.0 / MG::GetDiscretization()->OperatorPrefactor(sol);
  const vmg_float diag_inv = 1.0 / A.GetDiag();

  for (grid_iter = rhs.Iterators().Local().Begin(); grid_iter != rhs.Iterators().Local().End(); ++grid_iter) {

    temp = prefactor_inv * rhs.GetCorrectedVal(*grid_iter);

    for (stencil_iter=A.begin(); stencil_iter!=A.end(); ++stencil_iter)
      temp -= stencil_iter->Val() * sol.GetVal(*grid_iter + stencil_iter->Disp());

    sol(*grid_iter) = temp * diag_inv;

  }
}