MueLu_Constraint_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_CONSTRAINT_DECL_HPP
#define MUELU_CONSTRAINT_DECL_HPP

#include <Teuchos_SerialDenseMatrix.hpp>

#include <Xpetra_MultiVector_fwd.hpp>
#include <Xpetra_Matrix_fwd.hpp>
#include <Xpetra_CrsGraph_fwd.hpp>

#include "MueLu_ConfigDefs.hpp"
#include "MueLu_BaseClass.hpp"

namespace MueLu {


template <class Scalar        = DefaultScalar,
          class LocalOrdinal  = DefaultLocalOrdinal,
          class GlobalOrdinal = DefaultGlobalOrdinal,
          class Node          = DefaultNode>
class Constraint : public BaseClass {
#undef MUELU_CONSTRAINT_SHORT
#include "MueLu_UseShortNames.hpp"
 public:
  using MagnitudeType = typename Teuchos::ScalarTraits<Scalar>::magnitudeType;



  void Setup(const RCP<MultiVector>& B, const RCP<MultiVector>& Bc, RCP<const CrsGraph> Ppattern);

  MagnitudeType ResidualNorm(const RCP<const Matrix> P) const;




  void Apply(const Matrix& P, Matrix& Projected) const;


  RCP<const CrsGraph> GetPattern() const {
    return Ppattern_;
  }

 private:
  RCP<MultiVector> B_;
  RCP<MultiVector> Bc_;
  RCP<MultiVector> X_;                                    
  RCP<const CrsGraph> Ppattern_;                          
  ArrayRCP<Teuchos::SerialDenseMatrix<LO, SC> > XXtInv_;  
};

}  // namespace MueLu

#define MUELU_CONSTRAINT_SHORT
#endif  // MUELU_CONSTRAINT_DECL_HPP