{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE MagicHash #-}
{-# LANGUAGE NoImplicitPrelude #-}
{-# LANGUAGE TypeFamilies #-}
{-# LANGUAGE UnboxedTuples #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE CPP #-}
module Basement.String
( String(..)
, MutableString(..)
, create
, replicate
, length
, Encoding(..)
, fromBytes
, fromChunkBytes
, fromBytesUnsafe
, fromBytesLenient
, toBytes
, mutableValidate
, copy
, ValidationFailure(..)
, index
, null
, drop
, take
, splitAt
, revDrop
, revTake
, revSplitAt
, splitOn
, sub
, elem
, indices
, intersperse
, span
, spanEnd
, break
, breakEnd
, breakElem
, breakLine
, dropWhile
, singleton
, charMap
, snoc
, cons
, unsnoc
, uncons
, find
, findIndex
, sortBy
, filter
, reverse
, replace
, builderAppend
, builderBuild
, builderBuild_
, readInteger
, readIntegral
, readNatural
, readDouble
, readRational
, readFloatingExact
, upper
, lower
, caseFold
, isPrefixOf
, isSuffixOf
, isInfixOf
, stripPrefix
, stripSuffix
, all
, any
, lines
, words
, toBase64
, toBase64URL
, toBase64OpenBSD
) where
import Basement.UArray (UArray)
import qualified Basement.UArray as Vec
import qualified Basement.UArray as C
import qualified Basement.UArray.Mutable as MVec
import Basement.Block.Mutable (Block(..), MutableBlock(..))
import qualified Basement.Block.Mutable as MBLK
import Basement.Compat.Bifunctor
import Basement.Compat.Base
import Basement.Compat.Natural
import Basement.Compat.MonadTrans
import Basement.Compat.Primitive
import Basement.Types.OffsetSize
import Basement.Numerical.Additive
import Basement.Numerical.Subtractive
import Basement.Numerical.Multiplicative
import Basement.Numerical.Number
import Basement.Cast
import Basement.Monad
import Basement.PrimType
import Basement.FinalPtr
import Basement.IntegralConv
import Basement.Floating
import Basement.MutableBuilder
import Basement.String.CaseMapping (upperMapping, lowerMapping, foldMapping)
import Basement.UTF8.Table
import Basement.UTF8.Helper
import Basement.UTF8.Base
import Basement.UTF8.Types
import Basement.UArray.Base as C (onBackendPrim, onBackend, onBackendPure, offset, ValidRange(..), offsetsValidRange, MUArray(..), MUArrayBackend(..))
import Basement.Alg.Class (Indexable)
import qualified Basement.Alg.UTF8 as UTF8
import qualified Basement.Alg.String as Alg
import Basement.Types.Char7 (Char7(..), c7Upper, c7Lower)
import qualified Basement.Types.Char7 as Char7
import GHC.Prim
import GHC.ST
import GHC.Types
import GHC.Word
#if MIN_VERSION_base(4,9,0)
import GHC.Char
#endif
import qualified Data.List
import Data.Ratio
import Data.Char (toUpper, toLower)
import qualified Prelude
import qualified Basement.String.Encoding.Encoding as Encoder
import qualified Basement.String.Encoding.ASCII7 as Encoder
import qualified Basement.String.Encoding.UTF16 as Encoder
import qualified Basement.String.Encoding.UTF32 as Encoder
import qualified Basement.String.Encoding.ISO_8859_1 as Encoder
data EncoderUTF8 = EncoderUTF8
instance Encoder.Encoding EncoderUTF8 where
type Unit EncoderUTF8 = Word8
type Error EncoderUTF8 = ValidationFailure
encodingNext :: EncoderUTF8
-> (Offset (Unit EncoderUTF8) -> Unit EncoderUTF8)
-> Offset (Unit EncoderUTF8)
-> Either (Error EncoderUTF8) (Char, Offset (Unit EncoderUTF8))
encodingNext _ = \ofs :: Offset (Unit EncoderUTF8) -> Unit EncoderUTF8
ofs -> (Char, Offset Word8)
-> Either ValidationFailure (Char, Offset Word8)
forall a b. b -> Either a b
Right ((Char, Offset Word8)
-> Either ValidationFailure (Char, Offset Word8))
-> (Offset Word8 -> (Char, Offset Word8))
-> Offset Word8
-> Either ValidationFailure (Char, Offset Word8)
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. (Offset Word8 -> Word8) -> Offset Word8 -> (Char, Offset Word8)
nextWithIndexer Offset Word8 -> Word8
Offset (Unit EncoderUTF8) -> Unit EncoderUTF8
ofs
encodingWrite :: EncoderUTF8
-> Char
-> Builder
(UArray (Unit EncoderUTF8))
(MUArray (Unit EncoderUTF8))
(Unit EncoderUTF8)
st
err
()
encodingWrite _ = Char
-> Builder
(UArray (Unit EncoderUTF8))
(MUArray (Unit EncoderUTF8))
(Unit EncoderUTF8)
st
err
()
forall (st :: * -> *) err.
(PrimMonad st, Monad st) =>
Char -> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
writeWithBuilder
validate :: UArray Word8
-> Offset8
-> CountOf Word8
-> (Offset8, Maybe ValidationFailure)
validate :: UArray Word8
-> Offset Word8
-> CountOf Word8
-> (Offset Word8, Maybe ValidationFailure)
validate array :: UArray Word8
array ofsStart :: Offset Word8
ofsStart sz :: CountOf Word8
sz = (Block Word8
-> Offset Word8 -> (Offset Word8, Maybe ValidationFailure))
-> (Ptr Word8
-> Offset Word8 -> ST Any (Offset Word8, Maybe ValidationFailure))
-> UArray Word8
-> (Offset Word8, Maybe ValidationFailure)
forall ty a s.
(Block ty -> Offset ty -> a)
-> (Ptr ty -> Offset ty -> ST s a) -> UArray ty -> a
C.unsafeDewrap Block Word8
-> Offset Word8 -> (Offset Word8, Maybe ValidationFailure)
goBa Ptr Word8
-> Offset Word8 -> ST Any (Offset Word8, Maybe ValidationFailure)
goAddr UArray Word8
array
where
unTranslateOffset :: Offset a -> p (Offset a) c -> p (Offset a) c
unTranslateOffset start :: Offset a
start = (Offset a -> Offset a) -> p (Offset a) c -> p (Offset a) c
forall (p :: * -> * -> *) a b c.
Bifunctor p =>
(a -> b) -> p a c -> p b c
first (\e :: Offset a
e -> Offset a
e Offset a -> Offset a -> Offset a
forall a. Offset a -> Offset a -> Offset a
`offsetSub` Offset a
start)
goBa :: Block Word8
-> Offset Word8 -> (Offset Word8, Maybe ValidationFailure)
goBa ba :: Block Word8
ba start :: Offset Word8
start =
Offset Word8
-> (Offset Word8, Maybe ValidationFailure)
-> (Offset Word8, Maybe ValidationFailure)
forall (p :: * -> * -> *) a c.
Bifunctor p =>
Offset a -> p (Offset a) c -> p (Offset a) c
unTranslateOffset Offset Word8
start ((Offset Word8, Maybe ValidationFailure)
-> (Offset Word8, Maybe ValidationFailure))
-> (Offset Word8, Maybe ValidationFailure)
-> (Offset Word8, Maybe ValidationFailure)
forall a b. (a -> b) -> a -> b
$ Offset Word8
-> Block Word8
-> Offset Word8
-> (Offset Word8, Maybe ValidationFailure)
forall container.
Indexable container Word8 =>
Offset Word8
-> container
-> Offset Word8
-> (Offset Word8, Maybe ValidationFailure)
Alg.validate (Offset Word8
startOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+Offset Word8
end) Block Word8
ba (Offset Word8
start Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ Offset Word8
ofsStart)
goAddr :: Ptr Word8
-> Offset Word8 -> ST Any (Offset Word8, Maybe ValidationFailure)
goAddr ptr :: Ptr Word8
ptr@(Ptr !Addr#
_) start :: Offset Word8
start =
(Offset Word8, Maybe ValidationFailure)
-> ST Any (Offset Word8, Maybe ValidationFailure)
forall (f :: * -> *) a. Applicative f => a -> f a
pure ((Offset Word8, Maybe ValidationFailure)
-> ST Any (Offset Word8, Maybe ValidationFailure))
-> (Offset Word8, Maybe ValidationFailure)
-> ST Any (Offset Word8, Maybe ValidationFailure)
forall a b. (a -> b) -> a -> b
$ Offset Word8
-> (Offset Word8, Maybe ValidationFailure)
-> (Offset Word8, Maybe ValidationFailure)
forall (p :: * -> * -> *) a c.
Bifunctor p =>
Offset a -> p (Offset a) c -> p (Offset a) c
unTranslateOffset Offset Word8
start ((Offset Word8, Maybe ValidationFailure)
-> (Offset Word8, Maybe ValidationFailure))
-> (Offset Word8, Maybe ValidationFailure)
-> (Offset Word8, Maybe ValidationFailure)
forall a b. (a -> b) -> a -> b
$ Offset Word8
-> Ptr Word8
-> Offset Word8
-> (Offset Word8, Maybe ValidationFailure)
forall container.
Indexable container Word8 =>
Offset Word8
-> container
-> Offset Word8
-> (Offset Word8, Maybe ValidationFailure)
Alg.validate (Offset Word8
startOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+Offset Word8
end) Ptr Word8
ptr (Offset Word8
ofsStart Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ Offset Word8
start)
end :: Offset Word8
end = Offset Word8
ofsStart Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
mutableValidate :: PrimMonad prim
=> MVec.MUArray Word8 (PrimState prim)
-> Offset Word8
-> CountOf Word8
-> prim (Offset Word8, Maybe ValidationFailure)
mutableValidate :: MUArray Word8 (PrimState prim)
-> Offset Word8
-> CountOf Word8
-> prim (Offset Word8, Maybe ValidationFailure)
mutableValidate mba :: MUArray Word8 (PrimState prim)
mba ofsStart :: Offset Word8
ofsStart sz :: CountOf Word8
sz = do
Offset Word8 -> prim (Offset Word8, Maybe ValidationFailure)
loop Offset Word8
ofsStart
where
end :: Offset Word8
end = Offset Word8
ofsStart Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
loop :: Offset Word8 -> prim (Offset Word8, Maybe ValidationFailure)
loop ofs :: Offset Word8
ofs
| Offset Word8
ofs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
end = [Char] -> prim (Offset Word8, Maybe ValidationFailure)
forall a. HasCallStack => [Char] -> a
error "mutableValidate: internal error: went pass offset"
| Offset Word8
ofs Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
end = (Offset Word8, Maybe ValidationFailure)
-> prim (Offset Word8, Maybe ValidationFailure)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
end, Maybe ValidationFailure
forall a. Maybe a
Nothing)
| Bool
otherwise = do
(Offset Word8, Maybe ValidationFailure)
r <- Offset Word8 -> prim (Offset Word8, Maybe ValidationFailure)
one Offset Word8
ofs
case (Offset Word8, Maybe ValidationFailure)
r of
(nextOfs :: Offset Word8
nextOfs, Nothing) -> Offset Word8 -> prim (Offset Word8, Maybe ValidationFailure)
loop Offset Word8
nextOfs
(pos :: Offset Word8
pos, Just failure :: ValidationFailure
failure) -> (Offset Word8, Maybe ValidationFailure)
-> prim (Offset Word8, Maybe ValidationFailure)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
pos, ValidationFailure -> Maybe ValidationFailure
forall a. a -> Maybe a
Just ValidationFailure
failure)
one :: Offset Word8 -> prim (Offset Word8, Maybe ValidationFailure)
one pos :: Offset Word8
pos = do
StepASCII
h <- Word8 -> StepASCII
StepASCII (Word8 -> StepASCII) -> prim Word8 -> prim StepASCII
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> MUArray Word8 (PrimState prim) -> Offset Word8 -> prim Word8
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> prim ty
Vec.unsafeRead MUArray Word8 (PrimState prim)
mba Offset Word8
pos
let nbConts :: Int
nbConts = StepASCII -> Int
getNbBytes StepASCII
h
if Int
nbConts Int -> Int -> Bool
forall a. Eq a => a -> a -> Bool
== 0xff
then (Offset Word8, Maybe ValidationFailure)
-> prim (Offset Word8, Maybe ValidationFailure)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
pos, ValidationFailure -> Maybe ValidationFailure
forall a. a -> Maybe a
Just ValidationFailure
InvalidHeader)
else if Offset Word8
pos Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1 Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ Int -> Offset Word8
forall ty. Int -> Offset ty
Offset Int
nbConts Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
end
then (Offset Word8, Maybe ValidationFailure)
-> prim (Offset Word8, Maybe ValidationFailure)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
pos, ValidationFailure -> Maybe ValidationFailure
forall a. a -> Maybe a
Just ValidationFailure
MissingByte)
else do
case Int
nbConts of
0 -> (Offset Word8, Maybe ValidationFailure)
-> prim (Offset Word8, Maybe ValidationFailure)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
pos Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1, Maybe ValidationFailure
forall a. Maybe a
Nothing)
1 -> do
Word8
c1 <- MUArray Word8 (PrimState prim) -> Offset Word8 -> prim Word8
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> prim ty
Vec.unsafeRead MUArray Word8 (PrimState prim)
mba (Offset Word8
pos Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1)
if Word8 -> Bool
isContinuation Word8
c1
then (Offset Word8, Maybe ValidationFailure)
-> prim (Offset Word8, Maybe ValidationFailure)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
pos Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 2, Maybe ValidationFailure
forall a. Maybe a
Nothing)
else (Offset Word8, Maybe ValidationFailure)
-> prim (Offset Word8, Maybe ValidationFailure)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
pos, ValidationFailure -> Maybe ValidationFailure
forall a. a -> Maybe a
Just ValidationFailure
InvalidContinuation)
2 -> do
Word8
c1 <- MUArray Word8 (PrimState prim) -> Offset Word8 -> prim Word8
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> prim ty
Vec.unsafeRead MUArray Word8 (PrimState prim)
mba (Offset Word8
pos Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1)
Word8
c2 <- MUArray Word8 (PrimState prim) -> Offset Word8 -> prim Word8
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> prim ty
Vec.unsafeRead MUArray Word8 (PrimState prim)
mba (Offset Word8
pos Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 2)
if Word8 -> Bool
isContinuation Word8
c1 Bool -> Bool -> Bool
&& Word8 -> Bool
isContinuation Word8
c2
then (Offset Word8, Maybe ValidationFailure)
-> prim (Offset Word8, Maybe ValidationFailure)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
pos Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 3, Maybe ValidationFailure
forall a. Maybe a
Nothing)
else (Offset Word8, Maybe ValidationFailure)
-> prim (Offset Word8, Maybe ValidationFailure)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
pos, ValidationFailure -> Maybe ValidationFailure
forall a. a -> Maybe a
Just ValidationFailure
InvalidContinuation)
3 -> do
Word8
c1 <- MUArray Word8 (PrimState prim) -> Offset Word8 -> prim Word8
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> prim ty
Vec.unsafeRead MUArray Word8 (PrimState prim)
mba (Offset Word8
pos Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1)
Word8
c2 <- MUArray Word8 (PrimState prim) -> Offset Word8 -> prim Word8
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> prim ty
Vec.unsafeRead MUArray Word8 (PrimState prim)
mba (Offset Word8
pos Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 2)
Word8
c3 <- MUArray Word8 (PrimState prim) -> Offset Word8 -> prim Word8
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> prim ty
Vec.unsafeRead MUArray Word8 (PrimState prim)
mba (Offset Word8
pos Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 3)
if Word8 -> Bool
isContinuation Word8
c1 Bool -> Bool -> Bool
&& Word8 -> Bool
isContinuation Word8
c2 Bool -> Bool -> Bool
&& Word8 -> Bool
isContinuation Word8
c3
then (Offset Word8, Maybe ValidationFailure)
-> prim (Offset Word8, Maybe ValidationFailure)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
pos Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 4, Maybe ValidationFailure
forall a. Maybe a
Nothing)
else (Offset Word8, Maybe ValidationFailure)
-> prim (Offset Word8, Maybe ValidationFailure)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
pos, ValidationFailure -> Maybe ValidationFailure
forall a. a -> Maybe a
Just ValidationFailure
InvalidContinuation)
_ -> [Char] -> prim (Offset Word8, Maybe ValidationFailure)
forall a. HasCallStack => [Char] -> a
error "internal error"
nextWithIndexer :: (Offset Word8 -> Word8)
-> Offset Word8
-> (Char, Offset Word8)
nextWithIndexer :: (Offset Word8 -> Word8) -> Offset Word8 -> (Char, Offset Word8)
nextWithIndexer getter :: Offset Word8 -> Word8
getter off :: Offset Word8
off =
case Word# -> Int#
getNbBytes# Word#
h of
0# -> (Word# -> Char
toChar Word#
h, Offset Word8
off Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1)
1# -> (Word# -> Char
toChar (Word8 -> Word#
decode2 (Offset Word8 -> Word8
getter (Offset Word8 -> Word8) -> Offset Word8 -> Word8
forall a b. (a -> b) -> a -> b
$ Offset Word8
off Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1)), Offset Word8
off Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 2)
2# -> (Word# -> Char
toChar (Word8 -> Word8 -> Word#
decode3 (Offset Word8 -> Word8
getter (Offset Word8 -> Word8) -> Offset Word8 -> Word8
forall a b. (a -> b) -> a -> b
$ Offset Word8
off Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1) (Offset Word8 -> Word8
getter (Offset Word8 -> Word8) -> Offset Word8 -> Word8
forall a b. (a -> b) -> a -> b
$ Offset Word8
off Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 2)), Offset Word8
off Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 3)
3# -> (Word# -> Char
toChar (Word8 -> Word8 -> Word8 -> Word#
decode4 (Offset Word8 -> Word8
getter (Offset Word8 -> Word8) -> Offset Word8 -> Word8
forall a b. (a -> b) -> a -> b
$ Offset Word8
off Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1) (Offset Word8 -> Word8
getter (Offset Word8 -> Word8) -> Offset Word8 -> Word8
forall a b. (a -> b) -> a -> b
$ Offset Word8
off Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 2) (Offset Word8 -> Word8
getter (Offset Word8 -> Word8) -> Offset Word8 -> Word8
forall a b. (a -> b) -> a -> b
$ Offset Word8
off Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 3))
, Offset Word8
off Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 4)
r :: Int#
r -> [Char] -> (Char, Offset Word8)
forall a. HasCallStack => [Char] -> a
error ("next: internal error: invalid input: " [Char] -> [Char] -> [Char]
forall a. Semigroup a => a -> a -> a
<> Int -> [Char]
forall a. Show a => a -> [Char]
show (Int# -> Int
I# Int#
r) [Char] -> [Char] -> [Char]
forall a. Semigroup a => a -> a -> a
<> " " [Char] -> [Char] -> [Char]
forall a. Semigroup a => a -> a -> a
<> Word -> [Char]
forall a. Show a => a -> [Char]
show (Word# -> Word
W# Word#
h))
where
!(W8# h :: Word#
h) = Offset Word8 -> Word8
getter Offset Word8
off
toChar :: Word# -> Char
toChar :: Word# -> Char
toChar w :: Word#
w = Char# -> Char
C# (Int# -> Char#
chr# (Word# -> Int#
word2Int# Word#
w))
decode2 :: Word8 -> Word#
decode2 :: Word8 -> Word#
decode2 (W8# c1 :: Word#
c1) =
Word# -> Word# -> Word#
or# (Word# -> Int# -> Word#
uncheckedShiftL# (Word# -> Word# -> Word#
and# Word#
h 0x1f##) 6#)
(Word# -> Word# -> Word#
and# Word#
c1 0x3f##)
decode3 :: Word8 -> Word8 -> Word#
decode3 :: Word8 -> Word8 -> Word#
decode3 (W8# c1 :: Word#
c1) (W8# c2 :: Word#
c2) =
Word# -> Word# -> Word#
or# (Word# -> Int# -> Word#
uncheckedShiftL# (Word# -> Word# -> Word#
and# Word#
h 0xf##) 12#)
(Word# -> Word# -> Word#
or# (Word# -> Int# -> Word#
uncheckedShiftL# (Word# -> Word# -> Word#
and# Word#
c1 0x3f##) 6#)
(Word# -> Word# -> Word#
and# Word#
c2 0x3f##))
decode4 :: Word8 -> Word8 -> Word8 -> Word#
decode4 :: Word8 -> Word8 -> Word8 -> Word#
decode4 (W8# c1 :: Word#
c1) (W8# c2 :: Word#
c2) (W8# c3 :: Word#
c3) =
Word# -> Word# -> Word#
or# (Word# -> Int# -> Word#
uncheckedShiftL# (Word# -> Word# -> Word#
and# Word#
h 0x7##) 18#)
(Word# -> Word# -> Word#
or# (Word# -> Int# -> Word#
uncheckedShiftL# (Word# -> Word# -> Word#
and# Word#
c1 0x3f##) 12#)
(Word# -> Word# -> Word#
or# (Word# -> Int# -> Word#
uncheckedShiftL# (Word# -> Word# -> Word#
and# Word#
c2 0x3f##) 6#)
(Word# -> Word# -> Word#
and# Word#
c3 0x3f##))
)
writeWithBuilder :: (PrimMonad st, Monad st)
=> Char
-> Builder (UArray Word8) (MVec.MUArray Word8) Word8 st err ()
writeWithBuilder :: Char -> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
writeWithBuilder c :: Char
c
| Int# -> Bool
bool# (Word# -> Word# -> Int#
ltWord# Word#
x 0x80## ) = Builder (UArray Word8) (MUArray Word8) Word8 st err ()
encode1
| Int# -> Bool
bool# (Word# -> Word# -> Int#
ltWord# Word#
x 0x800## ) = Builder (UArray Word8) (MUArray Word8) Word8 st err ()
encode2
| Int# -> Bool
bool# (Word# -> Word# -> Int#
ltWord# Word#
x 0x10000##) = Builder (UArray Word8) (MUArray Word8) Word8 st err ()
encode3
| Bool
otherwise = Builder (UArray Word8) (MUArray Word8) Word8 st err ()
encode4
where
!(I# xi :: Int#
xi) = Char -> Int
forall a. Enum a => a -> Int
fromEnum Char
c
!x :: Word#
x = Int# -> Word#
int2Word# Int#
xi
encode1 :: Builder (UArray Word8) (MUArray Word8) Word8 st err ()
encode1 = Word8 -> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall ty (state :: * -> *) err.
(PrimType ty, PrimMonad state) =>
ty -> Builder (UArray ty) (MUArray ty) ty state err ()
Vec.builderAppend (Word# -> Word8
W8# Word#
x)
encode2 :: Builder (UArray Word8) (MUArray Word8) Word8 st err ()
encode2 = do
let x1 :: Word#
x1 = Word# -> Word# -> Word#
or# (Word# -> Int# -> Word#
uncheckedShiftRL# Word#
x 6#) 0xc0##
x2 :: Word#
x2 = Word# -> Word#
toContinuation Word#
x
Word8 -> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall ty (state :: * -> *) err.
(PrimType ty, PrimMonad state) =>
ty -> Builder (UArray ty) (MUArray ty) ty state err ()
Vec.builderAppend (Word# -> Word8
W8# Word#
x1) Builder (UArray Word8) (MUArray Word8) Word8 st err ()
-> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
-> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> Word8 -> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall ty (state :: * -> *) err.
(PrimType ty, PrimMonad state) =>
ty -> Builder (UArray ty) (MUArray ty) ty state err ()
Vec.builderAppend (Word# -> Word8
W8# Word#
x2)
encode3 :: Builder (UArray Word8) (MUArray Word8) Word8 st err ()
encode3 = do
let x1 :: Word#
x1 = Word# -> Word# -> Word#
or# (Word# -> Int# -> Word#
uncheckedShiftRL# Word#
x 12#) 0xe0##
x2 :: Word#
x2 = Word# -> Word#
toContinuation (Word# -> Int# -> Word#
uncheckedShiftRL# Word#
x 6#)
x3 :: Word#
x3 = Word# -> Word#
toContinuation Word#
x
Word8 -> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall ty (state :: * -> *) err.
(PrimType ty, PrimMonad state) =>
ty -> Builder (UArray ty) (MUArray ty) ty state err ()
Vec.builderAppend (Word# -> Word8
W8# Word#
x1) Builder (UArray Word8) (MUArray Word8) Word8 st err ()
-> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
-> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> Word8 -> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall ty (state :: * -> *) err.
(PrimType ty, PrimMonad state) =>
ty -> Builder (UArray ty) (MUArray ty) ty state err ()
Vec.builderAppend (Word# -> Word8
W8# Word#
x2) Builder (UArray Word8) (MUArray Word8) Word8 st err ()
-> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
-> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> Word8 -> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall ty (state :: * -> *) err.
(PrimType ty, PrimMonad state) =>
ty -> Builder (UArray ty) (MUArray ty) ty state err ()
Vec.builderAppend (Word# -> Word8
W8# Word#
x3)
encode4 :: Builder (UArray Word8) (MUArray Word8) Word8 st err ()
encode4 = do
let x1 :: Word#
x1 = Word# -> Word# -> Word#
or# (Word# -> Int# -> Word#
uncheckedShiftRL# Word#
x 18#) 0xf0##
x2 :: Word#
x2 = Word# -> Word#
toContinuation (Word# -> Int# -> Word#
uncheckedShiftRL# Word#
x 12#)
x3 :: Word#
x3 = Word# -> Word#
toContinuation (Word# -> Int# -> Word#
uncheckedShiftRL# Word#
x 6#)
x4 :: Word#
x4 = Word# -> Word#
toContinuation Word#
x
Word8 -> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall ty (state :: * -> *) err.
(PrimType ty, PrimMonad state) =>
ty -> Builder (UArray ty) (MUArray ty) ty state err ()
Vec.builderAppend (Word# -> Word8
W8# Word#
x1) Builder (UArray Word8) (MUArray Word8) Word8 st err ()
-> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
-> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> Word8 -> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall ty (state :: * -> *) err.
(PrimType ty, PrimMonad state) =>
ty -> Builder (UArray ty) (MUArray ty) ty state err ()
Vec.builderAppend (Word# -> Word8
W8# Word#
x2) Builder (UArray Word8) (MUArray Word8) Word8 st err ()
-> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
-> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> Word8 -> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall ty (state :: * -> *) err.
(PrimType ty, PrimMonad state) =>
ty -> Builder (UArray ty) (MUArray ty) ty state err ()
Vec.builderAppend (Word# -> Word8
W8# Word#
x3) Builder (UArray Word8) (MUArray Word8) Word8 st err ()
-> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
-> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> Word8 -> Builder (UArray Word8) (MUArray Word8) Word8 st err ()
forall ty (state :: * -> *) err.
(PrimType ty, PrimMonad state) =>
ty -> Builder (UArray ty) (MUArray ty) ty state err ()
Vec.builderAppend (Word# -> Word8
W8# Word#
x4)
toContinuation :: Word# -> Word#
toContinuation :: Word# -> Word#
toContinuation w :: Word#
w = Word# -> Word# -> Word#
or# (Word# -> Word# -> Word#
and# Word#
w 0x3f##) 0x80##
writeUTF8Char :: PrimMonad prim => MutableString (PrimState prim) -> Offset8 -> UTF8Char -> prim ()
writeUTF8Char :: MutableString (PrimState prim)
-> Offset Word8 -> UTF8Char -> prim ()
writeUTF8Char (MutableString mba :: MUArray Word8 (PrimState prim)
mba) i :: Offset Word8
i (UTF8_1 x1 :: Word8
x1) =
MUArray Word8 (PrimState prim) -> Offset Word8 -> Word8 -> prim ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> ty -> prim ()
Vec.unsafeWrite MUArray Word8 (PrimState prim)
mba Offset Word8
i Word8
x1
writeUTF8Char (MutableString mba :: MUArray Word8 (PrimState prim)
mba) i :: Offset Word8
i (UTF8_2 x1 :: Word8
x1 x2 :: Word8
x2) = do
MUArray Word8 (PrimState prim) -> Offset Word8 -> Word8 -> prim ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> ty -> prim ()
Vec.unsafeWrite MUArray Word8 (PrimState prim)
mba Offset Word8
i Word8
x1
MUArray Word8 (PrimState prim) -> Offset Word8 -> Word8 -> prim ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> ty -> prim ()
Vec.unsafeWrite MUArray Word8 (PrimState prim)
mba (Offset Word8
iOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+1) Word8
x2
writeUTF8Char (MutableString mba :: MUArray Word8 (PrimState prim)
mba) i :: Offset Word8
i (UTF8_3 x1 :: Word8
x1 x2 :: Word8
x2 x3 :: Word8
x3) = do
MUArray Word8 (PrimState prim) -> Offset Word8 -> Word8 -> prim ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> ty -> prim ()
Vec.unsafeWrite MUArray Word8 (PrimState prim)
mba Offset Word8
i Word8
x1
MUArray Word8 (PrimState prim) -> Offset Word8 -> Word8 -> prim ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> ty -> prim ()
Vec.unsafeWrite MUArray Word8 (PrimState prim)
mba (Offset Word8
iOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+1) Word8
x2
MUArray Word8 (PrimState prim) -> Offset Word8 -> Word8 -> prim ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> ty -> prim ()
Vec.unsafeWrite MUArray Word8 (PrimState prim)
mba (Offset Word8
iOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+2) Word8
x3
writeUTF8Char (MutableString mba :: MUArray Word8 (PrimState prim)
mba) i :: Offset Word8
i (UTF8_4 x1 :: Word8
x1 x2 :: Word8
x2 x3 :: Word8
x3 x4 :: Word8
x4) = do
MUArray Word8 (PrimState prim) -> Offset Word8 -> Word8 -> prim ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> ty -> prim ()
Vec.unsafeWrite MUArray Word8 (PrimState prim)
mba Offset Word8
i Word8
x1
MUArray Word8 (PrimState prim) -> Offset Word8 -> Word8 -> prim ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> ty -> prim ()
Vec.unsafeWrite MUArray Word8 (PrimState prim)
mba (Offset Word8
iOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+1) Word8
x2
MUArray Word8 (PrimState prim) -> Offset Word8 -> Word8 -> prim ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> ty -> prim ()
Vec.unsafeWrite MUArray Word8 (PrimState prim)
mba (Offset Word8
iOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+2) Word8
x3
MUArray Word8 (PrimState prim) -> Offset Word8 -> Word8 -> prim ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim) -> Offset ty -> ty -> prim ()
Vec.unsafeWrite MUArray Word8 (PrimState prim)
mba (Offset Word8
iOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+3) Word8
x4
{-# INLINE writeUTF8Char #-}
unsafeFreezeShrink :: PrimMonad prim => MutableString (PrimState prim) -> CountOf Word8 -> prim String
unsafeFreezeShrink :: MutableString (PrimState prim) -> CountOf Word8 -> prim String
unsafeFreezeShrink (MutableString mba :: MUArray Word8 (PrimState prim)
mba) s :: CountOf Word8
s = UArray Word8 -> String
String (UArray Word8 -> String) -> prim (UArray Word8) -> prim String
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> MUArray Word8 (PrimState prim)
-> CountOf Word8 -> prim (UArray Word8)
forall ty (prim :: * -> *).
(PrimType ty, PrimMonad prim) =>
MUArray ty (PrimState prim) -> CountOf ty -> prim (UArray ty)
Vec.unsafeFreezeShrink MUArray Word8 (PrimState prim)
mba CountOf Word8
s
{-# INLINE unsafeFreezeShrink #-}
null :: String -> Bool
null :: String -> Bool
null (String ba :: UArray Word8
ba) = UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
ba CountOf Word8 -> CountOf Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== 0
countCharMoreThanBytes :: CountOf Char -> UArray Word8 -> Bool
countCharMoreThanBytes :: CountOf Char -> UArray Word8 -> Bool
countCharMoreThanBytes (CountOf chars :: Int
chars) ba :: UArray Word8
ba = Int
chars Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
>= Int
bytes
where (CountOf bytes :: Int
bytes) = UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
ba
take :: CountOf Char -> String -> String
take :: CountOf Char -> String -> String
take n :: CountOf Char
n s :: String
s@(String ba :: UArray Word8
ba)
| CountOf Char
n CountOf Char -> CountOf Char -> Bool
forall a. Ord a => a -> a -> Bool
<= 0 = String
forall a. Monoid a => a
mempty
| CountOf Char -> UArray Word8 -> Bool
countCharMoreThanBytes CountOf Char
n UArray Word8
ba = String
s
| Bool
otherwise = UArray Word8 -> String
String (UArray Word8 -> String) -> UArray Word8 -> String
forall a b. (a -> b) -> a -> b
$ CountOf Word8 -> UArray Word8 -> UArray Word8
forall ty. CountOf ty -> UArray ty -> UArray ty
Vec.unsafeTake (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize (Offset Word8 -> CountOf Word8) -> Offset Word8 -> CountOf Word8
forall a b. (a -> b) -> a -> b
$ CountOf Char -> String -> Offset Word8
indexN CountOf Char
n String
s) UArray Word8
ba
drop :: CountOf Char -> String -> String
drop :: CountOf Char -> String -> String
drop n :: CountOf Char
n s :: String
s@(String ba :: UArray Word8
ba)
| CountOf Char
n CountOf Char -> CountOf Char -> Bool
forall a. Ord a => a -> a -> Bool
<= 0 = String
s
| CountOf Char -> UArray Word8 -> Bool
countCharMoreThanBytes CountOf Char
n UArray Word8
ba = String
forall a. Monoid a => a
mempty
| Bool
otherwise = UArray Word8 -> String
String (UArray Word8 -> String) -> UArray Word8 -> String
forall a b. (a -> b) -> a -> b
$ CountOf Word8 -> UArray Word8 -> UArray Word8
forall ty. CountOf ty -> UArray ty -> UArray ty
Vec.drop (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize (Offset Word8 -> CountOf Word8) -> Offset Word8 -> CountOf Word8
forall a b. (a -> b) -> a -> b
$ CountOf Char -> String -> Offset Word8
indexN CountOf Char
n String
s) UArray Word8
ba
splitAt :: CountOf Char -> String -> (String, String)
splitAt :: CountOf Char -> String -> (String, String)
splitAt n :: CountOf Char
n s :: String
s@(String ba :: UArray Word8
ba)
| CountOf Char
n CountOf Char -> CountOf Char -> Bool
forall a. Ord a => a -> a -> Bool
<= 0 = (String
forall a. Monoid a => a
mempty, String
s)
| CountOf Char -> UArray Word8 -> Bool
countCharMoreThanBytes CountOf Char
n UArray Word8
ba = (String
s, String
forall a. Monoid a => a
mempty)
| Bool
otherwise =
let (v1 :: UArray Word8
v1,v2 :: UArray Word8
v2) = CountOf Word8 -> UArray Word8 -> (UArray Word8, UArray Word8)
forall ty. CountOf ty -> UArray ty -> (UArray ty, UArray ty)
C.splitAt (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize (Offset Word8 -> CountOf Word8) -> Offset Word8 -> CountOf Word8
forall a b. (a -> b) -> a -> b
$ CountOf Char -> String -> Offset Word8
indexN CountOf Char
n String
s) UArray Word8
ba
in (UArray Word8 -> String
String UArray Word8
v1, UArray Word8 -> String
String UArray Word8
v2)
indexN :: CountOf Char -> String -> Offset Word8
indexN :: CountOf Char -> String -> Offset Word8
indexN !CountOf Char
n (String ba :: UArray Word8
ba) = (Block Word8 -> Offset Word8 -> Offset Word8)
-> (Ptr Word8 -> Offset Word8 -> ST Any (Offset Word8))
-> UArray Word8
-> Offset Word8
forall ty a s.
(Block ty -> Offset ty -> a)
-> (Ptr ty -> Offset ty -> ST s a) -> UArray ty -> a
Vec.unsafeDewrap Block Word8 -> Offset Word8 -> Offset Word8
goVec Ptr Word8 -> Offset Word8 -> ST Any (Offset Word8)
forall s. Ptr Word8 -> Offset Word8 -> ST s (Offset Word8)
goAddr UArray Word8
ba
where
goVec :: Block Word8 -> Offset Word8 -> Offset Word8
goVec :: Block Word8 -> Offset Word8 -> Offset Word8
goVec (Block !ByteArray#
ma) !Offset Word8
start = Offset Word8 -> Offset Char -> Offset Word8
loop Offset Word8
start 0
where
!len :: Offset Word8
len = Offset Word8
start Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
Vec.length UArray Word8
ba
loop :: Offset Word8 -> Offset Char -> Offset Word8
loop :: Offset Word8 -> Offset Char -> Offset Word8
loop !Offset Word8
idx !Offset Char
i
| Offset Word8
idx Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
>= Offset Word8
len Bool -> Bool -> Bool
|| Offset Char
i Offset Char -> CountOf Char -> Bool
forall ty. Offset ty -> CountOf ty -> Bool
.==# CountOf Char
n = CountOf Word8 -> Offset Word8
forall a. CountOf a -> Offset a
sizeAsOffset (Offset Word8
idx Offset Word8 -> Offset Word8 -> Difference (Offset Word8)
forall a. Subtractive a => a -> a -> Difference a
- Offset Word8
start)
| Bool
otherwise = Offset Word8 -> Offset Char -> Offset Word8
loop (Offset Word8
idx Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
d) (Offset Char
i Offset Char -> Offset Char -> Offset Char
forall a. Additive a => a -> a -> a
+ Int -> Offset Char
forall ty. Int -> Offset ty
Offset 1)
where d :: CountOf Word8
d = Word8 -> CountOf Word8
skipNextHeaderValue (ByteArray# -> Offset Word8 -> Word8
forall ty. PrimType ty => ByteArray# -> Offset ty -> ty
primBaIndex ByteArray#
ma Offset Word8
idx)
{-# INLINE goVec #-}
goAddr :: Ptr Word8 -> Offset Word8 -> ST s (Offset Word8)
goAddr :: Ptr Word8 -> Offset Word8 -> ST s (Offset Word8)
goAddr (Ptr ptr :: Addr#
ptr) !Offset Word8
start = Offset Word8 -> ST s (Offset Word8)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8 -> ST s (Offset Word8))
-> Offset Word8 -> ST s (Offset Word8)
forall a b. (a -> b) -> a -> b
$ Offset Word8 -> Offset Char -> Offset Word8
loop Offset Word8
start (Int -> Offset Char
forall ty. Int -> Offset ty
Offset 0)
where
!len :: Offset Word8
len = Offset Word8
start Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
Vec.length UArray Word8
ba
loop :: Offset Word8 -> Offset Char -> Offset Word8
loop :: Offset Word8 -> Offset Char -> Offset Word8
loop !Offset Word8
idx !Offset Char
i
| Offset Word8
idx Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
>= Offset Word8
len Bool -> Bool -> Bool
|| Offset Char
i Offset Char -> CountOf Char -> Bool
forall ty. Offset ty -> CountOf ty -> Bool
.==# CountOf Char
n = CountOf Word8 -> Offset Word8
forall a. CountOf a -> Offset a
sizeAsOffset (Offset Word8
idx Offset Word8 -> Offset Word8 -> Difference (Offset Word8)
forall a. Subtractive a => a -> a -> Difference a
- Offset Word8
start)
| Bool
otherwise = Offset Word8 -> Offset Char -> Offset Word8
loop (Offset Word8
idx Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
d) (Offset Char
i Offset Char -> Offset Char -> Offset Char
forall a. Additive a => a -> a -> a
+ Int -> Offset Char
forall ty. Int -> Offset ty
Offset 1)
where d :: CountOf Word8
d = Word8 -> CountOf Word8
skipNextHeaderValue (Addr# -> Offset Word8 -> Word8
forall ty. PrimType ty => Addr# -> Offset ty -> ty
primAddrIndex Addr#
ptr Offset Word8
idx)
{-# INLINE goAddr #-}
{-# INLINE indexN #-}
countFromStart :: String -> CountOf Char -> CountOf Char
countFromStart :: String -> CountOf Char -> CountOf Char
countFromStart s :: String
s sz :: CountOf Char
sz@(CountOf sz' :: Int
sz')
| CountOf Char
sz CountOf Char -> CountOf Char -> Bool
forall a. Ord a => a -> a -> Bool
>= CountOf Char
len = Int -> CountOf Char
forall ty. Int -> CountOf ty
CountOf 0
| Bool
otherwise = Int -> CountOf Char
forall ty. Int -> CountOf ty
CountOf (Int
len' Int -> Int -> Difference Int
forall a. Subtractive a => a -> a -> Difference a
- Int
sz')
where len :: CountOf Char
len@(CountOf len' :: Int
len') = String -> CountOf Char
length String
s
revTake :: CountOf Char -> String -> String
revTake :: CountOf Char -> String -> String
revTake n :: CountOf Char
n v :: String
v = CountOf Char -> String -> String
drop (String -> CountOf Char -> CountOf Char
countFromStart String
v CountOf Char
n) String
v
revDrop :: CountOf Char -> String -> String
revDrop :: CountOf Char -> String -> String
revDrop n :: CountOf Char
n v :: String
v = CountOf Char -> String -> String
take (String -> CountOf Char -> CountOf Char
countFromStart String
v CountOf Char
n) String
v
revSplitAt :: CountOf Char -> String -> (String, String)
revSplitAt :: CountOf Char -> String -> (String, String)
revSplitAt n :: CountOf Char
n v :: String
v = (CountOf Char -> String -> String
drop CountOf Char
idx String
v, CountOf Char -> String -> String
take CountOf Char
idx String
v) where idx :: CountOf Char
idx = String -> CountOf Char -> CountOf Char
countFromStart String
v CountOf Char
n
splitOn :: (Char -> Bool) -> String -> [String]
splitOn :: (Char -> Bool) -> String -> [String]
splitOn predicate :: Char -> Bool
predicate s :: String
s
| CountOf Word8
sz CountOf Word8 -> CountOf Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Int -> CountOf Word8
forall ty. Int -> CountOf ty
CountOf 0 = [String
forall a. Monoid a => a
mempty]
| Bool
otherwise = Offset Word8 -> Offset Word8 -> [String]
loop Offset Word8
forall a. Additive a => a
azero Offset Word8
forall a. Additive a => a
azero
where
!sz :: CountOf Word8
sz = String -> CountOf Word8
size String
s
end :: Offset Word8
end = Offset Word8
forall a. Additive a => a
azero Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
loop :: Offset Word8 -> Offset Word8 -> [String]
loop prevIdx :: Offset Word8
prevIdx idx :: Offset Word8
idx
| Offset Word8
idx Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
end = [String -> Offset Word8 -> Offset Word8 -> String
sub String
s Offset Word8
prevIdx Offset Word8
idx]
| Bool
otherwise =
let !(Step c :: Char
c idx' :: Offset Word8
idx') = String -> Offset Word8 -> Step
next String
s Offset Word8
idx
in if Char -> Bool
predicate Char
c
then String -> Offset Word8 -> Offset Word8 -> String
sub String
s Offset Word8
prevIdx Offset Word8
idx String -> [String] -> [String]
forall a. a -> [a] -> [a]
: Offset Word8 -> Offset Word8 -> [String]
loop Offset Word8
idx' Offset Word8
idx'
else Offset Word8 -> Offset Word8 -> [String]
loop Offset Word8
prevIdx Offset Word8
idx'
sub :: String -> Offset8 -> Offset8 -> String
sub :: String -> Offset Word8 -> Offset Word8 -> String
sub (String ba :: UArray Word8
ba) start :: Offset Word8
start end :: Offset Word8
end = UArray Word8 -> String
String (UArray Word8 -> String) -> UArray Word8 -> String
forall a b. (a -> b) -> a -> b
$ UArray Word8 -> Offset Word8 -> Offset Word8 -> UArray Word8
forall ty.
PrimType ty =>
UArray ty -> Offset ty -> Offset ty -> UArray ty
Vec.sub UArray Word8
ba Offset Word8
start Offset Word8
end
splitIndex :: Offset8 -> String -> (String, String)
splitIndex :: Offset Word8 -> String -> (String, String)
splitIndex idx :: Offset Word8
idx (String ba :: UArray Word8
ba) = (UArray Word8 -> String
String UArray Word8
v1, UArray Word8 -> String
String UArray Word8
v2)
where (v1 :: UArray Word8
v1,v2 :: UArray Word8
v2) = CountOf Word8 -> UArray Word8 -> (UArray Word8, UArray Word8)
forall ty. CountOf ty -> UArray ty -> (UArray ty, UArray ty)
C.splitAt (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
idx) UArray Word8
ba
break :: (Char -> Bool) -> String -> (String, String)
break :: (Char -> Bool) -> String -> (String, String)
break predicate :: Char -> Bool
predicate s :: String
s@(String ba :: UArray Word8
ba) = (forall s. ST s (String, String)) -> (String, String)
forall a. (forall s. ST s a) -> a
runST ((forall s. ST s (String, String)) -> (String, String))
-> (forall s. ST s (String, String)) -> (String, String)
forall a b. (a -> b) -> a -> b
$ UArray Word8
-> ((Offset Word8 -> Word8) -> ST s (String, String))
-> ST s (String, String)
forall (prim :: * -> *) ty a.
(PrimMonad prim, PrimType ty) =>
UArray ty -> ((Offset ty -> ty) -> prim a) -> prim a
Vec.unsafeIndexer UArray Word8
ba (Offset Word8 -> Word8) -> ST s (String, String)
forall st. (Offset Word8 -> Word8) -> ST st (String, String)
go
where
!sz :: CountOf Word8
sz = String -> CountOf Word8
size String
s
end :: Offset Word8
end = Offset Word8
forall a. Additive a => a
azero Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
go :: (Offset Word8 -> Word8) -> ST st (String, String)
go :: (Offset Word8 -> Word8) -> ST st (String, String)
go getIdx :: Offset Word8 -> Word8
getIdx = Offset Word8 -> ST st (String, String)
loop (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0)
where
!nextI :: Offset Word8 -> (Char, Offset Word8)
nextI = (Offset Word8 -> Word8) -> Offset Word8 -> (Char, Offset Word8)
nextWithIndexer Offset Word8 -> Word8
getIdx
loop :: Offset Word8 -> ST st (String, String)
loop idx :: Offset Word8
idx
| Offset Word8
idx Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
end = (String, String) -> ST st (String, String)
forall (m :: * -> *) a. Monad m => a -> m a
return (String
s, String
forall a. Monoid a => a
mempty)
| Bool
otherwise = do
let (c :: Char
c, idx' :: Offset Word8
idx') = Offset Word8 -> (Char, Offset Word8)
nextI Offset Word8
idx
case Char -> Bool
predicate Char
c of
True -> (String, String) -> ST st (String, String)
forall (m :: * -> *) a. Monad m => a -> m a
return ((String, String) -> ST st (String, String))
-> (String, String) -> ST st (String, String)
forall a b. (a -> b) -> a -> b
$ Offset Word8 -> String -> (String, String)
splitIndex Offset Word8
idx String
s
False -> Offset Word8 -> ST st (String, String)
loop Offset Word8
idx'
{-# INLINE loop #-}
{-# INLINE [2] break #-}
breakEnd :: (Char -> Bool) -> String -> (String, String)
breakEnd :: (Char -> Bool) -> String -> (String, String)
breakEnd predicate :: Char -> Bool
predicate s :: String
s@(String arr :: UArray Word8
arr)
| Offset Word8
k Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
end = (String
s, String
forall a. Monoid a => a
mempty)
| Bool
otherwise = Offset Word8 -> String -> (String, String)
splitIndex (Offset Word8
k Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Offset a -> Offset a -> Offset a
`offsetSub` Offset Word8
start) String
s
where
k :: Offset Word8
k = (Block Word8 -> Offset Word8)
-> (FinalPtr Word8 -> Ptr Word8 -> ST Any (Offset Word8))
-> UArray Word8
-> Offset Word8
forall ty a s.
(Block ty -> a)
-> (FinalPtr ty -> Ptr ty -> ST s a) -> UArray ty -> a
C.onBackend Block Word8 -> Offset Word8
goVec (\_ -> Offset Word8 -> ST Any (Offset Word8)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Offset Word8 -> ST Any (Offset Word8))
-> (Ptr Word8 -> Offset Word8)
-> Ptr Word8
-> ST Any (Offset Word8)
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. Ptr Word8 -> Offset Word8
goAddr) UArray Word8
arr
(C.ValidRange !Offset Word8
start !Offset Word8
end) = UArray Word8 -> ValidRange Word8
forall ty. UArray ty -> ValidRange ty
offsetsValidRange UArray Word8
arr
goVec :: Block Word8 -> Offset Word8
goVec ba :: Block Word8
ba@(Block !ByteArray#
_) = let k :: Offset Word8
k = (Char -> Bool)
-> Block Word8 -> Offset Word8 -> Offset Word8 -> Offset Word8
forall container.
Indexable container Word8 =>
(Char -> Bool)
-> container -> Offset Word8 -> Offset Word8 -> Offset Word8
Alg.revFindIndexPredicate Char -> Bool
predicate Block Word8
ba Offset Word8
start Offset Word8
end
in if Offset Word8
k Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
end then Offset Word8
end else Block Word8 -> Offset Word8 -> Offset Word8
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> Offset Word8
UTF8.nextSkip Block Word8
ba Offset Word8
k
goAddr :: Ptr Word8 -> Offset Word8
goAddr ptr :: Ptr Word8
ptr@(Ptr !Addr#
_) =
let k :: Offset Word8
k = (Char -> Bool)
-> Ptr Word8 -> Offset Word8 -> Offset Word8 -> Offset Word8
forall container.
Indexable container Word8 =>
(Char -> Bool)
-> container -> Offset Word8 -> Offset Word8 -> Offset Word8
Alg.revFindIndexPredicate Char -> Bool
predicate Ptr Word8
ptr Offset Word8
start Offset Word8
end
in if Offset Word8
k Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
end then Offset Word8
end else Ptr Word8 -> Offset Word8 -> Offset Word8
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> Offset Word8
UTF8.nextSkip Ptr Word8
ptr Offset Word8
k
{-# INLINE [2] breakEnd #-}
#if MIN_VERSION_base(4,9,0)
{-# RULES "break (== 'c')" [3] forall c . break (eqChar c) = breakElem c #-}
#else
{-# RULES "break (== 'c')" [3] forall c . break (== c) = breakElem c #-}
#endif
breakElem :: Char -> String -> (String, String)
breakElem :: Char -> String -> (String, String)
breakElem !Char
el s :: String
s@(String ba :: UArray Word8
ba)
| CountOf Word8
sz CountOf Word8 -> CountOf Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== 0 = (String
forall a. Monoid a => a
mempty, String
forall a. Monoid a => a
mempty)
| Bool
otherwise =
case Char -> UTF8Char
asUTF8Char Char
el of
UTF8_1 w :: Word8
w -> let !(v1 :: UArray Word8
v1,v2 :: UArray Word8
v2) = Word8 -> UArray Word8 -> (UArray Word8, UArray Word8)
forall ty. PrimType ty => ty -> UArray ty -> (UArray ty, UArray ty)
Vec.breakElem Word8
w UArray Word8
ba in (UArray Word8 -> String
String UArray Word8
v1, UArray Word8 -> String
String UArray Word8
v2)
_ -> (forall s. ST s (String, String)) -> (String, String)
forall a. (forall s. ST s a) -> a
runST ((forall s. ST s (String, String)) -> (String, String))
-> (forall s. ST s (String, String)) -> (String, String)
forall a b. (a -> b) -> a -> b
$ UArray Word8
-> ((Offset Word8 -> Word8) -> ST s (String, String))
-> ST s (String, String)
forall (prim :: * -> *) ty a.
(PrimMonad prim, PrimType ty) =>
UArray ty -> ((Offset ty -> ty) -> prim a) -> prim a
Vec.unsafeIndexer UArray Word8
ba (Offset Word8 -> Word8) -> ST s (String, String)
forall st. (Offset Word8 -> Word8) -> ST st (String, String)
go
where
sz :: CountOf Word8
sz = String -> CountOf Word8
size String
s
end :: Offset Word8
end = Offset Word8
forall a. Additive a => a
azero Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
go :: (Offset Word8 -> Word8) -> ST st (String, String)
go :: (Offset Word8 -> Word8) -> ST st (String, String)
go getIdx :: Offset Word8 -> Word8
getIdx = Offset Word8 -> ST st (String, String)
loop (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0)
where
!nextI :: Offset Word8 -> (Char, Offset Word8)
nextI = (Offset Word8 -> Word8) -> Offset Word8 -> (Char, Offset Word8)
nextWithIndexer Offset Word8 -> Word8
getIdx
loop :: Offset Word8 -> ST st (String, String)
loop idx :: Offset Word8
idx
| Offset Word8
idx Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
end = (String, String) -> ST st (String, String)
forall (m :: * -> *) a. Monad m => a -> m a
return (String
s, String
forall a. Monoid a => a
mempty)
| Bool
otherwise = do
let (c :: Char
c, idx' :: Offset Word8
idx') = Offset Word8 -> (Char, Offset Word8)
nextI Offset Word8
idx
case Char
el Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
== Char
c of
True -> (String, String) -> ST st (String, String)
forall (m :: * -> *) a. Monad m => a -> m a
return ((String, String) -> ST st (String, String))
-> (String, String) -> ST st (String, String)
forall a b. (a -> b) -> a -> b
$ Offset Word8 -> String -> (String, String)
splitIndex Offset Word8
idx String
s
False -> Offset Word8 -> ST st (String, String)
loop Offset Word8
idx'
breakLine :: String -> Either Bool (String, String)
breakLine :: String -> Either Bool (String, String)
breakLine (String arr :: UArray Word8
arr) = (UArray Word8 -> String)
-> (UArray Word8 -> String)
-> (UArray Word8, UArray Word8)
-> (String, String)
forall (p :: * -> * -> *) a b c d.
Bifunctor p =>
(a -> b) -> (c -> d) -> p a c -> p b d
bimap UArray Word8 -> String
String UArray Word8 -> String
String ((UArray Word8, UArray Word8) -> (String, String))
-> Either Bool (UArray Word8, UArray Word8)
-> Either Bool (String, String)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> UArray Word8 -> Either Bool (UArray Word8, UArray Word8)
Vec.breakLine UArray Word8
arr
span :: (Char -> Bool) -> String -> (String, String)
span :: (Char -> Bool) -> String -> (String, String)
span predicate :: Char -> Bool
predicate s :: String
s = (Char -> Bool) -> String -> (String, String)
break (Bool -> Bool
not (Bool -> Bool) -> (Char -> Bool) -> Char -> Bool
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. Char -> Bool
predicate) String
s
spanEnd :: (Char -> Bool) -> String -> (String, String)
spanEnd :: (Char -> Bool) -> String -> (String, String)
spanEnd predicate :: Char -> Bool
predicate s :: String
s = (Char -> Bool) -> String -> (String, String)
breakEnd (Bool -> Bool
not (Bool -> Bool) -> (Char -> Bool) -> Char -> Bool
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. Char -> Bool
predicate) String
s
dropWhile :: (Char -> Bool) -> String -> String
dropWhile :: (Char -> Bool) -> String -> String
dropWhile predicate :: Char -> Bool
predicate = (String, String) -> String
forall a b. (a, b) -> b
snd ((String, String) -> String)
-> (String -> (String, String)) -> String -> String
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. (Char -> Bool) -> String -> (String, String)
break (Bool -> Bool
not (Bool -> Bool) -> (Char -> Bool) -> Char -> Bool
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. Char -> Bool
predicate)
elem :: Char -> String -> Bool
elem :: Char -> String -> Bool
elem !Char
el s :: String
s@(String ba :: UArray Word8
ba) =
case Char -> UTF8Char
asUTF8Char Char
el of
UTF8_1 w :: Word8
w -> Word8 -> UArray Word8 -> Bool
forall ty. PrimType ty => ty -> UArray ty -> Bool
Vec.elem Word8
w UArray Word8
ba
_ -> (forall s. ST s Bool) -> Bool
forall a. (forall s. ST s a) -> a
runST ((forall s. ST s Bool) -> Bool) -> (forall s. ST s Bool) -> Bool
forall a b. (a -> b) -> a -> b
$ UArray Word8 -> ((Offset Word8 -> Word8) -> ST s Bool) -> ST s Bool
forall (prim :: * -> *) ty a.
(PrimMonad prim, PrimType ty) =>
UArray ty -> ((Offset ty -> ty) -> prim a) -> prim a
Vec.unsafeIndexer UArray Word8
ba (Offset Word8 -> Word8) -> ST s Bool
forall st. (Offset Word8 -> Word8) -> ST st Bool
go
where
sz :: CountOf Word8
sz = String -> CountOf Word8
size String
s
end :: Offset Word8
end = Offset Word8
forall a. Additive a => a
azero Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
go :: (Offset Word8 -> Word8) -> ST st Bool
go :: (Offset Word8 -> Word8) -> ST st Bool
go getIdx :: Offset Word8 -> Word8
getIdx = Offset Word8 -> ST st Bool
loop (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0)
where
!nextI :: Offset Word8 -> (Char, Offset Word8)
nextI = (Offset Word8 -> Word8) -> Offset Word8 -> (Char, Offset Word8)
nextWithIndexer Offset Word8 -> Word8
getIdx
loop :: Offset Word8 -> ST st Bool
loop !Offset Word8
idx
| Offset Word8
idx Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
end = Bool -> ST st Bool
forall (m :: * -> *) a. Monad m => a -> m a
return Bool
False
| Bool
otherwise = do
let (c :: Char
c, idx' :: Offset Word8
idx') = Offset Word8 -> (Char, Offset Word8)
nextI Offset Word8
idx
case Char
el Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
== Char
c of
True -> Bool -> ST st Bool
forall (m :: * -> *) a. Monad m => a -> m a
return Bool
True
False -> Offset Word8 -> ST st Bool
loop Offset Word8
idx'
intersperse :: Char -> String -> String
intersperse :: Char -> String -> String
intersperse sep :: Char
sep src :: String
src = case String -> CountOf Char
length String
src CountOf Char -> CountOf Char -> Difference (CountOf Char)
forall a. Subtractive a => a -> a -> Difference a
- 1 of
Nothing -> String
src
Just 0 -> String
src
Just gaps -> (forall s. ST s String) -> String
forall a. (forall s. ST s a) -> a
runST ((forall s. ST s String) -> String)
-> (forall s. ST s String) -> String
forall a b. (a -> b) -> a -> b
$ String
-> CountOf Word8
-> (String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8))
-> ST s String
forall s.
String
-> CountOf Word8
-> (String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8))
-> ST s String
unsafeCopyFrom String
src CountOf Word8
dstBytes String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8)
forall s.
String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8)
go
where
lastSrcI :: Offset Char
lastSrcI :: Offset Char
lastSrcI = 0 Offset Char -> CountOf Char -> Offset Char
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Char
gaps
dstBytes :: CountOf Word8
dstBytes = (String -> CountOf Word8
size String
src :: CountOf Word8) CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ (CountOf Char
gaps CountOf Char -> CountOf Word8 -> CountOf Word8
forall a n. (Additive a, IsNatural n) => n -> a -> a
`scale` Int -> CountOf Word8
charToBytes (Char -> Int
forall a. Enum a => a -> Int
fromEnum Char
sep))
go :: String -> Offset Char -> Offset8 -> MutableString s -> Offset8 -> ST s (Offset8, Offset8)
go :: String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8)
go src' :: String
src' srcI :: Offset Char
srcI srcIdx :: Offset Word8
srcIdx dst :: MutableString s
dst dstIdx :: Offset Word8
dstIdx
| Offset Char
srcI Offset Char -> Offset Char -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Char
lastSrcI = do
Offset Word8
nextDstIdx <- MutableString (PrimState (ST s))
-> Offset Word8 -> Char -> ST s (Offset Word8)
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
write MutableString s
MutableString (PrimState (ST s))
dst Offset Word8
dstIdx Char
c
(Offset Word8, Offset Word8) -> ST s (Offset Word8, Offset Word8)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
nextSrcIdx, Offset Word8
nextDstIdx)
| Bool
otherwise = do
Offset Word8
nextDstIdx <- MutableString (PrimState (ST s))
-> Offset Word8 -> Char -> ST s (Offset Word8)
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
write MutableString s
MutableString (PrimState (ST s))
dst Offset Word8
dstIdx Char
c
Offset Word8
nextDstIdx' <- MutableString (PrimState (ST s))
-> Offset Word8 -> Char -> ST s (Offset Word8)
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
write MutableString s
MutableString (PrimState (ST s))
dst Offset Word8
nextDstIdx Char
sep
(Offset Word8, Offset Word8) -> ST s (Offset Word8, Offset Word8)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8
nextSrcIdx, Offset Word8
nextDstIdx')
where
!(Step c :: Char
c nextSrcIdx :: Offset Word8
nextSrcIdx) = String -> Offset Word8 -> Step
next String
src' Offset Word8
srcIdx
unsafeCopyFrom :: String
-> CountOf Word8
-> (String -> Offset Char -> Offset8 -> MutableString s -> Offset8 -> ST s (Offset8, Offset8))
-> ST s String
unsafeCopyFrom :: String
-> CountOf Word8
-> (String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8))
-> ST s String
unsafeCopyFrom src :: String
src dstBytes :: CountOf Word8
dstBytes f :: String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8)
f = CountOf Word8 -> ST s (MutableString (PrimState (ST s)))
forall (prim :: * -> *).
PrimMonad prim =>
CountOf Word8 -> prim (MutableString (PrimState prim))
new CountOf Word8
dstBytes ST s (MutableString s)
-> (MutableString s -> ST s (MutableString s))
-> ST s (MutableString s)
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= Offset Char
-> Offset Word8
-> Offset Word8
-> (String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8))
-> MutableString s
-> ST s (MutableString s)
fill (Int -> Offset Char
forall ty. Int -> Offset ty
Offset 0) (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8)
f ST s (MutableString s)
-> (MutableString s -> ST s String) -> ST s String
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= MutableString s -> ST s String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> prim String
freeze
where
srcLen :: CountOf Char
srcLen = String -> CountOf Char
length String
src
end :: Offset Char
end = Int -> Offset Char
forall ty. Int -> Offset ty
Offset 0 Offset Char -> CountOf Char -> Offset Char
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Char
srcLen
fill :: Offset Char
-> Offset Word8
-> Offset Word8
-> (String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8))
-> MutableString s
-> ST s (MutableString s)
fill srcI :: Offset Char
srcI srcIdx :: Offset Word8
srcIdx dstIdx :: Offset Word8
dstIdx f' :: String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8)
f' dst' :: MutableString s
dst'
| Offset Char
srcI Offset Char -> Offset Char -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Char
end = MutableString s -> ST s (MutableString s)
forall (m :: * -> *) a. Monad m => a -> m a
return MutableString s
dst'
| Bool
otherwise = do (nextSrcIdx :: Offset Word8
nextSrcIdx, nextDstIdx :: Offset Word8
nextDstIdx) <- String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8)
f' String
src Offset Char
srcI Offset Word8
srcIdx MutableString s
dst' Offset Word8
dstIdx
Offset Char
-> Offset Word8
-> Offset Word8
-> (String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8))
-> MutableString s
-> ST s (MutableString s)
fill (Offset Char
srcI Offset Char -> Offset Char -> Offset Char
forall a. Additive a => a -> a -> a
+ Int -> Offset Char
forall ty. Int -> Offset ty
Offset 1) Offset Word8
nextSrcIdx Offset Word8
nextDstIdx String
-> Offset Char
-> Offset Word8
-> MutableString s
-> Offset Word8
-> ST s (Offset Word8, Offset Word8)
f' MutableString s
dst'
length :: String -> CountOf Char
length :: String -> CountOf Char
length (String arr :: UArray Word8
arr)
| Offset Word8
start Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
end = 0
| Bool
otherwise = (Block Word8 -> CountOf Char)
-> (FinalPtr Word8 -> Ptr Word8 -> ST Any (CountOf Char))
-> UArray Word8
-> CountOf Char
forall ty a s.
(Block ty -> a)
-> (FinalPtr ty -> Ptr ty -> ST s a) -> UArray ty -> a
C.onBackend Block Word8 -> CountOf Char
goVec (\_ -> CountOf Char -> ST Any (CountOf Char)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (CountOf Char -> ST Any (CountOf Char))
-> (Ptr Word8 -> CountOf Char)
-> Ptr Word8
-> ST Any (CountOf Char)
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. Ptr Word8 -> CountOf Char
goAddr) UArray Word8
arr
where
(C.ValidRange !Offset Word8
start !Offset Word8
end) = UArray Word8 -> ValidRange Word8
forall ty. UArray ty -> ValidRange ty
offsetsValidRange UArray Word8
arr
goVec :: Block Word8 -> CountOf Char
goVec ma :: Block Word8
ma = Block Word8 -> Offset Word8 -> Offset Word8 -> CountOf Char
forall container.
(Indexable container Word8, Indexable container Word64) =>
container -> Offset Word8 -> Offset Word8 -> CountOf Char
UTF8.length Block Word8
ma Offset Word8
start Offset Word8
end
goAddr :: Ptr Word8 -> CountOf Char
goAddr ptr :: Ptr Word8
ptr = Ptr Word8 -> Offset Word8 -> Offset Word8 -> CountOf Char
forall container.
(Indexable container Word8, Indexable container Word64) =>
container -> Offset Word8 -> Offset Word8 -> CountOf Char
UTF8.length Ptr Word8
ptr Offset Word8
start Offset Word8
end
replicate :: CountOf Char -> Char -> String
replicate :: CountOf Char -> Char -> String
replicate (CountOf n :: Int
n) c :: Char
c = (forall s. ST s String) -> String
forall a. (forall s. ST s a) -> a
runST (CountOf Word8 -> ST s (MutableString (PrimState (ST s)))
forall (prim :: * -> *).
PrimMonad prim =>
CountOf Word8 -> prim (MutableString (PrimState prim))
new CountOf Word8
nbBytes ST s (MutableString s)
-> (MutableString s -> ST s String) -> ST s String
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= MutableString s -> ST s String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> prim String
fill)
where
nbBytes :: CountOf Word8
nbBytes = Word -> CountOf Word8 -> CountOf Word8
forall a n. (Additive a, IsNatural n) => n -> a -> a
scale (Int -> Word
forall source destination.
Cast source destination =>
source -> destination
cast Int
n :: Word) CountOf Word8
sz
sz :: CountOf Word8
sz = Int -> CountOf Word8
charToBytes (Char -> Int
forall a. Enum a => a -> Int
fromEnum Char
c)
fill :: PrimMonad prim => MutableString (PrimState prim) -> prim String
fill :: MutableString (PrimState prim) -> prim String
fill ms :: MutableString (PrimState prim)
ms = Offset Word8 -> prim String
loop (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0)
where
loop :: Offset Word8 -> prim String
loop idx :: Offset Word8
idx
| Offset Word8
idx Offset Word8 -> CountOf Word8 -> Bool
forall ty. Offset ty -> CountOf ty -> Bool
.==# CountOf Word8
nbBytes = MutableString (PrimState prim) -> prim String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> prim String
freeze MutableString (PrimState prim)
ms
| Bool
otherwise = MutableString (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
write MutableString (PrimState prim)
ms Offset Word8
idx Char
c prim (Offset Word8) -> (Offset Word8 -> prim String) -> prim String
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= Offset Word8 -> prim String
loop
copy :: String -> String
copy :: String -> String
copy (String s :: UArray Word8
s) = UArray Word8 -> String
String (UArray Word8 -> UArray Word8
forall ty. PrimType ty => UArray ty -> UArray ty
Vec.copy UArray Word8
s)
singleton :: Char -> String
singleton :: Char -> String
singleton c :: Char
c = (forall s. ST s String) -> String
forall a. (forall s. ST s a) -> a
runST ((forall s. ST s String) -> String)
-> (forall s. ST s String) -> String
forall a b. (a -> b) -> a -> b
$ do
MutableString s
ms <- CountOf Word8 -> ST s (MutableString (PrimState (ST s)))
forall (prim :: * -> *).
PrimMonad prim =>
CountOf Word8 -> prim (MutableString (PrimState prim))
new CountOf Word8
nbBytes
Offset Word8
_ <- MutableString (PrimState (ST s))
-> Offset Word8 -> Char -> ST s (Offset Word8)
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
write MutableString s
MutableString (PrimState (ST s))
ms (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) Char
c
MutableString (PrimState (ST s)) -> ST s String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> prim String
freeze MutableString s
MutableString (PrimState (ST s))
ms
where
!nbBytes :: CountOf Word8
nbBytes = Int -> CountOf Word8
charToBytes (Char -> Int
forall a. Enum a => a -> Int
fromEnum Char
c)
create :: PrimMonad prim => CountOf Word8 -> (MutableString (PrimState prim) -> prim (Offset Word8)) -> prim String
create :: CountOf Word8
-> (MutableString (PrimState prim) -> prim (Offset Word8))
-> prim String
create sz :: CountOf Word8
sz f :: MutableString (PrimState prim) -> prim (Offset Word8)
f = do
MutableString (PrimState prim)
ms <- CountOf Word8 -> prim (MutableString (PrimState prim))
forall (prim :: * -> *).
PrimMonad prim =>
CountOf Word8 -> prim (MutableString (PrimState prim))
new CountOf Word8
sz
Offset Word8
filled <- MutableString (PrimState prim) -> prim (Offset Word8)
f MutableString (PrimState prim)
ms
if Offset Word8
filled Offset Word8 -> CountOf Word8 -> Bool
forall ty. Offset ty -> CountOf ty -> Bool
.==# CountOf Word8
sz
then MutableString (PrimState prim) -> prim String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> prim String
freeze MutableString (PrimState prim)
ms
else do
String
s <- MutableString (PrimState prim) -> prim String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> prim String
freeze MutableString (PrimState prim)
ms
let (String ba :: UArray Word8
ba) = String
s
String -> prim String
forall (f :: * -> *) a. Applicative f => a -> f a
pure (String -> prim String) -> String -> prim String
forall a b. (a -> b) -> a -> b
$ UArray Word8 -> String
String (UArray Word8 -> String) -> UArray Word8 -> String
forall a b. (a -> b) -> a -> b
$ CountOf Word8 -> UArray Word8 -> UArray Word8
forall ty. CountOf ty -> UArray ty -> UArray ty
C.take (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
filled) UArray Word8
ba
charMap :: (Char -> Char) -> String -> String
charMap :: (Char -> Char) -> String -> String
charMap f :: Char -> Char
f src :: String
src
| CountOf Word8
srcSz CountOf Word8 -> CountOf Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== 0 = String
forall a. Monoid a => a
mempty
| Bool
otherwise =
let !(elems :: [(String, CountOf Word8)]
elems, nbBytes :: CountOf Word8
nbBytes) = [(String, CountOf Word8)]
-> Offset Word8
-> CountOf Word8
-> ([(String, CountOf Word8)], CountOf Word8)
allocateAndFill [] (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) (Int -> CountOf Word8
forall ty. Int -> CountOf ty
CountOf 0)
in (forall s. ST s String) -> String
forall a. (forall s. ST s a) -> a
runST ((forall s. ST s String) -> String)
-> (forall s. ST s String) -> String
forall a b. (a -> b) -> a -> b
$ do
MutableString s
dest <- CountOf Word8 -> ST s (MutableString (PrimState (ST s)))
forall (prim :: * -> *).
PrimMonad prim =>
CountOf Word8 -> prim (MutableString (PrimState prim))
new CountOf Word8
nbBytes
MutableString (PrimState (ST s))
-> [(String, CountOf Word8)] -> Offset Word8 -> ST s ()
forall (m :: * -> *).
PrimMonad m =>
MutableString (PrimState m)
-> [(String, CountOf Word8)] -> Offset Word8 -> m ()
copyLoop MutableString s
MutableString (PrimState (ST s))
dest [(String, CountOf Word8)]
elems (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0 Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
nbBytes)
MutableString (PrimState (ST s)) -> ST s String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> prim String
freeze MutableString s
MutableString (PrimState (ST s))
dest
where
!srcSz :: CountOf Word8
srcSz = String -> CountOf Word8
size String
src
srcEnd :: Offset Word8
srcEnd = Offset Word8
forall a. Additive a => a
azero Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
srcSz
allocateAndFill :: [(String, CountOf Word8)]
-> Offset8
-> CountOf Word8
-> ([(String,CountOf Word8)], CountOf Word8)
allocateAndFill :: [(String, CountOf Word8)]
-> Offset Word8
-> CountOf Word8
-> ([(String, CountOf Word8)], CountOf Word8)
allocateAndFill acc :: [(String, CountOf Word8)]
acc idx :: Offset Word8
idx bytesWritten :: CountOf Word8
bytesWritten
| Offset Word8
idx Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
srcEnd = ([(String, CountOf Word8)]
acc, CountOf Word8
bytesWritten)
| Bool
otherwise =
let (el :: (String, CountOf Word8)
el@(_,addBytes :: CountOf Word8
addBytes), idx' :: Offset Word8
idx') = (forall s. ST s ((String, CountOf Word8), Offset Word8))
-> ((String, CountOf Word8), Offset Word8)
forall a. (forall s. ST s a) -> a
runST ((forall s. ST s ((String, CountOf Word8), Offset Word8))
-> ((String, CountOf Word8), Offset Word8))
-> (forall s. ST s ((String, CountOf Word8), Offset Word8))
-> ((String, CountOf Word8), Offset Word8)
forall a b. (a -> b) -> a -> b
$ do
let !diffBytes :: Difference (Offset Word8)
diffBytes = Offset Word8
srcEnd Offset Word8 -> Offset Word8 -> Difference (Offset Word8)
forall a. Subtractive a => a -> a -> Difference a
- Offset Word8
idx
!allocatedBytes :: CountOf Word8
allocatedBytes = if Difference (Offset Word8)
CountOf Word8
diffBytes CountOf Word8 -> CountOf Word8 -> Bool
forall a. Ord a => a -> a -> Bool
<= Int -> CountOf Word8
forall ty. Int -> CountOf ty
CountOf 4 then Int -> CountOf Word8
forall ty. Int -> CountOf ty
CountOf 4 else Difference (Offset Word8)
CountOf Word8
diffBytes
MutableString s
ms <- CountOf Word8 -> ST s (MutableString (PrimState (ST s)))
forall (prim :: * -> *).
PrimMonad prim =>
CountOf Word8 -> prim (MutableString (PrimState prim))
new CountOf Word8
allocatedBytes
(dstIdx :: CountOf Word8
dstIdx, srcIdx :: Offset Word8
srcIdx) <- MutableString (PrimState (ST s))
-> CountOf Word8
-> Offset Word8
-> ST s (CountOf Word8, Offset Word8)
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim)
-> CountOf Word8
-> Offset Word8
-> prim (CountOf Word8, Offset Word8)
fill MutableString s
MutableString (PrimState (ST s))
ms CountOf Word8
allocatedBytes Offset Word8
idx
String
s <- MutableString (PrimState (ST s)) -> ST s String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> prim String
freeze MutableString s
MutableString (PrimState (ST s))
ms
((String, CountOf Word8), Offset Word8)
-> ST s ((String, CountOf Word8), Offset Word8)
forall (m :: * -> *) a. Monad m => a -> m a
return ((String
s, CountOf Word8
dstIdx), Offset Word8
srcIdx)
in [(String, CountOf Word8)]
-> Offset Word8
-> CountOf Word8
-> ([(String, CountOf Word8)], CountOf Word8)
allocateAndFill ((String, CountOf Word8)
el (String, CountOf Word8)
-> [(String, CountOf Word8)] -> [(String, CountOf Word8)]
forall a. a -> [a] -> [a]
: [(String, CountOf Word8)]
acc) Offset Word8
idx' (CountOf Word8
bytesWritten CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ CountOf Word8
addBytes)
fill :: PrimMonad prim
=> MutableString (PrimState prim)
-> CountOf Word8
-> Offset8
-> prim (CountOf Word8, Offset8)
fill :: MutableString (PrimState prim)
-> CountOf Word8
-> Offset Word8
-> prim (CountOf Word8, Offset Word8)
fill mba :: MutableString (PrimState prim)
mba dsz :: CountOf Word8
dsz srcIdxOrig :: Offset Word8
srcIdxOrig =
Offset Word8 -> Offset Word8 -> prim (CountOf Word8, Offset Word8)
loop (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) Offset Word8
srcIdxOrig
where
endDst :: Offset Word8
endDst = (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
dsz
loop :: Offset Word8 -> Offset Word8 -> prim (CountOf Word8, Offset Word8)
loop dstIdx :: Offset Word8
dstIdx srcIdx :: Offset Word8
srcIdx
| Offset Word8
srcIdx Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
srcEnd = (CountOf Word8, Offset Word8) -> prim (CountOf Word8, Offset Word8)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
dstIdx, Offset Word8
srcIdx)
| Offset Word8
dstIdx Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
endDst = (CountOf Word8, Offset Word8) -> prim (CountOf Word8, Offset Word8)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
dstIdx, Offset Word8
srcIdx)
| Bool
otherwise =
let !(Step c :: Char
c srcIdx' :: Offset Word8
srcIdx') = String -> Offset Word8 -> Step
next String
src Offset Word8
srcIdx
c' :: Char
c' = Char -> Char
f Char
c
!nbBytes :: CountOf Word8
nbBytes = Int -> CountOf Word8
charToBytes (Char -> Int
forall a. Enum a => a -> Int
fromEnum Char
c')
in
if Offset Word8
dstIdx Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
nbBytes Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
<= CountOf Word8 -> Offset Word8
forall a. CountOf a -> Offset a
sizeAsOffset CountOf Word8
dsz
then do Offset Word8
dstIdx' <- MutableString (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
write MutableString (PrimState prim)
mba Offset Word8
dstIdx Char
c'
Offset Word8 -> Offset Word8 -> prim (CountOf Word8, Offset Word8)
loop Offset Word8
dstIdx' Offset Word8
srcIdx'
else (CountOf Word8, Offset Word8) -> prim (CountOf Word8, Offset Word8)
forall (m :: * -> *) a. Monad m => a -> m a
return (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
dstIdx, Offset Word8
srcIdx)
copyLoop :: MutableString (PrimState m)
-> [(String, CountOf Word8)] -> Offset Word8 -> m ()
copyLoop _ [] (Offset 0) = () -> m ()
forall (m :: * -> *) a. Monad m => a -> m a
return ()
copyLoop _ [] n :: Offset Word8
n = [Char] -> m ()
forall a. HasCallStack => [Char] -> a
error ("charMap invalid: " [Char] -> [Char] -> [Char]
forall a. Semigroup a => a -> a -> a
<> Offset Word8 -> [Char]
forall a. Show a => a -> [Char]
show Offset Word8
n)
copyLoop ms :: MutableString (PrimState m)
ms@(MutableString mba :: MUArray Word8 (PrimState m)
mba) ((String ba :: UArray Word8
ba, sz :: CountOf Word8
sz):xs :: [(String, CountOf Word8)]
xs) end :: Offset Word8
end = do
let start :: Offset Word8
start = Offset Word8
end Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetMinusE` CountOf Word8
sz
MUArray Word8 (PrimState m)
-> Offset Word8
-> UArray Word8
-> Offset Word8
-> CountOf Word8
-> m ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim)
-> Offset ty -> UArray ty -> Offset ty -> CountOf ty -> prim ()
Vec.unsafeCopyAtRO MUArray Word8 (PrimState m)
mba Offset Word8
start UArray Word8
ba (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) CountOf Word8
sz
MutableString (PrimState m)
-> [(String, CountOf Word8)] -> Offset Word8 -> m ()
copyLoop MutableString (PrimState m)
ms [(String, CountOf Word8)]
xs Offset Word8
start
snoc :: String -> Char -> String
snoc :: String -> Char -> String
snoc s :: String
s@(String ba :: UArray Word8
ba) c :: Char
c
| CountOf Word8
len CountOf Word8 -> CountOf Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Int -> CountOf Word8
forall ty. Int -> CountOf ty
CountOf 0 = Char -> String
singleton Char
c
| Bool
otherwise = (forall s. ST s String) -> String
forall a. (forall s. ST s a) -> a
runST ((forall s. ST s String) -> String)
-> (forall s. ST s String) -> String
forall a b. (a -> b) -> a -> b
$ do
MutableString s
ms <- CountOf Word8 -> ST s (MutableString (PrimState (ST s)))
forall (prim :: * -> *).
PrimMonad prim =>
CountOf Word8 -> prim (MutableString (PrimState prim))
new (CountOf Word8
len CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ CountOf Word8
nbBytes)
let (MutableString mba :: MUArray Word8 s
mba) = MutableString s
ms
MUArray Word8 (PrimState (ST s))
-> Offset Word8
-> UArray Word8
-> Offset Word8
-> CountOf Word8
-> ST s ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim)
-> Offset ty -> UArray ty -> Offset ty -> CountOf ty -> prim ()
Vec.unsafeCopyAtRO MUArray Word8 s
MUArray Word8 (PrimState (ST s))
mba (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) UArray Word8
ba (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) CountOf Word8
len
Offset Word8
_ <- MutableString (PrimState (ST s))
-> Offset Word8 -> Char -> ST s (Offset Word8)
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
write MutableString s
MutableString (PrimState (ST s))
ms (Offset Word8
forall a. Additive a => a
azero Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
len) Char
c
MutableString (PrimState (ST s)) -> ST s String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> prim String
freeze MutableString s
MutableString (PrimState (ST s))
ms
where
!len :: CountOf Word8
len = String -> CountOf Word8
size String
s
!nbBytes :: CountOf Word8
nbBytes = Int -> CountOf Word8
charToBytes (Char -> Int
forall a. Enum a => a -> Int
fromEnum Char
c)
cons :: Char -> String -> String
cons :: Char -> String -> String
cons c :: Char
c s :: String
s@(String ba :: UArray Word8
ba)
| CountOf Word8
len CountOf Word8 -> CountOf Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Int -> CountOf Word8
forall ty. Int -> CountOf ty
CountOf 0 = Char -> String
singleton Char
c
| Bool
otherwise = (forall s. ST s String) -> String
forall a. (forall s. ST s a) -> a
runST ((forall s. ST s String) -> String)
-> (forall s. ST s String) -> String
forall a b. (a -> b) -> a -> b
$ do
MutableString s
ms <- CountOf Word8 -> ST s (MutableString (PrimState (ST s)))
forall (prim :: * -> *).
PrimMonad prim =>
CountOf Word8 -> prim (MutableString (PrimState prim))
new (CountOf Word8
len CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ CountOf Word8
nbBytes)
let (MutableString mba :: MUArray Word8 s
mba) = MutableString s
ms
Offset Word8
idx <- MutableString (PrimState (ST s))
-> Offset Word8 -> Char -> ST s (Offset Word8)
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
write MutableString s
MutableString (PrimState (ST s))
ms (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) Char
c
MUArray Word8 (PrimState (ST s))
-> Offset Word8
-> UArray Word8
-> Offset Word8
-> CountOf Word8
-> ST s ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim)
-> Offset ty -> UArray ty -> Offset ty -> CountOf ty -> prim ()
Vec.unsafeCopyAtRO MUArray Word8 s
MUArray Word8 (PrimState (ST s))
mba Offset Word8
idx UArray Word8
ba (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) CountOf Word8
len
MutableString (PrimState (ST s)) -> ST s String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> prim String
freeze MutableString s
MutableString (PrimState (ST s))
ms
where
!len :: CountOf Word8
len = String -> CountOf Word8
size String
s
!nbBytes :: CountOf Word8
nbBytes = Int -> CountOf Word8
charToBytes (Char -> Int
forall a. Enum a => a -> Int
fromEnum Char
c)
unsnoc :: String -> Maybe (String, Char)
unsnoc :: String -> Maybe (String, Char)
unsnoc s :: String
s@(String arr :: UArray Word8
arr)
| CountOf Word8
sz CountOf Word8 -> CountOf Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== 0 = Maybe (String, Char)
forall a. Maybe a
Nothing
| Bool
otherwise =
let !(StepBack c :: Char
c idx :: Offset Word8
idx) = String -> Offset Word8 -> StepBack
prev String
s (CountOf Word8 -> Offset Word8
forall a. CountOf a -> Offset a
sizeAsOffset CountOf Word8
sz)
in (String, Char) -> Maybe (String, Char)
forall a. a -> Maybe a
Just (UArray Word8 -> String
String (UArray Word8 -> String) -> UArray Word8 -> String
forall a b. (a -> b) -> a -> b
$ CountOf Word8 -> UArray Word8 -> UArray Word8
forall ty. CountOf ty -> UArray ty -> UArray ty
Vec.take (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
idx) UArray Word8
arr, Char
c)
where
sz :: CountOf Word8
sz = String -> CountOf Word8
size String
s
uncons :: String -> Maybe (Char, String)
uncons :: String -> Maybe (Char, String)
uncons s :: String
s@(String ba :: UArray Word8
ba)
| String -> Bool
null String
s = Maybe (Char, String)
forall a. Maybe a
Nothing
| Bool
otherwise =
let !(Step c :: Char
c idx :: Offset Word8
idx) = String -> Offset Word8 -> Step
next String
s Offset Word8
forall a. Additive a => a
azero
in (Char, String) -> Maybe (Char, String)
forall a. a -> Maybe a
Just (Char
c, UArray Word8 -> String
String (UArray Word8 -> String) -> UArray Word8 -> String
forall a b. (a -> b) -> a -> b
$ CountOf Word8 -> UArray Word8 -> UArray Word8
forall ty. CountOf ty -> UArray ty -> UArray ty
Vec.drop (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
idx) UArray Word8
ba)
find :: (Char -> Bool) -> String -> Maybe Char
find :: (Char -> Bool) -> String -> Maybe Char
find predicate :: Char -> Bool
predicate s :: String
s = Offset Word8 -> Maybe Char
loop (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0)
where
!sz :: CountOf Word8
sz = String -> CountOf Word8
size String
s
end :: Offset Word8
end = Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0 Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
loop :: Offset Word8 -> Maybe Char
loop idx :: Offset Word8
idx
| Offset Word8
idx Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
end = Maybe Char
forall a. Maybe a
Nothing
| Bool
otherwise =
let !(Step c :: Char
c idx' :: Offset Word8
idx') = String -> Offset Word8 -> Step
next String
s Offset Word8
idx
in case Char -> Bool
predicate Char
c of
True -> Char -> Maybe Char
forall a. a -> Maybe a
Just Char
c
False -> Offset Word8 -> Maybe Char
loop Offset Word8
idx'
sortBy :: (Char -> Char -> Ordering) -> String -> String
sortBy :: (Char -> Char -> Ordering) -> String -> String
sortBy sortF :: Char -> Char -> Ordering
sortF s :: String
s = [Item String] -> String
forall l. IsList l => [Item l] -> l
fromList ([Item String] -> String) -> [Item String] -> String
forall a b. (a -> b) -> a -> b
$ (Char -> Char -> Ordering) -> [Char] -> [Char]
forall a. (a -> a -> Ordering) -> [a] -> [a]
Data.List.sortBy Char -> Char -> Ordering
sortF ([Char] -> [Item String]) -> [Char] -> [Item String]
forall a b. (a -> b) -> a -> b
$ String -> [Item String]
forall l. IsList l => l -> [Item l]
toList String
s
filter :: (Char -> Bool) -> String -> String
filter :: (Char -> Bool) -> String -> String
filter predicate :: Char -> Bool
predicate (String arr :: UArray Word8
arr) = (forall s. ST s String) -> String
forall a. (forall s. ST s a) -> a
runST ((forall s. ST s String) -> String)
-> (forall s. ST s String) -> String
forall a b. (a -> b) -> a -> b
$ do
(finalSize :: CountOf Word8
finalSize, dst :: MutableString s
dst) <- CountOf Word8
-> (MutableBlock Word8 (PrimState (ST s)) -> ST s (CountOf Word8))
-> ST s (CountOf Word8, MutableString (PrimState (ST s)))
forall (prim :: * -> *) a.
PrimMonad prim =>
CountOf Word8
-> (MutableBlock Word8 (PrimState prim) -> prim a)
-> prim (a, MutableString (PrimState prim))
newNative CountOf Word8
sz ((MutableBlock Word8 (PrimState (ST s)) -> ST s (CountOf Word8))
-> ST s (CountOf Word8, MutableString s))
-> (MutableBlock Word8 (PrimState (ST s)) -> ST s (CountOf Word8))
-> ST s (CountOf Word8, MutableString s)
forall a b. (a -> b) -> a -> b
$ \(MutableBlock mba :: MutableByteArray# (PrimState (ST s))
mba) ->
(Block Word8 -> ST s (CountOf Word8))
-> (FinalPtr Word8 -> ST s (CountOf Word8))
-> UArray Word8
-> ST s (CountOf Word8)
forall (prim :: * -> *) ty a.
PrimMonad prim =>
(Block ty -> prim a)
-> (FinalPtr ty -> prim a) -> UArray ty -> prim a
C.onBackendPrim (\ba :: Block Word8
ba@(Block !ByteArray#
_) -> (Char -> Bool)
-> CountOf Word8
-> MutableByteArray# s
-> Block Word8
-> Offset Word8
-> ST s (CountOf Word8)
forall s container.
Indexable container Word8 =>
(Char -> Bool)
-> CountOf Word8
-> MutableByteArray# s
-> container
-> Offset Word8
-> ST s (CountOf Word8)
Alg.copyFilter Char -> Bool
predicate CountOf Word8
sz MutableByteArray# s
MutableByteArray# (PrimState (ST s))
mba Block Word8
ba Offset Word8
start)
(\fptr :: FinalPtr Word8
fptr -> FinalPtr Word8
-> (Ptr Word8 -> ST s (CountOf Word8)) -> ST s (CountOf Word8)
forall (prim :: * -> *) p a.
PrimMonad prim =>
FinalPtr p -> (Ptr p -> prim a) -> prim a
withFinalPtr FinalPtr Word8
fptr ((Ptr Word8 -> ST s (CountOf Word8)) -> ST s (CountOf Word8))
-> (Ptr Word8 -> ST s (CountOf Word8)) -> ST s (CountOf Word8)
forall a b. (a -> b) -> a -> b
$ \ptr :: Ptr Word8
ptr@(Ptr !Addr#
_) -> (Char -> Bool)
-> CountOf Word8
-> MutableByteArray# s
-> Ptr Word8
-> Offset Word8
-> ST s (CountOf Word8)
forall s container.
Indexable container Word8 =>
(Char -> Bool)
-> CountOf Word8
-> MutableByteArray# s
-> container
-> Offset Word8
-> ST s (CountOf Word8)
Alg.copyFilter Char -> Bool
predicate CountOf Word8
sz MutableByteArray# s
MutableByteArray# (PrimState (ST s))
mba Ptr Word8
ptr Offset Word8
start)
UArray Word8
arr
CountOf Word8 -> MutableString (PrimState (ST s)) -> ST s String
forall (prim :: * -> *).
PrimMonad prim =>
CountOf Word8 -> MutableString (PrimState prim) -> prim String
freezeShrink CountOf Word8
finalSize MutableString s
MutableString (PrimState (ST s))
dst
where
!sz :: CountOf Word8
sz = UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
arr
!start :: Offset Word8
start = UArray Word8 -> Offset Word8
forall ty. UArray ty -> Offset ty
C.offset UArray Word8
arr
reverse :: String -> String
reverse :: String -> String
reverse (String arr :: UArray Word8
arr) = (forall s. ST s String) -> String
forall a. (forall s. ST s a) -> a
runST ((forall s. ST s String) -> String)
-> (forall s. ST s String) -> String
forall a b. (a -> b) -> a -> b
$ do
MutableString s
s <- CountOf Word8
-> (MutableBlock Word8 (PrimState (ST s)) -> ST s ())
-> ST s (MutableString (PrimState (ST s)))
forall (prim :: * -> *).
PrimMonad prim =>
CountOf Word8
-> (MutableBlock Word8 (PrimState prim) -> prim ())
-> prim (MutableString (PrimState prim))
newNative_ (UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
arr) ((MutableBlock Word8 (PrimState (ST s)) -> ST s ())
-> ST s (MutableString s))
-> (MutableBlock Word8 (PrimState (ST s)) -> ST s ())
-> ST s (MutableString s)
forall a b. (a -> b) -> a -> b
$ \(MutableBlock mba :: MutableByteArray# (PrimState (ST s))
mba) ->
(Block Word8 -> ST s ())
-> (FinalPtr Word8 -> ST s ()) -> UArray Word8 -> ST s ()
forall (prim :: * -> *) ty a.
PrimMonad prim =>
(Block ty -> prim a)
-> (FinalPtr ty -> prim a) -> UArray ty -> prim a
C.onBackendPrim
(\ba :: Block Word8
ba@(Block !ByteArray#
_) -> MutableByteArray# (PrimState (ST s))
-> Offset Word8
-> Block Word8
-> Offset Word8
-> Offset Word8
-> ST s ()
forall (prim :: * -> *) container.
(PrimMonad prim, Indexable container Word8) =>
MutableByteArray# (PrimState prim)
-> Offset Word8
-> container
-> Offset Word8
-> Offset Word8
-> prim ()
UTF8.reverse MutableByteArray# (PrimState (ST s))
mba 0 Block Word8
ba Offset Word8
start Offset Word8
end)
(\fptr :: FinalPtr Word8
fptr -> FinalPtr Word8 -> (Ptr Word8 -> ST s ()) -> ST s ()
forall (prim :: * -> *) p a.
PrimMonad prim =>
FinalPtr p -> (Ptr p -> prim a) -> prim a
withFinalPtr FinalPtr Word8
fptr ((Ptr Word8 -> ST s ()) -> ST s ())
-> (Ptr Word8 -> ST s ()) -> ST s ()
forall a b. (a -> b) -> a -> b
$ \ptr :: Ptr Word8
ptr@(Ptr !Addr#
_) -> MutableByteArray# (PrimState (ST s))
-> Offset Word8
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> ST s ()
forall (prim :: * -> *) container.
(PrimMonad prim, Indexable container Word8) =>
MutableByteArray# (PrimState prim)
-> Offset Word8
-> container
-> Offset Word8
-> Offset Word8
-> prim ()
UTF8.reverse MutableByteArray# (PrimState (ST s))
mba 0 Ptr Word8
ptr Offset Word8
start Offset Word8
end)
UArray Word8
arr
MutableString (PrimState (ST s)) -> ST s String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> prim String
freeze MutableString s
MutableString (PrimState (ST s))
s
where
!(C.ValidRange start :: Offset Word8
start end :: Offset Word8
end) = UArray Word8 -> ValidRange Word8
forall ty. UArray ty -> ValidRange ty
C.offsetsValidRange UArray Word8
arr
indices :: String -> String -> [Offset8]
indices :: String -> String -> [Offset Word8]
indices (String ned :: UArray Word8
ned) (String hy :: UArray Word8
hy) = UArray Word8 -> UArray Word8 -> [Offset Word8]
forall ty. PrimType ty => UArray ty -> UArray ty -> [Offset ty]
Vec.indices UArray Word8
ned UArray Word8
hy
replace :: String -> String -> String -> String
replace :: String -> String -> String -> String
replace (String needle :: UArray Word8
needle) (String replacement :: UArray Word8
replacement) (String haystack :: UArray Word8
haystack) =
UArray Word8 -> String
String (UArray Word8 -> String) -> UArray Word8 -> String
forall a b. (a -> b) -> a -> b
$ UArray Word8 -> UArray Word8 -> UArray Word8 -> UArray Word8
forall ty.
PrimType ty =>
UArray ty -> UArray ty -> UArray ty -> UArray ty
Vec.replace UArray Word8
needle UArray Word8
replacement UArray Word8
haystack
index :: String -> Offset Char -> Maybe Char
index :: String -> Offset Char -> Maybe Char
index s :: String
s n :: Offset Char
n
| Offset Word8
ofs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
>= Offset Word8
end = Maybe Char
forall a. Maybe a
Nothing
| Bool
otherwise =
let (Step !Char
c _) = String -> Offset Word8 -> Step
next String
s Offset Word8
ofs
in Char -> Maybe Char
forall a. a -> Maybe a
Just Char
c
where
!nbBytes :: CountOf Word8
nbBytes = String -> CountOf Word8
size String
s
end :: Offset Word8
end = 0 Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
nbBytes
ofs :: Offset Word8
ofs = CountOf Char -> String -> Offset Word8
indexN (Offset Char -> CountOf Char
forall a. Offset a -> CountOf a
offsetAsSize Offset Char
n) String
s
findIndex :: (Char -> Bool) -> String -> Maybe (Offset Char)
findIndex :: (Char -> Bool) -> String -> Maybe (Offset Char)
findIndex predicate :: Char -> Bool
predicate s :: String
s = Offset Char -> Offset Word8 -> Maybe (Offset Char)
loop 0 0
where
!sz :: CountOf Word8
sz = String -> CountOf Word8
size String
s
loop :: Offset Char -> Offset Word8 -> Maybe (Offset Char)
loop ofs :: Offset Char
ofs idx :: Offset Word8
idx
| Offset Word8
idx Offset Word8 -> CountOf Word8 -> Bool
forall ty. Offset ty -> CountOf ty -> Bool
.==# CountOf Word8
sz = Maybe (Offset Char)
forall a. Maybe a
Nothing
| Bool
otherwise =
let !(Step c :: Char
c idx' :: Offset Word8
idx') = String -> Offset Word8 -> Step
next String
s Offset Word8
idx
in case Char -> Bool
predicate Char
c of
True -> Offset Char -> Maybe (Offset Char)
forall a. a -> Maybe a
Just Offset Char
ofs
False -> Offset Char -> Offset Word8 -> Maybe (Offset Char)
loop (Offset Char
ofsOffset Char -> Offset Char -> Offset Char
forall a. Additive a => a -> a -> a
+1) Offset Word8
idx'
data Encoding
= ASCII7
| UTF8
| UTF16
| UTF32
| ISO_8859_1
deriving (Typeable, Typeable Encoding
DataType
Constr
Typeable Encoding =>
(forall (c :: * -> *).
(forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> Encoding -> c Encoding)
-> (forall (c :: * -> *).
(forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c Encoding)
-> (Encoding -> Constr)
-> (Encoding -> DataType)
-> (forall (t :: * -> *) (c :: * -> *).
Typeable t =>
(forall d. Data d => c (t d)) -> Maybe (c Encoding))
-> (forall (t :: * -> * -> *) (c :: * -> *).
Typeable t =>
(forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c Encoding))
-> ((forall b. Data b => b -> b) -> Encoding -> Encoding)
-> (forall r r'.
(r -> r' -> r)
-> r -> (forall d. Data d => d -> r') -> Encoding -> r)
-> (forall r r'.
(r' -> r -> r)
-> r -> (forall d. Data d => d -> r') -> Encoding -> r)
-> (forall u. (forall d. Data d => d -> u) -> Encoding -> [u])
-> (forall u. Int -> (forall d. Data d => d -> u) -> Encoding -> u)
-> (forall (m :: * -> *).
Monad m =>
(forall d. Data d => d -> m d) -> Encoding -> m Encoding)
-> (forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d) -> Encoding -> m Encoding)
-> (forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d) -> Encoding -> m Encoding)
-> Data Encoding
Encoding -> DataType
Encoding -> Constr
(forall b. Data b => b -> b) -> Encoding -> Encoding
(forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> Encoding -> c Encoding
(forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c Encoding
forall a.
Typeable a =>
(forall (c :: * -> *).
(forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> a -> c a)
-> (forall (c :: * -> *).
(forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c a)
-> (a -> Constr)
-> (a -> DataType)
-> (forall (t :: * -> *) (c :: * -> *).
Typeable t =>
(forall d. Data d => c (t d)) -> Maybe (c a))
-> (forall (t :: * -> * -> *) (c :: * -> *).
Typeable t =>
(forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c a))
-> ((forall b. Data b => b -> b) -> a -> a)
-> (forall r r'.
(r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> a -> r)
-> (forall r r'.
(r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> a -> r)
-> (forall u. (forall d. Data d => d -> u) -> a -> [u])
-> (forall u. Int -> (forall d. Data d => d -> u) -> a -> u)
-> (forall (m :: * -> *).
Monad m =>
(forall d. Data d => d -> m d) -> a -> m a)
-> (forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d) -> a -> m a)
-> (forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d) -> a -> m a)
-> Data a
forall u. Int -> (forall d. Data d => d -> u) -> Encoding -> u
forall u. (forall d. Data d => d -> u) -> Encoding -> [u]
forall r r'.
(r -> r' -> r)
-> r -> (forall d. Data d => d -> r') -> Encoding -> r
forall r r'.
(r' -> r -> r)
-> r -> (forall d. Data d => d -> r') -> Encoding -> r
forall (m :: * -> *).
Monad m =>
(forall d. Data d => d -> m d) -> Encoding -> m Encoding
forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d) -> Encoding -> m Encoding
forall (c :: * -> *).
(forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c Encoding
forall (c :: * -> *).
(forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> Encoding -> c Encoding
forall (t :: * -> *) (c :: * -> *).
Typeable t =>
(forall d. Data d => c (t d)) -> Maybe (c Encoding)
forall (t :: * -> * -> *) (c :: * -> *).
Typeable t =>
(forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c Encoding)
$cISO_8859_1 :: Constr
$cUTF32 :: Constr
$cUTF16 :: Constr
$cUTF8 :: Constr
$cASCII7 :: Constr
$tEncoding :: DataType
gmapMo :: (forall d. Data d => d -> m d) -> Encoding -> m Encoding
$cgmapMo :: forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d) -> Encoding -> m Encoding
gmapMp :: (forall d. Data d => d -> m d) -> Encoding -> m Encoding
$cgmapMp :: forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d) -> Encoding -> m Encoding
gmapM :: (forall d. Data d => d -> m d) -> Encoding -> m Encoding
$cgmapM :: forall (m :: * -> *).
Monad m =>
(forall d. Data d => d -> m d) -> Encoding -> m Encoding
gmapQi :: Int -> (forall d. Data d => d -> u) -> Encoding -> u
$cgmapQi :: forall u. Int -> (forall d. Data d => d -> u) -> Encoding -> u
gmapQ :: (forall d. Data d => d -> u) -> Encoding -> [u]
$cgmapQ :: forall u. (forall d. Data d => d -> u) -> Encoding -> [u]
gmapQr :: (r' -> r -> r)
-> r -> (forall d. Data d => d -> r') -> Encoding -> r
$cgmapQr :: forall r r'.
(r' -> r -> r)
-> r -> (forall d. Data d => d -> r') -> Encoding -> r
gmapQl :: (r -> r' -> r)
-> r -> (forall d. Data d => d -> r') -> Encoding -> r
$cgmapQl :: forall r r'.
(r -> r' -> r)
-> r -> (forall d. Data d => d -> r') -> Encoding -> r
gmapT :: (forall b. Data b => b -> b) -> Encoding -> Encoding
$cgmapT :: (forall b. Data b => b -> b) -> Encoding -> Encoding
dataCast2 :: (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c Encoding)
$cdataCast2 :: forall (t :: * -> * -> *) (c :: * -> *).
Typeable t =>
(forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c Encoding)
dataCast1 :: (forall d. Data d => c (t d)) -> Maybe (c Encoding)
$cdataCast1 :: forall (t :: * -> *) (c :: * -> *).
Typeable t =>
(forall d. Data d => c (t d)) -> Maybe (c Encoding)
dataTypeOf :: Encoding -> DataType
$cdataTypeOf :: Encoding -> DataType
toConstr :: Encoding -> Constr
$ctoConstr :: Encoding -> Constr
gunfold :: (forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c Encoding
$cgunfold :: forall (c :: * -> *).
(forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c Encoding
gfoldl :: (forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> Encoding -> c Encoding
$cgfoldl :: forall (c :: * -> *).
(forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> Encoding -> c Encoding
$cp1Data :: Typeable Encoding
Data, Encoding -> Encoding -> Bool
(Encoding -> Encoding -> Bool)
-> (Encoding -> Encoding -> Bool) -> Eq Encoding
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: Encoding -> Encoding -> Bool
$c/= :: Encoding -> Encoding -> Bool
== :: Encoding -> Encoding -> Bool
$c== :: Encoding -> Encoding -> Bool
Eq, Eq Encoding
Eq Encoding =>
(Encoding -> Encoding -> Ordering)
-> (Encoding -> Encoding -> Bool)
-> (Encoding -> Encoding -> Bool)
-> (Encoding -> Encoding -> Bool)
-> (Encoding -> Encoding -> Bool)
-> (Encoding -> Encoding -> Encoding)
-> (Encoding -> Encoding -> Encoding)
-> Ord Encoding
Encoding -> Encoding -> Bool
Encoding -> Encoding -> Ordering
Encoding -> Encoding -> Encoding
forall a.
Eq a =>
(a -> a -> Ordering)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> a)
-> (a -> a -> a)
-> Ord a
min :: Encoding -> Encoding -> Encoding
$cmin :: Encoding -> Encoding -> Encoding
max :: Encoding -> Encoding -> Encoding
$cmax :: Encoding -> Encoding -> Encoding
>= :: Encoding -> Encoding -> Bool
$c>= :: Encoding -> Encoding -> Bool
> :: Encoding -> Encoding -> Bool
$c> :: Encoding -> Encoding -> Bool
<= :: Encoding -> Encoding -> Bool
$c<= :: Encoding -> Encoding -> Bool
< :: Encoding -> Encoding -> Bool
$c< :: Encoding -> Encoding -> Bool
compare :: Encoding -> Encoding -> Ordering
$ccompare :: Encoding -> Encoding -> Ordering
$cp1Ord :: Eq Encoding
Ord, Int -> Encoding -> [Char] -> [Char]
[Encoding] -> [Char] -> [Char]
Encoding -> [Char]
(Int -> Encoding -> [Char] -> [Char])
-> (Encoding -> [Char])
-> ([Encoding] -> [Char] -> [Char])
-> Show Encoding
forall a.
(Int -> a -> [Char] -> [Char])
-> (a -> [Char]) -> ([a] -> [Char] -> [Char]) -> Show a
showList :: [Encoding] -> [Char] -> [Char]
$cshowList :: [Encoding] -> [Char] -> [Char]
show :: Encoding -> [Char]
$cshow :: Encoding -> [Char]
showsPrec :: Int -> Encoding -> [Char] -> [Char]
$cshowsPrec :: Int -> Encoding -> [Char] -> [Char]
Show, Int -> Encoding
Encoding -> Int
Encoding -> [Encoding]
Encoding -> Encoding
Encoding -> Encoding -> [Encoding]
Encoding -> Encoding -> Encoding -> [Encoding]
(Encoding -> Encoding)
-> (Encoding -> Encoding)
-> (Int -> Encoding)
-> (Encoding -> Int)
-> (Encoding -> [Encoding])
-> (Encoding -> Encoding -> [Encoding])
-> (Encoding -> Encoding -> [Encoding])
-> (Encoding -> Encoding -> Encoding -> [Encoding])
-> Enum Encoding
forall a.
(a -> a)
-> (a -> a)
-> (Int -> a)
-> (a -> Int)
-> (a -> [a])
-> (a -> a -> [a])
-> (a -> a -> [a])
-> (a -> a -> a -> [a])
-> Enum a
enumFromThenTo :: Encoding -> Encoding -> Encoding -> [Encoding]
$cenumFromThenTo :: Encoding -> Encoding -> Encoding -> [Encoding]
enumFromTo :: Encoding -> Encoding -> [Encoding]
$cenumFromTo :: Encoding -> Encoding -> [Encoding]
enumFromThen :: Encoding -> Encoding -> [Encoding]
$cenumFromThen :: Encoding -> Encoding -> [Encoding]
enumFrom :: Encoding -> [Encoding]
$cenumFrom :: Encoding -> [Encoding]
fromEnum :: Encoding -> Int
$cfromEnum :: Encoding -> Int
toEnum :: Int -> Encoding
$ctoEnum :: Int -> Encoding
pred :: Encoding -> Encoding
$cpred :: Encoding -> Encoding
succ :: Encoding -> Encoding
$csucc :: Encoding -> Encoding
Enum, Encoding
Encoding -> Encoding -> Bounded Encoding
forall a. a -> a -> Bounded a
maxBound :: Encoding
$cmaxBound :: Encoding
minBound :: Encoding
$cminBound :: Encoding
Bounded)
fromEncoderBytes :: ( Encoder.Encoding encoding
, PrimType (Encoder.Unit encoding)
)
=> encoding
-> UArray Word8
-> (String, Maybe ValidationFailure, UArray Word8)
fromEncoderBytes :: encoding
-> UArray Word8 -> (String, Maybe ValidationFailure, UArray Word8)
fromEncoderBytes enc :: encoding
enc bytes :: UArray Word8
bytes =
case (forall s.
ST
s (Either (Offset (Unit encoding), Error encoding) (UArray Word8)))
-> Either (Offset (Unit encoding), Error encoding) (UArray Word8)
forall a. (forall s. ST s a) -> a
runST ((forall s.
ST
s (Either (Offset (Unit encoding), Error encoding) (UArray Word8)))
-> Either (Offset (Unit encoding), Error encoding) (UArray Word8))
-> (forall s.
ST
s (Either (Offset (Unit encoding), Error encoding) (UArray Word8)))
-> Either (Offset (Unit encoding), Error encoding) (UArray Word8)
forall a b. (a -> b) -> a -> b
$ encoding
-> EncoderUTF8
-> UArray (Unit encoding)
-> ST
s
(Either
(Offset (Unit encoding), Error encoding)
(UArray (Unit EncoderUTF8)))
forall (st :: * -> *) input output.
(PrimMonad st, Monad st, Encoding input, PrimType (Unit input),
Encoding output, PrimType (Unit output)) =>
input
-> output
-> UArray (Unit input)
-> st
(Either (Offset (Unit input), Error input) (UArray (Unit output)))
Encoder.convertFromTo encoding
enc EncoderUTF8
EncoderUTF8 (UArray Word8 -> UArray (Unit encoding)
forall a b. (PrimType a, PrimType b) => UArray a -> UArray b
Vec.recast UArray Word8
bytes) of
Left (off :: Offset (Unit encoding)
off, _) ->
let (b1 :: UArray (Unit encoding)
b1, b2 :: UArray (Unit encoding)
b2) = CountOf (Unit encoding)
-> UArray (Unit encoding)
-> (UArray (Unit encoding), UArray (Unit encoding))
forall ty. CountOf ty -> UArray ty -> (UArray ty, UArray ty)
Vec.splitAt (Offset (Unit encoding) -> CountOf (Unit encoding)
forall a. Offset a -> CountOf a
offsetAsSize Offset (Unit encoding)
off) (UArray Word8 -> UArray (Unit encoding)
forall a b. (PrimType a, PrimType b) => UArray a -> UArray b
Vec.recast UArray Word8
bytes)
in (UArray Word8 -> String
String (UArray Word8 -> String) -> UArray Word8 -> String
forall a b. (a -> b) -> a -> b
$ UArray (Unit encoding) -> UArray Word8
forall a b. (PrimType a, PrimType b) => UArray a -> UArray b
Vec.recast UArray (Unit encoding)
b1, ValidationFailure -> Maybe ValidationFailure
forall a. a -> Maybe a
Just ValidationFailure
BuildingFailure, UArray (Unit encoding) -> UArray Word8
forall a b. (PrimType a, PrimType b) => UArray a -> UArray b
Vec.recast UArray (Unit encoding)
b2)
Right converted :: UArray Word8
converted -> (UArray Word8 -> String
String UArray Word8
converted, Maybe ValidationFailure
forall a. Maybe a
Nothing, UArray Word8
forall a. Monoid a => a
mempty)
fromBytes :: Encoding -> UArray Word8 -> (String, Maybe ValidationFailure, UArray Word8)
fromBytes :: Encoding
-> UArray Word8 -> (String, Maybe ValidationFailure, UArray Word8)
fromBytes ASCII7 bytes :: UArray Word8
bytes = ASCII7
-> UArray Word8 -> (String, Maybe ValidationFailure, UArray Word8)
forall encoding.
(Encoding encoding, PrimType (Unit encoding)) =>
encoding
-> UArray Word8 -> (String, Maybe ValidationFailure, UArray Word8)
fromEncoderBytes ASCII7
Encoder.ASCII7 UArray Word8
bytes
fromBytes ISO_8859_1 bytes :: UArray Word8
bytes = ISO_8859_1
-> UArray Word8 -> (String, Maybe ValidationFailure, UArray Word8)
forall encoding.
(Encoding encoding, PrimType (Unit encoding)) =>
encoding
-> UArray Word8 -> (String, Maybe ValidationFailure, UArray Word8)
fromEncoderBytes ISO_8859_1
Encoder.ISO_8859_1 UArray Word8
bytes
fromBytes UTF16 bytes :: UArray Word8
bytes = UTF16
-> UArray Word8 -> (String, Maybe ValidationFailure, UArray Word8)
forall encoding.
(Encoding encoding, PrimType (Unit encoding)) =>
encoding
-> UArray Word8 -> (String, Maybe ValidationFailure, UArray Word8)
fromEncoderBytes UTF16
Encoder.UTF16 UArray Word8
bytes
fromBytes UTF32 bytes :: UArray Word8
bytes = UTF32
-> UArray Word8 -> (String, Maybe ValidationFailure, UArray Word8)
forall encoding.
(Encoding encoding, PrimType (Unit encoding)) =>
encoding
-> UArray Word8 -> (String, Maybe ValidationFailure, UArray Word8)
fromEncoderBytes UTF32
Encoder.UTF32 UArray Word8
bytes
fromBytes UTF8 bytes :: UArray Word8
bytes
| UArray Word8 -> Bool
forall ty. UArray ty -> Bool
C.null UArray Word8
bytes = (String
forall a. Monoid a => a
mempty, Maybe ValidationFailure
forall a. Maybe a
Nothing, UArray Word8
forall a. Monoid a => a
mempty)
| Bool
otherwise =
case UArray Word8
-> Offset Word8
-> CountOf Word8
-> (Offset Word8, Maybe ValidationFailure)
validate UArray Word8
bytes (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) (UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
bytes) of
(_, Nothing) -> (UArray Word8 -> String
fromBytesUnsafe UArray Word8
bytes, Maybe ValidationFailure
forall a. Maybe a
Nothing, UArray Word8
forall a. Monoid a => a
mempty)
(pos :: Offset Word8
pos, Just vf :: ValidationFailure
vf) ->
let (b1 :: UArray Word8
b1, b2 :: UArray Word8
b2) = CountOf Word8 -> UArray Word8 -> (UArray Word8, UArray Word8)
forall ty. CountOf ty -> UArray ty -> (UArray ty, UArray ty)
C.splitAt (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
pos) UArray Word8
bytes
in (UArray Word8 -> String
fromBytesUnsafe UArray Word8
b1, ValidationFailure -> Maybe ValidationFailure
toErr ValidationFailure
vf, UArray Word8
b2)
where
toErr :: ValidationFailure -> Maybe ValidationFailure
toErr MissingByte = Maybe ValidationFailure
forall a. Maybe a
Nothing
toErr InvalidHeader = ValidationFailure -> Maybe ValidationFailure
forall a. a -> Maybe a
Just ValidationFailure
InvalidHeader
toErr InvalidContinuation = ValidationFailure -> Maybe ValidationFailure
forall a. a -> Maybe a
Just ValidationFailure
InvalidContinuation
toErr BuildingFailure = ValidationFailure -> Maybe ValidationFailure
forall a. a -> Maybe a
Just ValidationFailure
BuildingFailure
fromBytesLenient :: UArray Word8 -> (String, UArray Word8)
fromBytesLenient :: UArray Word8 -> (String, UArray Word8)
fromBytesLenient bytes :: UArray Word8
bytes
| UArray Word8 -> Bool
forall ty. UArray ty -> Bool
C.null UArray Word8
bytes = (String
forall a. Monoid a => a
mempty, UArray Word8
forall a. Monoid a => a
mempty)
| Bool
otherwise =
case UArray Word8
-> Offset Word8
-> CountOf Word8
-> (Offset Word8, Maybe ValidationFailure)
validate UArray Word8
bytes (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) (UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
bytes) of
(_, Nothing) -> (UArray Word8 -> String
fromBytesUnsafe UArray Word8
bytes, UArray Word8
forall a. Monoid a => a
mempty)
(_, Just BuildingFailure) -> [Char] -> (String, UArray Word8)
forall a. HasCallStack => [Char] -> a
error "fromBytesLenient: FIXME!"
(pos :: Offset Word8
pos, Just MissingByte) ->
let (b1 :: UArray Word8
b1,b2 :: UArray Word8
b2) = CountOf Word8 -> UArray Word8 -> (UArray Word8, UArray Word8)
forall ty. CountOf ty -> UArray ty -> (UArray ty, UArray ty)
C.splitAt (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
pos) UArray Word8
bytes
in (UArray Word8 -> String
fromBytesUnsafe UArray Word8
b1, UArray Word8
b2)
(pos :: Offset Word8
pos, Just InvalidHeader) ->
let (b1 :: UArray Word8
b1,b2 :: UArray Word8
b2) = CountOf Word8 -> UArray Word8 -> (UArray Word8, UArray Word8)
forall ty. CountOf ty -> UArray ty -> (UArray ty, UArray ty)
C.splitAt (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
pos) UArray Word8
bytes
(_,b3 :: UArray Word8
b3) = CountOf Word8 -> UArray Word8 -> (UArray Word8, UArray Word8)
forall ty. CountOf ty -> UArray ty -> (UArray ty, UArray ty)
C.splitAt 1 UArray Word8
b2
(s3 :: String
s3, r :: UArray Word8
r) = UArray Word8 -> (String, UArray Word8)
fromBytesLenient UArray Word8
b3
in ([String] -> String
forall a. Monoid a => [a] -> a
mconcat [UArray Word8 -> String
fromBytesUnsafe UArray Word8
b1,String
replacement, String
s3], UArray Word8
r)
(pos :: Offset Word8
pos, Just InvalidContinuation) ->
let (b1 :: UArray Word8
b1,b2 :: UArray Word8
b2) = CountOf Word8 -> UArray Word8 -> (UArray Word8, UArray Word8)
forall ty. CountOf ty -> UArray ty -> (UArray ty, UArray ty)
C.splitAt (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
pos) UArray Word8
bytes
(_,b3 :: UArray Word8
b3) = CountOf Word8 -> UArray Word8 -> (UArray Word8, UArray Word8)
forall ty. CountOf ty -> UArray ty -> (UArray ty, UArray ty)
C.splitAt 1 UArray Word8
b2
(s3 :: String
s3, r :: UArray Word8
r) = UArray Word8 -> (String, UArray Word8)
fromBytesLenient UArray Word8
b3
in ([String] -> String
forall a. Monoid a => [a] -> a
mconcat [UArray Word8 -> String
fromBytesUnsafe UArray Word8
b1,String
replacement, String
s3], UArray Word8
r)
where
replacement :: String
!replacement :: String
replacement = UArray Word8 -> String
fromBytesUnsafe (UArray Word8 -> String) -> UArray Word8 -> String
forall a b. (a -> b) -> a -> b
$ [Item (UArray Word8)] -> UArray Word8
forall l. IsList l => [Item l] -> l
fromList [0xef,0xbf,0xbd]
fromChunkBytes :: [UArray Word8] -> [String]
fromChunkBytes :: [UArray Word8] -> [String]
fromChunkBytes l :: [UArray Word8]
l = [UArray Word8] -> [String]
loop [UArray Word8]
l
where
loop :: [UArray Word8] -> [String]
loop [] = []
loop [bytes :: UArray Word8
bytes] =
case UArray Word8
-> Offset Word8
-> CountOf Word8
-> (Offset Word8, Maybe ValidationFailure)
validate UArray Word8
bytes (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) (UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
bytes) of
(_, Nothing) -> [UArray Word8 -> String
fromBytesUnsafe UArray Word8
bytes]
(_, Just err :: ValidationFailure
err) -> ValidationFailure -> [String]
forall a a. Show a => a -> a
doErr ValidationFailure
err
loop (bytes :: UArray Word8
bytes:cs :: [UArray Word8]
cs@(c1 :: UArray Word8
c1:c2 :: [UArray Word8]
c2)) =
case UArray Word8
-> Offset Word8
-> CountOf Word8
-> (Offset Word8, Maybe ValidationFailure)
validate UArray Word8
bytes (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) (UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
bytes) of
(_, Nothing) -> UArray Word8 -> String
fromBytesUnsafe UArray Word8
bytes String -> [String] -> [String]
forall a. a -> [a] -> [a]
: [UArray Word8] -> [String]
loop [UArray Word8]
cs
(pos :: Offset Word8
pos, Just MissingByte) ->
let (b1 :: UArray Word8
b1,b2 :: UArray Word8
b2) = CountOf Word8 -> UArray Word8 -> (UArray Word8, UArray Word8)
forall ty. CountOf ty -> UArray ty -> (UArray ty, UArray ty)
C.splitAt (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
pos) UArray Word8
bytes
in UArray Word8 -> String
fromBytesUnsafe UArray Word8
b1 String -> [String] -> [String]
forall a. a -> [a] -> [a]
: [UArray Word8] -> [String]
loop ((UArray Word8
b2 UArray Word8 -> UArray Word8 -> UArray Word8
forall a. Monoid a => a -> a -> a
`mappend` UArray Word8
c1) UArray Word8 -> [UArray Word8] -> [UArray Word8]
forall a. a -> [a] -> [a]
: [UArray Word8]
c2)
(_, Just err :: ValidationFailure
err) -> ValidationFailure -> [String]
forall a a. Show a => a -> a
doErr ValidationFailure
err
doErr :: a -> a
doErr err :: a
err = [Char] -> a
forall a. HasCallStack => [Char] -> a
error ("fromChunkBytes: " [Char] -> [Char] -> [Char]
forall a. Semigroup a => a -> a -> a
<> a -> [Char]
forall a. Show a => a -> [Char]
show a
err)
fromBytesUnsafe :: UArray Word8 -> String
fromBytesUnsafe :: UArray Word8 -> String
fromBytesUnsafe = UArray Word8 -> String
String
toEncoderBytes :: ( Encoder.Encoding encoding
, PrimType (Encoder.Unit encoding)
, Exception (Encoder.Error encoding)
)
=> encoding
-> UArray Word8
-> UArray Word8
toEncoderBytes :: encoding -> UArray Word8 -> UArray Word8
toEncoderBytes enc :: encoding
enc bytes :: UArray Word8
bytes = UArray (Unit encoding) -> UArray Word8
forall a b. (PrimType a, PrimType b) => UArray a -> UArray b
Vec.recast (UArray (Unit encoding) -> UArray Word8)
-> UArray (Unit encoding) -> UArray Word8
forall a b. (a -> b) -> a -> b
$
case (forall s.
ST
s
(Either
(Offset Word8, ValidationFailure) (UArray (Unit encoding))))
-> Either
(Offset Word8, ValidationFailure) (UArray (Unit encoding))
forall a. (forall s. ST s a) -> a
runST ((forall s.
ST
s
(Either
(Offset Word8, ValidationFailure) (UArray (Unit encoding))))
-> Either
(Offset Word8, ValidationFailure) (UArray (Unit encoding)))
-> (forall s.
ST
s
(Either
(Offset Word8, ValidationFailure) (UArray (Unit encoding))))
-> Either
(Offset Word8, ValidationFailure) (UArray (Unit encoding))
forall a b. (a -> b) -> a -> b
$ EncoderUTF8
-> encoding
-> UArray (Unit EncoderUTF8)
-> ST
s
(Either
(Offset (Unit EncoderUTF8), Error EncoderUTF8)
(UArray (Unit encoding)))
forall (st :: * -> *) input output.
(PrimMonad st, Monad st, Encoding input, PrimType (Unit input),
Encoding output, PrimType (Unit output)) =>
input
-> output
-> UArray (Unit input)
-> st
(Either (Offset (Unit input), Error input) (UArray (Unit output)))
Encoder.convertFromTo EncoderUTF8
EncoderUTF8 encoding
enc UArray Word8
UArray (Unit EncoderUTF8)
bytes of
Left _ -> [Char] -> UArray (Unit encoding)
forall a. HasCallStack => [Char] -> a
error "toEncoderBytes: FIXME!"
Right converted :: UArray (Unit encoding)
converted -> UArray (Unit encoding)
converted
toBytes :: Encoding -> String -> UArray Word8
toBytes :: Encoding -> String -> UArray Word8
toBytes UTF8 (String bytes :: UArray Word8
bytes) = UArray Word8
bytes
toBytes ASCII7 (String bytes :: UArray Word8
bytes) = ASCII7 -> UArray Word8 -> UArray Word8
forall encoding.
(Encoding encoding, PrimType (Unit encoding),
Exception (Error encoding)) =>
encoding -> UArray Word8 -> UArray Word8
toEncoderBytes ASCII7
Encoder.ASCII7 UArray Word8
bytes
toBytes ISO_8859_1 (String bytes :: UArray Word8
bytes) = ISO_8859_1 -> UArray Word8 -> UArray Word8
forall encoding.
(Encoding encoding, PrimType (Unit encoding),
Exception (Error encoding)) =>
encoding -> UArray Word8 -> UArray Word8
toEncoderBytes ISO_8859_1
Encoder.ISO_8859_1 UArray Word8
bytes
toBytes UTF16 (String bytes :: UArray Word8
bytes) = UTF16 -> UArray Word8 -> UArray Word8
forall encoding.
(Encoding encoding, PrimType (Unit encoding),
Exception (Error encoding)) =>
encoding -> UArray Word8 -> UArray Word8
toEncoderBytes UTF16
Encoder.UTF16 UArray Word8
bytes
toBytes UTF32 (String bytes :: UArray Word8
bytes) = UTF32 -> UArray Word8 -> UArray Word8
forall encoding.
(Encoding encoding, PrimType (Unit encoding),
Exception (Error encoding)) =>
encoding -> UArray Word8 -> UArray Word8
toEncoderBytes UTF32
Encoder.UTF32 UArray Word8
bytes
lines :: String -> [String]
lines :: String -> [String]
lines s :: String
s =
case String -> Either Bool (String, String)
breakLine String
s of
Left _ -> [String
s]
Right (line :: String
line,r :: String
r) -> String
line String -> [String] -> [String]
forall a. a -> [a] -> [a]
: String -> [String]
lines String
r
words :: String -> [String]
words :: String -> [String]
words = ([Char] -> String) -> [[Char]] -> [String]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [Char] -> String
forall l. IsList l => [Item l] -> l
fromList ([[Char]] -> [String])
-> (String -> [[Char]]) -> String -> [String]
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. [Char] -> [[Char]]
Prelude.words ([Char] -> [[Char]]) -> (String -> [Char]) -> String -> [[Char]]
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. String -> [Char]
forall l. IsList l => l -> [Item l]
toList
builderAppend :: PrimMonad state => Char -> Builder String MutableString Word8 state err ()
builderAppend :: Char -> Builder String MutableString Word8 state err ()
builderAppend c :: Char
c = State
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err)
state
()
-> Builder String MutableString Word8 state err ()
forall collection (mutCollection :: * -> *) step (state :: * -> *)
err a.
State
(Offset step,
BuildingState collection mutCollection step (PrimState state),
Maybe err)
state
a
-> Builder collection mutCollection step state err a
Builder (State
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err)
state
()
-> Builder String MutableString Word8 state err ())
-> State
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err)
state
()
-> Builder String MutableString Word8 state err ()
forall a b. (a -> b) -> a -> b
$ ((Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err)
-> state
((),
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err)))
-> State
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err)
state
()
forall s (m :: * -> *) a. (s -> m (a, s)) -> State s m a
State (((Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err)
-> state
((),
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err)))
-> State
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err)
state
())
-> ((Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err)
-> state
((),
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err)))
-> State
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err)
state
()
forall a b. (a -> b) -> a -> b
$ \(i :: Offset Word8
i, st :: BuildingState String MutableString Word8 (PrimState state)
st, e :: Maybe err
e) ->
if Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
i CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ CountOf Word8
nbBytes CountOf Word8 -> CountOf Word8 -> Bool
forall a. Ord a => a -> a -> Bool
>= BuildingState String MutableString Word8 (PrimState state)
-> CountOf Word8
forall collection (mutCollection :: * -> *) step state.
BuildingState collection mutCollection step state -> CountOf step
chunkSize BuildingState String MutableString Word8 (PrimState state)
st
then do
String
cur <- MutableString (PrimState state) -> CountOf Word8 -> state String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> CountOf Word8 -> prim String
unsafeFreezeShrink (BuildingState String MutableString Word8 (PrimState state)
-> MutableString (PrimState state)
forall collection (mutCollection :: * -> *) step state.
BuildingState collection mutCollection step state
-> mutCollection state
curChunk BuildingState String MutableString Word8 (PrimState state)
st) (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
i)
MutableString (PrimState state)
newChunk <- CountOf Word8 -> state (MutableString (PrimState state))
forall (prim :: * -> *).
PrimMonad prim =>
CountOf Word8 -> prim (MutableString (PrimState prim))
new (BuildingState String MutableString Word8 (PrimState state)
-> CountOf Word8
forall collection (mutCollection :: * -> *) step state.
BuildingState collection mutCollection step state -> CountOf step
chunkSize BuildingState String MutableString Word8 (PrimState state)
st)
MutableString (PrimState state)
-> Offset Word8 -> UTF8Char -> state ()
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim)
-> Offset Word8 -> UTF8Char -> prim ()
writeUTF8Char MutableString (PrimState state)
newChunk (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) UTF8Char
utf8Char
((),
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err))
-> state
((),
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err))
forall (m :: * -> *) a. Monad m => a -> m a
return ((), (CountOf Word8 -> Offset Word8
forall a. CountOf a -> Offset a
sizeAsOffset CountOf Word8
nbBytes, BuildingState String MutableString Word8 (PrimState state)
st { prevChunks :: [String]
prevChunks = String
cur String -> [String] -> [String]
forall a. a -> [a] -> [a]
: BuildingState String MutableString Word8 (PrimState state)
-> [String]
forall collection (mutCollection :: * -> *) step state.
BuildingState collection mutCollection step state -> [collection]
prevChunks BuildingState String MutableString Word8 (PrimState state)
st
, prevChunksSize :: CountOf Word8
prevChunksSize = Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
i CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ BuildingState String MutableString Word8 (PrimState state)
-> CountOf Word8
forall collection (mutCollection :: * -> *) step state.
BuildingState collection mutCollection step state -> CountOf step
prevChunksSize BuildingState String MutableString Word8 (PrimState state)
st
, curChunk :: MutableString (PrimState state)
curChunk = MutableString (PrimState state)
newChunk
}, Maybe err
e))
else do
MutableString (PrimState state)
-> Offset Word8 -> UTF8Char -> state ()
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim)
-> Offset Word8 -> UTF8Char -> prim ()
writeUTF8Char (BuildingState String MutableString Word8 (PrimState state)
-> MutableString (PrimState state)
forall collection (mutCollection :: * -> *) step state.
BuildingState collection mutCollection step state
-> mutCollection state
curChunk BuildingState String MutableString Word8 (PrimState state)
st) Offset Word8
i UTF8Char
utf8Char
((),
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err))
-> state
((),
(Offset Word8,
BuildingState String MutableString Word8 (PrimState state),
Maybe err))
forall (m :: * -> *) a. Monad m => a -> m a
return ((), (Offset Word8
i Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ CountOf Word8 -> Offset Word8
forall a. CountOf a -> Offset a
sizeAsOffset CountOf Word8
nbBytes, BuildingState String MutableString Word8 (PrimState state)
st, Maybe err
e))
where
utf8Char :: UTF8Char
utf8Char = Char -> UTF8Char
asUTF8Char Char
c
nbBytes :: CountOf Word8
nbBytes = UTF8Char -> CountOf Word8
numBytes UTF8Char
utf8Char
builderBuild :: PrimMonad m => Int -> Builder String MutableString Word8 m err () -> m (Either err String)
builderBuild :: Int
-> Builder String MutableString Word8 m err ()
-> m (Either err String)
builderBuild sizeChunksI :: Int
sizeChunksI sb :: Builder String MutableString Word8 m err ()
sb
| Int
sizeChunksI Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
<= 3 = Int
-> Builder String MutableString Word8 m err ()
-> m (Either err String)
forall (m :: * -> *) err.
PrimMonad m =>
Int
-> Builder String MutableString Word8 m err ()
-> m (Either err String)
builderBuild 64 Builder String MutableString Word8 m err ()
sb
| Bool
otherwise = do
MutableString (PrimState m)
firstChunk <- CountOf Word8 -> m (MutableString (PrimState m))
forall (prim :: * -> *).
PrimMonad prim =>
CountOf Word8 -> prim (MutableString (PrimState prim))
new CountOf Word8
sizeChunks
(i :: Offset Word8
i, st :: BuildingState String MutableString Word8 (PrimState m)
st, e :: Maybe err
e) <- ((),
(Offset Word8,
BuildingState String MutableString Word8 (PrimState m), Maybe err))
-> (Offset Word8,
BuildingState String MutableString Word8 (PrimState m), Maybe err)
forall a b. (a, b) -> b
snd (((),
(Offset Word8,
BuildingState String MutableString Word8 (PrimState m), Maybe err))
-> (Offset Word8,
BuildingState String MutableString Word8 (PrimState m), Maybe err))
-> m ((),
(Offset Word8,
BuildingState String MutableString Word8 (PrimState m), Maybe err))
-> m (Offset Word8,
BuildingState String MutableString Word8 (PrimState m), Maybe err)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> State
(Offset Word8,
BuildingState String MutableString Word8 (PrimState m), Maybe err)
m
()
-> (Offset Word8,
BuildingState String MutableString Word8 (PrimState m), Maybe err)
-> m ((),
(Offset Word8,
BuildingState String MutableString Word8 (PrimState m), Maybe err))
forall s (m :: * -> *) a. State s m a -> s -> m (a, s)
runState (Builder String MutableString Word8 m err ()
-> State
(Offset Word8,
BuildingState String MutableString Word8 (PrimState m), Maybe err)
m
()
forall collection (mutCollection :: * -> *) step (state :: * -> *)
err a.
Builder collection mutCollection step state err a
-> State
(Offset step,
BuildingState collection mutCollection step (PrimState state),
Maybe err)
state
a
runBuilder Builder String MutableString Word8 m err ()
sb) (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0, [String]
-> CountOf Word8
-> MutableString (PrimState m)
-> CountOf Word8
-> BuildingState String MutableString Word8 (PrimState m)
forall collection (mutCollection :: * -> *) step state.
[collection]
-> CountOf step
-> mutCollection state
-> CountOf step
-> BuildingState collection mutCollection step state
BuildingState [] (Int -> CountOf Word8
forall ty. Int -> CountOf ty
CountOf 0) MutableString (PrimState m)
firstChunk CountOf Word8
sizeChunks, Maybe err
forall a. Maybe a
Nothing)
case Maybe err
e of
Just err :: err
err -> Either err String -> m (Either err String)
forall (m :: * -> *) a. Monad m => a -> m a
return (err -> Either err String
forall a b. a -> Either a b
Left err
err)
Nothing -> do
String
cur <- MutableString (PrimState m) -> CountOf Word8 -> m String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> CountOf Word8 -> prim String
unsafeFreezeShrink (BuildingState String MutableString Word8 (PrimState m)
-> MutableString (PrimState m)
forall collection (mutCollection :: * -> *) step state.
BuildingState collection mutCollection step state
-> mutCollection state
curChunk BuildingState String MutableString Word8 (PrimState m)
st) (Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
i)
let totalSize :: CountOf Word8
totalSize = BuildingState String MutableString Word8 (PrimState m)
-> CountOf Word8
forall collection (mutCollection :: * -> *) step state.
BuildingState collection mutCollection step state -> CountOf step
prevChunksSize BuildingState String MutableString Word8 (PrimState m)
st CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ Offset Word8 -> CountOf Word8
forall a. Offset a -> CountOf a
offsetAsSize Offset Word8
i
UArray Word8
final <- CountOf Word8 -> m (MUArray Word8 (PrimState m))
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
CountOf ty -> prim (MUArray ty (PrimState prim))
Vec.new CountOf Word8
totalSize m (MUArray Word8 (PrimState m))
-> (MUArray Word8 (PrimState m) -> m (MUArray Word8 (PrimState m)))
-> m (MUArray Word8 (PrimState m))
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= CountOf Word8
-> [String]
-> MUArray Word8 (PrimState m)
-> m (MUArray Word8 (PrimState m))
forall (m :: * -> *).
PrimMonad m =>
CountOf Word8
-> [String]
-> MUArray Word8 (PrimState m)
-> m (MUArray Word8 (PrimState m))
fillFromEnd CountOf Word8
totalSize (String
cur String -> [String] -> [String]
forall a. a -> [a] -> [a]
: BuildingState String MutableString Word8 (PrimState m) -> [String]
forall collection (mutCollection :: * -> *) step state.
BuildingState collection mutCollection step state -> [collection]
prevChunks BuildingState String MutableString Word8 (PrimState m)
st) m (MUArray Word8 (PrimState m))
-> (MUArray Word8 (PrimState m) -> m (UArray Word8))
-> m (UArray Word8)
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= MUArray Word8 (PrimState m) -> m (UArray Word8)
forall (prim :: * -> *) ty.
PrimMonad prim =>
MUArray ty (PrimState prim) -> prim (UArray ty)
Vec.unsafeFreeze
Either err String -> m (Either err String)
forall (m :: * -> *) a. Monad m => a -> m a
return (Either err String -> m (Either err String))
-> (UArray Word8 -> Either err String)
-> UArray Word8
-> m (Either err String)
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. String -> Either err String
forall a b. b -> Either a b
Right (String -> Either err String)
-> (UArray Word8 -> String) -> UArray Word8 -> Either err String
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. UArray Word8 -> String
String (UArray Word8 -> m (Either err String))
-> UArray Word8 -> m (Either err String)
forall a b. (a -> b) -> a -> b
$ UArray Word8
final
where
sizeChunks :: CountOf Word8
sizeChunks = Int -> CountOf Word8
forall ty. Int -> CountOf ty
CountOf Int
sizeChunksI
fillFromEnd :: CountOf Word8
-> [String]
-> MUArray Word8 (PrimState m)
-> m (MUArray Word8 (PrimState m))
fillFromEnd _ [] mba :: MUArray Word8 (PrimState m)
mba = MUArray Word8 (PrimState m) -> m (MUArray Word8 (PrimState m))
forall (m :: * -> *) a. Monad m => a -> m a
return MUArray Word8 (PrimState m)
mba
fillFromEnd !CountOf Word8
end (String x :: UArray Word8
x:xs :: [String]
xs) mba :: MUArray Word8 (PrimState m)
mba = do
let sz :: CountOf Word8
sz = UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
Vec.length UArray Word8
x
let start :: CountOf Word8
start = CountOf Word8
end CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. CountOf a -> CountOf a -> CountOf a
`sizeSub` CountOf Word8
sz
MUArray Word8 (PrimState m)
-> Offset Word8
-> UArray Word8
-> Offset Word8
-> CountOf Word8
-> m ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MUArray ty (PrimState prim)
-> Offset ty -> UArray ty -> Offset ty -> CountOf ty -> prim ()
Vec.unsafeCopyAtRO MUArray Word8 (PrimState m)
mba (CountOf Word8 -> Offset Word8
forall a. CountOf a -> Offset a
sizeAsOffset CountOf Word8
start) UArray Word8
x (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) CountOf Word8
sz
CountOf Word8
-> [String]
-> MUArray Word8 (PrimState m)
-> m (MUArray Word8 (PrimState m))
fillFromEnd CountOf Word8
start [String]
xs MUArray Word8 (PrimState m)
mba
builderBuild_ :: PrimMonad m => Int -> Builder String MutableString Word8 m () () -> m String
builderBuild_ :: Int -> Builder String MutableString Word8 m () () -> m String
builderBuild_ sizeChunksI :: Int
sizeChunksI sb :: Builder String MutableString Word8 m () ()
sb = (() -> String) -> (String -> String) -> Either () String -> String
forall a c b. (a -> c) -> (b -> c) -> Either a b -> c
either (\() -> [Char] -> String
forall a. [Char] -> a
internalError "impossible output") String -> String
forall k (cat :: k -> k -> *) (a :: k). Category cat => cat a a
id (Either () String -> String) -> m (Either () String) -> m String
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Int
-> Builder String MutableString Word8 m () ()
-> m (Either () String)
forall (m :: * -> *) err.
PrimMonad m =>
Int
-> Builder String MutableString Word8 m err ()
-> m (Either err String)
builderBuild Int
sizeChunksI Builder String MutableString Word8 m () ()
sb
stringDewrap :: (Block Word8 -> Offset Word8 -> a)
-> (Ptr Word8 -> Offset Word8 -> ST s a)
-> String
-> a
stringDewrap :: (Block Word8 -> Offset Word8 -> a)
-> (Ptr Word8 -> Offset Word8 -> ST s a) -> String -> a
stringDewrap withBa :: Block Word8 -> Offset Word8 -> a
withBa withPtr :: Ptr Word8 -> Offset Word8 -> ST s a
withPtr (String ba :: UArray Word8
ba) = (Block Word8 -> Offset Word8 -> a)
-> (Ptr Word8 -> Offset Word8 -> ST s a) -> UArray Word8 -> a
forall ty a s.
(Block ty -> Offset ty -> a)
-> (Ptr ty -> Offset ty -> ST s a) -> UArray ty -> a
C.unsafeDewrap Block Word8 -> Offset Word8 -> a
withBa Ptr Word8 -> Offset Word8 -> ST s a
withPtr UArray Word8
ba
{-# INLINE stringDewrap #-}
readIntegral :: (HasNegation i, IntegralUpsize Word8 i, Additive i, Multiplicative i, IsIntegral i) => String -> Maybe i
readIntegral :: String -> Maybe i
readIntegral str :: String
str
| CountOf Word8
sz CountOf Word8 -> CountOf Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== 0 = Maybe i
forall a. Maybe a
Nothing
| Bool
otherwise = (Block Word8 -> Offset Word8 -> Maybe i)
-> (Ptr Word8 -> Offset Word8 -> ST Any (Maybe i))
-> String
-> Maybe i
forall a s.
(Block Word8 -> Offset Word8 -> a)
-> (Ptr Word8 -> Offset Word8 -> ST s a) -> String -> a
stringDewrap Block Word8 -> Offset Word8 -> Maybe i
withBa (\ptr :: Ptr Word8
ptr@(Ptr !Addr#
_) -> Maybe i -> ST Any (Maybe i)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Maybe i -> ST Any (Maybe i))
-> (Offset Word8 -> Maybe i) -> Offset Word8 -> ST Any (Maybe i)
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. Ptr Word8 -> Offset Word8 -> Maybe i
withPtr Ptr Word8
ptr) String
str
where
!sz :: CountOf Word8
sz = String -> CountOf Word8
size String
str
withBa :: Block Word8 -> Offset Word8 -> Maybe i
withBa ba :: Block Word8
ba ofs :: Offset Word8
ofs =
let negativeSign :: Bool
negativeSign = Block Word8 -> Offset Word8 -> Word8 -> Bool
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> Word8 -> Bool
UTF8.expectAscii Block Word8
ba Offset Word8
ofs 0x2d
startOfs :: Offset Word8
startOfs = if Bool
negativeSign then Offset Word8 -> Offset Word8
forall a. Enum a => a -> a
succ Offset Word8
ofs else Offset Word8
ofs
in case i
-> Block Word8
-> Offset Word8
-> Offset Word8
-> (# i, Bool, Offset Word8 #)
forall acc.
(IntegralUpsize Word8 acc, Additive acc, Multiplicative acc,
Integral acc) =>
acc
-> Block Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsBA 0 Block Word8
ba Offset Word8
endOfs Offset Word8
startOfs of
(# acc :: i
acc, True, endOfs' :: Offset Word8
endOfs' #) | Offset Word8
endOfs' Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
startOfs -> i -> Maybe i
forall a. a -> Maybe a
Just (i -> Maybe i) -> i -> Maybe i
forall a b. (a -> b) -> a -> b
$! if Bool
negativeSign then i -> i
forall a. HasNegation a => a -> a
negate i
acc else i
acc
_ -> Maybe i
forall a. Maybe a
Nothing
where !endOfs :: Offset Word8
endOfs = Offset Word8
ofs Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
withPtr :: Ptr Word8 -> Offset Word8 -> Maybe i
withPtr addr :: Ptr Word8
addr ofs :: Offset Word8
ofs =
let negativeSign :: Bool
negativeSign = Ptr Word8 -> Offset Word8 -> Word8 -> Bool
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> Word8 -> Bool
UTF8.expectAscii Ptr Word8
addr Offset Word8
ofs 0x2d
startOfs :: Offset Word8
startOfs = if Bool
negativeSign then Offset Word8 -> Offset Word8
forall a. Enum a => a -> a
succ Offset Word8
ofs else Offset Word8
ofs
in case i
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> (# i, Bool, Offset Word8 #)
forall acc.
(IntegralUpsize Word8 acc, Additive acc, Multiplicative acc,
Integral acc) =>
acc
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsPtr 0 Ptr Word8
addr Offset Word8
endOfs Offset Word8
startOfs of
(# acc :: i
acc, True, endOfs' :: Offset Word8
endOfs' #) | Offset Word8
endOfs' Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
startOfs -> i -> Maybe i
forall a. a -> Maybe a
Just (i -> Maybe i) -> i -> Maybe i
forall a b. (a -> b) -> a -> b
$! if Bool
negativeSign then i -> i
forall a. HasNegation a => a -> a
negate i
acc else i
acc
_ -> Maybe i
forall a. Maybe a
Nothing
where !endOfs :: Offset Word8
endOfs = Offset Word8
ofs Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
{-# SPECIALISE readIntegral :: String -> Maybe Integer #-}
{-# SPECIALISE readIntegral :: String -> Maybe Int #-}
readInteger :: String -> Maybe Integer
readInteger :: String -> Maybe Integer
readInteger = String -> Maybe Integer
forall i.
(HasNegation i, IntegralUpsize Word8 i, Additive i,
Multiplicative i, IsIntegral i) =>
String -> Maybe i
readIntegral
readNatural :: String -> Maybe Natural
readNatural :: String -> Maybe Natural
readNatural str :: String
str
| CountOf Word8
sz CountOf Word8 -> CountOf Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== 0 = Maybe Natural
forall a. Maybe a
Nothing
| Bool
otherwise = (Block Word8 -> Offset Word8 -> Maybe Natural)
-> (Ptr Word8 -> Offset Word8 -> ST Any (Maybe Natural))
-> String
-> Maybe Natural
forall a s.
(Block Word8 -> Offset Word8 -> a)
-> (Ptr Word8 -> Offset Word8 -> ST s a) -> String -> a
stringDewrap Block Word8 -> Offset Word8 -> Maybe Natural
withBa (\ptr :: Ptr Word8
ptr@(Ptr !Addr#
_) -> Maybe Natural -> ST Any (Maybe Natural)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Maybe Natural -> ST Any (Maybe Natural))
-> (Offset Word8 -> Maybe Natural)
-> Offset Word8
-> ST Any (Maybe Natural)
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. Ptr Word8 -> Offset Word8 -> Maybe Natural
withPtr Ptr Word8
ptr) String
str
where
!sz :: CountOf Word8
sz = String -> CountOf Word8
size String
str
withBa :: Block Word8 -> Offset Word8 -> Maybe Natural
withBa ba :: Block Word8
ba stringStart :: Offset Word8
stringStart =
case Natural
-> Block Word8
-> Offset Word8
-> Offset Word8
-> (# Natural, Bool, Offset Word8 #)
forall acc.
(IntegralUpsize Word8 acc, Additive acc, Multiplicative acc,
Integral acc) =>
acc
-> Block Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsBA 0 Block Word8
ba Offset Word8
eofs Offset Word8
stringStart of
(# acc :: Natural
acc, True, endOfs :: Offset Word8
endOfs #) | Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
stringStart -> Natural -> Maybe Natural
forall a. a -> Maybe a
Just Natural
acc
_ -> Maybe Natural
forall a. Maybe a
Nothing
where eofs :: Offset Word8
eofs = Offset Word8
stringStart Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
withPtr :: Ptr Word8 -> Offset Word8 -> Maybe Natural
withPtr addr :: Ptr Word8
addr stringStart :: Offset Word8
stringStart =
case Natural
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> (# Natural, Bool, Offset Word8 #)
forall acc.
(IntegralUpsize Word8 acc, Additive acc, Multiplicative acc,
Integral acc) =>
acc
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsPtr 0 Ptr Word8
addr Offset Word8
eofs Offset Word8
stringStart of
(# acc :: Natural
acc, True, endOfs :: Offset Word8
endOfs #) | Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
stringStart -> Natural -> Maybe Natural
forall a. a -> Maybe a
Just Natural
acc
_ -> Maybe Natural
forall a. Maybe a
Nothing
where eofs :: Offset Word8
eofs = Offset Word8
stringStart Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
readDouble :: String -> Maybe Double
readDouble :: String -> Maybe Double
readDouble s :: String
s =
String -> ReadFloatingCallback Double -> Maybe Double
forall a. String -> ReadFloatingCallback a -> Maybe a
readFloatingExact String
s (ReadFloatingCallback Double -> Maybe Double)
-> ReadFloatingCallback Double -> Maybe Double
forall a b. (a -> b) -> a -> b
$ \isNegative :: Bool
isNegative integral :: Natural
integral floatingDigits :: Word
floatingDigits mExponant :: Maybe Int
mExponant ->
Double -> Maybe Double
forall a. a -> Maybe a
Just (Double -> Maybe Double) -> Double -> Maybe Double
forall a b. (a -> b) -> a -> b
$ Bool -> Double -> Double
forall a. HasNegation a => Bool -> a -> a
applySign Bool
isNegative (Double -> Double) -> Double -> Double
forall a b. (a -> b) -> a -> b
$ case (Word
floatingDigits, Maybe Int
mExponant) of
(0, Nothing) -> Natural -> Double
naturalToDouble Natural
integral
(0, Just exponent :: Int
exponent) -> Int -> Double -> Double
withExponant Int
exponent (Double -> Double) -> Double -> Double
forall a b. (a -> b) -> a -> b
$ Natural -> Double
naturalToDouble Natural
integral
(floating :: Word
floating, Nothing) -> Word -> Double -> Double
forall a b.
(Divisible a, Integral b, Integral a, Num a) =>
b -> a -> a
applyFloating Word
floating (Double -> Double) -> Double -> Double
forall a b. (a -> b) -> a -> b
$ Natural -> Double
naturalToDouble Natural
integral
(floating :: Word
floating, Just exponent :: Int
exponent) -> Int -> Double -> Double
withExponant Int
exponent (Double -> Double) -> Double -> Double
forall a b. (a -> b) -> a -> b
$ Word -> Double -> Double
forall a b.
(Divisible a, Integral b, Integral a, Num a) =>
b -> a -> a
applyFloating Word
floating (Double -> Double) -> Double -> Double
forall a b. (a -> b) -> a -> b
$ Natural -> Double
naturalToDouble Natural
integral
where
applySign :: Bool -> a -> a
applySign True = a -> a
forall a. HasNegation a => a -> a
negate
applySign False = a -> a
forall k (cat :: k -> k -> *) (a :: k). Category cat => cat a a
id
withExponant :: Int -> Double -> Double
withExponant e :: Int
e v :: Double
v = Double
v Double -> Double -> Double
forall a. Multiplicative a => a -> a -> a
* Double -> Int -> Double
doubleExponant 10 Int
e
applyFloating :: b -> a -> a
applyFloating digits :: b
digits n :: a
n = a
n a -> a -> a
forall a. Divisible a => a -> a -> a
/ (10 a -> b -> a
forall a b. (Num a, Integral b) => a -> b -> a
Prelude.^ b
digits)
readRational :: String -> Maybe Prelude.Rational
readRational :: String -> Maybe Rational
readRational s :: String
s =
String -> ReadFloatingCallback Rational -> Maybe Rational
forall a. String -> ReadFloatingCallback a -> Maybe a
readFloatingExact String
s (ReadFloatingCallback Rational -> Maybe Rational)
-> ReadFloatingCallback Rational -> Maybe Rational
forall a b. (a -> b) -> a -> b
$ \isNegative :: Bool
isNegative integral :: Natural
integral floatingDigits :: Word
floatingDigits mExponant :: Maybe Int
mExponant ->
case Maybe Int
mExponant of
Just exponent :: Int
exponent
| Int
exponent Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
< -10000 Bool -> Bool -> Bool
|| Int
exponent Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
> 10000 -> Maybe Rational
forall a. Maybe a
Nothing
| Bool
otherwise -> Rational -> Maybe Rational
forall a. a -> Maybe a
Just (Rational -> Maybe Rational) -> Rational -> Maybe Rational
forall a b. (a -> b) -> a -> b
$ Bool -> Natural -> Integer
forall b a. (HasNegation b, IntegralUpsize a b) => Bool -> a -> b
modF Bool
isNegative Natural
integral Integer -> Integer -> Rational
forall a. Integral a => a -> a -> Ratio a
% (10 Integer -> Int -> Integer
forall a b. (Num a, Integral b) => a -> b -> a
Prelude.^ (Word -> Int
forall source destination.
Cast source destination =>
source -> destination
cast Word
floatingDigits Int -> Int -> Difference Int
forall a. Subtractive a => a -> a -> Difference a
- Int
exponent))
Nothing -> Rational -> Maybe Rational
forall a. a -> Maybe a
Just (Rational -> Maybe Rational) -> Rational -> Maybe Rational
forall a b. (a -> b) -> a -> b
$ Bool -> Natural -> Integer
forall b a. (HasNegation b, IntegralUpsize a b) => Bool -> a -> b
modF Bool
isNegative Natural
integral Integer -> Integer -> Rational
forall a. Integral a => a -> a -> Ratio a
% (10 Integer -> Word -> Integer
forall a b. (Num a, Integral b) => a -> b -> a
Prelude.^ Word
floatingDigits)
where
modF :: Bool -> a -> b
modF True = b -> b
forall a. HasNegation a => a -> a
negate (b -> b) -> (a -> b) -> a -> b
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. a -> b
forall a b. IntegralUpsize a b => a -> b
integralUpsize
modF False = a -> b
forall a b. IntegralUpsize a b => a -> b
integralUpsize
type ReadFloatingCallback a = Bool
-> Natural
-> Word
-> Maybe Int
-> Maybe a
readFloatingExact :: String -> ReadFloatingCallback a -> Maybe a
readFloatingExact :: String -> ReadFloatingCallback a -> Maybe a
readFloatingExact str :: String
str f :: ReadFloatingCallback a
f
| CountOf Word8
sz CountOf Word8 -> CountOf Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== 0 = Maybe a
forall a. Maybe a
Nothing
| Bool
otherwise = (Block Word8 -> Offset Word8 -> Maybe a)
-> (Ptr Word8 -> Offset Word8 -> ST Any (Maybe a))
-> String
-> Maybe a
forall a s.
(Block Word8 -> Offset Word8 -> a)
-> (Ptr Word8 -> Offset Word8 -> ST s a) -> String -> a
stringDewrap Block Word8 -> Offset Word8 -> Maybe a
withBa Ptr Word8 -> Offset Word8 -> ST Any (Maybe a)
withPtr String
str
where
!sz :: CountOf Word8
sz = String -> CountOf Word8
size String
str
withBa :: Block Word8 -> Offset Word8 -> Maybe a
withBa ba :: Block Word8
ba stringStart :: Offset Word8
stringStart =
let !isNegative :: Bool
isNegative = Block Word8 -> Offset Word8 -> Word8 -> Bool
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> Word8 -> Bool
UTF8.expectAscii Block Word8
ba Offset Word8
stringStart 0x2d
in Bool -> Offset Word8 -> Maybe a
consumeIntegral Bool
isNegative (if Bool
isNegative then Offset Word8
stringStartOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+1 else Offset Word8
stringStart)
where
eofs :: Offset Word8
eofs = Offset Word8
stringStart Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
consumeIntegral :: Bool -> Offset Word8 -> Maybe a
consumeIntegral !Bool
isNegative startOfs :: Offset Word8
startOfs =
case Natural
-> Block Word8
-> Offset Word8
-> Offset Word8
-> (# Natural, Bool, Offset Word8 #)
forall acc.
(IntegralUpsize Word8 acc, Additive acc, Multiplicative acc,
Integral acc) =>
acc
-> Block Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsBA 0 Block Word8
ba Offset Word8
eofs Offset Word8
startOfs of
(# acc :: Natural
acc, True , endOfs :: Offset Word8
endOfs #) | Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
startOfs -> ReadFloatingCallback a
f Bool
isNegative Natural
acc 0 Maybe Int
forall a. Maybe a
Nothing
(# acc :: Natural
acc, False, endOfs :: Offset Word8
endOfs #) | Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
startOfs ->
if Block Word8 -> Offset Word8 -> Word8 -> Bool
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> Word8 -> Bool
UTF8.expectAscii Block Word8
ba Offset Word8
endOfs 0x2e
then Bool -> Natural -> Offset Word8 -> Maybe a
consumeFloat Bool
isNegative Natural
acc (Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1)
else Bool -> Natural -> Word -> Offset Word8 -> Maybe a
consumeExponant Bool
isNegative Natural
acc 0 Offset Word8
endOfs
_ -> Maybe a
forall a. Maybe a
Nothing
consumeFloat :: Bool -> Natural -> Offset Word8 -> Maybe a
consumeFloat isNegative :: Bool
isNegative integral :: Natural
integral startOfs :: Offset Word8
startOfs =
case Natural
-> Block Word8
-> Offset Word8
-> Offset Word8
-> (# Natural, Bool, Offset Word8 #)
forall acc.
(IntegralUpsize Word8 acc, Additive acc, Multiplicative acc,
Integral acc) =>
acc
-> Block Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsBA Natural
integral Block Word8
ba Offset Word8
eofs Offset Word8
startOfs of
(# acc :: Natural
acc, True, endOfs :: Offset Word8
endOfs #) | Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
startOfs -> let (CountOf !Int
diff) = Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Difference (Offset Word8)
forall a. Subtractive a => a -> a -> Difference a
- Offset Word8
startOfs
in ReadFloatingCallback a
f Bool
isNegative Natural
acc (Int -> Word
forall source destination.
Cast source destination =>
source -> destination
cast Int
diff) Maybe Int
forall a. Maybe a
Nothing
(# acc :: Natural
acc, False, endOfs :: Offset Word8
endOfs #) | Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
startOfs -> let (CountOf !Int
diff) = Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Difference (Offset Word8)
forall a. Subtractive a => a -> a -> Difference a
- Offset Word8
startOfs
in Bool -> Natural -> Word -> Offset Word8 -> Maybe a
consumeExponant Bool
isNegative Natural
acc (Int -> Word
forall source destination.
Cast source destination =>
source -> destination
cast Int
diff) Offset Word8
endOfs
_ -> Maybe a
forall a. Maybe a
Nothing
consumeExponant :: Bool -> Natural -> Word -> Offset Word8 -> Maybe a
consumeExponant !Bool
isNegative !Natural
integral !Word
floatingDigits !Offset Word8
startOfs
| Offset Word8
startOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
eofs = ReadFloatingCallback a
f Bool
isNegative Natural
integral Word
floatingDigits Maybe Int
forall a. Maybe a
Nothing
| Bool
otherwise =
case Block Word8 -> Offset Word8 -> StepASCII
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> StepASCII
UTF8.nextAscii Block Word8
ba Offset Word8
startOfs of
StepASCII 0x45 -> Offset Word8 -> Maybe a
consumeExponantSign (Offset Word8
startOfsOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+1)
StepASCII 0x65 -> Offset Word8 -> Maybe a
consumeExponantSign (Offset Word8
startOfsOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+1)
_ -> Maybe a
forall a. Maybe a
Nothing
where
consumeExponantSign :: Offset Word8 -> Maybe a
consumeExponantSign ofs :: Offset Word8
ofs
| Offset Word8
ofs Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
eofs = Maybe a
forall a. Maybe a
Nothing
| Bool
otherwise = let exponentNegative :: Bool
exponentNegative = Block Word8 -> Offset Word8 -> Word8 -> Bool
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> Word8 -> Bool
UTF8.expectAscii Block Word8
ba Offset Word8
ofs 0x2d
in Bool -> Offset Word8 -> Maybe a
consumeExponantNumber Bool
exponentNegative (if Bool
exponentNegative then Offset Word8
ofs Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1 else Offset Word8
ofs)
consumeExponantNumber :: Bool -> Offset Word8 -> Maybe a
consumeExponantNumber exponentNegative :: Bool
exponentNegative ofs :: Offset Word8
ofs =
case Int
-> Block Word8
-> Offset Word8
-> Offset Word8
-> (# Int, Bool, Offset Word8 #)
forall acc.
(IntegralUpsize Word8 acc, Additive acc, Multiplicative acc,
Integral acc) =>
acc
-> Block Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsBA 0 Block Word8
ba Offset Word8
eofs Offset Word8
ofs of
(# acc :: Int
acc, True, endOfs :: Offset Word8
endOfs #) | Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
ofs -> ReadFloatingCallback a
f Bool
isNegative Natural
integral Word
floatingDigits (Int -> Maybe Int
forall a. a -> Maybe a
Just (Int -> Maybe Int) -> Int -> Maybe Int
forall a b. (a -> b) -> a -> b
$! if Bool
exponentNegative then Int -> Int
forall a. HasNegation a => a -> a
negate Int
acc else Int
acc)
_ -> Maybe a
forall a. Maybe a
Nothing
withPtr :: Ptr Word8 -> Offset Word8 -> ST Any (Maybe a)
withPtr ptr :: Ptr Word8
ptr@(Ptr !Addr#
_) stringStart :: Offset Word8
stringStart = Maybe a -> ST Any (Maybe a)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Maybe a -> ST Any (Maybe a)) -> Maybe a -> ST Any (Maybe a)
forall a b. (a -> b) -> a -> b
$
let !isNegative :: Bool
isNegative = Ptr Word8 -> Offset Word8 -> Word8 -> Bool
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> Word8 -> Bool
UTF8.expectAscii Ptr Word8
ptr Offset Word8
stringStart 0x2d
in Bool -> Offset Word8 -> Maybe a
consumeIntegral Bool
isNegative (if Bool
isNegative then Offset Word8
stringStartOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+1 else Offset Word8
stringStart)
where
eofs :: Offset Word8
eofs = Offset Word8
stringStart Offset Word8 -> CountOf Word8 -> Offset Word8
forall ty. Offset ty -> CountOf ty -> Offset ty
`offsetPlusE` CountOf Word8
sz
consumeIntegral :: Bool -> Offset Word8 -> Maybe a
consumeIntegral !Bool
isNegative startOfs :: Offset Word8
startOfs =
case Natural
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> (# Natural, Bool, Offset Word8 #)
forall acc.
(IntegralUpsize Word8 acc, Additive acc, Multiplicative acc,
Integral acc) =>
acc
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsPtr 0 Ptr Word8
ptr Offset Word8
eofs Offset Word8
startOfs of
(# acc :: Natural
acc, True , endOfs :: Offset Word8
endOfs #) | Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
startOfs -> ReadFloatingCallback a
f Bool
isNegative Natural
acc 0 Maybe Int
forall a. Maybe a
Nothing
(# acc :: Natural
acc, False, endOfs :: Offset Word8
endOfs #) | Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
startOfs ->
if Ptr Word8 -> Offset Word8 -> Word8 -> Bool
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> Word8 -> Bool
UTF8.expectAscii Ptr Word8
ptr Offset Word8
endOfs 0x2e
then Bool -> Natural -> Offset Word8 -> Maybe a
consumeFloat Bool
isNegative Natural
acc (Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1)
else Bool -> Natural -> Word -> Offset Word8 -> Maybe a
consumeExponant Bool
isNegative Natural
acc 0 Offset Word8
endOfs
_ -> Maybe a
forall a. Maybe a
Nothing
consumeFloat :: Bool -> Natural -> Offset Word8 -> Maybe a
consumeFloat isNegative :: Bool
isNegative integral :: Natural
integral startOfs :: Offset Word8
startOfs =
case Natural
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> (# Natural, Bool, Offset Word8 #)
forall acc.
(IntegralUpsize Word8 acc, Additive acc, Multiplicative acc,
Integral acc) =>
acc
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsPtr Natural
integral Ptr Word8
ptr Offset Word8
eofs Offset Word8
startOfs of
(# acc :: Natural
acc, True, endOfs :: Offset Word8
endOfs #) | Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
startOfs -> let (CountOf !Int
diff) = Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Difference (Offset Word8)
forall a. Subtractive a => a -> a -> Difference a
- Offset Word8
startOfs
in ReadFloatingCallback a
f Bool
isNegative Natural
acc (Int -> Word
forall source destination.
Cast source destination =>
source -> destination
cast Int
diff) Maybe Int
forall a. Maybe a
Nothing
(# acc :: Natural
acc, False, endOfs :: Offset Word8
endOfs #) | Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
startOfs -> let (CountOf !Int
diff) = Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Difference (Offset Word8)
forall a. Subtractive a => a -> a -> Difference a
- Offset Word8
startOfs
in Bool -> Natural -> Word -> Offset Word8 -> Maybe a
consumeExponant Bool
isNegative Natural
acc (Int -> Word
forall source destination.
Cast source destination =>
source -> destination
cast Int
diff) Offset Word8
endOfs
_ -> Maybe a
forall a. Maybe a
Nothing
consumeExponant :: Bool -> Natural -> Word -> Offset Word8 -> Maybe a
consumeExponant !Bool
isNegative !Natural
integral !Word
floatingDigits !Offset Word8
startOfs
| Offset Word8
startOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
eofs = ReadFloatingCallback a
f Bool
isNegative Natural
integral Word
floatingDigits Maybe Int
forall a. Maybe a
Nothing
| Bool
otherwise =
case Ptr Word8 -> Offset Word8 -> StepASCII
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> StepASCII
UTF8.nextAscii Ptr Word8
ptr Offset Word8
startOfs of
StepASCII 0x45 -> Offset Word8 -> Maybe a
consumeExponantSign (Offset Word8
startOfsOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+1)
StepASCII 0x65 -> Offset Word8 -> Maybe a
consumeExponantSign (Offset Word8
startOfsOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+1)
_ -> Maybe a
forall a. Maybe a
Nothing
where
consumeExponantSign :: Offset Word8 -> Maybe a
consumeExponantSign ofs :: Offset Word8
ofs
| Offset Word8
ofs Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
eofs = Maybe a
forall a. Maybe a
Nothing
| Bool
otherwise = let exponentNegative :: Bool
exponentNegative = Ptr Word8 -> Offset Word8 -> Word8 -> Bool
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> Word8 -> Bool
UTF8.expectAscii Ptr Word8
ptr Offset Word8
ofs 0x2d
in Bool -> Offset Word8 -> Maybe a
consumeExponantNumber Bool
exponentNegative (if Bool
exponentNegative then Offset Word8
ofs Offset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+ 1 else Offset Word8
ofs)
consumeExponantNumber :: Bool -> Offset Word8 -> Maybe a
consumeExponantNumber exponentNegative :: Bool
exponentNegative ofs :: Offset Word8
ofs =
case Int
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> (# Int, Bool, Offset Word8 #)
forall acc.
(IntegralUpsize Word8 acc, Additive acc, Multiplicative acc,
Integral acc) =>
acc
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsPtr 0 Ptr Word8
ptr Offset Word8
eofs Offset Word8
ofs of
(# acc :: Int
acc, True, endOfs :: Offset Word8
endOfs #) | Offset Word8
endOfs Offset Word8 -> Offset Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> Offset Word8
ofs -> ReadFloatingCallback a
f Bool
isNegative Natural
integral Word
floatingDigits (Int -> Maybe Int
forall a. a -> Maybe a
Just (Int -> Maybe Int) -> Int -> Maybe Int
forall a b. (a -> b) -> a -> b
$! if Bool
exponentNegative then Int -> Int
forall a. HasNegation a => a -> a
negate Int
acc else Int
acc)
_ -> Maybe a
forall a. Maybe a
Nothing
decimalDigitsBA :: (IntegralUpsize Word8 acc, Additive acc, Multiplicative acc, Integral acc)
=> acc
-> Block Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsBA :: acc
-> Block Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsBA startAcc :: acc
startAcc ba :: Block Word8
ba !Offset Word8
endOfs !Offset Word8
startOfs = acc -> Offset Word8 -> (# acc, Bool, Offset Word8 #)
loop acc
startAcc Offset Word8
startOfs
where
loop :: acc -> Offset Word8 -> (# acc, Bool, Offset Word8 #)
loop !acc
acc !Offset Word8
ofs
| Offset Word8
ofs Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
endOfs = (# acc
acc, Bool
True, Offset Word8
ofs #)
| Bool
otherwise =
case Block Word8 -> Offset Word8 -> StepDigit
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> StepDigit
UTF8.nextAsciiDigit Block Word8
ba Offset Word8
ofs of
sg :: StepDigit
sg@(StepDigit d :: Word8
d) | StepDigit -> Bool
isValidStepDigit StepDigit
sg -> acc -> Offset Word8 -> (# acc, Bool, Offset Word8 #)
loop (10 acc -> acc -> acc
forall a. Multiplicative a => a -> a -> a
* acc
acc acc -> acc -> acc
forall a. Additive a => a -> a -> a
+ Word8 -> acc
forall a b. IntegralUpsize a b => a -> b
integralUpsize Word8
d) (Offset Word8 -> Offset Word8
forall a. Enum a => a -> a
succ Offset Word8
ofs)
| Bool
otherwise -> (# acc
acc, Bool
False, Offset Word8
ofs #)
{-# SPECIALIZE decimalDigitsBA :: Integer -> Block Word8 -> Offset Word8 -> Offset Word8 -> (# Integer, Bool, Offset Word8 #) #-}
{-# SPECIALIZE decimalDigitsBA :: Natural -> Block Word8 -> Offset Word8 -> Offset Word8 -> (# Natural, Bool, Offset Word8 #) #-}
{-# SPECIALIZE decimalDigitsBA :: Int -> Block Word8 -> Offset Word8 -> Offset Word8 -> (# Int, Bool, Offset Word8 #) #-}
{-# SPECIALIZE decimalDigitsBA :: Word -> Block Word8 -> Offset Word8 -> Offset Word8 -> (# Word, Bool, Offset Word8 #) #-}
decimalDigitsPtr :: (IntegralUpsize Word8 acc, Additive acc, Multiplicative acc, Integral acc)
=> acc
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsPtr :: acc
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> (# acc, Bool, Offset Word8 #)
decimalDigitsPtr startAcc :: acc
startAcc ptr :: Ptr Word8
ptr !Offset Word8
endOfs !Offset Word8
startOfs = acc -> Offset Word8 -> (# acc, Bool, Offset Word8 #)
loop acc
startAcc Offset Word8
startOfs
where
loop :: acc -> Offset Word8 -> (# acc, Bool, Offset Word8 #)
loop !acc
acc !Offset Word8
ofs
| Offset Word8
ofs Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
endOfs = (# acc
acc, Bool
True, Offset Word8
ofs #)
| Bool
otherwise =
case Ptr Word8 -> Offset Word8 -> StepDigit
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> StepDigit
UTF8.nextAsciiDigit Ptr Word8
ptr Offset Word8
ofs of
sg :: StepDigit
sg@(StepDigit d :: Word8
d) | StepDigit -> Bool
isValidStepDigit StepDigit
sg -> acc -> Offset Word8 -> (# acc, Bool, Offset Word8 #)
loop (10 acc -> acc -> acc
forall a. Multiplicative a => a -> a -> a
* acc
acc acc -> acc -> acc
forall a. Additive a => a -> a -> a
+ Word8 -> acc
forall a b. IntegralUpsize a b => a -> b
integralUpsize Word8
d) (Offset Word8 -> Offset Word8
forall a. Enum a => a -> a
succ Offset Word8
ofs)
| Bool
otherwise -> (# acc
acc, Bool
False, Offset Word8
ofs #)
{-# SPECIALIZE decimalDigitsPtr :: Integer -> Ptr Word8 -> Offset Word8 -> Offset Word8 -> (# Integer, Bool, Offset Word8 #) #-}
{-# SPECIALIZE decimalDigitsPtr :: Natural -> Ptr Word8 -> Offset Word8 -> Offset Word8 -> (# Natural, Bool, Offset Word8 #) #-}
{-# SPECIALIZE decimalDigitsPtr :: Int -> Ptr Word8 -> Offset Word8 -> Offset Word8 -> (# Int, Bool, Offset Word8 #) #-}
{-# SPECIALIZE decimalDigitsPtr :: Word -> Ptr Word8 -> Offset Word8 -> Offset Word8 -> (# Word, Bool, Offset Word8 #) #-}
caseConvert :: (Char7 -> Char7) -> (Char -> CM) -> String -> String
caseConvert :: (Char7 -> Char7) -> (Char -> CM) -> String -> String
caseConvert opASCII :: Char7 -> Char7
opASCII op :: Char -> CM
op s :: String
s@(String arr :: UArray Word8
arr) = (forall s. ST s String) -> String
forall a. (forall s. ST s a) -> a
runST ((forall s. ST s String) -> String)
-> (forall s. ST s String) -> String
forall a b. (a -> b) -> a -> b
$ do
MutableBlock Word8 s
mba <- CountOf Word8 -> ST s (MutableBlock Word8 (PrimState (ST s)))
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
CountOf ty -> prim (MutableBlock ty (PrimState prim))
MBLK.new CountOf Word8
iLen
CountOf Word8
nL <- (Block Word8 -> ST s (CountOf Word8))
-> (FinalPtr Word8 -> ST s (CountOf Word8))
-> UArray Word8
-> ST s (CountOf Word8)
forall (prim :: * -> *) ty a.
PrimMonad prim =>
(Block ty -> prim a)
-> (FinalPtr ty -> prim a) -> UArray ty -> prim a
C.onBackendPrim
(\blk :: Block Word8
blk -> MutableBlock Word8 (PrimState (ST s))
-> Block Word8
-> Offset Word8
-> Offset Word8
-> ST s (CountOf Word8)
forall container (prim :: * -> *).
(Indexable container Word8, PrimMonad prim) =>
MutableBlock Word8 (PrimState prim)
-> container
-> Offset Word8
-> Offset Word8
-> prim (CountOf Word8)
go MutableBlock Word8 s
MutableBlock Word8 (PrimState (ST s))
mba Block Word8
blk (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) Offset Word8
start)
(\fptr :: FinalPtr Word8
fptr -> FinalPtr Word8
-> (Ptr Word8 -> ST s (CountOf Word8)) -> ST s (CountOf Word8)
forall (prim :: * -> *) p a.
PrimMonad prim =>
FinalPtr p -> (Ptr p -> prim a) -> prim a
withFinalPtr FinalPtr Word8
fptr ((Ptr Word8 -> ST s (CountOf Word8)) -> ST s (CountOf Word8))
-> (Ptr Word8 -> ST s (CountOf Word8)) -> ST s (CountOf Word8)
forall a b. (a -> b) -> a -> b
$ \ptr :: Ptr Word8
ptr -> MutableBlock Word8 (PrimState (ST s))
-> Ptr Word8
-> Offset Word8
-> Offset Word8
-> ST s (CountOf Word8)
forall container (prim :: * -> *).
(Indexable container Word8, PrimMonad prim) =>
MutableBlock Word8 (PrimState prim)
-> container
-> Offset Word8
-> Offset Word8
-> prim (CountOf Word8)
go MutableBlock Word8 s
MutableBlock Word8 (PrimState (ST s))
mba Ptr Word8
ptr (Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 0) Offset Word8
start)
UArray Word8
arr
MutableString s -> ST s String
forall (prim :: * -> *).
PrimMonad prim =>
MutableString (PrimState prim) -> prim String
freeze (MutableString s -> ST s String)
-> (MUArray Word8 s -> MutableString s)
-> MUArray Word8 s
-> ST s String
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. MUArray Word8 s -> MutableString s
forall st. MUArray Word8 st -> MutableString st
MutableString (MUArray Word8 s -> ST s String) -> MUArray Word8 s -> ST s String
forall a b. (a -> b) -> a -> b
$ Offset Word8
-> CountOf Word8 -> MUArrayBackend Word8 s -> MUArray Word8 s
forall ty st.
Offset ty -> CountOf ty -> MUArrayBackend ty st -> MUArray ty st
MVec.MUArray 0 CountOf Word8
nL (MutableBlock Word8 s -> MUArrayBackend Word8 s
forall ty st. MutableBlock ty st -> MUArrayBackend ty st
C.MUArrayMBA MutableBlock Word8 s
mba)
where
!(C.ValidRange start :: Offset Word8
start end :: Offset Word8
end) = UArray Word8 -> ValidRange Word8
forall ty. UArray ty -> ValidRange ty
C.offsetsValidRange UArray Word8
arr
!iLen :: CountOf Word8
iLen = 1 CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
arr
go :: (Indexable container Word8, PrimMonad prim)
=> MutableBlock Word8 (PrimState prim)
-> container
-> Offset Word8
-> Offset Word8
-> prim (CountOf Word8)
go :: MutableBlock Word8 (PrimState prim)
-> container
-> Offset Word8
-> Offset Word8
-> prim (CountOf Word8)
go !MutableBlock Word8 (PrimState prim)
dst !container
src = MutableBlock Word8 (PrimState prim)
-> CountOf Word8
-> CountOf Word8
-> Offset Word8
-> Offset Word8
-> prim (CountOf Word8)
loop MutableBlock Word8 (PrimState prim)
dst CountOf Word8
iLen 0
where
eSize :: Char -> CountOf Word8
eSize !Char
e = if Char
e Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
== '\0' then 0 else Int -> CountOf Word8
charToBytes (Char -> Int
forall a. Enum a => a -> Int
fromEnum Char
e)
loop :: MutableBlock Word8 (PrimState prim)
-> CountOf Word8
-> CountOf Word8
-> Offset Word8
-> Offset Word8
-> prim (CountOf Word8)
loop !MutableBlock Word8 (PrimState prim)
dst !CountOf Word8
allocLen !CountOf Word8
nLen !Offset Word8
dstIdx !Offset Word8
srcIdx
| Offset Word8
srcIdx Offset Word8 -> Offset Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== Offset Word8
end = CountOf Word8 -> prim (CountOf Word8)
forall (m :: * -> *) a. Monad m => a -> m a
return CountOf Word8
nLen
| CountOf Word8
nLen CountOf Word8 -> CountOf Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== CountOf Word8
allocLen = prim (CountOf Word8)
realloc
| StepASCII -> Bool
headerIsAscii StepASCII
h = do
MutableBlock Word8 (PrimState prim)
-> Offset Word8 -> Char7 -> prim ()
forall (prim :: * -> *) container.
(PrimMonad prim, RandomAccess container prim Word8) =>
container -> Offset Word8 -> Char7 -> prim ()
UTF8.writeASCII MutableBlock Word8 (PrimState prim)
dst Offset Word8
dstIdx (Char7 -> Char7
opASCII (Char7 -> Char7) -> Char7 -> Char7
forall a b. (a -> b) -> a -> b
$ Word8 -> Char7
Char7 (Word8 -> Char7) -> Word8 -> Char7
forall a b. (a -> b) -> a -> b
$ StepASCII -> Word8
stepAsciiRawValue StepASCII
h)
MutableBlock Word8 (PrimState prim)
-> CountOf Word8
-> CountOf Word8
-> Offset Word8
-> Offset Word8
-> prim (CountOf Word8)
loop MutableBlock Word8 (PrimState prim)
dst CountOf Word8
allocLen (CountOf Word8
nLen CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ 1) (Offset Word8
dstIdxOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 1) (Offset Word8
srcIdxOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 1)
| Bool
otherwise = do
let !(CM c1 :: Char
c1 c2 :: Char
c2 c3 :: Char
c3) = Char -> CM
op Char
c
!(Step c :: Char
c nextSrcIdx :: Offset Word8
nextSrcIdx) = StepASCII -> container -> Offset Word8 -> Step
forall container.
Indexable container Word8 =>
StepASCII -> container -> Offset Word8 -> Step
UTF8.nextWith StepASCII
h container
src (Offset Word8
srcIdxOffset Word8 -> Offset Word8 -> Offset Word8
forall a. Additive a => a -> a -> a
+Int -> Offset Word8
forall ty. Int -> Offset ty
Offset 1)
Offset Word8
nextDstIdx <- MutableBlock Word8 (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
forall (prim :: * -> *) container.
(PrimMonad prim, RandomAccess container prim Word8) =>
container -> Offset Word8 -> Char -> prim (Offset Word8)
UTF8.writeUTF8 MutableBlock Word8 (PrimState prim)
dst Offset Word8
dstIdx Char
c1
if Char
c2 Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
== '\0'
then MutableBlock Word8 (PrimState prim)
-> CountOf Word8
-> CountOf Word8
-> Offset Word8
-> Offset Word8
-> prim (CountOf Word8)
loop MutableBlock Word8 (PrimState prim)
dst CountOf Word8
allocLen (CountOf Word8
nLen CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ Int -> CountOf Word8
charToBytes (Char -> Int
forall a. Enum a => a -> Int
fromEnum Char
c1)) Offset Word8
nextDstIdx Offset Word8
nextSrcIdx
else do
let !cSize :: CountOf Word8
cSize = Char -> CountOf Word8
eSize Char
c1 CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ Char -> CountOf Word8
eSize Char
c2 CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ Char -> CountOf Word8
eSize Char
c3
Offset Word8
nextDstIdx <- MutableBlock Word8 (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
forall (prim :: * -> *) container.
(PrimMonad prim, RandomAccess container prim Word8) =>
container -> Offset Word8 -> Char -> prim (Offset Word8)
UTF8.writeUTF8 MutableBlock Word8 (PrimState prim)
dst Offset Word8
nextDstIdx Char
c2
Offset Word8
nextDstIdx <- if Char
c3 Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
== '\0' then Offset Word8 -> prim (Offset Word8)
forall (m :: * -> *) a. Monad m => a -> m a
return Offset Word8
nextDstIdx else MutableBlock Word8 (PrimState prim)
-> Offset Word8 -> Char -> prim (Offset Word8)
forall (prim :: * -> *) container.
(PrimMonad prim, RandomAccess container prim Word8) =>
container -> Offset Word8 -> Char -> prim (Offset Word8)
UTF8.writeUTF8 MutableBlock Word8 (PrimState prim)
dst Offset Word8
nextDstIdx Char
c3
MutableBlock Word8 (PrimState prim)
-> CountOf Word8
-> CountOf Word8
-> Offset Word8
-> Offset Word8
-> prim (CountOf Word8)
loop MutableBlock Word8 (PrimState prim)
dst CountOf Word8
allocLen (CountOf Word8
nLen CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ CountOf Word8
cSize) Offset Word8
nextDstIdx Offset Word8
nextSrcIdx
where
{-# NOINLINE realloc #-}
realloc :: prim (CountOf Word8)
realloc = do
let nAll :: CountOf Word8
nAll = CountOf Word8
allocLen CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ CountOf Word8
allocLen CountOf Word8 -> CountOf Word8 -> CountOf Word8
forall a. Additive a => a -> a -> a
+ 1
MutableBlock Word8 (PrimState prim)
nDst <- CountOf Word8 -> prim (MutableBlock Word8 (PrimState prim))
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
CountOf ty -> prim (MutableBlock ty (PrimState prim))
MBLK.new CountOf Word8
nAll
MutableBlock Word8 (PrimState prim)
-> Offset Word8
-> MutableBlock Word8 (PrimState prim)
-> Offset Word8
-> CountOf Word8
-> prim ()
forall (prim :: * -> *) ty.
(PrimMonad prim, PrimType ty) =>
MutableBlock ty (PrimState prim)
-> Offset ty
-> MutableBlock ty (PrimState prim)
-> Offset ty
-> CountOf ty
-> prim ()
MBLK.unsafeCopyElements MutableBlock Word8 (PrimState prim)
nDst 0 MutableBlock Word8 (PrimState prim)
dst 0 CountOf Word8
nLen
MutableBlock Word8 (PrimState prim)
-> CountOf Word8
-> CountOf Word8
-> Offset Word8
-> Offset Word8
-> prim (CountOf Word8)
loop MutableBlock Word8 (PrimState prim)
nDst CountOf Word8
nAll CountOf Word8
nLen Offset Word8
dstIdx Offset Word8
srcIdx
h :: StepASCII
h = container -> Offset Word8 -> StepASCII
forall container.
Indexable container Word8 =>
container -> Offset Word8 -> StepASCII
UTF8.nextAscii container
src Offset Word8
srcIdx
upper :: String -> String
upper :: String -> String
upper = (Char7 -> Char7) -> (Char -> CM) -> String -> String
caseConvert Char7 -> Char7
c7Upper Char -> CM
upperMapping
lower :: String -> String
lower :: String -> String
lower = (Char7 -> Char7) -> (Char -> CM) -> String -> String
caseConvert Char7 -> Char7
c7Lower Char -> CM
lowerMapping
caseFold :: String -> String
caseFold :: String -> String
caseFold = (Char7 -> Char7) -> (Char -> CM) -> String -> String
caseConvert Char7 -> Char7
c7Upper Char -> CM
foldMapping
isPrefixOf :: String -> String -> Bool
isPrefixOf :: String -> String -> Bool
isPrefixOf (String needle :: UArray Word8
needle) (String haystack :: UArray Word8
haystack) = UArray Word8 -> UArray Word8 -> Bool
forall ty. PrimType ty => UArray ty -> UArray ty -> Bool
C.isPrefixOf UArray Word8
needle UArray Word8
haystack
isSuffixOf :: String -> String -> Bool
isSuffixOf :: String -> String -> Bool
isSuffixOf (String needle :: UArray Word8
needle) (String haystack :: UArray Word8
haystack)
| CountOf Word8
needleLen CountOf Word8 -> CountOf Word8 -> Bool
forall a. Ord a => a -> a -> Bool
> CountOf Word8
hayLen = Bool
False
| Bool
otherwise = UArray Word8
needle UArray Word8 -> UArray Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== CountOf Word8 -> UArray Word8 -> UArray Word8
forall ty. CountOf ty -> UArray ty -> UArray ty
C.revTake CountOf Word8
needleLen UArray Word8
haystack
where
needleLen :: CountOf Word8
needleLen = UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
needle
hayLen :: CountOf Word8
hayLen = UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
haystack
isInfixOf :: String -> String -> Bool
isInfixOf :: String -> String -> Bool
isInfixOf (String needle :: UArray Word8
needle) (String haystack :: UArray Word8
haystack)
= Maybe (CountOf Word8) -> UArray Word8 -> Bool
loop (CountOf Word8
hayLen CountOf Word8 -> CountOf Word8 -> Difference (CountOf Word8)
forall a. Subtractive a => a -> a -> Difference a
- CountOf Word8
needleLen) UArray Word8
haystack
where
needleLen :: CountOf Word8
needleLen = UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
needle
hayLen :: CountOf Word8
hayLen = UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
haystack
loop :: Maybe (CountOf Word8) -> UArray Word8 -> Bool
loop Nothing _ = Bool
False
loop (Just cnt :: CountOf Word8
cnt) haystack' :: UArray Word8
haystack' = UArray Word8
needle UArray Word8 -> UArray Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== CountOf Word8 -> UArray Word8 -> UArray Word8
forall ty. CountOf ty -> UArray ty -> UArray ty
C.take CountOf Word8
needleLen UArray Word8
haystack' Bool -> Bool -> Bool
|| Maybe (CountOf Word8) -> UArray Word8 -> Bool
loop (CountOf Word8
cntCountOf Word8 -> CountOf Word8 -> Difference (CountOf Word8)
forall a. Subtractive a => a -> a -> Difference a
-1) (CountOf Word8 -> UArray Word8 -> UArray Word8
forall ty. CountOf ty -> UArray ty -> UArray ty
C.drop 1 UArray Word8
haystack')
stripPrefix :: String -> String -> Maybe String
stripPrefix :: String -> String -> Maybe String
stripPrefix (String suffix :: UArray Word8
suffix) (String arr :: UArray Word8
arr)
| UArray Word8 -> UArray Word8 -> Bool
forall ty. PrimType ty => UArray ty -> UArray ty -> Bool
C.isPrefixOf UArray Word8
suffix UArray Word8
arr = String -> Maybe String
forall a. a -> Maybe a
Just (String -> Maybe String) -> String -> Maybe String
forall a b. (a -> b) -> a -> b
$ UArray Word8 -> String
String (UArray Word8 -> String) -> UArray Word8 -> String
forall a b. (a -> b) -> a -> b
$ CountOf Word8 -> UArray Word8 -> UArray Word8
forall ty. CountOf ty -> UArray ty -> UArray ty
C.drop (UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
suffix) UArray Word8
arr
| Bool
otherwise = Maybe String
forall a. Maybe a
Nothing
stripSuffix :: String -> String -> Maybe String
stripSuffix :: String -> String -> Maybe String
stripSuffix (String prefix :: UArray Word8
prefix) (String arr :: UArray Word8
arr)
| UArray Word8 -> UArray Word8 -> Bool
forall ty. PrimType ty => UArray ty -> UArray ty -> Bool
C.isSuffixOf UArray Word8
prefix UArray Word8
arr = String -> Maybe String
forall a. a -> Maybe a
Just (String -> Maybe String) -> String -> Maybe String
forall a b. (a -> b) -> a -> b
$ UArray Word8 -> String
String (UArray Word8 -> String) -> UArray Word8 -> String
forall a b. (a -> b) -> a -> b
$ CountOf Word8 -> UArray Word8 -> UArray Word8
forall ty. CountOf ty -> UArray ty -> UArray ty
C.revDrop (UArray Word8 -> CountOf Word8
forall ty. UArray ty -> CountOf ty
C.length UArray Word8
prefix) UArray Word8
arr
| Bool
otherwise = Maybe String
forall a. Maybe a
Nothing
all :: (Char -> Bool) -> String -> Bool
all :: (Char -> Bool) -> String -> Bool
all predicate :: Char -> Bool
predicate (String arr :: UArray Word8
arr) = (Block Word8 -> Bool)
-> (FinalPtr Word8 -> Ptr Word8 -> ST Any Bool)
-> UArray Word8
-> Bool
forall ty a s.
(Block ty -> a)
-> (FinalPtr ty -> Ptr ty -> ST s a) -> UArray ty -> a
C.onBackend Block Word8 -> Bool
goBA (\_ -> Bool -> ST Any Bool
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Bool -> ST Any Bool)
-> (Ptr Word8 -> Bool) -> Ptr Word8 -> ST Any Bool
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. Ptr Word8 -> Bool
goAddr) UArray Word8
arr
where
!(C.ValidRange start :: Offset Word8
start end :: Offset Word8
end) = UArray Word8 -> ValidRange Word8
forall ty. UArray ty -> ValidRange ty
C.offsetsValidRange UArray Word8
arr
goBA :: Block Word8 -> Bool
goBA ba :: Block Word8
ba = (Char -> Bool)
-> Block Word8 -> Offset Word8 -> Offset Word8 -> Bool
forall container.
Indexable container Word8 =>
(Char -> Bool) -> container -> Offset Word8 -> Offset Word8 -> Bool
UTF8.all Char -> Bool
predicate Block Word8
ba Offset Word8
start Offset Word8
end
goAddr :: Ptr Word8 -> Bool
goAddr addr :: Ptr Word8
addr = (Char -> Bool) -> Ptr Word8 -> Offset Word8 -> Offset Word8 -> Bool
forall container.
Indexable container Word8 =>
(Char -> Bool) -> container -> Offset Word8 -> Offset Word8 -> Bool
UTF8.all Char -> Bool
predicate Ptr Word8
addr Offset Word8
start Offset Word8
end
any :: (Char -> Bool) -> String -> Bool
any :: (Char -> Bool) -> String -> Bool
any predicate :: Char -> Bool
predicate (String arr :: UArray Word8
arr) = (Block Word8 -> Bool)
-> (FinalPtr Word8 -> Ptr Word8 -> ST Any Bool)
-> UArray Word8
-> Bool
forall ty a s.
(Block ty -> a)
-> (FinalPtr ty -> Ptr ty -> ST s a) -> UArray ty -> a
C.onBackend Block Word8 -> Bool
goBA (\_ -> Bool -> ST Any Bool
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Bool -> ST Any Bool)
-> (Ptr Word8 -> Bool) -> Ptr Word8 -> ST Any Bool
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. Ptr Word8 -> Bool
goAddr) UArray Word8
arr
where
!(C.ValidRange start :: Offset Word8
start end :: Offset Word8
end) = UArray Word8 -> ValidRange Word8
forall ty. UArray ty -> ValidRange ty
C.offsetsValidRange UArray Word8
arr
goBA :: Block Word8 -> Bool
goBA ba :: Block Word8
ba = (Char -> Bool)
-> Block Word8 -> Offset Word8 -> Offset Word8 -> Bool
forall container.
Indexable container Word8 =>
(Char -> Bool) -> container -> Offset Word8 -> Offset Word8 -> Bool
UTF8.any Char -> Bool
predicate Block Word8
ba Offset Word8
start Offset Word8
end
goAddr :: Ptr Word8 -> Bool
goAddr addr :: Ptr Word8
addr = (Char -> Bool) -> Ptr Word8 -> Offset Word8 -> Offset Word8 -> Bool
forall container.
Indexable container Word8 =>
(Char -> Bool) -> container -> Offset Word8 -> Offset Word8 -> Bool
UTF8.any Char -> Bool
predicate Ptr Word8
addr Offset Word8
start Offset Word8
end
toBase64 :: String -> String
toBase64 :: String -> String
toBase64 (String src :: UArray Word8
src) = UArray Word8 -> String
fromBytesUnsafe (UArray Word8 -> String)
-> (Bool -> UArray Word8) -> Bool -> String
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. Addr# -> UArray Word8 -> Bool -> UArray Word8
forall ty.
PrimType ty =>
Addr# -> UArray ty -> Bool -> UArray Word8
Vec.toBase64Internal Addr#
set UArray Word8
src (Bool -> String) -> Bool -> String
forall a b. (a -> b) -> a -> b
$ Bool
True
where
!set :: Addr#
set = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"#
toBase64URL :: Bool -> String -> String
toBase64URL :: Bool -> String -> String
toBase64URL padded :: Bool
padded (String src :: UArray Word8
src) = UArray Word8 -> String
fromBytesUnsafe (UArray Word8 -> String)
-> (Bool -> UArray Word8) -> Bool -> String
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. Addr# -> UArray Word8 -> Bool -> UArray Word8
forall ty.
PrimType ty =>
Addr# -> UArray ty -> Bool -> UArray Word8
Vec.toBase64Internal Addr#
set UArray Word8
src (Bool -> String) -> Bool -> String
forall a b. (a -> b) -> a -> b
$ Bool
padded
where
!set :: Addr#
set = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_"#
toBase64OpenBSD :: String -> String
toBase64OpenBSD :: String -> String
toBase64OpenBSD (String src :: UArray Word8
src) = UArray Word8 -> String
fromBytesUnsafe (UArray Word8 -> String)
-> (Bool -> UArray Word8) -> Bool -> String
forall k (cat :: k -> k -> *) (b :: k) (c :: k) (a :: k).
Category cat =>
cat b c -> cat a b -> cat a c
. Addr# -> UArray Word8 -> Bool -> UArray Word8
forall ty.
PrimType ty =>
Addr# -> UArray ty -> Bool -> UArray Word8
Vec.toBase64Internal Addr#
set UArray Word8
src (Bool -> String) -> Bool -> String
forall a b. (a -> b) -> a -> b
$ Bool
False
where
!set :: Addr#
set = "./ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"#