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

fix some unnecessary lambdas in assign

This commit is contained in:
microproofs 2023-07-28 12:39:20 -04:00 committed by Kasey
parent 03dd13dc7d
commit f03ed41e03
2 changed files with 16 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
crates/aiken-lang/src/gen_uplc.rs
View File

@ -687,22 +687,21 @@ impl<'a> CodeGenerator<'a> {
let val = AirTree::local_var(name, tipo.clone());
if tipo.is_primitive() {
AirTree::let_assignment(name, assignment.hoist_over(val))
if non_opaque_tipo.is_primitive() {
assignment
} else {
let expect = self.expect_type_assign(
&non_opaque_tipo,
val.clone(),
val,
&mut index_map,
pattern.location(),
);
let assign_expect = AirTree::let_assignment("_", expect);
AirTree::let_assignment(
name,
assignment.hoist_over(assign_expect.hoist_over(val)),
)
let sequence = vec![assignment, assign_expect];
AirTree::UnhoistedSequence(sequence)
}
} else {
AirTree::let_assignment(name, value)
@ -721,28 +720,27 @@ impl<'a> CodeGenerator<'a> {
let name = &format!("__discard_expect_{}", name);
let mut index_map = IndexMap::new();
let tipo = convert_opaque_type(tipo, &self.data_types);
let non_opaque_tipo = convert_opaque_type(tipo, &self.data_types);
let assignment = AirTree::let_assignment(name, value);
let val = AirTree::local_var(name, tipo.clone());
if tipo.is_primitive() {
AirTree::let_assignment(name, assignment.hoist_over(val))
if non_opaque_tipo.is_primitive() {
assignment.hoist_over(val)
} else {
let expect = self.expect_type_assign(
&tipo,
val.clone(),
&non_opaque_tipo,
val,
&mut index_map,
pattern.location(),
);
let assign_expect = AirTree::let_assignment("_", expect);
AirTree::let_assignment(
name,
assignment.hoist_over(assign_expect.hoist_over(val)),
)
let sequence = vec![assignment, assign_expect];
AirTree::UnhoistedSequence(sequence)
}
} else if !props.remove_unused {
AirTree::let_assignment(name, value)

7
crates/aiken-project/src/tests/gen_uplc.rs
View File

@ -73,7 +73,6 @@ fn assert_uplc(source_code: &str, expected: Term<Name>, should_fail: bool) {
};
let expected = optimize::aiken_optimize_and_intern(expected);
// println!("expected: {}", expected.to_pretty());
let expected: Program<DeBruijn> = expected.try_into().unwrap();
@ -103,7 +102,6 @@ fn assert_uplc(source_code: &str, expected: Term<Name>, should_fail: bool) {
term: expected,
};
println!("expected: {}", expected.to_pretty());
let expected = optimize::aiken_optimize_and_intern(expected);
@ -4212,10 +4210,7 @@ fn expect_head_cast_data_with_tail() {
)
.apply(Term::var("tail_2"))
.apply(
Term::unit()
.lambda("_")
.apply(Term::un_i_data().apply(Term::var("list_item")))
.lambda("list_item"),
Term::un_i_data().apply(Term::var("list_item")).lambda("list_item")
),
)
.lambda("tail_2")