@@ -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,11 +192,12 @@ 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)
@@ -211,17 +213,23 @@ 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_)
220−import Data.Text.Internal (Text(..), empty, empty_, firstf, mul, safe, text)
227+import 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)
224−import Data.Text.Internal.Unsafe.Char (unsafeChr, unsafeWrite)
232+import Data.Text.Internal.Unsafe.Char (unsafeChr)
225import qualified Data.Text.Internal.Functions as F
233import qualified Data.Text.Internal.Functions as F
226import qualified Data.Text.Internal.Encoding.Utf16 as U16
234import qualified Data.Text.Internal.Encoding.Utf16 as U16
227import Data.Text.Internal.Search (indices)
235import Data.Text.Internal.Search (indices)
@@ -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
@@ -349,6 +364,14 @@ instance Exts.IsList Text where
349instance NFData Text where rnf !_ = ()
364instance 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,54 +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−{-# RULES "TEXT empty literal"
423− unstream (S.map safe (S.streamList []))
426−{-# RULES "TEXT singleton literal" forall a.
427− unstream (S.map safe (S.streamList [a]))
430−-- | /O(1)/ Convert a character into a Text. Subject to fusion.
431−-- Performs replacement on invalid scalar values.
432−singleton :: Char -> Text
433−singleton = unstream . S.singleton . safe
434−{-# INLINE [1] singleton #-}
436−{-# RULES "TEXT singleton" forall a.
437− unstream (S.singleton (safe a))
440−-- This is intended to reduce inlining bloat.
441−singleton_ :: Char -> Text
442−singleton_ c = Text (A.run x) 0 len
443− where x :: ST s (A.MArray s)
444− x = do arr <- A.new len
445− _ <- unsafeWrite arr 0 d
447− len | d < '\x10000' = 1
450−{-# NOINLINE singleton_ #-}
452-- -----------------------------------------------------------------------------
433-- -----------------------------------------------------------------------------
453-- * Basic functions
434-- * Basic functions
@@ -505,8 +486,8 @@ head t = S.head (stream t)
505uncons :: Text -> Maybe (Char, Text)
486uncons :: Text -> Maybe (Char, Text)
506uncons t@(Text arr off len)
487uncons t@(Text arr off len)
507 | len <= 0 = Nothing
488 | len <= 0 = Nothing
489+ | otherwise = Just $ let !(Iter c d) = iter t 0
508− | otherwise = Just (c, text arr (off+d) (len-d))
490+ in (c, text arr (off+d) (len-d))
509− where Iter c d = iter t 0
510{-# INLINE [1] uncons #-}
491{-# INLINE [1] uncons #-}
512-- | Lifted from Control.Arrow and specialized.
493-- | Lifted from Control.Arrow and specialized.
@@ -591,7 +572,9 @@ isSingleton = S.isSingleton . stream
591-- Subject to fusion.
572-- Subject to fusion.
592length :: Text -> Int
573length :: Text -> Int
593length t = S.length (stream t)
574length t = S.length (stream t)
594−{-# 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.
596-- | /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.
597-- Subject to fusion.
580-- Subject to fusion.
@@ -610,32 +593,32 @@ compareLength t n = S.compareLengthI (stream t) n
612"TEXT ==N/length -> compareLength/==EQ" [~1] forall t n.
595"TEXT ==N/length -> compareLength/==EQ" [~1] forall t n.
613− (==) (length t) n = compareLength t n == EQ
596+ eqInt (length t) n = compareLength t n == EQ
617"TEXT /=N/length -> compareLength//=EQ" [~1] forall t n.
600"TEXT /=N/length -> compareLength//=EQ" [~1] forall t n.
618− (/=) (length t) n = compareLength t n /= EQ
601+ neInt (length t) n = compareLength t n /= EQ
622"TEXT <N/length -> compareLength/==LT" [~1] forall t n.
605"TEXT <N/length -> compareLength/==LT" [~1] forall t n.
623− (<) (length t) n = compareLength t n == LT
606+ ltInt (length t) n = compareLength t n == LT
627"TEXT <=N/length -> compareLength//=GT" [~1] forall t n.
610"TEXT <=N/length -> compareLength//=GT" [~1] forall t n.
628− (<=) (length t) n = compareLength t n /= GT
611+ leInt (length t) n = compareLength t n /= GT
632"TEXT >N/length -> compareLength/==GT" [~1] forall t n.
615"TEXT >N/length -> compareLength/==GT" [~1] forall t n.
633− (>) (length t) n = compareLength t n == GT
616+ gtInt (length t) n = compareLength t n == GT
637"TEXT >=N/length -> compareLength//=LT" [~1] forall t n.
620"TEXT >=N/length -> compareLength//=LT" [~1] forall t n.
638− (>=) (length t) n = compareLength t n /= LT
621+ geInt (length t) n = compareLength t n /= LT
641-- -----------------------------------------------------------------------------
624-- -----------------------------------------------------------------------------
@@ -666,22 +649,36 @@ reverse :: Text -> Text
666reverse t = S.reverse (stream t)
649reverse t = S.reverse (stream t)
667{-# INLINE reverse #-}
650{-# INLINE reverse #-}
669−-- | /O(m+n)/ Replace every occurrence of @needle@ in @haystack@ with @repl@.
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:
671−-- If @needle@ occurs in @repl@, it won't be replaced twice:
657+-- @
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:
673-- > replace "oo" "foo" "oo" == "foo"
666-- > replace "oo" "foo" "oo" == "foo"
675−-- If several instances of @needle@ overlap, only the first one will be
668+-- In cases where several instances of @needle@ overlap, only the
676−-- replaced:
669+-- first one will be replaced:
678-- > replace "ofo" "bar" "ofofo" == "barfo"
671-- > replace "ofo" "bar" "ofofo" == "barfo"
680-- In (unlikely) bad cases, this function's time complexity degrades
673-- In (unlikely) bad cases, this function's time complexity degrades
681-- towards /O(n*m)/.
674-- towards /O(n*m)/.
682−replace :: Text -- ^ @needle@ to search for
675+replace :: Text
676+ -- ^ @needle@ to search for. If this string is empty, an
677+ -- error will occur.
683− -> Text -- ^ @repl@ to replace @needle@ with
679+ -- ^ @replacement@ to replace @needle@ with.
684− -> Text -- ^ @haystack@ in which to search
681+ -- ^ @haystack@ in which to search.
686replace needle@(Text _ _ neeLen)
683replace needle@(Text _ _ neeLen)
687 (Text repArr repOff repLen)
684 (Text repArr repOff repLen)
@@ -744,7 +741,7 @@ replace needle@(Text _ _ neeLen)
744-- instead of itself.
741-- instead of itself.
745toCaseFold :: Text -> Text
742toCaseFold :: Text -> Text
746toCaseFold t = unstream (S.toCaseFold (stream t))
743toCaseFold t = unstream (S.toCaseFold (stream t))
747−{-# INLINE [0] toCaseFold #-}
744+{-# INLINE toCaseFold #-}
749-- | /O(n)/ Convert a string to lower case, using simple case
746-- | /O(n)/ Convert a string to lower case, using simple case
750-- conversion. Subject to fusion.
747-- conversion. Subject to fusion.
@@ -918,13 +915,13 @@ concatMap f = concat . foldr ((:) . f) []
918{-# INLINE concatMap #-}
915{-# INLINE concatMap #-}
920-- | /O(n)/ 'any' @p@ @t@ determines whether any character in the
917-- | /O(n)/ 'any' @p@ @t@ determines whether any character in the
921−-- 'Text' @t@ satisifes the predicate @p@. Subject to fusion.
918+-- 'Text' @t@ satisfies the predicate @p@. Subject to fusion.
922any :: (Char -> Bool) -> Text -> Bool
919any :: (Char -> Bool) -> Text -> Bool
923any p t = S.any p (stream t)
920any p t = S.any p (stream t)
924{-# INLINE any #-}
921{-# INLINE any #-}
926-- | /O(n)/ 'all' @p@ @t@ determines whether all characters in the
923-- | /O(n)/ 'all' @p@ @t@ determines whether all characters in the
927−-- 'Text' @t@ satisify the predicate @p@. Subject to fusion.
924+-- 'Text' @t@ satisfy the predicate @p@. Subject to fusion.
928all :: (Char -> Bool) -> Text -> Bool
925all :: (Char -> Bool) -> Text -> Bool
929all p t = S.all p (stream t)
926all p t = S.all p (stream t)
930{-# INLINE all #-}
927{-# INLINE all #-}
@@ -1155,6 +1152,27 @@ takeWhile p t@(Text arr off len) = loop 0
1155 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
1158-- | /O(n)/ 'dropWhile' @p@ @t@ returns the suffix remaining after
1176-- | /O(n)/ 'dropWhile' @p@ @t@ returns the suffix remaining after
1159-- 'takeWhile' @p@ @t@. Subject to fusion.
1177-- 'takeWhile' @p@ @t@. Subject to fusion.
1160dropWhile :: (Char -> Bool) -> Text -> Text
1178dropWhile :: (Char -> Bool) -> Text -> Text
@@ -1173,8 +1191,9 @@ dropWhile p t@(Text arr off len) = loop 0 0
1175-- | /O(n)/ 'dropWhileEnd' @p@ @t@ returns the prefix remaining after
1193-- | /O(n)/ 'dropWhileEnd' @p@ @t@ returns the prefix remaining after
1176−-- dropping characters that fail the predicate @p@ from the end of
1194+-- dropping characters that satisfy the predicate @p@ from the end of
1177-- @t@. Subject to fusion.
1195-- @t@. Subject to fusion.
1178-- Examples:
1197-- Examples:
1180-- > dropWhileEnd (=='.') "foo..." == "foo"
1199-- > dropWhileEnd (=='.') "foo..." == "foo"
@@ -1194,7 +1213,7 @@ dropWhileEnd p t@(Text arr off len) = loop (len-1) len
1196-- | /O(n)/ 'dropAround' @p@ @t@ returns the substring remaining after
1215-- | /O(n)/ 'dropAround' @p@ @t@ returns the substring remaining after
1197−-- dropping characters that fail the predicate @p@ from both the
1216+-- dropping characters that satisfy the predicate @p@ from both the
1198-- beginning and end of @t@. Subject to fusion.
1217-- beginning and end of @t@. Subject to fusion.
1199dropAround :: (Char -> Bool) -> Text -> Text
1218dropAround :: (Char -> Bool) -> Text -> Text
1200dropAround p = dropWhile p . dropWhileEnd p
1219dropAround p = dropWhile p . dropWhileEnd p
@@ -1289,9 +1308,9 @@ tails t | null t = [empty]
1289-- copies to create substrings; they just construct new 'Text's that
1308-- copies to create substrings; they just construct new 'Text's that
1290-- are slices of the original.
1309-- are slices of the original.
1292−-- | /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'
1293−-- 'Text' argument, consuming the delimiter. An empty delimiter is
1312+-- argument (which cannot be empty), consuming the delimiter. An empty
1294−-- invalid, and will cause an error to be raised.
1313+-- delimiter is invalid, and will cause an error to be raised.
1296-- Examples:
1315-- Examples:
@@ -1304,9 +1323,16 @@ tails t | null t = [empty]
1304-- > intercalate s . splitOn s == id
1323-- > intercalate s . splitOn s == id
1305-- > splitOn (singleton c) == split (==c)
1324-- > splitOn (singleton c) == split (==c)
1326+-- (Note: the string @s@ to split on above cannot be empty.)
1307-- In (unlikely) bad cases, this function's time complexity degrades
1328-- In (unlikely) bad cases, this function's time complexity degrades
1308-- towards /O(n*m)/.
1329-- towards /O(n*m)/.
1309−splitOn :: Text -> Text -> [Text]
1330+splitOn :: Text
1331+ -- ^ String to split on. If this string is empty, an error
1310splitOn pat@(Text _ _ l) src@(Text arr off len)
1336splitOn pat@(Text _ _ l) src@(Text arr off len)
1311 | l <= 0 = emptyError "splitOn"
1337 | l <= 0 = emptyError "splitOn"
1312 | isSingleton pat = split (== unsafeHead pat) src
1338 | isSingleton pat = split (== unsafeHead pat) src
@@ -1514,7 +1540,7 @@ countChar c t = S.countChar c (stream t)
1514-- equivalent to a pair of 'unpack' operations.
1540-- equivalent to a pair of 'unpack' operations.
1515zip :: Text -> Text -> [(Char,Char)]
1541zip :: Text -> Text -> [(Char,Char)]
1516zip a b = S.unstreamList $ S.zipWith (,) (stream a) (stream b)
1542zip a b = S.unstreamList $ S.zipWith (,) (stream a) (stream b)
1517−{-# INLINE [0] zip #-}
1543+{-# INLINE zip #-}
1519-- | /O(n)/ 'zipWith' generalises 'zip' by zipping with the function
1545-- | /O(n)/ 'zipWith' generalises 'zip' by zipping with the function
1520-- given as the first argument, instead of a tupling function.
1546-- given as the first argument, instead of a tupling function.
@@ -1522,7 +1548,7 @@ zip a b = S.unstreamList $ S.zipWith (,) (stream a) (stream b)
1522zipWith :: (Char -> Char -> Char) -> Text -> Text -> Text
1548zipWith :: (Char -> Char -> Char) -> Text -> Text -> Text
1523zipWith f t1 t2 = unstream (S.zipWith g (stream t1) (stream t2))
1549zipWith f t1 t2 = unstream (S.zipWith g (stream t1) (stream t2))
1524 where g a b = safe (f a b)
1550 where g a b = safe (f a b)
1525−{-# INLINE [0] zipWith #-}
1551+{-# INLINE zipWith #-}
1527-- | /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
1528-- representing white space.
1554-- representing white space.