summaryrefslogtreecommitdiff
path: root/dhall/src/semantics/nze
diff options
context:
space:
mode:
authorBasile Henry2020-03-02 10:24:18 +0100
committerBasile Henry2020-03-02 10:31:10 +0100
commitd5874d9dcedc15eaccb942cc8f45f26b2335ed2d (patch)
tree906e09e2686356c91a3a2ea4c373ad99827c7c13 /dhall/src/semantics/nze
parent1a702c9c8e6addd481fff49f970cc5c2ef7b5d56 (diff)
Normalization for ProjectionByExpr
Diffstat (limited to 'dhall/src/semantics/nze')
-rw-r--r--dhall/src/semantics/nze/normalize.rs37
1 files changed, 34 insertions, 3 deletions
diff --git a/dhall/src/semantics/nze/normalize.rs b/dhall/src/semantics/nze/normalize.rs
index 604db8f..dedd659 100644
--- a/dhall/src/semantics/nze/normalize.rs
+++ b/dhall/src/semantics/nze/normalize.rs
@@ -372,11 +372,42 @@ pub(crate) fn normalize_one_layer(expr: ExprKind<Nir>, env: &NzEnv) -> NirKind {
},
_ => Ret::Expr(expr),
},
+ PartialExpr(ExprKind::Projection(v2, _)) => {
+ return normalize_one_layer(
+ ExprKind::Field(v2.clone(), l.clone()),
+ env,
+ )
+ }
_ => Ret::Expr(expr),
},
- ExprKind::ProjectionByExpr(_, _) => {
- unimplemented!("selection by expression")
- }
+
+ ExprKind::ProjectionByExpr(ref v, ref t) => match dbg!(v).kind() {
+ RecordLit(kvs) => match dbg!(t).kind() {
+ RecordType(kts) => Ret::NirKind(RecordLit(
+ kts.iter()
+ .filter_map(|(l, _)| {
+ kvs.get(l).map(|x| (l.clone(), x.clone()))
+ })
+ .collect(),
+ )),
+ _ => Ret::Expr(expr),
+ },
+ _ => match dbg!(t).kind() {
+ RecordType(kts) => {
+ use crate::syntax::map::DupTreeSet;
+ use std::iter::FromIterator;
+
+ let ts = DupTreeSet::from_iter(
+ kts.iter().map(|(l, _)| l.clone()),
+ );
+ return normalize_one_layer(
+ ExprKind::Projection(v.clone(), ts),
+ env,
+ );
+ }
+ _ => Ret::Expr(expr),
+ },
+ },
ExprKind::Merge(ref handlers, ref variant, _) => {
match handlers.kind() {