ForeignSpec.hs

{-# LANGUAGE MagicHash           #-}
{-# LANGUAGE OverloadedStrings   #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE UnliftedFFITypes    #-}

module Z.ForeignSpec where

import qualified Data.List                as List
import           GHC.Exts
import           System.IO.Unsafe
import           Test.Hspec
import           Test.Hspec.QuickCheck
import           Test.QuickCheck
import           Test.QuickCheck.Function
import           Test.QuickCheck.Property
import qualified Z.Data.Array             as A
import qualified Z.Data.CBytes            as CB
import qualified Z.Data.JSON              as JSON
import qualified Z.Data.Vector.Base       as V
import           Z.Foreign

spec :: Spec
spec = describe "Foreign" $ do
    describe "pass prim array list to foreign" $ do
        prop "sum first should be equal(unsafe FFI)" $ \ xss ->
            let pas = List.map (V.pack . getNonEmpty) xss :: [A.PrimArray Int]
                s = sum (List.map (List.head . getNonEmpty) xss)
            in unsafeDupablePerformIO (withPrimArrayListUnsafe pas sum_first_unsafe) === s

        prop "sum first should be equal(safe FFI)" $ \ xss ->
            let pas = List.map (V.pack . getNonEmpty) xss :: [A.PrimArray Int]
                s = sum (List.map (List.head . getNonEmpty) xss)
            in unsafeDupablePerformIO (withPrimArrayListSafe pas sum_first_safe) === s

--------------------------------------------------------------------------------

foreign import ccall unsafe sum_first_unsafe :: BAArray# Int -> Int -> IO Int
foreign import ccall safe sum_first_safe :: Ptr (Ptr Int) -> Int -> IO Int