Ifpack_Krylov.h

/*@HEADER
// ***********************************************************************
//
//       Ifpack: Object-Oriented Algebraic Preconditioner Package
//                 Copyright (2002) Sandia Corporation
//
// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
// license for use of this work by or on behalf of the U.S. Government.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
// 1. Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
//
// 3. Neither the name of the Corporation nor the names of the
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// Questions? Contact Michael A. Heroux (maherou@sandia.gov)
//
// ***********************************************************************
//@HEADER
*/

#ifndef IFPACK_KRYLOV_H
#define IFPACK_KRYLOV_H

#include "Ifpack_ConfigDefs.h"
#include "Ifpack_Preconditioner.h"
#include "Teuchos_RefCountPtr.hpp"
#include "Ifpack_PointRelaxation.h"
#include "Ifpack_BlockRelaxation.h"
#include "Ifpack_SparseContainer.h"
#include "Ifpack_DenseContainer.h"
#include "Ifpack_Amesos.h"
#ifdef HAVE_IFPACK_AZTECOO
#include "AztecOO.h"
#endif

namespace Teuchos {
  class ParameterList;
}

class Epetra_MultiVector;
class Epetra_Vector;
class Epetra_Map;
class Epetra_Comm;
class Epetra_Time;
class Epetra_Vector;
class Epetra_Operator;
class Epetra_RowMatrix;
#ifdef HAVE_IFPACK_AZTECOO
class AztecOO;
#endif


#ifdef HAVE_IFPACK_EPETRAEXT
class EpetraExt_PointToBlockDiagPermute;
#endif


class Ifpack_Krylov : public Ifpack_Preconditioner {

  typedef double                SC;
  typedef Epetra_MultiVector    MV;
  typedef Epetra_Operator       OP;
  typedef Epetra_RowMatrix      RM;

public:

  Ifpack_Krylov(Epetra_Operator* Matrix);

  Ifpack_Krylov(Epetra_RowMatrix* Matrix);

  virtual ~Ifpack_Krylov() {};


  virtual inline int SetUseTranspose(bool UseTranspose_in)
  {
    UseTranspose_ = UseTranspose_in;
    return(0);
  }




  virtual int Apply(const Epetra_MultiVector& X, Epetra_MultiVector& Y) const;


  virtual int ApplyInverse(const Epetra_MultiVector& X, Epetra_MultiVector& Y) const;

  virtual double NormInf() const
  {
    return(-1.0);
  }


  virtual const char * Label() const
  {
    return(Label_.c_str());
  }

  virtual bool UseTranspose() const
  {
    return(UseTranspose_);
  }

  virtual bool HasNormInf() const
  {
    return(false);
  }

  virtual const Epetra_Comm & Comm() const;

  virtual const Epetra_Map & OperatorDomainMap() const;

  virtual const Epetra_Map & OperatorRangeMap() const;

  virtual int Initialize();

  virtual bool IsInitialized() const
  {
    return(IsInitialized_);
  }

  virtual inline bool IsComputed() const
  {
    return(IsComputed_);
  }

  virtual int Compute();

  virtual const Epetra_RowMatrix& Matrix() const
  {
    return(*Matrix_);
  }

  virtual double Condest(const Ifpack_CondestType CT = Ifpack_Cheap,
                         const int MaxIters = 1550,
                         const double Tol = 1e-9,
                         Epetra_RowMatrix* Matrix_in = 0);

  virtual double Condest() const
  {
    return(Condest_);
  }

  virtual int SetParameters(Teuchos::ParameterList& List);

  virtual std::ostream& Print(std::ostream & os) const;



  virtual int NumInitialize() const
  {
    return(NumInitialize_);
  }

  virtual int NumCompute() const
  {
    return(NumCompute_);
  }

  virtual int NumApplyInverse() const
  {
    return(NumApplyInverse_);
  }

  virtual double InitializeTime() const
  {
    return(InitializeTime_);
  }

  virtual double ComputeTime() const
  {
    return(ComputeTime_);
  }

  virtual double ApplyInverseTime() const
  {
    return(ApplyInverseTime_);
  }

  virtual double InitializeFlops() const
  {
    return(0.0);
  }

  virtual double ComputeFlops() const
  {
    return(ComputeFlops_);
  }

  virtual double ApplyInverseFlops() const
  {
    return(ApplyInverseFlops_);
  }

private:

  // @}
  // @{ \name Private methods

  virtual void SetLabel();

  Ifpack_Krylov(const Ifpack_Krylov& /* rhs */)
  {}

  Ifpack_Krylov& operator=(const Ifpack_Krylov& /* rhs */)
  {
    return(*this);
  }

  // @{ Initializations, timing and flops
  bool IsInitialized_;
  bool IsComputed_;
  int NumInitialize_;
  int NumCompute_;
  mutable int NumApplyInverse_;
  double InitializeTime_;
  double ComputeTime_;
  mutable double ApplyInverseTime_;
  double ComputeFlops_;
  mutable double ApplyInverseFlops_;
  // @}

  // @{ Settings
  int Iterations_;
  double Tolerance_;
  int SolverType_;
  int PreconditionerType_;
  int NumSweeps_;
  int BlockSize_;
  double DampingParameter_;
  bool UseTranspose_;
  double Condest_;
#if 0
  // Unused; commented out to avoid build warnings.

  bool ComputeCondest_;
#endif // 0
  std::string Label_;

  // @{ Other data
  int NumMyRows_;
  int NumMyNonzeros_;
  long long NumGlobalRows_;
  long long NumGlobalNonzeros_;
  Teuchos::RefCountPtr<Epetra_Operator> Operator_;
  Teuchos::RefCountPtr<Epetra_RowMatrix> Matrix_;

  bool IsRowMatrix_;
  Teuchos::RefCountPtr<Epetra_Time> Time_;
  bool ZeroStartingSolution_;

  // Aztec solver
#ifdef HAVE_IFPACK_AZTECOO
  Teuchos::RCP<AztecOO> AztecSolver_;
#endif

  // Inner preconditioner
  Teuchos::RCP<Ifpack_Preconditioner> IfpackPrec_;

};


#endif // IFPACK_KRYLOV_H