add deleteat!

trixi-framework · Jul 30, 2024 · bee9a26 · bee9a26
1 parent 20b724f
commit bee9a26
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/preprocessing/geometries/polygon.jl b/src/preprocessing/geometries/polygon.jl
@@ -101,6 +101,16 @@ end
 
 @inline Base.eltype(::Polygon{NDIMS, ELTYPE}) where {NDIMS, ELTYPE} = ELTYPE
 
+@inline function Base.deleteat!(polygon::Polygon, indices)
+    (; edge_vertices, edge_normals, edge_vertices_ids) = polygon
+
+    deleteat!(edge_vertices, indices)
+    deleteat!(edge_vertices_ids, indices)
+    deleteat!(edge_normals, indices)
+
+    return polygon
+end
+
 @inline nfaces(mesh::Polygon) = length(mesh.edge_normals)
 
 @inline function face_vertices(edge, geometry::Polygon)

diff --git a/src/preprocessing/geometries/triangle_mesh.jl b/src/preprocessing/geometries/triangle_mesh.jl
@@ -140,6 +140,17 @@ end
 
 @inline Base.eltype(::TriangleMesh{NDIMS, ELTYPE}) where {NDIMS, ELTYPE} = ELTYPE
 
+@inline function Base.deleteat!(mesh::TriangleMesh, indices)
+    (; face_vertices, face_vertices_ids, face_edges_ids, face_normals) = mesh
+
+    deleteat!(face_vertices, indices)
+    deleteat!(face_vertices_ids, indices)
+    deleteat!(face_edges_ids, indices)
+    deleteat!(face_normals, indices)
+
+    return mesh
+end
+
 @inline nfaces(mesh::TriangleMesh) = length(mesh.face_normals)
 
 @inline function face_vertices(triangle, geometry::TriangleMesh)