diff options
| author | bors <bors@rust-lang.org> | 2024-05-09 04:04:09 +0000 |
|---|---|---|
| committer | bors <bors@rust-lang.org> | 2024-05-09 04:04:09 +0000 |
| commit | 37dc766378895b6e0e0c8d87a3f1e567f210a744 (patch) | |
| tree | c81795ea112432f6a6c5339ff43512f0da777886 /compiler/rustc_borrowck/src | |
| parent | ee9a9f84c599492eb9613292394d6d05d9f9e6ba (diff) | |
| parent | 58a06b6a9986149937d7efa97eba83df207a1c7c (diff) | |
| download | rust-37dc766378895b6e0e0c8d87a3f1e567f210a744.tar.gz rust-37dc766378895b6e0e0c8d87a3f1e567f210a744.zip | |
Auto merge of #124831 - nnethercote:rustc_data_structures-cleanups, r=michaelwoerister
`rustc_data_structures` cleanups Some improvements I found while looking through this code. r? `@michaelwoerister`
Diffstat (limited to 'compiler/rustc_borrowck/src')
| -rw-r--r-- | compiler/rustc_borrowck/src/type_check/liveness/local_use_map.rs | 42 |
1 files changed, 32 insertions, 10 deletions
diff --git a/compiler/rustc_borrowck/src/type_check/liveness/local_use_map.rs b/compiler/rustc_borrowck/src/type_check/liveness/local_use_map.rs index da5456692ab..ccd9fb25739 100644 --- a/compiler/rustc_borrowck/src/type_check/liveness/local_use_map.rs +++ b/compiler/rustc_borrowck/src/type_check/liveness/local_use_map.rs @@ -1,4 +1,3 @@ -use rustc_data_structures::vec_linked_list as vll; use rustc_index::IndexVec; use rustc_middle::mir::visit::{PlaceContext, Visitor}; use rustc_middle::mir::{Body, Local, Location}; @@ -37,9 +36,12 @@ pub(crate) struct LocalUseMap { /// we add for each local variable. first_drop_at: IndexVec<Local, Option<AppearanceIndex>>, - appearances: IndexVec<AppearanceIndex, Appearance>, + appearances: Appearances, } +// The `Appearance::next` field effectively embeds a linked list within `Appearances`. +type Appearances = IndexVec<AppearanceIndex, Appearance>; + struct Appearance { point_index: PointIndex, next: Option<AppearanceIndex>, @@ -49,14 +51,34 @@ rustc_index::newtype_index! { pub struct AppearanceIndex {} } -impl vll::LinkElem for Appearance { - type LinkIndex = AppearanceIndex; +fn appearances_iter( + first: Option<AppearanceIndex>, + appearances: &Appearances, +) -> impl Iterator<Item = AppearanceIndex> + '_ { + AppearancesIter { appearances, current: first } +} + +// Iterates over `Appearances` by following `next` fields. +struct AppearancesIter<'a> { + appearances: &'a Appearances, + current: Option<AppearanceIndex>, +} - fn next(elem: &Self) -> Option<AppearanceIndex> { - elem.next +impl<'a> Iterator for AppearancesIter<'a> { + type Item = AppearanceIndex; + + fn next(&mut self) -> Option<AppearanceIndex> { + if let Some(c) = self.current { + self.current = self.appearances[c].next; + Some(c) + } else { + None + } } } +//----------------------------------------------------------------------------- + impl LocalUseMap { pub(crate) fn build( live_locals: &[Local], @@ -86,17 +108,17 @@ impl LocalUseMap { } pub(crate) fn defs(&self, local: Local) -> impl Iterator<Item = PointIndex> + '_ { - vll::iter(self.first_def_at[local], &self.appearances) + appearances_iter(self.first_def_at[local], &self.appearances) .map(move |aa| self.appearances[aa].point_index) } pub(crate) fn uses(&self, local: Local) -> impl Iterator<Item = PointIndex> + '_ { - vll::iter(self.first_use_at[local], &self.appearances) + appearances_iter(self.first_use_at[local], &self.appearances) .map(move |aa| self.appearances[aa].point_index) } pub(crate) fn drops(&self, local: Local) -> impl Iterator<Item = PointIndex> + '_ { - vll::iter(self.first_drop_at[local], &self.appearances) + appearances_iter(self.first_drop_at[local], &self.appearances) .map(move |aa| self.appearances[aa].point_index) } } @@ -146,7 +168,7 @@ impl LocalUseMapBuild<'_> { fn insert( elements: &DenseLocationMap, first_appearance: &mut Option<AppearanceIndex>, - appearances: &mut IndexVec<AppearanceIndex, Appearance>, + appearances: &mut Appearances, location: Location, ) { let point_index = elements.point_from_location(location); |
