module Data.Primitive.SIMD.Word8X16 (Word8X16) where
import Data.Primitive.SIMD.Class
import GHC.Word
import GHC.Types
import GHC.Prim
import GHC.Ptr
import GHC.ST
import Foreign.Storable
import Control.Monad.Primitive
import Data.Primitive.Types
import Data.Primitive.ByteArray
import Data.Primitive.Addr
import Data.Monoid
import Data.Typeable
import qualified Data.Vector.Primitive as PV
import qualified Data.Vector.Primitive.Mutable as PMV
import Data.Vector.Unboxed (Unbox)
import qualified Data.Vector.Unboxed as UV
import Data.Vector.Generic (Vector(..))
import Data.Vector.Generic.Mutable (MVector(..))
data Word8X16 = Word8X16 Word8X16# deriving Typeable
abs' :: Word8 -> Word8
abs' (W8# x) = W8# (abs# x)
abs# :: Word# -> Word#
abs# x = case abs (W8# x) of
W8# y -> y
signum' :: Word8 -> Word8
signum' (W8# x) = W8# (signum# x)
signum# :: Word# -> Word#
signum# x = case signum (W8# x) of
W8# y -> y
instance Eq Word8X16 where
a == b = case unpackWord8X16 a of
(x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11, x12, x13, x14, x15, x16) -> case unpackWord8X16 b of
(y1, y2, y3, y4, y5, y6, y7, y8, y9, y10, y11, y12, y13, y14, y15, y16) -> x1 == y1 && x2 == y2 && x3 == y3 && x4 == y4 && x5 == y5 && x6 == y6 && x7 == y7 && x8 == y8 && x9 == y9 && x10 == y10 && x11 == y11 && x12 == y12 && x13 == y13 && x14 == y14 && x15 == y15 && x16 == y16
instance Ord Word8X16 where
a `compare` b = case unpackWord8X16 a of
(x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11, x12, x13, x14, x15, x16) -> case unpackWord8X16 b of
(y1, y2, y3, y4, y5, y6, y7, y8, y9, y10, y11, y12, y13, y14, y15, y16) -> x1 `compare` y1 <> x2 `compare` y2 <> x3 `compare` y3 <> x4 `compare` y4 <> x5 `compare` y5 <> x6 `compare` y6 <> x7 `compare` y7 <> x8 `compare` y8 <> x9 `compare` y9 <> x10 `compare` y10 <> x11 `compare` y11 <> x12 `compare` y12 <> x13 `compare` y13 <> x14 `compare` y14 <> x15 `compare` y15 <> x16 `compare` y16
instance Show Word8X16 where
showsPrec _ a s = case unpackWord8X16 a of
(x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11, x12, x13, x14, x15, x16) -> "Word8X16 (" ++ shows x1 (", " ++ shows x2 (", " ++ shows x3 (", " ++ shows x4 (", " ++ shows x5 (", " ++ shows x6 (", " ++ shows x7 (", " ++ shows x8 (", " ++ shows x9 (", " ++ shows x10 (", " ++ shows x11 (", " ++ shows x12 (", " ++ shows x13 (", " ++ shows x14 (", " ++ shows x15 (", " ++ shows x16 (")" ++ s))))))))))))))))
instance Num Word8X16 where
(+) = plusWord8X16
() = minusWord8X16
(*) = timesWord8X16
negate = mapVector negate
abs = mapVector abs'
signum = mapVector signum'
fromInteger = broadcastVector . fromInteger
instance Bounded Word8X16 where
minBound = broadcastVector minBound
maxBound = broadcastVector maxBound
instance Storable Word8X16 where
sizeOf x = vectorSize x * elementSize x
alignment = sizeOf
peek (Ptr a) = readOffAddr (Addr a) 0
poke (Ptr a) = writeOffAddr (Addr a) 0
instance SIMDVector Word8X16 where
type Elem Word8X16 = Word8
type ElemTuple Word8X16 = (Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8)
nullVector = broadcastVector 0
vectorSize _ = 16
elementSize _ = 1
broadcastVector = broadcastWord8X16
unsafeInsertVector = unsafeInsertWord8X16
packVector = packWord8X16
unpackVector = unpackWord8X16
mapVector = mapWord8X16
zipVector = zipWord8X16
foldVector = foldWord8X16
instance SIMDIntVector Word8X16 where
quotVector = quotWord8X16
remVector = remWord8X16
instance Prim Word8X16 where
sizeOf# a = let !(I# x) = sizeOf a in x
alignment# a = let !(I# x) = alignment a in x
indexByteArray# ba i = indexWord8X16Array (ByteArray ba) (I# i)
readByteArray# mba i s = let (ST r) = readWord8X16Array (MutableByteArray mba) (I# i) in r s
writeByteArray# mba i v s = let (ST r) = writeWord8X16Array (MutableByteArray mba) (I# i) v in case r s of { (# s', _ #) -> s' }
setByteArray# mba off n v s = let (ST r) = setByteArrayGeneric (MutableByteArray mba) (I# off) (I# n) v in case r s of { (# s', _ #) -> s' }
indexOffAddr# addr i = indexWord8X16OffAddr (Addr addr) (I# i)
readOffAddr# addr i s = let (ST r) = readWord8X16OffAddr (Addr addr) (I# i) in r s
writeOffAddr# addr i v s = let (ST r) = writeWord8X16OffAddr (Addr addr) (I# i) v in case r s of { (# s', _ #) -> s' }
setOffAddr# addr off n v s = let (ST r) = setOffAddrGeneric (Addr addr) (I# off) (I# n) v in case r s of { (# s', _ #) -> s' }
newtype instance UV.Vector Word8X16 = V_Word8X16 (PV.Vector Word8X16)
newtype instance UV.MVector s Word8X16 = MV_Word8X16 (PMV.MVector s Word8X16)
instance Vector UV.Vector Word8X16 where
basicUnsafeFreeze (MV_Word8X16 v) = V_Word8X16 <$> PV.unsafeFreeze v
basicUnsafeThaw (V_Word8X16 v) = MV_Word8X16 <$> PV.unsafeThaw v
basicLength (V_Word8X16 v) = PV.length v
basicUnsafeSlice start len (V_Word8X16 v) = V_Word8X16(PV.unsafeSlice start len v)
basicUnsafeIndexM (V_Word8X16 v) = PV.unsafeIndexM v
basicUnsafeCopy (MV_Word8X16 m) (V_Word8X16 v) = PV.unsafeCopy m v
elemseq _ = seq
instance MVector UV.MVector Word8X16 where
basicLength (MV_Word8X16 v) = PMV.length v
basicUnsafeSlice start len (MV_Word8X16 v) = MV_Word8X16(PMV.unsafeSlice start len v)
basicOverlaps (MV_Word8X16 v) (MV_Word8X16 w) = PMV.overlaps v w
basicUnsafeNew len = MV_Word8X16 <$> PMV.unsafeNew len
#if MIN_VERSION_vector(0,11,0)
basicInitialize (MV_Word8X16 v) = basicInitialize v
#endif
basicUnsafeRead (MV_Word8X16 v) = PMV.unsafeRead v
basicUnsafeWrite (MV_Word8X16 v) = PMV.unsafeWrite v
instance Unbox Word8X16
broadcastWord8X16 :: Word8 -> Word8X16
broadcastWord8X16 (W8# x) = Word8X16 (broadcastWord8X16# x)
packWord8X16 :: (Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8) -> Word8X16
packWord8X16 (W8# x1, W8# x2, W8# x3, W8# x4, W8# x5, W8# x6, W8# x7, W8# x8, W8# x9, W8# x10, W8# x11, W8# x12, W8# x13, W8# x14, W8# x15, W8# x16) = Word8X16 (packWord8X16# (# x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11, x12, x13, x14, x15, x16 #))
unpackWord8X16 :: Word8X16 -> (Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8, Word8)
unpackWord8X16 (Word8X16 m1) = case unpackWord8X16# m1 of
(# x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11, x12, x13, x14, x15, x16 #) -> (W8# x1, W8# x2, W8# x3, W8# x4, W8# x5, W8# x6, W8# x7, W8# x8, W8# x9, W8# x10, W8# x11, W8# x12, W8# x13, W8# x14, W8# x15, W8# x16)
unsafeInsertWord8X16 :: Word8X16 -> Word8 -> Int -> Word8X16
unsafeInsertWord8X16 (Word8X16 m1) (W8# y) _i@(I# ip) = Word8X16 (insertWord8X16# m1 y (ip -# 0#))
mapWord8X16 :: (Word8 -> Word8) -> Word8X16 -> Word8X16
mapWord8X16 f = mapWord8X16# (\ x -> case f (W8# x) of { W8# y -> y})
mapWord8X16# :: (Word# -> Word#) -> Word8X16 -> Word8X16
mapWord8X16# f = \ v -> case unpackWord8X16 v of
(W8# x1, W8# x2, W8# x3, W8# x4, W8# x5, W8# x6, W8# x7, W8# x8, W8# x9, W8# x10, W8# x11, W8# x12, W8# x13, W8# x14, W8# x15, W8# x16) -> packWord8X16 (W8# (f x1), W8# (f x2), W8# (f x3), W8# (f x4), W8# (f x5), W8# (f x6), W8# (f x7), W8# (f x8), W8# (f x9), W8# (f x10), W8# (f x11), W8# (f x12), W8# (f x13), W8# (f x14), W8# (f x15), W8# (f x16))
zipWord8X16 :: (Word8 -> Word8 -> Word8) -> Word8X16 -> Word8X16 -> Word8X16
zipWord8X16 f = \ v1 v2 -> case unpackWord8X16 v1 of
(x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11, x12, x13, x14, x15, x16) -> case unpackWord8X16 v2 of
(y1, y2, y3, y4, y5, y6, y7, y8, y9, y10, y11, y12, y13, y14, y15, y16) -> packWord8X16 (f x1 y1, f x2 y2, f x3 y3, f x4 y4, f x5 y5, f x6 y6, f x7 y7, f x8 y8, f x9 y9, f x10 y10, f x11 y11, f x12 y12, f x13 y13, f x14 y14, f x15 y15, f x16 y16)
foldWord8X16 :: (Word8 -> Word8 -> Word8) -> Word8X16 -> Word8
foldWord8X16 f' = \ v -> case unpackWord8X16 v of
(x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11, x12, x13, x14, x15, x16) -> x1 `f` x2 `f` x3 `f` x4 `f` x5 `f` x6 `f` x7 `f` x8 `f` x9 `f` x10 `f` x11 `f` x12 `f` x13 `f` x14 `f` x15 `f` x16
where f !x !y = f' x y
plusWord8X16 :: Word8X16 -> Word8X16 -> Word8X16
plusWord8X16 (Word8X16 m1_1) (Word8X16 m1_2) = Word8X16 (plusWord8X16# m1_1 m1_2)
minusWord8X16 :: Word8X16 -> Word8X16 -> Word8X16
minusWord8X16 (Word8X16 m1_1) (Word8X16 m1_2) = Word8X16 (minusWord8X16# m1_1 m1_2)
timesWord8X16 :: Word8X16 -> Word8X16 -> Word8X16
timesWord8X16 (Word8X16 m1_1) (Word8X16 m1_2) = Word8X16 (timesWord8X16# m1_1 m1_2)
quotWord8X16 :: Word8X16 -> Word8X16 -> Word8X16
quotWord8X16 (Word8X16 m1_1) (Word8X16 m1_2) = Word8X16 (quotWord8X16# m1_1 m1_2)
remWord8X16 :: Word8X16 -> Word8X16 -> Word8X16
remWord8X16 (Word8X16 m1_1) (Word8X16 m1_2) = Word8X16 (remWord8X16# m1_1 m1_2)
indexWord8X16Array :: ByteArray -> Int -> Word8X16
indexWord8X16Array (ByteArray a) (I# i) = Word8X16 (indexWord8X16Array# a i)
readWord8X16Array :: PrimMonad m => MutableByteArray (PrimState m) -> Int -> m Word8X16
readWord8X16Array (MutableByteArray a) (I# i) = primitive (\ s0 -> case readWord8X16Array# a ((i *# 1#) +# 0#) s0 of
(# s1, m1 #) -> (# s1, Word8X16 m1 #))
writeWord8X16Array :: PrimMonad m => MutableByteArray (PrimState m) -> Int -> Word8X16 -> m ()
writeWord8X16Array (MutableByteArray a) (I# i) (Word8X16 m1) = primitive_ (writeWord8X16Array# a ((i *# 1#) +# 0#) m1)
indexWord8X16OffAddr :: Addr -> Int -> Word8X16
indexWord8X16OffAddr (Addr a) (I# i) = Word8X16 (indexWord8X16OffAddr# (plusAddr# a (i *# 16#)) 0#)
readWord8X16OffAddr :: PrimMonad m => Addr -> Int -> m Word8X16
readWord8X16OffAddr (Addr a) (I# i) = primitive (\ s0 -> case (\ addr i' -> readWord8X16OffAddr# (plusAddr# addr i') 0#) a ((i *# 16#) +# 0#) s0 of
(# s1, m1 #) -> (# s1, Word8X16 m1 #))
writeWord8X16OffAddr :: PrimMonad m => Addr -> Int -> Word8X16 -> m ()
writeWord8X16OffAddr (Addr a) (I# i) (Word8X16 m1) = primitive_ (writeWord8X16OffAddr# (plusAddr# a ((i *# 16#) +# 0#)) 0# m1)