refactor(plot): better shared face finding for HFB plotting

Author: wpbonelli

autotest/test_gridutil.py

@@ -8,6 +8,7 @@
     get_disu_kwargs,
     get_disv_kwargs,
     get_lni,
+    get_shared_face_3d,
     uniform_flow_field,
 )
 
@@ -186,3 +187,207 @@ def test_uniform_flow_field(qx, qy, qz, nlay, nrow, ncol):
 
     # TODO: check flowja
     # print(flowja.shape)
+
+
+def test_get_shared_face_3d_structured_horizontal():
+    """Test get_shared_face_3d for horizontal barrier in structured grid.
+
+    Tests same layer, adjacent cells.
+    """
+    from flopy.discretization import StructuredGrid
+
+    # Create a simple 2x2x1 structured grid
+    nlay, nrow, ncol = 1, 2, 2
+    delr = np.array([1.0, 1.0])
+    delc = np.array([1.0, 1.0])
+    top = np.array([[10.0, 10.0], [10.0, 10.0]])
+    botm = np.array([[[0.0, 0.0], [0.0, 0.0]]])
+
+    grid = StructuredGrid(delc=delc, delr=delr, top=top, botm=botm)
+
+    # Test horizontal barrier between cells (0, 0, 0) and (0, 0, 1)
+    # These are horizontally adjacent
+    cellid1 = (0, 0, 0)
+    cellid2 = (0, 0, 1)
+
+    face = get_shared_face_3d(grid, cellid1, cellid2)
+
+    assert face is not None
+    assert len(face) == 4  # Vertical face should have 4 vertices
+
+    # Check that face is vertical (two z-coordinates: top and bottom)
+    z_coords = sorted({v[2] for v in face})
+    assert len(z_coords) == 2
+    assert z_coords[0] == 0.0  # bottom
+    assert z_coords[1] == 10.0  # top
+
+    # Check x-coordinate is at the shared boundary (x=1.0)
+    x_coords = {v[0] for v in face}
+    assert 1.0 in x_coords
+
+
+def test_get_shared_face_3d_structured_vertical():
+    """Test get_shared_face_3d for vertical barrier in structured grid.
+
+    Tests different layers, same horizontal position.
+    """
+    from flopy.discretization import StructuredGrid
+
+    # Create a simple 2x2x2 structured grid (2 layers)
+    nlay, nrow, ncol = 2, 2, 2
+    delr = np.array([1.0, 1.0])
+    delc = np.array([1.0, 1.0])
+    top = np.array([[10.0, 10.0], [10.0, 10.0]])
+    botm = np.array([[[5.0, 5.0], [5.0, 5.0]], [[0.0, 0.0], [0.0, 0.0]]])
+
+    grid = StructuredGrid(delc=delc, delr=delr, top=top, botm=botm)
+
+    # Test vertical barrier between cells (0, 0, 0) and (1, 0, 0)
+    # These are vertically adjacent
+    cellid1 = (0, 0, 0)
+    cellid2 = (1, 0, 0)
+
+    face = get_shared_face_3d(grid, cellid1, cellid2)
+
+    assert face is not None
+    assert len(face) == 2  # Horizontal face should have 2 vertices (the edge)
+
+    # Check that face is horizontal (all z-coordinates the same at interface)
+    z_coords = {v[2] for v in face}
+    assert len(z_coords) == 1
+    assert next(iter(z_coords)) == 5.0  # Interface between layers
+
+
+def test_get_shared_face_3d_vertex():
+    """Test get_shared_face_3d for vertex grid."""
+    from flopy.discretization import VertexGrid
+
+    # Create a simple 2-cell vertex grid
+    nlay = 2
+    ncpl = 2
+
+    # Two square cells side by side
+    vertices = [
+        [0, 0.0, 0.0],
+        [1, 1.0, 0.0],
+        [2, 2.0, 0.0],
+        [3, 0.0, 1.0],
+        [4, 1.0, 1.0],
+        [5, 2.0, 1.0],
+    ]
+
+    cell2d = [
+        [0, 0.5, 0.5, 4, 0, 1, 4, 3],
+        [1, 1.5, 0.5, 4, 1, 2, 5, 4],
+    ]
+
+    top = np.array([10.0, 10.0])
+    botm = np.array([[5.0, 5.0], [0.0, 0.0]])
+
+    grid = VertexGrid(vertices=vertices, cell2d=cell2d, top=top, botm=botm, nlay=nlay)
+
+    # Test horizontal barrier between cells (0, 0) and (0, 1) in same layer
+    cellid1 = (0, 0)
+    cellid2 = (0, 1)
+
+    face = get_shared_face_3d(grid, cellid1, cellid2)
+
+    assert face is not None
+    assert len(face) == 4  # Vertical face
+
+    # Test vertical barrier between cells (0, 0) and (1, 0) in different layers
+    cellid1 = (0, 0)
+    cellid2 = (1, 0)
+
+    face = get_shared_face_3d(grid, cellid1, cellid2)
+
+    assert face is not None
+    assert len(face) == 2  # Horizontal face
+
+    # Check all z-coordinates are the same (interface elevation)
+    z_coords = {v[2] for v in face}
+    assert len(z_coords) == 1
+    assert next(iter(z_coords)) == 5.0
+
+
+def test_get_shared_face_3d_unstructured():
+    """Test get_shared_face_3d for unstructured grid with explicit 3D vertices."""
+    from flopy.discretization import UnstructuredGrid
+
+    # Create a simple 2-cell unstructured grid with 3D vertices
+    # Two hexahedral cells stacked vertically
+    vertices = [
+        # Bottom layer vertices (z=0)
+        [0, 0.0, 0.0, 0.0],
+        [1, 1.0, 0.0, 0.0],
+        [2, 1.0, 1.0, 0.0],
+        [3, 0.0, 1.0, 0.0],
+        # Middle layer vertices (z=5)
+        [4, 0.0, 0.0, 5.0],
+        [5, 1.0, 0.0, 5.0],
+        [6, 1.0, 1.0, 5.0],
+        [7, 0.0, 1.0, 5.0],
+        # Top layer vertices (z=10)
+        [8, 0.0, 0.0, 10.0],
+        [9, 1.0, 0.0, 10.0],
+        [10, 1.0, 1.0, 10.0],
+        [11, 0.0, 1.0, 10.0],
+    ]
+
+    iverts = [
+        [0, 1, 2, 3, 4, 5, 6, 7],  # Bottom cell
+        [4, 5, 6, 7, 8, 9, 10, 11],  # Top cell
+    ]
+
+    xc = np.array([0.5, 0.5])
+    yc = np.array([0.5, 0.5])
+    top = np.array([10.0, 10.0])
+    botm = np.array([0.0, 5.0])
+
+    grid = UnstructuredGrid(
+        vertices=vertices,
+        iverts=iverts,
+        xcenters=xc,
+        ycenters=yc,
+        top=top,
+        botm=botm,
+        ncpl=[2],
+    )
+
+    # Test shared face between two vertically stacked cells
+    cellid1 = (0,)
+    cellid2 = (1,)
+
+    face = get_shared_face_3d(grid, cellid1, cellid2)
+
+    assert face is not None
+    assert len(face) == 4  # Should have 4 shared vertices at the interface
+
+    # Check all shared vertices are at z=5.0 (the interface)
+    z_coords = {v[2] for v in face}
+    assert len(z_coords) == 1
+    assert next(iter(z_coords)) == 5.0
+
+
+def test_get_shared_face_3d_error_cases():
+    """Test error handling in get_shared_face_3d."""
+    from flopy.discretization import StructuredGrid
+
+    nlay, nrow, ncol = 1, 2, 2
+    delr = np.array([1.0, 1.0])
+    delc = np.array([1.0, 1.0])
+    top = np.array([[10.0, 10.0], [10.0, 10.0]])
+    botm = np.array([[[0.0, 0.0], [0.0, 0.0]]])
+
+    grid = StructuredGrid(delc=delc, delr=delr, top=top, botm=botm)
+
+    # Test with same cellid
+    with pytest.raises(ValueError, match="cellid1 and cellid2 must be different"):
+        get_shared_face_3d(grid, (0, 0, 0), (0, 0, 0))
+
+    # Test with non-adjacent cells (should return None)
+    cellid1 = (0, 0, 0)
+    cellid2 = (0, 1, 1)  # Diagonally opposite, not adjacent
+
+    face = get_shared_face_3d(grid, cellid1, cellid2)
+    assert face is None

flopy/plot/crosssection.py

@@ -13,7 +13,7 @@
 from . import plotutil
 from .plotutil import (
     get_shared_face_3d,
-    is_vertical_barrier,
+    is_vertical,
     to_mp7_endpoints,
     to_mp7_pathlines,
 )
@@ -781,22 +781,6 @@ def plot_grid(self, **kwargs):
             ax.add_collection(col)
         return col
 
-    def _cellid_to_node(self, cellid):
-        """
-        Convert a cellid tuple to a node number.
-
-        Parameters
-        ----------
-        cellid : tuple
-            Cell identifier
-
-        Returns
-        -------
-        int
-            Node number
-        """
-        return self.mg.get_node([cellid])[0]
-
     def _plot_vertical_hfb_lines(self, color=None, **kwargs):
         """
         Plot vertical HFBs as lines at layer interfaces.
@@ -848,8 +832,8 @@ def _plot_vertical_hfb_lines(self, color=None, **kwargs):
 
             # Get the x-coordinates along the cross section for this cell
             # Need to find this cell in projpts - convert both cellids to nodes
-            node1 = self._cellid_to_node(cellid1)
-            node2 = self._cellid_to_node(cellid2)
+            node1 = self.mg.get_node([cellid1])[0]
+            node2 = self.mg.get_node([cellid2])[0]
 
             # Get x-coordinates from either node's projection
             xcoords = []
@@ -958,11 +942,11 @@ def plot_bc(self, name=None, package=None, kper=0, color=None, head=None, **kwar
                             cellid1 = tuple(entry["cellid1"])
                             cellid2 = tuple(entry["cellid2"])
 
-                            if is_vertical_barrier(self.mg, cellid1, cellid2):
-                                # Store vertical HFBs for line plotting
+                            if not is_vertical(self.mg, cellid1, cellid2):
+                                # horizontal face, vertical barrier
                                 vertical_hfbs.append((cellid1, cellid2))
                             else:
-                                # Horizontal barriers - plot both cells as patches
+                                # vertical face, horizontal barrier
                                 cellids.append(list(entry["cellid1"]))
                                 cellids.append(list(entry["cellid2"]))
 

flopy/plot/map.py

@@ -13,7 +13,7 @@
 from .plotutil import (
     get_shared_face,
     get_shared_face_3d,
-    is_vertical_barrier,
+    is_vertical,
     to_mp7_pathlines,
 )
 
@@ -469,20 +469,18 @@ def _plot_barrier_bc(self, barrier_data, color=None, name=None, **kwargs):
         """
         ax = kwargs.pop("ax", self.ax)
 
-        # Separate horizontal and vertical barriers
         horizontal_line_segments = []
         vertical_cell_indices = []
 
         for cellid1, cellid2 in barrier_data:
             # Only plot barriers on the current layer (for layered grids)
             # For DISU (len==1), plot all barriers since there's no layer filtering
-            if len(cellid1) >= 2:  # DIS or DISV (has layer info)
+            if len(cellid1) >= 2:
                 if cellid1[0] != self.layer and cellid2[0] != self.layer:
                     continue
 
-            # Check if this is a vertical barrier
-            if is_vertical_barrier(self.mg, cellid1, cellid2):
-                # For vertical barriers, plot the cells on the current layer
+            # horizontal face, vertical barrier
+            if not is_vertical(self.mg, cellid1, cellid2):
                 if len(cellid1) >= 2:
                     if cellid1[0] == self.layer:
                         if len(cellid1) == 3:
@@ -499,7 +497,7 @@ def _plot_barrier_bc(self, barrier_data, color=None, name=None, **kwargs):
                         else:
                             vertical_cell_indices.append([self.layer, cellid2[1]])
             else:
-                # Horizontal barrier - plot as line on shared face
+                # vertical face, horizontal barrier
                 shared_face = get_shared_face(self.mg, cellid1, cellid2)
                 if shared_face is not None:
                     horizontal_line_segments.append(shared_face)