PlayaEpetraMatrix.hpp

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

#ifndef PLAYA_EPETRAMATRIX_HPP
#define PLAYA_EPETRAMATRIX_HPP

#include "PlayaLoadableMatrix.hpp"
#include "PlayaLinearOperatorDecl.hpp"
#include "PlayaLinearOpWithSpacesDecl.hpp"
#include "PlayaRowAccessibleOp.hpp"
#include "PlayaPrintable.hpp"
#include "PlayaILUFactorizableOp.hpp"
#include "Epetra_CrsMatrix.h"
#include "PlayaLinearOpWithSpacesDecl.hpp"

namespace Playa
{
using namespace Teuchos;



class EpetraMatrix : public LinearOpWithSpaces<double>,
                     public LoadableMatrix<double>,
                     public RowAccessibleOp<double>,
                     public Printable,
                     public ILUFactorizableOp<double>
{
public:

  EpetraMatrix(const Epetra_CrsGraph& graph,
    const VectorSpace<double>& domain,
    const VectorSpace<double>& range);

  EpetraMatrix(const RCP<Epetra_CrsMatrix>& mat,
    const VectorSpace<double>& domain,
    const VectorSpace<double>& range);

  virtual void apply(
    Teuchos::ETransp applyType,
    const Vector<double>& in,
    Vector<double> out) const ;
  


  virtual void addToRow(int globalRowIndex,
    int nElemsToInsert,
    const int* globalColumnIndices,
    const double* elementValues) ;


  virtual void addToElementBatch(int numRows, 
    int rowBlockSize,
    const int* globalRowIndices,
    int numColumnsPerRow,
    const int* globalColumnIndices,
    const double* values,
    const int* skipRow);

  virtual void zero() ;
  

  virtual void getILUKPreconditioner(int fillLevels,
    int overlapFill,
    double relaxationValue,
    double relativeThreshold,
    double absoluteThreshold,
    LeftOrRight leftOrRight,
    Preconditioner<double>& rtn) const ;

  
  void getRow(const int& row, 
    Teuchos::Array<int>& indices, 
    Teuchos::Array<double>& values) const;
  

  virtual void print(std::ostream& os) const ;

  std::ostream& describe(
    std::ostream                         &out
    ,const Teuchos::EVerbosityLevel      verbLevel
    ,const std::string                   leadingIndent
    , const std::string                   indentSpacer
    ) const 
    {
      out << leadingIndent << indentSpacer << this->description() << std::endl;
      return out;
    }
  std::string description() const ;

  static Epetra_CrsMatrix& getConcrete(const LinearOperator<double>& A);

  static RCP<const Epetra_CrsMatrix> getConcretePtr(const LinearOperator<double>& A);

  const Epetra_CrsMatrix* crsMatrix() const ;

private:

  Epetra_CrsMatrix* crsMatrix();

  RCP<Epetra_CrsMatrix> matrix_;

  const Epetra_Map& getRangeMap() const;
  const Epetra_Map& getDomainMap() const;
};

  

}

#endif