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

fix up code gen tests

This commit is contained in:
microproofs 2024-02-03 20:09:34 -05:00 committed by Kasey
parent 4ab3b61200
commit 6b97ab71fe
2 changed files with 303 additions and 309 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -994,9 +994,7 @@ impl<'a> CodeGenerator<'a> {
let val = AirTree::local_var(&tail_name, tipo.clone()); let val = AirTree::local_var(&tail_name, tipo.clone());
elems.push(tail_name.clone()); let then = if tail_name != "_" {
if tail_name != "_" {
self.assignment( self.assignment(
tail, tail,
val, val,
@ -1012,15 +1010,18 @@ impl<'a> CodeGenerator<'a> {
) )
} else { } else {
then then
} };
elems.push(tail_name);
then
} }
}; };
let then = elements let then = elements
.iter() .iter()
.enumerate() .enumerate()
.rev() .rfold(then, |then, (index, elem)| {
.fold(then, |then, (index, elem)| {
let elem_name = match elem { let elem_name = match elem {
Pattern::Var { name, .. } => name.to_string(), Pattern::Var { name, .. } => name.to_string(),
Pattern::Assign { name, .. } => name.to_string(), Pattern::Assign { name, .. } => name.to_string(),
@ -1041,9 +1042,7 @@ impl<'a> CodeGenerator<'a> {
let val = AirTree::local_var(&elem_name, list_elem_type.clone()); let val = AirTree::local_var(&elem_name, list_elem_type.clone());
elems.push(elem_name.clone()); let then = if elem_name != "_" {
if elem_name != "_" {
self.assignment( self.assignment(
elem, elem,
val, val,
@ -1059,16 +1058,20 @@ impl<'a> CodeGenerator<'a> {
) )
} else { } else {
then then
} };
elems.push(elem_name);
then
}); });
let names = elems.iter().rev().cloned().collect(); elems.reverse();
let list_access = if elements.is_empty() { let list_access = if elements.is_empty() {
AirTree::list_empty(value, props.msg_func) AirTree::list_empty(value, props.msg_func)
} else { } else {
AirTree::list_access( AirTree::list_access(
names, elems,
tipo.clone(), tipo.clone(),
tail.is_some(), tail.is_some(),
value, value,
@ -1115,12 +1118,10 @@ impl<'a> CodeGenerator<'a> {
let mut fields = vec![]; let mut fields = vec![];
let then = let then = arguments
arguments
.iter() .iter()
.enumerate() .enumerate()
.rev() .rfold(then, |then, (index, arg)| {
.fold(then, |then, (index, arg)| {
let label = arg.label.clone().unwrap_or_default(); let label = arg.label.clone().unwrap_or_default();
let field_index = if let Some(field_map) = &field_map { let field_index = if let Some(field_map) = &field_map {
@ -1156,13 +1157,7 @@ impl<'a> CodeGenerator<'a> {
let val = AirTree::local_var(&field_name, arg_type.clone()); let val = AirTree::local_var(&field_name, arg_type.clone());
fields.push(( let then = if field_name != "_" {
field_index,
field_name.to_string(),
arg_type.clone(),
));
if field_name != "_" {
self.assignment( self.assignment(
&arg.value, &arg.value,
val, val,
@ -1178,14 +1173,14 @@ impl<'a> CodeGenerator<'a> {
) )
} else { } else {
then then
} };
fields.push((field_index, field_name, arg_type.clone()));
then
}); });
let indices = fields fields.reverse();
.iter()
.rev()
.map(|(index, name, tipo)| (*index, name.to_string(), tipo.clone()))
.collect_vec();
// This `value` is either value param that was passed in or // This `value` is either value param that was passed in or
// local var // local var
@ -1199,15 +1194,14 @@ impl<'a> CodeGenerator<'a> {
let local_value = AirTree::local_var(&constructor_name, tipo.clone()); let local_value = AirTree::local_var(&constructor_name, tipo.clone());
let then = if check_replaceable_opaque_type(tipo, &self.data_types) { let then = if check_replaceable_opaque_type(tipo, &self.data_types) {
AirTree::let_assignment(&indices[0].1, local_value, then) AirTree::let_assignment(&fields[0].1, local_value, then)
} else { } else {
let (is_expect, msg) = if props.full_check { let (is_expect, msg) = if props.full_check {
(true, props.msg_func.clone()) (true, props.msg_func.clone())
} else { } else {
(false, None) (false, None)
}; };
AirTree::fields_expose(indices, local_value, msg, is_expect) AirTree::fields_expose(fields, local_value, msg, is_expect).hoist_over(then)
.hoist_over(then)
}; };
// TODO: See if we can combine these two if-conditions; // TODO: See if we can combine these two if-conditions;
@ -1260,11 +1254,7 @@ impl<'a> CodeGenerator<'a> {
let mut fields = vec![]; let mut fields = vec![];
let then = elems let then = elems.iter().enumerate().rfold(then, |then, (index, arg)| {
.iter()
.enumerate()
.rev()
.fold(then, |then, (index, arg)| {
let tuple_name = match &arg { let tuple_name = match &arg {
Pattern::Var { name, .. } => name.to_string(), Pattern::Var { name, .. } => name.to_string(),
Pattern::Assign { name, .. } => name.to_string(), Pattern::Assign { name, .. } => name.to_string(),
@ -1292,9 +1282,7 @@ impl<'a> CodeGenerator<'a> {
let val = AirTree::local_var(&tuple_name, arg_type.clone()); let val = AirTree::local_var(&tuple_name, arg_type.clone());
fields.push(tuple_name.clone()); let then = if "_" != tuple_name {
if "_" != tuple_name {
self.assignment( self.assignment(
arg, arg,
val, val,
@ -1310,9 +1298,15 @@ impl<'a> CodeGenerator<'a> {
) )
} else { } else {
then then
} };
fields.push(tuple_name);
then
}); });
fields.reverse();
let (is_expect, msg) = if props.full_check { let (is_expect, msg) = if props.full_check {
(true, props.msg_func) (true, props.msg_func)
} else { } else {
@ -1543,7 +1537,7 @@ impl<'a> CodeGenerator<'a> {
let mut tuple_expect_items = vec![]; let mut tuple_expect_items = vec![];
let then = tuple_inner_types.iter().enumerate().rev().fold( let then = tuple_inner_types.iter().enumerate().rfold(
AirTree::void(), AirTree::void(),
|then, (index, arg)| { |then, (index, arg)| {
let tuple_index_name = format!( let tuple_index_name = format!(
@ -1565,19 +1559,19 @@ impl<'a> CodeGenerator<'a> {
}, },
); );
let tuple_index_names = tuple_expect_items.iter().rev().cloned().collect_vec(); tuple_expect_items.reverse();
let tuple_access = AirTree::tuple_access( let tuple_access = AirTree::tuple_access(
tuple_index_names, tuple_expect_items,
tipo.clone(), tipo.clone(),
AirTree::local_var(&tuple_name, tipo.clone()), AirTree::local_var(&tuple_name, tipo.clone()),
msg_func, msg_func,
true, true,
); );
let tupl_access = tuple_access.hoist_over(then); let tuple_access = tuple_access.hoist_over(then);
AirTree::let_assignment(&tuple_name, value, tupl_access) AirTree::let_assignment(&tuple_name, value, tuple_access)
// Constructor // Constructor
} else { } else {
@ -1634,7 +1628,7 @@ impl<'a> CodeGenerator<'a> {
|acc, (index, constr)| { |acc, (index, constr)| {
let mut constr_args = vec![]; let mut constr_args = vec![];
let constr_then = constr.arguments.iter().enumerate().rev().fold( let constr_then = constr.arguments.iter().enumerate().rfold(
AirTree::void(), AirTree::void(),
|then, (index, arg)| { |then, (index, arg)| {
let arg_name = let arg_name =
@ -1647,8 +1641,8 @@ impl<'a> CodeGenerator<'a> {
AirTree::let_assignment( AirTree::let_assignment(
"_", "_",
self.expect_type_assign( self.expect_type_assign(
&arg_tipo, &arg_tipo.clone(),
AirTree::local_var(&arg_name, arg_tipo.clone()), AirTree::local_var(arg_name, arg_tipo),
defined_data_types, defined_data_types,
location, location,
msg_term.clone(), msg_term.clone(),
@ -1657,14 +1651,7 @@ impl<'a> CodeGenerator<'a> {
) )
}, },
); );
constr_args.reverse();
let indices = constr_args
.iter()
.rev()
.map(|(index, arg_name, arg_tipo)| {
(*index, arg_name.clone(), (*arg_tipo).clone())
})
.collect_vec();
let assign = if constr_args.is_empty() { let assign = if constr_args.is_empty() {
AirTree::fields_empty( AirTree::fields_empty(
@ -1679,7 +1666,7 @@ impl<'a> CodeGenerator<'a> {
) )
} else { } else {
AirTree::fields_expose( AirTree::fields_expose(
indices, constr_args,
AirTree::local_var( AirTree::local_var(
format!( format!(
"__constr_var_span_{}_{}", "__constr_var_span_{}_{}",
@ -2185,52 +2172,9 @@ impl<'a> CodeGenerator<'a> {
let mut elems = vec![]; let mut elems = vec![];
let elems_then = elements
.iter()
.enumerate()
.rfold(then, |then, (index, elem)| {
// TODO: Turn 'Pattern' into another type instead of using strings and
// expecting a special magic string '_'.
let elem_name = match elem {
Pattern::Var { name, .. } => name.to_string(),
Pattern::Assign { name, .. } => name.to_string(),
Pattern::Discard { .. } => "_".to_string(),
_ => format!(
"elem_{}_span_{}_{}",
index,
elem.location().start,
elem.location().end
),
};
let mut elem_props = ClauseProperties::init_inner(
list_elem_type,
elem_name.clone(),
elem_name.clone(),
props.final_clause,
);
let then = if elem_name != "_" {
self.nested_clause_condition(
elem,
list_elem_type,
&mut elem_props,
then,
)
} else {
then
};
elems.push(elem_name);
*complex_clause = *complex_clause || elem_props.complex_clause;
then
});
let mut list_tail = None; let mut list_tail = None;
let tail_then = tail.iter().rev().fold(elems_then, |then, elem| { let elems_then = tail.iter().rfold(then, |inner_then, elem| {
assert!(!elements.is_empty()); assert!(!elements.is_empty());
let tail = defined_tails.get(elements.len() - 1); let tail = defined_tails.get(elements.len() - 1);
let elem_name = match elem.as_ref() { let elem_name = match elem.as_ref() {
@ -2255,10 +2199,15 @@ impl<'a> CodeGenerator<'a> {
list_tail = Some((tail.unwrap().to_string(), elem_name.to_string())); list_tail = Some((tail.unwrap().to_string(), elem_name.to_string()));
} }
let then = if elem_name != "_" { let inner_then = if elem_name != "_" {
self.nested_clause_condition(elem, subject_tipo, &mut elem_props, then) self.nested_clause_condition(
elem,
subject_tipo,
&mut elem_props,
inner_then,
)
} else { } else {
then inner_then
}; };
if props.final_clause && defined_tails.is_empty() { if props.final_clause && defined_tails.is_empty() {
@ -2267,14 +2216,61 @@ impl<'a> CodeGenerator<'a> {
*complex_clause = *complex_clause || elem_props.complex_clause; *complex_clause = *complex_clause || elem_props.complex_clause;
then inner_then
}); });
let defined_heads = elems.iter().rev().cloned().collect_vec(); let elems_then =
elements
.iter()
.enumerate()
.rfold(elems_then, |elems_then, (index, elem)| {
// TODO: Turn 'Pattern' into another type instead of using strings and
// expecting a special magic string '_'.
let elem_name = match elem {
Pattern::Var { name, .. } => name.to_string(),
Pattern::Assign { name, .. } => name.to_string(),
Pattern::Discard { .. } => "_".to_string(),
_ => format!(
"elem_{}_span_{}_{}",
index,
elem.location().start,
elem.location().end
),
};
let mut elem_props = ClauseProperties::init_inner(
list_elem_type,
elem_name.clone(),
elem_name.clone(),
props.final_clause,
);
let elems_then = if elem_name != "_" {
self.nested_clause_condition(
elem,
list_elem_type,
&mut elem_props,
elems_then,
)
} else {
elems_then
};
elems.push(elem_name);
*complex_clause = *complex_clause || elem_props.complex_clause;
elems_then
});
elems.reverse();
// This case is really only possible with something like
// when_tuple_empty_lists
let list_assign = if props.final_clause && defined_tails.is_empty() { let list_assign = if props.final_clause && defined_tails.is_empty() {
AirTree::list_access( AirTree::list_access(
defined_heads, elems,
subject_tipo.clone(), subject_tipo.clone(),
tail.is_some(), tail.is_some(),
AirTree::local_var(&props.original_subject_name, subject_tipo.clone()), AirTree::local_var(&props.original_subject_name, subject_tipo.clone()),
@ -2285,10 +2281,10 @@ impl<'a> CodeGenerator<'a> {
ExpectLevel::None, ExpectLevel::None,
) )
} else { } else {
assert!(defined_tails.len() >= defined_heads.len()); assert!(defined_tails.len() >= elems.len());
AirTree::list_expose( AirTree::list_expose(
defined_heads elems
.into_iter() .into_iter()
.zip(defined_tails) .zip(defined_tails)
.filter(|(head, _)| head != "_") .filter(|(head, _)| head != "_")
@ -2299,7 +2295,7 @@ impl<'a> CodeGenerator<'a> {
) )
}; };
(AirTree::void(), list_assign.hoist_over(tail_then)) (AirTree::void(), list_assign.hoist_over(elems_then))
} }
Pattern::Constructor { Pattern::Constructor {
name, name,
@ -2309,7 +2305,7 @@ impl<'a> CodeGenerator<'a> {
.. ..
} => { } => {
if subject_tipo.is_bool() { if subject_tipo.is_bool() {
(AirTree::bool(name == "True"), AirTree::no_op()) (AirTree::bool(name == "True"), then)
} else { } else {
assert!( assert!(
matches!(function_tipo.as_ref().clone(), Type::Fn { .. }) matches!(function_tipo.as_ref().clone(), Type::Fn { .. })
@ -2345,9 +2341,11 @@ impl<'a> CodeGenerator<'a> {
let mut fields = vec![]; let mut fields = vec![];
let next_then = arguments.iter().enumerate().rev().fold( let next_then =
then, arguments
|inner_then, (index, arg)| { .iter()
.enumerate()
.rfold(then, |inner_then, (index, arg)| {
let label = arg.label.clone().unwrap_or_default(); let label = arg.label.clone().unwrap_or_default();
let field_index = if let Some(field_map) = &field_map { let field_index = if let Some(field_map) = &field_map {
@ -2399,8 +2397,7 @@ impl<'a> CodeGenerator<'a> {
fields.push((field_index, field_name, arg_type.clone())); fields.push((field_index, field_name, arg_type.clone()));
statement statement
}, });
);
fields.reverse(); fields.reverse();
@ -2441,8 +2438,7 @@ impl<'a> CodeGenerator<'a> {
elems elems
.iter() .iter()
.enumerate() .enumerate()
.rev() .rfold(then, |inner_then, (index, element)| {
.fold(then, |inner_then, (index, element)| {
let elem_name = match element { let elem_name = match element {
Pattern::Var { name, .. } => name.to_string(), Pattern::Var { name, .. } => name.to_string(),
Pattern::Assign { name, .. } => name.to_string(), Pattern::Assign { name, .. } => name.to_string(),
@ -2510,7 +2506,7 @@ impl<'a> CodeGenerator<'a> {
} }
}); });
let tuple_name_assigns = previous_defined_names.into_iter().rev().fold( let tuple_name_assigns = previous_defined_names.into_iter().rfold(
next_then, next_then,
|inner_then, (index, prev_name, name)| { |inner_then, (index, prev_name, name)| {
AirTree::let_assignment( AirTree::let_assignment(
@ -2746,12 +2742,10 @@ impl<'a> CodeGenerator<'a> {
) -> AirTree { ) -> AirTree {
let mut arg_names = vec![]; let mut arg_names = vec![];
let checked_args = let checked_args = arguments
arguments
.iter() .iter()
.enumerate() .enumerate()
.rev() .rfold(body, |inner_then, (index, arg)| {
.fold(body, |inner_then, (index, arg)| {
let arg_name = arg.arg_name.get_variable_name().unwrap_or("_").to_string(); let arg_name = arg.arg_name.get_variable_name().unwrap_or("_").to_string();
let arg_span = arg.location; let arg_span = arg.location;

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

@ -5805,9 +5805,9 @@ fn opaque_value_in_datum() {
.lambda("ctx") .lambda("ctx")
.lambda("red") .lambda("red")
.lambda("dat") .lambda("dat")
.constr_fields_exposer()
.lambda("expect[(_,amount)]=val.inner.inner") .lambda("expect[(_,amount)]=val.inner.inner")
.apply(Term::string("expect [(_, amount)] = val.inner.inner")) .apply(Term::string("expect [(_, amount)] = val.inner.inner"))
.constr_fields_exposer()
.lambda("dat:Dat") .lambda("dat:Dat")
.apply(Term::string("dat: Dat")) .apply(Term::string("dat: Dat"))
.constr_index_exposer(), .constr_index_exposer(),