7534f3c · text · git.iko.soy

git.iko.soy/text

on master help sign in

Implement takeWhileEnd

7534f3cPaul Capron11 years ago

Data/Text.hs+22

compare with master

unifiedsplit

@@ -129,6 +129,7 @@ module Data.Text

129 , drop

130 , dropEnd

131 , takeWhile

132+ , takeWhileEnd

132 , dropWhile

133 , dropWhile

133 , dropWhileEnd

134 , dropWhileEnd

134 , dropAround

135 , dropAround

@@ -1124,6 +1125,27 @@ takeWhile p t@(Text arr off len) = loop 0

1124 unstream (S.takeWhile p (stream t)) = takeWhile p t

1125 unstream (S.takeWhile p (stream t)) = takeWhile p t

1125 #-}

1126 #-}

1126

1127

1128+-- | /O(n)/ 'takeWhileEnd', applied to a predicate @p@ and a 'Text',

1129+-- returns the longest suffix (possibly empty) of elements that

1130+-- satisfy @p@. Subject to fusion.

1131+-- Examples:

1132+--

1133+-- > takeWhileEnd (=='o') "foo" == "oo"

1134+takeWhileEnd :: (Char -> Bool) -> Text -> Text

1135+takeWhileEnd p t@(Text arr off len) = loop (len-1) len

1136+ where loop !i !l | l <= 0 = t

1137+ | p c = loop (i+d) (l+d)

1138+ | otherwise = text arr (off+l) (len-l)

1139+ where (c,d) = reverseIter t i

1140+{-# INLINE [1] takeWhileEnd #-}

1141+

1142+{-# RULES

1143+"TEXT takeWhileEnd -> fused" [~1] forall p t.

1144+ takeWhileEnd p t = S.reverse (S.takeWhile p (S.reverseStream t))

1145+"TEXT takeWhileEnd -> unfused" [1] forall p t.

1146+ S.reverse (S.takeWhile p (S.reverseStream t)) = takeWhileEnd p t

1147+ #-}

1148+

1127-- | /O(n)/ 'dropWhile' @p@ @t@ returns the suffix remaining after

1149-- | /O(n)/ 'dropWhile' @p@ @t@ returns the suffix remaining after

1128-- 'takeWhile' @p@ @t@. Subject to fusion.

1150-- 'takeWhile' @p@ @t@. Subject to fusion.

1129dropWhile :: (Char -> Bool) -> Text -> Text

1151dropWhile :: (Char -> Bool) -> Text -> Text

Data/Text/Lazy.hs+15

@@ -140,6 +140,7 @@ module Data.Text.Lazy

140 , drop

141 , dropEnd

142 , takeWhile

143+ , takeWhileEnd

143 , dropWhile

144 , dropWhile

144 , dropWhileEnd

145 , dropWhileEnd

145 , dropAround

146 , dropAround

@@ -1119,6 +1120,20 @@ takeWhile p t0 = takeWhile' t0

1119"LAZY TEXT takeWhile -> unfused" [1] forall p t.

1120"LAZY TEXT takeWhile -> unfused" [1] forall p t.

1120 unstream (S.takeWhile p (stream t)) = takeWhile p t

1121 unstream (S.takeWhile p (stream t)) = takeWhile p t

1121 #-}

1122 #-}

1123+-- | /O(n)/ 'takeWhileEnd', applied to a predicate @p@ and a 'Text',

1124+-- returns the longest suffix (possibly empty) of elements that

1125+-- satisfy @p@.

1126+-- Examples:

1127+--

1128+-- > takeWhileEnd (=='o') "foo" == "oo"

1129+takeWhileEnd :: (Char -> Bool) -> Text -> Text

1130+takeWhileEnd p = takeChunk empty . L.reverse . toChunks

1131+ where takeChunk acc [] = acc

1132+ takeChunk acc (t:ts) = if T.length t' < T.length t

1133+ then (Chunk t' acc)

1134+ else takeChunk (Chunk t' acc) ts

1135+ where t' = T.takeWhileEnd p t

1136+{-# INLINE takeWhileEnd #-}

1122

1137

1123-- | /O(n)/ 'dropWhile' @p@ @t@ returns the suffix remaining after

1138-- | /O(n)/ 'dropWhile' @p@ @t@ returns the suffix remaining after

1124-- 'takeWhile' @p@ @t@. Subject to fusion.

1139-- 'takeWhile' @p@ @t@. Subject to fusion.

tests/Tests/Properties.hs+6

@@ -516,6 +516,10 @@ sf_takeWhile q p = (L.takeWhile p . L.filter q) `eqP`

516 (unpackS . S.takeWhile p . S.filter q)

517t_takeWhile p = L.takeWhile p `eqP` (unpackS . T.takeWhile p)

518tl_takeWhile p = L.takeWhile p `eqP` (unpackS . TL.takeWhile p)

519+t_takeWhileEnd p = (L.reverse . L.takeWhile p . L.reverse) `eqP`

520+ (unpackS . T.takeWhileEnd p)

521+tl_takeWhileEnd p = (L.reverse . L.takeWhile p . L.reverse) `eqP`

522+ (unpackS . TL.takeWhileEnd p)

519s_dropWhile p = L.dropWhile p `eqP` (unpackS . S.dropWhile p)

523s_dropWhile p = L.dropWhile p `eqP` (unpackS . S.dropWhile p)

520s_dropWhile_s p = L.dropWhile p `eqP` (unpackS . S.unstream . S.dropWhile p)

524s_dropWhile_s p = L.dropWhile p `eqP` (unpackS . S.unstream . S.dropWhile p)

521sf_dropWhile q p = (L.dropWhile p . L.filter q) `eqP`

525sf_dropWhile q p = (L.dropWhile p . L.filter q) `eqP`

@@ -1142,6 +1146,8 @@ tests =

1142 testProperty "sf_takeWhile" sf_takeWhile,

1146 testProperty "sf_takeWhile" sf_takeWhile,

1143 testProperty "t_takeWhile" t_takeWhile,

1147 testProperty "t_takeWhile" t_takeWhile,

1144 testProperty "tl_takeWhile" tl_takeWhile,

1148 testProperty "tl_takeWhile" tl_takeWhile,

1149+ testProperty "t_takeWhileEnd" t_takeWhileEnd,

1150+ testProperty "tl_takeWhileEnd" tl_takeWhileEnd,

1145 testProperty "sf_dropWhile" sf_dropWhile,

1151 testProperty "sf_dropWhile" sf_dropWhile,

1146 testProperty "s_dropWhile" s_dropWhile,

1152 testProperty "s_dropWhile" s_dropWhile,

1147 testProperty "s_dropWhile_s" s_dropWhile_s,

1153 testProperty "s_dropWhile_s" s_dropWhile_s,