bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

fix node classification

Browse source
This commit is contained in:
Ralf Jung 2019-02-05 21:49:53 +01:00
parent 27ce224a8f
commit 84d75dbd7e
2 changed files with 19 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/librustc/hir/map/mod.rs
View file

@ -371,14 +371,11 @@ impl<'hir> Map<'hir> {
let def_id = self.local_def_id(variant.node.data.id());
Some(Def::Variant(def_id))
}
Node::AnonConst(item) => {
let def_id = self.local_def_id(item.id);
Some(Def::Const(def_id))
}
Node::StructCtor(variant) => {
let def_id = self.local_def_id(variant.id());
Some(Def::Fn(def_id))
Some(Def::StructCtor(def_id, def::CtorKind::from_hir(variant)))
}
Node::AnonConst(_) |
Node::Field(_) |
Node::Expr(_) |
Node::Stmt(_) |

23
src/librustc_mir/util/pretty.rs
View file

@ -595,13 +595,15 @@ fn write_mir_sig(
use rustc::hir::def::Def;
trace!("write_mir_sig: {:?}", src.instance);
let descr = tcx.describe_def(src.def_id()).unwrap();
let descr = tcx.describe_def(src.def_id());
match (descr, src.promoted) {
(_, Some(i)) => write!(w, "{:?} in ", i)?,
(Def::Fn(_), _) | (Def::Method(_), _) => write!(w, "fn")?,
(Def::Const(_), _) => write!(w, "const")?,
(Def::Static(_, /*is_mutbl*/false), _) => write!(w, "static")?,
(Def::Static(_, /*is_mutbl*/true), _) => write!(w, "static mut")?,
(Some(Def::Fn(_)), _) | (Some(Def::Method(_)), _) => write!(w, "fn ")?,
(Some(Def::StructCtor(..)), _) => write!(w, "struct ")?,
(Some(Def::Const(_)), _) => write!(w, "const ")?,
(Some(Def::Static(_, /*is_mutbl*/false)), _) => write!(w, "static ")?,
(Some(Def::Static(_, /*is_mutbl*/true)), _) => write!(w, "static mut ")?,
(None, _) => {}, // things like anon const, not an item
_ => bug!("Unexpected def description {:?}", descr),
}
@ -611,7 +613,10 @@ fn write_mir_sig(
})?;
match (descr, src.promoted) {
(Def::Fn(_), None) | (Def::Method(_), None) => {
(Some(Def::Fn(_)), None) |
(Some(Def::Method(_)), None) |
(Some(Def::StructCtor(..)), None) =>
{
write!(w, "(")?;
// fn argument types.
@ -624,11 +629,10 @@ fn write_mir_sig(
write!(w, ") -> {}", mir.return_ty())?;
}
(Def::Const(_), _) | (Def::Static(_, _), _) | (_, Some(_)) => {
_ => {
assert_eq!(mir.arg_count, 0);
write!(w, ": {} =", mir.return_ty())?;
}
_ => bug!("Unexpected def description {:?}", descr),
}
if let Some(yield_ty) = mir.yield_ty {
@ -636,6 +640,9 @@ fn write_mir_sig(
writeln!(w, "yields {}", yield_ty)?;
}
write!(w, " ")?;
// Next thing that gets printed is the opening {
Ok(())
}