MueLu_VectorDroppingDistanceLaplacian_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_VECTORDROPPINGDISTANCELAPLACIAN_DECL_HPP
#define MUELU_VECTORDROPPINGDISTANCELAPLACIAN_DECL_HPP

#include "MueLu_VectorDroppingBase.hpp"
#include "MueLu_BoundaryDetection.hpp"
#include "MueLu_ClassicalDropping.hpp"
#include "MueLu_CutDrop.hpp"
#include "MueLu_DroppingCommon.hpp"
#include "MueLu_DistanceLaplacianDropping.hpp"
#include "MueLu_MatrixConstruction.hpp"
#include "MueLu_Utilities.hpp"

namespace MueLu {

template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node, Misc::StrengthMeasure SoC>
class VectorDroppingDistanceLaplacian : public VectorDroppingBase<Scalar, LocalOrdinal, GlobalOrdinal, Node> {
 public:
  using matrix_type             = Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>;
  using crs_matrix_type         = Xpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>;
  using GraphType               = Xpetra::CrsGraph<LocalOrdinal, GlobalOrdinal, Node>;
  using local_matrix_type       = typename crs_matrix_type::local_matrix_type;
  using local_graph_type        = typename GraphType::local_graph_type;
  using rowptr_type             = typename local_graph_type::row_map_type::non_const_type;
  using device_type             = typename Node::device_type;
  using memory_space            = typename device_type::memory_space;
  using results_view            = Kokkos::View<DecisionType*, memory_space>;
  using magnitudeType           = typename Teuchos::ScalarTraits<Scalar>::magnitudeType;
  using boundary_nodes_type     = typename MueLu::LWGraph_kokkos<LocalOrdinal, GlobalOrdinal, Node>::boundary_nodes_type;
  using Level                   = MueLu::Level;
  using nnz_count_type          = Kokkos::pair<LocalOrdinal, LocalOrdinal>;
  using block_indices_view_type = typename Kokkos::View<LocalOrdinal*, typename Node::device_type>;

  template <class DistanceFunctorType>
  static void runDroppingFunctors_on_dlap_inner(matrix_type& A,
                                                matrix_type& mergedA,
                                                typename matrix_type::local_ordinal_type blkPartSize,
                                                block_indices_view_type& rowTranslation,
                                                block_indices_view_type& colTranslation,
                                                results_view& results,
                                                rowptr_type& filtered_rowptr,
                                                rowptr_type& graph_rowptr,
                                                nnz_count_type& nnz,
                                                boundary_nodes_type& boundaryNodes,
                                                const std::string& droppingMethod,
                                                const magnitudeType threshold,
                                                const bool aggregationMayCreateDirichlet,
                                                const bool symmetrizeDroppedGraph,
                                                const bool useBlocking,
                                                DistanceFunctorType& dist2,
                                                Level& level,
                                                const Factory& factory) {
    auto lclA                        = A.getLocalMatrixDevice();
    auto preserve_diagonals          = Misc::KeepDiagonalFunctor(lclA, results);
    auto mark_singletons_as_boundary = Misc::MarkSingletonVectorFunctor(lclA, rowTranslation, boundaryNodes, results);

    if (droppingMethod == "point-wise") {
      auto dist_laplacian_dropping = DistanceLaplacian::make_vector_drop_functor<SoC>(A, mergedA, threshold, dist2, results, rowTranslation, colTranslation);

      if (aggregationMayCreateDirichlet) {
        if (symmetrizeDroppedGraph) {
          auto drop_boundaries = Misc::VectorSymmetricDropBoundaryFunctor(mergedA, rowTranslation, colTranslation, boundaryNodes, results);
          VectorDroppingDistanceLaplacian::runDroppingFunctors(A, mergedA, blkPartSize, rowTranslation, colTranslation, results, filtered_rowptr, graph_rowptr, nnz, useBlocking, level, factory,
                                                               dist_laplacian_dropping,
                                                               drop_boundaries,
                                                               preserve_diagonals,
                                                               mark_singletons_as_boundary);
        } else {
          auto drop_boundaries = Misc::VectorDropBoundaryFunctor(lclA, rowTranslation, boundaryNodes, results);
          VectorDroppingDistanceLaplacian::runDroppingFunctors(A, mergedA, blkPartSize, rowTranslation, colTranslation, results, filtered_rowptr, graph_rowptr, nnz, useBlocking, level, factory,
                                                               dist_laplacian_dropping,
                                                               drop_boundaries,
                                                               preserve_diagonals,
                                                               mark_singletons_as_boundary);
        }
      } else {
        if (symmetrizeDroppedGraph) {
          auto drop_boundaries = Misc::VectorSymmetricDropBoundaryFunctor(mergedA, rowTranslation, colTranslation, boundaryNodes, results);
          VectorDroppingDistanceLaplacian::runDroppingFunctors(A, mergedA, blkPartSize, rowTranslation, colTranslation, results, filtered_rowptr, graph_rowptr, nnz, useBlocking, level, factory,
                                                               dist_laplacian_dropping,
                                                               drop_boundaries,
                                                               preserve_diagonals);
        } else {
          auto drop_boundaries = Misc::VectorDropBoundaryFunctor(lclA, rowTranslation, boundaryNodes, results);
          VectorDroppingDistanceLaplacian::runDroppingFunctors(A, mergedA, blkPartSize, rowTranslation, colTranslation, results, filtered_rowptr, graph_rowptr, nnz, useBlocking, level, factory,
                                                               dist_laplacian_dropping,
                                                               drop_boundaries,
                                                               preserve_diagonals);
        }
      }
    } else if (droppingMethod == "cut-drop") {
      auto comparison = CutDrop::make_dlap_comparison_functor<SoC>(A, dist2, results);
      auto cut_drop   = CutDrop::CutDropFunctor(comparison, threshold);

      if (symmetrizeDroppedGraph) {
        auto drop_boundaries = Misc::VectorSymmetricDropBoundaryFunctor(mergedA, rowTranslation, colTranslation, boundaryNodes, results);
        VectorDroppingDistanceLaplacian::runDroppingFunctors(A, mergedA, blkPartSize, rowTranslation, colTranslation, results, filtered_rowptr, graph_rowptr, nnz, useBlocking, level, factory,
                                                             drop_boundaries,
                                                             preserve_diagonals,
                                                             cut_drop);
      } else {
        auto drop_boundaries = Misc::VectorDropBoundaryFunctor(lclA, rowTranslation, boundaryNodes, results);
        VectorDroppingDistanceLaplacian::runDroppingFunctors(A, mergedA, blkPartSize, rowTranslation, colTranslation, results, filtered_rowptr, graph_rowptr, nnz, useBlocking, level, factory,
                                                             drop_boundaries,
                                                             preserve_diagonals,
                                                             cut_drop);
      }
    }
  }

  static void runDroppingFunctors_on_dlap(matrix_type& A,
                                          matrix_type& mergedA,
                                          LocalOrdinal blkPartSize,
                                          block_indices_view_type& rowTranslation,
                                          block_indices_view_type& colTranslation,
                                          results_view& results,
                                          rowptr_type& filtered_rowptr,
                                          rowptr_type& graph_rowptr,
                                          nnz_count_type& nnz,
                                          boundary_nodes_type& boundaryNodes,
                                          const std::string& droppingMethod,
                                          const magnitudeType threshold,
                                          const bool aggregationMayCreateDirichlet,
                                          const bool symmetrizeDroppedGraph,
                                          const bool useBlocking,
                                          const std::string& distanceLaplacianMetric,
                                          Teuchos::Array<double>& dlap_weights,
                                          LocalOrdinal interleaved_blocksize,
                                          Level& level,
                                          const Factory& factory);
};

}  // namespace MueLu
#endif