MueLu_UzawaSmoother_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_UZAWASMOOTHER_DECL_HPP_
#define MUELU_UZAWASMOOTHER_DECL_HPP_

#include "MueLu_ConfigDefs.hpp"

#include <Teuchos_ParameterList.hpp>

// Xpetra
#include <Xpetra_MapExtractor_fwd.hpp>
#include <Xpetra_MultiVectorFactory_fwd.hpp>
#include <Xpetra_Matrix_fwd.hpp>

// MueLu
#include "MueLu_UzawaSmoother_fwd.hpp"
#include "MueLu_SmootherPrototype.hpp"
#include "MueLu_FactoryBase_fwd.hpp"
#include "MueLu_FactoryManagerBase_fwd.hpp"
#include "MueLu_SmootherBase_fwd.hpp"

#include "MueLu_SchurComplementFactory_fwd.hpp"
#include "MueLu_FactoryManager_fwd.hpp"

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 UzawaSmoother : public SmootherPrototype<Scalar, LocalOrdinal, GlobalOrdinal, Node> {
  typedef Xpetra::MapExtractor<Scalar, LocalOrdinal, GlobalOrdinal, Node> MapExtractorClass;

#undef MUELU_UZAWASMOOTHER_SHORT
#include "MueLu_UseShortNames.hpp"

 public:


  RCP<const ParameterList> GetValidParameterList() const;

  void DeclareInput(Level &currentLevel) const;

  void SetVelocityPredictionFactoryManager(RCP<FactoryManager> FactManager);

  void SetSchurCompFactoryManager(RCP<FactoryManager> FactManager);

  void AddFactoryManager(RCP<const FactoryManagerBase> FactManager, int pos);




  void Setup(Level &currentLevel);

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

  RCP<SmootherPrototype> Copy() const;



  std::string description() const;

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

  size_t getNodeSmootherComplexity() const;


 private:
  std::string type_ = "Uzawa";

  RCP<const FactoryBase> AFact_;  

  RCP<Matrix> A_ = Teuchos::null;  //<! internal blocked operator "A" generated by AFact_

  RCP<const MapExtractorClass> rangeMapExtractor_;   
  RCP<const MapExtractorClass> domainMapExtractor_;  

  Teuchos::RCP<Vector> diagFinv_;  
  Teuchos::RCP<Matrix> F_;         
  Teuchos::RCP<Matrix> G_;         
  Teuchos::RCP<Matrix> D_;         
  Teuchos::RCP<Matrix> Z_;         

  Teuchos::RCP<SmootherBase> velPredictSmoo_;  
  Teuchos::RCP<SmootherBase> schurCompSmoo_;   

  std::vector<Teuchos::RCP<const FactoryManagerBase> > FactManager_;

};  // class UzawaSmoother

}  // namespace MueLu

#define MUELU_UZAWASMOOTHER_SHORT

#endif /* MUELU_UZAWASMOOTHER_DECL_HPP_ */