HigherOrderCO/Bend
1
1
Fork 0
mirror of https://github.com/HigherOrderCO/Bend.git synced 2024-08-16 07:10:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix pruning of native hvm defs

Browse Source
This commit is contained in:
Nicolas Abril 2024-06-15 19:31:33 +02:00
parent 8b5555c490
commit 9fc8a726af
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/fun/transform/definition_pruning.rs
View File

@ -49,8 +49,19 @@ impl Ctx<'_> {
}
}
fn rm_def(book: &mut Book, def_name: &Name) {
if book.defs.contains_key(def_name) {
book.defs.shift_remove(def_name);
} else if book.hvm_defs.contains_key(def_name) {
book.hvm_defs.shift_remove(def_name);
} else {
unreachable!()
}
}
// Remove unused definitions.
for def in self.book.defs.keys().cloned().collect::<Vec<_>>() {
let names = self.book.defs.keys().cloned().chain(self.book.hvm_defs.keys().cloned()).collect::<Vec<_>>();
for def in names {
if let Some(use_) = used.get(&def) {
match use_ {
Used::Main => {
@ -60,7 +71,7 @@ impl Ctx<'_> {
// Used by a non-builtin definition.
// Prune if `prune_all`, otherwise show a warning.
if prune_all {
self.book.defs.shift_remove(&def);
rm_def(self.book, &def);
} else {
self.info.add_rule_warning("Definition is unused.", WarningType::UnusedDefinition, def);
}
@ -69,7 +80,7 @@ impl Ctx<'_> {
// Unused, but a user-defined constructor.
// Prune if `prune_all`, otherwise nothing.
if prune_all {
self.book.defs.shift_remove(&def);
rm_def(self.book, &def);
} else {
// Don't show warning if it's a user-defined constructor.
}
@ -77,7 +88,7 @@ impl Ctx<'_> {
}
} else {
// Unused builtin, can always be pruned.
self.book.defs.shift_remove(&def);
rm_def(self.book, &def);
}
}
}