Add composition of plutarch functions

Plutonomicon · Jan 18, 2022 · b5977fe · b5977fe
1 parent 231267e
commit b5977fe
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/Plutarch.hs b/Plutarch.hs
@@ -29,6 +29,7 @@ module Plutarch (
   printScript,
   (#$),
   (#),
+  (#.),
   pinl,
   PCon (..),
   PMatch (..),
@@ -85,6 +86,17 @@ infixl 8 #
 
 infixr 0 #$
 
+{- |
+  Composition of Plutarch level functions
+
+  >>> f #. g #$ x
+  f (g x)
+-}
+(#.) :: Term s (b :--> c) -> Term s (a :--> b) -> Term s (a :--> c)
+(#.) bc ab = plam $ \a -> bc #$ ab # a
+
+infixr 7 #.
+
 {- $plam
  Lambda abstraction.
 
@@ -234,3 +246,4 @@ pfix = phoistAcyclic $
     plam $ \f ->
       (plam $ \(x :: Term s POpaque) -> f # (plam $ \(v :: Term s POpaque) -> (punsafeCoerce x) # x # v))
         # punsafeCoerce (plam $ \(x :: Term s POpaque) -> f # (plam $ \(v :: Term s POpaque) -> (punsafeCoerce x) # x # v))
+