15#ifndef dealii_mg_transfer_global_coarsening_h
16#define dealii_mg_transfer_global_coarsening_h
44 class MGTwoLevelTransferImplementation;
49 template <
int dim,
int spacedim>
53template <
int dim,
typename Number>
143 const unsigned int degree,
151 std::vector<unsigned int>
153 const unsigned int max_degree,
166 template <
int dim,
int spacedim>
167 std::vector<std::shared_ptr<const Triangulation<dim, spacedim>>>
187 template <
int dim,
int spacedim>
188 std::vector<std::shared_ptr<const Triangulation<dim, spacedim>>>
192 const bool preserve_fine_triangulation,
193 const bool repartition_fine_triangulation);
200 template <
int dim,
int spacedim>
201 std::vector<std::shared_ptr<const Triangulation<dim, spacedim>>>
205 const bool repartition_fine_triangulation =
false);
218template <
typename VectorType>
226 "This class is currently only implemented for vectors of "
227 "type LinearAlgebra::distributed::Vector.");
232 using Number =
typename VectorType::value_type;
266 virtual std::pair<bool, bool>
268 const std::shared_ptr<const Utilities::MPI::Partitioner>
270 const std::shared_ptr<const Utilities::MPI::Partitioner>
324 template <
int dim, std::
size_t w
idth,
typename IndexType>
325 std::pair<bool, bool>
327 const std::shared_ptr<const Utilities::MPI::Partitioner>
333 &constraint_info_coarse,
334 std::vector<unsigned
int> &dof_indices_fine);
378 std::shared_ptr<const Utilities::MPI::Partitioner>
427template <
int dim,
typename VectorType>
435 "This class is currently only implemented for vectors of "
436 "type LinearAlgebra::distributed::Vector.");
441 using Number =
typename VectorType::value_type;
519 const unsigned int dof_no_fine,
521 const unsigned int dof_no_coarse);
533 const unsigned int fe_degree_coarse);
539 interpolate(VectorType &dst,
const VectorType &src)
const override;
545 std::pair<bool, bool>
547 const std::shared_ptr<const Utilities::MPI::Partitioner>
561 const VectorType &src)
const override;
565 const VectorType &src)
const override;
637 internal::MatrixFreeFunctions::
638 ConstraintInfo<dim, VectorizedArrayType, types::global_dof_index>
644 internal::MatrixFreeFunctions::
645 ConstraintInfo<dim, VectorizedArrayType, types::global_dof_index>
731template <
int dim,
typename VectorType>
739 "This class is currently only implemented for vectors of "
740 "type LinearAlgebra::distributed::Vector.");
742 using Number =
typename VectorType::value_type;
822 interpolate(VectorType &dst,
const VectorType &src)
const override;
828 std::pair<bool, bool>
830 const std::shared_ptr<const Utilities::MPI::Partitioner>
844 boost::signals2::connection
850 boost::signals2::connection
856 boost::signals2::connection
862 boost::signals2::connection
872 const VectorType &src)
const override;
879 const VectorType &src)
const override;
885 template <
int n_components>
888 const VectorType &src)
const;
893 template <
int n_components>
926 std::shared_ptr<NonMatching::MappingInfo<dim, dim, Number>>
mapping_info;
932 internal::MatrixFreeFunctions::
933 ConstraintInfo<dim, VectorizedArrayType, unsigned int>
987template <
int dim,
typename Number>
989 LinearAlgebra::distributed::Vector<Number>>
1018 template <
typename MGTwoLevelTransferObject>
1020 const std::function<
void(
const unsigned int,
VectorType &)>
1026 template <
typename MGTwoLevelTransferObject>
1040 build(
const std::vector<std::shared_ptr<const Utilities::MPI::Partitioner>>
1092 const std::vector<std::shared_ptr<const Utilities::MPI::Partitioner>>
1106 const std::function<
void(
const unsigned int,
VectorType &)>
1149 template <
class InVector>
1153 const InVector &src)
const;
1164 template <
class OutVector>
1188 template <
class InVector>
1192 const InVector &src)
const;
1202 template <
class InVector>
1257 std::pair<const DoFHandler<dim> *,
unsigned int>
1272 template <
typename MGTwoLevelTransferObject>
1280 template <
class InVector>
1284 const InVector &vector_reference,
1285 const bool omit_zeroing_entries)
const;
1310 std::vector<std::shared_ptr<const Utilities::MPI::Partitioner>>
1321template <
int dim,
typename Number>
1367 const std::vector<MGConstrainedDoFs> &mg_constrained_dofs);
1403template <
int dim,
typename VectorType>
1407template <
int dim,
typename VectorType>
1419template <
int dim,
typename Number>
1420template <
typename MGTwoLevelTransferObject>
1423 const std::function<
void(
const unsigned int, VectorType &)>
1424 &initialize_dof_vector)
1426 this->transfer.
clear();
1427 this->internal_transfer.clear();
1429 this->initialize_transfer_references(transfer);
1430 this->build(initialize_dof_vector);
1435template <
int dim,
typename Number>
1436template <
typename MGTwoLevelTransferObject>
1441 this->initialize_transfer_references(transfer);
1446template <
int dim,
typename Number>
1447template <
typename MGTwoLevelTransferObject>
1452 const unsigned int min_level = transfer.
min_level();
1453 const unsigned int max_level = transfer.
max_level();
1455 this->transfer.
resize(min_level, max_level);
1457 for (
unsigned int l = min_level;
l <= max_level; ++
l)
1465template <
int dim,
typename Number>
1466template <
class InVector>
1469 const unsigned int level,
1471 const InVector &vec_reference,
1472 const bool omit_zeroing_entries)
const
1474 std::shared_ptr<const Utilities::MPI::Partitioner> partitioner;
1476 if (external_partitioners.empty())
1478 partitioner = vec_reference.get_partitioner();
1485 partitioner = external_partitioners[
level - transfer.
min_level()];
1491 if (vec.get_partitioner().get() == partitioner.get())
1493 if (omit_zeroing_entries ==
false)
1499 if (vec.size() == partitioner->size() &&
1500 vec.locally_owned_size() == partitioner->locally_owned_size())
1502 if (omit_zeroing_entries ==
false)
1508 vec.reinit(partitioner, omit_zeroing_entries);
1513template <
int dim,
typename Number>
1514template <
class InVector>
1518 const InVector &src)
const
1520 assert_dof_handler(dof_handler);
1524 const bool zero_out_values =
1525 (this->perform_plain_copy ==
false &&
1526 this->perform_renumbered_plain_copy ==
false) ||
1529 this->initialize_dof_vector(
level, dst[
level], src, !zero_out_values);
1532 if (this->perform_plain_copy)
1534 dst[dst.
max_level()].copy_locally_owned_data_from(src);
1536 else if (this->perform_renumbered_plain_copy)
1540 for (
unsigned int i = 0; i < this->copy_indices.back().n_cols(); ++i)
1541 dst_level.local_element(this->copy_indices.back()(1, i)) =
1542 src.local_element(i);
1546 this->ghosted_global_vector = src;
1547 this->ghosted_global_vector.update_ghost_values();
1553 auto &dst_level = dst[
l];
1555 const auto copy_unknowns = [&](
const auto &indices) {
1556 for (
unsigned int i = 0; i < indices.n_cols(); ++i)
1557 dst_level.local_element(indices(1, i)) =
1558 this->ghosted_global_vector.local_element(indices(0, i));
1561 copy_unknowns(this->copy_indices[l]);
1562 copy_unknowns(this->copy_indices_level_mine[l]);
1571template <
int dim,
typename Number>
1572template <
class OutVector>
1579 assert_dof_handler(dof_handler);
1581 if (this->perform_plain_copy)
1583 dst.zero_out_ghost_values();
1584 dst.copy_locally_owned_data_from(src[src.
max_level()]);
1586 else if (this->perform_renumbered_plain_copy)
1588 const auto &src_level = src[src.
max_level()];
1589 dst.zero_out_ghost_values();
1590 for (
unsigned int i = 0; i < this->copy_indices.back().n_cols(); ++i)
1591 dst.local_element(i) =
1592 src_level.local_element(this->copy_indices.back()(1, i));
1599 auto &ghosted_vector = this->ghosted_level_vector[
l];
1601 if (this->ghosted_level_vector[l].size() > 0)
1602 ghosted_vector = src[l];
1604 const auto *
const ghosted_vector_ptr =
1605 (this->ghosted_level_vector[
l].size() > 0) ? &ghosted_vector :
1608 ghosted_vector_ptr->update_ghost_values();
1610 const auto copy_unknowns = [&](
const auto &indices) {
1611 for (
unsigned int i = 0; i < indices.n_cols(); ++i)
1612 dst.local_element(indices(0, i)) =
1613 ghosted_vector_ptr->local_element(indices(1, i));
1616 copy_unknowns(this->copy_indices[l]);
1617 copy_unknowns(this->copy_indices_global_mine[l]);
1625template <
int dim,
typename Number>
1626template <
class InVector>
1629 const InVector &src)
const
1633 this->interpolate_to_mg(dof_handler_dummy, dst, src);
1638template <
int dim,
typename Number>
1639template <
class InVector>
1643 const InVector &src)
const
1645 assert_dof_handler(dof_handler);
1647 const unsigned int min_level = transfer.
min_level();
1648 const unsigned int max_level = transfer.
max_level();
1655 const bool zero_out_values =
false;
1656 this->initialize_dof_vector(
level, dst[
level], src, !zero_out_values);
1659 if (this->perform_plain_copy)
1661 dst[max_level].copy_locally_owned_data_from(src);
1663 for (
unsigned int l = max_level;
l > min_level; --
l)
1664 this->transfer[l]->
interpolate(dst[l - 1], dst[l]);
1666 else if (this->perform_renumbered_plain_copy)
1668 auto &dst_level = dst[max_level];
1670 for (
unsigned int i = 0; i < this->solution_copy_indices.back().n_cols();
1672 dst_level.local_element(this->solution_copy_indices.back()(1, i)) =
1673 src.local_element(i);
1675 for (
unsigned int l = max_level;
l > min_level; --
l)
1676 this->transfer[l]->
interpolate(dst[l - 1], dst[l]);
1680 this->solution_ghosted_global_vector = src;
1681 this->solution_ghosted_global_vector.update_ghost_values();
1683 for (
unsigned int l = max_level + 1;
l != min_level;)
1687 auto &dst_level = dst[
l];
1689 const auto copy_unknowns = [&](
const auto &indices) {
1690 for (
unsigned int i = 0; i < indices.n_cols(); ++i)
1691 dst_level.local_element(indices(1, i)) =
1692 this->solution_ghosted_global_vector.local_element(
1696 copy_unknowns(this->solution_copy_indices[l]);
1697 copy_unknowns(this->solution_copy_indices_level_mine[l]);
1702 this->transfer[
l]->interpolate(dst[l - 1], dst[l]);
std::function< void(const unsigned int, LinearAlgebra::distributed::Vector< Number > &)> initialize_dof_vector
SmartPointer< const MGConstrainedDoFs > mg_constrained_dofs
void resize(const unsigned int new_minlevel, const unsigned int new_maxlevel, Args &&...args)
unsigned int max_level() const
unsigned int min_level() const
void initialize_constraints(const MGConstrainedDoFs &mg_constrained_dofs)
MGTransferBlockMF(const std::vector< MGConstrainedDoFs > &mg_constrained_dofs)
MGTransferBlockMF(const MGConstrainedDoFs &mg_constrained_dofs)
void build(const std::vector< const DoFHandler< dim > * > &dof_handler)
std::vector< MGTransferMF< dim, Number > > transfer_operators_internal
MGTransferBlockMF(const MGTransferMF< dim, Number > &transfer_operator)
MGTransferBlockMF()=default
void build(const DoFHandler< dim > &dof_handler)
const MGTransferMF< dim, Number > & get_matrix_free_transfer(const unsigned int b) const override
std::vector< SmartPointer< const MGTransferMF< dim, Number > > > transfer_operators
void initialize_constraints(const std::vector< MGConstrainedDoFs > &mg_constrained_dofs)
std::pair< const DoFHandler< dim > *, unsigned int > get_dof_handler_fine() const
void copy_to_mg(const DoFHandler< dim > &dof_handler, MGLevelObject< VectorType > &dst, const InVector &src) const
unsigned int min_level() const
void build(const std::function< void(const unsigned int, VectorType &)> &initialize_dof_vector)
void assert_dof_handler(const DoFHandler< dim > &dof_handler_out) const
void interpolate_to_mg(const DoFHandler< dim > &dof_handler, MGLevelObject< VectorType > &dst, const InVector &src) const
std::vector< std::shared_ptr< const Utilities::MPI::Partitioner > > external_partitioners
void prolongate_and_add(const unsigned int to_level, VectorType &dst, const VectorType &src) const override
unsigned int max_level() const
MGTransferMF(const MGLevelObject< MGTwoLevelTransferObject > &transfer, const std::function< void(const unsigned int, VectorType &)> &initialize_dof_vector={})
void build(const DoFHandler< dim > &dof_handler, const std::function< void(const unsigned int, VectorType &)> &initialize_dof_vector)
void build(const DoFHandler< dim > &dof_handler, const std::vector< std::shared_ptr< const Utilities::MPI::Partitioner > > &external_partitioners={})
void build(const std::vector< std::shared_ptr< const Utilities::MPI::Partitioner > > &external_partitioners={})
void initialize_dof_vector(const unsigned int level, VectorType &vector, const InVector &vector_reference, const bool omit_zeroing_entries) const
void initialize_constraints(const MGConstrainedDoFs &mg_constrained_dofs)
void initialize_internal_transfer(const DoFHandler< dim > &dof_handler, const SmartPointer< const MGConstrainedDoFs > &mg_constrained_dofs)
void initialize_two_level_transfers(const MGLevelObject< MGTwoLevelTransferObject > &transfer)
MGTransferMF(const MGConstrainedDoFs &mg_constrained_dofs)
MGLevelObject< MGTwoLevelTransfer< dim, VectorType > > internal_transfer
void fill_and_communicate_copy_indices_global_coarsening(const DoFHandler< dim > &dof_handler)
MGLevelObject< SmartPointer< MGTwoLevelTransferBase< VectorType > > > transfer
std::size_t memory_consumption() const
virtual void restrict_and_add(const unsigned int from_level, VectorType &dst, const VectorType &src) const override
void initialize_transfer_references(const MGLevelObject< MGTwoLevelTransferObject > &transfer)
void prolongate(const unsigned int to_level, VectorType &dst, const VectorType &src) const override
void copy_from_mg(const DoFHandler< dim > &dof_handler, OutVector &dst, const MGLevelObject< VectorType > &src) const
void interpolate_to_mg(MGLevelObject< VectorType > &dst, const InVector &src) const
AlignedVector< Number > buffer_coarse_embedded
void zero_out_ghost_values(const VectorType &vec) const
std::shared_ptr< const Utilities::MPI::Partitioner > partitioner_fine_embedded
std::shared_ptr< const Utilities::MPI::Partitioner > partitioner_coarse
virtual std::size_t memory_consumption() const =0
AlignedVector< Number > buffer_fine_embedded
void prolongate_and_add(VectorType &dst, const VectorType &src) const
bool fine_element_is_continuous
LinearAlgebra::distributed::Vector< Number > vec_fine
void update_ghost_values(const VectorType &vec) const
LinearAlgebra::distributed::Vector< Number > vec_coarse
typename VectorType::value_type Number
bool vec_fine_needs_ghost_update
virtual void restrict_and_add_internal(LinearAlgebra::distributed::Vector< Number > &dst, const LinearAlgebra::distributed::Vector< Number > &src) const =0
virtual std::pair< bool, bool > enable_inplace_operations_if_possible(const std::shared_ptr< const Utilities::MPI::Partitioner > &partitioner_coarse, const std::shared_ptr< const Utilities::MPI::Partitioner > &partitioner_fine)=0
void restrict_and_add(VectorType &dst, const VectorType &src) const
std::shared_ptr< const Utilities::MPI::Partitioner > partitioner_coarse_embedded
std::pair< bool, bool > internal_enable_inplace_operations_if_possible(const std::shared_ptr< const Utilities::MPI::Partitioner > &partitioner_coarse, const std::shared_ptr< const Utilities::MPI::Partitioner > &partitioner_fine, bool &vec_fine_needs_ghost_update, internal::MatrixFreeFunctions::ConstraintInfo< dim, VectorizedArray< Number, width >, IndexType > &constraint_info_coarse, std::vector< unsigned int > &dof_indices_fine)
std::shared_ptr< const Utilities::MPI::Partitioner > partitioner_fine
virtual void interpolate(VectorType &dst, const VectorType &src) const =0
void compress(VectorType &vec, const VectorOperation::values op) const
virtual void prolongate_and_add_internal(LinearAlgebra::distributed::Vector< Number > &dst, const LinearAlgebra::distributed::Vector< Number > &src) const =0
std::shared_ptr< NonMatching::MappingInfo< dim, dim, Number > > mapping_info
boost::signals2::connection connect_restriction(const std::function< void(const bool)> &slot)
void reinit(const DoFHandler< dim > &dof_handler_fine, const DoFHandler< dim > &dof_handler_coarse, const Mapping< dim > &mapping_fine, const Mapping< dim > &mapping_coarse, const AffineConstraints< Number > &constraint_fine=AffineConstraints< Number >(), const AffineConstraints< Number > &constraint_coarse=AffineConstraints< Number >())
AdditionalData additional_data
std::unique_ptr< FiniteElement< dim > > fe_coarse
std::vector< Number > rpe_input_output
void restrict_and_add_internal_comp(VectorType &dst, const VectorType &src) const
boost::signals2::connection connect_prolongation_cell_loop(const std::function< void(const bool)> &slot)
std::vector< Number > rpe_buffer
std::size_t memory_consumption() const override
std::pair< bool, bool > enable_inplace_operations_if_possible(const std::shared_ptr< const Utilities::MPI::Partitioner > &partitioner_coarse, const std::shared_ptr< const Utilities::MPI::Partitioner > &partitioner_fine) override
Utilities::MPI::RemotePointEvaluation< dim > rpe
void restrict_and_add_internal(VectorType &dst, const VectorType &src) const override
std::vector< unsigned int > level_dof_indices_fine_ptrs
mg::SignalsNonNested signals_non_nested
internal::MatrixFreeFunctions::ConstraintInfo< dim, VectorizedArrayType, unsigned int > constraint_info
boost::signals2::connection connect_prolongation(const std::function< void(const bool)> &slot)
boost::signals2::connection connect_restriction_cell_loop(const std::function< void(const bool)> &slot)
void prolongate_and_add_internal_comp(VectorType &dst, const VectorType &src) const
MGTwoLevelTransferNonNested(const AdditionalData &data=AdditionalData())
void interpolate(VectorType &dst, const VectorType &src) const override
std::vector< unsigned int > level_dof_indices_fine
typename VectorType::value_type Number
unsigned int mg_level_fine
void prolongate_and_add_internal(VectorType &dst, const VectorType &src) const override
SmartPointer< const DoFHandler< dim > > dof_handler_fine
SmartPointer< const DoFHandler< dim > > dof_handler_fine
void reinit_geometric_transfer(const DoFHandler< dim > &dof_handler_fine, const DoFHandler< dim > &dof_handler_coarse, const AffineConstraints< Number > &constraint_fine=AffineConstraints< Number >(), const AffineConstraints< Number > &constraint_coarse=AffineConstraints< Number >(), const unsigned int mg_level_fine=numbers::invalid_unsigned_int, const unsigned int mg_level_coarse=numbers::invalid_unsigned_int)
internal::MatrixFreeFunctions::ConstraintInfo< dim, VectorizedArrayType, types::global_dof_index > constraint_info_fine
friend class internal::MGTwoLevelTransferImplementation
std::vector< MGTransferScheme > schemes
typename VectorType::value_type Number
std::vector< unsigned char > weights_are_compressed
void interpolate(VectorType &dst, const VectorType &src) const override
internal::MatrixFreeFunctions::ConstraintInfo< dim, VectorizedArrayType, types::global_dof_index > constraint_info_coarse
std::unique_ptr< MatrixFreeRelatedData > matrix_free_data
static bool fast_polynomial_transfer_supported(const unsigned int fe_degree_fine, const unsigned int fe_degree_coarse)
AlignedVector< VectorizedArrayType > weights
void reinit_polynomial_transfer(const DoFHandler< dim > &dof_handler_fine, const DoFHandler< dim > &dof_handler_coarse, const AffineConstraints< Number > &constraint_fine=AffineConstraints< Number >(), const AffineConstraints< Number > &constraint_coarse=AffineConstraints< Number >(), const unsigned int mg_level_fine=numbers::invalid_unsigned_int, const unsigned int mg_level_coarse=numbers::invalid_unsigned_int)
unsigned int mg_level_fine
std::pair< bool, bool > enable_inplace_operations_if_possible(const std::shared_ptr< const Utilities::MPI::Partitioner > &partitioner_coarse, const std::shared_ptr< const Utilities::MPI::Partitioner > &partitioner_fine) override
void restrict_and_add_internal(VectorType &dst, const VectorType &src) const override
unsigned int n_components
std::vector< unsigned int > weights_start
void reinit(const MatrixFree< dim, Number > &matrix_free_fine, const unsigned int dof_no_fine, const MatrixFree< dim, Number > &matrix_free_coarse, const unsigned int dof_no_coarse)
void reinit(const DoFHandler< dim > &dof_handler_fine, const DoFHandler< dim > &dof_handler_coarse, const AffineConstraints< Number > &constraint_fine=AffineConstraints< Number >(), const AffineConstraints< Number > &constraint_coarse=AffineConstraints< Number >(), const unsigned int mg_level_fine=numbers::invalid_unsigned_int, const unsigned int mg_level_coarse=numbers::invalid_unsigned_int)
std::size_t memory_consumption() const override
void prolongate_and_add_internal(VectorType &dst, const VectorType &src) const override
Abstract base class for mapping classes.
static constexpr std::size_t size()
#define DEAL_II_NAMESPACE_OPEN
#define DEAL_II_NAMESPACE_CLOSE
#define Assert(cond, exc)
#define AssertDimension(dim1, dim2)
static ::ExceptionBase & ExcInternalError()
Tensor< 2, dim, Number > l(const Tensor< 2, dim, Number > &F, const Tensor< 2, dim, Number > &dF_dt)
T & get_underlying_value(T &p)
static const unsigned int invalid_unsigned_int
bool enforce_all_points_found
AdditionalData(const double tolerance=1e-6, const unsigned int rtree_level=0, const bool enforce_all_points_found=true)
unsigned int degree_coarse
unsigned int n_dofs_per_cell_coarse
unsigned int n_coarse_cells
AlignedVector< double > restriction_matrix
internal::MatrixFreeFunctions::ShapeInfo< double > shape_info_coarse
unsigned int n_dofs_per_cell_fine
AlignedVector< double > prolongation_matrix
boost::signals2::signal< void(const bool before)> prolongation_cell_loop
boost::signals2::signal< void(const bool before)> restriction
boost::signals2::signal< void(const bool before)> restriction_cell_loop
boost::signals2::signal< void(const bool before)> prolongation