waalge/aiken
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make sure clean up doesn't add case and constr to comp time eval. Also had fun with test_1 mint script context validator

Browse Source
This commit is contained in:
microproofs
2025-01-09 21:23:48 +07:00
parent 33392f1532
commit 2c214186b6
4 changed files with 47 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
crates/uplc/src/optimize.rs
View File

@@ -38,5 +38,5 @@ pub fn aiken_optimize_and_intern(program: Program<Name>) -> Program<Name> {
}
}
prog.clean_up()
prog.clean_up(true)
}

45
crates/uplc/src/optimize/shrinker.rs
View File

@@ -984,9 +984,6 @@ impl Term<Name> {
inline_lambda,
);
}
Term::Constr { .. } => todo!(),
Term::Case { .. } => todo!(),
other => other.traverse_uplc_with_helper(
scope,
arg_stack,
@@ -1012,8 +1009,40 @@ impl Term<Name> {
}
}
Term::Case { .. } => todo!(),
Term::Constr { .. } => todo!(),
Term::Case { constr, branches } => {
let constr = Rc::make_mut(constr);
constr.traverse_uplc_with_helper(
scope,
vec![],
id_gen,
with,
context,
inline_lambda,
);
for branch in branches {
branch.traverse_uplc_with_helper(
scope,
arg_stack.clone(),
id_gen,
with,
context,
inline_lambda,
);
}
}
Term::Constr { fields, .. } => {
for field in fields {
field.traverse_uplc_with_helper(
scope,
vec![],
id_gen,
with,
context,
inline_lambda,
);
}
}
Term::Builtin(func) => {
let mut args = vec![];
@@ -2051,10 +2080,12 @@ impl Program<Name> {
.0
}
pub fn clean_up(self) -> Self {
pub fn clean_up(self, case: bool) -> Self {
self.traverse_uplc_with(true, &mut |id, term, _arg_stack, scope, context| {
term.remove_no_inlines(id, vec![], scope, context);
term.case_constr_apply_reducer(id, vec![], scope, context);
if case {
term.case_constr_apply_reducer(id, vec![], scope, context);
}
})
.0
}