module Data.Array.Repa.Repr.Delayed
( D, Array(..)
, fromFunction, toFunction
, delay)
where
import Data.Array.Repa.Eval.Elt
import Data.Array.Repa.Eval.Cursored
import Data.Array.Repa.Eval.Chunked
import Data.Array.Repa.Eval.Fill
import Data.Array.Repa.Index
import Data.Array.Repa.Shape
import Data.Array.Repa.Base
import GHC.Exts
data D
data instance Array D sh e
= ADelayed
sh
(sh -> e)
instance Repr D a where
index (ADelayed _ f) ix = f ix
linearIndex (ADelayed sh f) ix = f (fromIndex sh ix)
extent (ADelayed sh _)
= sh
deepSeqArray (ADelayed sh f) y
= sh `deepSeq` f `seq` y
instance (Fillable r2 e, Shape sh) => Fill D r2 sh e where
fillP (ADelayed sh getElem) marr
= fillChunkedP (size sh) (unsafeWriteMArr marr) (getElem . fromIndex sh)
fillS (ADelayed sh getElem) marr
= fillChunkedS (size sh) (unsafeWriteMArr marr) (getElem . fromIndex sh)
instance (Fillable r2 e, Elt e) => FillRange D r2 DIM2 e where
fillRangeP (ADelayed (Z :. _h :. w) getElem) marr
(Z :. y0 :. x0) (Z :. y1 :. x1)
= fillBlock2P (unsafeWriteMArr marr)
getElem
w x0 y0 x1 y1
fillRangeS (ADelayed (Z :. _h :. (I# w)) getElem) marr
(Z :. (I# y0) :. (I# x0)) (Z :. (I# y1) :. (I# x1))
= fillBlock2S (unsafeWriteMArr marr)
getElem
w x0 y0 x1 y1
fromFunction :: sh -> (sh -> a) -> Array D sh a
fromFunction sh f
= ADelayed sh f
toFunction
:: (Shape sh, Repr r1 a)
=> Array r1 sh a -> (sh, sh -> a)
toFunction arr
= case delay arr of
ADelayed sh f -> (sh, f)
delay :: (Shape sh, Repr r e)
=> Array r sh e -> Array D sh e
delay arr = ADelayed (extent arr) (index arr)