bjoernager/rust
bjoernager/rust
1
Fork 0
Code Issues Pull requests Activity

rustc_index: Add some map-like APIs to IndexVec

Browse source
This commit is contained in:
Vadim Petrochenkov 2021-09-22 01:59:46 +03:00
parent ac2d9fc509
commit fbe5e5c0ee
4 changed files with 28 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

15
compiler/rustc_index/src/vec.rs
View file

@ -720,6 +720,21 @@ impl<I: Idx, T> IndexVec<I, T> {
}
}
/// `IndexVec` is often used as a map, so it provides some map-like APIs.
impl<I: Idx, T> IndexVec<I, Option<T>> {
#[inline]
pub fn insert(&mut self, index: I, value: T) -> Option<T> {
self.ensure_contains_elem(index, || None);
self[index].replace(value)
}
#[inline]
pub fn get_or_insert_with(&mut self, index: I, value: impl FnOnce() -> T) -> &mut T {
self.ensure_contains_elem(index, || None);
self[index].get_or_insert_with(value)
}
}
impl<I: Idx, T: Clone> IndexVec<I, T> {
#[inline]
pub fn resize(&mut self, new_len: usize, value: T) {