48template <
typename VectorType>
52 this->mg_constrained_dofs = &mg_c;
57template <
typename VectorType>
62 this->mg_constrained_dofs = &mg_c;
67template <
typename VectorType>
72 prolongation_matrices.resize(0);
73 prolongation_sparsities.resize(0);
74 interface_dofs.resize(0);
79template <
typename VectorType>
83 const VectorType & src)
const
85 Assert((to_level >= 1) && (to_level <= prolongation_matrices.size()),
86 ExcIndexRange(to_level, 1, prolongation_matrices.size() + 1));
88 if (this->mg_constrained_dofs !=
nullptr &&
89 this->mg_constrained_dofs->get_user_constraint_matrix(to_level - 1)
93 VectorType copy_src(src);
94 this->mg_constrained_dofs->get_user_constraint_matrix(to_level - 1)
95 .distribute(copy_src);
96 prolongation_matrices[to_level - 1]->vmult(dst, copy_src);
100 prolongation_matrices[to_level - 1]->vmult(dst, src);
106template <
typename VectorType>
110 const VectorType & src)
const
112 Assert((from_level >= 1) && (from_level <= prolongation_matrices.size()),
113 ExcIndexRange(from_level, 1, prolongation_matrices.size() + 1));
116 prolongation_matrices[from_level - 1]->Tvmult_add(dst, src);
128 const unsigned int level,
129 std::vector<types::global_dof_index> &dof_indices)
131 if (mg_constrained_dofs !=
nullptr &&
133 for (
auto &ind : dof_indices)
134 if (mg_constrained_dofs->get_level_constraints(
level)
135 .is_identity_constrained(ind))
149template <
typename VectorType>
150template <
int dim,
int spacedim>
157 "The underlying DoFHandler object has not had its "
158 "distribute_mg_dofs() function called, but this is a prerequisite "
159 "for multigrid transfers. You will need to call this function, "
160 "probably close to where you already call distribute_dofs()."));
162 const unsigned int n_levels =
166 this->sizes.resize(n_levels);
167 for (
unsigned int l = 0; l < n_levels; ++l)
168 this->sizes[l] = dof_handler.
n_dofs(l);
180 prolongation_matrices.resize(0);
181 prolongation_sparsities.resize(0);
182 prolongation_matrices.reserve(n_levels - 1);
183 prolongation_sparsities.reserve(n_levels - 1);
185 for (
unsigned int i = 0; i < n_levels - 1; ++i)
187 prolongation_sparsities.emplace_back(
189 prolongation_matrices.emplace_back(
196 std::vector<types::global_dof_index> dof_indices_parent(dofs_per_cell);
197 std::vector<types::global_dof_index> dof_indices_child(dofs_per_cell);
198 std::vector<types::global_dof_index> entries(dofs_per_cell);
217 level_p1_relevant_dofs);
220 level_p1_relevant_dofs);
222 if (cell->has_children() && cell->is_locally_owned_on_level())
224 cell->get_mg_dof_indices(dof_indices_parent);
226 replace(this->mg_constrained_dofs,
level, dof_indices_parent);
228 Assert(cell->n_children() ==
231 for (
unsigned int child = 0; child < cell->n_children(); ++child)
236 child, cell->refinement_case());
238 Assert(prolongation.
n() != 0, ExcNoProlongation());
240 cell->child(child)->get_mg_dof_indices(dof_indices_child);
242 replace(this->mg_constrained_dofs,
249 for (
unsigned int i = 0; i < dofs_per_cell; ++i)
252 for (
unsigned int j = 0; j < dofs_per_cell; ++j)
253 if (prolongation(i, j) != 0)
254 entries.push_back(dof_indices_parent[j]);
262#ifdef DEAL_II_WITH_MPI
264 VectorType>::requires_distributed_sparsity_pattern)
280 *prolongation_matrices[
level],
281 *prolongation_sparsities[
level],
299 if (cell->has_children() && cell->is_locally_owned_on_level())
301 cell->get_mg_dof_indices(dof_indices_parent);
303 replace(this->mg_constrained_dofs,
level, dof_indices_parent);
305 Assert(cell->n_children() ==
308 for (
unsigned int child = 0; child < cell->n_children(); ++child)
312 child, cell->refinement_case());
314 if (this->mg_constrained_dofs !=
nullptr &&
316 for (
unsigned int j = 0; j < dofs_per_cell; ++j)
318 level, dof_indices_parent[j]))
319 for (
unsigned int i = 0; i < dofs_per_cell; ++i)
320 prolongation(i, j) = 0.;
322 cell->child(child)->get_mg_dof_indices(dof_indices_child);
324 replace(this->mg_constrained_dofs,
329 for (
unsigned int i = 0; i < dofs_per_cell; ++i)
330 prolongation_matrices[
level]->set(dof_indices_child[i],
332 dof_indices_parent.data(),
340 this->fill_and_communicate_copy_indices(dof_handler);
345template <
typename VectorType>
349 for (
unsigned int level = 0;
level < prolongation_matrices.size(); ++
level)
351 os <<
"Level " <<
level << std::endl;
352 prolongation_matrices[
level]->print(os);
359template <
typename VectorType>
364 for (
unsigned int i = 0; i < prolongation_matrices.size(); ++i)
365 result += prolongation_matrices[i]->memory_consumption() +
366 prolongation_sparsities[i]->memory_consumption();
373#include "mg_transfer_prebuilt.inst"
const std::vector< std::pair< size_type, number > > * get_constraint_entries(const size_type line_n) const
size_type n_constraints() const
const FiniteElement< dim, spacedim > & get_fe(const types::fe_index index=0) const
const IndexSet & locally_owned_mg_dofs(const unsigned int level) const
const Triangulation< dim, spacedim > & get_triangulation() const
types::global_dof_index n_dofs() const
MPI_Comm get_communicator() const
bool has_level_dofs() const
void add_entries(const size_type row, ForwardIterator begin, ForwardIterator end, const bool indices_are_unique_and_sorted=false)
const IndexSet & row_index_set() const
void reinit(const size_type m, const size_type n, const IndexSet &rowset=IndexSet())
unsigned int n_dofs_per_cell() const
virtual const FullMatrix< double > & get_prolongation_matrix(const unsigned int child, const RefinementCase< dim > &refinement_case=RefinementCase< dim >::isotropic_refinement) const
bool is_boundary_index(const unsigned int level, const types::global_dof_index index) const
bool have_boundary_indices() const
const AffineConstraints< double > & get_level_constraints(const unsigned int level) const
std::size_t memory_consumption() const
virtual void restrict_and_add(const unsigned int from_level, VectorType &dst, const VectorType &src) const override
void build(const DoFHandler< dim, spacedim > &dof_handler)
std::size_t memory_consumption() const
MGTransferPrebuilt()=default
virtual void prolongate(const unsigned int to_level, VectorType &dst, const VectorType &src) const override
void print_matrices(std::ostream &os) const
void initialize_constraints(const MGConstrainedDoFs &mg_constrained_dofs)
virtual unsigned int n_global_levels() const
#define DEAL_II_NAMESPACE_OPEN
#define DEAL_II_NAMESPACE_CLOSE
IteratorRange< cell_iterator > cell_iterators_on_level(const unsigned int level) const
static ::ExceptionBase & ExcNotImplemented()
#define Assert(cond, exc)
static ::ExceptionBase & ExcInternalError()
static ::ExceptionBase & ExcIndexRange(std::size_t arg1, std::size_t arg2, std::size_t arg3)
static ::ExceptionBase & ExcMessage(std::string arg1)
static void reinit(Matrix &matrix, Sparsity &sparsity, int level, const SparsityPatternType &sp, const DoFHandler< dim, spacedim > &)