73 lines
2.8 KiB
Rust
73 lines
2.8 KiB
Rust
// Copyright 2016 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
//! A pass that simplifies branches when their condition is known.
|
|
|
|
use rustc::ty::{TyCtxt, ParamEnv};
|
|
use rustc::mir::*;
|
|
use transform::{MirPass, MirSource};
|
|
|
|
use std::borrow::Cow;
|
|
|
|
pub struct SimplifyBranches { label: String }
|
|
|
|
impl SimplifyBranches {
|
|
pub fn new(label: &str) -> Self {
|
|
SimplifyBranches { label: format!("SimplifyBranches-{}", label) }
|
|
}
|
|
}
|
|
|
|
impl MirPass for SimplifyBranches {
|
|
fn name<'a>(&'a self) -> Cow<'a, str> {
|
|
Cow::Borrowed(&self.label)
|
|
}
|
|
|
|
fn run_pass<'a, 'tcx>(&self,
|
|
tcx: TyCtxt<'a, 'tcx, 'tcx>,
|
|
_src: MirSource,
|
|
mir: &mut Mir<'tcx>) {
|
|
for block in mir.basic_blocks_mut() {
|
|
let terminator = block.terminator_mut();
|
|
terminator.kind = match terminator.kind {
|
|
TerminatorKind::SwitchInt {
|
|
discr: Operand::Constant(ref c), switch_ty, ref values, ref targets, ..
|
|
} => {
|
|
let switch_ty = ParamEnv::empty().and(switch_ty);
|
|
if let Some(constint) = c.literal.assert_bits(tcx, switch_ty) {
|
|
let (otherwise, targets) = targets.split_last().unwrap();
|
|
let mut ret = TerminatorKind::Goto { target: *otherwise };
|
|
for (&v, t) in values.iter().zip(targets.iter()) {
|
|
if v == constint {
|
|
ret = TerminatorKind::Goto { target: *t };
|
|
break;
|
|
}
|
|
}
|
|
ret
|
|
} else {
|
|
continue
|
|
}
|
|
},
|
|
TerminatorKind::Assert {
|
|
target, cond: Operand::Constant(ref c), expected, ..
|
|
} if (c.literal.assert_bool(tcx) == Some(true)) == expected => {
|
|
TerminatorKind::Goto { target }
|
|
},
|
|
TerminatorKind::FalseEdges { real_target, .. } => {
|
|
TerminatorKind::Goto { target: real_target }
|
|
},
|
|
TerminatorKind::FalseUnwind { real_target, .. } => {
|
|
TerminatorKind::Goto { target: real_target }
|
|
},
|
|
_ => continue
|
|
};
|
|
}
|
|
}
|
|
}
|