repartitioning_policy_tools.h

Go to the documentation of this file.

// ---------------------------------------------------------------------
//
// Copyright (C) 2021 by the deal.II authors
//
// This file is part of the deal.II library.
//
// The deal.II library is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE.md at
// the top level directory of deal.II.
//
// ---------------------------------------------------------------------
 
#ifndef dealii_distributed_repartitioning_policy_tools_h
#define dealii_distributed_repartitioning_policy_tools_h
 
#include <deal.II/grid/tria.h>
 
#include <deal.II/lac/la_parallel_vector.h>
 
DEAL_II_NAMESPACE_OPEN
 
namespace RepartitioningPolicyTools
{
  template <int dim, int spacedim = dim>
  class Base : public Subscriptor
  {
  public:
    virtual LinearAlgebra::distributed::Vector<double>
    partition(const Triangulation<dim, spacedim> &tria_coarse_in) const = 0;
 
    virtual ~Base() = default;
  };
 
  template <int dim, int spacedim = dim>
  class DefaultPolicy : public Base<dim, spacedim>
  {
  public:
    DefaultPolicy(const bool tighten = false);
 
    virtual LinearAlgebra::distributed::Vector<double>
    partition(
      const Triangulation<dim, spacedim> &tria_coarse_in) const override;
 
  private:
    const bool tighten;
  };
 
  template <int dim, int spacedim = dim>
  class FirstChildPolicy : public Base<dim, spacedim>
  {
  public:
    FirstChildPolicy(const Triangulation<dim, spacedim> &tria_fine);
 
    virtual LinearAlgebra::distributed::Vector<double>
    partition(
      const Triangulation<dim, spacedim> &tria_coarse_in) const override;
 
  private:
    const unsigned int n_coarse_cells;
 
    const unsigned int n_global_levels;
 
    IndexSet is_level_partitions;
  };
 
  template <int dim, int spacedim = dim>
  class MinimalGranularityPolicy : public Base<dim, spacedim>
  {
  public:
    MinimalGranularityPolicy(const unsigned int n_min_cells);
 
    virtual LinearAlgebra::distributed::Vector<double>
    partition(const Triangulation<dim, spacedim> &tria_in) const override;
 
  private:
    const unsigned int n_min_cells;
  };
 
  template <int dim, int spacedim = dim>
  class CellWeightPolicy : public Base<dim, spacedim>
  {
  public:
    CellWeightPolicy(
      const std::function<unsigned int(
        const typename Triangulation<dim, spacedim>::cell_iterator &,
        const typename Triangulation<dim, spacedim>::CellStatus)>
        &weighting_function);
 
    virtual LinearAlgebra::distributed::Vector<double>
    partition(const Triangulation<dim, spacedim> &tria_in) const override;
 
  private:
    const std::function<
      unsigned int(const typename Triangulation<dim, spacedim>::cell_iterator &,
                   const typename Triangulation<dim, spacedim>::CellStatus)>
      weighting_function;
  };
 
} // namespace RepartitioningPolicyTools
 
DEAL_II_NAMESPACE_CLOSE
 
#endif