MueLu_SparseConstraint_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_SPARSECONSTRAINT_DECL_HPP
#define MUELU_SPARSECONSTRAINT_DECL_HPP
 
#include "Teuchos_ScalarTraits.hpp"
 
#include <Xpetra_MultiVector_fwd.hpp>
#include <Xpetra_Matrix.hpp>
#include <Xpetra_CrsGraph_fwd.hpp>
#include <Xpetra_MapFactory_fwd.hpp>
 
#include "MueLu_ConfigDefs.hpp"
#include <MueLu_Utilities_fwd.hpp>
#include "MueLu_Constraint.hpp"
 
namespace MueLu {
 
template <class Scalar        = DefaultScalar,
          class LocalOrdinal  = DefaultLocalOrdinal,
          class GlobalOrdinal = DefaultGlobalOrdinal,
          class Node          = DefaultNode>
class SparseConstraint
  : public Constraint<Scalar, LocalOrdinal, GlobalOrdinal, Node> {
#undef MUELU_SPARSECONSTRAINT_SHORT
#include "MueLu_UseShortNames.hpp"
 public:
  using MagnitudeType = typename Teuchos::ScalarTraits<Scalar>::magnitudeType;
 
 
 
  SparseConstraint() = default;
 
  SparseConstraint(const RCP<Matrix>& P_nodal, const RCP<Matrix>& D, const RCP<Matrix>& Dc, RCP<const CrsGraph> Ppattern, const std::string& solverType);
 
 
  void Setup();
 
  typename CrsGraph::local_graph_type FindBlocks(RCP<const CrsGraph>& /*XXt*/) override;
 
  MagnitudeType ResidualNorm(RCP<const Matrix> P) const override;
 
  void AssignMatrixEntriesToConstraintVector(const Matrix& A, MultiVector& vecC) const;
 
 private:
  RCP<Matrix> P_nodal_;
 
  RCP<Matrix> D_;
 
  RCP<Matrix> Dc_;
  RCP<Matrix> big_Dc_;
 
  RCP<const CrsGraph> RHS_pattern_;
 
  Kokkos::View<bool*> block_is_singular_;
};
 
}  // namespace MueLu
 
#define MUELU_SPARSECONSTRAINT_SHORT
#endif