use crate::middle::cstore::CrateStore;
use crate::session::CrateDisambiguator;
use crate::session::Session;
-use std::iter::repeat;
-use syntax::ast::{NodeId, CRATE_NODE_ID};
+use crate::util::nodemap::FxHashMap;
+use syntax::ast::NodeId;
use syntax::source_map::SourceMap;
use syntax_pos::Span;
source_map: &'a SourceMap,
/// The node map
- map: Vec<Option<Entry<'hir>>>,
+ map: FxHashMap<HirId, Entry<'hir>>,
/// The parent of this node
parent_node: hir::HirId,
let mut collector = NodeCollector {
krate,
source_map: sess.source_map(),
- map: repeat(None).take(sess.current_node_id_count()).collect(),
+ map: FxHashMap::with_capacity_and_hasher(sess.current_node_id_count(),
+ Default::default()),
parent_node: hir::CRATE_HIR_ID,
current_signature_dep_index: root_mod_sig_dep_index,
current_full_dep_index: root_mod_full_dep_index,
hcx,
hir_body_nodes,
};
- collector.insert_entry(CRATE_NODE_ID, Entry {
- parent: CRATE_NODE_ID,
- parent_hir: hir::CRATE_HIR_ID,
+ collector.insert_entry(hir::CRATE_HIR_ID, Entry {
+ parent: hir::CRATE_HIR_ID,
dep_node: root_mod_sig_dep_index,
node: Node::Crate,
});
crate_disambiguator: CrateDisambiguator,
cstore: &dyn CrateStore,
commandline_args_hash: u64)
- -> (Vec<Option<Entry<'hir>>>, Svh)
+ -> (FxHashMap<HirId, Entry<'hir>>, Svh)
{
self.hir_body_nodes.sort_unstable_by_key(|bn| bn.0);
(self.map, svh)
}
- fn insert_entry(&mut self, id: NodeId, entry: Entry<'hir>) {
+ fn insert_entry(&mut self, id: HirId, entry: Entry<'hir>) {
debug!("hir_map: {:?} => {:?}", id, entry);
- self.map[id.as_usize()] = Some(entry);
+ self.map.insert(id, entry);
}
fn insert(&mut self, span: Span, hir_id: HirId, node: Node<'hir>) {
let entry = Entry {
- parent: self.hir_to_node_id[&self.parent_node],
- parent_hir: self.parent_node,
+ parent: self.parent_node,
dep_node: if self.currently_in_body {
self.current_full_dep_index
} else {
node,
};
- let node_id = self.hir_to_node_id[&hir_id];
-
// Make sure that the DepNode of some node coincides with the HirId
// owner of that node.
if cfg!(debug_assertions) {
- assert_eq!(self.definitions.node_to_hir_id(node_id), hir_id);
+ let node_id = self.hir_to_node_id[&hir_id];
+ assert_eq!(self.definitions.node_to_hir_id(node_id), hir_id);
if hir_id.owner != self.current_dep_node_owner {
let node_str = match self.definitions.opt_def_index(node_id) {
}
}
- self.insert_entry(node_id, entry);
+ self.insert_entry(hir_id, entry);
}
fn with_parent<F: FnOnce(&mut Self)>(
use rustc_target::spec::abi::Abi;
use rustc_data_structures::svh::Svh;
-use syntax::ast::{self, Name, NodeId, CRATE_NODE_ID};
+use syntax::ast::{self, Name, NodeId};
use syntax::source_map::Spanned;
use syntax::ext::base::MacroKind;
use syntax_pos::{Span, DUMMY_SP};
/// Represents an entry and its parent `NodeId`.
#[derive(Copy, Clone, Debug)]
pub struct Entry<'hir> {
- parent: NodeId,
- parent_hir: HirId,
+ parent: HirId,
dep_node: DepNodeIndex,
node: Node<'hir>,
}
impl<'hir> Entry<'hir> {
- fn parent_node(self) -> Option<NodeId> {
+ fn parent_node(self) -> Option<HirId> {
match self.node {
Node::Crate | Node::MacroDef(_) => None,
_ => Some(self.parent),
///
/// Also, indexing is pretty quick when you've got a vector and
/// plain old integers.
- map: Vec<Option<Entry<'hir>>>,
+ map: FxHashMap<HirId, Entry<'hir>>,
definitions: &'hir Definitions,
/// read recorded). If the function just returns a DefId or
/// NodeId, no actual content was returned, so no read is needed.
pub fn read(&self, id: NodeId) {
- if let Some(entry) = self.map[id.as_usize()] {
+ let hir_id = self.node_to_hir_id(id);
+ if let Some(entry) = self.map.get(&hir_id) {
self.dep_graph.read_index(entry.dep_node);
} else {
bug!("called `HirMap::read()` with invalid `NodeId`: {:?}", id)
#[inline]
pub fn local_def_id(&self, node: NodeId) -> DefId {
self.opt_local_def_id(node).unwrap_or_else(|| {
+ let hir_id = self.node_to_hir_id(node);
bug!("local_def_id: no entry for `{}`, which has a map of `{:?}`",
- node, self.find_entry(node))
+ node, self.find_entry(hir_id))
})
}
// FIXME(@ljedrz): replace the NodeId variant
#[inline]
pub fn local_def_id_from_hir_id(&self, hir_id: HirId) -> DefId {
- let node_id = self.hir_to_node_id(hir_id);
- self.opt_local_def_id(node_id).unwrap_or_else(|| {
+ self.opt_local_def_id_from_hir_id(hir_id).unwrap_or_else(|| {
bug!("local_def_id_from_hir_id: no entry for `{:?}`, which has a map of `{:?}`",
- hir_id, self.find_entry(node_id))
+ hir_id, self.find_entry(hir_id))
})
}
self.map.len()
}
- fn find_entry(&self, id: NodeId) -> Option<Entry<'hir>> {
- self.map.get(id.as_usize()).cloned().unwrap_or(None)
+ fn find_entry(&self, id: HirId) -> Option<Entry<'hir>> {
+ self.map.get(&id).cloned()
}
pub fn krate(&self) -> &'hir Crate {
}
pub fn fn_decl(&self, node_id: ast::NodeId) -> Option<FnDecl> {
- if let Some(entry) = self.find_entry(node_id) {
+ let hir_id = self.node_to_hir_id(node_id);
+ if let Some(entry) = self.find_entry(hir_id) {
entry.fn_decl().cloned()
} else {
bug!("no entry for node_id `{}`", node_id)
/// which this is the body of, i.e., a `fn`, `const` or `static`
/// item (possibly associated), a closure, or a `hir::AnonConst`.
pub fn body_owner(&self, BodyId { hir_id }: BodyId) -> NodeId {
- let node_id = self.hir_to_node_id(hir_id);
- let parent = self.get_parent_node(node_id);
- assert!(self.map[parent.as_usize()].map_or(false, |e| e.is_body_owner(hir_id)));
- parent
+ let parent = self.get_parent_node_by_hir_id(hir_id);
+ assert!(self.map.get(&parent).map_or(false, |e| e.is_body_owner(hir_id)));
+ self.hir_to_node_id(parent)
}
pub fn body_owner_def_id(&self, id: BodyId) -> DefId {
/// Given a `NodeId`, returns the `BodyId` associated with it,
/// if the node is a body owner, otherwise returns `None`.
pub fn maybe_body_owned_by(&self, id: NodeId) -> Option<BodyId> {
- if let Some(entry) = self.find_entry(id) {
+ let hir_id = self.node_to_hir_id(id);
+ if let Some(entry) = self.find_entry(hir_id) {
if self.dep_graph.is_fully_enabled() {
- let hir_id_owner = self.node_to_hir_id(id).owner;
+ let hir_id_owner = hir_id.owner;
let def_path_hash = self.definitions.def_path_hash(hir_id_owner);
self.dep_graph.read(def_path_hash.to_dep_node(DepKind::HirBody));
}
&self.forest.krate.attrs
}
- pub fn get_module(&self, module: DefId) -> (&'hir Mod, Span, HirId) {
- let node_id = self.as_local_node_id(module).unwrap();
- let hir_id = self.node_to_hir_id(node_id);
- self.read(node_id);
- match self.find_entry(node_id).unwrap().node {
+ pub fn get_module(&self, module: DefId) -> (&'hir Mod, Span, HirId)
+ {
+ let hir_id = self.as_local_hir_id(module).unwrap();
+ self.read_by_hir_id(hir_id);
+ match self.find_entry(hir_id).unwrap().node {
Node::Item(&Item {
span,
node: ItemKind::Mod(ref m),
/// Retrieves the `Node` corresponding to `id`, returning `None` if cannot be found.
pub fn find(&self, id: NodeId) -> Option<Node<'hir>> {
- let result = self.find_entry(id).and_then(|entry| {
+ let hir_id = self.node_to_hir_id(id);
+ let result = self.find_entry(hir_id).and_then(|entry| {
if let Node::Crate = entry.node {
None
} else {
}
});
if result.is_some() {
- self.read(id);
+ self.read_by_hir_id(hir_id);
}
result
}
/// from a node to the root of the ast (unless you get the same ID back here
/// that can happen if the ID is not in the map itself or is just weird).
pub fn get_parent_node(&self, id: NodeId) -> NodeId {
+ let hir_id = self.node_to_hir_id(id);
if self.dep_graph.is_fully_enabled() {
- let hir_id_owner = self.node_to_hir_id(id).owner;
+ let hir_id_owner = hir_id.owner;
let def_path_hash = self.definitions.def_path_hash(hir_id_owner);
self.dep_graph.read(def_path_hash.to_dep_node(DepKind::HirBody));
}
- self.find_entry(id).and_then(|x| x.parent_node()).unwrap_or(id)
+ self.find_entry(hir_id)
+ .and_then(|x| x.parent_node())
+ .map(|x| self.hir_to_node_id(x))
+ .unwrap_or(id)
}
// FIXME(@ljedrz): replace the NodeId variant
/// is not an error, since items in the crate module have the crate root as
/// parent.
fn walk_parent_nodes<F, F2>(&self,
- start_id: NodeId,
+ start_id: HirId,
found: F,
bail_early: F2)
- -> Result<NodeId, NodeId>
+ -> Result<HirId, HirId>
where F: Fn(&Node<'hir>) -> bool, F2: Fn(&Node<'hir>) -> bool
{
let mut id = start_id;
loop {
- let parent_node = self.get_parent_node(id);
- if parent_node == CRATE_NODE_ID {
- return Ok(CRATE_NODE_ID);
+ let parent_node = self.get_parent_node_by_hir_id(id);
+ if parent_node == CRATE_HIR_ID {
+ return Ok(CRATE_HIR_ID);
}
if parent_node == id {
return Err(id);
}
};
- let node_id = self.hir_to_node_id(id);
- self.walk_parent_nodes(node_id, match_fn, match_non_returning_block)
- .ok()
- .map(|return_node_id| self.node_to_hir_id(return_node_id))
+ self.walk_parent_nodes(id, match_fn, match_non_returning_block).ok()
}
/// Retrieves the `NodeId` for `id`'s parent item, or `id` itself if no
/// in the HIR which is recorded by the map and is an item, either an item
/// in a module, trait, or impl.
pub fn get_parent(&self, id: NodeId) -> NodeId {
- match self.walk_parent_nodes(id, |node| match *node {
+ let hir_id = self.node_to_hir_id(id);
+ let parent_hid = match self.walk_parent_nodes(hir_id, |node| match *node {
Node::Item(_) |
Node::ForeignItem(_) |
Node::TraitItem(_) |
}, |_| false) {
Ok(id) => id,
Err(id) => id,
- }
+ };
+
+ self.hir_to_node_id(parent_hid)
}
// FIXME(@ljedrz): replace the NodeId variant
/// Returns the `NodeId` of `id`'s nearest module parent, or `id` itself if no
/// module parent is in this map.
pub fn get_module_parent_node(&self, id: NodeId) -> NodeId {
- match self.walk_parent_nodes(id, |node| match *node {
+ let hir_id = self.node_to_hir_id(id);
+ let parent_hid = match self.walk_parent_nodes(hir_id, |node| match *node {
Node::Item(&Item { node: ItemKind::Mod(_), .. }) => true,
_ => false,
}, |_| false) {
Ok(id) => id,
Err(id) => id,
- }
+ };
+
+ self.hir_to_node_id(parent_hid)
}
/// Returns the nearest enclosing scope. A scope is an item or block.
/// and associated types probably shouldn't, for example. Behavior in this
/// regard should be expected to be highly unstable.
pub fn get_enclosing_scope(&self, id: NodeId) -> Option<NodeId> {
- self.walk_parent_nodes(id, |node| match *node {
+ let hir_id = self.node_to_hir_id(id);
+ let parent_hid = self.walk_parent_nodes(hir_id, |node| match *node {
Node::Item(_) |
Node::ForeignItem(_) |
Node::TraitItem(_) |
Node::ImplItem(_) |
Node::Block(_) => true,
_ => false,
- }, |_| false).ok()
+ }, |_| false).ok();
+
+ parent_hid.map(|hid| self.hir_to_node_id(hid))
}
pub fn get_parent_did(&self, id: NodeId) -> DefId {
}
pub fn get_foreign_abi(&self, id: NodeId) -> Abi {
- let parent = self.get_parent(id);
+ let hir_id = self.node_to_hir_id(id);
+ let parent = self.get_parent_item(hir_id);
if let Some(entry) = self.find_entry(parent) {
if let Entry {
node: Node::Item(Item { node: ItemKind::ForeignMod(ref nm), .. }), .. } = entry
{
- self.read(id); // reveals some of the content of a node
+ self.read_by_hir_id(hir_id); // reveals some of the content of a node
return nm.abi;
}
}
- bug!("expected foreign mod or inlined parent, found {}", self.node_to_string(parent))
+ bug!("expected foreign mod or inlined parent, found {}", self.hir_to_string(parent))
}
// FIXME(@ljedrz): replace the NodeId variant
map: self,
item_name: parts.last().unwrap(),
in_which: &parts[..parts.len() - 1],
- idx: CRATE_NODE_ID,
+ idx: ast::CRATE_NODE_ID,
}
}
pub fn span(&self, id: NodeId) -> Span {
- self.read(id); // reveals span from node
- match self.find_entry(id).map(|entry| entry.node) {
+ let hir_id = self.node_to_hir_id(id);
+ self.read_by_hir_id(hir_id); // reveals span from node
+ match self.find_entry(hir_id).map(|entry| entry.node) {
Some(Node::Item(item)) => item.span,
Some(Node::ForeignItem(foreign_item)) => foreign_item.span,
Some(Node::TraitItem(trait_method)) => trait_method.span,
return None;
}
self.idx = NodeId::from_u32(self.idx.as_u32() + 1);
- let name = match self.map.find_entry(idx).map(|entry| entry.node) {
+ let hir_idx = self.map.node_to_hir_id(idx);
+ let name = match self.map.find_entry(hir_idx).map(|entry| entry.node) {
Some(Node::Item(n)) => n.name(),
Some(Node::ForeignItem(n)) => n.name(),
Some(Node::TraitItem(n)) => n.name(),
)
};
- if log_enabled!(::log::Level::Debug) {
- // This only makes sense for ordered stores; note the
- // enumerate to count the number of entries.
- let (entries_less_1, _) = map.iter().filter_map(|x| *x).enumerate().last()
- .expect("AST map was empty after folding?");
-
- let entries = entries_less_1 + 1;
- let vector_length = map.len();
- debug!("The AST map has {} entries with a maximum of {}: occupancy {:.1}%",
- entries, vector_length, (entries as f64 / vector_length as f64) * 100.);
- }
-
let map = Map {
forest,
dep_graph: forest.dep_graph.clone(),