1 files changed, 30 insertions, 1 deletions
diff --git a/ice40/arch.cc b/ice40/arch.cc
index 786d8ba1..ceafca17 100644
--- a/ice40/arch.cc
+++ b/ice40/arch.cc
@@ -255,11 +255,40 @@ BelId Arch::getBelByName(IdString name) const
     return ret;
 }
 
+BelId Arch::getBelByLocation(int x, int y, int z) const
+{
+    if (bel_by_loc.empty()) {
+        for (int i = 0; i < chip_info->num_bels; i++)
+            bel_by_loc[getBelLocation(BelId{i})] = i;
+    }
+
+    auto it = bel_by_loc.find(Loc{x, y, z});
+    if (it != bel_by_loc.end())
+        return BelId{it->second};
+    return BelId();
+}
+
+BelRange Arch::getBelsByTile(int x, int y) const
+{
+    // In iCE40 chipdb bels at the same tile are consecutive and dense z ordinates are used
+    BelRange br;
+    br.b.cursor = Arch::getBelByLocation(x, y, 0);
+    br.e.cursor = br.b.cursor;
+
+    if (br.e.cursor != -1) {
+        while (br.e.cursor < chip_info->num_bels &&
+               chip_info->bel_data[br.e.cursor].x == x &&
+               chip_info->bel_data[br.e.cursor].y == y)
+            br.e.cursor++;
+    }
+
+    return br;
+}
+
 BelRange Arch::getBelsAtSameTile(BelId bel) const
 {
     BelRange br;
     NPNR_ASSERT(bel != BelId());
-    // This requires Bels at the same tile are consecutive
     int x = chip_info->bel_data[bel.index].x;
     int y = chip_info->bel_data[bel.index].y;
     int start = bel.index, end = bel.index;