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

feat implement tuple index

This commit is contained in:
Kasey White 2022-12-30 22:03:32 -05:00 committed by Kasey
parent 3514e66234
commit e495eefb34
2 changed files with 51 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
crates/aiken-lang/src/air.rs
View File

@ -196,8 +196,7 @@ pub enum Air {
TupleIndex {
scope: Vec<u64>,
tipo: Arc<Type>,
index: u64,
name: String,
index: usize,
},
Todo {

53
crates/aiken-lang/src/uplc.rs
View File

@ -525,8 +525,15 @@ impl<'a> CodeGenerator<'a> {
self.build_ir(then, ir_stack, scope);
}
TypedExpr::TupleIndex { .. } => {
todo!("Tuple indexing not implementing yet");
TypedExpr::TupleIndex {
tipo, index, tuple, ..
} => {
ir_stack.push(Air::TupleIndex {
scope: scope.clone(),
tipo: tuple.tipo(),
index: *index,
});
self.build_ir(tuple, ir_stack, scope);
}
TypedExpr::ErrorTerm { tipo, label, .. } => {
@ -3962,7 +3969,47 @@ impl<'a> CodeGenerator<'a> {
arg_stack.push(term);
}
Air::TupleIndex { .. } => todo!(),
Air::TupleIndex { tipo, index, .. } => {
let mut term = arg_stack.pop().unwrap();
if matches!(tipo.get_uplc_type(), UplcType::Pair(_, _)) {
if index == 0 {
term = convert_data_to_type(
apply_wrap(
Term::Builtin(DefaultFunction::FstPair)
.force_wrap()
.force_wrap(),
term,
),
&tipo.get_inner_types()[0],
);
} else {
term = convert_data_to_type(
apply_wrap(
Term::Builtin(DefaultFunction::SndPair)
.force_wrap()
.force_wrap(),
term,
),
&tipo.get_inner_types()[1],
);
}
} else {
self.needs_field_access = true;
term = apply_wrap(
apply_wrap(
Term::Var(Name {
text: CONSTR_GET_FIELD.to_string(),
unique: 0.into(),
}),
term,
),
Term::Constant(UplcConstant::Integer(index as i128)),
);
}
arg_stack.push(term);
}
Air::TupleAccessor { tipo, names, .. } => {
let inner_types = tipo.get_inner_types();
let value = arg_stack.pop().unwrap();