Ifpack_Chebyshev.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.
//
// This library 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.
//
// This library 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 this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
// USA
// Questions? Contact Michael A. Heroux (maherou@sandia.gov)
//
// ***********************************************************************
//@HEADER
*/

#ifndef IFPACK_CHEBYSHEV_H
#define IFPACK_CHEBYSHEV_H

#include "Ifpack_ConfigDefs.h"
#include "Ifpack_Preconditioner.h"
#include "Teuchos_RefCountPtr.hpp"

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_EPETRAEXT
class EpetraExt_PointToBlockDiagPermute;
#endif


class Ifpack_Chebyshev : public Ifpack_Preconditioner {

public:



  Ifpack_Chebyshev(const Epetra_Operator* Matrix);


  Ifpack_Chebyshev(const Epetra_RowMatrix* Matrix);

  virtual ~Ifpack_Chebyshev() {};


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




  virtual inline 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 double GetLambdaMax(){return LambdaMax_;}

  virtual double GetLambdaMin(){return LambdaMin_;}
  
 

  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 ostream& Print(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_);
  }

  // @}
  // @{ \name Utility methods

  static int PowerMethod(const Epetra_Operator& Operator,
                         const Epetra_Vector& InvPointDiagonal,
                         const int MaximumIterations, 
                         double& LambdaMax);

  static int CG(const Epetra_Operator& Operator, 
                const Epetra_Vector& InvPointDiagonal, 
                const int MaximumIterations, 
                double& lambda_min, double& lambda_max);
  
#ifdef HAVE_IFPACK_EPETRAEXT

  // WARNING: This only works in Block Mode.
  int CG(const int MaximumIterations, 
         double& lambda_min, double& lambda_max);
  // WARNING: This only works in Block Mode.
  int PowerMethod(const int MaximumIterations,double& lambda_max);
#endif
private:
  
  // @}
  // @{ \name Private methods
  
  virtual void SetLabel();

  Ifpack_Chebyshev(const Ifpack_Chebyshev& rhs)
  {}
  
  Ifpack_Chebyshev& operator=(const Ifpack_Chebyshev& 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 PolyDegree_;
  bool UseTranspose_;
  double Condest_;
  bool ComputeCondest_;
  double EigRatio_;
  int EigMaxIters_;  
  string Label_;
  double LambdaMin_;
  double LambdaMax_;
  double MinDiagonalValue_;
  // @}

  // @{ Other data
  int NumMyRows_;
  int NumMyNonzeros_;
  int NumGlobalRows_;
  int NumGlobalNonzeros_;
  Teuchos::RefCountPtr<const Epetra_Operator> Operator_;
  Teuchos::RefCountPtr<const Epetra_RowMatrix> Matrix_;
  mutable Teuchos::RefCountPtr<Epetra_Vector> InvDiagonal_;
  bool UseBlockMode_;
#ifdef HAVE_IFPACK_EPETRAEXT

  Teuchos::ParameterList BlockList_;
  Teuchos::RefCountPtr<EpetraExt_PointToBlockDiagPermute> InvBlockDiagonal_;
#endif


  bool SolveNormalEquations_;

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

  // @}

};


#endif // IFPACK_CHEBYSHEV_H