Can the FFI deal with arrays? If so, how?

后端 未结 4 1257
终归单人心
终归单人心 2021-01-01 14:37

I\'m pretty sure it\'s possible to send arrays through the FFI, but I can\'t find any examples. For instance, I have a Haskell array that I send to a int foo(int*)

相关标签:
4条回答
  • 2021-01-01 15:11

    The FFI specification is quite readable, so you might want to just sit down and work through the whole thing. However, for this specific question, you can jump to the "Marshalling" section, particularly the Ptr and Storable subsections, which outline what's available for this.

    0 讨论(0)
  • 2021-01-01 15:15

    To convert a FFI Ptr into a Haskell list, you can use:

    peekArray0 :: (Storable a, Eq a) => a -> Ptr a -> IO [a]
    

    http://hackage.haskell.org/packages/archive/base/4.2.0.1/doc/html/Foreign-Marshal-Array.html#v%3ApeekArray0

    0 讨论(0)
  • 2021-01-01 15:24

    If you use the Data.Vector library you could use Data.Vector.Storable for your needs. Then you can use functions such as unsafeToForeignPtr or unsafeWith to access the underlying foreign pointer. This allows you to call C-code without any copying or marshaling taking place.

    If you want to create a vector from a C-array you can use unsafeFromForeignPtr.

    For your examples you can use (assuming c_foo does not modify it's arguments)

    import Foreign.Ptr
    import Foreign.C.Types
    import System.IO.Unsafe (unsafePerformIO)
    import qualified Data.Vector.Storable as SV
    
    foreign import ccall unsafe "foo" c_foo :: Ptr CInt -> CInt
    
    haskellFoo :: SV.Vector CInt -> CInt
    haskellFoo sv = unsafePerformIO $
        SV.unsafeWith sv $ \ptr -> return (c_foo ptr)
    

    This can be be golfed to:

    haskellFoo sv = unsafePerformIO $
        SV.unsafeWith sv (return . c_foo)
    

    Note that if your C-function modifies the data, then you shouldn't do this, instead you should make a copy of the data to not break referential transparency.

    If you want to use the standard Array-type you could use withStorableArray from Data.Array.Storable in the same way.

    0 讨论(0)
  • As in C, an array is basically a pointer to the array's first member. You get the other elements by doing arithmetic on the pointer. The Ptr is a member of Num, so you can use usual arithmetic operations.

    0 讨论(0)
提交回复
热议问题