Fix expansion of facets to edges for interior facets (#1132)

kinnala · Jun 13, 2024 · a8d5b09 · a8d5b09
1 parent 02a80c3
commit a8d5b09
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 11 deletions.
diff --git a/skfem/element/__init__.py b/skfem/element/__init__.py
@@ -54,7 +54,8 @@
     ElementTetP2: ElementTriP2,
     ElementHex0: ElementQuad0,
     ElementHex1: ElementQuad1,
-    ElementHex2: ElementQuad2
+    ElementHex2: ElementQuad2,
+    ElementHex1DG: ElementQuad1,
 }
 
 # for backwards compatibility

diff --git a/skfem/mesh/mesh.py b/skfem/mesh/mesh.py
@@ -458,16 +458,8 @@ def _expand_facets(self, ix: ndarray) -> Tuple[ndarray, ndarray]:
         """
         vertices = np.unique(self.facets[:, ix].flatten())
 
-        if self.dim() == 3:
-            edge_candidates = self.t2e[:, self.f2t[0, ix]].flatten()
-            # subset of edges that share all points with the given facets
-            subset = np.nonzero(
-                np.prod(np.isin(self.edges[:, edge_candidates],
-                                self.facets[:, ix].flatten()),
-                        axis=0)
-            )[0]
-            edges = np.intersect1d(self.boundary_edges(),
-                                   edge_candidates[subset])
+        if self.dim() == 3 and self.bndelem is not None:
+            edges = np.unique(self.f2e[:, ix])
         else:
             edges = np.array([], dtype=np.int64)