Teko_DiagonalPreconditionerOp.hpp

// @HEADER
// *****************************************************************************
//      Teko: A package for block and physics based preconditioning
//
// Copyright 2010 NTESS and the Teko contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef __Teko_DiagonalPreconditionerOp_hpp__
#define __Teko_DiagonalPreconditionerOp_hpp__
#include "Teko_Utilities.hpp"
 
#include "Teko_ImplicitLinearOp.hpp"
 
namespace Tpetra {
namespace Ext {
template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
class PointToBlockDiagPermute;
}
}  // namespace Tpetra
 
namespace Teko {
 
class DiagonalPreconditionerOp : public ImplicitLinearOp {
 public:
  DiagonalPreconditionerOp(Teuchos::RCP<Tpetra::Ext::PointToBlockDiagPermute<ST, LO, GO, NT>> BDP,
                           const VectorSpace range, const VectorSpace domain);
 
  virtual VectorSpace range() const { return range_; }
 
  virtual VectorSpace domain() const { return domain_; }
 
  virtual void implicitApply(const MultiVector &x, MultiVector &y, const double alpha = 1.0,
                             const double beta = 0.0) const;
 
  virtual void describe(Teuchos::FancyOStream &out_arg,
                        const Teuchos::EVerbosityLevel verbLevel) const;
 
  Teuchos::RCP<Tpetra::Ext::PointToBlockDiagPermute<ST, LO, GO, NT>> get_BDP() const {
    return BDP_;
  }
 
 private:
  Teuchos::RCP<Tpetra::Ext::PointToBlockDiagPermute<ST, LO, GO, NT>> BDP_;
  const VectorSpace range_, domain_;
};
 
}  // end namespace Teko
 
#endif