mg_transfer_internal.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) 2016 - 2024 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_mg_transfer_internal_h
#define dealii_mg_transfer_internal_h
 
#include <deal.II/base/config.h>
 
#include <deal.II/base/mg_level_object.h>
 
#include <deal.II/dofs/dof_handler.h>
 
#include <deal.II/lac/la_parallel_vector.h>
 
#include <deal.II/multigrid/mg_constrained_dofs.h>
 
DEAL_II_NAMESPACE_OPEN
 
namespace internal
{
  namespace MGTransfer
  {
    template <int dim, int spacedim>
    void
    fill_copy_indices(
      const DoFHandler<dim, spacedim> &dof_handler,
      const MGConstrainedDoFs         *mg_constrained_dofs,
      std::vector<std::vector<
        std::pair<types::global_dof_index, types::global_dof_index>>>
        &copy_indices,
      std::vector<std::vector<
        std::pair<types::global_dof_index, types::global_dof_index>>>
        &copy_indices_global_mine,
      std::vector<std::vector<
        std::pair<types::global_dof_index, types::global_dof_index>>>
                &copy_indices_level_mine,
      const bool skip_interface_dofs = true);
 
 
 
    template <int dim>
    unsigned int
    compute_shift_within_children(const unsigned int child,
                                  const unsigned int fe_shift_1d,
                                  const unsigned int fe_degree);
 
    template <typename Number>
    struct ElementInfo
    {
      unsigned int fe_degree;
 
      bool element_is_continuous;
 
      unsigned int n_components;
 
      unsigned int n_child_cell_dofs;
 
      std::vector<unsigned int> lexicographic_numbering;
 
      std::vector<Number> prolongation_matrix_1d;
    };
 
    template <int dim, typename Number>
    void
    setup_transfer(
      const DoFHandler<dim>   &dof_handler,
      const MGConstrainedDoFs *mg_constrained_dofs,
      const std::vector<std::shared_ptr<const Utilities::MPI::Partitioner>>
                                             &external_partitioners,
      ElementInfo<Number>                    &elem_info,
      std::vector<std::vector<unsigned int>> &level_dof_indices,
      std::vector<std::vector<std::pair<unsigned int, unsigned int>>>
                                &parent_child_connect,
      std::vector<unsigned int> &n_owned_level_cells,
      std::vector<std::vector<std::vector<unsigned short>>> &dirichlet_indices,
      std::vector<std::vector<Number>>                      &weights_on_refined,
      std::vector<Table<2, unsigned int>> &copy_indices_global_mine,
      MGLevelObject<std::shared_ptr<const Utilities::MPI::Partitioner>>
        &vector_partitioners);
 
 
 
    void
    resolve_identity_constraints(
      const MGConstrainedDoFs              *mg_constrained_dofs,
      const unsigned int                    level,
      std::vector<types::global_dof_index> &dof_indices);
 
  } // namespace MGTransfer
} // namespace internal
 
DEAL_II_NAMESPACE_CLOSE
 
#endif