Teko_MultPreconditionerFactory.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_MultiPreconditionerFactory_hpp__
#define __Teko_MultiPreconditionerFactory_hpp__

#include "Teko_BlockPreconditionerFactory.hpp"
#include "Teko_Utilities.hpp"
#include "Teko_BlockImplicitLinearOp.hpp"

namespace Teko {

class MultPrecondState : public Teko::BlockPreconditionerState {
 public:
  MultPrecondState() {}

  Teuchos::RCP<BlockPreconditionerState> StateOne_;
  Teuchos::RCP<BlockPreconditionerState> StateTwo_;
};

class MultPrecsLinearOp : public Teko::BlockImplicitLinearOp {
 public:
  MultPrecsLinearOp(const Teko::LinearOp &A, const Teko::LinearOp &M1, const Teko::LinearOp &M2)
      : A_(A), M1_(M1), M2_(M2) {}

  virtual Teko::VectorSpace range() const { return M1_->range(); }
  virtual Teko::VectorSpace domain() const { return M1_->domain(); }
  virtual void implicitApply(const Teko::BlockedMultiVector &r, Teko::BlockedMultiVector &y,
                             const double alpha = 1.0, const double beta = 0.0) const;

 protected:
  using Teko::BlockImplicitLinearOp::implicitApply;

  Teko::LinearOp A_, M1_, M2_;

 private:
  // hide me!
  MultPrecsLinearOp();
  MultPrecsLinearOp(const MultPrecsLinearOp &);
};

class MultPreconditionerFactory : public Teko::BlockPreconditionerFactory {
 public:
  MultPreconditionerFactory(
      const Teuchos::RCP<const Teko::BlockPreconditionerFactory> &FirstFactory,
      const Teuchos::RCP<const Teko::BlockPreconditionerFactory> &SecondFactory);

  MultPreconditionerFactory();

  Teko::LinearOp buildPreconditionerOperator(Teko::BlockedLinearOp &blo,
                                             Teko::BlockPreconditionerState &state) const;

  virtual Teuchos::RCP<Teko::PreconditionerState> buildPreconditionerState() const;

 protected:
  using Teko::BlockPreconditionerFactory::buildPreconditionerOperator;

  // class members
  Teuchos::RCP<const Teko::BlockPreconditionerFactory> FirstFactory_;
  Teuchos::RCP<const Teko::BlockPreconditionerFactory> SecondFactory_;

  virtual void initializeFromParameterList(const Teuchos::ParameterList &pl);
};

}  // end namespace Teko

#endif