cell_weights.h

// ---------------------------------------------------------------------
//
// Copyright (C) 2018 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_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/hp/dof_handler.h>
 
 
DEAL_II_NAMESPACE_OPEN
 
namespace parallel
{
  template <int dim, int spacedim = dim>
  class CellWeights
  {
  public:
    CellWeights(const ::hp::DoFHandler<dim, spacedim> &dof_handler);
 
    ~CellWeights();
 
    void
    register_constant_weighting(const unsigned int factor = 1000);
 
    void
    register_ndofs_weighting(const unsigned int factor = 1000);
 
    void
    register_ndofs_squared_weighting(const unsigned int factor = 1000);
 
    void
    register_custom_weighting(
      const std::function<unsigned int(
        const FiniteElement<dim, spacedim> &,
        const typename hp::DoFHandler<dim, spacedim>::cell_iterator &)>
        custom_function);
 
  private:
    SmartPointer<const ::hp::DoFHandler<dim, spacedim>, CellWeights>
      dof_handler;
 
    SmartPointer<const parallel::Triangulation<dim, spacedim>, CellWeights>
      triangulation;
 
    std::function<unsigned int(
      const FiniteElement<dim, spacedim> &,
      const typename hp::DoFHandler<dim, spacedim>::cell_iterator &)>
      weighting_function;
 
    boost::signals2::connection tria_listener;
 
    unsigned int
    weight_callback(
      const typename Triangulation<dim, spacedim>::cell_iterator &cell,
      const typename Triangulation<dim, spacedim>::CellStatus     status);
  };
} // namespace parallel
 
 
DEAL_II_NAMESPACE_CLOSE
 
#endif