ops::{Index, IndexMut},
};
+pub mod map;
+
#[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct RawId(u32);
--- /dev/null
+//! A map from arena IDs to some other type. Space requirement is O(highest ID).
+
+use std::marker::PhantomData;
+
+use super::ArenaId;
+
+/// A map from arena IDs to some other type. Space requirement is O(highest ID).
+#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
+pub struct ArenaMap<ID, T> {
+ v: Vec<Option<T>>,
+ _ty: PhantomData<ID>,
+}
+
+impl<ID: ArenaId, T> ArenaMap<ID, T> {
+ pub fn insert(&mut self, id: ID, t: T) {
+ let idx = Self::to_idx(id);
+ if self.v.capacity() <= idx {
+ self.v.reserve(idx + 1 - self.v.capacity());
+ }
+ if self.v.len() <= idx {
+ while self.v.len() <= idx {
+ self.v.push(None);
+ }
+ }
+ self.v[idx] = Some(t);
+ }
+
+ pub fn get(&self, id: ID) -> Option<&T> {
+ self.v.get(Self::to_idx(id)).and_then(|it| it.as_ref())
+ }
+
+ pub fn values(&self) -> impl Iterator<Item = &T> {
+ self.v.iter().filter_map(|o| o.as_ref())
+ }
+
+ pub fn values_mut(&mut self) -> impl Iterator<Item = &mut T> {
+ self.v.iter_mut().filter_map(|o| o.as_mut())
+ }
+
+ pub fn iter(&self) -> impl Iterator<Item = (ID, &T)> {
+ self.v
+ .iter()
+ .enumerate()
+ .filter_map(|(idx, o)| Some((Self::from_idx(idx), o.as_ref()?)))
+ }
+
+ fn to_idx(id: ID) -> usize {
+ u32::from(id.into_raw()) as usize
+ }
+
+ fn from_idx(idx: usize) -> ID {
+ ID::from_raw((idx as u32).into())
+ }
+}
+
+impl<ID: ArenaId, T> std::ops::Index<ID> for ArenaMap<ID, T> {
+ type Output = T;
+ fn index(&self, id: ID) -> &T {
+ self.v[Self::to_idx(id)].as_ref().unwrap()
+ }
+}
+
+impl<ID, T> Default for ArenaMap<ID, T> {
+ fn default() -> Self {
+ ArenaMap {
+ v: Vec::new(),
+ _ty: PhantomData,
+ }
+ }
+}
use rustc_hash::FxHashMap;
-use ra_arena::{Arena, RawId, impl_arena_id};
+use ra_arena::{Arena, RawId, impl_arena_id, map::ArenaMap};
use ra_db::{LocalSyntaxPtr, Cancelable};
use ra_syntax::{SyntaxNodeRef, ast::{self, AstNode, LoopBodyOwner, ArgListOwner, NameOwner}};
pub struct BodySyntaxMapping {
body: Arc<Body>,
expr_syntax_mapping: FxHashMap<LocalSyntaxPtr, ExprId>,
- expr_syntax_mapping_back: FxHashMap<ExprId, LocalSyntaxPtr>,
+ expr_syntax_mapping_back: ArenaMap<ExprId, LocalSyntaxPtr>,
pat_syntax_mapping: FxHashMap<LocalSyntaxPtr, PatId>,
- pat_syntax_mapping_back: FxHashMap<PatId, LocalSyntaxPtr>,
+ pat_syntax_mapping_back: ArenaMap<PatId, LocalSyntaxPtr>,
}
impl Body {
impl BodySyntaxMapping {
pub fn expr_syntax(&self, expr: ExprId) -> Option<LocalSyntaxPtr> {
- self.expr_syntax_mapping_back.get(&expr).cloned()
+ self.expr_syntax_mapping_back.get(expr).cloned()
}
pub fn syntax_expr(&self, ptr: LocalSyntaxPtr) -> Option<ExprId> {
self.expr_syntax_mapping.get(&ptr).cloned()
.cloned()
}
pub fn pat_syntax(&self, pat: PatId) -> Option<LocalSyntaxPtr> {
- self.pat_syntax_mapping_back.get(&pat).cloned()
+ self.pat_syntax_mapping_back.get(pat).cloned()
}
pub fn syntax_pat(&self, ptr: LocalSyntaxPtr) -> Option<PatId> {
self.pat_syntax_mapping.get(&ptr).cloned()
exprs: Arena<ExprId, Expr>,
pats: Arena<PatId, Pat>,
expr_syntax_mapping: FxHashMap<LocalSyntaxPtr, ExprId>,
- expr_syntax_mapping_back: FxHashMap<ExprId, LocalSyntaxPtr>,
+ expr_syntax_mapping_back: ArenaMap<ExprId, LocalSyntaxPtr>,
pat_syntax_mapping: FxHashMap<LocalSyntaxPtr, PatId>,
- pat_syntax_mapping_back: FxHashMap<PatId, LocalSyntaxPtr>,
+ pat_syntax_mapping_back: ArenaMap<PatId, LocalSyntaxPtr>,
}
impl ExprCollector {
exprs: Arena::default(),
pats: Arena::default(),
expr_syntax_mapping: FxHashMap::default(),
- expr_syntax_mapping_back: FxHashMap::default(),
+ expr_syntax_mapping_back: ArenaMap::default(),
pat_syntax_mapping: FxHashMap::default(),
- pat_syntax_mapping_back: FxHashMap::default(),
+ pat_syntax_mapping_back: ArenaMap::default(),
}
}
use std::{fmt, mem};
use log;
-use rustc_hash::FxHashMap;
use ena::unify::{InPlaceUnificationTable, UnifyKey, UnifyValue, NoError};
+use ra_arena::map::ArenaMap;
use ra_db::Cancelable;
/// The result of type inference: A mapping from expressions and patterns to types.
#[derive(Clone, PartialEq, Eq, Debug)]
pub struct InferenceResult {
- type_of_expr: FxHashMap<ExprId, Ty>,
- type_of_pat: FxHashMap<PatId, Ty>,
+ type_of_expr: ArenaMap<ExprId, Ty>,
+ type_of_pat: ArenaMap<PatId, Ty>,
}
impl Index<ExprId> for InferenceResult {
type Output = Ty;
fn index(&self, expr: ExprId) -> &Ty {
- self.type_of_expr.get(&expr).unwrap_or(&Ty::Unknown)
+ self.type_of_expr.get(expr).unwrap_or(&Ty::Unknown)
}
}
type Output = Ty;
fn index(&self, pat: PatId) -> &Ty {
- self.type_of_pat.get(&pat).unwrap_or(&Ty::Unknown)
+ self.type_of_pat.get(pat).unwrap_or(&Ty::Unknown)
}
}
module: Module,
impl_block: Option<ImplBlock>,
var_unification_table: InPlaceUnificationTable<TypeVarId>,
- type_of_expr: FxHashMap<ExprId, Ty>,
- type_of_pat: FxHashMap<PatId, Ty>,
+ type_of_expr: ArenaMap<ExprId, Ty>,
+ type_of_pat: ArenaMap<PatId, Ty>,
/// The return type of the function being inferred.
return_ty: Ty,
}
impl_block: Option<ImplBlock>,
) -> Self {
InferenceContext {
- type_of_expr: FxHashMap::default(),
- type_of_pat: FxHashMap::default(),
+ type_of_expr: ArenaMap::default(),
+ type_of_pat: ArenaMap::default(),
var_unification_table: InPlaceUnificationTable::new(),
return_ty: Ty::Unknown, // set in collect_fn_signature
db,
}
fn resolve_all(mut self) -> InferenceResult {
- let mut expr_types = mem::replace(&mut self.type_of_expr, FxHashMap::default());
+ let mut expr_types = mem::replace(&mut self.type_of_expr, ArenaMap::default());
for ty in expr_types.values_mut() {
let resolved = self.resolve_ty_completely(mem::replace(ty, Ty::Unknown));
*ty = resolved;
}
- let mut pat_types = mem::replace(&mut self.type_of_pat, FxHashMap::default());
+ let mut pat_types = mem::replace(&mut self.type_of_pat, ArenaMap::default());
for ty in pat_types.values_mut() {
let resolved = self.resolve_ty_completely(mem::replace(ty, Ty::Unknown));
*ty = resolved;
// resolve locally
let name = path.as_ident().cloned().unwrap_or_else(Name::self_param);
if let Some(scope_entry) = self.scopes.resolve_local_name(expr, name) {
- let ty = ctry!(self.type_of_pat.get(&scope_entry.pat()));
+ let ty = ctry!(self.type_of_pat.get(scope_entry.pat()));
let ty = self.resolve_ty_as_possible(ty.clone());
return Ok(Some(ty));
};
let inference_result = func.infer(&db).unwrap();
let body_syntax_mapping = func.body_syntax_mapping(&db).unwrap();
let mut types = Vec::new();
- for (pat, ty) in &inference_result.type_of_pat {
- let syntax_ptr = if let Some(sp) = body_syntax_mapping.pat_syntax(*pat) {
+ for (pat, ty) in inference_result.type_of_pat.iter() {
+ let syntax_ptr = if let Some(sp) = body_syntax_mapping.pat_syntax(pat) {
sp
} else {
continue;
};
types.push((syntax_ptr, ty));
}
- for (expr, ty) in &inference_result.type_of_expr {
- let syntax_ptr = if let Some(sp) = body_syntax_mapping.expr_syntax(*expr) {
+ for (expr, ty) in inference_result.type_of_expr.iter() {
+ let syntax_ptr = if let Some(sp) = body_syntax_mapping.expr_syntax(expr) {
sp
} else {
continue;