1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
module Data.Array.Repa.Repr.ByteString
( B, Array (..)
, fromByteString, toByteString)
where
import Data.Array.Repa.Shape
import Data.Array.Repa.Base
import Data.Array.Repa.Repr.Delayed
import Data.Word
import qualified Data.ByteString as B
import qualified Data.ByteString.Unsafe as BU
import Data.ByteString (ByteString)
-- | Strict ByteStrings arrays are represented as ForeignPtr buffers of Word8
data B
-- | Read elements from a `ByteString`.
instance Source B Word8 where
data Array B sh Word8
= AByteString !sh !ByteString
linearIndex (AByteString _ bs) ix
= bs `B.index` ix
{-# INLINE linearIndex #-}
unsafeLinearIndex (AByteString _ bs) ix
= bs `BU.unsafeIndex` ix
{-# INLINE unsafeLinearIndex #-}
extent (AByteString sh _)
= sh
{-# INLINE extent #-}
deepSeqArray (AByteString sh bs) x
= sh `deepSeq` bs `seq` x
{-# INLINE deepSeqArray #-}
deriving instance Show sh
=> Show (Array B sh Word8)
deriving instance Read sh
=> Read (Array B sh Word8)
-- Conversions ----------------------------------------------------------------
-- | O(1). Wrap a `ByteString` as an array.
fromByteString
:: sh -> ByteString -> Array B sh Word8
fromByteString sh bs
= AByteString sh bs
{-# INLINE fromByteString #-}
-- | O(1). Unpack a `ByteString` from an array.
toByteString :: Array B sh Word8 -> ByteString
toByteString (AByteString _ bs) = bs
{-# INLINE toByteString #-}
|
