Skip to content

Commit fb84bb1

Browse files
scottcarrscottcarr
committed
track Location in visitor, combine Location
1 parent 42001ed commit fb84bb1

File tree

18 files changed

+274
-251
lines changed

18 files changed

+274
-251
lines changed

‎src/librustc/mir/repr.rs

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1229,3 +1229,14 @@ impl<'a, 'b> GraphSuccessors<'b> for Mir<'a> {
12291229
type Item = BasicBlock;
12301230
type Iter = IntoIter<BasicBlock>;
12311231
}
1232+
1233+
#[derive(Copy, Clone, PartialEq, Eq, Debug, Hash, Ord, PartialOrd)]
1234+
pub struct Location {
1235+
/// the location is within this block
1236+
pub block: BasicBlock,
1237+
1238+
/// the location is the start of the this statement; or, if `statement_index`
1239+
/// == num-statements, then the start of the terminator.
1240+
pub statement_index: usize,
1241+
}
1242+

‎src/librustc/mir/visit.rs

Lines changed: 119 additions & 86 deletions
Large diffs are not rendered by default.

‎src/librustc_borrowck/borrowck/mir/dataflow/impls.rs

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,9 @@
99
// except according to those terms.
1010

1111
use rustc::ty::TyCtxt;
12-
use rustc::mir::repr::{self, Mir};
12+
use rustc::mir::repr::{self, Mir, Location};
1313
use rustc_data_structures::indexed_vec::Idx;
1414

15-
use super::super::gather_moves::{Location};
1615
use super::super::gather_moves::{MoveOutIndex, MovePathIndex};
1716
use super::super::MoveDataParamEnv;
1817
use super::super::DropFlagState;
@@ -252,7 +251,7 @@ impl<'a, 'tcx> BitDenotation for MaybeInitializedLvals<'a, 'tcx> {
252251
{
253252
drop_flag_effects_for_location(
254253
self.tcx, self.mir, ctxt,
255-
Location { block: bb, index: idx },
254+
Location { block: bb, statement_index: idx },
256255
|path, s| Self::update_bits(sets, path, s)
257256
)
258257
}
@@ -265,7 +264,7 @@ impl<'a, 'tcx> BitDenotation for MaybeInitializedLvals<'a, 'tcx> {
265264
{
266265
drop_flag_effects_for_location(
267266
self.tcx, self.mir, ctxt,
268-
Location { block: bb, index: statements_len },
267+
Location { block: bb, statement_index: statements_len },
269268
|path, s| Self::update_bits(sets, path, s)
270269
)
271270
}
@@ -314,7 +313,7 @@ impl<'a, 'tcx> BitDenotation for MaybeUninitializedLvals<'a, 'tcx> {
314313
{
315314
drop_flag_effects_for_location(
316315
self.tcx, self.mir, ctxt,
317-
Location { block: bb, index: idx },
316+
Location { block: bb, statement_index: idx },
318317
|path, s| Self::update_bits(sets, path, s)
319318
)
320319
}
@@ -327,7 +326,7 @@ impl<'a, 'tcx> BitDenotation for MaybeUninitializedLvals<'a, 'tcx> {
327326
{
328327
drop_flag_effects_for_location(
329328
self.tcx, self.mir, ctxt,
330-
Location { block: bb, index: statements_len },
329+
Location { block: bb, statement_index: statements_len },
331330
|path, s| Self::update_bits(sets, path, s)
332331
)
333332
}
@@ -375,7 +374,7 @@ impl<'a, 'tcx> BitDenotation for DefinitelyInitializedLvals<'a, 'tcx> {
375374
{
376375
drop_flag_effects_for_location(
377376
self.tcx, self.mir, ctxt,
378-
Location { block: bb, index: idx },
377+
Location { block: bb, statement_index: idx },
379378
|path, s| Self::update_bits(sets, path, s)
380379
)
381380
}
@@ -388,7 +387,7 @@ impl<'a, 'tcx> BitDenotation for DefinitelyInitializedLvals<'a, 'tcx> {
388387
{
389388
drop_flag_effects_for_location(
390389
self.tcx, self.mir, ctxt,
391-
Location { block: bb, index: statements_len },
390+
Location { block: bb, statement_index: statements_len },
392391
|path, s| Self::update_bits(sets, path, s)
393392
)
394393
}
@@ -431,7 +430,7 @@ impl<'a, 'tcx> BitDenotation for MovingOutStatements<'a, 'tcx> {
431430
let path_map = &move_data.path_map;
432431
let rev_lookup = &move_data.rev_lookup;
433432

434-
let loc = Location { block: bb, index: idx };
433+
let loc = Location { block: bb, statement_index: idx };
435434
debug!("stmt {:?} at loc {:?} moves out of move_indexes {:?}",
436435
stmt, loc, &loc_map[loc]);
437436
for move_index in &loc_map[loc] {
@@ -468,7 +467,7 @@ impl<'a, 'tcx> BitDenotation for MovingOutStatements<'a, 'tcx> {
468467
let (mir, move_data) = (self.mir, &ctxt.move_data);
469468
let term = mir[bb].terminator();
470469
let loc_map = &move_data.loc_map;
471-
let loc = Location { block: bb, index: statements_len };
470+
let loc = Location { block: bb, statement_index: statements_len };
472471
debug!("terminator {:?} at loc {:?} moves out of move_indexes {:?}",
473472
term, loc, &loc_map[loc]);
474473
let bits_per_block = self.bits_per_block(ctxt);

‎src/librustc_borrowck/borrowck/mir/elaborate_drops.rs

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
// except according to those terms.
1010

1111
use indexed_set::IdxSetBuf;
12-
use super::gather_moves::{MoveData, MovePathIndex, MovePathContent, Location};
12+
use super::gather_moves::{MoveData, MovePathIndex, MovePathContent};
1313
use super::dataflow::{MaybeInitializedLvals, MaybeUninitializedLvals};
1414
use super::dataflow::{DataflowResults};
1515
use super::{drop_flag_effects_for_location, on_all_children_bits};
@@ -146,9 +146,9 @@ impl<'b, 'tcx> ElaborateDropsCtxt<'b, 'tcx> {
146146
dead: self.flow_uninits.sets().on_entry_set_for(loc.block.index())
147147
.to_owned(),
148148
};
149-
for stmt in 0..loc.index {
149+
for stmt in 0..loc.statement_index {
150150
data.apply_location(self.tcx, self.mir, self.env,
151-
Location { block: loc.block, index: stmt });
151+
Location { block: loc.block, statement_index: stmt });
152152
}
153153
data
154154
}
@@ -226,7 +226,7 @@ impl<'b, 'tcx> ElaborateDropsCtxt<'b, 'tcx> {
226226

227227
let init_data = self.initialization_data_at(Location {
228228
block: bb,
229-
index: data.statements.len()
229+
statement_index: data.statements.len()
230230
});
231231

232232
let path = self.move_data().rev_lookup.find(location);
@@ -249,7 +249,7 @@ impl<'b, 'tcx> ElaborateDropsCtxt<'b, 'tcx> {
249249
fn elaborate_drops(&mut self)
250250
{
251251
for (bb, data) in self.mir.basic_blocks().iter_enumerated() {
252-
let loc = Location { block: bb, index: data.statements.len() };
252+
let loc = Location { block: bb, statement_index: data.statements.len() };
253253
let terminator = data.terminator();
254254

255255
let resume_block = self.patch.resume_block();
@@ -359,9 +359,9 @@ impl<'b, 'tcx> ElaborateDropsCtxt<'b, 'tcx> {
359359
unwind: Some(unwind)
360360
}, bb);
361361
on_all_children_bits(self.tcx, self.mir, self.move_data(), path, |child| {
362-
self.set_drop_flag(Location { block: target, index: 0 },
362+
self.set_drop_flag(Location { block: target, statement_index: 0 },
363363
child, DropFlagState::Present);
364-
self.set_drop_flag(Location { block: unwind, index: 0 },
364+
self.set_drop_flag(Location { block: unwind, statement_index: 0 },
365365
child, DropFlagState::Present);
366366
});
367367
}
@@ -741,7 +741,7 @@ impl<'b, 'tcx> ElaborateDropsCtxt<'b, 'tcx> {
741741
let drop_block = self.drop_block(c);
742742
if update_drop_flag {
743743
self.set_drop_flag(
744-
Location { block: drop_block, index: 0 },
744+
Location { block: drop_block, statement_index: 0 },
745745
c.path,
746746
DropFlagState::Absent
747747
);
@@ -927,7 +927,7 @@ impl<'b, 'tcx> ElaborateDropsCtxt<'b, 'tcx> {
927927
}
928928

929929
fn drop_flags_on_init(&mut self) {
930-
let loc = Location { block: START_BLOCK, index: 0 };
930+
let loc = Location { block: START_BLOCK, statement_index: 0 };
931931
let span = self.patch.source_info_for_location(self.mir, loc).span;
932932
let false_ = self.constant_bool(span, false);
933933
for flag in self.drop_flags.values() {
@@ -942,7 +942,7 @@ impl<'b, 'tcx> ElaborateDropsCtxt<'b, 'tcx> {
942942
} = data.terminator().kind {
943943
assert!(!self.patch.is_patched(bb));
944944

945-
let loc = Location { block: tgt, index: 0 };
945+
let loc = Location { block: tgt, statement_index: 0 };
946946
let path = self.move_data().rev_lookup.find(lv);
947947
on_all_children_bits(
948948
self.tcx, self.mir, self.move_data(), path,
@@ -953,7 +953,7 @@ impl<'b, 'tcx> ElaborateDropsCtxt<'b, 'tcx> {
953953
}
954954

955955
fn drop_flags_for_args(&mut self) {
956-
let loc = Location { block: START_BLOCK, index: 0 };
956+
let loc = Location { block: START_BLOCK, statement_index: 0 };
957957
super::drop_flag_effects_for_function_entry(
958958
self.tcx, self.mir, self.env, |path, ds| {
959959
self.set_drop_flag(loc, path, ds);
@@ -993,7 +993,7 @@ impl<'b, 'tcx> ElaborateDropsCtxt<'b, 'tcx> {
993993
}
994994
}
995995
}
996-
let loc = Location { block: bb, index: i };
996+
let loc = Location { block: bb, statement_index: i };
997997
super::drop_flag_effects_for_location(
998998
self.tcx, self.mir, self.env, loc, |path, ds| {
999999
if ds == DropFlagState::Absent || allow_initializations {
@@ -1011,7 +1011,7 @@ impl<'b, 'tcx> ElaborateDropsCtxt<'b, 'tcx> {
10111011
} = data.terminator().kind {
10121012
assert!(!self.patch.is_patched(bb));
10131013

1014-
let loc = Location { block: bb, index: data.statements.len() };
1014+
let loc = Location { block: bb, statement_index: data.statements.len() };
10151015
let path = self.move_data().rev_lookup.find(lv);
10161016
on_all_children_bits(
10171017
self.tcx, self.mir, self.move_data(), path,

‎src/librustc_borrowck/borrowck/mir/gather_moves.rs

Lines changed: 9 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -160,8 +160,8 @@ impl Index<Location> for LocMap {
160160
type Output = [MoveOutIndex];
161161
fn index(&self, index: Location) -> &Self::Output {
162162
assert!(index.block.index() < self.map.len());
163-
assert!(index.index < self.map[index.block.index()].len());
164-
&self.map[index.block.index()][index.index]
163+
assert!(index.statement_index < self.map[index.block.index()].len());
164+
&self.map[index.block.index()][index.statement_index]
165165
}
166166
}
167167

@@ -200,21 +200,6 @@ impl fmt::Debug for MoveOut {
200200
}
201201
}
202202

203-
#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord)]
204-
pub struct Location {
205-
/// block where action is located
206-
pub block: BasicBlock,
207-
/// index within above block; statement when < statments.len) or
208-
/// the terminator (when = statements.len).
209-
pub index: usize,
210-
}
211-
212-
impl fmt::Debug for Location {
213-
fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
214-
write!(fmt, "{:?}[{}]", self.block, self.index)
215-
}
216-
}
217-
218203
#[derive(Debug)]
219204
pub struct MovePathData<'tcx> {
220205
move_paths: Vec<MovePath<'tcx>>,
@@ -571,7 +556,7 @@ fn gather_moves<'a, 'tcx>(mir: &Mir<'tcx>, tcx: TyCtxt<'a, 'tcx, 'tcx>) -> MoveD
571556
};
572557

573558
for (i, stmt) in bb_data.statements.iter().enumerate() {
574-
let source = Location { block: bb, index: i };
559+
let source = Location { block: bb, statement_index: i };
575560
match stmt.kind {
576561
StatementKind::Assign(ref lval, ref rval) => {
577562
bb_ctxt.builder.create_move_path(lval);
@@ -627,7 +612,7 @@ fn gather_moves<'a, 'tcx>(mir: &Mir<'tcx>, tcx: TyCtxt<'a, 'tcx, 'tcx>) -> MoveD
627612

628613
TerminatorKind::Return => {
629614
let source = Location { block: bb,
630-
index: bb_data.statements.len() };
615+
statement_index: bb_data.statements.len() };
631616
if let FnOutput::FnConverging(_) = bb_ctxt.builder.mir.return_ty {
632617
debug!("gather_moves Return on_move_out_lval return {:?}", source);
633618
bb_ctxt.on_move_out_lval(SK::Return, &Lvalue::ReturnPointer, source);
@@ -639,7 +624,7 @@ fn gather_moves<'a, 'tcx>(mir: &Mir<'tcx>, tcx: TyCtxt<'a, 'tcx, 'tcx>) -> MoveD
639624

640625
TerminatorKind::If { ref cond, targets: _ } => {
641626
let source = Location { block: bb,
642-
index: bb_data.statements.len() };
627+
statement_index: bb_data.statements.len() };
643628
bb_ctxt.on_operand(SK::If, cond, source);
644629
}
645630

@@ -670,20 +655,20 @@ fn gather_moves<'a, 'tcx>(mir: &Mir<'tcx>, tcx: TyCtxt<'a, 'tcx, 'tcx>) -> MoveD
670655

671656
TerminatorKind::Drop { ref location, target: _, unwind: _ } => {
672657
let source = Location { block: bb,
673-
index: bb_data.statements.len() };
658+
statement_index: bb_data.statements.len() };
674659
bb_ctxt.on_move_out_lval(SK::Drop, location, source);
675660
}
676661
TerminatorKind::DropAndReplace { ref location, ref value, .. } => {
677662
let assigned_path = bb_ctxt.builder.move_path_for(location);
678663
bb_ctxt.path_map.fill_to(assigned_path.index());
679664

680665
let source = Location { block: bb,
681-
index: bb_data.statements.len() };
666+
statement_index: bb_data.statements.len() };
682667
bb_ctxt.on_operand(SK::Use, value, source);
683668
}
684669
TerminatorKind::Call { ref func, ref args, ref destination, cleanup: _ } => {
685670
let source = Location { block: bb,
686-
index: bb_data.statements.len() };
671+
statement_index: bb_data.statements.len() };
687672
bb_ctxt.on_operand(SK::CallFn, func, source);
688673
for arg in args {
689674
debug!("gather_moves Call on_operand {:?} {:?}", arg, source);
@@ -758,7 +743,7 @@ impl<'b, 'a: 'b, 'tcx: 'a> BlockContext<'b, 'a, 'tcx> {
758743
stmt_kind: StmtKind,
759744
lval: &Lvalue<'tcx>,
760745
source: Location) {
761-
let i = source.index;
746+
let i = source.statement_index;
762747
let index = MoveOutIndex::new(self.moves.len());
763748

764749
let path = self.builder.move_path_for(lval);

‎src/librustc_borrowck/borrowck/mir/mod.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ use rustc::hir;
1919
use rustc::hir::intravisit::{FnKind};
2020

2121
use rustc::mir::repr;
22-
use rustc::mir::repr::{BasicBlock, BasicBlockData, Mir, Statement, Terminator};
22+
use rustc::mir::repr::{BasicBlock, BasicBlockData, Mir, Statement, Terminator, Location};
2323
use rustc::session::Session;
2424
use rustc::ty::{self, TyCtxt};
2525

@@ -35,7 +35,7 @@ use self::dataflow::{DataflowOperator};
3535
use self::dataflow::{Dataflow, DataflowAnalysis, DataflowResults};
3636
use self::dataflow::{MaybeInitializedLvals, MaybeUninitializedLvals};
3737
use self::dataflow::{DefinitelyInitializedLvals};
38-
use self::gather_moves::{MoveData, MovePathIndex, Location};
38+
use self::gather_moves::{MoveData, MovePathIndex};
3939
use self::gather_moves::{MovePathContent, MovePathData};
4040

4141
fn has_rustc_mir_with(attrs: &[ast::Attribute], name: &str) -> Option<P<MetaItem>> {
@@ -367,7 +367,7 @@ fn drop_flag_effects_for_location<'a, 'tcx, F>(
367367
}
368368

369369
let block = &mir[loc.block];
370-
match block.statements.get(loc.index) {
370+
match block.statements.get(loc.statement_index) {
371371
Some(stmt) => match stmt.kind {
372372
repr::StatementKind::Assign(ref lvalue, _) => {
373373
debug!("drop_flag_effects: assignment {:?}", stmt);

‎src/librustc_borrowck/borrowck/mir/patch.rs

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
// option. This file may not be copied, modified, or distributed
99
// except according to those terms.
1010

11-
use super::gather_moves::Location;
1211
use rustc::ty::Ty;
1312
use rustc::mir::repr::*;
1413
use rustc_data_structures::indexed_vec::{IndexVec, Idx};
@@ -89,7 +88,7 @@ impl<'tcx> MirPatch<'tcx> {
8988
};
9089
Location {
9190
block: bb,
92-
index: offset
91+
statement_index: offset
9392
}
9493
}
9594

@@ -149,12 +148,12 @@ impl<'tcx> MirPatch<'tcx> {
149148
}
150149
debug!("MirPatch: adding statement {:?} at loc {:?}+{}",
151150
stmt, loc, delta);
152-
loc.index += delta;
151+
loc.statement_index += delta;
153152
let source_info = Self::source_info_for_index(
154153
&mir[loc.block], loc
155154
);
156155
mir[loc.block].statements.insert(
157-
loc.index, Statement {
156+
loc.statement_index, Statement {
158157
source_info: source_info,
159158
kind: stmt
160159
});
@@ -163,7 +162,7 @@ impl<'tcx> MirPatch<'tcx> {
163162
}
164163

165164
pub fn source_info_for_index(data: &BasicBlockData, loc: Location) -> SourceInfo {
166-
match data.statements.get(loc.index) {
165+
match data.statements.get(loc.statement_index) {
167166
Some(stmt) => stmt.source_info,
168167
None => data.terminator().source_info
169168
}

‎src/librustc_metadata/decoder.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ use rustc_const_math::ConstInt;
4343

4444
use rustc::mir;
4545
use rustc::mir::visit::MutVisitor;
46+
use rustc::mir::repr::Location;
4647

4748
use std::cell::Cell;
4849
use std::io;
@@ -857,7 +858,7 @@ pub fn maybe_get_item_mir<'a, 'tcx>(cdata: Cmd,
857858
impl<'v, 'cdata, 'codemap> mir::visit::MutVisitor<'v>
858859
for MirDefIdAndSpanTranslator<'cdata, 'codemap>
859860
{
860-
fn visit_def_id(&mut self, def_id: &mut DefId) {
861+
fn visit_def_id(&mut self, def_id: &mut DefId, _: Location) {
861862
*def_id = translate_def_id(self.crate_metadata, *def_id);
862863
}
863864

‎src/librustc_mir/build/cfg.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313

1414
//! Routines for manipulating the control-flow graph.
1515
16-
use build::{CFG, Location};
16+
use build::CFG;
1717
use rustc::mir::repr::*;
1818

1919
impl<'tcx> CFG<'tcx> {

0 commit comments

Comments
 (0)