PlayaAztecSolver.hpp

/* @HEADER@ */
//   
/* @HEADER@ */

#ifndef PLAYA_AZTECSOLVER_HPP
#define PLAYA_AZTECSOLVER_HPP

#include "PlayaDefs.hpp"
#include "PlayaLinearSolverBaseDecl.hpp"
#include "PlayaHandleable.hpp"
#include "PlayaPrintable.hpp"
#include "Teuchos_Describable.hpp"

#include "Teuchos_Array.hpp"
#include "Teuchos_RefCountPtr.hpp"
#include "Teuchos_ParameterList.hpp"
#include <map>

#include "AztecOO.h"

#define AZ_recursive_iterate 10001

#define HAVE_ML


namespace Playa
{
  using namespace Teuchos;

  class AztecSolver : public LinearSolverBase<double>,
                      public Playa::Handleable<LinearSolverBase<double> >,
                      public Printable,
                      public Describable
  {
  public:
    AztecSolver(const Teuchos::map<int, int>& aztecOptions,
                const Teuchos::map<int, double>& aztecParameters);

    AztecSolver(const Teuchos::ParameterList& params);

    virtual ~AztecSolver(){;}

    virtual void updateTolerance(const double& tol);


    void setUserPrec(const LinearOperator<double>& P,
         const LinearSolver<double>& pSolver);

    void print(std::ostream& os) const 
    {
      os << description() << std::endl;
    }
    
    std::string description() const {return "AztecSolver";}

    

    virtual SolverState<double> solve(const LinearOperator<double>& op,
                                      const Vector<double>& rhs,
                                      Vector<double>& soln) const ;

    virtual RCP<LinearSolverBase<double> > getRcp() 
    {return rcp(this);}


  protected:

  private:
    
    void setupML(Epetra_RowMatrix* A) const ;

    mutable Array<int> options_;

    mutable Array<double> parameters_;

    bool useML_;

    bool useIfpack_;

    bool useUserPrec_;

    bool aztec_recursive_iterate_;

    mutable ParameterList precParams_;

    mutable RCP<Epetra_Operator> userPrec_;

    mutable Array<double> aztec_status;

    mutable Array<int> aztec_proc_config;


    static Teuchos::map<string,int>& paramMap() 
    {static Teuchos::map<string,int> rtn; return rtn;}
    
    static void initParamMap();

    
  };
  
}

#endif