summaryrefslogtreecommitdiff
path: root/dhall/src
diff options
context:
space:
mode:
authorBasile Henry2020-11-01 23:33:38 +0100
committerBasile Henry2020-11-01 23:33:38 +0100
commit7066e625fc945e66d8af91c633e762fe5f7b17f5 (patch)
treec661bf84414d299c995222add6815a400fd78866 /dhall/src
parentaaac028d21b02827a0c79b34728fb104e5649149 (diff)
Handle partially abstract with
Diffstat (limited to '')
-rw-r--r--dhall/src/operations/normalization.rs37
1 files changed, 29 insertions, 8 deletions
diff --git a/dhall/src/operations/normalization.rs b/dhall/src/operations/normalization.rs
index 908ed93..f147f85 100644
--- a/dhall/src/operations/normalization.rs
+++ b/dhall/src/operations/normalization.rs
@@ -306,16 +306,24 @@ pub fn normalize_operation(opkind: &OpKind<Nir>) -> Ret {
let mut visited: Vec<(&Label, HashMap<Label, Nir>)> = Vec::new();
let mut to_create = Vec::new();
+ // To be used when an abstract entry is reached
+ let mut abstract_nir = None;
+
for label in labels {
match current_nk {
- None => to_create.push(label),
+ None => to_create.push(label.clone()),
Some(nk) => match nk {
RecordLit(kvs) => {
current_nk =
kvs.get(label).map(|nir| nir.kind().clone());
visited.push((label, kvs));
}
- _ => return nothing_to_do(),
+ // Handle partially abstract case
+ nir => {
+ abstract_nir = Some(Nir::from_kind(nir));
+ to_create.push(label.clone());
+ current_nk = None;
+ }
},
}
}
@@ -323,12 +331,25 @@ pub fn normalize_operation(opkind: &OpKind<Nir>) -> Ret {
// Create Nir for record bottom up
let mut nir = expr.clone();
- while let Some(label) = to_create.pop() {
- let rec = RecordLit(FromIterator::from_iter(once((
- label.clone(),
- nir,
- ))));
- nir = Nir::from_kind(rec);
+ match abstract_nir {
+ // No abstract nir, creating singleton records
+ None => {
+ while let Some(label) = to_create.pop() {
+ let rec = RecordLit(FromIterator::from_iter(once((
+ label.clone(),
+ nir,
+ ))));
+ nir = Nir::from_kind(rec);
+ }
+ }
+ // Abstract nir, creating with op
+ Some(abstract_nir) => {
+ nir = Nir::from_kind(Op(OpKind::With(
+ abstract_nir,
+ to_create,
+ nir,
+ )));
+ }
}
// Update visited records