16 #ifndef dealii_mg_transfer_matrix_free_h
17 #define dealii_mg_transfer_matrix_free_h
55 template <
int dim,
typename Number>
106 const std::vector<std::shared_ptr<const Utilities::MPI::Partitioner>>
107 &external_partitioners =
108 std::vector<std::shared_ptr<const Utilities::MPI::Partitioner>>());
116 const std::function<
void(
const unsigned int,
136 const unsigned int to_level,
142 const unsigned int to_level,
166 const unsigned int from_level,
184 template <
typename BlockVectorType2>
189 const BlockVectorType2 &src)
const;
240 std::vector<std::vector<std::pair<unsigned int, unsigned int>>>
292 template <
int degree>
295 const unsigned int to_level,
302 template <
int degree>
316 template <
int dim,
typename Number,
typename TransferType>
318 :
public MGTransferBase<LinearAlgebra::distributed::BlockVector<Number>>
341 const unsigned int to_level,
347 const unsigned int to_level,
371 const unsigned int from_level,
385 template <
typename BlockVectorType2>
390 const BlockVectorType2 &src)
const;
395 template <
typename BlockVectorType2>
400 const BlockVectorType2 &src)
const;
405 template <
typename BlockVectorType2>
409 BlockVectorType2 &dst,
416 template <
typename BlockVectorType2>
420 BlockVectorType2 &dst,
435 virtual const TransferType &
460 template <
int dim,
typename Number>
464 MGTransferMatrixFree<dim, Number>>
483 const std::vector<MGConstrainedDoFs> &mg_constrained_dofs);
501 const std::vector<MGConstrainedDoFs> &mg_constrained_dofs);
546 template <
int dim,
typename Number>
547 template <
typename BlockVectorType2>
552 const BlockVectorType2 &src)
const
554 const unsigned int min_level = dst.min_level();
555 const unsigned int max_level = dst.max_level();
561 const auto &fe = dof_handler.
get_fe();
565 dst[
level].locally_owned_size() !=
567 dst[
level].reinit(this->vector_partitioners[
level]);
570 this->copy_to_mg(dof_handler, dst, src,
true);
577 dst[max_level].update_ghost_values();
584 if (dst[
level].get_partitioner().get() ==
585 this->vector_partitioners[
level].get())
589 ghosted_fine.
reinit(this->vector_partitioners[
level]);
592 input = &ghosted_fine;
595 std::vector<Number> dof_values_coarse(fe.n_dofs_per_cell());
598 std::vector<types::global_dof_index> dof_indices(fe.n_dofs_per_cell());
600 if (cell->is_locally_owned_on_level())
605 if (cell->is_active())
608 std::fill(dof_values_coarse.begin(), dof_values_coarse.end(), 0.);
609 for (
unsigned int child = 0; child < cell->n_children(); ++child)
611 cell->child(child)->get_mg_dof_indices(dof_indices);
614 this->mg_constrained_dofs,
level, dof_indices);
616 for (
unsigned int i = 0; i < fe.n_dofs_per_cell(); ++i)
617 dof_values_fine(i) = (*input)(dof_indices[i]);
618 fe.get_restriction_matrix(child, cell->refinement_case())
619 .vmult(tmp, dof_values_fine);
620 for (
unsigned int i = 0; i < fe.n_dofs_per_cell(); ++i)
621 if (fe.restriction_is_additive(i))
622 dof_values_coarse[i] += tmp[i];
623 else if (tmp(i) != 0.)
624 dof_values_coarse[i] = tmp[i];
626 cell->get_mg_dof_indices(dof_indices);
627 for (
unsigned int i = 0; i < fe.n_dofs_per_cell(); ++i)
630 dst[
level - 1](dof_indices[i]) = dof_values_coarse[i];
637 dst[
level].zero_out_ghost_values();
642 template <
int dim,
typename Number,
typename TransferType>
643 template <
typename BlockVectorType2>
648 const BlockVectorType2 &src)
const
652 "This object was initialized with support for usage with one "
653 "DoFHandler for each block, but this method assumes that "
654 "the same DoFHandler is used for all the blocks!"));
655 const std::vector<const DoFHandler<dim> *> mg_dofs(src.n_blocks(),
658 copy_to_mg(mg_dofs, dst, src);
663 template <
int dim,
typename Number,
typename TransferType>
664 template <
typename BlockVectorType2>
669 const BlockVectorType2 &src)
const
671 const unsigned int n_blocks = src.n_blocks();
677 const unsigned int min_level = dst.min_level();
678 const unsigned int max_level = dst.max_level();
686 min_level, max_level);
690 const unsigned int data_block = same_for_all ? 0 :
b;
691 get_matrix_free_transfer(data_block)
692 .copy_to_mg(*dof_handler[
b], dst_non_block, src.block(
b));
694 for (
unsigned int l = min_level;
l <= max_level; ++
l)
695 dst[
l].block(
b) = dst_non_block[
l];
702 template <
int dim,
typename Number,
typename TransferType>
703 template <
typename BlockVectorType2>
707 BlockVectorType2 &dst,
713 "This object was initialized with support for usage with one "
714 "DoFHandler for each block, but this method assumes that "
715 "the same DoFHandler is used for all the blocks!"));
716 const std::vector<const DoFHandler<dim> *> mg_dofs(dst.n_blocks(),
719 copy_from_mg(mg_dofs, dst, src);
722 template <
int dim,
typename Number,
typename TransferType>
723 template <
typename BlockVectorType2>
727 BlockVectorType2 &dst,
731 const unsigned int n_blocks = dst.n_blocks();
737 const unsigned int min_level = src.min_level();
738 const unsigned int max_level = src.max_level();
740 for (
unsigned int l = min_level;
l <= max_level; ++
l)
745 min_level, max_level);
749 for (
unsigned int l = min_level;
l <= max_level; ++
l)
751 src_non_block[
l].reinit(src[
l].block(
b));
752 src_non_block[
l] = src[
l].block(
b);
754 const unsigned int data_block = same_for_all ? 0 :
b;
755 get_matrix_free_transfer(data_block)
756 .copy_from_mg(*dof_handler[
b], dst.block(
b), src_non_block);
const Triangulation< dim, spacedim > & get_triangulation() const
types::global_dof_index n_dofs() const
const IndexSet & locally_owned_mg_dofs(const unsigned int level) const
const FiniteElement< dim, spacedim > & get_fe(const types::fe_index index=0) const
size_type n_elements() const
bool is_element(const size_type index) const
void update_ghost_values() const
void copy_locally_owned_data_from(const Vector< Number2, MemorySpace > &src)
void reinit(const size_type size, const bool omit_zeroing_entries=false)
std::function< void(const unsigned int, LinearAlgebra::distributed::Vector< Number > &)> initialize_dof_vector
SmartPointer< const MGConstrainedDoFs > mg_constrained_dofs
virtual void prolongate(const unsigned int to_level, LinearAlgebra::distributed::BlockVector< Number > &dst, const LinearAlgebra::distributed::BlockVector< Number > &src) const override
virtual const TransferType & get_matrix_free_transfer(const unsigned int b) const =0
virtual void prolongate_and_add(const unsigned int to_level, LinearAlgebra::distributed::BlockVector< Number > &dst, const LinearAlgebra::distributed::BlockVector< Number > &src) const override
void copy_from_mg(const std::vector< const DoFHandler< dim > * > &dof_handler, BlockVectorType2 &dst, const MGLevelObject< LinearAlgebra::distributed::BlockVector< Number >> &src) const
void copy_to_mg(const std::vector< const DoFHandler< dim > * > &dof_handler, MGLevelObject< LinearAlgebra::distributed::BlockVector< Number >> &dst, const BlockVectorType2 &src) const
virtual void restrict_and_add(const unsigned int from_level, LinearAlgebra::distributed::BlockVector< Number > &dst, const LinearAlgebra::distributed::BlockVector< Number > &src) const override
MGTransferBlockMatrixFreeBase(const bool same_for_all)
void copy_to_mg(const DoFHandler< dim > &dof_handler, MGLevelObject< LinearAlgebra::distributed::BlockVector< Number >> &dst, const BlockVectorType2 &src) const
void copy_from_mg(const DoFHandler< dim > &dof_handler, BlockVectorType2 &dst, const MGLevelObject< LinearAlgebra::distributed::BlockVector< Number >> &src) const
static const bool supports_dof_handler_vector
std::vector< MGTransferMatrixFree< dim, Number > > matrix_free_transfer_vector
void build(const DoFHandler< dim > &dof_handler)
std::size_t memory_consumption() const
MGTransferBlockMatrixFree()=default
virtual ~MGTransferBlockMatrixFree() override=default
const MGTransferMatrixFree< dim, Number > & get_matrix_free_transfer(const unsigned int b) const override
void initialize_constraints(const MGConstrainedDoFs &mg_constrained_dofs)
void do_restrict_add(const unsigned int from_level, LinearAlgebra::distributed::Vector< Number > &dst, const LinearAlgebra::distributed::Vector< Number > &src) const
void do_prolongate_add(const unsigned int to_level, LinearAlgebra::distributed::Vector< Number > &dst, const LinearAlgebra::distributed::Vector< Number > &src) const
virtual ~MGTransferMatrixFree() override=default
std::vector< std::vector< unsigned int > > level_dof_indices
unsigned int n_components
virtual void prolongate(const unsigned int to_level, LinearAlgebra::distributed::Vector< Number > &dst, const LinearAlgebra::distributed::Vector< Number > &src) const override
unsigned int n_child_cell_dofs
void initialize_constraints(const MGConstrainedDoFs &mg_constrained_dofs)
std::vector< std::vector< std::vector< unsigned short > > > dirichlet_indices
MGLevelObject< std::shared_ptr< const Utilities::MPI::Partitioner > > vector_partitioners
AlignedVector< VectorizedArray< Number > > evaluation_data
void build(const DoFHandler< dim > &dof_handler, const std::vector< std::shared_ptr< const Utilities::MPI::Partitioner >> &external_partitioners=std::vector< std::shared_ptr< const Utilities::MPI::Partitioner >>())
std::vector< std::vector< std::pair< unsigned int, unsigned int > > > parent_child_connect
virtual void prolongate_and_add(const unsigned int to_level, LinearAlgebra::distributed::Vector< Number > &dst, const LinearAlgebra::distributed::Vector< Number > &src) const override
std::size_t memory_consumption() const
std::vector< AlignedVector< VectorizedArray< Number > > > weights_on_refined
std::vector< unsigned int > n_owned_level_cells
void interpolate_to_mg(const DoFHandler< dim > &dof_handler, MGLevelObject< LinearAlgebra::distributed::Vector< Number >> &dst, const BlockVectorType2 &src) const
AlignedVector< VectorizedArray< Number > > prolongation_matrix_1d
bool element_is_continuous
virtual void restrict_and_add(const unsigned int from_level, LinearAlgebra::distributed::Vector< Number > &dst, const LinearAlgebra::distributed::Vector< Number > &src) const override
virtual unsigned int n_global_levels() const
void update_ghost_values() 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 & ExcDimensionMismatch(std::size_t arg1, std::size_t arg2)
#define Assert(cond, exc)
#define AssertDimension(dim1, dim2)
static ::ExceptionBase & ExcMessage(std::string arg1)
std::enable_if_t< IsBlockVector< VectorType >::value, unsigned int > n_blocks(const VectorType &vector)
std::enable_if_t< IsBlockVector< VectorType >::value, void > collect_sizes(VectorType &vector)
Tensor< 2, dim, Number > l(const Tensor< 2, dim, Number > &F, const Tensor< 2, dim, Number > &dF_dt)
SymmetricTensor< 2, dim, Number > b(const Tensor< 2, dim, Number > &F)
void resolve_identity_constraints(const MGConstrainedDoFs *mg_constrained_dofs, const unsigned int level, std::vector< types::global_dof_index > &dof_indices)