module Data.Vector.HFixed (
Arity
, ArityC
, HVector(..)
, tupleSize
, HVectorF(..)
, tupleSizeF
, Proxy(..)
, ContVec
, ContVecF(..)
, asCVec
, asCVecF
, convert
, head
, tail
, cons
, concat
, ValueAt
, Index
, index
, set
, element
, elementCh
, mk0
, mk1
, mk2
, mk3
, mk4
, mk5
, fold
, foldr
, foldl
, foldrF
, foldlF
, foldrNatF
, foldlNatF
, mapM_
, unfoldr
, unfoldrF
, replicate
, replicateM
, replicateF
, replicateNatF
, zipWith
, zipWithF
, zipWithNatF
, zipFold
, zipFoldF
, monomorphize
, monomorphizeF
, mapNat
, sequence
, sequence_
, sequenceF
, wrap
, unwrap
, distribute
, distributeF
, eq
, compare
, rnf
) where
import Control.Applicative (Applicative(..),(<$>))
import qualified Control.DeepSeq as NF
import Data.Coerce (coerce)
import Data.Functor.Compose (Compose(..))
import Data.Functor.Identity (Identity(..))
import Data.Monoid (Monoid,All(..))
import Prelude ( Functor(..),Eq(..),Ord,Bool,Ordering
, id,(.),($),seq)
import qualified Prelude
import Data.Vector.HFixed.Class hiding (cons,consF)
import Data.Vector.Fixed.Cont (Peano)
import qualified Data.Vector.Fixed as F
import qualified Data.Vector.HFixed.Cont as C
asCVec :: ContVec xs -> ContVec xs
asCVec = id
asCVecF :: ContVecF f xs -> ContVecF f xs
asCVecF = id
convert :: (HVector v, HVector w, Elems v ~ Elems w)
=> v -> w
convert v = inspect v construct
tail :: (HVector v, HVector w, (a : Elems w) ~ Elems v)
=> v -> w
tail = C.vector . C.tail . C.cvec
head :: (HVector v, Elems v ~ (a : as), Arity as)
=> v -> a
head = C.head . C.cvec
cons :: (HVector v, HVector w, Elems w ~ (a : Elems v))
=> a -> v -> w
cons a = C.vector . C.cons a . C.cvec
concat :: ( HVector v, HVector u, HVector w
, Elems w ~ (Elems v ++ Elems u)
)
=> v -> u -> w
concat v u = C.vector $ C.concat (C.cvec v) (C.cvec u)
index :: (Index n (Elems v), HVector v) => v -> proxy n -> ValueAt n (Elems v)
index = C.index . C.cvec
set :: (Index n (Elems v), HVector v)
=> proxy n -> ValueAt n (Elems v) -> v -> v
set n x = C.vector
. C.set n x
. C.cvec
element :: forall n v a f proxy.
( Index (Peano n) (Elems v)
, ValueAt (Peano n) (Elems v) ~ a
, HVector v
, Functor f
)
=> proxy n -> (a -> f a) -> (v -> f v)
element _ f v = inspect v
$ lensF (Proxy @ (Peano n)) f construct
elementCh :: forall n v w a b f proxy.
( Index (Peano n) (Elems v)
, ValueAt (Peano n) (Elems v) ~ a
, HVector v
, HVector w
, Elems w ~ NewElems (Peano n) (Elems v) b
, Functor f
)
=> proxy n -> (a -> f b) -> (v -> f w)
elementCh _ f v = inspect v
$ lensChF (Proxy @ (Peano n)) f construct
fold :: HVector v => v -> Fn Identity (Elems v) r -> r
fold v f = inspect v (TFun f)
foldr :: (HVector v, ArityC c (Elems v))
=> Proxy c -> (forall a. c a => a -> b -> b) -> b -> v -> b
foldr c f b0 = C.foldrF c (\(Identity a) b -> f a b) b0 . C.cvec
foldl :: (HVector v, ArityC c (Elems v))
=> Proxy c -> (forall a. c a => b -> a -> b) -> b -> v -> b
foldl c f b0 = C.foldlF c (\b (Identity a) -> f b a) b0 . C.cvec
foldrF :: (HVectorF v, ArityC c (ElemsF v))
=> Proxy c -> (forall a. c a => f a -> b -> b) -> b -> v f -> b
foldrF c f b0 = C.foldrF c f b0 . C.cvecF
foldlF :: (HVectorF v, ArityC c (ElemsF v))
=> Proxy c -> (forall a. c a => b -> f a -> b) -> b -> v f -> b
foldlF c f b0 = C.foldlF c f b0 . C.cvecF
foldrNatF :: (HVectorF v)
=> (forall a. f a -> b -> b) -> b -> v f -> b
foldrNatF f b0 = C.foldrNatF f b0 . C.cvecF
foldlNatF :: (HVectorF v)
=> (forall a. b -> f a -> b) -> b -> v f -> b
foldlNatF f b0 = C.foldlNatF f b0 . C.cvecF
mapM_ :: (HVector v, ArityC c (Elems v), Applicative f)
=> Proxy c -> (forall a. c a => a -> f ()) -> v -> f ()
mapM_ c f = foldl c (\m a -> m *> f a) (pure ())
unfoldr :: (HVector v, ArityC c (Elems v))
=> Proxy c -> (forall a. c a => b -> (a,b)) -> b -> v
unfoldr c f = C.vector . C.unfoldrF c (\b -> let (a,b') = f b in (Identity a, b'))
unfoldrF :: (HVectorF v, ArityC c (ElemsF v))
=> Proxy c -> (forall a. c a => b -> (f a,b)) -> b -> v f
unfoldrF c f = C.vectorF . C.unfoldrF c f
mk0 :: forall v. (HVector v, Elems v ~ '[]) => v
mk0 = coerce (construct :: Fun '[] v)
mk1 :: forall v a. (HVector v, Elems v ~ '[a])
=> a -> v
mk1 = coerce (construct :: Fun '[a] v)
mk2 :: forall v a b. (HVector v, Elems v ~ '[a,b])
=> a -> b -> v
mk2 = coerce (construct :: Fun '[a,b] v)
mk3 :: forall v a b c. (HVector v, Elems v ~ '[a,b,c])
=> a -> b -> c -> v
mk3 = coerce (construct :: Fun '[a,b,c] v)
mk4 :: forall v a b c d. (HVector v, Elems v ~ '[a,b,c,d])
=> a -> b -> c -> d -> v
mk4 = coerce (construct :: Fun '[a,b,c,d] v)
mk5 :: forall v a b c d e. (HVector v, Elems v ~ '[a,b,c,d,e])
=> a -> b -> c -> d -> e -> v
mk5 = coerce (construct :: Fun '[a,b,c,d,e] v)
mapNat :: (HVectorF v)
=> (forall a. f a -> g a) -> v f -> v g
mapNat f = C.vectorF . C.mapNat f . C.cvecF
sequence
:: ( Applicative f, HVectorF v, HVector w, ElemsF v ~ Elems w )
=> v f -> f w
sequence
= fmap C.vector
. C.sequenceF
. C.mapNat (Compose . fmap Identity)
. C.cvecF
sequence_ :: (Applicative f, HVectorF v) => v f -> f ()
sequence_ = foldlNatF (\m a -> m <* a) (pure ())
sequenceF :: ( Applicative f, HVectorF v) => v (f `Compose` g) -> f (v g)
sequenceF v = C.vectorF <$> C.sequenceF (C.cvecF v)
wrap :: ( HVector v, HVectorF w, Elems v ~ ElemsF w )
=> (forall a. a -> f a) -> v -> w f
wrap f = C.vectorF . C.mapNat (f . runIdentity) . C.cvec
unwrap :: ( HVectorF v, HVector w, ElemsF v ~ Elems w )
=> (forall a. f a -> a) -> v f -> w
unwrap f = C.vector . C.mapNat (Identity . f) . C.cvecF
distribute
:: ( Functor f, HVector v, HVectorF w, Elems v ~ ElemsF w )
=> f v -> w f
distribute
= C.vectorF
. mapNat (fmap runIdentity . getCompose)
. C.distributeF
. fmap C.cvec
distributeF
:: ( Functor f, HVectorF v)
=> f (v g) -> v (f `Compose` g)
distributeF = C.vectorF . C.distributeF . fmap C.cvecF
replicate :: (HVector v, ArityC c (Elems v))
=> Proxy c -> (forall x. c x => x) -> v
replicate c x = C.vector $ C.replicateF c (Identity x)
replicateM :: (HVector v, Applicative f, ArityC c (Elems v))
=> Proxy c -> (forall a. c a => f a) -> f v
replicateM c x
= fmap C.vector
$ C.sequenceF
$ C.replicateF c (Compose $ fmap Identity x)
replicateNatF :: (HVectorF v, Arity (ElemsF v))
=> (forall a. f a) -> v f
replicateNatF x = C.vectorF $ C.replicateNatF x
replicateF :: (HVectorF v, ArityC c (ElemsF v))
=> Proxy c -> (forall a. c a => f a) -> v f
replicateF c x = C.vectorF $ C.replicateF c x
zipWith :: (HVector v, ArityC c (Elems v))
=> Proxy c -> (forall a. c a => a -> a -> a) -> v -> v -> v
zipWith c f v u
= C.vector
$ C.zipWithF c (\(Identity a) (Identity b) -> Identity (f a b)) (C.cvec v) (C.cvec u)
zipWithF :: (HVectorF v, ArityC c (ElemsF v))
=> Proxy c -> (forall a. c a => f a -> g a -> h a) -> v f -> v g -> v h
zipWithF c f v u
= C.vectorF $ C.zipWithF c f (C.cvecF v) (C.cvecF u)
zipWithNatF :: (HVectorF v)
=> (forall a. f a -> g a -> h a) -> v f -> v g -> v h
zipWithNatF f v u
= C.vectorF $ C.zipWithNatF f (C.cvecF v) (C.cvecF u)
zipFold :: (HVector v, ArityC c (Elems v), Monoid m)
=> Proxy c -> (forall a. c a => a -> a -> m) -> v -> v -> m
zipFold c f v u
= C.zipFoldF c (\(Identity a) (Identity b) -> f a b) (C.cvec v) (C.cvec u)
zipFoldF :: (HVectorF v, ArityC c (ElemsF v), Monoid m)
=> Proxy c -> (forall a. c a => f a -> f a -> m) -> v f -> v f -> m
zipFoldF c f v u
= C.zipFoldF c f (C.cvecF v) (C.cvecF u)
monomorphize :: ( HVector v
, Peano n ~ Len (Elems v)
, ArityC c (Elems v))
=> Proxy c -> (forall a. c a => a -> x)
-> v -> F.ContVec n x
monomorphize c f = C.monomorphizeF c (f . runIdentity) . C.cvec
monomorphizeF :: ( HVectorF v
, Peano n ~ Len (ElemsF v)
, ArityC c (ElemsF v)
)
=> Proxy c -> (forall a. c a => f a -> x)
-> v f -> F.ContVec n x
monomorphizeF c f = C.monomorphizeF c f . C.cvecF
eq :: (HVector v, ArityC Eq (Elems v)) => v -> v -> Bool
eq v u = getAll $ zipFold (Proxy :: Proxy Eq) (\x y -> All (x == y)) v u
compare :: (HVector v, ArityC Ord (Elems v)) => v -> v -> Ordering
compare = zipFold (Proxy :: Proxy Ord) Prelude.compare
rnf :: (HVector v, ArityC NF.NFData (Elems v)) => v -> ()
rnf = foldl (Proxy :: Proxy NF.NFData) (\r a -> NF.rnf a `seq` r) ()