Ifpack_CrsRiluk.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_CRSRILUK_H_
#define _IFPACK_CRSRILUK_H_
 
#include "Ifpack_ConfigDefs.h"
#include "Ifpack_ScalingType.h"
#include "Ifpack_IlukGraph.h"
#include "Epetra_ConfigDefs.h"
#include "Epetra_CompObject.h"
#include "Epetra_Operator.h"
#include "Epetra_CrsMatrix.h"
#include "Epetra_Object.h"
#include "Epetra_MultiVector.h"
#include "Epetra_Vector.h"
#include "Epetra_Map.h"
class Epetra_Comm;
class Epetra_VbrMatrix;
class Epetra_RowMatrix;
 
#include "Teuchos_RefCountPtr.hpp"
 
namespace Teuchos {
  class ParameterList;
}
 
 
class Ifpack_CrsRiluk: public Epetra_Object, public Epetra_CompObject, public virtual Epetra_Operator {
 
  // Give ostream << function some access to private and protected data/functions.
 
  friend std::ostream& operator << (std::ostream& os, const Ifpack_CrsRiluk& A);
 
 public:
 
  Ifpack_CrsRiluk(const Ifpack_IlukGraph & Graph_in);
 
  Ifpack_CrsRiluk(const Ifpack_CrsRiluk & Matrix);
 
  virtual ~Ifpack_CrsRiluk();
 
 
  int InitValues(const Epetra_CrsMatrix &A);
 
 
#ifndef EPETRA_NO_32BIT_GLOBAL_INDICES // FIXME LONG LONG
  int InitValues(const Epetra_VbrMatrix &A);
#endif
 
  bool ValuesInitialized() const {return(ValuesInitialized_);};
 
  void SetRelaxValue( double RelaxValue) {RelaxValue_ = RelaxValue; return;}
 
  void SetAbsoluteThreshold( double Athresh) {Athresh_ = Athresh; return;}
 
  void SetRelativeThreshold( double Rthresh) {Rthresh_ = Rthresh; return;}
 
  void SetOverlapMode( Epetra_CombineMode OverlapMode) {OverlapMode_ = OverlapMode; return;}
 
  /* This method is only available if the Teuchos package is enabled.
     This method recognizes four parameter names: relax_value,
     absolute_threshold, relative_threshold and overlap_mode. These names are
     case insensitive, and in each case except overlap_mode, the ParameterEntry
     must have type double. For overlap_mode, the ParameterEntry must have
     type Epetra_CombineMode.
  */
  int SetParameters(const Teuchos::ParameterList& parameterlist,
                    bool cerr_warning_if_unused=false);
 
 
  int Factor();
 
  bool Factored() const {return(Factored_);};
 
 
  // Mathematical functions.
 
 
 
  int Solve(bool Trans, const Epetra_MultiVector& X, Epetra_MultiVector& Y) const;
 
 
  int Multiply(bool Trans, const Epetra_MultiVector& X, Epetra_MultiVector& Y) const;
 
 
  int Condest(bool Trans, double & ConditionNumberEstimate) const;
  // Attribute access functions
 
  double GetRelaxValue() {return RelaxValue_;}
 
  double GetAbsoluteThreshold() {return Athresh_;}
 
  double GetRelativeThreshold() {return Rthresh_;}
 
  Epetra_CombineMode GetOverlapMode() {return OverlapMode_;}
 
 
#ifndef EPETRA_NO_32BIT_GLOBAL_INDICES
  int NumGlobalRows() const {return(Graph().NumGlobalRows());};
 
  int NumGlobalCols() const {return(Graph().NumGlobalCols());};
 
  int NumGlobalNonzeros() const {return(L().NumGlobalNonzeros()+U().NumGlobalNonzeros());};
 
  virtual int NumGlobalBlockDiagonals() const {return(Graph().NumGlobalBlockDiagonals());};
#endif
 
  long long NumGlobalRows64() const {return(Graph().NumGlobalRows64());};
  long long NumGlobalCols64() const {return(Graph().NumGlobalCols64());};
  long long NumGlobalNonzeros64() const {return(L().NumGlobalNonzeros64()+U().NumGlobalNonzeros64());};
  virtual long long NumGlobalBlockDiagonals64() const {return(Graph().NumGlobalBlockDiagonals64());};
 
  int NumMyRows() const {return(Graph().NumMyRows());};
 
  int NumMyCols() const {return(Graph().NumMyCols());};
 
  int NumMyNonzeros() const {return(L().NumMyNonzeros()+U().NumMyNonzeros());};
 
  virtual int NumMyBlockDiagonals() const {return(Graph().NumMyBlockDiagonals());};
 
  virtual int NumMyDiagonals() const {return(NumMyDiagonals_);};
 
#ifndef EPETRA_NO_32BIT_GLOBAL_INDICES
  int IndexBase() const {return(Graph().IndexBase());};
#endif
  long long IndexBase64() const {return(Graph().IndexBase64());};
 
  const Ifpack_IlukGraph & Graph() const {return(Graph_);};
 
  const Epetra_CrsMatrix & L() const {return(*L_);};
 
  const Epetra_Vector & D() const {return(*D_);};
 
  const Epetra_CrsMatrix & U() const {return(*U_);};
 
 
    const char * Label() const {return(Epetra_Object::Label());};
 
 
  int SetUseTranspose(bool UseTranspose_in) {UseTranspose_ = UseTranspose_in; return(0);};
 
 
  int Apply(const Epetra_MultiVector& X, Epetra_MultiVector& Y) const {
    return(Multiply(Ifpack_CrsRiluk::UseTranspose(), X, Y));};
 
 
  int ApplyInverse(const Epetra_MultiVector& X, Epetra_MultiVector& Y) const {
    return(Solve(Ifpack_CrsRiluk::UseTranspose(), X, Y));};
 
    double NormInf() const {return(0.0);};
 
    bool HasNormInf() const {return(false);};
 
    bool UseTranspose() const {return(UseTranspose_);};
 
    const Epetra_Map & OperatorDomainMap() const {return(U_->OperatorDomainMap());};
 
    const Epetra_Map & OperatorRangeMap() const{return(L_->OperatorRangeMap());};
 
    const Epetra_Comm & Comm() const{return(Comm_);};
 
 protected:
  void SetFactored(bool Flag) {Factored_ = Flag;};
  void SetValuesInitialized(bool Flag) {ValuesInitialized_ = Flag;};
  bool Allocated() const {return(Allocated_);};
  int SetAllocated(bool Flag) {Allocated_ = Flag; return(0);};
  int BlockGraph2PointGraph(const Epetra_CrsGraph & BG, Epetra_CrsGraph & PG, bool Upper);
 
 private:
 
 
  int AllocateCrs();
  int AllocateVbr();
  int InitAllValues(const Epetra_RowMatrix & A, int MaxNumEntries);
  int BlockMap2PointMap(const Epetra_BlockMap & BlockMap, Teuchos::RefCountPtr<Epetra_Map>* PointMap);
  int GenerateXY(bool Trans,
                 const Epetra_MultiVector& Xin, const Epetra_MultiVector& Yin,
                 Teuchos::RefCountPtr<Epetra_MultiVector>* Xout,
                 Teuchos::RefCountPtr<Epetra_MultiVector>* Yout) const;
  bool UserMatrixIsVbr_;
  bool UserMatrixIsCrs_;
  bool IsOverlapped_;
  const Ifpack_IlukGraph & Graph_;
  Teuchos::RefCountPtr<Epetra_Map> IlukRowMap_;
  Teuchos::RefCountPtr<Epetra_Map> IlukDomainMap_;
  Teuchos::RefCountPtr<Epetra_Map> IlukRangeMap_;
  Teuchos::RefCountPtr<const Epetra_Map> U_DomainMap_;
  Teuchos::RefCountPtr<const Epetra_Map> L_RangeMap_;
  const Epetra_Comm & Comm_;
  Teuchos::RefCountPtr<Epetra_CrsMatrix> L_;
  Teuchos::RefCountPtr<Epetra_CrsMatrix> U_;
  Teuchos::RefCountPtr<Epetra_CrsGraph> L_Graph_;
  Teuchos::RefCountPtr<Epetra_CrsGraph> U_Graph_;
  Teuchos::RefCountPtr<Epetra_Vector> D_;
  bool UseTranspose_;
 
  int NumMyDiagonals_;
  bool Allocated_;
  bool ValuesInitialized_;
  bool Factored_;
  double RelaxValue_;
  double Athresh_;
  double Rthresh_;
  mutable double Condest_;
 
  mutable Teuchos::RefCountPtr<Epetra_MultiVector> OverlapX_;
  mutable Teuchos::RefCountPtr<Epetra_MultiVector> OverlapY_;
  mutable Teuchos::RefCountPtr<Epetra_MultiVector> VbrX_;
  mutable Teuchos::RefCountPtr<Epetra_MultiVector> VbrY_;
  Epetra_CombineMode OverlapMode_;
 
 
};
 
std::ostream& operator << (std::ostream& os, const Ifpack_CrsRiluk& A);
 
#endif /* _IFPACK_CRSRILUK_H_ */