Add bounds calculation for geometric sets.

Author: pshriwise

src/pydagmc/dagnav.py

@@ -407,14 +407,14 @@ def category(self, category: str):
     def groups(self) -> list[Group]:
         """Get list of groups containing this DAGMC set."""
         return [group for group in self.model.groups if self in group]
-    
+
     def _metadata_group(self, prefix: str) -> list[Group]:
         """Get single group containing this DAGMC set, with matching prefix."""
         for group in self.model.groups:
             if self in group and prefix in group.name:
-                return group            
+                return group
         return None
-    
+
     def _metadata_group_name(self, prefix: str) -> Optional[str]:
         group = self._metadata_group(prefix)
         if group is not None:
@@ -479,6 +479,20 @@ def triangle_coords(self):
 
         return self.model.mb.get_coords(conn.flatten()).reshape(-1, 3)
 
+    @property
+    def bounds(self):
+        """Returns the axis-aligned bounding box for all triangles under this set.
+
+        Returns
+        -------
+        tuple(float, float, float, float, float, float)
+            (xmin, xmax, ymin, ymax, zmin, zmax)
+        """
+        coords = self.triangle_coords
+        min_coord = np.min(coords, axis=0)
+        max_coord = np.max(coords, axis=0)
+        return min_coord, max_coord
+
     def get_triangle_conn_and_coords(self, compress=False):
         """Returns the triangle connectivity and coordinates for all triangles under this set.
 
@@ -623,7 +637,7 @@ def boundary_group(self) -> Optional[Group]:
     def boundary(self) -> Optional[str]:
         """Name of the boundary assigned to this surface."""
         return self._metadata_group_name(self._boundary_prefix)
-        
+
     @boundary.setter
     def boundary(self, name: Optional[str]):
         self._set_metadata_group(self._boundary_prefix, name)

test/test_dagnav.py

@@ -858,6 +858,24 @@ def test_area(fuel_pin_model):
         pytest.approx(model.surfaces[surf_id].area, exp_area)
 
 
+def test_bounds(fuel_pin_model):
+    model = fuel_pin_model
+    vol1_bounds = model.volumes_by_id[1].bounds
+    expected_bounds = (-7.0, 7.0, -7.0, 7.0, -20.0, 20.0)
+    for val, exp_val in zip(vol1_bounds, expected_bounds):
+        pytest.approx(val, exp_val)
+
+    exp_surface_bounds = {1: (-7.0, 7.0, -7.0, 7.0, -20.0, 20.0),
+                          2: (-9.0, 9.0, -9.0, 9.0, -20.0, 20.0),
+                          3: (-10.0, 10.0, -10.0, 10.0, -20.0, 20.0),}
+
+    # test all surface bounds
+    for surf_id, expected_bounds in exp_surface_bounds.items():
+        surface_bounds = model.surfaces_by_id[surf_id].bounds
+        for val, exp_val in zip(surface_bounds, expected_bounds):
+            pytest.approx(val, exp_val)
+
+
 def test_add_groups(fuel_pin_model):
     model = fuel_pin_model
     volumes = model.volumes_by_id