1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
// Copyright (C) 2021 Garth N. Wells
//
// This file is part of DOLFINx (https://www.fenicsproject.org)
//
// SPDX-License-Identifier: LGPL-3.0-or-later
#ifdef HAS_ADIOS2
#include <algorithm>
#include <catch2/catch_test_macros.hpp>
#include <concepts>
#include <dolfinx/fem/Function.h>
#include <dolfinx/fem/FunctionSpace.h>
#include <dolfinx/fem/utils.h>
#include <dolfinx/io/ADIOS2Writers.h>
#include <dolfinx/mesh/Mesh.h>
#include <dolfinx/mesh/generation.h>
#include <mpi.h>
using namespace dolfinx;
namespace
{
template <std::floating_point T>
void test_fides_mesh()
{
auto part = mesh::create_cell_partitioner(mesh::GhostMode::shared_facet);
auto mesh = std::make_shared<mesh::Mesh<T>>(
mesh::create_rectangle<T>(MPI_COMM_WORLD, {{{0.0, 0.0}, {1.0, 1.0}}},
{22, 12}, mesh::CellType::triangle, part));
std::filesystem::path f = "test_mesh" + std::to_string(sizeof(T)) + ".bp";
io::FidesWriter writer(mesh->comm(), f, mesh);
writer.write(0.0);
auto x = mesh->geometry().x();
// Move all coordinates of the mesh geometry
std::ranges::transform(x, x.begin(), [](auto x) { return x + 1; });
writer.write(0.2);
// Only move x coordinate
for (std::size_t i = 0; i < x.size(); i += 3)
x[i] -= 0.5;
writer.write(0.4);
}
template <std::floating_point T>
void test_fides_function()
{
auto mesh = std::make_shared<mesh::Mesh<T>>(
mesh::create_rectangle<T>(MPI_COMM_WORLD, {{{0.0, 0.0}, {1.0, 1.0}}},
{22, 12}, mesh::CellType::triangle));
// Create a Basix continuous Lagrange element of degree 1
basix::FiniteElement e = basix::create_element<T>(
basix::element::family::P,
mesh::cell_type_to_basix_type(mesh::CellType::triangle), 1,
basix::element::lagrange_variant::unset,
basix::element::dpc_variant::unset, false);
// Create a scalar function space
auto V = std::make_shared<fem::FunctionSpace<T>>(
fem::create_functionspace(mesh, e));
// Create a finite element Function
auto u = std::make_shared<fem::Function<T>>(V);
auto v = std::make_shared<fem::Function<std::complex<T>>>(V);
std::filesystem::path f0 = "test_u0" + std::to_string(sizeof(T)) + ".bp";
io::FidesWriter<T> writer0(mesh->comm(), f0, {u});
writer0.write(0.0);
std::filesystem::path f1 = "test_u1" + std::to_string(sizeof(T)) + ".bp";
io::FidesWriter<T> writer1(mesh->comm(), f1, {u, v});
writer1.write(0.0);
}
template <std::floating_point T>
void test_vtx_reuse_mesh()
{
auto mesh = std::make_shared<mesh::Mesh<T>>(
mesh::create_rectangle<T>(MPI_COMM_WORLD, {{{0.0, 0.0}, {1.0, 1.0}}},
{22, 12}, mesh::CellType::triangle));
// Create a Basix continuous Lagrange element of degree 1
basix::FiniteElement e = basix::create_element<T>(
basix::element::family::P,
mesh::cell_type_to_basix_type(mesh::CellType::triangle), 1,
basix::element::lagrange_variant::unset,
basix::element::dpc_variant::unset, false);
// Create a scalar function space
auto V = std::make_shared<fem::FunctionSpace<T>>(
fem::create_functionspace(mesh, e));
// Create a finite element Function
auto u = std::make_shared<fem::Function<T>>(V);
auto v = std::make_shared<fem::Function<std::complex<T>>>(V);
std::filesystem::path f
= "test_vtx_reuse_mesh" + std::to_string(sizeof(T)) + ".bp";
io::VTXWriter<T> writer(mesh->comm(), f, {u, v}, "BPFile",
io::VTXMeshPolicy::reuse);
writer.write(0);
std::ranges::fill(u->x()->mutable_array(), 1);
writer.write(1);
}
} // namespace
TEST_CASE("Fides output")
{
CHECK_NOTHROW(test_fides_mesh<float>());
CHECK_NOTHROW(test_fides_mesh<double>());
CHECK_NOTHROW(test_fides_function<float>());
CHECK_NOTHROW(test_fides_function<double>());
}
TEST_CASE("VTX reuse mesh")
{
CHECK_NOTHROW(test_vtx_reuse_mesh<float>());
CHECK_NOTHROW(test_vtx_reuse_mesh<double>());
}
#endif
|
