#include "MachDeps.h"
module Data.Primitive.SIMD.Int64X2 (Int64X2) 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 Int64X2 = 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 Int64X2 where
a == b = case unpackInt64X2 a of
(x1, x2) -> case unpackInt64X2 b of
(y1, y2) -> x1 == y1 && x2 == y2
instance Ord Int64X2 where
a `compare` b = case unpackInt64X2 a of
(x1, x2) -> case unpackInt64X2 b of
(y1, y2) -> x1 `compare` y1 <> x2 `compare` y2
instance Show Int64X2 where
showsPrec _ a s = case unpackInt64X2 a of
(x1, x2) -> "Int64X2 (" ++ shows x1 (", " ++ shows x2 (")" ++ s))
instance Num Int64X2 where
(+) = plusInt64X2
() = minusInt64X2
(*) = timesInt64X2
negate = negateInt64X2
abs = mapVector abs'
signum = mapVector signum'
fromInteger = broadcastVector . fromInteger
instance Bounded Int64X2 where
minBound = broadcastVector minBound
maxBound = broadcastVector maxBound
instance Storable Int64X2 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 Int64X2 where
type Elem Int64X2 = Int64
type ElemTuple Int64X2 = (Int64, Int64)
nullVector = broadcastVector 0
vectorSize _ = 2
elementSize _ = 8
broadcastVector = broadcastInt64X2
unsafeInsertVector = unsafeInsertInt64X2
packVector = packInt64X2
unpackVector = unpackInt64X2
mapVector = mapInt64X2
zipVector = zipInt64X2
foldVector = foldInt64X2
instance SIMDIntVector Int64X2 where
quotVector = quotInt64X2
remVector = remInt64X2
instance Prim Int64X2 where
sizeOf# a = let !(I# x) = sizeOf a in x
alignment# a = let !(I# x) = alignment a in x
indexByteArray# ba i = indexInt64X2Array (ByteArray ba) (I# i)
readByteArray# mba i s = let (ST r) = readInt64X2Array (MutableByteArray mba) (I# i) in r s
writeByteArray# mba i v s = let (ST r) = writeInt64X2Array (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 = indexInt64X2OffAddr (Addr addr) (I# i)
readOffAddr# addr i s = let (ST r) = readInt64X2OffAddr (Addr addr) (I# i) in r s
writeOffAddr# addr i v s = let (ST r) = writeInt64X2OffAddr (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 Int64X2 = V_Int64X2 (PV.Vector Int64X2)
newtype instance UV.MVector s Int64X2 = MV_Int64X2 (PMV.MVector s Int64X2)
instance Vector UV.Vector Int64X2 where
basicUnsafeFreeze (MV_Int64X2 v) = V_Int64X2 <$> PV.unsafeFreeze v
basicUnsafeThaw (V_Int64X2 v) = MV_Int64X2 <$> PV.unsafeThaw v
basicLength (V_Int64X2 v) = PV.length v
basicUnsafeSlice start len (V_Int64X2 v) = V_Int64X2(PV.unsafeSlice start len v)
basicUnsafeIndexM (V_Int64X2 v) = PV.unsafeIndexM v
basicUnsafeCopy (MV_Int64X2 m) (V_Int64X2 v) = PV.unsafeCopy m v
elemseq _ = seq
instance MVector UV.MVector Int64X2 where
basicLength (MV_Int64X2 v) = PMV.length v
basicUnsafeSlice start len (MV_Int64X2 v) = MV_Int64X2(PMV.unsafeSlice start len v)
basicOverlaps (MV_Int64X2 v) (MV_Int64X2 w) = PMV.overlaps v w
basicUnsafeNew len = MV_Int64X2 <$> PMV.unsafeNew len
#if MIN_VERSION_vector(0,11,0)
basicInitialize (MV_Int64X2 v) = basicInitialize v
#endif
basicUnsafeRead (MV_Int64X2 v) = PMV.unsafeRead v
basicUnsafeWrite (MV_Int64X2 v) = PMV.unsafeWrite v
instance Unbox Int64X2
broadcastInt64X2 :: Int64 -> Int64X2
broadcastInt64X2 (I64# x) = Int64X2 (broadcastInt64X2# x)
packInt64X2 :: (Int64, Int64) -> Int64X2
packInt64X2 (I64# x1, I64# x2) = Int64X2 (packInt64X2# (# x1, x2 #))
unpackInt64X2 :: Int64X2 -> (Int64, Int64)
unpackInt64X2 (Int64X2 m1) = case unpackInt64X2# m1 of
(# x1, x2 #) -> (I64# x1, I64# x2)
unsafeInsertInt64X2 :: Int64X2 -> Int64 -> Int -> Int64X2
unsafeInsertInt64X2 (Int64X2 m1) (I64# y) _i@(I# ip) = Int64X2 (insertInt64X2# m1 y (ip -# 0#))
mapInt64X2 :: (Int64 -> Int64) -> Int64X2 -> Int64X2
mapInt64X2 f = mapInt64X2# (\ x -> case f (I64# x) of { I64# y -> y})
mapInt64X2# :: (RealInt64# -> RealInt64#) -> Int64X2 -> Int64X2
mapInt64X2# f = \ v -> case unpackInt64X2 v of
(I64# x1, I64# x2) -> packInt64X2 (I64# (f x1), I64# (f x2))
zipInt64X2 :: (Int64 -> Int64 -> Int64) -> Int64X2 -> Int64X2 -> Int64X2
zipInt64X2 f = \ v1 v2 -> case unpackInt64X2 v1 of
(x1, x2) -> case unpackInt64X2 v2 of
(y1, y2) -> packInt64X2 (f x1 y1, f x2 y2)
foldInt64X2 :: (Int64 -> Int64 -> Int64) -> Int64X2 -> Int64
foldInt64X2 f' = \ v -> case unpackInt64X2 v of
(x1, x2) -> x1 `f` x2
where f !x !y = f' x y
plusInt64X2 :: Int64X2 -> Int64X2 -> Int64X2
plusInt64X2 (Int64X2 m1_1) (Int64X2 m1_2) = Int64X2 (plusInt64X2# m1_1 m1_2)
minusInt64X2 :: Int64X2 -> Int64X2 -> Int64X2
minusInt64X2 (Int64X2 m1_1) (Int64X2 m1_2) = Int64X2 (minusInt64X2# m1_1 m1_2)
timesInt64X2 :: Int64X2 -> Int64X2 -> Int64X2
timesInt64X2 (Int64X2 m1_1) (Int64X2 m1_2) = Int64X2 (timesInt64X2# m1_1 m1_2)
quotInt64X2 :: Int64X2 -> Int64X2 -> Int64X2
quotInt64X2 (Int64X2 m1_1) (Int64X2 m1_2) = Int64X2 (quotInt64X2# m1_1 m1_2)
remInt64X2 :: Int64X2 -> Int64X2 -> Int64X2
remInt64X2 (Int64X2 m1_1) (Int64X2 m1_2) = Int64X2 (remInt64X2# m1_1 m1_2)
negateInt64X2 :: Int64X2 -> Int64X2
negateInt64X2 (Int64X2 m1_1) = Int64X2 (negateInt64X2# m1_1)
indexInt64X2Array :: ByteArray -> Int -> Int64X2
indexInt64X2Array (ByteArray a) (I# i) = Int64X2 (indexInt64X2Array# a i)
readInt64X2Array :: PrimMonad m => MutableByteArray (PrimState m) -> Int -> m Int64X2
readInt64X2Array (MutableByteArray a) (I# i) = primitive (\ s0 -> case readInt64X2Array# a ((i *# 1#) +# 0#) s0 of
(# s1, m1 #) -> (# s1, Int64X2 m1 #))
writeInt64X2Array :: PrimMonad m => MutableByteArray (PrimState m) -> Int -> Int64X2 -> m ()
writeInt64X2Array (MutableByteArray a) (I# i) (Int64X2 m1) = primitive_ (writeInt64X2Array# a ((i *# 1#) +# 0#) m1)
indexInt64X2OffAddr :: Addr -> Int -> Int64X2
indexInt64X2OffAddr (Addr a) (I# i) = Int64X2 (indexInt64X2OffAddr# (plusAddr# a (i *# 16#)) 0#)
readInt64X2OffAddr :: PrimMonad m => Addr -> Int -> m Int64X2
readInt64X2OffAddr (Addr a) (I# i) = primitive (\ s0 -> case (\ addr i' -> readInt64X2OffAddr# (plusAddr# addr i') 0#) a ((i *# 16#) +# 0#) s0 of
(# s1, m1 #) -> (# s1, Int64X2 m1 #))
writeInt64X2OffAddr :: PrimMonad m => Addr -> Int -> Int64X2 -> m ()
writeInt64X2OffAddr (Addr a) (I# i) (Int64X2 m1) = primitive_ (writeInt64X2OffAddr# (plusAddr# a ((i *# 16#) +# 0#)) 0# m1)