module Graphics.Image.Interface.Repa.Storable (
RSS(..), RPS(..), Image(..)
) where
import qualified Data.Array.Repa.Eval as R
import qualified Data.Array.Repa.Repr.ForeignPtr as R
import Data.Typeable (Typeable)
import Data.Vector.Unboxed (Unbox)
import Foreign.Storable
import Graphics.Image.Interface as I
import Graphics.Image.Interface.Repa.Generic
import qualified Graphics.Image.Interface.Vector.Storable as IVS
import Prelude as P
data RSS = RSS deriving Typeable
data RPS = RPS deriving Typeable
instance Show RSS where
show _ = "RepaSequentialStorable"
instance Show RPS where
show _ = "RepaParallelStorable"
instance SuperClass RSS cs e => BaseArray RSS cs e where
type SuperClass RSS cs e =
(ColorSpace cs e, Unbox (Pixel cs e),
Storable e, Storable (Pixel cs e),
R.Elt e, R.Elt (Pixel cs e))
newtype Image RSS cs e = SSImage (RImage R.F (Pixel cs e))
dims (SSImage img) = dimsR img
instance BaseArray RSS cs e => Array RSS cs e where
type Manifest RSS = IVS.VS
type Vector RSS = Vector IVS.VS
makeImage !sz f = SSImage (makeImageR sz f)
makeImageWindowed !sz !wIx !wSz f = SSImage . makeImageWindowedR sz wIx wSz f
scalar = SSImage . scalarR
index00 (SSImage img) = index00R img
map f (SSImage img) = SSImage (mapR f img)
imap f (SSImage img) = SSImage (imapR f img)
zipWith f (SSImage img1) (SSImage img2) = SSImage (zipWithR f img1 img2)
izipWith f (SSImage img1) (SSImage img2) = SSImage (izipWithR f img1 img2)
traverse (SSImage img) f g = SSImage (traverseR img f g)
traverse2 (SSImage img1) (SSImage img2) f g = SSImage (traverse2R img1 img2 f g)
transpose (SSImage img) = SSImage (transposeR img)
backpermute !sz g (SSImage img) = SSImage (backpermuteR sz g img)
fromLists = SSImage . fromListsR
fold f !px0 (SSImage img) = foldR Sequential f px0 img
foldIx f !px0 (SSImage img) = foldIxR Sequential f px0 img
eq (SSImage img1) (SSImage img2) = eqR Sequential img1 img2
compute (SSImage img) = SSImage (computeR Sequential img)
(|*|) (SSImage img1) (SSImage img2) = SSImage (multR Sequential img1 img2)
toManifest (SSImage img) = fromVector (dimsR img) $ toVectorStorableR Sequential img
toVector (SSImage img) = toVectorStorableR Sequential img
fromVector !sz = SSImage . fromVectorStorableR sz
instance SuperClass RPS cs e => BaseArray RPS cs e where
type SuperClass RPS cs e =
(ColorSpace cs e,
Storable e, Storable (Pixel cs e), Unbox (Pixel cs e),
R.Elt e, R.Elt (Pixel cs e))
newtype Image RPS cs e = PSImage (RImage R.F (Pixel cs e))
dims (PSImage img) = dimsR img
instance BaseArray RPS cs e => Array RPS cs e where
type Manifest RPS = IVS.VS
type Vector RPS = Vector IVS.VS
makeImage !sz f = PSImage (makeImageR sz f)
makeImageWindowed !sz !wIx !wSz f = PSImage . makeImageWindowedR sz wIx wSz f
scalar = PSImage . scalarR
index00 (PSImage img) = index00R img
map f (PSImage img) = PSImage (mapR f img)
imap f (PSImage img) = PSImage (imapR f img)
zipWith f (PSImage img1) (PSImage img2) = PSImage (zipWithR f img1 img2)
izipWith f (PSImage img1) (PSImage img2) = PSImage (izipWithR f img1 img2)
traverse (PSImage img) f g = PSImage (traverseR img f g)
traverse2 (PSImage img1) (PSImage img2) f g = PSImage (traverse2R img1 img2 f g)
transpose (PSImage img) = PSImage (transposeR img)
backpermute !sz g (PSImage img) = PSImage (backpermuteR sz g img)
fromLists = PSImage . fromListsR
fold f !px0 (PSImage img) = foldR Parallel f px0 img
foldIx f !px0 (PSImage img) = foldIxR Parallel f px0 img
eq (PSImage img1) (PSImage img2) = eqR Parallel img1 img2
compute (PSImage img) = PSImage (computeR Parallel img)
(|*|) (PSImage img1) (PSImage img2) = PSImage (multR Parallel img1 img2)
toManifest (PSImage img) = fromVector (dimsR img) $ toVectorStorableR Parallel img
toVector (PSImage img) = toVectorStorableR Parallel img
fromVector !sz = PSImage . fromVectorStorableR sz