mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

spans are now indexmapped

Browse Source
This commit is contained in:
Oğuz Ağcayazı 2023-10-09 13:03:58 +03:00
parent 5f079dd2ff
commit 77df2cd9a5
2 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
compiler/rustc_smir/src/rustc_internal/mod.rs
View File

@ -30,7 +30,7 @@ impl<'tcx> Index<stable_mir::ty::Span> for Tables<'tcx> {
#[inline(always)]
fn index(&self, index: stable_mir::ty::Span) -> &Self::Output {
&self.spans[index.0]
&self.spans.get_index(index.0).unwrap().0
}
}
@ -106,7 +106,6 @@ impl<'tcx> Tables<'tcx> {
}
fn create_alloc_id(&mut self, aid: AllocId) -> stable_mir::AllocId {
// FIXME: this becomes inefficient when we have too many ids
if let Some(i) = self.alloc_ids.get(&aid) {
return *i;
} else {
@ -117,14 +116,13 @@ impl<'tcx> Tables<'tcx> {
}
pub(crate) fn create_span(&mut self, span: Span) -> stable_mir::ty::Span {
for (i, &sp) in self.spans.iter().enumerate() {
if sp == span {
return stable_mir::ty::Span(i);
}
if let Some(i) = self.spans.get(&span) {
return *i;
} else {
let id = self.spans.len();
self.spans.insert(span, stable_mir::ty::Span(id));
stable_mir::ty::Span(id)
}
let id = self.spans.len();
self.spans.push(span);
stable_mir::ty::Span(id)
}
}
@ -138,7 +136,7 @@ pub fn run(tcx: TyCtxt<'_>, f: impl FnOnce()) {
tcx,
def_ids: fx::FxIndexMap::default(),
alloc_ids: fx::FxIndexMap::default(),
spans: vec![],
spans: fx::FxIndexMap::default(),
types: vec![],
},
f,

2
compiler/rustc_smir/src/rustc_smir/mod.rs
View File

@ -197,7 +197,7 @@ pub struct Tables<'tcx> {
pub tcx: TyCtxt<'tcx>,
pub def_ids: FxIndexMap<DefId, stable_mir::DefId>,
pub alloc_ids: FxIndexMap<AllocId, stable_mir::AllocId>,
pub spans: Vec<rustc_span::Span>,
pub spans: FxIndexMap<rustc_span::Span, Span>,
pub types: Vec<MaybeStable<stable_mir::ty::TyKind, Ty<'tcx>>>,
}