-
Notifications
You must be signed in to change notification settings - Fork 708
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Added test for SolutionTransfer with hp-refinement and manually set f…
…lags.
- Loading branch information
1 parent
4646a31
commit 3b2b22d
Showing
2 changed files
with
143 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,138 @@ | ||
// --------------------------------------------------------------------- | ||
// | ||
// Copyright (C) 2023 by the deal.II authors | ||
// | ||
// This file is part of the deal.II library. | ||
// | ||
// The deal.II library is free software; you can use it, redistribute | ||
// it, and/or modify it under the terms of the GNU Lesser General | ||
// Public License as published by the Free Software Foundation; either | ||
// version 2.1 of the License, or (at your option) any later version. | ||
// The full text of the license can be found in the file LICENSE at | ||
// the top level of the deal.II distribution. | ||
// | ||
// --------------------------------------------------------------------- | ||
|
||
|
||
|
||
// Test distributed SolutionTransfer with hp-refinement and manually set flags. | ||
|
||
|
||
#include <deal.II/distributed/solution_transfer.h> | ||
#include <deal.II/distributed/tria.h> | ||
|
||
#include <deal.II/dofs/dof_handler.h> | ||
#include <deal.II/dofs/dof_tools.h> | ||
|
||
#include <deal.II/fe/fe_q.h> | ||
|
||
#include <deal.II/grid/grid_generator.h> | ||
|
||
#include <deal.II/hp/fe_collection.h> | ||
|
||
#include <deal.II/lac/la_parallel_vector.h> | ||
|
||
#include "../tests.h" | ||
|
||
|
||
template <int dim> | ||
void | ||
test() | ||
{ | ||
parallel::distributed::Triangulation<dim> tria(MPI_COMM_WORLD); | ||
GridGenerator::subdivided_hyper_cube(tria, 2); | ||
tria.refine_global(1); | ||
|
||
hp::FECollection<dim> fes; | ||
for (unsigned int d = 1; d <= 3; ++d) | ||
fes.push_back(FE_Q<dim>(d)); | ||
|
||
DoFHandler<dim> dofh(tria); | ||
for (const auto &cell : dofh.active_cell_iterators()) | ||
if (cell->is_locally_owned()) | ||
cell->set_active_fe_index(1); | ||
|
||
dofh.distribute_dofs(fes); | ||
IndexSet locally_owned_dofs = dofh.locally_owned_dofs(); | ||
IndexSet locally_relevant_dofs = | ||
DoFTools::extract_locally_relevant_dofs(dofh); | ||
|
||
// set up solution | ||
LinearAlgebra::distributed::Vector<double> solution; | ||
solution.reinit(locally_owned_dofs, | ||
locally_relevant_dofs, | ||
dofh.get_communicator()); | ||
|
||
for (unsigned int i = 0; i < solution.size(); ++i) | ||
if (locally_owned_dofs.is_element(i)) | ||
solution(i) = i; | ||
solution.update_ghost_values(); | ||
|
||
// set refine/coarsen flags manually | ||
for (const auto &cell : dofh.active_cell_iterators()) | ||
if (cell->is_locally_owned()) | ||
{ | ||
const std::string parent_id_string = cell->parent()->id().to_string(); | ||
|
||
if (parent_id_string == "0_0:") | ||
{ | ||
// parent 0: h refinement | ||
cell->set_refine_flag(); | ||
} | ||
else if (parent_id_string == "1_0:") | ||
{ | ||
// parent 1: h coarsening | ||
cell->set_coarsen_flag(); | ||
} | ||
else if (parent_id_string == "2_0:") | ||
{ | ||
// parent 2: p refinement | ||
const auto super_fe_index = | ||
fes.next_in_hierarchy(cell->active_fe_index()); | ||
cell->set_future_fe_index(super_fe_index); | ||
} | ||
else if (parent_id_string == "3_0:") | ||
{ | ||
// parent 3: p coarsening | ||
const auto sub_fe_index = | ||
fes.previous_in_hierarchy(cell->active_fe_index()); | ||
cell->set_future_fe_index(sub_fe_index); | ||
} | ||
else | ||
{ | ||
Assert(false, ExcInternalError()); | ||
} | ||
} | ||
|
||
// initiate refinement and transfer | ||
parallel::distributed:: | ||
SolutionTransfer<dim, LinearAlgebra::distributed::Vector<double>> | ||
soltrans(dofh); | ||
soltrans.prepare_for_coarsening_and_refinement(solution); | ||
|
||
tria.execute_coarsening_and_refinement(); | ||
|
||
dofh.distribute_dofs(fes); | ||
locally_owned_dofs = dofh.locally_owned_dofs(); | ||
locally_relevant_dofs = DoFTools::extract_locally_relevant_dofs(dofh); | ||
|
||
solution.reinit(locally_owned_dofs, | ||
locally_relevant_dofs, | ||
dofh.get_communicator()); | ||
soltrans.interpolate(solution); | ||
|
||
// make sure no processor is hanging | ||
MPI_Barrier(MPI_COMM_WORLD); | ||
|
||
deallog << "OK" << std::endl; | ||
} | ||
|
||
|
||
int | ||
main(int argc, char **argv) | ||
{ | ||
Utilities::MPI::MPI_InitFinalize mpi_initialization(argc, argv, 1); | ||
MPILogInitAll log; | ||
|
||
test<2>(); | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
|
||
DEAL:0::OK | ||
|
||
DEAL:1::OK | ||
|