Add option to mir::MutVisitor
to not invalidate CFG.
This also applies that option to some uses of the visitor
This commit is contained in:
parent
9ee22ff7e8
commit
7547084ff6
8 changed files with 101 additions and 72 deletions
|
@ -80,6 +80,8 @@ macro_rules! make_mir_visitor {
|
|||
self.super_body(body);
|
||||
}
|
||||
|
||||
extra_body_methods!($($mutability)?);
|
||||
|
||||
fn visit_basic_block_data(
|
||||
&mut self,
|
||||
block: BasicBlock,
|
||||
|
@ -287,63 +289,7 @@ macro_rules! make_mir_visitor {
|
|||
&mut self,
|
||||
body: &$($mutability)? Body<'tcx>,
|
||||
) {
|
||||
let span = body.span;
|
||||
if let Some(gen) = &$($mutability)? body.generator {
|
||||
if let Some(yield_ty) = $(& $mutability)? gen.yield_ty {
|
||||
self.visit_ty(
|
||||
yield_ty,
|
||||
TyContext::YieldTy(SourceInfo::outermost(span))
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
// for best performance, we want to use an iterator rather
|
||||
// than a for-loop, to avoid calling `body::Body::invalidate` for
|
||||
// each basic block.
|
||||
#[allow(unused_macro_rules)]
|
||||
macro_rules! basic_blocks {
|
||||
(mut) => (body.basic_blocks_mut().iter_enumerated_mut());
|
||||
() => (body.basic_blocks().iter_enumerated());
|
||||
}
|
||||
for (bb, data) in basic_blocks!($($mutability)?) {
|
||||
self.visit_basic_block_data(bb, data);
|
||||
}
|
||||
|
||||
for scope in &$($mutability)? body.source_scopes {
|
||||
self.visit_source_scope_data(scope);
|
||||
}
|
||||
|
||||
self.visit_ty(
|
||||
$(& $mutability)? body.return_ty(),
|
||||
TyContext::ReturnTy(SourceInfo::outermost(body.span))
|
||||
);
|
||||
|
||||
for local in body.local_decls.indices() {
|
||||
self.visit_local_decl(local, & $($mutability)? body.local_decls[local]);
|
||||
}
|
||||
|
||||
#[allow(unused_macro_rules)]
|
||||
macro_rules! type_annotations {
|
||||
(mut) => (body.user_type_annotations.iter_enumerated_mut());
|
||||
() => (body.user_type_annotations.iter_enumerated());
|
||||
}
|
||||
|
||||
for (index, annotation) in type_annotations!($($mutability)?) {
|
||||
self.visit_user_type_annotation(
|
||||
index, annotation
|
||||
);
|
||||
}
|
||||
|
||||
for var_debug_info in &$($mutability)? body.var_debug_info {
|
||||
self.visit_var_debug_info(var_debug_info);
|
||||
}
|
||||
|
||||
self.visit_span($(& $mutability)? body.span);
|
||||
|
||||
for const_ in &$($mutability)? body.required_consts {
|
||||
let location = START_BLOCK.start_location();
|
||||
self.visit_constant(const_, location);
|
||||
}
|
||||
super_body!(self, body, $($mutability, true)?);
|
||||
}
|
||||
|
||||
fn super_basic_block_data(&mut self,
|
||||
|
@ -982,12 +928,7 @@ macro_rules! make_mir_visitor {
|
|||
body: &$($mutability)? Body<'tcx>,
|
||||
location: Location
|
||||
) {
|
||||
#[allow(unused_macro_rules)]
|
||||
macro_rules! basic_blocks {
|
||||
(mut) => (body.basic_blocks_mut());
|
||||
() => (body.basic_blocks());
|
||||
}
|
||||
let basic_block = & $($mutability)? basic_blocks!($($mutability)?)[location.block];
|
||||
let basic_block = & $($mutability)? basic_blocks!(body, $($mutability, true)?)[location.block];
|
||||
if basic_block.statements.len() == location.statement_index {
|
||||
if let Some(ref $($mutability)? terminator) = basic_block.terminator {
|
||||
self.visit_terminator(terminator, location)
|
||||
|
@ -1002,6 +943,94 @@ macro_rules! make_mir_visitor {
|
|||
}
|
||||
}
|
||||
|
||||
macro_rules! basic_blocks {
|
||||
($body:ident, mut, true) => {
|
||||
$body.basic_blocks.as_mut()
|
||||
};
|
||||
($body:ident, mut, false) => {
|
||||
$body.basic_blocks.as_mut_preserves_cfg()
|
||||
};
|
||||
($body:ident,) => {
|
||||
$body.basic_blocks()
|
||||
};
|
||||
}
|
||||
|
||||
macro_rules! basic_blocks_iter {
|
||||
($body:ident, mut, $invalidate:tt) => {
|
||||
basic_blocks!($body, mut, $invalidate).iter_enumerated_mut()
|
||||
};
|
||||
($body:ident,) => {
|
||||
basic_blocks!($body,).iter_enumerated()
|
||||
};
|
||||
}
|
||||
|
||||
macro_rules! extra_body_methods {
|
||||
(mut) => {
|
||||
fn visit_body_preserves_cfg(&mut self, body: &mut Body<'tcx>) {
|
||||
self.super_body_preserves_cfg(body);
|
||||
}
|
||||
|
||||
fn super_body_preserves_cfg(&mut self, body: &mut Body<'tcx>) {
|
||||
super_body!(self, body, mut, false);
|
||||
}
|
||||
};
|
||||
() => {};
|
||||
}
|
||||
|
||||
macro_rules! super_body {
|
||||
($self:ident, $body:ident, $($mutability:ident, $invalidate:tt)?) => {
|
||||
let span = $body.span;
|
||||
if let Some(gen) = &$($mutability)? $body.generator {
|
||||
if let Some(yield_ty) = $(& $mutability)? gen.yield_ty {
|
||||
$self.visit_ty(
|
||||
yield_ty,
|
||||
TyContext::YieldTy(SourceInfo::outermost(span))
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
for (bb, data) in basic_blocks_iter!($body, $($mutability, $invalidate)?) {
|
||||
$self.visit_basic_block_data(bb, data);
|
||||
}
|
||||
|
||||
for scope in &$($mutability)? $body.source_scopes {
|
||||
$self.visit_source_scope_data(scope);
|
||||
}
|
||||
|
||||
$self.visit_ty(
|
||||
$(& $mutability)? $body.return_ty(),
|
||||
TyContext::ReturnTy(SourceInfo::outermost($body.span))
|
||||
);
|
||||
|
||||
for local in $body.local_decls.indices() {
|
||||
$self.visit_local_decl(local, & $($mutability)? $body.local_decls[local]);
|
||||
}
|
||||
|
||||
#[allow(unused_macro_rules)]
|
||||
macro_rules! type_annotations {
|
||||
(mut) => ($body.user_type_annotations.iter_enumerated_mut());
|
||||
() => ($body.user_type_annotations.iter_enumerated());
|
||||
}
|
||||
|
||||
for (index, annotation) in type_annotations!($($mutability)?) {
|
||||
$self.visit_user_type_annotation(
|
||||
index, annotation
|
||||
);
|
||||
}
|
||||
|
||||
for var_debug_info in &$($mutability)? $body.var_debug_info {
|
||||
$self.visit_var_debug_info(var_debug_info);
|
||||
}
|
||||
|
||||
$self.visit_span($(& $mutability)? $body.span);
|
||||
|
||||
for const_ in &$($mutability)? $body.required_consts {
|
||||
let location = START_BLOCK.start_location();
|
||||
$self.visit_constant(const_, location);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
macro_rules! visit_place_fns {
|
||||
(mut) => {
|
||||
fn tcx<'a>(&'a self) -> TyCtxt<'tcx>;
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue