Thyra_MultiVectorPreconditioner.hpp

Go to the documentation of this file.

// @HEADER
// ****************************************************************************
//                Tempus: Copyright (2017) Sandia Corporation
//
// Distributed under BSD 3-clause license (See accompanying file Copyright.txt)
// ****************************************************************************
// @HEADER

#ifndef Thyra_MultiVectorPreconditioner_hpp
#define Thyra_MultiVectorPreconditioner_hpp

#include "Thyra_PreconditionerBase.hpp"
#include "Teuchos_ConstNonconstObjectContainer.hpp"
#include "Thyra_MultiVectorLinearOp.hpp"
#include "Thyra_DefaultMultiVectorProductVectorSpace.hpp"

namespace Thyra {

template <class Scalar>
class MultiVectorPreconditioner : virtual public PreconditionerBase<Scalar> {
 public:

  MultiVectorPreconditioner() {}

  void nonconstInitialize(
      const RCP<PreconditionerBase<Scalar> > &prec,
      const RCP<const DefaultMultiVectorProductVectorSpace<Scalar> >
          &multiVecRange,
      const RCP<const DefaultMultiVectorProductVectorSpace<Scalar> >
          &multiVecDomain)
  {
    validateInitialize(prec, multiVecRange, multiVecDomain);
    prec_           = prec;
    multiVecRange_  = multiVecRange;
    multiVecDomain_ = multiVecDomain;
  }

  void initialize(const RCP<const PreconditionerBase<Scalar> > &prec,
                  const RCP<const DefaultMultiVectorProductVectorSpace<Scalar> >
                      &multiVecRange,
                  const RCP<const DefaultMultiVectorProductVectorSpace<Scalar> >
                      &multiVecDomain)
  {
    validateInitialize(prec, multiVecRange, multiVecDomain);
    prec_           = prec;
    multiVecRange_  = multiVecRange;
    multiVecDomain_ = multiVecDomain;
  }

  RCP<PreconditionerBase<Scalar> > getNonconstPreconditioner()
  {
    return prec_.getNonconstObj();
  }

  RCP<const PreconditionerBase<Scalar> > getPreconditioner() const
  {
    return prec_.getConstObj();
  }

  void uninitialize()
  {
    prec_.uninitialize();
    multiVecRange_  = Teuchos::null;
    multiVecDomain_ = Teuchos::null;
  }



  bool isLeftPrecOpConst() const
  {
    return prec_.getConstObj()->isLeftPrecOpConst();
  }

  Teuchos::RCP<LinearOpBase<Scalar> > getNonconstLeftPrecOp()
  {
    return nonconstMultiVectorLinearOp(
        prec_.getNonconstObj()->getNonconstLeftPrecOp(), multiVecRange_,
        multiVecDomain_);
  }

  Teuchos::RCP<const LinearOpBase<Scalar> > getLeftPrecOp() const
  {
    return multiVectorLinearOp(prec_.getConstObj()->getLeftPrecOp(),
                               multiVecRange_, multiVecDomain_);
  }

  bool isRightPrecOpConst() const
  {
    return prec_.getConstObj()->isRightPrecOpConst();
  }

  Teuchos::RCP<LinearOpBase<Scalar> > getNonconstRightPrecOp()
  {
    return nonconstMultiVectorLinearOp(
        prec_.getNonconstObj()->getNonconstRightPrecOp(), multiVecRange_,
        multiVecDomain_);
  }

  Teuchos::RCP<const LinearOpBase<Scalar> > getRightPrecOp() const
  {
    return multiVectorLinearOp(prec_.getConstObj()->getRightPrecOp(),
                               multiVecRange_, multiVecDomain_);
  }

  bool isUnspecifiedPrecOpConst() const
  {
    return prec_.getConstObj()->isUnspecifiedPrecOpConst();
  }

  Teuchos::RCP<LinearOpBase<Scalar> > getNonconstUnspecifiedPrecOp()
  {
    return nonconstMultiVectorLinearOp(
        prec_.getNonconstObj()->getNonconstUnspecifiedPrecOp(), multiVecRange_,
        multiVecDomain_);
  }

  Teuchos::RCP<const LinearOpBase<Scalar> > getUnspecifiedPrecOp() const
  {
    return multiVectorLinearOp(prec_.getNonconstObj()->getUnspecifiedPrecOp(),
                               multiVecRange_, multiVecDomain_);
  }


 private:
  // //////////////////////////////
  // Private types

  typedef Teuchos::ConstNonconstObjectContainer<PreconditionerBase<Scalar> >
      CNPB;

  // //////////////////////////////
  // Private data members

  CNPB prec_;
  RCP<const DefaultMultiVectorProductVectorSpace<Scalar> > multiVecRange_;
  RCP<const DefaultMultiVectorProductVectorSpace<Scalar> > multiVecDomain_;

  // //////////////////////////////
  // Private member functions

  static void validateInitialize(
      const RCP<const PreconditionerBase<Scalar> > &prec,
      const RCP<const DefaultMultiVectorProductVectorSpace<Scalar> >
          &multiVecRange,
      const RCP<const DefaultMultiVectorProductVectorSpace<Scalar> >
          &multiVecDomain)
  {
#ifdef TEUCHOS_DEBUG
    TEUCHOS_TEST_FOR_EXCEPT(is_null(prec));
    TEUCHOS_TEST_FOR_EXCEPT(is_null(multiVecRange));
    TEUCHOS_TEST_FOR_EXCEPT(is_null(multiVecDomain));
    TEUCHOS_TEST_FOR_EXCEPT(multiVecRange->numBlocks() !=
                            multiVecDomain->numBlocks());
#else
    (void)prec;
    (void)multiVecRange;
    (void)multiVecDomain;
#endif
  }
};

template <class Scalar>
RCP<MultiVectorPreconditioner<Scalar> > multiVectorPreconditioner()
{
  return Teuchos::rcp(new MultiVectorPreconditioner<Scalar>());
}

template <class Scalar>
RCP<MultiVectorPreconditioner<Scalar> > nonconstMultiVectorPreconditioner(
    const RCP<PreconditionerBase<Scalar> > &prec,
    const RCP<const DefaultMultiVectorProductVectorSpace<Scalar> >
        &multiVecRange,
    const RCP<const DefaultMultiVectorProductVectorSpace<Scalar> >
        &multiVecDomain)
{
  RCP<MultiVectorPreconditioner<Scalar> > mvprec =
      Teuchos::rcp(new MultiVectorPreconditioner<Scalar>());
  mvprec->nonconstInitialize(prec, multiVecRange, multiVecDomain);
  return mvprec;
}

template <class Scalar>
RCP<MultiVectorPreconditioner<Scalar> > multiVectorPreconditioner(
    const RCP<const PreconditionerBase<Scalar> > &prec,
    const RCP<const DefaultMultiVectorProductVectorSpace<Scalar> >
        &multiVecRange,
    const RCP<const DefaultMultiVectorProductVectorSpace<Scalar> >
        &multiVecDomain)
{
  RCP<MultiVectorPreconditioner<Scalar> > mvprec =
      Teuchos::rcp(new MultiVectorPreconditioner<Scalar>());
  mvprec->initialize(prec, multiVecRange, multiVecDomain);
  return mvprec;
}

}  // end namespace Thyra

#endif