From a9d66b594e29f02bbe4ed283e2ffde46ca59516e Mon Sep 17 00:00:00 2001 From: Mathias Anselmann Date: Wed, 5 Jun 2019 16:29:27 +0200 Subject: [PATCH] Added small test for extract_dofs() with DoFHandler and hp::DoFHandler in parallel --- tests/dofs/dof_tools_12a.cc | 133 +++++++++++++++++++++++ tests/dofs/dof_tools_12a.mpirun=3.output | 21 ++++ tests/dofs/dof_tools_12a.output | 21 ++++ 3 files changed, 175 insertions(+) create mode 100644 tests/dofs/dof_tools_12a.cc create mode 100644 tests/dofs/dof_tools_12a.mpirun=3.output create mode 100644 tests/dofs/dof_tools_12a.output diff --git a/tests/dofs/dof_tools_12a.cc b/tests/dofs/dof_tools_12a.cc new file mode 100644 index 0000000000..8d698a81e7 --- /dev/null +++ b/tests/dofs/dof_tools_12a.cc @@ -0,0 +1,133 @@ +// --------------------------------------------------------------------- +// +// Copyright (C) 2003 - 2018 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.md at +// the top level directory of deal.II. +// +// --------------------------------------------------------------------- +#include + +#include + +#include +#include +#include + +#include +#include + +#include +#include +#include + +#include +#include +#include +#include + +#include "../tests.h" + +// check +// DoFTools::extract_dofs as in dof_tools_12 in parallel for fewer elements + +void +output_bool_vector(std::vector &v) +{ + for (unsigned int i = 0; i < v.size(); ++i) + deallog << (v[i] ? '1' : '0'); + deallog << std::endl; +} + + +template +void +check_this(const DoFHandlerType &dof_handler) +{ + IndexSet locally_relevant_dofs; + DoFTools::extract_locally_relevant_dofs(dof_handler, locally_relevant_dofs); + + std::vector selected_dofs(dof_handler.n_locally_owned_dofs()); + std::vector mask(dof_handler.get_fe().n_components(), false); + + // only select first component + mask[0] = true; + DoFTools::extract_dofs(dof_handler, ComponentMask(mask), selected_dofs); + if (Utilities::MPI::this_mpi_process(MPI_COMM_WORLD) == 0) + { + output_bool_vector(selected_dofs); + } + + // also select last component + mask.back() = true; + DoFTools::extract_dofs(dof_handler, ComponentMask(mask), selected_dofs); + if (Utilities::MPI::this_mpi_process(MPI_COMM_WORLD) == 0) + { + output_bool_vector(selected_dofs); + } +} + + +template +void +check(const FiniteElement &fe, const std::string &name) +{ + deallog << "Checking " << name << " in " << dim << "d:" << std::endl; + + // create tria and dofhandler + // objects. set different boundary + // and sub-domain ids + parallel::distributed::Triangulation tria(MPI_COMM_WORLD); + GridGenerator::hyper_cube(tria, 0., 1.); + tria.refine_global(1); + for (unsigned int ref = 0; ref < 2; ++ref) + { + for (auto &cell : tria.active_cell_iterators()) + if (cell->is_locally_owned() && cell->center()(0) < .5 && + cell->center()(1) < .5) + cell->set_refine_flag(); + tria.execute_coarsening_and_refinement(); + } + + DoFHandler dof_handler(tria); + dof_handler.distribute_dofs(fe); + + // setup hp DoFHandler + hp::FECollection fe_collection(fe); + hp::DoFHandler hp_dof_handler(tria); + hp_dof_handler.distribute_dofs(fe_collection); + + check_this>(dof_handler); + check_this>(hp_dof_handler); +} + + +#define CHECK(EL, deg, dim) \ + { \ + FE_##EL EL(deg); \ + check(EL, #EL #deg); \ + } + +#define CHECK_ALL(EL, deg) \ + { \ + CHECK(EL, deg, 2); \ + CHECK(EL, deg, 3); \ + } + +int +main(int argc, char **argv) +{ + Utilities::MPI::MPI_InitFinalize mpi_initialization(argc, argv, 1); + mpi_initlog(); + CHECK_ALL(Q, 1); + + CHECK_ALL(Q, 2); + + return 0; +} diff --git a/tests/dofs/dof_tools_12a.mpirun=3.output b/tests/dofs/dof_tools_12a.mpirun=3.output new file mode 100644 index 0000000000..c14f45a857 --- /dev/null +++ b/tests/dofs/dof_tools_12a.mpirun=3.output @@ -0,0 +1,21 @@ + +DEAL::Checking Q1 in 2d: +DEAL::111111111111111 +DEAL::111111111111111 +DEAL::111111111111111 +DEAL::111111111111111 +DEAL::Checking Q1 in 3d: +DEAL::111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::Checking Q2 in 2d: +DEAL::111111111111111111111111111111111111111111111 +DEAL::111111111111111111111111111111111111111111111 +DEAL::111111111111111111111111111111111111111111111 +DEAL::111111111111111111111111111111111111111111111 +DEAL::Checking Q2 in 3d: +DEAL::11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 diff --git a/tests/dofs/dof_tools_12a.output b/tests/dofs/dof_tools_12a.output new file mode 100644 index 0000000000..531d3dd93c --- /dev/null +++ b/tests/dofs/dof_tools_12a.output @@ -0,0 +1,21 @@ + +DEAL::Checking Q1 in 2d: +DEAL::11111111111111111111111111111111111111111 +DEAL::11111111111111111111111111111111111111111 +DEAL::11111111111111111111111111111111111111111 +DEAL::11111111111111111111111111111111111111111 +DEAL::Checking Q1 in 3d: +DEAL::11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::Checking Q2 in 2d: +DEAL::111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::Checking Q2 in 3d: +DEAL::11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 +DEAL::11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111 -- 2.39.5