module Graphics.Image.Interface.Vector.Storable (
VS(..), S, Image(..), fromStorableVector, toStorableVector, fromIx, toIx, checkDims
) where
import Prelude hiding (map, zipWith)
#if !MIN_VERSION_base(4,8,0)
import Data.Functor
#endif
import qualified Data.Vector.Storable as VS
import Graphics.Image.Interface as I
import Graphics.Image.Interface.Vector.Generic
data VS = VS
data S
type instance Repr (V S) = VS.Vector
instance Show S where
show _ = "Storable"
instance Show VS where
show _ = "VectorStorable"
instance SuperClass VS cs e => BaseArray VS cs e where
type SuperClass VS cs e =
(ColorSpace cs e, Num (Pixel cs e), VS.Storable (Pixel cs e))
data Image VS cs e = VSImage !(Image (V S) cs e)
dims (VSImage img) = dims img
instance (MArray VS cs e, BaseArray VS cs e) => Array VS cs e where
type Manifest VS = VS
makeImage !sh = VSImage . makeImage sh
makeImageWindowed !sh !window f g = VSImage $ makeImageWindowed sh window f g
scalar = VSImage . scalar
index00 (VSImage img) = index00 img
map f (VSImage img) = VSImage $ I.map f img
imap f (VSImage img) = VSImage $ I.imap f img
zipWith f (VSImage img1) (VSImage img2) = VSImage $ I.zipWith f img1 img2
izipWith f (VSImage img1) (VSImage img2) = VSImage $ I.izipWith f img1 img2
traverse (VSImage img) f g = VSImage $ I.traverse img f g
traverse2 (VSImage img1) (VSImage img2) f g = VSImage $ I.traverse2 img1 img2 f g
transpose (VSImage img) = VSImage $ I.transpose img
backpermute !sz f (VSImage img) = VSImage $ I.backpermute sz f img
fromLists = VSImage . I.fromLists
fold f !px0 (VSImage img) = fold f px0 img
foldIx f !px0 (VSImage img) = foldIx f px0 img
(|*|) (VSImage img1) (VSImage img2) = VSImage (img1 |*| img2)
eq (VSImage img1) (VSImage img2) = img1 == img2
compute (VSImage img) = VSImage $! compute img
toManifest = id
instance BaseArray VS cs e => MArray VS cs e where
data MImage s VS cs e = MVSImage (MImage s (V S) cs e)
unsafeIndex (VSImage img) = unsafeIndex img
deepSeqImage (VSImage img) = deepSeqImage img
foldl f !px0 (VSImage img) = I.foldl f px0 img
foldr f !px0 (VSImage img) = I.foldr f px0 img
makeImageM !sh f = VSImage <$> makeImageM sh f
mapM f (VSImage img) = VSImage <$> I.mapM f img
mapM_ f (VSImage img) = I.mapM_ f img
foldM f !px0 (VSImage img) = I.foldM f px0 img
foldM_ f !px0 (VSImage img) = I.foldM_ f px0 img
mdims (MVSImage mimg) = mdims mimg
thaw (VSImage img) = MVSImage <$> I.thaw img
freeze (MVSImage img) = VSImage <$> I.freeze img
new !ix = MVSImage <$> I.new ix
read (MVSImage img) = I.read img
write (MVSImage img) = I.write img
swap (MVSImage img) = I.swap img
toStorableVector :: Array VS cs e => Image VS cs e -> VS.Vector (Pixel cs e)
toStorableVector (VSImage img) = toVector img
fromStorableVector :: Array VS cs e => (Int, Int) -> VS.Vector (Pixel cs e) -> Image VS cs e
fromStorableVector !sz !v = VSImage $ fromVector sz v