-{-# LANGUAGE BangPatterns, CPP, ForeignFunctionInterface, GeneralizedNewtypeDeriving, MagicHash,
- UnliftedFFITypes #-}
--- This module lifts material from Brian O'Sullivan's text package
--- especially Data.Text.Encoding in order to define a pipes-appropriate
--- streamDecodeUtf8
-module Pipes.Text.Internal
- ( Decoding(..)
- , streamDecodeUtf8With
- , streamDecodeUtf8
+module Pipes.Text.Internal
+ (module Pipes.Text.Internal.Codec
+ , module Pipes.Text.Internal.Decoding
) where
-import Control.Exception (evaluate, try)
-#if __GLASGOW_HASKELL__ >= 702
-import Control.Monad.ST.Unsafe (unsafeIOToST, unsafeSTToIO)
-import Control.Monad.ST (ST, runST)
-#else
-import Control.Monad.ST (unsafeIOToST, unsafeSTToIO, ST, runST)
-#endif
-import Data.Bits ((.&.))
-import Data.ByteString as B
-import Data.ByteString.Internal as B
-import Data.Text ()
-import Data.Text.Encoding.Error (OnDecodeError, UnicodeException, strictDecode)
-import Data.Text.Internal (Text(..), safe, textP)
-import Data.Word (Word8, Word32)
-import Foreign.C.Types (CSize)
-import Foreign.ForeignPtr (withForeignPtr)
-import Foreign.Marshal.Utils (with)
-import Foreign.Ptr (Ptr, minusPtr, nullPtr, plusPtr)
-import Foreign.Storable (Storable, peek, poke)
-import GHC.Base hiding (ord)
-import GHC.Word
-import qualified Data.Text.Array as A
-import GHC.Exts (Char(..), Int(..), chr#, ord#, word2Int#)
-import GHC.Word (Word8(..), Word16(..), Word32(..))
-
-import Data.Text.Unsafe (unsafeDupablePerformIO)
-
-#include "pipes_text_cbits.h"
-
--- | A stream oriented decoding result.
-data Decoding = Some Text ByteString (ByteString -> Decoding)
- | Other Text ByteString
-instance Show Decoding where
- showsPrec d (Some t bs _) = showParen (d > prec) $
- showString "Some " . showsPrec prec' t .
- showChar ' ' . showsPrec prec' bs .
- showString " _"
- where prec = 10; prec' = prec + 1
- showsPrec d (Other t bs) = showParen (d > prec) $
- showString "Other " . showsPrec prec' t .
- showChar ' ' . showsPrec prec' bs .
- showString " _"
- where prec = 10; prec' = prec + 1
-
-newtype CodePoint = CodePoint Word32 deriving (Eq, Show, Num, Storable)
-newtype DecoderState = DecoderState Word32 deriving (Eq, Show, Num, Storable)
-
--- | Decode, in a stream oriented way, a 'ByteString' containing UTF-8
--- encoded text that is known to be valid.
---
--- If the input contains any invalid UTF-8 data, an exception will be
--- thrown (either by this function or a continuation) that cannot be
--- caught in pure code. For more control over the handling of invalid
--- data, use 'streamDecodeUtf8With'.
-streamDecodeUtf8 :: ByteString -> Decoding
-streamDecodeUtf8 = streamDecodeUtf8With (Just strictDecode)
-
--- | Decode, in a stream oriented way, a 'ByteString' containing UTF-8
--- encoded text.
-streamDecodeUtf8With :: Maybe OnDecodeError -> ByteString -> Decoding
-streamDecodeUtf8With mErr = case mErr of
- Nothing -> decodeWith False strictDecode
- Just onErr -> decodeWith True onErr
- where
- -- We create a slightly larger than necessary buffer to accommodate a
- -- potential surrogate pair started in the last buffer
- decodeWith replace onErr = decodeChunk 0 0
- where
- decodeChunk :: CodePoint -> DecoderState -> ByteString -> Decoding
- decodeChunk codepoint0 state0 bs@(PS fp off len) =
- runST $ (unsafeIOToST . decodeChunkToBuffer) =<< A.new (len+1)
- where
- decodeChunkToBuffer :: A.MArray s -> IO Decoding
- decodeChunkToBuffer dest = withForeignPtr fp $ \ptr ->
- with (0::CSize) $ \destOffPtr ->
- with codepoint0 $ \codepointPtr ->
- with state0 $ \statePtr ->
- with nullPtr $ \curPtrPtr ->
- let end = ptr `plusPtr` (off + len)
- loop curPtr = do
- poke curPtrPtr curPtr
- curPtr' <- c_decode_utf8_with_state (A.maBA dest) destOffPtr
- curPtrPtr end codepointPtr statePtr
- state <- peek statePtr
- case state of
- UTF8_REJECT ->
- -- We encountered an encoding error
- if replace
- then do
- x <- peek curPtr'
- case onErr desc (Just x) of
- Nothing -> loop $ curPtr' `plusPtr` 1
- Just c -> do
- destOff <- peek destOffPtr
- w <- unsafeSTToIO $
- unsafeWrite dest (fromIntegral destOff) (safe c)
- poke destOffPtr (destOff + fromIntegral w)
- poke statePtr 0
- loop $ curPtr' `plusPtr` 1
- else do
- n <- peek destOffPtr
- chunkText <- unsafeSTToIO $ do
- arr <- A.unsafeFreeze dest
- return $! textP arr 0 (fromIntegral n)
- lastPtr <- peek curPtrPtr
- let left = lastPtr `minusPtr` curPtr
- return $ Other chunkText (B.drop left bs)
- _ -> do
- -- We encountered the end of the buffer while decoding
- n <- peek destOffPtr
- codepoint <- peek codepointPtr
- chunkText <- unsafeSTToIO $ do
- arr <- A.unsafeFreeze dest
- return $! textP arr 0 (fromIntegral n)
- lastPtr <- peek curPtrPtr
- let left = lastPtr `minusPtr` curPtr
- return $ Some chunkText (B.drop left bs)
- (decodeChunk codepoint state)
- in loop (ptr `plusPtr` off)
- desc = "Data.Text.Encoding.streamDecodeUtf8With: Invalid UTF-8 stream"
-
-ord :: Char -> Int
-ord (C# c#) = I# (ord# c#)
-{-# INLINE ord #-}
-
-
-unsafeWrite :: A.MArray s -> Int -> Char -> ST s Int
-unsafeWrite marr i c
- | n < 0x10000 = do
- A.unsafeWrite marr i (fromIntegral n)
- return 1
- | otherwise = do
- A.unsafeWrite marr i lo
- A.unsafeWrite marr (i+1) hi
- return 2
- where n = ord c
- m = n - 0x10000
- lo = fromIntegral $ (m `shiftR` 10) + 0xD800
- hi = fromIntegral $ (m .&. 0x3FF) + 0xDC00
- shiftR (I# x#) (I# i#) = I# (x# `iShiftRA#` i#)
-{-# INLINE unsafeWrite #-}
-
-foreign import ccall unsafe "_hs_pipes_text_decode_utf8_state" c_decode_utf8_with_state
- :: MutableByteArray# s -> Ptr CSize
- -> Ptr (Ptr Word8) -> Ptr Word8
- -> Ptr CodePoint -> Ptr DecoderState -> IO (Ptr Word8)
\ No newline at end of file
+import Pipes.Text.Internal.Codec
+import Pipes.Text.Internal.Decoding
\ No newline at end of file