Add branchTerm function

ollef · May 25, 2024 · 94475bf · 94475bf
1 parent fbbac10
commit 94475bf
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/Low/Syntax.hs b/src/Low/Syntax.hs
@@ -1,5 +1,6 @@
 {-# LANGUAGE DeriveAnyClass #-}
 {-# LANGUAGE DeriveGeneric #-}
+{-# LANGUAGE LambdaCase #-}
 
 module Low.Syntax where
 
@@ -41,6 +42,11 @@ data Branch v
   | LiteralBranch !Literal !(Term v)
   deriving (Eq, Show, Generic, Hashable)
 
+branchTerm :: Branch v -> Term v
+branchTerm = \case
+  ConstructorBranch _ t -> t
+  LiteralBranch _ t -> t
+
 data Function v
   = Body !PassBy !(Term v)
   | Parameter !Name !PassBy !(Scope Function v)