1 files changed, 27 insertions, 0 deletions
diff --git a/gowin/arch.h b/gowin/arch.h
index 5591744d..f12c604e 100644
--- a/gowin/arch.h
+++ b/gowin/arch.h
@@ -422,6 +422,31 @@ struct Arch : BaseCtx
     // Get the TimingClockingInfo of a port
     TimingClockingInfo getPortClockingInfo(const CellInfo *cell, IdString port, int index) const;
 
+    bool isValidBelForCellType(IdString cell_type, BelId bel) const { return cell_type == getBelType(bel); }
+
+    const std::vector<IdString> &getCellTypes() const { return cell_types; }
+
+    std::vector<BelBucketId> getBelBuckets() const { return cell_types; }
+
+    IdString getBelBucketName(BelBucketId bucket) const { return bucket; }
+
+    BelBucketId getBelBucketByName(IdString name) const { return name; }
+
+    BelBucketId getBelBucketForBel(BelId bel) const { return getBelType(bel); }
+
+    BelBucketId getBelBucketForCellType(IdString cell_type) const { return cell_type; }
+
+    std::vector<BelId> getBelsInBucket(BelBucketId bucket) const
+    {
+        std::vector<BelId> bels;
+        for (BelId bel : getBels()) {
+            if (getBelType(bel) == bucket) {
+                bels.push_back(bel);
+            }
+        }
+        return bels;
+    }
+
     bool isValidBelForCell(CellInfo *cell, BelId bel) const;
     bool isBelLocationValid(BelId bel) const;
 
@@ -434,6 +459,8 @@ struct Arch : BaseCtx
     // Internal usage
     void assignArchInfo();
     bool cellsCompatible(const CellInfo **cells, int count) const;
+
+    std::vector<IdString> cell_types;
 };
 
 NEXTPNR_NAMESPACE_END