diff mbox series

[113/125] gccrs: borrowck: BIR: Place tree traverse API

Message ID 20240801145809.366388-115-arthur.cohen@embecosm.com
State New
Headers show
Series [001/125] Rust: Make 'tree'-level 'MAIN_NAME_P' work | expand

Commit Message

Arthur Cohen Aug. 1, 2024, 2:57 p.m. UTC 
From: Jakub Dupak <dev@jakubdupak.com>

gcc/rust/ChangeLog:

	* checks/errors/borrowck/rust-bir-place.h:
	Create place tree traverse API.

Signed-off-by: Jakub Dupak <dev@jakubdupak.com>
---
 .../checks/errors/borrowck/rust-bir-place.h   | 23 +++++++++++++++++++
 1 file changed, 23 insertions(+)
diff mbox series

Patch

diff --git a/gcc/rust/checks/errors/borrowck/rust-bir-place.h b/gcc/rust/checks/errors/borrowck/rust-bir-place.h
index ccc820121cc..e62ec3557ad 100644
--- a/gcc/rust/checks/errors/borrowck/rust-bir-place.h
+++ b/gcc/rust/checks/errors/borrowck/rust-bir-place.h
@@ -229,6 +229,29 @@  public:
 		       places[place].tyty});
   }
 
+  template <typename FN> void for_each_path_from_root (PlaceId var, FN fn) const
+  {
+    PlaceId current = var;
+    current = places[current].path.first_child;
+    while (current != INVALID_PLACE)
+      {
+	fn (current);
+	for_each_path_from_root (current, fn);
+	current = places[current].path.next_sibling;
+      }
+  }
+
+  template <typename FN>
+  void for_each_path_segment (PlaceId place_id, FN fn) const
+  {
+    PlaceId current = place_id;
+    while (current != INVALID_PLACE)
+      {
+	fn (current);
+	current = places[current].path.parent;
+      }
+  }
+
 private:
   static bool is_type_copy (TyTy::BaseType *ty)
   {