232 lines
6 KiB
Elm
232 lines
6 KiB
Elm
module List where
|
|
|
|
{-| A library for manipulating lists of values. Every value in a
|
|
list must have the same type.
|
|
|
|
# Basics
|
|
@docs (::), (++), length, reverse, map
|
|
|
|
# Sub-lists
|
|
@docs head, tail, last, filter, take, drop
|
|
|
|
# Putting Lists Together
|
|
@docs concat, concatMap, join, intersperse, zip, zipWith
|
|
|
|
# Taking Lists Apart
|
|
@docs split, partition, unzip
|
|
|
|
# Folds
|
|
@docs foldr, foldl, foldr1, foldl1, scanl, scanl1
|
|
|
|
# Special Folds
|
|
@docs sum, product, maximum, minimum, all, any, and, or
|
|
-}
|
|
|
|
import open Basics
|
|
import Native.List
|
|
|
|
{-| Add an element to the front of a list `(1 :: [2,3] == [1,2,3])` -}
|
|
(::) : a -> [a] -> [a]
|
|
(::) = Native.List.cons
|
|
|
|
{-| Puts two appendable things together:
|
|
|
|
[1,1] ++ [2,3] == [1,1,2,3]
|
|
"abc" ++ "123" == "abc123"
|
|
-}
|
|
(++) : appendable -> appendable -> appendable
|
|
(++) = Native.List.append
|
|
|
|
infixr 5 ::
|
|
infixr 5 ++
|
|
|
|
{-| Extract the first element of a list. List must be non-empty.
|
|
`(head [1,2,3] == 1)`
|
|
-}
|
|
head : [a] -> a
|
|
head = Native.List.head
|
|
|
|
{-| Extract the elements after the head of the list. List must be non-empty.
|
|
`(tail [1,2,3] == [2,3])`
|
|
-}
|
|
tail : [a] -> [a]
|
|
tail = Native.List.tail
|
|
|
|
{-| Extract the last element of a list. List must be non-empty.
|
|
`(last [1,2,3] == 3)`
|
|
-}
|
|
last : [a] -> a
|
|
last = Native.List.last
|
|
|
|
{-| Check if a list is empty `(isEmpty [] == True)` -}
|
|
isEmpty : [a] -> Bool
|
|
isEmpty xs =
|
|
case xs of
|
|
[] -> True
|
|
_ -> False
|
|
|
|
{-| Apply a function to every element of a list: `(map sqrt [1,4,9] == [1,2,3])` -}
|
|
map : (a -> b) -> [a] -> [b]
|
|
map = Native.List.map
|
|
|
|
{-| Reduce a list from the left: `(foldl (::) [] "gateman" == "nametag")` -}
|
|
foldl : (a -> b -> b) -> b -> [a] -> b
|
|
foldl = Native.List.foldl
|
|
|
|
{-| Reduce a list from the right: `(foldr (+) 0 [1,2,3] == 6)` -}
|
|
foldr : (a -> b -> b) -> b -> [a] -> b
|
|
foldr = Native.List.foldr
|
|
|
|
{-| Reduce a list from the left without a base case. List must be non-empty. -}
|
|
foldl1 : (a -> a -> a) -> [a] -> a
|
|
foldl1 = Native.List.foldl1
|
|
|
|
{-| Reduce a list from the right without a base case. List must be non-empty. -}
|
|
foldr1 : (a -> a -> a) -> [a] -> a
|
|
foldr1 = Native.List.foldr1
|
|
|
|
{-| Reduce a list from the left, building up all of the intermediate results into a list.
|
|
|
|
scanl (+) 0 [1,2,3,4] == [0,1,3,6,10]
|
|
-}
|
|
scanl : (a -> b -> b) -> b -> [a] -> [b]
|
|
scanl = Native.List.scanl
|
|
|
|
{-| Same as scanl but it doesn't require a base case. List must be non-empty.
|
|
|
|
scanl1 (+) [1,2,3,4] == [1,3,6,10]
|
|
-}
|
|
scanl1 : (a -> a -> a) -> [a] -> [a]
|
|
scanl1 = Native.List.scanl1
|
|
|
|
{-| Keep only elements that satisfy the predicate:
|
|
`(filter isLower "AaBbCc" == "abc")`
|
|
-}
|
|
filter : (a -> Bool) -> [a] -> [a]
|
|
filter = Native.List.filter
|
|
|
|
{-| Determine the length of a list: `(length "innumerable" == 11)` -}
|
|
length : [a] -> Int
|
|
length = Native.List.length
|
|
|
|
{-| Reverse a list. `(reverse [1..4] == [4,3,2,1])` -}
|
|
reverse : [a] -> [a]
|
|
reverse = Native.List.reverse
|
|
|
|
{-| Check to see if all elements satisfy the predicate. -}
|
|
all : (a -> Bool) -> [a] -> Bool
|
|
all = Native.List.all
|
|
|
|
{-| Check to see if any elements satisfy the predicate. -}
|
|
any : (a -> Bool) -> [a] -> Bool
|
|
any = Native.List.any
|
|
|
|
{-| Check to see if all elements are True. -}
|
|
and : [Bool] -> Bool
|
|
and = foldl (&&) True
|
|
|
|
{-| Check to see if any elements are True. -}
|
|
or : [Bool] -> Bool
|
|
or = foldl (||) False
|
|
|
|
{-| Concatenate a list of appendable things:
|
|
|
|
concat ["tree","house"] == "treehouse"
|
|
-}
|
|
concat : [appendable] -> appendable
|
|
concat = Native.List.concat
|
|
|
|
{-| Map a given function onto a list and flatten the resulting lists.
|
|
|
|
concatMap f xs == concat (map f xs)
|
|
-}
|
|
concatMap : (a -> appendable) -> [a] -> appendable
|
|
concatMap f list = concat (map f list)
|
|
|
|
{-| Get the sum of the list elements. `(sum [1..4] == 10)` -}
|
|
sum : [number] -> number
|
|
sum = foldl (+) 0
|
|
|
|
{-| Get the product of the list elements. `(product [1..4] == 24)` -}
|
|
product : [number] -> number
|
|
product = foldl (*) 1
|
|
|
|
{-| Find the maximum element in a non-empty list: `maximum [1,4,2] == 4` -}
|
|
maximum : [comparable] -> comparable
|
|
maximum = foldl1 max
|
|
|
|
{-| Find the minimum element in a non-empty list: `minimum [3,2,1] == 1` -}
|
|
minimum : [comparable] -> comparable
|
|
minimum = foldl1 min
|
|
|
|
{-| Split a list based on the predicate. -}
|
|
partition : (a -> Bool) -> [a] -> ([a],[a])
|
|
partition pred lst =
|
|
case lst of
|
|
[] -> ([],[])
|
|
x::xs -> let (bs,cs) = partition pred xs in
|
|
if pred x then (x::bs,cs) else (bs,x::cs)
|
|
|
|
{-| Combine two lists, combining them into tuples pairwise.
|
|
If one list is longer, the extra elements are dropped.
|
|
|
|
zip [1,2,3] [6,7] == [(1,6),(2,7)]
|
|
zip == zipWith (,)
|
|
-}
|
|
zip : [a] -> [b] -> [(a,b)]
|
|
zip = Native.List.zip
|
|
|
|
{-| Combine two lists, combining them with the given function.
|
|
If one list is longer, the extra elements are dropped.
|
|
|
|
zipWith (+) [1,2,3] [1,2,3,4] == [2,4,6]
|
|
-}
|
|
zipWith : (a -> b -> c) -> [a] -> [b] -> [c]
|
|
zipWith = Native.List.zipWith
|
|
|
|
{-| Decompose a list of tuples. -}
|
|
unzip : [(a,b)] -> ([a],[b])
|
|
unzip pairs =
|
|
case pairs of
|
|
[] -> ([],[])
|
|
(x,y)::ps -> let (xs,ys) = (unzip ps) in (x::xs,y::ys)
|
|
|
|
{-| Split a list with a given seperator.
|
|
|
|
split "," "hello,there,friend" == ["hello", "there", "friend"]
|
|
-}
|
|
split : [a] -> [a] -> [[a]]
|
|
split = Native.List.split
|
|
|
|
{-| Places the given value between all of the lists in the second
|
|
argument and concatenates the result.
|
|
|
|
join "a" ["H","w","ii","n"] == "Hawaiian"
|
|
-}
|
|
join : appendable -> [appendable] -> appendable
|
|
join = Native.List.join
|
|
|
|
{-| Places the given value between all members of the given list.
|
|
|
|
intersperse ' ' "INCEPTION" == "I N C E P T I O N"
|
|
-}
|
|
intersperse : a -> [a] -> [a]
|
|
intersperse sep xs =
|
|
case xs of
|
|
a::b::cs -> a :: sep :: intersperse sep (b::cs)
|
|
[a] -> [a]
|
|
[] -> []
|
|
|
|
{-| Take the first n members of a list: `(take 2 [1,2,3,4] == [1,2])` -}
|
|
take : Int -> [a] -> [a]
|
|
take = Native.List.take
|
|
|
|
{-| Drop the first n members of a list: `(drop 2 [1,2,3,4] == [3,4])` -}
|
|
drop : Int -> [a] -> [a]
|
|
drop = Native.List.drop
|
|
|
|
{-| Creates a list with *n* copies of a value:
|
|
`(repeat 3 (0,0) == [(0,0),(0,0),(0,0)]`
|
|
-}
|
|
repeat : Int -> a -> [a]
|
|
repeat = Native.List.repeat
|