language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-30 23:31:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

turn count_targets_help into a loop

Browse source
This commit is contained in:
Folkert 2021-01-12 22:16:33 +01:00
parent c4e5af554b
commit 30023ac86b
1 changed files with 27 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

49
compiler/mono/src/decision_tree.rs
View file

@ -1631,32 +1631,37 @@ fn count_targets(decision_tree: &Decider<u64>) -> MutMap<u64, u64> {
result
}
fn count_targets_help(decision_tree: &Decider<u64>, targets: &mut MutMap<u64, u64>) {
fn count_targets_help(initial: &Decider<u64>, targets: &mut MutMap<u64, u64>) {
use Decider::*;
match decision_tree {
Leaf(target) => match targets.get_mut(target) {
None => {
targets.insert(*target, 1);
let mut stack = vec![initial];
while let Some(decision_tree) = stack.pop() {
match decision_tree {
Leaf(target) => match targets.get_mut(target) {
None => {
targets.insert(*target, 1);
}
Some(current) => {
*current += 1;
}
},
Chain {
success, failure, ..
} => {
stack.push(success);
stack.push(failure);
}
Some(current) => {
*current += 1;
}
},
Chain {
success, failure, ..
} => {
count_targets_help(success, targets);
count_targets_help(failure, targets);
}
FanOut {
tests, fallback, ..
} => {
stack.push(fallback);
FanOut {
tests, fallback, ..
} => {
count_targets_help(fallback, targets);
for (_, decider) in tests {
count_targets_help(decider, targets);
for (_, decider) in tests {
stack.push(decider);
}
}
}
}