Teko_TpetraBlockPreconditioner.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_TpetraBlockPreconditioner_hpp__
#define __Teko_TpetraBlockPreconditioner_hpp__

// #include "Teko_BlockPreconditionerFactory.hpp"
#include "Teko_PreconditionerFactory.hpp"
#include "Teko_TpetraInverseOpWrapper.hpp"
#include "Teko_ConfigDefs.hpp"

namespace Teko {
namespace TpetraHelpers {

class TpetraBlockPreconditioner : public TpetraInverseOpWrapper {
 public:
  TpetraBlockPreconditioner(const Teuchos::RCP<const PreconditionerFactory> &bfp);

  virtual void initPreconditioner(bool clearOld = false);

  virtual void buildPreconditioner(const Teuchos::RCP<const Tpetra::Operator<ST, LO, GO, NT> > &A,
                                   bool clear = true);

  virtual void buildPreconditioner(const Teuchos::RCP<const Tpetra::Operator<ST, LO, GO, NT> > &A,
                                   const Tpetra::MultiVector<ST, LO, GO, NT> &mv,
                                   bool clear = true);

  virtual void rebuildPreconditioner(
      const Teuchos::RCP<const Tpetra::Operator<ST, LO, GO, NT> > &A);

  virtual void rebuildPreconditioner(const Teuchos::RCP<const Tpetra::Operator<ST, LO, GO, NT> > &A,
                                     const Tpetra::MultiVector<ST, LO, GO, NT> &mv);

  virtual Teuchos::RCP<PreconditionerState> getPreconditionerState();

  virtual Teuchos::RCP<const PreconditionerState> getPreconditionerState() const;

 protected:
  Teuchos::RCP<const Thyra::LinearOpBase<ST> > extractLinearOp(
      const Teuchos::RCP<const Tpetra::Operator<ST, LO, GO, NT> > &A) const;
  Teuchos::RCP<const MappingStrategy> extractMappingStrategy(
      const Teuchos::RCP<const Tpetra::Operator<ST, LO, GO, NT> > &A) const;

  TpetraBlockPreconditioner();
  TpetraBlockPreconditioner(const TpetraBlockPreconditioner &);

  // Teuchos::RCP<const BlockPreconditionerFactory> preconFactory_;
  Teuchos::RCP<const PreconditionerFactory> preconFactory_;
  Teuchos::RCP<Thyra::PreconditionerBase<ST> > preconObj_;
  bool firstBuildComplete_;
};

}  // end namespace TpetraHelpers
}  // end namespace Teko

#endif