#include "MachDeps.h"
module Data.Primitive.SIMD.Int64X4 (Int64X4) where
import Data.Primitive.SIMD.Class
import GHC.Int
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(..))
#if WORD_SIZE_IN_BITS == 64
type RealInt64# = Int#
#elif WORD_SIZE_IN_BITS == 32
type RealInt64# = Int64#
#else
#error "WORD_SIZE_IN_BITS is neither 64 or 32"
#endif
data Int64X4 = Int64X4 Int64X2# Int64X2# deriving Typeable
abs' :: Int64 -> Int64
abs' (I64# x) = I64# (abs# x)
abs# :: RealInt64# -> RealInt64#
abs# x = case abs (I64# x) of
I64# y -> y
signum' :: Int64 -> Int64
signum' (I64# x) = I64# (signum# x)
signum# :: RealInt64# -> RealInt64#
signum# x = case signum (I64# x) of
I64# y -> y
instance Eq Int64X4 where
a == b = case unpackInt64X4 a of
(x1, x2, x3, x4) -> case unpackInt64X4 b of
(y1, y2, y3, y4) -> x1 == y1 && x2 == y2 && x3 == y3 && x4 == y4
instance Ord Int64X4 where
a `compare` b = case unpackInt64X4 a of
(x1, x2, x3, x4) -> case unpackInt64X4 b of
(y1, y2, y3, y4) -> x1 `compare` y1 <> x2 `compare` y2 <> x3 `compare` y3 <> x4 `compare` y4
instance Show Int64X4 where
showsPrec _ a s = case unpackInt64X4 a of
(x1, x2, x3, x4) -> "Int64X4 (" ++ shows x1 (", " ++ shows x2 (", " ++ shows x3 (", " ++ shows x4 (")" ++ s))))
instance Num Int64X4 where
(+) = plusInt64X4
() = minusInt64X4
(*) = timesInt64X4
negate = negateInt64X4
abs = mapVector abs'
signum = mapVector signum'
fromInteger = broadcastVector . fromInteger
instance Bounded Int64X4 where
minBound = broadcastVector minBound
maxBound = broadcastVector maxBound
instance Storable Int64X4 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 Int64X4 where
type Elem Int64X4 = Int64
type ElemTuple Int64X4 = (Int64, Int64, Int64, Int64)
nullVector = broadcastVector 0
vectorSize _ = 4
elementSize _ = 8
broadcastVector = broadcastInt64X4
unsafeInsertVector = unsafeInsertInt64X4
packVector = packInt64X4
unpackVector = unpackInt64X4
mapVector = mapInt64X4
zipVector = zipInt64X4
foldVector = foldInt64X4
sumVector = sumInt64X4
instance SIMDIntVector Int64X4 where
quotVector = quotInt64X4
remVector = remInt64X4
instance Prim Int64X4 where
sizeOf# a = let !(I# x) = sizeOf a in x
alignment# a = let !(I# x) = alignment a in x
indexByteArray# ba i = indexInt64X4Array (ByteArray ba) (I# i)
readByteArray# mba i s = let (ST r) = readInt64X4Array (MutableByteArray mba) (I# i) in r s
writeByteArray# mba i v s = let (ST r) = writeInt64X4Array (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 = indexInt64X4OffAddr (Addr addr) (I# i)
readOffAddr# addr i s = let (ST r) = readInt64X4OffAddr (Addr addr) (I# i) in r s
writeOffAddr# addr i v s = let (ST r) = writeInt64X4OffAddr (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 Int64X4 = V_Int64X4 (PV.Vector Int64X4)
newtype instance UV.MVector s Int64X4 = MV_Int64X4 (PMV.MVector s Int64X4)
instance Vector UV.Vector Int64X4 where
basicUnsafeFreeze (MV_Int64X4 v) = V_Int64X4 <$> PV.unsafeFreeze v
basicUnsafeThaw (V_Int64X4 v) = MV_Int64X4 <$> PV.unsafeThaw v
basicLength (V_Int64X4 v) = PV.length v
basicUnsafeSlice start len (V_Int64X4 v) = V_Int64X4(PV.unsafeSlice start len v)
basicUnsafeIndexM (V_Int64X4 v) = PV.unsafeIndexM v
basicUnsafeCopy (MV_Int64X4 m) (V_Int64X4 v) = PV.unsafeCopy m v
elemseq _ = seq
instance MVector UV.MVector Int64X4 where
basicLength (MV_Int64X4 v) = PMV.length v
basicUnsafeSlice start len (MV_Int64X4 v) = MV_Int64X4(PMV.unsafeSlice start len v)
basicOverlaps (MV_Int64X4 v) (MV_Int64X4 w) = PMV.overlaps v w
basicUnsafeNew len = MV_Int64X4 <$> PMV.unsafeNew len
#if MIN_VERSION_vector(0,11,0)
basicInitialize (MV_Int64X4 v) = basicInitialize v
#endif
basicUnsafeRead (MV_Int64X4 v) = PMV.unsafeRead v
basicUnsafeWrite (MV_Int64X4 v) = PMV.unsafeWrite v
instance Unbox Int64X4
broadcastInt64X4 :: Int64 -> Int64X4
broadcastInt64X4 (I64# x) = case broadcastInt64X2# x of
v -> Int64X4 v v
packInt64X4 :: (Int64, Int64, Int64, Int64) -> Int64X4
packInt64X4 (I64# x1, I64# x2, I64# x3, I64# x4) = Int64X4 (packInt64X2# (# x1, x2 #)) (packInt64X2# (# x3, x4 #))
unpackInt64X4 :: Int64X4 -> (Int64, Int64, Int64, Int64)
unpackInt64X4 (Int64X4 m1 m2) = case unpackInt64X2# m1 of
(# x1, x2 #) -> case unpackInt64X2# m2 of
(# x3, x4 #) -> (I64# x1, I64# x2, I64# x3, I64# x4)
unsafeInsertInt64X4 :: Int64X4 -> Int64 -> Int -> Int64X4
unsafeInsertInt64X4 (Int64X4 m1 m2) (I64# y) _i@(I# ip) | _i < 2 = Int64X4 (insertInt64X2# m1 y (ip -# 0#)) m2
| otherwise = Int64X4 m1 (insertInt64X2# m2 y (ip -# 2#))
mapInt64X4 :: (Int64 -> Int64) -> Int64X4 -> Int64X4
mapInt64X4 f = mapInt64X4# (\ x -> case f (I64# x) of { I64# y -> y})
mapInt64X4# :: (RealInt64# -> RealInt64#) -> Int64X4 -> Int64X4
mapInt64X4# f = \ v -> case unpackInt64X4 v of
(I64# x1, I64# x2, I64# x3, I64# x4) -> packInt64X4 (I64# (f x1), I64# (f x2), I64# (f x3), I64# (f x4))
zipInt64X4 :: (Int64 -> Int64 -> Int64) -> Int64X4 -> Int64X4 -> Int64X4
zipInt64X4 f = \ v1 v2 -> case unpackInt64X4 v1 of
(x1, x2, x3, x4) -> case unpackInt64X4 v2 of
(y1, y2, y3, y4) -> packInt64X4 (f x1 y1, f x2 y2, f x3 y3, f x4 y4)
foldInt64X4 :: (Int64 -> Int64 -> Int64) -> Int64X4 -> Int64
foldInt64X4 f' = \ v -> case unpackInt64X4 v of
(x1, x2, x3, x4) -> x1 `f` x2 `f` x3 `f` x4
where f !x !y = f' x y
sumInt64X4 :: Int64X4 -> Int64
sumInt64X4 (Int64X4 x1 x2) = case unpackInt64X2# (plusInt64X2# x1 x2) of
(# y1, y2 #) -> I64# y1 + I64# y2
plusInt64X4 :: Int64X4 -> Int64X4 -> Int64X4
plusInt64X4 (Int64X4 m1_1 m2_1) (Int64X4 m1_2 m2_2) = Int64X4 (plusInt64X2# m1_1 m1_2) (plusInt64X2# m2_1 m2_2)
minusInt64X4 :: Int64X4 -> Int64X4 -> Int64X4
minusInt64X4 (Int64X4 m1_1 m2_1) (Int64X4 m1_2 m2_2) = Int64X4 (minusInt64X2# m1_1 m1_2) (minusInt64X2# m2_1 m2_2)
timesInt64X4 :: Int64X4 -> Int64X4 -> Int64X4
timesInt64X4 (Int64X4 m1_1 m2_1) (Int64X4 m1_2 m2_2) = Int64X4 (timesInt64X2# m1_1 m1_2) (timesInt64X2# m2_1 m2_2)
quotInt64X4 :: Int64X4 -> Int64X4 -> Int64X4
quotInt64X4 (Int64X4 m1_1 m2_1) (Int64X4 m1_2 m2_2) = Int64X4 (quotInt64X2# m1_1 m1_2) (quotInt64X2# m2_1 m2_2)
remInt64X4 :: Int64X4 -> Int64X4 -> Int64X4
remInt64X4 (Int64X4 m1_1 m2_1) (Int64X4 m1_2 m2_2) = Int64X4 (remInt64X2# m1_1 m1_2) (remInt64X2# m2_1 m2_2)
negateInt64X4 :: Int64X4 -> Int64X4
negateInt64X4 (Int64X4 m1_1 m2_1) = Int64X4 (negateInt64X2# m1_1) (negateInt64X2# m2_1)
indexInt64X4Array :: ByteArray -> Int -> Int64X4
indexInt64X4Array (ByteArray a) (I# i) = Int64X4 (indexInt64X2Array# a ((i *# 2#) +# 0#)) (indexInt64X2Array# a ((i *# 2#) +# 1#))
readInt64X4Array :: PrimMonad m => MutableByteArray (PrimState m) -> Int -> m Int64X4
readInt64X4Array (MutableByteArray a) (I# i) = primitive (\ s0 -> case readInt64X2Array# a ((i *# 2#) +# 0#) s0 of
(# s1, m1 #) -> case readInt64X2Array# a ((i *# 2#) +# 1#) s1 of
(# s2, m2 #) -> (# s2, Int64X4 m1 m2 #))
writeInt64X4Array :: PrimMonad m => MutableByteArray (PrimState m) -> Int -> Int64X4 -> m ()
writeInt64X4Array (MutableByteArray a) (I# i) (Int64X4 m1 m2) = primitive_ (writeInt64X2Array# a ((i *# 2#) +# 0#) m1) >> primitive_ (writeInt64X2Array# a ((i *# 2#) +# 1#) m2)
indexInt64X4OffAddr :: Addr -> Int -> Int64X4
indexInt64X4OffAddr (Addr a) (I# i) = Int64X4 (indexInt64X2OffAddr# (plusAddr# a ((i *# 32#) +# 0#)) 0#) (indexInt64X2OffAddr# (plusAddr# a ((i *# 32#) +# 16#)) 0#)
readInt64X4OffAddr :: PrimMonad m => Addr -> Int -> m Int64X4
readInt64X4OffAddr (Addr a) (I# i) = primitive (\ s0 -> case (\ addr i' -> readInt64X2OffAddr# (plusAddr# addr i') 0#) a ((i *# 32#) +# 0#) s0 of
(# s1, m1 #) -> case (\ addr i' -> readInt64X2OffAddr# (plusAddr# addr i') 0#) a ((i *# 32#) +# 16#) s1 of
(# s2, m2 #) -> (# s2, Int64X4 m1 m2 #))
writeInt64X4OffAddr :: PrimMonad m => Addr -> Int -> Int64X4 -> m ()
writeInt64X4OffAddr (Addr a) (I# i) (Int64X4 m1 m2) = primitive_ (writeInt64X2OffAddr# (plusAddr# a ((i *# 32#) +# 0#)) 0# m1) >> primitive_ (writeInt64X2OffAddr# (plusAddr# a ((i *# 32#) +# 16#)) 0# m2)