{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE CPP #-}
{-# LANGUAGE DeriveTraversable #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TypeApplications #-}
{-# LANGUAGE TypeFamilies #-}
module Data.Chimera
(
memoize
, memoizeFix
, Chimera
, VChimera
, UChimera
, tabulate
, tabulateFix
, tabulateFix'
, iterate
, cycle
, index
, toList
, tabulateM
, tabulateFixM
, iterateM
, mapSubvectors
, zipSubvectors
) where
import Prelude hiding ((^), (*), div, fromIntegral, not, and, or, cycle, iterate, drop)
import Control.Applicative
import Control.Monad.Fix
import Control.Monad.Zip
import Data.Bits
import Data.Functor.Identity
import qualified Data.Vector as V
import qualified Data.Vector.Generic as G
import qualified Data.Vector.Unboxed as U
#ifdef MIN_VERSION_mtl
import Control.Monad.Reader (MonadReader, ask, local)
#endif
#ifdef MIN_VERSION_distributive
import Data.Distributive
#ifdef MIN_VERSION_adjunctions
import qualified Data.Functor.Rep as Rep
#endif
#endif
import Data.Chimera.Compat
import Data.Chimera.FromIntegral
newtype Chimera v a = Chimera { forall (v :: * -> *) a. Chimera v a -> Vector (v a)
_unChimera :: V.Vector (v a) }
deriving ((forall a b. (a -> b) -> Chimera v a -> Chimera v b)
-> (forall a b. a -> Chimera v b -> Chimera v a)
-> Functor (Chimera v)
forall a b. a -> Chimera v b -> Chimera v a
forall a b. (a -> b) -> Chimera v a -> Chimera v b
forall (v :: * -> *) a b.
Functor v =>
a -> Chimera v b -> Chimera v a
forall (v :: * -> *) a b.
Functor v =>
(a -> b) -> Chimera v a -> Chimera v b
forall (f :: * -> *).
(forall a b. (a -> b) -> f a -> f b)
-> (forall a b. a -> f b -> f a) -> Functor f
$cfmap :: forall (v :: * -> *) a b.
Functor v =>
(a -> b) -> Chimera v a -> Chimera v b
fmap :: forall a b. (a -> b) -> Chimera v a -> Chimera v b
$c<$ :: forall (v :: * -> *) a b.
Functor v =>
a -> Chimera v b -> Chimera v a
<$ :: forall a b. a -> Chimera v b -> Chimera v a
Functor, (forall m. Monoid m => Chimera v m -> m)
-> (forall m a. Monoid m => (a -> m) -> Chimera v a -> m)
-> (forall m a. Monoid m => (a -> m) -> Chimera v a -> m)
-> (forall a b. (a -> b -> b) -> b -> Chimera v a -> b)
-> (forall a b. (a -> b -> b) -> b -> Chimera v a -> b)
-> (forall b a. (b -> a -> b) -> b -> Chimera v a -> b)
-> (forall b a. (b -> a -> b) -> b -> Chimera v a -> b)
-> (forall a. (a -> a -> a) -> Chimera v a -> a)
-> (forall a. (a -> a -> a) -> Chimera v a -> a)
-> (forall a. Chimera v a -> [a])
-> (forall a. Chimera v a -> Bool)
-> (forall a. Chimera v a -> Int)
-> (forall a. Eq a => a -> Chimera v a -> Bool)
-> (forall a. Ord a => Chimera v a -> a)
-> (forall a. Ord a => Chimera v a -> a)
-> (forall a. Num a => Chimera v a -> a)
-> (forall a. Num a => Chimera v a -> a)
-> Foldable (Chimera v)
forall a. Eq a => a -> Chimera v a -> Bool
forall a. Num a => Chimera v a -> a
forall a. Ord a => Chimera v a -> a
forall m. Monoid m => Chimera v m -> m
forall a. Chimera v a -> Bool
forall a. Chimera v a -> Int
forall a. Chimera v a -> [a]
forall a. (a -> a -> a) -> Chimera v a -> a
forall m a. Monoid m => (a -> m) -> Chimera v a -> m
forall b a. (b -> a -> b) -> b -> Chimera v a -> b
forall a b. (a -> b -> b) -> b -> Chimera v a -> b
forall (v :: * -> *) a.
(Foldable v, Eq a) =>
a -> Chimera v a -> Bool
forall (v :: * -> *) a. (Foldable v, Num a) => Chimera v a -> a
forall (v :: * -> *) a. (Foldable v, Ord a) => Chimera v a -> a
forall (v :: * -> *) m. (Foldable v, Monoid m) => Chimera v m -> m
forall (v :: * -> *) a. Foldable v => Chimera v a -> Bool
forall (v :: * -> *) a. Foldable v => Chimera v a -> Int
forall (v :: * -> *) a. Foldable v => Chimera v a -> [a]
forall (v :: * -> *) a.
Foldable v =>
(a -> a -> a) -> Chimera v a -> a
forall (v :: * -> *) m a.
(Foldable v, Monoid m) =>
(a -> m) -> Chimera v a -> m
forall (v :: * -> *) b a.
Foldable v =>
(b -> a -> b) -> b -> Chimera v a -> b
forall (v :: * -> *) a b.
Foldable v =>
(a -> b -> b) -> b -> Chimera v a -> b
forall (t :: * -> *).
(forall m. Monoid m => t m -> m)
-> (forall m a. Monoid m => (a -> m) -> t a -> m)
-> (forall m a. Monoid m => (a -> m) -> t a -> m)
-> (forall a b. (a -> b -> b) -> b -> t a -> b)
-> (forall a b. (a -> b -> b) -> b -> t a -> b)
-> (forall b a. (b -> a -> b) -> b -> t a -> b)
-> (forall b a. (b -> a -> b) -> b -> t a -> b)
-> (forall a. (a -> a -> a) -> t a -> a)
-> (forall a. (a -> a -> a) -> t a -> a)
-> (forall a. t a -> [a])
-> (forall a. t a -> Bool)
-> (forall a. t a -> Int)
-> (forall a. Eq a => a -> t a -> Bool)
-> (forall a. Ord a => t a -> a)
-> (forall a. Ord a => t a -> a)
-> (forall a. Num a => t a -> a)
-> (forall a. Num a => t a -> a)
-> Foldable t
$cfold :: forall (v :: * -> *) m. (Foldable v, Monoid m) => Chimera v m -> m
fold :: forall m. Monoid m => Chimera v m -> m
$cfoldMap :: forall (v :: * -> *) m a.
(Foldable v, Monoid m) =>
(a -> m) -> Chimera v a -> m
foldMap :: forall m a. Monoid m => (a -> m) -> Chimera v a -> m
$cfoldMap' :: forall (v :: * -> *) m a.
(Foldable v, Monoid m) =>
(a -> m) -> Chimera v a -> m
foldMap' :: forall m a. Monoid m => (a -> m) -> Chimera v a -> m
$cfoldr :: forall (v :: * -> *) a b.
Foldable v =>
(a -> b -> b) -> b -> Chimera v a -> b
foldr :: forall a b. (a -> b -> b) -> b -> Chimera v a -> b
$cfoldr' :: forall (v :: * -> *) a b.
Foldable v =>
(a -> b -> b) -> b -> Chimera v a -> b
foldr' :: forall a b. (a -> b -> b) -> b -> Chimera v a -> b
$cfoldl :: forall (v :: * -> *) b a.
Foldable v =>
(b -> a -> b) -> b -> Chimera v a -> b
foldl :: forall b a. (b -> a -> b) -> b -> Chimera v a -> b
$cfoldl' :: forall (v :: * -> *) b a.
Foldable v =>
(b -> a -> b) -> b -> Chimera v a -> b
foldl' :: forall b a. (b -> a -> b) -> b -> Chimera v a -> b
$cfoldr1 :: forall (v :: * -> *) a.
Foldable v =>
(a -> a -> a) -> Chimera v a -> a
foldr1 :: forall a. (a -> a -> a) -> Chimera v a -> a
$cfoldl1 :: forall (v :: * -> *) a.
Foldable v =>
(a -> a -> a) -> Chimera v a -> a
foldl1 :: forall a. (a -> a -> a) -> Chimera v a -> a
$ctoList :: forall (v :: * -> *) a. Foldable v => Chimera v a -> [a]
toList :: forall a. Chimera v a -> [a]
$cnull :: forall (v :: * -> *) a. Foldable v => Chimera v a -> Bool
null :: forall a. Chimera v a -> Bool
$clength :: forall (v :: * -> *) a. Foldable v => Chimera v a -> Int
length :: forall a. Chimera v a -> Int
$celem :: forall (v :: * -> *) a.
(Foldable v, Eq a) =>
a -> Chimera v a -> Bool
elem :: forall a. Eq a => a -> Chimera v a -> Bool
$cmaximum :: forall (v :: * -> *) a. (Foldable v, Ord a) => Chimera v a -> a
maximum :: forall a. Ord a => Chimera v a -> a
$cminimum :: forall (v :: * -> *) a. (Foldable v, Ord a) => Chimera v a -> a
minimum :: forall a. Ord a => Chimera v a -> a
$csum :: forall (v :: * -> *) a. (Foldable v, Num a) => Chimera v a -> a
sum :: forall a. Num a => Chimera v a -> a
$cproduct :: forall (v :: * -> *) a. (Foldable v, Num a) => Chimera v a -> a
product :: forall a. Num a => Chimera v a -> a
Foldable, Functor (Chimera v)
Foldable (Chimera v)
Functor (Chimera v)
-> Foldable (Chimera v)
-> (forall (f :: * -> *) a b.
Applicative f =>
(a -> f b) -> Chimera v a -> f (Chimera v b))
-> (forall (f :: * -> *) a.
Applicative f =>
Chimera v (f a) -> f (Chimera v a))
-> (forall (m :: * -> *) a b.
Monad m =>
(a -> m b) -> Chimera v a -> m (Chimera v b))
-> (forall (m :: * -> *) a.
Monad m =>
Chimera v (m a) -> m (Chimera v a))
-> Traversable (Chimera v)
forall (t :: * -> *).
Functor t
-> Foldable t
-> (forall (f :: * -> *) a b.
Applicative f =>
(a -> f b) -> t a -> f (t b))
-> (forall (f :: * -> *) a. Applicative f => t (f a) -> f (t a))
-> (forall (m :: * -> *) a b.
Monad m =>
(a -> m b) -> t a -> m (t b))
-> (forall (m :: * -> *) a. Monad m => t (m a) -> m (t a))
-> Traversable t
forall {v :: * -> *}. Traversable v => Functor (Chimera v)
forall {v :: * -> *}. Traversable v => Foldable (Chimera v)
forall (v :: * -> *) (m :: * -> *) a.
(Traversable v, Monad m) =>
Chimera v (m a) -> m (Chimera v a)
forall (v :: * -> *) (f :: * -> *) a.
(Traversable v, Applicative f) =>
Chimera v (f a) -> f (Chimera v a)
forall (v :: * -> *) (m :: * -> *) a b.
(Traversable v, Monad m) =>
(a -> m b) -> Chimera v a -> m (Chimera v b)
forall (v :: * -> *) (f :: * -> *) a b.
(Traversable v, Applicative f) =>
(a -> f b) -> Chimera v a -> f (Chimera v b)
forall (m :: * -> *) a.
Monad m =>
Chimera v (m a) -> m (Chimera v a)
forall (f :: * -> *) a.
Applicative f =>
Chimera v (f a) -> f (Chimera v a)
forall (m :: * -> *) a b.
Monad m =>
(a -> m b) -> Chimera v a -> m (Chimera v b)
forall (f :: * -> *) a b.
Applicative f =>
(a -> f b) -> Chimera v a -> f (Chimera v b)
$ctraverse :: forall (v :: * -> *) (f :: * -> *) a b.
(Traversable v, Applicative f) =>
(a -> f b) -> Chimera v a -> f (Chimera v b)
traverse :: forall (f :: * -> *) a b.
Applicative f =>
(a -> f b) -> Chimera v a -> f (Chimera v b)
$csequenceA :: forall (v :: * -> *) (f :: * -> *) a.
(Traversable v, Applicative f) =>
Chimera v (f a) -> f (Chimera v a)
sequenceA :: forall (f :: * -> *) a.
Applicative f =>
Chimera v (f a) -> f (Chimera v a)
$cmapM :: forall (v :: * -> *) (m :: * -> *) a b.
(Traversable v, Monad m) =>
(a -> m b) -> Chimera v a -> m (Chimera v b)
mapM :: forall (m :: * -> *) a b.
Monad m =>
(a -> m b) -> Chimera v a -> m (Chimera v b)
$csequence :: forall (v :: * -> *) (m :: * -> *) a.
(Traversable v, Monad m) =>
Chimera v (m a) -> m (Chimera v a)
sequence :: forall (m :: * -> *) a.
Monad m =>
Chimera v (m a) -> m (Chimera v a)
Traversable)
type VChimera = Chimera V.Vector
type UChimera = Chimera U.Vector
instance Applicative (Chimera V.Vector) where
pure :: forall a. a -> Chimera Vector a
pure = (Word -> a) -> Chimera Vector a
forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate ((Word -> a) -> Chimera Vector a)
-> (a -> Word -> a) -> a -> Chimera Vector a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> Word -> a
forall a b. a -> b -> a
const
<*> :: forall a b.
Chimera Vector (a -> b) -> Chimera Vector a -> Chimera Vector b
(<*>) = (Vector (a -> b) -> Vector a -> Vector b)
-> Chimera Vector (a -> b) -> Chimera Vector a -> Chimera Vector b
forall (u :: * -> *) a (v :: * -> *) b (w :: * -> *) c.
(Vector u a, Vector v b, Vector w c) =>
(u a -> v b -> w c) -> Chimera u a -> Chimera v b -> Chimera w c
zipSubvectors Vector (a -> b) -> Vector a -> Vector b
forall a b. Vector (a -> b) -> Vector a -> Vector b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
(<*>)
#if __GLASGOW_HASKELL__ > 801
liftA2 :: forall a b c.
(a -> b -> c)
-> Chimera Vector a -> Chimera Vector b -> Chimera Vector c
liftA2 a -> b -> c
f = (Vector a -> Vector b -> Vector c)
-> Chimera Vector a -> Chimera Vector b -> Chimera Vector c
forall (u :: * -> *) a (v :: * -> *) b (w :: * -> *) c.
(Vector u a, Vector v b, Vector w c) =>
(u a -> v b -> w c) -> Chimera u a -> Chimera v b -> Chimera w c
zipSubvectors ((a -> b -> c) -> Vector a -> Vector b -> Vector c
forall a b c. (a -> b -> c) -> Vector a -> Vector b -> Vector c
forall (f :: * -> *) a b c.
Applicative f =>
(a -> b -> c) -> f a -> f b -> f c
liftA2 a -> b -> c
f)
#endif
instance Monad (Chimera V.Vector) where
Chimera Vector a
m >>= :: forall a b.
Chimera Vector a -> (a -> Chimera Vector b) -> Chimera Vector b
>>= a -> Chimera Vector b
f = (Word -> b) -> Chimera Vector b
forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate ((Word -> b) -> Chimera Vector b)
-> (Word -> b) -> Chimera Vector b
forall a b. (a -> b) -> a -> b
$ \Word
w -> Chimera Vector b -> Word -> b
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index (a -> Chimera Vector b
f (Chimera Vector a -> Word -> a
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index Chimera Vector a
m Word
w)) Word
w
instance MonadFix (Chimera V.Vector) where
mfix :: forall a. (a -> Chimera Vector a) -> Chimera Vector a
mfix = (Word -> a) -> Chimera Vector a
forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate ((Word -> a) -> Chimera Vector a)
-> ((a -> Chimera Vector a) -> Word -> a)
-> (a -> Chimera Vector a)
-> Chimera Vector a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (a -> Word -> a) -> Word -> a
forall a. (a -> Word -> a) -> Word -> a
forall (m :: * -> *) a. MonadFix m => (a -> m a) -> m a
mfix ((a -> Word -> a) -> Word -> a)
-> ((a -> Chimera Vector a) -> a -> Word -> a)
-> (a -> Chimera Vector a)
-> Word
-> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Chimera Vector a -> Word -> a)
-> (a -> Chimera Vector a) -> a -> Word -> a
forall a b. (a -> b) -> (a -> a) -> a -> b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap Chimera Vector a -> Word -> a
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index
instance MonadZip (Chimera V.Vector) where
mzip :: forall a b.
Chimera Vector a -> Chimera Vector b -> Chimera Vector (a, b)
mzip Chimera Vector a
as Chimera Vector b
bs = (Word -> (a, b)) -> Chimera Vector (a, b)
forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate (\Word
w -> (Chimera Vector a -> Word -> a
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index Chimera Vector a
as Word
w, Chimera Vector b -> Word -> b
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index Chimera Vector b
bs Word
w))
mzipWith :: forall a b c.
(a -> b -> c)
-> Chimera Vector a -> Chimera Vector b -> Chimera Vector c
mzipWith a -> b -> c
f Chimera Vector a
as Chimera Vector b
bs = (Word -> c) -> Chimera Vector c
forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate ((Word -> c) -> Chimera Vector c)
-> (Word -> c) -> Chimera Vector c
forall a b. (a -> b) -> a -> b
$ \Word
w -> a -> b -> c
f (Chimera Vector a -> Word -> a
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index Chimera Vector a
as Word
w) (Chimera Vector b -> Word -> b
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index Chimera Vector b
bs Word
w)
#ifdef MIN_VERSION_mtl
instance MonadReader Word (Chimera V.Vector) where
ask :: Chimera Vector Word
ask = (Word -> Word) -> Chimera Vector Word
forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate Word -> Word
forall a. a -> a
id
local :: forall a. (Word -> Word) -> Chimera Vector a -> Chimera Vector a
local = (Chimera Vector a -> (Word -> Word) -> Chimera Vector a)
-> (Word -> Word) -> Chimera Vector a -> Chimera Vector a
forall a b c. (a -> b -> c) -> b -> a -> c
flip ((Chimera Vector a -> (Word -> Word) -> Chimera Vector a)
-> (Word -> Word) -> Chimera Vector a -> Chimera Vector a)
-> (Chimera Vector a -> (Word -> Word) -> Chimera Vector a)
-> (Word -> Word)
-> Chimera Vector a
-> Chimera Vector a
forall a b. (a -> b) -> a -> b
$ ((Word -> a) -> Chimera Vector a
forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate ((Word -> a) -> Chimera Vector a)
-> ((Word -> Word) -> Word -> a)
-> (Word -> Word)
-> Chimera Vector a
forall b c a. (b -> c) -> (a -> b) -> a -> c
.) (((Word -> Word) -> Word -> a)
-> (Word -> Word) -> Chimera Vector a)
-> (Chimera Vector a -> (Word -> Word) -> Word -> a)
-> Chimera Vector a
-> (Word -> Word)
-> Chimera Vector a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Word -> a) -> (Word -> Word) -> Word -> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
(.) ((Word -> a) -> (Word -> Word) -> Word -> a)
-> (Chimera Vector a -> Word -> a)
-> Chimera Vector a
-> (Word -> Word)
-> Word
-> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Chimera Vector a -> Word -> a
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index
#endif
#ifdef MIN_VERSION_distributive
instance Distributive (Chimera V.Vector) where
distribute :: forall (f :: * -> *) a.
Functor f =>
f (Chimera Vector a) -> Chimera Vector (f a)
distribute = (Word -> f a) -> Chimera Vector (f a)
forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate ((Word -> f a) -> Chimera Vector (f a))
-> (f (Chimera Vector a) -> Word -> f a)
-> f (Chimera Vector a)
-> Chimera Vector (f a)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Word -> f (Chimera Vector a) -> f a)
-> f (Chimera Vector a) -> Word -> f a
forall a b c. (a -> b -> c) -> b -> a -> c
flip ((Chimera Vector a -> a) -> f (Chimera Vector a) -> f a
forall a b. (a -> b) -> f a -> f b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap ((Chimera Vector a -> a) -> f (Chimera Vector a) -> f a)
-> (Word -> Chimera Vector a -> a)
-> Word
-> f (Chimera Vector a)
-> f a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Chimera Vector a -> Word -> a) -> Word -> Chimera Vector a -> a
forall a b c. (a -> b -> c) -> b -> a -> c
flip Chimera Vector a -> Word -> a
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index)
collect :: forall (f :: * -> *) a b.
Functor f =>
(a -> Chimera Vector b) -> f a -> Chimera Vector (f b)
collect a -> Chimera Vector b
f = (Word -> f b) -> Chimera Vector (f b)
forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate ((Word -> f b) -> Chimera Vector (f b))
-> (f a -> Word -> f b) -> f a -> Chimera Vector (f b)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Word -> f a -> f b) -> f a -> Word -> f b
forall a b c. (a -> b -> c) -> b -> a -> c
flip ((a -> b) -> f a -> f b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
(<$>) ((a -> b) -> f a -> f b) -> (Word -> a -> b) -> Word -> f a -> f b
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ((Chimera Vector b -> b) -> (a -> Chimera Vector b) -> a -> b
forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> Chimera Vector b
f) ((Chimera Vector b -> b) -> a -> b)
-> (Word -> Chimera Vector b -> b) -> Word -> a -> b
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Chimera Vector b -> Word -> b) -> Word -> Chimera Vector b -> b
forall a b c. (a -> b -> c) -> b -> a -> c
flip Chimera Vector b -> Word -> b
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index)
#ifdef MIN_VERSION_adjunctions
instance Rep.Representable (Chimera V.Vector) where
type Rep (Chimera V.Vector) = Word
tabulate :: forall a. (Rep (Chimera Vector) -> a) -> Chimera Vector a
tabulate = (Word -> a) -> Chimera Vector a
(Rep (Chimera Vector) -> a) -> Chimera Vector a
forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate
index :: forall a. Chimera Vector a -> Rep (Chimera Vector) -> a
index = Chimera Vector a -> Word -> a
Chimera Vector a -> Rep (Chimera Vector) -> a
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index
#endif
#endif
bits :: Int
bits :: Int
bits = Word -> Int
fbs (Word
0 :: Word)
tabulate :: G.Vector v a => (Word -> a) -> Chimera v a
tabulate :: forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate Word -> a
f = Identity (Chimera v a) -> Chimera v a
forall a. Identity a -> a
runIdentity (Identity (Chimera v a) -> Chimera v a)
-> Identity (Chimera v a) -> Chimera v a
forall a b. (a -> b) -> a -> b
$ (Word -> Identity a) -> Identity (Chimera v a)
forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
(Word -> m a) -> m (Chimera v a)
tabulateM (a -> Identity a
forall a. a -> Identity a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (a -> Identity a) -> (Word -> a) -> Word -> Identity a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Word -> a
f)
tabulateM
:: forall m v a.
(Monad m, G.Vector v a)
=> (Word -> m a)
-> m (Chimera v a)
tabulateM :: forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
(Word -> m a) -> m (Chimera v a)
tabulateM Word -> m a
f = Vector (v a) -> Chimera v a
forall (v :: * -> *) a. Vector (v a) -> Chimera v a
Chimera (Vector (v a) -> Chimera v a)
-> m (Vector (v a)) -> m (Chimera v a)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Int -> (Int -> m (v a)) -> m (Vector (v a))
forall (m :: * -> *) a.
Monad m =>
Int -> (Int -> m a) -> m (Vector a)
V.generateM (Int
bits Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
1) Int -> m (v a)
tabulateSubVector
where
tabulateSubVector :: Int -> m (v a)
tabulateSubVector :: Int -> m (v a)
tabulateSubVector Int
0 = a -> v a
forall (v :: * -> *) a. Vector v a => a -> v a
G.singleton (a -> v a) -> m a -> m (v a)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Word -> m a
f Word
0
tabulateSubVector Int
i = Int -> (Int -> m a) -> m (v a)
forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
Int -> (Int -> m a) -> m (v a)
G.generateM Int
ii (\Int
j -> Word -> m a
f (Int -> Word
int2word (Int
ii Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
j)))
where
ii :: Int
ii = Int
1 Int -> Int -> Int
forall a. Bits a => a -> Int -> a
`unsafeShiftL` (Int
i Int -> Int -> Int
forall a. Num a => a -> a -> a
- Int
1)
{-# SPECIALIZE tabulateM :: G.Vector v a => (Word -> Identity a) -> Identity (Chimera v a) #-}
tabulateFix :: G.Vector v a => ((Word -> a) -> Word -> a) -> Chimera v a
tabulateFix :: forall (v :: * -> *) a.
Vector v a =>
((Word -> a) -> Word -> a) -> Chimera v a
tabulateFix (Word -> a) -> Word -> a
uf = Identity (Chimera v a) -> Chimera v a
forall a. Identity a -> a
runIdentity (Identity (Chimera v a) -> Chimera v a)
-> Identity (Chimera v a) -> Chimera v a
forall a b. (a -> b) -> a -> b
$ ((Word -> Identity a) -> Word -> Identity a)
-> Identity (Chimera v a)
forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
((Word -> m a) -> Word -> m a) -> m (Chimera v a)
tabulateFixM ((a -> Identity a
forall a. a -> Identity a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (a -> Identity a) -> (Word -> a) -> Word -> Identity a
forall b c a. (b -> c) -> (a -> b) -> a -> c
.) ((Word -> a) -> Word -> Identity a)
-> ((Word -> Identity a) -> Word -> a)
-> (Word -> Identity a)
-> Word
-> Identity a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Word -> a) -> Word -> a
uf ((Word -> a) -> Word -> a)
-> ((Word -> Identity a) -> Word -> a)
-> (Word -> Identity a)
-> Word
-> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Identity a -> a
forall a. Identity a -> a
runIdentity (Identity a -> a) -> (Word -> Identity a) -> Word -> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
.))
tabulateFix' :: G.Vector v a => ((Word -> a) -> Word -> a) -> Chimera v a
tabulateFix' :: forall (v :: * -> *) a.
Vector v a =>
((Word -> a) -> Word -> a) -> Chimera v a
tabulateFix' (Word -> a) -> Word -> a
uf = Identity (Chimera v a) -> Chimera v a
forall a. Identity a -> a
runIdentity (Identity (Chimera v a) -> Chimera v a)
-> Identity (Chimera v a) -> Chimera v a
forall a b. (a -> b) -> a -> b
$ ((Word -> Identity a) -> Word -> Identity a)
-> Identity (Chimera v a)
forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
((Word -> m a) -> Word -> m a) -> m (Chimera v a)
tabulateFixM' ((a -> Identity a
forall a. a -> Identity a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (a -> Identity a) -> (Word -> a) -> Word -> Identity a
forall b c a. (b -> c) -> (a -> b) -> a -> c
.) ((Word -> a) -> Word -> Identity a)
-> ((Word -> Identity a) -> Word -> a)
-> (Word -> Identity a)
-> Word
-> Identity a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Word -> a) -> Word -> a
uf ((Word -> a) -> Word -> a)
-> ((Word -> Identity a) -> Word -> a)
-> (Word -> Identity a)
-> Word
-> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Identity a -> a
forall a. Identity a -> a
runIdentity (Identity a -> a) -> (Word -> Identity a) -> Word -> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
.))
tabulateFixM
:: forall m v a.
(Monad m, G.Vector v a)
=> ((Word -> m a) -> Word -> m a)
-> m (Chimera v a)
tabulateFixM :: forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
((Word -> m a) -> Word -> m a) -> m (Chimera v a)
tabulateFixM = Strategy -> ((Word -> m a) -> Word -> m a) -> m (Chimera v a)
forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
Strategy -> ((Word -> m a) -> Word -> m a) -> m (Chimera v a)
tabulateFixM_ Strategy
Downwards
{-# SPECIALIZE tabulateFixM :: G.Vector v a => ((Word -> Identity a) -> Word -> Identity a) -> Identity (Chimera v a) #-}
tabulateFixM'
:: forall m v a.
(Monad m, G.Vector v a)
=> ((Word -> m a) -> Word -> m a)
-> m (Chimera v a)
tabulateFixM' :: forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
((Word -> m a) -> Word -> m a) -> m (Chimera v a)
tabulateFixM' = Strategy -> ((Word -> m a) -> Word -> m a) -> m (Chimera v a)
forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
Strategy -> ((Word -> m a) -> Word -> m a) -> m (Chimera v a)
tabulateFixM_ Strategy
Full
data Strategy = Full | Downwards
tabulateFixM_
:: forall m v a.
(Monad m, G.Vector v a)
=> Strategy
-> ((Word -> m a) -> Word -> m a)
-> m (Chimera v a)
tabulateFixM_ :: forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
Strategy -> ((Word -> m a) -> Word -> m a) -> m (Chimera v a)
tabulateFixM_ Strategy
strat (Word -> m a) -> Word -> m a
f = m (Chimera v a)
result
where
result :: m (Chimera v a)
result :: m (Chimera v a)
result = Vector (v a) -> Chimera v a
forall (v :: * -> *) a. Vector (v a) -> Chimera v a
Chimera (Vector (v a) -> Chimera v a)
-> m (Vector (v a)) -> m (Chimera v a)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Int -> (Int -> m (v a)) -> m (Vector (v a))
forall (m :: * -> *) a.
Monad m =>
Int -> (Int -> m a) -> m (Vector a)
V.generateM (Int
bits Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
1) Int -> m (v a)
tabulateSubVector
tabulateSubVector :: Int -> m (v a)
tabulateSubVector :: Int -> m (v a)
tabulateSubVector Int
0 = a -> v a
forall (v :: * -> *) a. Vector v a => a -> v a
G.singleton (a -> v a) -> m a -> m (v a)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> case Strategy
strat of
Strategy
Downwards -> ((Word -> m a) -> Word -> m a) -> Word -> m a
forall a. (a -> a) -> a
fix (Word -> m a) -> Word -> m a
f Word
0
Strategy
Full -> (Word -> m a) -> Word -> m a
f (\Word
k -> (Chimera v a -> Word -> a) -> Word -> Chimera v a -> a
forall a b c. (a -> b -> c) -> b -> a -> c
flip Chimera v a -> Word -> a
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index Word
k (Chimera v a -> a) -> m (Chimera v a) -> m a
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> m (Chimera v a)
result) Word
0
tabulateSubVector Int
i = m (v a)
subResult
where
subResult :: m (v a)
subResult = Int -> (Int -> m a) -> m (v a)
forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
Int -> (Int -> m a) -> m (v a)
G.generateM Int
ii (\Int
j -> (Word -> m a) -> Word -> m a
f Word -> m a
fixF (Int -> Word
int2word (Int
ii Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
j)))
subResultBoxed :: m (Vector a)
subResultBoxed = Int -> (Int -> m a) -> m (Vector a)
forall (m :: * -> *) a.
Monad m =>
Int -> (Int -> m a) -> m (Vector a)
V.generateM Int
ii (\Int
j -> (Word -> m a) -> Word -> m a
f Word -> m a
fixF (Int -> Word
int2word (Int
ii Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
j)))
ii :: Int
ii = Int
1 Int -> Int -> Int
forall a. Bits a => a -> Int -> a
`unsafeShiftL` (Int
i Int -> Int -> Int
forall a. Num a => a -> a -> a
- Int
1)
fixF :: Word -> m a
fixF :: Word -> m a
fixF Word
k
| Word
k Word -> Word -> Bool
forall a. Ord a => a -> a -> Bool
< Int -> Word
int2word Int
ii
= (Chimera v a -> Word -> a) -> Word -> Chimera v a -> a
forall a b c. (a -> b -> c) -> b -> a -> c
flip Chimera v a -> Word -> a
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index Word
k (Chimera v a -> a) -> m (Chimera v a) -> m a
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> m (Chimera v a)
result
| Word
k Word -> Word -> Bool
forall a. Ord a => a -> a -> Bool
<= Int -> Word
int2word Int
ii Word -> Int -> Word
forall a. Bits a => a -> Int -> a
`shiftL` Int
1 Word -> Word -> Word
forall a. Num a => a -> a -> a
- Word
1
= (Vector a -> Int -> a
forall a. Vector a -> Int -> a
`V.unsafeIndex` (Word -> Int
word2int Word
k Int -> Int -> Int
forall a. Num a => a -> a -> a
- Int
ii)) (Vector a -> a) -> m (Vector a) -> m a
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> m (Vector a)
subResultBoxed
| Bool
otherwise
= case Strategy
strat of
Strategy
Downwards -> (Word -> m a) -> Word -> m a
f Word -> m a
fixF Word
k
Strategy
Full -> (Chimera v a -> Word -> a) -> Word -> Chimera v a -> a
forall a b c. (a -> b -> c) -> b -> a -> c
flip Chimera v a -> Word -> a
forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index Word
k (Chimera v a -> a) -> m (Chimera v a) -> m a
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> m (Chimera v a)
result
iterate :: G.Vector v a => (a -> a) -> a -> Chimera v a
iterate :: forall (v :: * -> *) a. Vector v a => (a -> a) -> a -> Chimera v a
iterate a -> a
f = Identity (Chimera v a) -> Chimera v a
forall a. Identity a -> a
runIdentity (Identity (Chimera v a) -> Chimera v a)
-> (a -> Identity (Chimera v a)) -> a -> Chimera v a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (a -> Identity a) -> a -> Identity (Chimera v a)
forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
(a -> m a) -> a -> m (Chimera v a)
iterateM (a -> Identity a
forall a. a -> Identity a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (a -> Identity a) -> (a -> a) -> a -> Identity a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> a
f)
iterateM :: forall m v a. (Monad m, G.Vector v a) => (a -> m a) -> a -> m (Chimera v a)
iterateM :: forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
(a -> m a) -> a -> m (Chimera v a)
iterateM a -> m a
f a
seed = do
a
nextSeed <- a -> m a
f a
seed
let z :: v a
z = a -> v a
forall (v :: * -> *) a. Vector v a => a -> v a
G.singleton a
seed
Vector (v a)
zs <- Int -> (v a -> m (v a)) -> v a -> m (Vector (v a))
forall (m :: * -> *) a.
Monad m =>
Int -> (a -> m a) -> a -> m (Vector a)
V.iterateNM Int
bits v a -> m (v a)
go (a -> v a
forall (v :: * -> *) a. Vector v a => a -> v a
G.singleton a
nextSeed)
Chimera v a -> m (Chimera v a)
forall a. a -> m a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Chimera v a -> m (Chimera v a)) -> Chimera v a -> m (Chimera v a)
forall a b. (a -> b) -> a -> b
$ Vector (v a) -> Chimera v a
forall (v :: * -> *) a. Vector (v a) -> Chimera v a
Chimera (Vector (v a) -> Chimera v a) -> Vector (v a) -> Chimera v a
forall a b. (a -> b) -> a -> b
$ v a
z v a -> Vector (v a) -> Vector (v a)
forall a. a -> Vector a -> Vector a
`V.cons` Vector (v a)
zs
where
go :: v a -> m (v a)
go :: v a -> m (v a)
go v a
vec = do
a
nextSeed <- a -> m a
f (v a -> a
forall (v :: * -> *) a. Vector v a => v a -> a
G.unsafeLast v a
vec)
Int -> (a -> m a) -> a -> m (v a)
forall (m :: * -> *) (v :: * -> *) a.
(Monad m, Vector v a) =>
Int -> (a -> m a) -> a -> m (v a)
G.iterateNM (v a -> Int
forall (v :: * -> *) a. Vector v a => v a -> Int
G.length v a
vec Int -> Int -> Int
forall a. Bits a => a -> Int -> a
`shiftL` Int
1) a -> m a
f a
nextSeed
{-# SPECIALIZE iterateM :: G.Vector v a => (a -> Identity a) -> a -> Identity (Chimera v a) #-}
index :: G.Vector v a => Chimera v a -> Word -> a
index :: forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index (Chimera Vector (v a)
vs) Word
i =
(Vector (v a)
vs Vector (v a) -> Int -> v a
forall a. Vector a -> Int -> a
`V.unsafeIndex` (Word -> Int
fbs Word
i Int -> Int -> Int
forall a. Num a => a -> a -> a
- Int
lz))
v a -> Int -> a
forall (v :: * -> *) a. Vector v a => v a -> Int -> a
`G.unsafeIndex`
Word -> Int
word2int (Word
i Word -> Word -> Word
forall a. Bits a => a -> a -> a
.&. Word -> Word
forall a. Bits a => a -> a
complement ((Word
1 Word -> Int -> Word
forall a. Bits a => a -> Int -> a
`shiftL` (Word -> Int
fbs Word
i Int -> Int -> Int
forall a. Num a => a -> a -> a
- Int
1)) Word -> Int -> Word
forall a. Bits a => a -> Int -> a
`unsafeShiftR` Int
lz))
where
lz :: Int
!lz :: Int
lz = Word -> Int
word2int (Word -> Word
clz Word
i)
{-# INLINE index #-}
toList :: G.Vector v a => Chimera v a -> [a]
toList :: forall (v :: * -> *) a. Vector v a => Chimera v a -> [a]
toList (Chimera Vector (v a)
vs) = (v a -> [a]) -> Vector (v a) -> [a]
forall m a. Monoid m => (a -> m) -> Vector a -> m
forall (t :: * -> *) m a.
(Foldable t, Monoid m) =>
(a -> m) -> t a -> m
foldMap v a -> [a]
forall (v :: * -> *) a. Vector v a => v a -> [a]
G.toList Vector (v a)
vs
cycle :: G.Vector v a => v a -> Chimera v a
cycle :: forall (v :: * -> *) a. Vector v a => v a -> Chimera v a
cycle v a
vec = case Word
l of
Word
0 -> [Char] -> Chimera v a
forall a. HasCallStack => [Char] -> a
error [Char]
"Data.Chimera.cycle: empty list"
Word
_ -> (Word -> a) -> Chimera v a
forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate (v a -> Int -> a
forall (v :: * -> *) a. Vector v a => v a -> Int -> a
G.unsafeIndex v a
vec (Int -> a) -> (Word -> Int) -> Word -> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Word -> Int
word2int (Word -> Int) -> (Word -> Word) -> Word -> Int
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Word -> Word -> Word
forall a. Integral a => a -> a -> a
`rem` Word
l))
where
l :: Word
l = Int -> Word
int2word (Int -> Word) -> Int -> Word
forall a b. (a -> b) -> a -> b
$ v a -> Int
forall (v :: * -> *) a. Vector v a => v a -> Int
G.length v a
vec
memoize :: (Word -> a) -> (Word -> a)
memoize :: forall a. (Word -> a) -> Word -> a
memoize = forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index @V.Vector (Chimera Vector a -> Word -> a)
-> ((Word -> a) -> Chimera Vector a) -> (Word -> a) -> Word -> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Word -> a) -> Chimera Vector a
forall (v :: * -> *) a. Vector v a => (Word -> a) -> Chimera v a
tabulate
memoizeFix :: ((Word -> a) -> Word -> a) -> (Word -> a)
memoizeFix :: forall a. ((Word -> a) -> Word -> a) -> Word -> a
memoizeFix = forall (v :: * -> *) a. Vector v a => Chimera v a -> Word -> a
index @V.Vector (Chimera Vector a -> Word -> a)
-> (((Word -> a) -> Word -> a) -> Chimera Vector a)
-> ((Word -> a) -> Word -> a)
-> Word
-> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ((Word -> a) -> Word -> a) -> Chimera Vector a
forall (v :: * -> *) a.
Vector v a =>
((Word -> a) -> Word -> a) -> Chimera v a
tabulateFix
mapSubvectors
:: (G.Vector u a, G.Vector v b)
=> (u a -> v b)
-> Chimera u a
-> Chimera v b
mapSubvectors :: forall (u :: * -> *) a (v :: * -> *) b.
(Vector u a, Vector v b) =>
(u a -> v b) -> Chimera u a -> Chimera v b
mapSubvectors u a -> v b
f (Chimera Vector (u a)
bs) = Vector (v b) -> Chimera v b
forall (v :: * -> *) a. Vector (v a) -> Chimera v a
Chimera ((u a -> v b) -> Vector (u a) -> Vector (v b)
forall a b. (a -> b) -> Vector a -> Vector b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap u a -> v b
f Vector (u a)
bs)
zipSubvectors
:: (G.Vector u a, G.Vector v b, G.Vector w c)
=> (u a -> v b -> w c)
-> Chimera u a
-> Chimera v b
-> Chimera w c
zipSubvectors :: forall (u :: * -> *) a (v :: * -> *) b (w :: * -> *) c.
(Vector u a, Vector v b, Vector w c) =>
(u a -> v b -> w c) -> Chimera u a -> Chimera v b -> Chimera w c
zipSubvectors u a -> v b -> w c
f (Chimera Vector (u a)
bs1) (Chimera Vector (v b)
bs2) = Vector (w c) -> Chimera w c
forall (v :: * -> *) a. Vector (v a) -> Chimera v a
Chimera ((u a -> v b -> w c) -> Vector (u a) -> Vector (v b) -> Vector (w c)
forall a b c. (a -> b -> c) -> Vector a -> Vector b -> Vector c
forall (m :: * -> *) a b c.
MonadZip m =>
(a -> b -> c) -> m a -> m b -> m c
mzipWith u a -> v b -> w c
f Vector (u a)
bs1 Vector (v b)
bs2)