author Ian Lynagh Fri, 14 Mar 2008 19:30:37 +0000 (19:30 +0000) committer Ian Lynagh Fri, 14 Mar 2008 19:30:37 +0000 (19:30 +0000)
Patch from Russell O'Connor, trac proposal #974.

 Data/Either.hs patch | blob | history

module Data.Either (
Either(..),
-   either       -- :: (a -> c) -> (b -> c) -> Either a b -> c
+   either,           -- :: (a -> c) -> (b -> c) -> Either a b -> c
+   lefts,            -- :: [Either a b] -> [a]
+   rights,           -- :: [Either a b] -> [b]
+   partitionEithers, -- :: [Either a b] -> ([a],[b])
) where

+import Data.Tuple
+
import GHC.Base

+{-
+-- just for testing
+import Test.QuickCheck
+-}
+
{-|

The 'Either' type represents values with two possibilities: a value of
@@ -40,3 +50,36 @@ either                  :: (a -> c) -> (b -> c) -> Either a b -> c
either f _ (Left x)     =  f x
either _ g (Right y)    =  g y
+
+-- | Extracts from a list of 'Either' all the 'Left' elements
+-- All the 'Left' elements are extracted in order.
+
+lefts   :: [Either a b] -> [a]
+lefts x = [a | Left a <- x]
+
+-- | Extracts from a list of 'Either' all the 'Right' elements
+-- All the 'Right' elements are extracted in order.
+
+rights   :: [Either a b] -> [b]
+rights x = [a | Right a <- x]
+
+-- | Partitions a list of 'Either' into two lists
+-- All the 'Left' elements are extracted, in order, to the first
+-- component of the output.  Similarly the 'Right' elements are extracted
+-- to the second component of the output.
+
+partitionEithers :: [Either a b] -> ([a],[b])
+partitionEithers = foldr (either left right) ([],[])
+ where
+  left  a (l, r) = (a:l, r)
+  right a (l, r) = (l, a:r)
+
+{-
+{--------------------------------------------------------------------
+  Testing
+--------------------------------------------------------------------}
+prop_partitionEithers :: [Either Int Int] -> Bool
+prop_partitionEithers x =
+  partitionEithers x == (lefts x, rights x)
+-}
+