Stokhos_EpetraMultiVectorOperator.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//                           Stokhos Package
//
// Copyright 2009 NTESS and the Stokhos contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef STOKHOS_EPETRA_MULTIVECTOR_OPERATOR_HPP
#define STOKHOS_EPETRA_MULTIVECTOR_OPERATOR_HPP

#include "Teuchos_RCP.hpp"
#include "Epetra_Operator.h"
#include "Epetra_MultiVector.h"

namespace Stokhos {
    
  class EpetraMultiVectorOperator : public Epetra_Operator {
      
  public:

    EpetraMultiVectorOperator(
      const Teuchos::RCP<const Epetra_MultiVector>& multi_vec,
      bool is_multi_vec_transposed);

    EpetraMultiVectorOperator(
      const Teuchos::RCP<Epetra_MultiVector>& multi_vec,
      bool is_multi_vec_transposed);
    
    virtual ~EpetraMultiVectorOperator();

    Teuchos::RCP<const Epetra_MultiVector>
    getMultiVector() const { return multi_vec; }

    Teuchos::RCP<Epetra_MultiVector>
    getMultiVector() { return nonconst_multi_vec; }
    
    virtual int SetUseTranspose(bool UseTranspose);
    
    virtual int Apply(const Epetra_MultiVector& Input, 
                      Epetra_MultiVector& Result) const;

    virtual int ApplyInverse(const Epetra_MultiVector& X, 
                             Epetra_MultiVector& Y) const;
    
    virtual double NormInf() const;
    
    virtual const char* Label () const;
  
    virtual bool UseTranspose() const;
    
    virtual bool HasNormInf() const;

    virtual const Epetra_Comm & Comm() const;

    virtual const Epetra_Map& OperatorDomainMap () const;

    virtual const Epetra_Map& OperatorRangeMap () const;

  private:
    
    EpetraMultiVectorOperator(const EpetraMultiVectorOperator&);
    
    EpetraMultiVectorOperator& operator=(const EpetraMultiVectorOperator&);
    
  protected:
    
    std::string label;
    
    Teuchos::RCP<const Epetra_MultiVector> multi_vec;

    Teuchos::RCP<Epetra_MultiVector> nonconst_multi_vec;
    
    bool is_multi_vec_transposed;

    bool useTranspose;

    Teuchos::RCP<Epetra_Map> domain_map;

  }; // class EpetraMultiVectorOperator
  
} // namespace Stokhos

#endif // STOKHOS_EPETRA_MULTIVECTOR_OPERATOR_HPP