Wildmeshing Toolkit
Loading...
Searching...
No Matches
TetMesh.h
1#pragma once
2
3#include <wmtk/utils/VectorUtils.h>
4#include <type_traits>
5#include <wmtk/AttributeCollection.hpp>
6#include <wmtk/Types.hpp>
7#include <wmtk/simplex/RawSimplex.hpp>
8#include <wmtk/utils/Logger.hpp>
9
10#include <tbb/concurrent_vector.h>
11#include <tbb/enumerable_thread_specific.h>
12#include <tbb/spin_mutex.h>
13
14#include <array>
15#include <cassert>
16#include <limits>
17#include <map>
18#include <optional>
19#include <queue>
20#include <vector>
21
22namespace wmtk {
24{
25private:
30 static constexpr std::array<std::array<int, 2>, 6> m_local_edges = {
31 {{{0, 1}}, {{1, 2}}, {{0, 2}}, {{0, 3}}, {{1, 3}}, {{2, 3}}}};
32
33 static constexpr std::array<int, 4> m_map_vertex2edge = {{0, 0, 1, 3}};
34 static constexpr std::array<int, 4> m_map_vertex2oppo_face = {{3, 1, 2, 0}};
35 static constexpr std::array<int, 6> m_map_edge2face = {{0, 0, 0, 1, 2, 1}};
36 static constexpr std::array<std::array<int, 3>, 4> m_local_faces = {
37 {{{0, 1, 2}}, {{0, 2, 3}}, {{0, 1, 3}}, {{1, 2, 3}}}}; // sorted local vids
38 static constexpr std::array<std::array<int, 3>, 4> m_local_edges_in_a_face = {
39 {{{0, 1, 2}}, {{2, 5, 3}}, {{3, 4, 0}}, {{5, 1, 4}}}};
40
41public:
42 // Cell Tuple Navigator
48 class Tuple
49 {
50 size_t m_global_vid = std::numeric_limits<size_t>::max();
51 size_t m_local_eid = std::numeric_limits<size_t>::max();
52 size_t m_local_fid = std::numeric_limits<size_t>::max();
53 size_t m_global_tid = std::numeric_limits<size_t>::max();
54
55 int m_hash = 0;
56
57 private:
67 Tuple(const TetMesh& m, size_t vid, size_t local_eid, size_t local_fid, size_t tid);
68
69 public:
70 Tuple() {}
71
72 friend TetMesh;
73
81 bool is_valid(const TetMesh& m) const;
88 bool is_boundary_edge(const TetMesh& m) const;
95 bool is_boundary_face(const TetMesh& m) const;
96
101 void print_info() const;
102
108 void print_info(const TetMesh& m) const;
109
115 size_t vid(const TetMesh& m) const;
116
125 size_t eid(const TetMesh& m) const;
126
134 size_t fid(const TetMesh& m) const;
135
142 size_t tid(const TetMesh& m) const;
143
150 Tuple switch_vertex(const TetMesh& m) const;
157 Tuple switch_edge(const TetMesh& m) const;
164 Tuple switch_face(const TetMesh& m) const;
165
173 std::optional<Tuple> switch_tetrahedron(const TetMesh& m) const;
174 std::optional<Tuple> switch_tetrahedron_slow(const TetMesh& m) const;
175
176
178
181 void check_validity(const TetMesh& m) const;
182 friend bool operator==(const Tuple& a, const Tuple& t)
183 {
184 return (
185 std::tie(a.m_global_vid, a.m_local_eid, a.m_local_fid, a.m_global_tid, a.m_hash) ==
186 std::tie(t.m_global_vid, t.m_local_eid, t.m_local_fid, t.m_global_tid, t.m_hash));
187 }
188 friend bool operator<(const Tuple& a, const Tuple& t)
189 {
190 return (
191 std::tie(a.m_global_vid, a.m_local_eid, a.m_local_fid, a.m_global_tid, a.m_hash) <
192 std::tie(t.m_global_vid, t.m_local_eid, t.m_local_fid, t.m_global_tid, t.m_hash));
193 }
194 };
195
201 {
202 Tuple m_tuple;
203 const TetMesh& m_mesh;
204
205 public:
206 SmartTuple(const TetMesh& mesh, const Tuple& t)
207 : m_mesh(mesh)
208 , m_tuple(t)
209 {}
210
211 const Tuple& tuple() { return m_tuple; }
212 const TetMesh& mesh() { return m_mesh; }
213
214 SmartTuple& operator=(const SmartTuple& t)
215 {
216 m_tuple = t.m_tuple;
217 return *this;
218 }
219
220 bool is_valid() const { return m_tuple.is_valid(m_mesh); }
221 bool is_boundary_edge() const { return m_tuple.is_boundary_edge(m_mesh); }
222 bool is_boundary_face() const { return m_tuple.is_boundary_face(m_mesh); }
223 size_t vid() const { return m_tuple.vid(m_mesh); }
224 size_t eid() const { return m_tuple.eid(m_mesh); }
225 size_t fid() const { return m_tuple.fid(m_mesh); }
226 size_t tid() const { return m_tuple.tid(m_mesh); }
227 SmartTuple switch_vertex() const { return {m_mesh, m_tuple.switch_vertex(m_mesh)}; }
228 SmartTuple switch_edge() const { return {m_mesh, m_tuple.switch_edge(m_mesh)}; }
229 SmartTuple switch_face() const { return {m_mesh, m_tuple.switch_face(m_mesh)}; }
230 std::optional<SmartTuple> switch_tetrahedron() const
231 {
232 const std::optional<Tuple> t = m_tuple.switch_tetrahedron(m_mesh);
233 if (t) {
234 return std::optional<SmartTuple>({m_mesh, t.value()});
235 }
236 return {};
237 }
238 void check_validity() const { return m_tuple.check_validity(m_mesh); }
239 };
240
247 {
248 public:
249 std::vector<size_t> m_conn_tets; // todo: always keep it sorted
250 bool m_is_removed = false;
251
252 size_t& operator[](const size_t index)
253 {
254 assert(index < m_conn_tets.size());
255 return m_conn_tets[index];
256 }
257
258 size_t operator[](const size_t index) const
259 {
260 assert(index < m_conn_tets.size());
261 return m_conn_tets[index];
262 }
263
264 friend bool operator==(const VertexConnectivity& l, const VertexConnectivity& r)
265 {
266 return std::tie(l.m_conn_tets, l.m_is_removed) ==
267 std::tie(r.m_conn_tets, r.m_is_removed); // keep the same order
268 }
269
270 void print_info() {}
271 };
272
278 {
279 public:
280 std::array<size_t, 4> m_indices;
281 bool m_is_removed = false;
282
283 int hash = 0;
284
285 size_t& operator[](size_t index)
286 {
287 assert(index < 4);
288 return m_indices[index];
289 }
290
291 size_t operator[](size_t index) const
292 {
293 assert(index < 4);
294 return m_indices[index];
295 }
296
297 int find(size_t v_id) const
298 {
299 for (int j = 0; j < 4; j++) {
300 if (v_id == m_indices[j]) return j;
301 }
302 return -1;
303 }
304
305 int find_local_edge(size_t v1_id, size_t v2_id) const
306 {
307 std::array<int, 2> e;
308 for (int j = 0; j < 4; j++) {
309 if (v1_id == m_indices[j])
310 e[0] = j;
311 else if (v2_id == m_indices[j])
312 e[1] = j;
313 }
314 if (e[0] > e[1]) std::swap(e[0], e[1]);
315 int i =
316 std::find(m_local_edges.begin(), m_local_edges.end(), e) - m_local_edges.begin();
317 if (i >= m_local_edges.size()) return -1;
318 return i;
319 }
320
321 int find_local_face(size_t v1_id, size_t v2_id, size_t v3_id) const
322 {
323 std::array<int, 3> f;
324 for (int j = 0; j < 4; j++) {
325 if (v1_id == m_indices[j])
326 f[0] = j;
327 else if (v2_id == m_indices[j])
328 f[1] = j;
329 else if (v3_id == m_indices[j])
330 f[2] = j;
331 }
332 std::sort(f.begin(), f.end());
333 int i =
334 std::find(m_local_faces.begin(), m_local_faces.end(), f) - m_local_faces.begin();
335 if (i >= m_local_edges.size()) return -1;
336 return i;
337 }
338
339 friend bool operator==(const TetrahedronConnectivity& l, const TetrahedronConnectivity& r)
340 {
341 return std::tie(l.m_indices, l.m_is_removed, l.hash) ==
342 std::tie(r.m_indices, r.m_is_removed, r.hash); // keep the same order
343 }
344
345 void print_info() {}
346 };
347
348 TetMesh();
349 virtual ~TetMesh() = default;
355 size_t vert_capacity() const { return current_vert_size; }
361 size_t tet_capacity() const { return current_tet_size; }
362
367 size_t vertex_size() const
368 {
369 int cnt = 0;
370 for (auto i = 0; i < vert_capacity(); i++) {
371 if (!m_vertex_connectivity[i].m_is_removed) cnt++;
372 }
373 return cnt;
374 }
379 size_t tet_size() const
380 {
381 int cnt = 0;
382 for (auto i = 0; i < tet_capacity(); i++) {
383 if (!m_tet_connectivity[i].m_is_removed) cnt++;
384 }
385 return cnt;
386 }
396 void init(size_t n_vertices, const std::vector<std::array<size_t, 4>>& tets);
397 void init_with_isolated_vertices(
398 size_t n_vertices,
399 const std::vector<std::array<size_t, 4>>& tets);
400
406 void init(const MatrixXi& T);
407
415 bool split_edge(const Tuple& t, std::vector<Tuple>& new_tets);
423 virtual bool collapse_edge(const Tuple& t, std::vector<Tuple>& new_tets);
424
425 bool link_condition(const Tuple& t);
426
445 const Tuple& loc0,
446 std::vector<Tuple>& new_edges,
447 size_t& v1_id,
448 Tuple& new_loc,
449 std::map<size_t, wmtk::TetMesh::VertexConnectivity>& rollback_vert_conn,
450 std::vector<size_t>& n1_t_ids_copy,
451 std::vector<size_t>& new_tet_id,
452 std::vector<TetrahedronConnectivity>& old_tets);
453
462 bool collapse_edge_check_topology(const std::vector<size_t>& new_tet_id);
463
477 size_t& v1_id,
478 std::map<size_t, wmtk::TetMesh::VertexConnectivity>& rollback_vert_conn,
479 std::vector<size_t>& n1_t_ids,
480 std::vector<size_t>& new_tet_id,
481 std::vector<TetrahedronConnectivity>& old_tets);
482
491 bool swap_edge_56(const Tuple& t, std::vector<Tuple>& new_tets);
500 bool swap_edge_44(const Tuple& t, std::vector<Tuple>& new_tets);
509 bool swap_edge(const Tuple& t, std::vector<Tuple>& new_tets);
516 bool swap_face(const Tuple& t, std::vector<Tuple>& new_tets);
524 bool smooth_vertex(const Tuple& t);
525
534 bool split_tet(const Tuple& t, std::vector<Tuple>& new_tets);
535
546 bool split_face(const Tuple& t, std::vector<Tuple>& new_tets);
547
556 const std::vector<Tuple>& intersected_tets,
557 const std::vector<Tuple>& intersected_edges,
558 std::vector<size_t>& new_edge_vids,
559 std::vector<size_t>& new_center_vids,
560 std::vector<std::array<size_t, 4>>& center_split_tets);
561
571 bool insert_point(const Tuple& t, std::vector<Tuple>& new_tets);
572 virtual bool insert_point_before(const Tuple& t) { return true; };
573 virtual bool insert_point_after(std::vector<Tuple>& new_tets) { return true; };
579 void consolidate_mesh();
580
586 std::vector<Tuple> get_edges() const;
592 std::vector<Tuple> get_faces() const;
598 std::vector<Tuple> get_vertices() const;
604 std::vector<Tuple> get_tets() const;
609 // virtual void for_each_edge(const std::function<void(const TetMesh::Tuple&)>&);
610
611 // TODO: make this concurrent
616 virtual void for_each_face(const std::function<void(const TetMesh::Tuple&)>&);
617 // /**
618 // * @brief looping through all the unique vertices and perform the given function
619 // *
620 // */
621 // virtual void for_each_vertex(const std::function<void(const TetMesh::Tuple&)>&);
622 // /**
623 // * @brief looping through all the unique tet and perform the given function
624 // *
625 // */
626 // virtual void for_each_tetra(const std::function<void(const TetMesh::Tuple&)>&);
627
628public:
629 template <typename T>
630 using vector = tbb::concurrent_vector<T>;
631
632public:
633 AbstractAttributeContainer* p_vertex_attrs = nullptr;
634 AbstractAttributeContainer* p_edge_attrs = nullptr;
635 AbstractAttributeContainer* p_face_attrs = nullptr;
636 AbstractAttributeContainer* p_tet_attrs = nullptr;
637 // AbstractAttributeContainer vertex_attrs, edge_attrs, face_attrs, tet_attrs;
638
639
640private:
641 // Stores the connectivity of the mesh
642 vector<VertexConnectivity> m_vertex_connectivity;
643 vector<TetrahedronConnectivity> m_tet_connectivity;
644 std::atomic_long current_vert_size;
645 std::atomic_long current_tet_size;
646 tbb::spin_mutex vertex_connectivity_lock;
647 tbb::spin_mutex tet_connectivity_lock;
648 bool vertex_connectivity_synchronizing_flag = false;
649 bool tet_connectivity_synchronizing_flag = false;
650 int MAX_THREADS = 128;
651
652 int m_t_empty_slot = 0;
653 int m_v_empty_slot = 0;
654 int get_next_empty_slot_t();
655 int get_next_empty_slot_v();
656
657 // TODO: subdivide_tets function should not be in the TetMesh API.
658 void subdivide_tets(
659 const std::vector<size_t> t_ids,
660 const std::vector<bool>& mark_surface,
661 const std::map<std::array<size_t, 2>, size_t>& map_edge2vid,
662 std::map<std::array<size_t, 3>, std::vector<std::array<size_t, 5>>>& new_face_vids,
663 const std::vector<size_t>& new_vids,
664 std::vector<size_t>& new_tids,
665 std::vector<size_t>& new_center_vids,
666 std::vector<std::array<size_t, 4>>& center_split_tets);
667 void subdivide_a_tet(
668 size_t t_id,
669 const std::array<int, 6>& new_v_ids,
670 bool mark_surface,
671 std::map<std::array<size_t, 3>, std::vector<std::array<size_t, 5>>>& new_face_vids,
672 std::vector<size_t>& new_tids,
673 std::vector<size_t>& new_center_vids,
674 std::vector<std::array<size_t, 4>>& center_split_tets);
675
676public:
677 virtual bool invariants(const std::vector<Tuple>&) { return true; }
678
679protected:
680 virtual bool triangle_insertion_before(const std::vector<Tuple>& faces) { return true; }
681 virtual bool triangle_insertion_after(const std::vector<std::vector<Tuple>>&) { return true; }
682
684 // Checks if the split should be performed or not (user controlled)
691 virtual bool split_edge_before(const Tuple& t) { return true; } // check edge condition
692 // This function computes the attributes for the added simplices
693 // if it returns false then the operation is undone
700 virtual bool split_edge_after(const Tuple& t) { return true; } // check tet condition
701
703 // Checks if the collapse should be performed or not (user controlled)
711 virtual bool collapse_edge_before(const Tuple& t) { return true; }
712 // If it returns false then the operation is undone (the tuple indexes a vertex and tet that
713 // survived)
720 virtual bool collapse_edge_after(const Tuple& t) { return true; }
728 virtual bool swap_edge_44_before(const Tuple& t) { return true; }
739 virtual double swap_edge_44_energy(
740 const std::vector<std::array<size_t, 4>>& tets,
741 const int op_case)
742 {
743 return -op_case;
744 }
751 virtual bool swap_edge_44_after(const Tuple& t) { return true; }
759 virtual bool swap_edge_56_before(const Tuple& t) { return true; }
770 virtual double swap_edge_56_energy(
771 const std::vector<std::array<size_t, 4>>& tets,
772 const int op_case)
773 {
774 return -op_case;
775 }
782 virtual bool swap_edge_56_after(const Tuple& t) { return true; }
790 virtual bool swap_edge_before(const Tuple& t) { return true; }
797 virtual bool swap_edge_after(const Tuple& t) { return true; }
805 virtual bool swap_face_before(const Tuple& t) { return true; }
812 virtual bool swap_face_after(const Tuple& t) { return true; }
819 virtual bool smooth_before(const Tuple& t) { return true; }
826 virtual bool smooth_after(const Tuple& t) { return true; }
827
834 virtual bool split_face_before(const Tuple& t) { return true; }
835
843 virtual bool split_face_after(const Tuple& t) { return true; }
844
851 virtual bool split_tet_before(const Tuple& t) { return true; }
852
860 virtual bool split_tet_after(const Tuple& t) { return true; }
861
862 // virtual void resize_vertex_mutex(size_t v) {}
863
864public:
871 Tuple tuple_from_edge(size_t tid, int local_eid) const;
877 Tuple tuple_from_edge(const std::array<size_t, 2>& vids) const;
878
885 Tuple tuple_from_face(size_t tid, int local_fid) const;
886
892 std::tuple<Tuple, size_t> tuple_from_face(const std::array<size_t, 3>& vids) const;
893 std::tuple<Tuple, size_t> tuple_from_face(const simplex::Face& f) const;
894
900 Tuple tuple_from_vertex(size_t vid) const;
901
907 Tuple tuple_from_tet(size_t tid) const;
908
913 Tuple tuple_from_vids(size_t vid0, size_t vid1, size_t vid2, size_t vid3) const;
914
915 simplex::Tet simplex_from_tet(const Tuple& t) const;
916 simplex::Tet simplex_from_tet(const size_t tid) const;
917
921 Tuple switch_vertex(const Tuple& t) const
922 {
923 auto loc = t.switch_vertex(*this);
925 return loc;
926 }
930 Tuple switch_edge(const Tuple& t) const
931 {
932 auto loc = t.switch_edge(*this);
934 return loc;
935 }
939 Tuple switch_face(const Tuple& t) const
940 {
941 auto loc = t.switch_face(*this);
943 return loc;
944 }
948 std::optional<Tuple> switch_tetrahedron(const Tuple& t) const
949 {
950 auto loc = t.switch_tetrahedron(*this);
951 if (loc.has_value()) check_tuple_validity(loc.value());
952 return loc;
953 }
954
961 std::vector<Tuple> get_one_ring_tets_for_vertex(const Tuple& t) const;
968 std::vector<size_t> get_one_ring_tids_for_vertex(const Tuple& t) const;
969 std::vector<size_t> get_one_ring_tids_for_vertex(const size_t vid) const;
970
977 std::vector<Tuple> get_one_ring_vertices_for_vertex(const Tuple& t) const;
985 std::vector<size_t> get_one_ring_vids_for_vertex(size_t vid, std::vector<size_t>& cache);
992 std::vector<size_t> get_one_ring_vids_for_vertex(size_t vid) const;
997 std::vector<size_t> get_one_ring_vids_for_vertex_adj(size_t vid) const;
1002 std::vector<size_t> get_one_ring_vids_for_vertex_adj(size_t vid, std::vector<size_t>& cache);
1003
1010 std::vector<Tuple> get_incident_tets_for_edge(const Tuple& t) const;
1011 std::vector<Tuple> get_incident_tets_for_edge(const size_t vid0, const size_t vid1) const;
1012
1013 std::vector<size_t> get_incident_tids_for_edge(const Tuple& t) const;
1014 std::vector<size_t> get_incident_tids_for_edge(const size_t vid0, const size_t vid1) const;
1015
1022 std::vector<Tuple> get_one_ring_tets_for_edge(const Tuple& t) const;
1023
1030 std::vector<std::array<size_t, 3>> vertex_adjacent_boundary_faces(const Tuple& t) const;
1035 std::array<Tuple, 4> oriented_tet_vertices(const Tuple& t) const;
1040 std::array<size_t, 4> oriented_tet_vids(const Tuple& t) const;
1041 std::array<size_t, 4> oriented_tet_vids(const size_t tid) const;
1048 std::array<Tuple, 3> get_face_vertices(const Tuple& t) const;
1049 std::array<size_t, 3> get_face_vids(const Tuple& t) const;
1050
1058 std::array<Tuple, 6> tet_edges(const Tuple& t) const;
1062 void check_tuple_validity(const Tuple& t) const { t.check_validity(*this); }
1075 void remove_tets_by_ids(const std::vector<size_t>& tids)
1076 {
1077 for (size_t tid : tids) {
1078 m_tet_connectivity[tid].m_is_removed = true;
1079 for (int j = 0; j < 4; j++)
1080 vector_erase(m_vertex_connectivity[m_tet_connectivity[tid][j]].m_conn_tets, tid);
1081 }
1082 for (auto& v : m_vertex_connectivity) {
1083 if (v.m_is_removed) continue;
1084 if (v.m_conn_tets.empty()) v.m_is_removed = true;
1085 }
1086 }
1087 bool m_collapse_check_link_condition = true; // classical link condition
1088 bool m_collapse_check_topology = false; // sanity check
1089 bool m_collapse_check_manifold = true; // manifoldness check after collapse
1090
1091private:
1092 std::map<size_t, VertexConnectivity> operation_update_connectivity_impl(
1093 std::vector<size_t>& affected_tid,
1094 const std::vector<std::array<size_t, 4>>& new_tet_conn);
1095 void operation_failure_rollback_imp(
1096 std::map<size_t, VertexConnectivity>& rollback_vert_conn,
1097 const std::vector<size_t>& affected,
1098 const std::vector<size_t>& new_tet_id,
1099 const std::vector<TetrahedronConnectivity>& old_tets);
1100 std::map<size_t, VertexConnectivity> operation_update_connectivity_impl(
1101 const std::vector<size_t>& remove_id,
1102 const std::vector<std::array<size_t, 4>>& new_tet_conn,
1103 std::vector<size_t>& allocate_id);
1104 static std::vector<TetrahedronConnectivity> record_old_tet_connectivity(
1105 const TetMesh::vector<TetrahedronConnectivity>& conn,
1106 const std::vector<size_t>& tets)
1107 {
1108 std::vector<TetrahedronConnectivity> tet_conn;
1109 for (size_t i : tets) {
1110 tet_conn.push_back(conn[i]);
1111 }
1112 return tet_conn;
1113 }
1114
1115public:
1116 void start_protect_attributes()
1117 {
1118 if (p_vertex_attrs) {
1119 p_vertex_attrs->begin_protect();
1120 }
1121 if (p_edge_attrs) {
1122 p_edge_attrs->begin_protect();
1123 }
1124 if (p_face_attrs) {
1125 p_face_attrs->begin_protect();
1126 }
1127 if (p_tet_attrs) {
1128 p_tet_attrs->begin_protect();
1129 }
1130 }
1131
1132 void release_protect_attributes()
1133 {
1134 if (p_vertex_attrs) {
1135 p_vertex_attrs->end_protect();
1136 }
1137 if (p_edge_attrs) {
1138 p_edge_attrs->end_protect();
1139 }
1140 if (p_face_attrs) {
1141 p_face_attrs->end_protect();
1142 }
1143 if (p_tet_attrs) {
1144 p_tet_attrs->end_protect();
1145 }
1146 }
1147
1148 void rollback_protected_attributes()
1149 {
1150 if (p_vertex_attrs) {
1151 p_vertex_attrs->rollback();
1152 }
1153 if (p_edge_attrs) {
1154 p_edge_attrs->rollback();
1155 }
1156 if (p_face_attrs) {
1157 p_face_attrs->rollback();
1158 }
1159 if (p_tet_attrs) {
1160 p_tet_attrs->rollback();
1161 }
1162 }
1163
1164public:
1166 {
1167 tbb::spin_mutex mutex;
1168 int owner = std::numeric_limits<int>::max();
1169
1170 public:
1171 bool trylock() { return mutex.try_lock(); }
1172
1173 void unlock()
1174 {
1175 mutex.unlock();
1176 reset_owner();
1177 }
1178
1179 int get_owner() { return owner; }
1180
1181 void set_owner(int n) { owner = n; }
1182
1183 void reset_owner() { owner = INT_MAX; }
1184 };
1185
1186private:
1187 tbb::concurrent_vector<VertexMutex> m_vertex_mutex;
1188
1189 bool try_set_vertex_mutex(const Tuple& v, int threadid)
1190 {
1191 bool got = m_vertex_mutex[v.vid(*this)].trylock();
1192 if (got) m_vertex_mutex[v.vid(*this)].set_owner(threadid);
1193 return got;
1194 }
1195 bool try_set_vertex_mutex(size_t vid, int threadid)
1196 {
1197 bool got = m_vertex_mutex[vid].trylock();
1198 if (got) m_vertex_mutex[vid].set_owner(threadid);
1199 return got;
1200 }
1201
1202 void unlock_vertex_mutex(const Tuple& v) { m_vertex_mutex[v.vid(*this)].unlock(); }
1203 void unlock_vertex_mutex(size_t vid) { m_vertex_mutex[vid].unlock(); }
1204
1205protected:
1206 void resize_vertex_mutex(size_t v) { m_vertex_mutex.grow_to_at_least(v); }
1207
1208public:
1209 tbb::enumerable_thread_specific<std::vector<size_t>> mutex_release_stack;
1210 tbb::enumerable_thread_specific<std::vector<size_t>> get_one_ring_cache;
1211
1212 // void init(size_t n_vertices, const std::vector<std::array<size_t, 4>>& tets);
1213 int release_vertex_mutex_in_stack();
1214
1215 // helpers
1223 bool try_set_vertex_mutex_two_ring(const Tuple& v, int threadid);
1231 bool try_set_vertex_mutex_two_ring_vid(const Tuple& v, int threadid);
1240 bool try_set_vertex_mutex_two_ring_vid(size_t v, int threadid);
1241
1242 // can be called
1251 bool try_set_edge_mutex_two_ring(const Tuple& e, int threadid = 0);
1260 bool try_set_face_mutex_two_ring(const Tuple& f, int threadid = 0);
1272 const Tuple& v1,
1273 const Tuple& v2,
1274 const Tuple& v3,
1275 int threadid = 0);
1286 bool try_set_face_mutex_two_ring(size_t v1, size_t v2, size_t v3, int threadid = 0);
1294 bool try_set_vertex_mutex_one_ring(const Tuple& v, int threadid = 0);
1295
1296public:
1301 void for_each_edge(const std::function<void(const TetMesh::Tuple&)>&);
1306 void for_each_vertex(const std::function<void(const TetMesh::Tuple&)>&);
1311 void for_each_tetra(const std::function<void(const TetMesh::Tuple&)>&);
1312 int NUM_THREADS = 0;
1313};
1314
1315
1316} // namespace wmtk
Definition TetMesh.h:201
a Tuple refers to a global vid and a global tet id, and a local edge id and local face id
Definition TetMesh.h:49
Tuple switch_face(const TetMesh &m) const
Definition TetMeshTuple.cpp:253
bool is_valid(const TetMesh &m) const
Definition TetMeshTuple.cpp:57
bool is_boundary_face(const TetMesh &m) const
Definition TetMeshTuple.cpp:17
void print_info() const
prints the tuple
Definition TetMeshTuple.cpp:68
std::optional< Tuple > switch_tetrahedron(const TetMesh &m) const
Definition TetMeshTuple.cpp:273
size_t eid(const TetMesh &m) const
Definition TetMeshTuple.cpp:111
Tuple switch_edge(const TetMesh &m) const
Definition TetMeshTuple.cpp:238
Tuple switch_vertex(const TetMesh &m) const
Definition TetMeshTuple.cpp:224
size_t tid(const TetMesh &m) const
Definition TetMeshTuple.cpp:219
size_t vid(const TetMesh &m) const
Definition TetMeshTuple.cpp:106
size_t fid(const TetMesh &m) const
Definition TetMeshTuple.cpp:131
bool is_boundary_edge(const TetMesh &m) const
Definition TetMeshTuple.cpp:34
void check_validity(const TetMesh &m) const
check Tuple validity and connectivity validity
Definition TetMeshTuple.cpp:358
Definition TetMesh.h:247
Definition TetMesh.h:1166
Definition TetMesh.h:24
virtual bool swap_edge_44_after(const Tuple &t)
User specified modifications and desideratas for after a 4-4 edge swap.
Definition TetMesh.h:751
std::vector< Tuple > get_one_ring_vertices_for_vertex(const Tuple &t) const
Get the one ring vertices for a vertex.
Definition TetMesh.cpp:712
virtual bool swap_edge_44_before(const Tuple &t)
User specified preparations and desideratas for an 4-4 edge swap before changing the connectivity.
Definition TetMesh.h:728
virtual bool split_tet_before(const Tuple &t)
User specified preparations and desideratas for a tet split before changing the connectivity.
Definition TetMesh.h:851
size_t tet_size() const
get the number of unremoved tets
Definition TetMesh.h:379
std::vector< Tuple > get_faces() const
Definition TetMesh.cpp:219
std::array< Tuple, 6 > tet_edges(const Tuple &t) const
get the 6 edges of a tet represented by Tuples
Definition TetMesh.cpp:677
virtual bool smooth_after(const Tuple &t)
User specified modifications and desideratas for after smoothing a vertex.
Definition TetMesh.h:826
void consolidate_mesh()
cleans up the deleted vertices or tetrahedra, fixes the corresponding indices, and reset the version ...
Definition TetMesh.cpp:836
size_t vert_capacity() const
get the current largest global vid
Definition TetMesh.h:355
virtual bool collapse_edge_after(const Tuple &t)
User specified modifications and desideratas for after an edge collapse.
Definition TetMesh.h:720
virtual bool swap_edge_after(const Tuple &t)
User specified modifications and desideratas for after a 3-2 edge swap.
Definition TetMesh.h:797
bool smooth_vertex(const Tuple &t)
Definition TetMesh.cpp:339
bool insert_point(const Tuple &t, std::vector< Tuple > &new_tets)
Insert a point into a tetmesh inside a tet. In general position, this split a tet into 4....
Definition TetMeshTriangleInsertionConn.cpp:412
bool split_edge(const Tuple &t, std::vector< Tuple > &new_tets)
Definition TetMeshEdgeSplittingConn.cpp:6
Tuple switch_face(const Tuple &t) const
wrapper function from Tuple::switch_face
Definition TetMesh.h:939
std::array< Tuple, 4 > oriented_tet_vertices(const Tuple &t) const
Definition TetMesh.cpp:637
void for_each_vertex(const std::function< void(const TetMesh::Tuple &)> &)
perform the given function for each vertex
Definition TetMesh.cpp:1355
bool try_set_vertex_mutex_one_ring(const Tuple &v, int threadid=0)
try lock the one-ring neighboring traingles' incident vertices
Definition TetMesh.cpp:1269
std::array< size_t, 4 > oriented_tet_vids(const Tuple &t) const
Definition TetMesh.cpp:649
bool collapse_edge_check_topology(const std::vector< size_t > &new_tet_id)
Check topology after collapse connectivity change. This is a sanity check and should not be necessary...
Definition TetMeshEdgeCollapsingConn.cpp:419
bool try_set_vertex_mutex_two_ring_vid(const Tuple &v, int threadid)
try lock the two-ring neighboring traingles' incident vertices using vids
Definition TetMesh.cpp:974
Tuple tuple_from_tet(size_t tid) const
get a Tuple from global tetra index
Definition TetMesh.cpp:580
void subdivide_tets(const std::vector< size_t > t_ids, const std::vector< bool > &mark_surface, const std::map< std::array< size_t, 2 >, size_t > &map_edge2vid, std::map< std::array< size_t, 3 >, std::vector< std::array< size_t, 5 > > > &new_face_vids, const std::vector< size_t > &new_vids, std::vector< size_t > &new_tids, std::vector< size_t > &new_center_vids, std::vector< std::array< size_t, 4 > > &center_split_tets)
Definition TetMeshTriangleInsertionConn.cpp:141
void init(size_t n_vertices, const std::vector< std::array< size_t, 4 > > &tets)
Definition TetMesh.cpp:77
void subdivide_a_tet(size_t t_id, const std::array< int, 6 > &new_v_ids, bool mark_surface, std::map< std::array< size_t, 3 >, std::vector< std::array< size_t, 5 > > > &new_face_vids, std::vector< size_t > &new_tids, std::vector< size_t > &new_center_vids, std::vector< std::array< size_t, 4 > > &center_split_tets)
Definition TetMeshTriangleInsertionConn.cpp:234
virtual bool swap_edge_56_before(const Tuple &t)
User specified preparations and desideratas for a 5-6 edge swap before changing the connectivity.
Definition TetMesh.h:759
std::vector< Tuple > get_incident_tets_for_edge(const Tuple &t) const
Get the incident tets for edge.
Definition TetMesh.cpp:781
virtual bool split_face_after(const Tuple &t)
Compute the attributes for the added simplices.
Definition TetMesh.h:843
bool swap_edge_56(const Tuple &t, std::vector< Tuple > &new_tets)
Definition TetMeshSwapMeshConnectivity.cpp:690
bool split_face(const Tuple &t, std::vector< Tuple > &new_tets)
Split a face in 3 faces.
Definition TetMeshFaceSplittingConn.cpp:6
bool link_condition(const Tuple &t)
Definition TetMeshEdgeCollapsingConn.cpp:310
std::vector< Tuple > get_one_ring_tets_for_edge(const Tuple &t) const
Get the one ring tets for edge.
Definition TetMesh.cpp:816
virtual bool split_face_before(const Tuple &t)
User specified preparations and desideratas for a face split before changing the connectivity.
Definition TetMesh.h:834
virtual double swap_edge_44_energy(const std::vector< std::array< size_t, 4 > > &tets, const int op_case)
User specified energy to decide which of the 4 possible orientations should be chosen.
Definition TetMesh.h:739
Tuple tuple_from_vertex(size_t vid) const
get a Tuple from global vertex index
Definition TetMesh.cpp:567
virtual void for_each_face(const std::function< void(const TetMesh::Tuple &)> &)
looping through all the unique edges and perform the given function
Definition TetMesh.cpp:205
std::vector< size_t > get_one_ring_vids_for_vertex(size_t vid, std::vector< size_t > &cache)
Get the one ring vids for vertex.
Definition TetMesh.cpp:757
std::optional< Tuple > switch_tetrahedron(const Tuple &t) const
wrapper function from Tuple::switch_tetrahedron
Definition TetMesh.h:948
std::vector< Tuple > get_tets() const
Definition TetMesh.cpp:308
virtual bool smooth_before(const Tuple &t)
User specified preparations and desideratas for smoothing a vertex.
Definition TetMesh.h:819
virtual bool split_edge_before(const Tuple &t)
User specified preparations and desideratas for an edge split before changing the connectivity.
Definition TetMesh.h:691
std::vector< size_t > get_one_ring_vids_for_vertex_adj(size_t vid) const
Duplicate of the function TetMesh::get_one_ring_vids_for_vertex.
Definition TetMesh.cpp:743
virtual bool swap_edge_before(const Tuple &t)
User specified preparations and desideratas for an 3-2 edge swap before changing the conenctivity.
Definition TetMesh.h:790
void triangle_insertion(const std::vector< Tuple > &intersected_tets, const std::vector< Tuple > &intersected_edges, std::vector< size_t > &new_edge_vids, std::vector< size_t > &new_center_vids, std::vector< std::array< size_t, 4 > > &center_split_tets)
Insert a triangle into a tetmesh, with known intersection information.
Definition TetMeshTriangleInsertionConn.cpp:11
bool check_mesh_connectivity_validity() const
checks the validity of the connectivity of the mesh. Including the validity of each Tuple
Definition TetMesh.cpp:234
bool split_tet(const Tuple &t, std::vector< Tuple > &new_tets)
Split a tet in 4 tets.
Definition TetMeshTetSplittingConn.cpp:6
Tuple tuple_from_edge(size_t tid, int local_eid) const
get a Tuple from global tetra index and local edge index (from 0-5).
Definition TetMesh.cpp:353
size_t vertex_size() const
get the number of unremoved verticies
Definition TetMesh.h:367
Tuple switch_vertex(const Tuple &t) const
wrapper function from Tuple::switch_vertex
Definition TetMesh.h:921
virtual bool swap_face_before(const Tuple &t)
User specified preparations and desideratas for an 2-3 face swap befroe changing the geometry.
Definition TetMesh.h:805
bool try_set_edge_mutex_two_ring(const Tuple &e, int threadid=0)
try lock the two-ring neighboring triangles' incident vertices for the two ends of an edge
Definition TetMesh.cpp:1023
bool swap_edge(const Tuple &t, std::vector< Tuple > &new_tets)
3-2 edge swap
Definition TetMeshSwapMeshConnectivity.cpp:120
Tuple switch_edge(const Tuple &t) const
wrapper function from Tuple::switch_edge
Definition TetMesh.h:930
std::vector< Tuple > get_edges() const
Definition TetMesh.cpp:172
std::vector< size_t > get_one_ring_tids_for_vertex(const Tuple &t) const
Get the one ring tids for vertex.
Definition TetMesh.cpp:690
virtual bool split_tet_after(const Tuple &t)
Compute the attributes for the added simplices.
Definition TetMesh.h:860
void for_each_edge(const std::function< void(const TetMesh::Tuple &)> &)
perform the given function for each edge
Definition TetMesh.cpp:1294
bool collapse_edge_conn(const Tuple &loc0, std::vector< Tuple > &new_edges, size_t &v1_id, Tuple &new_loc, std::map< size_t, wmtk::TetMesh::VertexConnectivity > &rollback_vert_conn, std::vector< size_t > &n1_t_ids_copy, std::vector< size_t > &new_tet_id, std::vector< TetrahedronConnectivity > &old_tets)
Definition TetMeshEdgeCollapsingConn.cpp:449
void check_tuple_validity(const Tuple &t) const
Definition TetMesh.h:1062
void collapse_edge_rollback(size_t &v1_id, std::map< size_t, wmtk::TetMesh::VertexConnectivity > &rollback_vert_conn, std::vector< size_t > &n1_t_ids, std::vector< size_t > &new_tet_id, std::vector< TetrahedronConnectivity > &old_tets)
Definition TetMeshEdgeCollapsingConn.cpp:438
bool try_set_vertex_mutex_two_ring(const Tuple &v, int threadid)
try lock the two-ring neighboring traingles' incident vertices
Definition TetMesh.cpp:952
virtual bool collapse_edge(const Tuple &t, std::vector< Tuple > &new_tets)
Definition TetMeshEdgeCollapsingConn.cpp:263
bool try_set_face_mutex_two_ring(const Tuple &f, int threadid=0)
try lock the two-ring neighboring triangles' incident vertices for the 3 vertices of a face
Definition TetMesh.cpp:1069
Tuple tuple_from_face(size_t tid, int local_fid) const
get a Tuple from global tetra index and local face index (from 0-3).
Definition TetMesh.cpp:363
virtual bool split_edge_after(const Tuple &t)
This function computes the attributes for the added simplices. User specified modifications and desid...
Definition TetMesh.h:700
static constexpr std::array< std::array< int, 2 >, 6 > m_local_edges
local edges within a tet
Definition TetMesh.h:30
std::map< size_t, VertexConnectivity > operation_update_connectivity_impl(std::vector< size_t > &affected_tid, const std::vector< std::array< size_t, 4 > > &new_tet_conn)
Definition TetMeshSwapMeshConnectivity.cpp:43
std::array< Tuple, 3 > get_face_vertices(const Tuple &t) const
Get the 3 vertices of a face represented by Tuple.
Definition TetMesh.cpp:659
virtual bool swap_edge_56_after(const Tuple &t)
User specified modifications and desideratas for after a 5-6 edge swap.
Definition TetMesh.h:782
virtual bool collapse_edge_before(const Tuple &t)
User specified preparations and desideratas for an edge collapse before changing the connectivity.
Definition TetMesh.h:711
std::vector< Tuple > get_vertices() const
Definition TetMesh.cpp:323
size_t tet_capacity() const
get the current largest global tid
Definition TetMesh.h:361
std::vector< Tuple > get_one_ring_tets_for_vertex(const Tuple &t) const
Get the one ring tets for a vertex.
Definition TetMesh.cpp:701
void for_each_tetra(const std::function< void(const TetMesh::Tuple &)> &)
perform the given function for each tet
Definition TetMesh.cpp:1327
bool swap_edge_44(const Tuple &t, std::vector< Tuple > &new_tets)
Definition TetMeshSwapMeshConnectivity.cpp:472
virtual double swap_edge_56_energy(const std::vector< std::array< size_t, 4 > > &tets, const int op_case)
User specified energy to decide which of the 5 possible orientations should be chosen.
Definition TetMesh.h:770
bool swap_face(const Tuple &t, std::vector< Tuple > &new_tets)
2-3 face swap
Definition TetMeshSwapMeshConnectivity.cpp:231
void remove_tets_by_ids(const std::vector< size_t > &tids)
remove the tetrahedrons in the mesh that have given tet ids
Definition TetMesh.h:1075
Tuple tuple_from_vids(size_t vid0, size_t vid1, size_t vid2, size_t vid3) const
Get a Tuple from global vertex IDs.
Definition TetMesh.cpp:591
std::vector< std::array< size_t, 3 > > vertex_adjacent_boundary_faces(const Tuple &t) const
Definition TetMesh.cpp:915
virtual bool swap_face_after(const Tuple &t)
User specified modifications and desideratas for after a 2-3 face swap.
Definition TetMesh.h:812
Definition RawSimplex.hpp:57
Definition RawSimplex.hpp:71