Get Bounding box of arch.Layout. (#79)

* Adding method to get total bounds of the layout

* adding error when no positions are found

* adding tests

* changing condition of failure

* adding failure check

* Apply suggestion from @Copilot

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* run linter

---------

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: weinbe58

src/bloqade/shuttle/arch.py

@@ -1,4 +1,5 @@
 from dataclasses import dataclass, field
+from itertools import chain
 from typing import cast
 
 import numpy as np
@@ -41,6 +42,31 @@ def __eq__(self, other):
             self.static_traps == other.static_traps and self.fillable == other.fillable
         )
 
+    def bounding_box(self) -> tuple[float, float, float, float]:
+        """Get the bounding box (xmin, xmax, ymin, ymax) of the layout."""
+        xmin = float("inf")
+        xmax = float("-inf")
+        ymin = float("inf")
+        ymax = float("-inf")
+
+        for zone in chain(self.static_traps.values(), self.special_grid.values()):
+            if zone.x_init is not None:
+                xmin = min(xmin, zone.x_init)
+                xmax = max(xmax, zone.x_init + zone.width)
+            if zone.y_init is not None:
+                ymin = min(ymin, zone.y_init)
+                ymax = max(ymax, zone.y_init + zone.height)
+
+        if (
+            xmin == float("inf")
+            or xmax == float("-inf")
+            or ymin == float("inf")
+            or ymax == float("-inf")
+        ):
+            raise ValueError("Layout has incomplete bounding box data.")
+
+        return xmin, xmax, ymin, ymax
+
     @staticmethod
     def _plot_zone(zone: Grid, ax, name: str, **plot_options):
         from matplotlib.axes import Axes  # type: ignore

test/unit/test_arch.py

@@ -1,3 +1,4 @@
+import pytest
 from bloqade.geometry.dialects.grid import Grid
 
 from bloqade.shuttle.arch import Layout
@@ -28,3 +29,16 @@ def test_layout():
         {"test"},
     )
     assert layout != 1
+    assert layout.bounding_box() == (0.0, 15.0, 0.0, 15.0)
+
+
+def test_bounding_box_no_positions():
+    layout = Layout(
+        {"test": Grid.from_positions([], [])},
+        {"test"},
+        {"test"},
+        {"test"},
+    )
+
+    with pytest.raises(ValueError):
+        layout.bounding_box()