MueLu_SemiCoarsenPFactory_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_SEMICOARSENPFACTORY_DECL_HPP
#define MUELU_SEMICOARSENPFACTORY_DECL_HPP

#include <Xpetra_Matrix_fwd.hpp>

#include "MueLu_ConfigDefs.hpp"
#include "MueLu_PFactory.hpp"
#include "MueLu_SemiCoarsenPFactory_fwd.hpp"

#include "MueLu_Level_fwd.hpp"

#define VERTICAL 1
#define HORIZONTAL 2
#define GRID_SUPPLIED -1
#define NUM_ZPTS 0
#define ORIENTATION 1

namespace MueLu {

template <class Scalar        = DefaultScalar,
          class LocalOrdinal  = DefaultLocalOrdinal,
          class GlobalOrdinal = DefaultGlobalOrdinal,
          class Node          = DefaultNode>
class SemiCoarsenPFactory : public PFactory {
#undef MUELU_SEMICOARSENPFACTORY_SHORT
#include "MueLu_UseShortNames.hpp"

 public:


  SemiCoarsenPFactory()
    : bTransferCoordinates_(false) {}

  virtual ~SemiCoarsenPFactory() {}

  RCP<const ParameterList> GetValidParameterList() const;



  void DeclareInput(Level& fineLevel, Level& coarseLevel) const;




  void Build(Level& fineLevel, Level& coarseLevel) const;
  void BuildP(Level& fineLevel, Level& coarseLevel) const;


 private:
  LO FindCpts(LO const PtsPerLine, LO const CoarsenRate, LO const Thin, LO** LayerCpts) const;
  LO MakeSemiCoarsenP(LO const Ntotal, LO const nz, LO const CoarsenRate, LO const LayerId[],
                      LO const VertLineId[], LO const DofsPerNode, RCP<Matrix>& Amat,
                      RCP<Matrix>& P, RCP<const Map>& coarseMap,
                      const RCP<MultiVector> fineNullspace, RCP<MultiVector>& coarseNullspace, RCP<Matrix>& R, bool buildRestriction, bool doLinear) const;
  void RevertToPieceWiseConstant(RCP<Matrix>& P, LO BlkSize) const;

  mutable bool bTransferCoordinates_;  //< boolean which is true if coordinate information is available to be transferred to coarse coordinate information
};                                     // class SemiCoarsenPFactory

}  // namespace MueLu

#define MUELU_SEMICOARSENPFACTORY_SHORT
#endif  // MUELU_SEMICOARSENPFACTORY_DECL_HPP