|
8 | 8 | // option. This file may not be copied, modified, or distributed
|
9 | 9 | // except according to those terms.
|
10 | 10 |
|
11 |
| -//use std::{iter, slice}; |
12 |
| -//use rustc_data_structures::indexed_vec::IndexVec; |
13 |
| -//use rustc_data_structures::graph_algorithms::{Graph, GraphPredecessors, GraphSuccessors}; |
14 |
| -//use mir::repr::{Mir, BasicBlock, START_BLOCK}; |
15 |
| -//use mir::cache::Cache; |
| 11 | +use std::{iter, slice}; |
| 12 | +use rustc_data_structures::indexed_vec::IndexVec; |
| 13 | +use rustc_data_structures::graph_algorithms::{Graph, GraphPredecessors, GraphSuccessors}; |
| 14 | +use mir::repr::{Mir, BasicBlock, START_BLOCK}; |
| 15 | +use mir::cache::Cache; |
16 | 16 |
|
17 |
| -//impl MirCfg { |
18 |
| -// pub fn new<'a, 'tcx>(mir: &Mir, cache: &Cache) -> Self { |
19 |
| -// MirCfg { |
20 |
| -// predecessors: cache.predecessors(mir).clone(), |
21 |
| -// successors: calculate_successors(mir), |
22 |
| -// n_nodes: mir.basic_blocks().len(), |
23 |
| -// start_node: START_BLOCK, |
24 |
| -// } |
25 |
| -// } |
26 |
| -//} |
27 |
| -// |
28 |
| -//#[derive(Clone, Debug)] |
29 |
| -//pub struct MirCfg { |
30 |
| -// predecessors: IndexVec<BasicBlock,Vec<BasicBlock>>, |
31 |
| -// successors: IndexVec<BasicBlock,Vec<BasicBlock>>, |
32 |
| -// start_node: BasicBlock, |
33 |
| -// n_nodes: usize, |
34 |
| -//} |
35 |
| -// |
36 |
| -//impl Graph for MirCfg { |
37 |
| -// |
38 |
| -// type Node = BasicBlock; |
39 |
| -// |
40 |
| -// fn num_nodes(&self) -> usize { self.n_nodes } |
41 |
| -// |
42 |
| -// fn start_node(&self) -> Self::Node { self.start_node } |
43 |
| -// |
44 |
| -// fn predecessors<'graph>(&'graph self, node: Self::Node) |
45 |
| -// -> <Self as GraphPredecessors<'graph>>::Iter |
46 |
| -// { |
47 |
| -// self.predecessors[node].iter().cloned() |
48 |
| -// } |
49 |
| -// fn successors<'graph>(&'graph self, node: Self::Node) |
50 |
| -// -> <Self as GraphSuccessors<'graph>>::Iter |
51 |
| -// { |
52 |
| -// self.successors[node].iter().cloned() |
53 |
| -// } |
54 |
| -//} |
55 |
| -// |
56 |
| -//impl<'g> GraphPredecessors<'g> for MirCfg { |
57 |
| -// type Item = BasicBlock; |
58 |
| -// type Iter = iter::Cloned<slice::Iter<'g, BasicBlock>>; |
59 |
| -//} |
60 |
| -// |
61 |
| -//impl<'g> GraphSuccessors<'g> for MirCfg { |
62 |
| -// type Item = BasicBlock; |
63 |
| -// type Iter = iter::Cloned<slice::Iter<'g, BasicBlock>>; |
64 |
| -//} |
65 |
| -// |
66 |
| -//fn calculate_successors<'a, 'tcx>(mir: &'a Mir<'tcx>) -> IndexVec<BasicBlock, Vec<BasicBlock>> { |
67 |
| -// let mut successors = IndexVec::from_elem(vec![], mir.basic_blocks()); |
68 |
| -// for (bb, data) in mir.basic_blocks().iter_enumerated() { |
69 |
| -// if let Some(ref term) = data.terminator { |
70 |
| -// successors[bb].append(term.successors().to_mut()); |
71 |
| -// } |
72 |
| -// } |
73 |
| -// for ss in successors.iter_mut() { |
74 |
| -// ss.sort(); |
75 |
| -// ss.dedup(); |
76 |
| -// } |
77 |
| -// successors |
78 |
| -//} |
| 17 | +impl MirCfg { |
| 18 | + pub fn new<'a, 'tcx>(mir: &Mir, cache: &Cache) -> Self { |
| 19 | + MirCfg { |
| 20 | + predecessors: cache.predecessors(mir).clone(), |
| 21 | + successors: calculate_successors(mir), |
| 22 | + n_nodes: mir.basic_blocks().len(), |
| 23 | + start_node: START_BLOCK, |
| 24 | + } |
| 25 | + } |
| 26 | +} |
| 27 | + |
| 28 | +#[derive(Clone, Debug)] |
| 29 | +pub struct MirCfg { |
| 30 | + predecessors: IndexVec<BasicBlock,Vec<BasicBlock>>, |
| 31 | + successors: IndexVec<BasicBlock,Vec<BasicBlock>>, |
| 32 | + start_node: BasicBlock, |
| 33 | + n_nodes: usize, |
| 34 | +} |
| 35 | + |
| 36 | +impl Graph for MirCfg { |
| 37 | + |
| 38 | + type Node = BasicBlock; |
| 39 | + |
| 40 | + fn num_nodes(&self) -> usize { self.n_nodes } |
| 41 | + |
| 42 | + fn start_node(&self) -> Self::Node { self.start_node } |
| 43 | + |
| 44 | + fn predecessors<'graph>(&'graph self, node: Self::Node) |
| 45 | + -> <Self as GraphPredecessors<'graph>>::Iter |
| 46 | + { |
| 47 | + self.predecessors[node].iter().cloned() |
| 48 | + } |
| 49 | + fn successors<'graph>(&'graph self, node: Self::Node) |
| 50 | + -> <Self as GraphSuccessors<'graph>>::Iter |
| 51 | + { |
| 52 | + self.successors[node].iter().cloned() |
| 53 | + } |
| 54 | +} |
| 55 | + |
| 56 | +impl<'g> GraphPredecessors<'g> for MirCfg { |
| 57 | + type Item = BasicBlock; |
| 58 | + type Iter = iter::Cloned<slice::Iter<'g, BasicBlock>>; |
| 59 | +} |
| 60 | + |
| 61 | +impl<'g> GraphSuccessors<'g> for MirCfg { |
| 62 | + type Item = BasicBlock; |
| 63 | + type Iter = iter::Cloned<slice::Iter<'g, BasicBlock>>; |
| 64 | +} |
| 65 | + |
| 66 | +fn calculate_successors<'a, 'tcx>(mir: &'a Mir<'tcx>) -> IndexVec<BasicBlock, Vec<BasicBlock>> { |
| 67 | + let mut successors = IndexVec::from_elem(vec![], mir.basic_blocks()); |
| 68 | + for (bb, data) in mir.basic_blocks().iter_enumerated() { |
| 69 | + if let Some(ref term) = data.terminator { |
| 70 | + successors[bb].append(term.successors().to_mut()); |
| 71 | + } |
| 72 | + } |
| 73 | + for ss in successors.iter_mut() { |
| 74 | + ss.sort(); |
| 75 | + ss.dedup(); |
| 76 | + } |
| 77 | + successors |
| 78 | +} |
0 commit comments