CoCg.cxx

// Copyright (C) 2003-2009 Marc Duruflé
//
// This file is part of the linear-algebra library Seldon,
// http://seldon.sourceforge.net/.
//
// Seldon is free software; you can redistribute it and/or modify it under the
// terms of the GNU Lesser General Public License as published by the Free
// Software Foundation; either version 2.1 of the License, or (at your option)
// any later version.
//
// Seldon is distributed in the hope that it will be useful, but WITHOUT ANY
// WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
// FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for
// more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with Seldon. If not, see http://www.gnu.org/licenses/.
 
 
#ifndef SELDON_FILE_ITERATIVE_COCG_CXX
 
namespace Seldon
{
 
 
#ifdef SELDON_WITH_VIRTUAL
  template<class T, class Vector1>
  int CoCg(const VirtualMatrix<T>& A, Vector1& x, const Vector1& b,
           Preconditioner_Base<T>& M,
           Iteration<typename ClassComplexType<T>::Treal>& iter)
#else
  template <class Titer, class Matrix1, class Vector1, class Preconditioner>
  int CoCg(const Matrix1& A, Vector1& x, const Vector1& b,
           Preconditioner& M, Iteration<Titer> & iter)
#endif
  {
    const int N = A.GetM();
    if (N <= 0)
      return 0;
 
    typedef typename Vector1::value_type Complexe;
    Complexe rho, rho_1, alpha, beta, delta;
    Complexe zero, one;
    SetComplexZero(zero);
    SetComplexOne(one);
    rho = one;
    rho_1 = zero;
 
    Vector1 p(b), q(b), r(b), z(b);
    p.Fill(zero); q.Fill(zero); r.Fill(zero); z.Fill(zero);
 
    // for implementation see Cg
    // we initialize iter
    int success_init = iter.Init(b);
    if (success_init != 0)
      return iter.ErrorCode();
 
    Copy(b, r);
    if (!iter.IsInitGuess_Null())
      iter.MltAdd(-one, A, x, one, r);
    else
      x.Fill(zero);
 
    iter.SetNumberIteration(0);
        
    // Loop until the stopping criteria are reached
    while (! iter.Finished(r))
      {
        // preconditioning
        M.Solve(A, r, z);
        
        // instead of (bar(r),z) in CG we compute (r,z)
        rho = DotProd(r, z);
 
        if (rho == zero)
          {
            iter.Fail(1, "Cocg breakdown #1");
            break;
          }
 
        if (iter.First())
          Copy(z, p);
        else
          {
            beta = rho / rho_1;
            Mlt(beta, p);
            Add(one, z, p);
          }
        
        // product matrix vector
        iter.Mlt(A, p, q);
 
        delta = DotProd(p, q);
        if (delta == zero)
          {
            iter.Fail(2, "Cocg breakdown #2");
            break;
          }
        alpha = rho / delta;
 
        Add(alpha, p, x);
        Add(-alpha, q, r);
 
        rho_1 = rho;
        ++iter;
      }
 
    return iter.ErrorCode();
  }
 
 
} // end namespace
 
#define SELDON_FILE_ITERATIVE_COCG_CXX
#endif