From a033c07a27740fdfc7cb6454eb05b08d8c0157ea Mon Sep 17 00:00:00 2001
From: Anatole Storck <storckanatole@gmail.com>
Date: Fri, 13 Feb 2026 11:32:30 +0100
Subject: [PATCH 1/2] add bbox rejection for finding cpu intersections with a
 YTRegion

---
 yt/frontends/ramses/hilbert.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/yt/frontends/ramses/hilbert.py b/yt/frontends/ramses/hilbert.py
index 5f203c8f798..d7ca15e3249 100644
--- a/yt/frontends/ramses/hilbert.py
+++ b/yt/frontends/ramses/hilbert.py
@@ -75,14 +75,16 @@ def get_intersecting_cpus(
     dx_cond: float | None = None,
     factor: float = 4.0,
     bound_keys: Optional["np.ndarray[Any, np.dtype[np.float64]]"] = None,
+    bbox: Optional[Any] = None,
 ) -> set[int]:
     """
     Find the subset of CPUs that intersect the bbox in a recursive fashion.
     """
+    if bbox is None:
+        bbox = region.get_bbox()
     if LE is None:
         LE = np.array([0, 0, 0], dtype="d")
     if dx_cond is None:
-        bbox = region.get_bbox()
         dx_cond = float((bbox[1] - bbox[0]).min().to("code_length"))
     if bound_keys is None:
         ncpu = ds.parameters["ncpu"]
@@ -90,6 +92,14 @@ def get_intersecting_cpus(
         bound_keys[:ncpu] = [ds.hilbert_indices[icpu + 1][0] for icpu in range(ncpu)]
         bound_keys[ncpu] = ds.hilbert_indices[ncpu][1]
 
+    # bbox rejection to avoid expensive selector checks. (greatly improves performance for small disks)
+    bbox_min = bbox[0].to("code_length").d
+    bbox_max = bbox[1].to("code_length").d
+    cube_min = LE
+    cube_max = LE + dx
+    if np.any(cube_max < bbox_min) or np.any(cube_min > bbox_max):
+        return set()
+
     # If the current dx is smaller than the smallest size of the bbox
     if dx < dx_cond / factor:
         # Finish recursion
@@ -111,7 +121,7 @@ def get_intersecting_cpus(
                 if bbox_intersects(region.selector, LE_new, dx):
                     ret.update(
                         get_intersecting_cpus(
-                            ds, region, LE_new, dx, dx_cond, factor, bound_keys
+                            ds, region, LE_new, dx, dx_cond, factor, bound_keys, bbox
                         )
                     )
     return ret

From e1524be10cafec8e8d86240ee52db734b747fed6 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 25 Feb 2026 10:05:23 +0000
Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 yt/frontends/ramses/hilbert.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/yt/frontends/ramses/hilbert.py b/yt/frontends/ramses/hilbert.py
index d7ca15e3249..70e697abfab 100644
--- a/yt/frontends/ramses/hilbert.py
+++ b/yt/frontends/ramses/hilbert.py
@@ -75,7 +75,7 @@ def get_intersecting_cpus(
     dx_cond: float | None = None,
     factor: float = 4.0,
     bound_keys: Optional["np.ndarray[Any, np.dtype[np.float64]]"] = None,
-    bbox: Optional[Any] = None,
+    bbox: Any | None = None,
 ) -> set[int]:
     """
     Find the subset of CPUs that intersect the bbox in a recursive fashion.