| 279 | } |
| 280 | |
| 281 | fn assign_loop_levels(&mut self) { |
| 282 | let mut stack: SmallVec<[Loop; 8]> = SmallVec::new(); |
| 283 | for lp in self.loops.keys() { |
| 284 | if self.loops[lp].level == LoopLevel::invalid() { |
| 285 | stack.push(lp); |
| 286 | while let Some(&lp) = stack.last() { |
| 287 | if let Some(parent) = self.loops[lp].parent.into() { |
| 288 | if self.loops[parent].level != LoopLevel::invalid() { |
| 289 | self.loops[lp].level = self.loops[parent].level.inc(); |
| 290 | stack.pop(); |
| 291 | } else { |
| 292 | stack.push(parent); |
| 293 | } |
| 294 | } else { |
| 295 | self.loops[lp].level = LoopLevel::root().inc(); |
| 296 | stack.pop(); |
| 297 | } |
| 298 | } |
| 299 | } |
| 300 | } |
| 301 | } |
| 302 | } |
| 303 | |
| 304 | #[cfg(test)] |