diff options
Diffstat (limited to 'gcc')
| -rw-r--r-- | gcc/rust/util/rust-hir-map.cc | 19 | ||||
| -rw-r--r-- | gcc/rust/util/rust-hir-map.h | 3 |
2 files changed, 22 insertions, 0 deletions
diff --git a/gcc/rust/util/rust-hir-map.cc b/gcc/rust/util/rust-hir-map.cc index 4b71312..786fc5a 100644 --- a/gcc/rust/util/rust-hir-map.cc +++ b/gcc/rust/util/rust-hir-map.cc @@ -719,5 +719,24 @@ Mappings::iterate_impl_blocks (std::function<bool (HirId, HIR::ImplBlock *)> cb) } } +void +Mappings::iterate_trait_items ( + std::function<bool (HIR::TraitItem *, HIR::Trait *)> cb) +{ + for (auto it = hirTraitItemMappings.begin (); + it != hirTraitItemMappings.end (); it++) + { + for (auto iy = it->second.begin (); iy != it->second.end (); iy++) + { + HirId trait_item_id = iy->first; + HIR::TraitItem *trait_item = iy->second; + HIR::Trait *trait = lookup_trait_item_mapping (trait_item_id); + + if (!cb (trait_item, trait)) + return; + } + } +} + } // namespace Analysis } // namespace Rust diff --git a/gcc/rust/util/rust-hir-map.h b/gcc/rust/util/rust-hir-map.h index 3ed3d47..ec08170 100644 --- a/gcc/rust/util/rust-hir-map.h +++ b/gcc/rust/util/rust-hir-map.h @@ -474,6 +474,9 @@ public: void iterate_impl_blocks (std::function<bool (HirId, HIR::ImplBlock *)> cb); + void iterate_trait_items ( + std::function<bool (HIR::TraitItem *item, HIR::Trait *)> cb); + bool is_impl_item (HirId id) { HirId parent_impl_block_id = UNKNOWN_HIRID; |