15#ifndef dealii_mg_transfer_global_coarsening_h
16#define dealii_mg_transfer_global_coarsening_h
43 class MGTwoLevelTransferImplementation;
48 template <
int dim,
int spacedim>
52template <
int dim,
typename Number>
142 const unsigned int degree,
150 std::vector<unsigned int>
152 const unsigned int max_degree,
165 template <
int dim,
int spacedim>
166 std::vector<std::shared_ptr<const Triangulation<dim, spacedim>>>
186 template <
int dim,
int spacedim>
187 std::vector<std::shared_ptr<const Triangulation<dim, spacedim>>>
191 const bool preserve_fine_triangulation,
192 const bool repartition_fine_triangulation);
199 template <
int dim,
int spacedim>
200 std::vector<std::shared_ptr<const Triangulation<dim, spacedim>>>
204 const bool repartition_fine_triangulation =
false);
217template <
typename VectorType>
225 "This class is currently only implemented for vectors of "
226 "type LinearAlgebra::distributed::Vector.");
231 using Number =
typename VectorType::value_type;
266 const std::shared_ptr<const Utilities::MPI::Partitioner>
268 const std::shared_ptr<const Utilities::MPI::Partitioner>
322 template <
int dim, std::
size_t w
idth,
typename IndexType>
325 const std::shared_ptr<const Utilities::MPI::Partitioner>
331 &constraint_info_coarse,
332 std::vector<unsigned
int> &dof_indices_fine);
376 std::shared_ptr<const Utilities::MPI::Partitioner>
405template <
int dim,
typename VectorType>
413 "This class is currently only implemented for vectors of "
414 "type LinearAlgebra::distributed::Vector.");
419 using Number =
typename VectorType::value_type;
496 const unsigned int fe_degree_coarse);
502 interpolate(VectorType &dst,
const VectorType &src)
const override;
510 const std::shared_ptr<const Utilities::MPI::Partitioner>
524 const VectorType &src)
const override;
528 const VectorType &src)
const override;
609 internal::MatrixFreeFunctions::
610 ConstraintInfo<dim, VectorizedArrayType, types::global_dof_index>
616 internal::MatrixFreeFunctions::
617 ConstraintInfo<dim, VectorizedArrayType, types::global_dof_index>
656template <
int dim,
typename VectorType>
664 "This class is currently only implemented for vectors of "
665 "type LinearAlgebra::distributed::Vector.");
667 using Number =
typename VectorType::value_type;
747 interpolate(VectorType &dst,
const VectorType &src)
const override;
755 const std::shared_ptr<const Utilities::MPI::Partitioner>
769 boost::signals2::connection
775 boost::signals2::connection
781 boost::signals2::connection
787 boost::signals2::connection
797 const VectorType &src)
const override;
804 const VectorType &src)
const override;
810 template <
int n_components>
813 const VectorType &src)
const;
818 template <
int n_components>
841 std::shared_ptr<NonMatching::MappingInfo<dim, dim, Number>>
mapping_info;
847 internal::MatrixFreeFunctions::
848 ConstraintInfo<dim, VectorizedArrayType, unsigned int>
902template <
int dim,
typename Number>
904 LinearAlgebra::distributed::Vector<Number>>
933 template <
typename MGTwoLevelTransferObject>
935 const std::function<
void(
const unsigned int,
VectorType &)>
941 template <
typename MGTwoLevelTransferObject>
955 build(
const std::vector<std::shared_ptr<const Utilities::MPI::Partitioner>>
1007 const std::vector<std::shared_ptr<const Utilities::MPI::Partitioner>>
1021 const std::function<
void(
const unsigned int,
VectorType &)>
1064 template <
class InVector>
1068 const InVector &src)
const;
1079 template <
class OutVector>
1103 template <
class InVector>
1107 const InVector &src)
const;
1117 template <
class InVector>
1172 std::pair<const DoFHandler<dim> *,
unsigned int>
1187 template <
typename MGTwoLevelTransferObject>
1195 template <
class InVector>
1199 const InVector &vector_reference,
1200 const bool omit_zeroing_entries)
const;
1225 std::vector<std::shared_ptr<const Utilities::MPI::Partitioner>>
1236template <
int dim,
typename Number>
1282 const std::vector<MGConstrainedDoFs> &mg_constrained_dofs);
1318template <
int dim,
typename VectorType>
1322template <
int dim,
typename VectorType>
1334template <
int dim,
typename Number>
1335template <
typename MGTwoLevelTransferObject>
1338 const std::function<
void(
const unsigned int, VectorType &)>
1339 &initialize_dof_vector)
1341 this->transfer.
clear();
1342 this->internal_transfer.clear();
1344 this->initialize_transfer_references(transfer);
1345 this->build(initialize_dof_vector);
1350template <
int dim,
typename Number>
1351template <
typename MGTwoLevelTransferObject>
1356 this->initialize_transfer_references(transfer);
1361template <
int dim,
typename Number>
1362template <
typename MGTwoLevelTransferObject>
1367 const unsigned int min_level =
transfer.min_level();
1368 const unsigned int max_level =
transfer.max_level();
1370 this->transfer.
resize(min_level, max_level);
1372 for (
unsigned int l = min_level;
l <= max_level; ++
l)
1380template <
int dim,
typename Number>
1381template <
class InVector>
1384 const unsigned int level,
1386 const InVector &vec_reference,
1387 const bool omit_zeroing_entries)
const
1389 std::shared_ptr<const Utilities::MPI::Partitioner> partitioner;
1391 if (external_partitioners.empty())
1393 partitioner = vec_reference.get_partitioner();
1400 partitioner = external_partitioners[
level -
transfer.min_level()];
1406 if (vec.get_partitioner().get() == partitioner.get())
1408 if (omit_zeroing_entries ==
false)
1414 if (vec.size() == partitioner->size() &&
1415 vec.locally_owned_size() == partitioner->locally_owned_size())
1417 if (omit_zeroing_entries ==
false)
1423 vec.reinit(partitioner, omit_zeroing_entries);
1428template <
int dim,
typename Number>
1429template <
class InVector>
1433 const InVector &src)
const
1435 assert_dof_handler(dof_handler);
1439 const bool zero_out_values =
1440 (this->perform_plain_copy ==
false &&
1441 this->perform_renumbered_plain_copy ==
false) ||
1444 this->initialize_dof_vector(
level, dst[
level], src, !zero_out_values);
1447 if (this->perform_plain_copy)
1449 dst[dst.
max_level()].copy_locally_owned_data_from(src);
1451 else if (this->perform_renumbered_plain_copy)
1455 for (
unsigned int i = 0; i < this->copy_indices.back().n_cols(); ++i)
1456 dst_level.local_element(this->copy_indices.back()(1, i)) =
1457 src.local_element(i);
1461 this->ghosted_global_vector = src;
1462 this->ghosted_global_vector.update_ghost_values();
1468 auto &dst_level = dst[
l];
1470 const auto copy_unknowns = [&](
const auto &indices) {
1471 for (
unsigned int i = 0; i < indices.n_cols(); ++i)
1472 dst_level.local_element(indices(1, i)) =
1473 this->ghosted_global_vector.local_element(indices(0, i));
1476 copy_unknowns(this->copy_indices[l]);
1477 copy_unknowns(this->copy_indices_level_mine[l]);
1486template <
int dim,
typename Number>
1487template <
class OutVector>
1494 assert_dof_handler(dof_handler);
1496 if (this->perform_plain_copy)
1498 dst.zero_out_ghost_values();
1499 dst.copy_locally_owned_data_from(src[src.
max_level()]);
1501 else if (this->perform_renumbered_plain_copy)
1503 const auto &src_level = src[src.
max_level()];
1504 dst.zero_out_ghost_values();
1505 for (
unsigned int i = 0; i < this->copy_indices.back().n_cols(); ++i)
1506 dst.local_element(i) =
1507 src_level.local_element(this->copy_indices.back()(1, i));
1514 auto &ghosted_vector = this->ghosted_level_vector[
l];
1516 if (this->ghosted_level_vector[l].size() > 0)
1517 ghosted_vector = src[l];
1519 const auto *
const ghosted_vector_ptr =
1520 (this->ghosted_level_vector[
l].size() > 0) ? &ghosted_vector :
1523 ghosted_vector_ptr->update_ghost_values();
1525 const auto copy_unknowns = [&](
const auto &indices) {
1526 for (
unsigned int i = 0; i < indices.n_cols(); ++i)
1527 dst.local_element(indices(0, i)) =
1528 ghosted_vector_ptr->local_element(indices(1, i));
1531 copy_unknowns(this->copy_indices[l]);
1532 copy_unknowns(this->copy_indices_global_mine[l]);
1540template <
int dim,
typename Number>
1541template <
class InVector>
1544 const InVector &src)
const
1548 this->interpolate_to_mg(dof_handler_dummy, dst, src);
1553template <
int dim,
typename Number>
1554template <
class InVector>
1558 const InVector &src)
const
1560 assert_dof_handler(dof_handler);
1562 const unsigned int min_level =
transfer.min_level();
1563 const unsigned int max_level =
transfer.max_level();
1570 const bool zero_out_values =
false;
1571 this->initialize_dof_vector(
level, dst[
level], src, !zero_out_values);
1574 if (this->perform_plain_copy)
1576 dst[max_level].copy_locally_owned_data_from(src);
1578 for (
unsigned int l = max_level;
l > min_level; --
l)
1579 this->transfer[l]->
interpolate(dst[l - 1], dst[l]);
1581 else if (this->perform_renumbered_plain_copy)
1583 auto &dst_level = dst[max_level];
1585 for (
unsigned int i = 0; i < this->solution_copy_indices.back().n_cols();
1587 dst_level.local_element(this->solution_copy_indices.back()(1, i)) =
1588 src.local_element(i);
1590 for (
unsigned int l = max_level;
l > min_level; --
l)
1591 this->transfer[l]->
interpolate(dst[l - 1], dst[l]);
1595 this->solution_ghosted_global_vector = src;
1596 this->solution_ghosted_global_vector.update_ghost_values();
1598 for (
unsigned int l = max_level + 1;
l != min_level;)
1602 auto &dst_level = dst[
l];
1604 const auto copy_unknowns = [&](
const auto &indices) {
1605 for (
unsigned int i = 0; i < indices.n_cols(); ++i)
1606 dst_level.local_element(indices(1, i)) =
1607 this->solution_ghosted_global_vector.local_element(
1611 copy_unknowns(this->solution_copy_indices[l]);
1612 copy_unknowns(this->solution_copy_indices_level_mine[l]);
1617 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
virtual void 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
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
void 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)
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
void restrict_and_add(VectorType &dst, const VectorType &src) const
std::shared_ptr< const Utilities::MPI::Partitioner > partitioner_coarse_embedded
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
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::size_t memory_consumption() const 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
void 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
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
AlignedVector< VectorizedArrayType > weights_compressed
void 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
typename VectorType::value_type Number
void interpolate(VectorType &dst, const VectorType &src) const override
internal::MatrixFreeFunctions::ConstraintInfo< dim, VectorizedArrayType, types::global_dof_index > constraint_info_coarse
static bool fast_polynomial_transfer_supported(const unsigned int fe_degree_fine, const unsigned int fe_degree_coarse)
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
void restrict_and_add_internal(VectorType &dst, const VectorType &src) const override
std::vector< Number > weights
unsigned int n_components
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.
#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
AlignedVector< double > restriction_matrix_1d
unsigned int n_dofs_per_cell_coarse
unsigned int n_coarse_cells
AlignedVector< double > prolongation_matrix_1d
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