Wildmeshing Toolkit
test_simplicial_embedding.cpp
Go to the documentation of this file.
1 #include <catch2/catch_test_macros.hpp>
2 
3 #include <tools/DEBUG_TetMesh.hpp>
4 #include <tools/DEBUG_TriMesh.hpp>
5 #include <tools/TetMesh_examples.hpp>
6 #include <tools/TriMesh_examples.hpp>
7 #include <wmtk/Mesh.hpp>
8 #include <wmtk/components/simplicial_embedding/simplicial_embedding.hpp>
9 #include <wmtk/io/ParaviewWriter.hpp>
10 
11 using namespace wmtk;
12 using namespace components;
13 
14 const std::filesystem::path data_dir = WMTK_DATA_DIR;
15 
16 TEST_CASE(
17  "simplicial_embedding_component_tri",
18  "[components][simplicial_embedding][trimesh][2D][scheduler]")
19 {
20  tests::DEBUG_TriMesh m = wmtk::tests::hex_plus_two_with_position();
21 
22  SimplicialEmbeddingOptions options;
23  options.value = 1;
24  options.tag_attributes[PrimitiveType::Vertex] =
25  m.register_attribute<int64_t>("vertex_tag", wmtk::PrimitiveType::Vertex, 1);
26  options.tag_attributes[PrimitiveType::Edge] =
27  m.register_attribute<int64_t>("edge_tag", wmtk::PrimitiveType::Edge, 1);
28  options.tag_attributes[PrimitiveType::Triangle] =
29  m.register_attribute<int64_t>("face_tag", wmtk::PrimitiveType::Triangle, 1);
30  options.pass_through_attributes.emplace_back(
31  m.get_attribute_handle<double>("vertices", PrimitiveType::Vertex));
32 
33  SECTION("points_in_2d_case")
34  {
35  // 0---1---2
36  // / \ / \ / \ .
37  // 3---4---5---6
38  // \ / \ / .
39  // 7---8
40  // set 0 1 4 5 6
41  {
42  const std::vector<Tuple>& vertex_tuples = m.get_all(wmtk::PrimitiveType::Vertex);
43  attribute::Accessor<int64_t> acc_vertex_tag =
44  m.create_accessor<int64_t>(options.tag_attributes[PrimitiveType::Vertex]);
45  acc_vertex_tag.scalar_attribute(vertex_tuples[0]) = options.value;
46  acc_vertex_tag.scalar_attribute(vertex_tuples[1]) = options.value;
47  acc_vertex_tag.scalar_attribute(vertex_tuples[4]) = options.value;
48  acc_vertex_tag.scalar_attribute(vertex_tuples[5]) = options.value;
49  acc_vertex_tag.scalar_attribute(vertex_tuples[6]) = options.value;
50  }
51 
52  simplicial_embedding(m, options);
53 
54  CHECK(m.get_all(PrimitiveType::Vertex).size() == 15);
55 
56  if (false) {
57  wmtk::io::ParaviewWriter writer(
58  data_dir / "simplicial_embedding_result_0d_case",
59  "vertices",
60  m,
61  true,
62  true,
63  true,
64  false);
65  m.serialize(writer);
66  }
67  }
68  SECTION("edges_in_2d_case")
69  {
70  // 0---1---2
71  // / \ / \ / \ .
72  // 3---4---5---6
73  // \ / \ / .
74  // 7---8
75  // set vertex 0 1 4 5 6 7
76  // set edge 4-5 5-1 1-4 4-7 7-3
77  {
78  const std::vector<Tuple>& vertex_tuples = m.get_all(wmtk::PrimitiveType::Vertex);
79  attribute::Accessor<int64_t> acc_vertex_tag =
80  m.create_accessor<int64_t>(options.tag_attributes[PrimitiveType::Vertex]);
81  acc_vertex_tag.scalar_attribute(vertex_tuples[0]) = options.value;
82  acc_vertex_tag.scalar_attribute(vertex_tuples[1]) = options.value;
83  acc_vertex_tag.scalar_attribute(vertex_tuples[3]) = options.value;
84  acc_vertex_tag.scalar_attribute(vertex_tuples[4]) = options.value;
85  acc_vertex_tag.scalar_attribute(vertex_tuples[5]) = options.value;
86  acc_vertex_tag.scalar_attribute(vertex_tuples[6]) = options.value;
87  acc_vertex_tag.scalar_attribute(vertex_tuples[7]) = options.value;
88  attribute::Accessor<int64_t> acc_edge_tag =
89  m.create_accessor<int64_t>(options.tag_attributes[PrimitiveType::Edge]);
90  acc_edge_tag.scalar_attribute(m.edge_tuple_with_vs_and_t(4, 5, 2)) = options.value;
91  acc_edge_tag.scalar_attribute(m.edge_tuple_with_vs_and_t(5, 1, 2)) = options.value;
92  acc_edge_tag.scalar_attribute(m.edge_tuple_with_vs_and_t(1, 4, 2)) = options.value;
93  acc_edge_tag.scalar_attribute(m.edge_tuple_with_vs_and_t(7, 4, 5)) = options.value;
94  acc_edge_tag.scalar_attribute(m.edge_tuple_with_vs_and_t(7, 3, 5)) = options.value;
95  }
96 
97  simplicial_embedding(m, options);
98 
99  CHECK(m.get_all(PrimitiveType::Triangle).size() == 17);
100  CHECK(m.get_all(PrimitiveType::Vertex).size() == 15);
101 
102  if (false) {
103  ParaviewWriter writer(
104  data_dir / "simplicial_embedding_result_1d_case",
105  "vertices",
106  m,
107  true,
108  true,
109  true,
110  false);
111  m.serialize(writer);
112  }
113  }
114 }
115 
116 TEST_CASE(
117  "simplicial_embedding_component_tet",
118  "[components][simplicial_embedding][tetmesh][3D][scheduler][.]")
119 {
120  using namespace tests_3d;
121  // 0 ---------- 4
122  // / \\ // \ .
123  // / \ \ // \ .
124  // / \ \ // \ .
125  // / \ \3 \ .
126  // 1 --------- 2/ -------- 5 tuple edge 2-3
127  // \ / /\ \ / .
128  // \ / / \\ / .
129  // \ // \\ / .
130  // \ // \ / .
131  // 6 -----------7
132  //
133  DEBUG_TetMesh m = six_cycle_tets();
134 
135  SimplicialEmbeddingOptions options;
136  options.value = 1;
137  options.tag_attributes[PrimitiveType::Vertex] =
138  m.register_attribute<int64_t>("vertex_tag", wmtk::PrimitiveType::Vertex, 1);
139  options.tag_attributes[PrimitiveType::Edge] =
140  m.register_attribute<int64_t>("edge_tag", wmtk::PrimitiveType::Edge, 1);
141  options.tag_attributes[PrimitiveType::Triangle] =
142  m.register_attribute<int64_t>("face_tag", wmtk::PrimitiveType::Triangle, 1);
143  options.pass_through_attributes.emplace_back(
144  m.get_attribute_handle<double>("vertices", PrimitiveType::Vertex));
145 
146  SECTION("points_in_3d_case")
147  {
148  {
149  const std::vector<Tuple>& vertex_tuples = m.get_all(wmtk::PrimitiveType::Vertex);
150  attribute::Accessor<int64_t> acc_vertex_tag =
151  m.create_accessor<int64_t>(options.tag_attributes[PrimitiveType::Vertex]);
152  acc_vertex_tag.scalar_attribute(vertex_tuples[1]) = options.value;
153  acc_vertex_tag.scalar_attribute(vertex_tuples[2]) = options.value;
154  acc_vertex_tag.scalar_attribute(vertex_tuples[3]) = options.value;
155  }
156  simplicial_embedding(m, options);
157 
158  CHECK(false); // TODO add real checks
159 
160  if (false) {
161  ParaviewWriter writer(
162  data_dir / "simplicial_embedding_result_points_3d_case",
163  "vertices",
164  m,
165  true,
166  true,
167  true,
168  true);
169  m.serialize(writer);
170  }
171  }
172  SECTION("edges_in_3d_case")
173  {
174  {
175  const std::vector<Tuple>& vertex_tuples = m.get_all(wmtk::PrimitiveType::Vertex);
176  attribute::Accessor<int64_t> acc_vertex_tag =
177  m.create_accessor<int64_t>(options.tag_attributes[PrimitiveType::Vertex]);
178  acc_vertex_tag.scalar_attribute(vertex_tuples[0]) = options.value;
179  acc_vertex_tag.scalar_attribute(vertex_tuples[1]) = options.value;
180  acc_vertex_tag.scalar_attribute(vertex_tuples[2]) = options.value;
181  acc_vertex_tag.scalar_attribute(vertex_tuples[3]) = options.value;
182  acc_vertex_tag.scalar_attribute(vertex_tuples[5]) = options.value;
183  acc_vertex_tag.scalar_attribute(vertex_tuples[7]) = options.value;
184  attribute::Accessor<int64_t> acc_edge_tag =
185  m.create_accessor<int64_t>(options.tag_attributes[PrimitiveType::Edge]);
186  acc_edge_tag.scalar_attribute(m.edge_tuple_with_vs_and_t(0, 1, 0)) = options.value;
187  acc_edge_tag.scalar_attribute(m.edge_tuple_with_vs_and_t(0, 2, 0)) = options.value;
188  acc_edge_tag.scalar_attribute(m.edge_tuple_with_vs_and_t(0, 3, 0)) = options.value;
189  acc_edge_tag.scalar_attribute(m.edge_tuple_with_vs_and_t(1, 2, 0)) = options.value;
190  acc_edge_tag.scalar_attribute(m.edge_tuple_with_vs_and_t(2, 3, 0)) = options.value;
191  acc_edge_tag.scalar_attribute(m.edge_tuple_with_vs_and_t(3, 1, 0)) = options.value;
192  acc_edge_tag.scalar_attribute(m.edge_tuple_with_vs_and_t(2, 5, 2)) = options.value;
193  }
194  simplicial_embedding(m, options);
195 
196  CHECK(false); // TODO add real checks
197 
198  if (false) {
199  ParaviewWriter writer(
200  data_dir / "simplicial_embedding_result_edges_3d_case",
201  "vertices",
202  m,
203  true,
204  true,
205  true,
206  true);
207  m.serialize(writer);
208  }
209  }
210 }
wmtk::attribute::Accessor::scalar_attribute
T & scalar_attribute(const ArgType &t)
wmtk::components::simplicial_embedding
void simplicial_embedding(Mesh &mesh, const SimplicialEmbeddingOptions &options)
Definition: simplicial_embedding.cpp:13
wmtk
Definition: Accessor.hpp:6
simplicial_embedding.hpp
wmtk::components::SimplicialEmbeddingOptions::tag_attributes
std::map< PrimitiveType, attribute::MeshAttributeHandle > tag_attributes
All simplex dimensions must have an int64_t scalar attribute where the tags are stored.
Definition: SimplicialEmbeddingOptions.hpp:12
wmtk::components::SimplicialEmbeddingOptions::value
int64_t value
The value that should be simplicially embedded.
Definition: SimplicialEmbeddingOptions.hpp:16
wmtk::components::SimplicialEmbeddingOptions::pass_through_attributes
std::vector< attribute::MeshAttributeHandle > pass_through_attributes
Other attributes that should be processed with the default behavior.
Definition: SimplicialEmbeddingOptions.hpp:20
data_dir
const std::filesystem::path data_dir
Definition: test_simplicial_embedding.cpp:14
TEST_CASE
TEST_CASE("simplicial_embedding_component_tri", "[components][simplicial_embedding][trimesh][2D][scheduler]")
Definition: test_simplicial_embedding.cpp:16