cell_weights.h

Go to the documentation of this file.

// -----------------------------------------------------------------------------
//
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception OR LGPL-2.1-or-later
// Copyright (C) 2018 - 2026 by the deal.II authors
//
// This file is part of the deal.II library.
//
// Detailed license information governing the source code and contributions
// can be found in LICENSE.md and CONTRIBUTING.md at the top level directory.
//
// -----------------------------------------------------------------------------
 
#ifndef dealii_distributed_cell_weights_h
#define dealii_distributed_cell_weights_h
 
#include <deal.II/base/config.h>
 
#include <deal.II/distributed/tria_base.h>
 
#include <deal.II/dofs/dof_handler.h>
 
#include <deal.II/fe/fe.h>
 
#include <deal.II/grid/cell_status.h>
#include <deal.II/grid/tria.h>
 
#include <boost/signals2/connection.hpp>
 
#include <functional>
#include <vector>
 
 
DEAL_II_NAMESPACE_OPEN
 
namespace parallel
{
  template <int dim, int spacedim = dim>
  class CellWeights
  {
  public:
    using WeightingFunction = std::function<
      unsigned int(const typename DoFHandler<dim, spacedim>::cell_iterator &,
                   const FiniteElement<dim, spacedim> &)>;
 
    static WeightingFunction
    constant_weighting(const unsigned int factor = 1);
 
    static WeightingFunction
    ndofs_weighting(const std::pair<float, float> &coefficients);
 
    static WeightingFunction
    ndofs_weighting(const std::vector<std::pair<float, float>> &coefficients);
 
    static std::vector<unsigned int>
    precompute_weights(const hp::FECollection<dim, spacedim> &fe_collection,
                       const WeightingFunction &weighting_function);
 
    CellWeights() = default;
 
    CellWeights(const DoFHandler<dim, spacedim> &dof_handler,
                const WeightingFunction         &weighting_function);
 
    CellWeights(const DoFHandler<dim, spacedim> &dof_handler,
                const std::vector<unsigned int> &precomputed_weights);
 
    ~CellWeights();
 
    void
    reinit(const DoFHandler<dim, spacedim> &dof_handler,
           const WeightingFunction         &weighting_function);
 
    void
    reinit(const DoFHandler<dim, spacedim> &dof_handler,
           const std::vector<unsigned int> &precomputed_weights);
 
    static std::function<unsigned int(
      const typename ::Triangulation<dim, spacedim>::cell_iterator &cell,
      const CellStatus status)>
    make_weighting_callback(const DoFHandler<dim, spacedim> &dof_handler,
                            const WeightingFunction &weighting_function);
 
    static std::function<unsigned int(
      const typename ::Triangulation<dim, spacedim>::cell_iterator &cell,
      const CellStatus status)>
    make_weighting_callback(
      const DoFHandler<dim, spacedim> &dof_handler,
      const std::vector<unsigned int> &precomputed_weights);
 
  private:
    boost::signals2::connection connection;
 
    static unsigned int
    weighting_callback(
      const typename ::Triangulation<dim, spacedim>::cell_iterator &cell,
      const CellStatus                                  status,
      const DoFHandler<dim, spacedim>                  &dof_handler,
      const parallel::TriangulationBase<dim, spacedim> &triangulation,
      const WeightingFunction                          &weighting_function);
 
    static unsigned int
    weighting_callback(
      const typename ::Triangulation<dim, spacedim>::cell_iterator &cell,
      const CellStatus                                  status,
      const DoFHandler<dim, spacedim>                  &dof_handler,
      const parallel::TriangulationBase<dim, spacedim> &triangulation,
      const std::vector<unsigned int>                  &precomputed_weights);
  };
} // namespace parallel
 
 
DEAL_II_NAMESPACE_CLOSE
 
#endif