aztecoo_solver.h

Go to the documentation of this file.

// This file is part of HermesCommon
//
// Copyright (c) 2009 hp-FEM group at the University of Nevada, Reno (UNR).
// Email: hpfem-group@unr.edu, home page: http://www.hpfem.org/.
//
// Hermes2D is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published
// by the Free Software Foundation; either version 2 of the License,
// or (at your option) any later version.
//
// Hermes2D 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Hermes2D; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
#ifndef __HERMES_COMMON_AZTECOO_SOLVER_H_
#define __HERMES_COMMON_AZTECOO_SOLVER_H_
#include "../../config.h"
#ifdef HAVE_AZTECOO
#include "epetra.h"
#include "solvers/linear_matrix_solver.h"
#include "precond_ifpack.h"
#include "precond_ml.h"
#include <AztecOO.h>
#include <EpetraExt_Permutation.h>

namespace Hermes
{
  namespace Solvers
  {
    template <typename Scalar>
    class HERMES_API AztecOOSolver : public IterSolver < Scalar >
    {
    public:
      AztecOOSolver(EpetraMatrix<Scalar> *m, EpetraVector<Scalar> *rhs);
      void free();
      virtual ~AztecOOSolver();

      void set_solver(const char *solver);
      void set_tolerance(double tol);
      void set_max_iters(int iters);

      virtual void set_precond(const char *name);

      virtual void set_precond(Precond<Scalar> *pc);

      virtual void set_reuse_scheme(MatrixStructureReuseScheme reuse_scheme);

      virtual void use_node_wise_ordering(unsigned int num_pdes);
      virtual void use_equations_wise_ordering();

      virtual void solve();
      virtual void solve(Scalar* initial_guess);

      virtual int get_matrix_size();

      virtual int get_num_iters();
      virtual double get_residual_norm();

      void set_option(int option, int value);

      void set_param(int param, double value);

    protected:
      AztecOO aztec;
      EpetraMatrix<Scalar> *m;
      EpetraVector<Scalar> *rhs;

      int *row_perm;
      int *col_perm;
      EpetraExt::Permutation<Epetra_CrsMatrix> *P;
      EpetraExt::Permutation<Epetra_CrsMatrix> *Q;
      EpetraMatrix<Scalar> *final_matrix;

      void create_permutation_vectors();
      void free_permutation_data();

      EpetraPrecond<Scalar> *pc;

      template<typename T> friend LinearMatrixSolver<T>* create_linear_solver(Matrix<T>* matrix, Vector<T>* rhs, bool use_direct_solver);
    };
  }
}
#endif
#endif