Teko_ProbingPreconditionerFactory.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_ProbingPreconditionerFactory_hpp__
#define __Teko_ProbingPreconditionerFactory_hpp__
 
#include "Teko_PreconditionerState.hpp"
#include "Teko_PreconditionerFactory.hpp"
#include "Teko_ConfigDefs.hpp"
 
#include "Tpetra_CrsGraph.hpp"
#include "Tpetra_CrsMatrix.hpp"
 
namespace Teko {
 
class ProbingPreconditionerFactory : public virtual Teko::PreconditionerFactory {
 public:
 
 
  ProbingPreconditionerFactory();
 
 
  LinearOp buildPreconditionerOperator(LinearOp& lo, PreconditionerState& state) const override;
 
  void initializeFromParameterList(const Teuchos::ParameterList& pl) override;
 
  void setGraphOperator(const Teko::LinearOp& graphOp);
  void setGraph(const Teuchos::RCP<const Tpetra::CrsGraph<LO, GO, NT> >& graph);
 
  void setInverseFactory(const Teuchos::RCP<Teko::InverseFactory>& invFactory) {
    invFactory_ = invFactory;
  }
 
 protected:
  Teuchos::RCP<const Tpetra::CrsGraph<LO, GO, NT> > graph_;
  Teuchos::RCP<Teko::InverseFactory> invFactory_;
 
 private:
  Teuchos::RCP<Tpetra::CrsMatrix<ST, LO, GO, NT> > probe(const LinearOp& lo) const;
};
 
}  // end namespace Teko
 
#endif