ListPair.map2, based on code by escalier@riseup.net

2 files changed, 14 insertions, 0 deletions

diff --git a/lib/ur/listPair.ur b/lib/ur/listPair.ur
index 0182af19..d69993f9 100644
--- a/lib/ur/listPair.ur
+++ b/lib/ur/listPair.ur
@@ -33,3 +33,14 @@ fun all [a] [b] f =
     in
         all'
     end
+
+fun map2 [a] [b] [c] (f : a -> b -> c) =
+    let
+        fun map2' ls1 ls2 =
+            case (ls1, ls2) of
+                ([], []) => []
+              | (x1 :: ls1, x2 :: ls2) => f x1 x2 :: map2' ls1 ls2
+              | _ => error <xml>ListPair.map2: Unequal list lengths</xml>
+    in
+        map2'
+    end
diff --git a/lib/ur/listPair.urs b/lib/ur/listPair.urs
index f2287c13..218b9420 100644
--- a/lib/ur/listPair.urs
+++ b/lib/ur/listPair.urs
@@ -5,3 +5,6 @@ val mapX : a ::: Type -> b ::: Type -> ctx ::: {Unit}
            -> (a -> b -> xml ctx [] []) -> list a -> list b -> xml ctx [] []
 
 val all : a ::: Type -> b ::: Type -> (a -> b -> bool) -> list a -> list b -> bool
+
+val map2 : a ::: Type -> b ::: Type -> c ::: Type
+           -> (a -> b -> c) -> list a -> list b -> list c