Provide positional information when visiting ty, substs and closure_substs in MIR

Nashenas88 · Nashenas88 · commit 059a9a16d54f · 2017-07-18T22:31:38.000-04:00
diff --git a/src/librustc/mir/visit.rs b/src/librustc/mir/visit.rs
@@ -210,17 +210,20 @@ macro_rules! make_mir_visitor {
             }
 
             fn visit_ty(&mut self,
-                        ty: & $($mutability)* Ty<'tcx>) {
+                        ty: & $($mutability)* Ty<'tcx>,
+                        _: PositionalInfo) {
                 self.super_ty(ty);
             }
 
             fn visit_substs(&mut self,
-                            substs: & $($mutability)* &'tcx Substs<'tcx>) {
+                            substs: & $($mutability)* &'tcx Substs<'tcx>,
+                            _: Location) {
                 self.super_substs(substs);
             }
 
             fn visit_closure_substs(&mut self,
-                                    substs: & $($mutability)* ClosureSubsts<'tcx>) {
+                                    substs: & $($mutability)* ClosureSubsts<'tcx>,
+                                    _: Location) {
                 self.super_closure_substs(substs);
             }
 
@@ -266,7 +269,7 @@ macro_rules! make_mir_visitor {
                     self.visit_visibility_scope_data(scope);
                 }
 
-                self.visit_ty(&$($mutability)* mir.return_ty);
+                self.visit_ty(&$($mutability)* mir.return_ty, PositionalInfo::Span(mir.span));
 
                 for local_decl in &$($mutability)* mir.local_decls {
                     self.visit_local_decl(local_decl);
@@ -385,7 +388,7 @@ macro_rules! make_mir_visitor {
                                                 ref values,
                                                 ref targets } => {
                         self.visit_operand(discr, source_location);
-                        self.visit_ty(switch_ty);
+                        self.visit_ty(switch_ty, PositionalInfo::Location(source_location));
                         for value in &values[..] {
                             self.visit_const_int(value, source_location);
                         }
@@ -489,7 +492,7 @@ macro_rules! make_mir_visitor {
                                  ref $($mutability)* operand,
                                  ref $($mutability)* ty) => {
                         self.visit_operand(operand, location);
-                        self.visit_ty(ty);
+                        self.visit_ty(ty, PositionalInfo::Location(location));
                     }
 
                     Rvalue::BinaryOp(_bin_op,
@@ -511,28 +514,28 @@ macro_rules! make_mir_visitor {
                     }
 
                     Rvalue::NullaryOp(_op, ref $($mutability)* ty) => {
-                        self.visit_ty(ty);
+                        self.visit_ty(ty, PositionalInfo::Location(location));
                     }
 
                     Rvalue::Aggregate(ref $($mutability)* kind,
                                       ref $($mutability)* operands) => {
                         let kind = &$($mutability)* **kind;
                         match *kind {
                             AggregateKind::Array(ref $($mutability)* ty) => {
-                                self.visit_ty(ty);
+                                self.visit_ty(ty, PositionalInfo::Location(location));
                             }
                             AggregateKind::Tuple => {
                             }
                             AggregateKind::Adt(_adt_def,
                                                _variant_index,
                                                ref $($mutability)* substs,
                                                _active_field_index) => {
-                                self.visit_substs(substs);
+                                self.visit_substs(substs, location);
                             }
                             AggregateKind::Closure(ref $($mutability)* def_id,
                                                    ref $($mutability)* closure_substs) => {
                                 self.visit_def_id(def_id, location);
-                                self.visit_closure_substs(closure_substs);
+                                self.visit_closure_substs(closure_substs, location);
                             }
                         }
 
@@ -581,7 +584,7 @@ macro_rules! make_mir_visitor {
                     ref $($mutability)* ty,
                 } = *static_;
                 self.visit_def_id(def_id, location);
-                self.visit_ty(ty);
+                self.visit_ty(ty, PositionalInfo::Location(location));
             }
 
             fn super_projection(&mut self,
@@ -611,7 +614,7 @@ macro_rules! make_mir_visitor {
                     ProjectionElem::Subslice { from: _, to: _ } => {
                     }
                     ProjectionElem::Field(_field, ref $($mutability)* ty) => {
-                        self.visit_ty(ty);
+                        self.visit_ty(ty, PositionalInfo::Location(location));
                     }
                     ProjectionElem::Index(ref $($mutability)* operand) => {
                         self.visit_operand(operand, location);
@@ -635,7 +638,7 @@ macro_rules! make_mir_visitor {
                     is_user_variable: _,
                 } = *local_decl;
 
-                self.visit_ty(ty);
+                self.visit_ty(ty, PositionalInfo::SourceInfo(*source_info));
                 self.visit_source_info(source_info);
             }
 
@@ -658,7 +661,7 @@ macro_rules! make_mir_visitor {
                 } = *constant;
 
                 self.visit_span(span);
-                self.visit_ty(ty);
+                self.visit_ty(ty, PositionalInfo::Location(location));
                 self.visit_literal(literal, location);
             }
 
@@ -669,7 +672,7 @@ macro_rules! make_mir_visitor {
                     Literal::Item { ref $($mutability)* def_id,
                                     ref $($mutability)* substs } => {
                         self.visit_def_id(def_id, location);
-                        self.visit_substs(substs);
+                        self.visit_substs(substs, location);
                     }
                     Literal::Value { ref $($mutability)* value } => {
                         self.visit_const_val(value, location);
@@ -734,6 +737,12 @@ macro_rules! make_mir_visitor {
 make_mir_visitor!(Visitor,);
 make_mir_visitor!(MutVisitor,mut);
 
+pub enum PositionalInfo {
+    Location(Location),
+    SourceInfo(SourceInfo),
+    Span(Span),
+}
+
 #[derive(Copy, Clone, Debug, PartialEq, Eq)]
 pub enum LvalueContext<'tcx> {
     // Appears as LHS of an assignment
diff --git a/src/librustc_mir/build/mod.rs b/src/librustc_mir/build/mod.rs
@@ -17,7 +17,7 @@ use rustc::hir::def_id::DefId;
 use rustc::middle::region::CodeExtent;
 use rustc::mir::*;
 use rustc::mir::transform::MirSource;
-use rustc::mir::visit::MutVisitor;
+use rustc::mir::visit::{MutVisitor, PositionalInfo};
 use rustc::ty::{self, Ty, TyCtxt};
 use rustc::ty::subst::Substs;
 use rustc::util::nodemap::NodeMap;
@@ -143,7 +143,7 @@ struct GlobalizeMir<'a, 'gcx: 'a> {
 }
 
 impl<'a, 'gcx: 'tcx, 'tcx> MutVisitor<'tcx> for GlobalizeMir<'a, 'gcx> {
-    fn visit_ty(&mut self, ty: &mut Ty<'tcx>) {
+    fn visit_ty(&mut self, ty: &mut Ty<'tcx>, _: PositionalInfo) {
         if let Some(lifted) = self.tcx.lift(ty) {
             *ty = lifted;
         } else {
@@ -153,7 +153,7 @@ impl<'a, 'gcx: 'tcx, 'tcx> MutVisitor<'tcx> for GlobalizeMir<'a, 'gcx> {
         }
     }
 
-    fn visit_substs(&mut self, substs: &mut &'tcx Substs<'tcx>) {
+    fn visit_substs(&mut self, substs: &mut &'tcx Substs<'tcx>, _: Location) {
         if let Some(lifted) = self.tcx.lift(substs) {
             *substs = lifted;
         } else {
diff --git a/src/librustc_mir/transform/erase_regions.rs b/src/librustc_mir/transform/erase_regions.rs
@@ -15,7 +15,7 @@
 use rustc::ty::subst::Substs;
 use rustc::ty::{Ty, TyCtxt, ClosureSubsts};
 use rustc::mir::*;
-use rustc::mir::visit::MutVisitor;
+use rustc::mir::visit::{MutVisitor, PositionalInfo};
 use rustc::mir::transform::{MirPass, MirSource};
 
 struct EraseRegionsVisitor<'a, 'tcx: 'a> {
@@ -31,12 +31,12 @@ impl<'a, 'tcx> EraseRegionsVisitor<'a, 'tcx> {
 }
 
 impl<'a, 'tcx> MutVisitor<'tcx> for EraseRegionsVisitor<'a, 'tcx> {
-    fn visit_ty(&mut self, ty: &mut Ty<'tcx>) {
+    fn visit_ty(&mut self, ty: &mut Ty<'tcx>, _: PositionalInfo) {
         let old_ty = *ty;
         *ty = self.tcx.erase_regions(&old_ty);
     }
 
-    fn visit_substs(&mut self, substs: &mut &'tcx Substs<'tcx>) {
+    fn visit_substs(&mut self, substs: &mut &'tcx Substs<'tcx>, _: Location) {
         *substs = self.tcx.erase_regions(&{*substs});
     }
 
@@ -62,7 +62,8 @@ impl<'a, 'tcx> MutVisitor<'tcx> for EraseRegionsVisitor<'a, 'tcx> {
     }
 
     fn visit_closure_substs(&mut self,
-                            substs: &mut ClosureSubsts<'tcx>) {
+                            substs: &mut ClosureSubsts<'tcx>,
+                            _: Location) {
         *substs = self.tcx.erase_regions(substs);
     }
 
diff --git a/src/librustc_passes/mir_stats.rs b/src/librustc_passes/mir_stats.rs
@@ -279,7 +279,8 @@ impl<'a, 'tcx> mir_visit::Visitor<'tcx> for StatCollector<'a, 'tcx> {
     }
 
     fn visit_closure_substs(&mut self,
-                            substs: &ClosureSubsts<'tcx>) {
+                            substs: &ClosureSubsts<'tcx>,
+                            _: Location) {
         self.record("ClosureSubsts", substs);
         self.super_closure_substs(substs);
     }

Original file line number	Diff line number	Diff line change
`@@ -15,7 +15,7 @@`
`15`	`15`	`use rustc::ty::subst::Substs;`
`16`	`16`	`use rustc::ty::{Ty, TyCtxt, ClosureSubsts};`
`17`	`17`	`use rustc::mir::*;`
`18`		`-use rustc::mir::visit::MutVisitor;`
	`18`	`+use rustc::mir::visit::{MutVisitor, PositionalInfo};`
`19`	`19`	`use rustc::mir::transform::{MirPass, MirSource};`
`20`	`20`
`21`	`21`	`struct EraseRegionsVisitor<'a, 'tcx: 'a> {`
`@@ -31,12 +31,12 @@ impl<'a, 'tcx> EraseRegionsVisitor<'a, 'tcx> {`
`31`	`31`	`}`
`32`	`32`
`33`	`33`	`impl<'a, 'tcx> MutVisitor<'tcx> for EraseRegionsVisitor<'a, 'tcx> {`
`34`		`- fn visit_ty(&mut self, ty: &mut Ty<'tcx>) {`
	`34`	`+ fn visit_ty(&mut self, ty: &mut Ty<'tcx>, _: PositionalInfo) {`
`35`	`35`	`let old_ty = *ty;`
`36`	`36`	`*ty = self.tcx.erase_regions(&old_ty);`
`37`	`37`	`}`
`38`	`38`
`39`		`- fn visit_substs(&mut self, substs: &mut &'tcx Substs<'tcx>) {`
	`39`	`+ fn visit_substs(&mut self, substs: &mut &'tcx Substs<'tcx>, _: Location) {`
`40`	`40`	`substs = self.tcx.erase_regions(&{substs});`
`41`	`41`	`}`
`42`	`42`
`@@ -62,7 +62,8 @@ impl<'a, 'tcx> MutVisitor<'tcx> for EraseRegionsVisitor<'a, 'tcx> {`
`62`	`62`	`}`
`63`	`63`
`64`	`64`	`fn visit_closure_substs(&mut self,`
`65`		`- substs: &mut ClosureSubsts<'tcx>) {`
	`65`	`+ substs: &mut ClosureSubsts<'tcx>,`
	`66`	`+ _: Location) {`
`66`	`67`	`*substs = self.tcx.erase_regions(substs);`
`67`	`68`	`}`
`68`	`69`
Original file line number	Diff line number	Diff line change
`@@ -279,7 +279,8 @@ impl<'a, 'tcx> mir_visit::Visitor<'tcx> for StatCollector<'a, 'tcx> {`
`279`	`279`	`}`
`280`	`280`
`281`	`281`	`fn visit_closure_substs(&mut self,`
`282`		`- substs: &ClosureSubsts<'tcx>) {`
	`282`	`+ substs: &ClosureSubsts<'tcx>,`
	`283`	`+ _: Location) {`
`283`	`284`	`self.record("ClosureSubsts", substs);`
`284`	`285`	`self.super_closure_substs(substs);`
`285`	`286`	`}`