MueLu_TrilinosSmoother_decl.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//        MueLu: A package for multigrid based preconditioning
//
// Copyright 2012 NTESS and the MueLu contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef MUELU_TRILINOSSMOOTHER_DECL_HPP
#define MUELU_TRILINOSSMOOTHER_DECL_HPP

#include <Teuchos_ParameterList.hpp>

#include "MueLu_ConfigDefs.hpp"
#include "MueLu_TrilinosSmoother_fwd.hpp"
#include "MueLu_SmootherPrototype.hpp"

#include "MueLu_FactoryBase_fwd.hpp"
#include "MueLu_IfpackSmoother_fwd.hpp"

#if defined(HAVE_MUELU_IFPACK2)
#include "MueLu_Ifpack2Smoother_fwd.hpp"
#include "MueLu_Ifpack2Smoother.hpp"
#endif

#if defined(HAVE_MUELU_BELOS)
#include "MueLu_BelosSmoother_fwd.hpp"
#endif

#if defined(HAVE_MUELU_STRATIMIKOS)
#include "MueLu_StratimikosSmoother_fwd.hpp"
#endif

// Note: TrilinosSmoother is a SmootherPrototype that cannot be turned into a smoother using Setup().
//       When this prototype is cloned using Copy(), the clone is an Ifpack or an Ifpack2 smoother.
//       The clone can be used as a smoother after calling Setup().

namespace MueLu {

template <class Scalar        = SmootherPrototype<>::scalar_type,
          class LocalOrdinal  = typename SmootherPrototype<Scalar>::local_ordinal_type,
          class GlobalOrdinal = typename SmootherPrototype<Scalar, LocalOrdinal>::global_ordinal_type,
          class Node          = typename SmootherPrototype<Scalar, LocalOrdinal, GlobalOrdinal>::node_type>
class TrilinosSmoother : public SmootherPrototype<Scalar, LocalOrdinal, GlobalOrdinal, Node> {
#undef MUELU_TRILINOSSMOOTHER_SHORT
#include "MueLu_UseShortNames.hpp"

 public:


  TrilinosSmoother(const std::string& type = "", const Teuchos::ParameterList& paramList = Teuchos::ParameterList(), const LO& overlap = 0);

  virtual ~TrilinosSmoother() {}




  void DeclareInput(Level& currentLevel) const;




  void Setup(Level& currentLevel);

  void Apply(MultiVector& X, const MultiVector& B, bool InitialGuessIsZero = false) const;


  void SetFactory(const std::string& varName, const RCP<const FactoryBase>& factory);

  RCP<SmootherPrototype> Copy() const;

#ifndef _MSC_VER
  template <class Scalar2, class LocalOrdinal2, class GlobalOrdinal2, class Node2>
  friend class TrilinosSmoother;
#endif

  // As a temporary solution.
  // See https://software.sandia.gov/bugzilla/show_bug.cgi?id=5283#c5 for what I proposed to do
  static std::string Ifpack2ToIfpack1Type(const std::string& type);

  // As a temporary solution.
  // See https://software.sandia.gov/bugzilla/show_bug.cgi?id=5283#c5 for what I proposed to do
  /* TODO:
    ifpackList.set("type", "Chebyshev");
    ifpackList.set("chebyshev: degree", (int) 1);
    ifpackList.set("chebyshev: max eigenvalue", (double) 2.0);
    ifpackList.set("chebyshev: min eigenvalue", (double) 1.0);
    ifpackList.set("chebyshev: zero starting solution", false);
  */
  static Teuchos::ParameterList Ifpack2ToIfpack1Param(const Teuchos::ParameterList& ifpack2List);



  std::string description() const;

  // using MueLu::Describable::describe; // overloading, not hiding
  // void describe(Teuchos::FancyOStream &out, const VerbLevel verbLevel = Default) const {
  void print(Teuchos::FancyOStream& out, const VerbLevel verbLevel = Default) const;

  RCP<SmootherPrototype> getSmoother() { return s_; }

  size_t getNodeSmootherComplexity() const { return s_->getNodeSmootherComplexity(); }


 private:
  std::string type_;

  LO overlap_;

  RCP<FactoryBase> AFact_;

  //
  // Underlying Smoother
  //

  RCP<SmootherPrototype> sEpetra_, sTpetra_, sBelos_, sStratimikos_;
  mutable RCP<SmootherPrototype> s_;

  // Records for the case if something goes wrong
  bool triedEpetra_, triedTpetra_, triedBelos_, triedStratimikos_;
  std::string errorEpetra_, errorTpetra_, errorBelos_, errorStratimikos_;

};  // class TrilinosSmoother

}  // namespace MueLu

#define MUELU_TRILINOSSMOOTHER_SHORT
#endif  // MUELU_TRILINOSSMOOTHER_DECL_HPP