{-# LANGUAGE Safe, QuasiQuotes, TypeOperators #-}
module Data.Invertible.List
( cons
, uncons
, consMaybe
, repLen
, map
, reverse
, transpose
, lookup
, index
, zip
, zip3
, zip4
, zip5
, zip6
, zip7
, zipWith
, interleave
, lines
, words
) where
import Prelude hiding (map, reverse, lookup, zip, zip3, unzip, zipWith, lines, words)
import Control.Arrow ((***))
import qualified Data.List as L
import Data.Tuple (swap)
import Data.Invertible.Bijection
import Data.Invertible.TH
import Data.Invertible.Internal
cons :: Maybe (a, [a]) <-> [a]
cons :: forall a. Maybe (a, [a]) <-> [a]
cons =
[biCase|
Just (a, l) <-> a:l
Nothing <-> []
|]
uncons :: [a] <-> Maybe (a, [a])
uncons :: forall a. [a] <-> Maybe (a, [a])
uncons = forall (a :: * -> * -> *) b c. Bijection a b c -> Bijection a c b
invert forall a. Maybe (a, [a]) <-> [a]
cons
consMaybe :: (Maybe a, [a]) <-> [a]
consMaybe :: forall a. (Maybe a, [a]) <-> [a]
consMaybe =
[biCase|
(Just a, l) <-> a:l
(Nothing, l) <-> l
|]
repLen :: Int <-> [()]
repLen :: Int <-> [()]
repLen = (forall a. Int -> a -> [a]
`L.replicate` ()) forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: forall (t :: * -> *) a. Foldable t => t a -> Int
L.length
map :: (a <-> b) -> [a] <-> [b]
map :: forall a b. (a <-> b) -> [a] <-> [b]
map (a -> b
f :<->: b -> a
g) = forall a b. (a -> b) -> [a] -> [b]
L.map a -> b
f forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: forall a b. (a -> b) -> [a] -> [b]
L.map b -> a
g
reverse :: [a] <-> [a]
reverse :: forall a. [a] <-> [a]
reverse = forall (a :: * -> * -> *) b. a b b -> Bijection a b b
involution forall a. [a] -> [a]
L.reverse
transpose :: [[a]] <-> [[a]]
transpose :: forall a. [[a]] <-> [[a]]
transpose = forall (a :: * -> * -> *) b. a b b -> Bijection a b b
involution forall a. [[a]] -> [[a]]
L.transpose
lookup :: (Eq a, Eq b) => [(a, b)] -> Maybe a <-> Maybe b
lookup :: forall a b. (Eq a, Eq b) => [(a, b)] -> Maybe a <-> Maybe b
lookup [(a, b)]
l = (forall a b c. (a -> b -> c) -> b -> a -> c
flip forall a b. Eq a => a -> [(a, b)] -> Maybe b
L.lookup [(a, b)]
l forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<<) forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: (forall a b c. (a -> b -> c) -> b -> a -> c
flip forall a b. Eq a => a -> [(a, b)] -> Maybe b
L.lookup (forall a b. (a -> b) -> [a] -> [b]
L.map forall a b. (a, b) -> (b, a)
swap [(a, b)]
l) forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<<)
index :: Eq a => [a] -> Maybe a <-> Maybe Int
index :: forall a. Eq a => [a] -> Maybe a <-> Maybe Int
index [a]
l = (forall a b c. (a -> b -> c) -> b -> a -> c
flip forall a. Eq a => a -> [a] -> Maybe Int
L.elemIndex [a]
l forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<<) forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: (forall {a} {a}. (Ord a, Num a, Enum a) => [a] -> a -> Maybe a
idx [a]
l forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<<) where
idx :: [a] -> a -> Maybe a
idx [a]
_ a
i | a
i forall a. Ord a => a -> a -> Bool
< a
0 = forall a. Maybe a
Nothing
idx [] a
_ = forall a. Maybe a
Nothing
idx (a
x:[a]
_) a
0 = forall a. a -> Maybe a
Just a
x
idx (a
_:[a]
r) a
i = [a] -> a -> Maybe a
idx [a]
r forall a b. (a -> b) -> a -> b
$ forall a. Enum a => a -> a
pred a
i
zip :: ([a], [b]) <-> [(a, b)]
zip :: forall a b. ([a], [b]) <-> [(a, b)]
zip = forall a b c. (a -> b -> c) -> (a, b) -> c
uncurry forall a b. [a] -> [b] -> [(a, b)]
L.zip forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: forall a b. [(a, b)] -> ([a], [b])
L.unzip
zip3 :: ([a], [b], [c]) <-> [(a, b, c)]
zip3 :: forall a b c. ([a], [b], [c]) <-> [(a, b, c)]
zip3 = (\([a]
a,[b]
b,[c]
c) -> forall a b c. [a] -> [b] -> [c] -> [(a, b, c)]
L.zip3 [a]
a [b]
b [c]
c) forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: forall a b c. [(a, b, c)] -> ([a], [b], [c])
L.unzip3
zip4 :: ([a], [b], [c], [d]) <-> [(a, b, c, d)]
zip4 :: forall a b c d. ([a], [b], [c], [d]) <-> [(a, b, c, d)]
zip4 = (\([a]
a,[b]
b,[c]
c,[d]
d) -> forall a b c d. [a] -> [b] -> [c] -> [d] -> [(a, b, c, d)]
L.zip4 [a]
a [b]
b [c]
c [d]
d) forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: forall a b c d. [(a, b, c, d)] -> ([a], [b], [c], [d])
L.unzip4
zip5 :: ([a], [b], [c], [d], [e]) <-> [(a, b, c, d, e)]
zip5 :: forall a b c d e. ([a], [b], [c], [d], [e]) <-> [(a, b, c, d, e)]
zip5 = (\([a]
a,[b]
b,[c]
c,[d]
d,[e]
e) -> forall a b c d e.
[a] -> [b] -> [c] -> [d] -> [e] -> [(a, b, c, d, e)]
L.zip5 [a]
a [b]
b [c]
c [d]
d [e]
e) forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: forall a b c d e. [(a, b, c, d, e)] -> ([a], [b], [c], [d], [e])
L.unzip5
zip6 :: ([a], [b], [c], [d], [e], [f]) <-> [(a, b, c, d, e, f)]
zip6 :: forall a b c d e f.
([a], [b], [c], [d], [e], [f]) <-> [(a, b, c, d, e, f)]
zip6 = (\([a]
a,[b]
b,[c]
c,[d]
d,[e]
e,[f]
f) -> forall a b c d e f.
[a] -> [b] -> [c] -> [d] -> [e] -> [f] -> [(a, b, c, d, e, f)]
L.zip6 [a]
a [b]
b [c]
c [d]
d [e]
e [f]
f) forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: forall a b c d e f.
[(a, b, c, d, e, f)] -> ([a], [b], [c], [d], [e], [f])
L.unzip6
zip7 :: ([a], [b], [c], [d], [e], [f], [g]) <-> [(a, b, c, d, e, f, g)]
zip7 :: forall a b c d e f g.
([a], [b], [c], [d], [e], [f], [g]) <-> [(a, b, c, d, e, f, g)]
zip7 = (\([a]
a,[b]
b,[c]
c,[d]
d,[e]
e,[f]
f,[g]
g) -> forall a b c d e f g.
[a]
-> [b]
-> [c]
-> [d]
-> [e]
-> [f]
-> [g]
-> [(a, b, c, d, e, f, g)]
L.zip7 [a]
a [b]
b [c]
c [d]
d [e]
e [f]
f [g]
g) forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: forall a b c d e f g.
[(a, b, c, d, e, f, g)] -> ([a], [b], [c], [d], [e], [f], [g])
L.unzip7
zipWith :: (a, b) <-> c -> ([a], [b]) <-> [(c)]
zipWith :: forall a b c. ((a, b) <-> c) -> ([a], [b]) <-> [c]
zipWith ((a, b) -> c
f :<->: c -> (a, b)
g) = forall a b c. (a -> b -> c) -> (a, b) -> c
uncurry (forall a b c. (a -> b -> c) -> [a] -> [b] -> [c]
L.zipWith (forall a b c. ((a, b) -> c) -> a -> b -> c
curry (a, b) -> c
f)) forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: forall a b. [(a, b)] -> ([a], [b])
L.unzip forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a b. (a -> b) -> [a] -> [b]
L.map c -> (a, b)
g
interleave :: ([a], [a]) <-> [a]
interleave :: forall a. ([a], [a]) <-> [a]
interleave = forall a b c. (a -> b -> c) -> (a, b) -> c
uncurry forall {a}. [a] -> [a] -> [a]
f forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: forall {a}. [a] -> ([a], [a])
g where
f :: [a] -> [a] -> [a]
f (a
x:[a]
xl) (a
y:[a]
yl) = a
xforall a. a -> [a] -> [a]
:a
yforall a. a -> [a] -> [a]
:[a] -> [a] -> [a]
f [a]
xl [a]
yl
f [] [a]
l = [a]
l
f [a]
l [] = [a]
l
g :: [a] -> ([a], [a])
g (a
x:a
y:[a]
l) = (a
xforall a. a -> [a] -> [a]
:) forall (a :: * -> * -> *) b c b' c'.
Arrow a =>
a b c -> a b' c' -> a (b, b') (c, c')
*** (a
yforall a. a -> [a] -> [a]
:) forall a b. (a -> b) -> a -> b
$ [a] -> ([a], [a])
g [a]
l
g [a]
l = ([a]
l, [])
lines :: String <-> [String]
lines :: String <-> [String]
lines = String -> [String]
L.lines forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: [String] -> String
L.unlines
words :: String <-> [String]
words :: String <-> [String]
words = String -> [String]
L.words forall (a :: * -> * -> *) b c. a b c -> a c b -> Bijection a b c
:<->: [String] -> String
L.unwords