@@ -129,6 +129,7 @@ module Data.Text
130 , dropEnd
130 , dropEnd
131 , takeWhile
131 , takeWhile
132 , dropWhile
133 , dropWhile
133 , dropWhileEnd
134 , dropWhileEnd
134 , dropAround
135 , dropAround
@@ -191,15 +192,16 @@ module Data.Text
192 -- * Low level operations
193 -- * Low level operations
196import Prelude (Char, Bool(..), Int, Maybe(..), String,
198import Prelude (Char, Bool(..), Int, Maybe(..), String,
197 Eq(..), Ord(..), Ordering(..), (++),
199 Eq(..), Ord(..), Ordering(..), (++),
198− Read(..), Show(..),
200+ Read(..),
199 (&&), (||), (+), (-), (.), ($), ($!), (>>),
201 (&&), (||), (+), (-), (.), ($), ($!), (>>),
200 not, return, otherwise, quot)
202 not, return, otherwise, quot)
201#if defined(HAVE_DEEPSEQ)
203#if defined(HAVE_DEEPSEQ)
202−import Control.DeepSeq (NFData)
204+import Control.DeepSeq (NFData(rnf))
204#if defined(ASSERTS)
206#if defined(ASSERTS)
205import Control.Exception (assert)
207import Control.Exception (assert)
@@ -211,13 +213,19 @@ import Control.Monad (foldM)
211import Control.Monad.ST (ST)
213import Control.Monad.ST (ST)
212import qualified Data.Text.Array as A
214import qualified Data.Text.Array as A
213import qualified Data.List as L
215import qualified Data.List as L
216+import Data.Binary (Binary(get, put))
214import Data.Monoid (Monoid(..))
217import Data.Monoid (Monoid(..))
218+#if MIN_VERSION_base(4,9,0)
219+import Data.Semigroup (Semigroup(..))
215import Data.String (IsString(..))
221import Data.String (IsString(..))
216import qualified Data.Text.Internal.Fusion as S
222import qualified Data.Text.Internal.Fusion as S
217import qualified Data.Text.Internal.Fusion.Common as S
223import qualified Data.Text.Internal.Fusion.Common as S
224+import Data.Text.Encoding (decodeUtf8', encodeUtf8)
218import Data.Text.Internal.Fusion (stream, reverseStream, unstream)
225import Data.Text.Internal.Fusion (stream, reverseStream, unstream)
219import Data.Text.Internal.Private (span_)
226import Data.Text.Internal.Private (span_)
220import Data.Text.Internal (Text(..), empty, firstf, mul, safe, text)
227import Data.Text.Internal (Text(..), empty, firstf, mul, safe, text)
228+import Data.Text.Show (singleton, unpack, unpackCString#)
221import qualified Prelude as P
229import qualified Prelude as P
222import Data.Text.Unsafe (Iter(..), iter, iter_, lengthWord16, reverseIter,
230import Data.Text.Unsafe (Iter(..), iter, iter_, lengthWord16, reverseIter,
223 reverseIter_, unsafeHead, unsafeTail)
231 reverseIter_, unsafeHead, unsafeTail)
@@ -230,15 +238,13 @@ import Data.ByteString (ByteString)
230import qualified Data.Text.Lazy as L
238import qualified Data.Text.Lazy as L
231import Data.Int (Int64)
239import Data.Int (Int64)
233−#if __GLASGOW_HASKELL__ >= 702
241+import GHC.Base (eqInt, neInt, gtInt, geInt, ltInt, leInt)
234−import qualified GHC.CString as GHC
236−import qualified GHC.Base as GHC
238#if __GLASGOW_HASKELL__ >= 708
242#if __GLASGOW_HASKELL__ >= 708
239import qualified GHC.Exts as Exts
243import qualified GHC.Exts as Exts
241−import GHC.Prim (Addr#)
245+#if MIN_VERSION_base(4,7,0)
246+import Text.Printf (PrintfArg, formatArg, formatString)
243-- $strict
249-- $strict
@@ -324,15 +330,24 @@ instance Eq Text where
324instance Ord Text where
330instance Ord Text where
325 compare = compareText
331 compare = compareText
327−instance Show Text where
328− showsPrec p ps r = showsPrec p (unpack ps) r
330instance Read Text where
333instance Read Text where
331 readsPrec p str = [(pack x,y) | (x,y) <- readsPrec p str]
334 readsPrec p str = [(pack x,y) | (x,y) <- readsPrec p str]
336+#if MIN_VERSION_base(4,9,0)
337+-- Semigroup orphan instances for older GHCs are provided by
338+-- 'semigroups` package
340+instance Semigroup Text where
333instance Monoid Text where
344instance Monoid Text where
334 mempty = empty
345 mempty = empty
346+#if MIN_VERSION_base(4,9,0)
347+ mappend = (<>) -- future-proof definition
335 mappend = append
349 mappend = append
336 mconcat = concat
351 mconcat = concat
338instance IsString Text where
353instance IsString Text where
@@ -346,9 +361,17 @@ instance Exts.IsList Text where
348#if defined(HAVE_DEEPSEQ)
363#if defined(HAVE_DEEPSEQ)
349−instance NFData Text
364+instance NFData Text where rnf !_ = ()
367+instance Binary Text where
368+ put t = put (encodeUtf8 t)
371+ case decodeUtf8' bs of
372+ P.Left exn -> P.fail (P.show exn)
373+ P.Right a -> P.return a
352-- | This instance preserves data abstraction at the cost of inefficiency.
375-- | This instance preserves data abstraction at the cost of inefficiency.
353-- We omit reflection services for the sake of data abstraction.
376-- We omit reflection services for the sake of data abstraction.
@@ -372,6 +395,12 @@ instance Data Text where
372 _ -> P.error "gunfold"
395 _ -> P.error "gunfold"
373 dataTypeOf _ = textDataType
396 dataTypeOf _ = textDataType
398+#if MIN_VERSION_base(4,7,0)
399+-- | Only defined for @base-4.7.0.0@ and later
400+instance PrintfArg Text where
401+ formatArg txt = formatString $ unpack txt
375packConstr :: Constr
404packConstr :: Constr
376packConstr = mkConstr textDataType "pack" [] Prefix
405packConstr = mkConstr textDataType "pack" [] Prefix
@@ -401,30 +430,6 @@ pack :: String -> Text
401pack = unstream . S.map safe . S.streamList
430pack = unstream . S.map safe . S.streamList
402{-# INLINE [1] pack #-}
431{-# INLINE [1] pack #-}
404−-- | /O(n)/ Convert a 'Text' into a 'String'. Subject to fusion.
405−unpack :: Text -> String
406−unpack = S.unstreamList . stream
407−{-# INLINE [1] unpack #-}
409−-- | /O(n)/ Convert a literal string into a Text. Subject to fusion.
410−unpackCString# :: Addr# -> Text
411−unpackCString# addr# = unstream (S.streamCString# addr#)
412−{-# NOINLINE unpackCString# #-}
414−{-# RULES "TEXT literal" forall a.
415− unstream (S.map safe (S.streamList (GHC.unpackCString# a)))
416− = unpackCString# a #-}
418−{-# RULES "TEXT literal UTF8" forall a.
419− unstream (S.map safe (S.streamList (GHC.unpackCStringUtf8# a)))
420− = unpackCString# a #-}
422−-- | /O(1)/ Convert a character into a Text. Subject to fusion.
423−-- Performs replacement on invalid scalar values.
424−singleton :: Char -> Text
425−singleton = unstream . S.singleton . safe
426−{-# INLINE [1] singleton #-}
428-- -----------------------------------------------------------------------------
433-- -----------------------------------------------------------------------------
429-- * Basic functions
434-- * Basic functions
@@ -481,8 +486,8 @@ head t = S.head (stream t)
481uncons :: Text -> Maybe (Char, Text)
486uncons :: Text -> Maybe (Char, Text)
482uncons t@(Text arr off len)
487uncons t@(Text arr off len)
483 | len <= 0 = Nothing
488 | len <= 0 = Nothing
489+ | otherwise = Just $ let !(Iter c d) = iter t 0
484− | otherwise = Just (c, text arr (off+d) (len-d))
490+ in (c, text arr (off+d) (len-d))
485− where Iter c d = iter t 0
486{-# INLINE [1] uncons #-}
491{-# INLINE [1] uncons #-}
488-- | Lifted from Control.Arrow and specialized.
493-- | Lifted from Control.Arrow and specialized.
@@ -567,7 +572,9 @@ isSingleton = S.isSingleton . stream
567-- Subject to fusion.
572-- Subject to fusion.
568length :: Text -> Int
573length :: Text -> Int
569length t = S.length (stream t)
574length t = S.length (stream t)
570−{-# INLINE length #-}
575+{-# INLINE [0] length #-}
576+-- length needs to be phased after the compareN/length rules otherwise
577+-- it may inline before the rules have an opportunity to fire.
572-- | /O(n)/ Compare the count of characters in a 'Text' to a number.
579-- | /O(n)/ Compare the count of characters in a 'Text' to a number.
573-- Subject to fusion.
580-- Subject to fusion.
@@ -586,32 +593,32 @@ compareLength t n = S.compareLengthI (stream t) n
588"TEXT ==N/length -> compareLength/==EQ" [~1] forall t n.
595"TEXT ==N/length -> compareLength/==EQ" [~1] forall t n.
589− (==) (length t) n = compareLength t n == EQ
596+ eqInt (length t) n = compareLength t n == EQ
593"TEXT /=N/length -> compareLength//=EQ" [~1] forall t n.
600"TEXT /=N/length -> compareLength//=EQ" [~1] forall t n.
594− (/=) (length t) n = compareLength t n /= EQ
601+ neInt (length t) n = compareLength t n /= EQ
598"TEXT <N/length -> compareLength/==LT" [~1] forall t n.
605"TEXT <N/length -> compareLength/==LT" [~1] forall t n.
599− (<) (length t) n = compareLength t n == LT
606+ ltInt (length t) n = compareLength t n == LT
603"TEXT <=N/length -> compareLength//=GT" [~1] forall t n.
610"TEXT <=N/length -> compareLength//=GT" [~1] forall t n.
604− (<=) (length t) n = compareLength t n /= GT
611+ leInt (length t) n = compareLength t n /= GT
608"TEXT >N/length -> compareLength/==GT" [~1] forall t n.
615"TEXT >N/length -> compareLength/==GT" [~1] forall t n.
609− (>) (length t) n = compareLength t n == GT
616+ gtInt (length t) n = compareLength t n == GT
613"TEXT >=N/length -> compareLength//=LT" [~1] forall t n.
620"TEXT >=N/length -> compareLength//=LT" [~1] forall t n.
614− (>=) (length t) n = compareLength t n /= LT
621+ geInt (length t) n = compareLength t n /= LT
617-- -----------------------------------------------------------------------------
624-- -----------------------------------------------------------------------------
@@ -642,11 +649,37 @@ reverse :: Text -> Text
642reverse t = S.reverse (stream t)
649reverse t = S.reverse (stream t)
643{-# INLINE reverse #-}
650{-# INLINE reverse #-}
645−-- | /O(m+n)/ Replace every occurrence of one substring with another.
652+-- | /O(m+n)/ Replace every non-overlapping occurrence of @needle@ in
653+-- @haystack@ with @replacement@.
655+-- This function behaves as though it was defined as follows:
658+-- replace needle replacement haystack =
659+-- 'intercalate' replacement ('splitOn' needle haystack)
662+-- As this suggests, each occurrence is replaced exactly once. So if
663+-- @needle@ occurs in @replacement@, that occurrence will /not/ itself
664+-- be replaced recursively:
666+-- > replace "oo" "foo" "oo" == "foo"
668+-- In cases where several instances of @needle@ overlap, only the
669+-- first one will be replaced:
671+-- > replace "ofo" "bar" "ofofo" == "barfo"
647-- In (unlikely) bad cases, this function's time complexity degrades
673-- In (unlikely) bad cases, this function's time complexity degrades
648-- towards /O(n*m)/.
674-- towards /O(n*m)/.
649−replace :: Text -> Text -> Text -> Text
675+replace :: Text
676+ -- ^ @needle@ to search for. If this string is empty, an
677+ -- error will occur.
679+ -- ^ @replacement@ to replace @needle@ with.
681+ -- ^ @haystack@ in which to search.
650replace needle@(Text _ _ neeLen)
683replace needle@(Text _ _ neeLen)
651 (Text repArr repOff repLen)
684 (Text repArr repOff repLen)
652 haystack@(Text hayArr hayOff hayLen)
685 haystack@(Text hayArr hayOff hayLen)
@@ -708,7 +741,7 @@ replace needle@(Text _ _ neeLen)
708-- instead of itself.
741-- instead of itself.
709toCaseFold :: Text -> Text
742toCaseFold :: Text -> Text
710toCaseFold t = unstream (S.toCaseFold (stream t))
743toCaseFold t = unstream (S.toCaseFold (stream t))
711−{-# INLINE [0] toCaseFold #-}
744+{-# INLINE toCaseFold #-}
713-- | /O(n)/ Convert a string to lower case, using simple case
746-- | /O(n)/ Convert a string to lower case, using simple case
714-- conversion. Subject to fusion.
747-- conversion. Subject to fusion.
@@ -882,13 +915,13 @@ concatMap f = concat . foldr ((:) . f) []
882{-# INLINE concatMap #-}
915{-# INLINE concatMap #-}
884-- | /O(n)/ 'any' @p@ @t@ determines whether any character in the
917-- | /O(n)/ 'any' @p@ @t@ determines whether any character in the
885−-- 'Text' @t@ satisifes the predicate @p@. Subject to fusion.
918+-- 'Text' @t@ satisfies the predicate @p@. Subject to fusion.
886any :: (Char -> Bool) -> Text -> Bool
919any :: (Char -> Bool) -> Text -> Bool
887any p t = S.any p (stream t)
920any p t = S.any p (stream t)
888{-# INLINE any #-}
921{-# INLINE any #-}
890-- | /O(n)/ 'all' @p@ @t@ determines whether all characters in the
923-- | /O(n)/ 'all' @p@ @t@ determines whether all characters in the
891−-- 'Text' @t@ satisify the predicate @p@. Subject to fusion.
924+-- 'Text' @t@ satisfy the predicate @p@. Subject to fusion.
892all :: (Char -> Bool) -> Text -> Bool
925all :: (Char -> Bool) -> Text -> Bool
893all p t = S.all p (stream t)
926all p t = S.all p (stream t)
894{-# INLINE all #-}
927{-# INLINE all #-}
@@ -1119,6 +1152,27 @@ takeWhile p t@(Text arr off len) = loop 0
1119 unstream (S.takeWhile p (stream t)) = takeWhile p t
1152 unstream (S.takeWhile p (stream t)) = takeWhile p t
1155+-- | /O(n)/ 'takeWhileEnd', applied to a predicate @p@ and a 'Text',
1156+-- returns the longest suffix (possibly empty) of elements that
1157+-- satisfy @p@. Subject to fusion.
1160+-- > takeWhileEnd (=='o') "foo" == "oo"
1161+takeWhileEnd :: (Char -> Bool) -> Text -> Text
1162+takeWhileEnd p t@(Text arr off len) = loop (len-1) len
1163+ where loop !i !l | l <= 0 = t
1164+ | p c = loop (i+d) (l+d)
1165+ | otherwise = text arr (off+l) (len-l)
1166+ where (c,d) = reverseIter t i
1167+{-# INLINE [1] takeWhileEnd #-}
1170+"TEXT takeWhileEnd -> fused" [~1] forall p t.
1171+ takeWhileEnd p t = S.reverse (S.takeWhile p (S.reverseStream t))
1172+"TEXT takeWhileEnd -> unfused" [1] forall p t.
1173+ S.reverse (S.takeWhile p (S.reverseStream t)) = takeWhileEnd p t
1122-- | /O(n)/ 'dropWhile' @p@ @t@ returns the suffix remaining after
1176-- | /O(n)/ 'dropWhile' @p@ @t@ returns the suffix remaining after
1123-- 'takeWhile' @p@ @t@. Subject to fusion.
1177-- 'takeWhile' @p@ @t@. Subject to fusion.
1124dropWhile :: (Char -> Bool) -> Text -> Text
1178dropWhile :: (Char -> Bool) -> Text -> Text
@@ -1137,8 +1191,9 @@ dropWhile p t@(Text arr off len) = loop 0 0
1139-- | /O(n)/ 'dropWhileEnd' @p@ @t@ returns the prefix remaining after
1193-- | /O(n)/ 'dropWhileEnd' @p@ @t@ returns the prefix remaining after
1140−-- dropping characters that fail the predicate @p@ from the end of
1194+-- dropping characters that satisfy the predicate @p@ from the end of
1141-- @t@. Subject to fusion.
1195-- @t@. Subject to fusion.
1142-- Examples:
1197-- Examples:
1144-- > dropWhileEnd (=='.') "foo..." == "foo"
1199-- > dropWhileEnd (=='.') "foo..." == "foo"
@@ -1158,7 +1213,7 @@ dropWhileEnd p t@(Text arr off len) = loop (len-1) len
1160-- | /O(n)/ 'dropAround' @p@ @t@ returns the substring remaining after
1215-- | /O(n)/ 'dropAround' @p@ @t@ returns the substring remaining after
1161−-- dropping characters that fail the predicate @p@ from both the
1216+-- dropping characters that satisfy the predicate @p@ from both the
1162-- beginning and end of @t@. Subject to fusion.
1217-- beginning and end of @t@. Subject to fusion.
1163dropAround :: (Char -> Bool) -> Text -> Text
1218dropAround :: (Char -> Bool) -> Text -> Text
1164dropAround p = dropWhile p . dropWhileEnd p
1219dropAround p = dropWhile p . dropWhileEnd p
@@ -1253,9 +1308,9 @@ tails t | null t = [empty]
1253-- copies to create substrings; they just construct new 'Text's that
1308-- copies to create substrings; they just construct new 'Text's that
1254-- are slices of the original.
1309-- are slices of the original.
1256−-- | /O(m+n)/ Break a 'Text' into pieces separated by the first
1311+-- | /O(m+n)/ Break a 'Text' into pieces separated by the first 'Text'
1257−-- 'Text' argument, consuming the delimiter. An empty delimiter is
1312+-- argument (which cannot be empty), consuming the delimiter. An empty
1258−-- invalid, and will cause an error to be raised.
1313+-- delimiter is invalid, and will cause an error to be raised.
1260-- Examples:
1315-- Examples:
@@ -1268,9 +1323,16 @@ tails t | null t = [empty]
1268-- > intercalate s . splitOn s == id
1323-- > intercalate s . splitOn s == id
1269-- > splitOn (singleton c) == split (==c)
1324-- > splitOn (singleton c) == split (==c)
1326+-- (Note: the string @s@ to split on above cannot be empty.)
1271-- In (unlikely) bad cases, this function's time complexity degrades
1328-- In (unlikely) bad cases, this function's time complexity degrades
1272-- towards /O(n*m)/.
1329-- towards /O(n*m)/.
1273−splitOn :: Text -> Text -> [Text]
1330+splitOn :: Text
1331+ -- ^ String to split on. If this string is empty, an error
1274splitOn pat@(Text _ _ l) src@(Text arr off len)
1336splitOn pat@(Text _ _ l) src@(Text arr off len)
1275 | l <= 0 = emptyError "splitOn"
1337 | l <= 0 = emptyError "splitOn"
1276 | isSingleton pat = split (== unsafeHead pat) src
1338 | isSingleton pat = split (== unsafeHead pat) src
@@ -1478,7 +1540,7 @@ countChar c t = S.countChar c (stream t)
1478-- equivalent to a pair of 'unpack' operations.
1540-- equivalent to a pair of 'unpack' operations.
1479zip :: Text -> Text -> [(Char,Char)]
1541zip :: Text -> Text -> [(Char,Char)]
1480zip a b = S.unstreamList $ S.zipWith (,) (stream a) (stream b)
1542zip a b = S.unstreamList $ S.zipWith (,) (stream a) (stream b)
1481−{-# INLINE [0] zip #-}
1543+{-# INLINE zip #-}
1483-- | /O(n)/ 'zipWith' generalises 'zip' by zipping with the function
1545-- | /O(n)/ 'zipWith' generalises 'zip' by zipping with the function
1484-- given as the first argument, instead of a tupling function.
1546-- given as the first argument, instead of a tupling function.
@@ -1486,7 +1548,7 @@ zip a b = S.unstreamList $ S.zipWith (,) (stream a) (stream b)
1486zipWith :: (Char -> Char -> Char) -> Text -> Text -> Text
1548zipWith :: (Char -> Char -> Char) -> Text -> Text -> Text
1487zipWith f t1 t2 = unstream (S.zipWith g (stream t1) (stream t2))
1549zipWith f t1 t2 = unstream (S.zipWith g (stream t1) (stream t2))
1488 where g a b = safe (f a b)
1550 where g a b = safe (f a b)
1489−{-# INLINE [0] zipWith #-}
1551+{-# INLINE zipWith #-}
1491-- | /O(n)/ Breaks a 'Text' up into a list of words, delimited by 'Char's
1553-- | /O(n)/ Breaks a 'Text' up into a list of words, delimited by 'Char's
1492-- representing white space.
1554-- representing white space.