git.iko.soy/text
on master help sign in
Implement lazy dropEnd
6207edfBryan O'Sullivan
Data/Text/Lazy.hs+18
compare with master
unifiedsplit
@@ -130,6 +130,7 @@ module Data.Text.Lazy
130 -- ** Breaking strings
130 -- ** Breaking strings
131 , take
131 , take
132 , drop
132 , drop
133+ , dropEnd
133 , takeWhile
134 , takeWhile
134 , dropWhile
135 , dropWhile
135 , dropWhileEnd
136 , dropWhileEnd
@@ -957,6 +958,23 @@ drop i t0
957 unstream (S.drop n (stream t)) = drop n t
958 unstream (S.drop n (stream t)) = drop n t
958 #-}
959 #-}
959
960
961+-- | /O(n)/ 'dropEnd' @n@ @t@ returns the prefix remaining after
962+-- dropping @n@ characters from the end of @t@.
963+--
964+-- Examples:
965+--
966+-- > dropEnd 3 "foobar" == "foo"
967+dropEnd :: Int64 -> Text -> Text
968+dropEnd n t0
969+ | n <= 0 = t0
970+ | otherwise = dropChunk n . L.reverse . toChunks $ t0
971+ where dropChunk m [] = empty
972+ dropChunk m (t:ts)
973+ | m >= l = dropChunk (m-l) ts
974+ | otherwise = fromChunks . L.reverse $
975+ T.dropEnd (fromIntegral m) t : ts
976+ where l = fromIntegral (T.length t)
977+
960-- | /O(n)/ 'dropWords' @n@ returns the suffix with @n@ 'Word16'
978-- | /O(n)/ 'dropWords' @n@ returns the suffix with @n@ 'Word16'
961-- values dropped, or the empty 'Text' if @n@ is greater than the
979-- values dropped, or the empty 'Text' if @n@ is greater than the
962-- number of 'Word16' values present.
980-- number of 'Word16' values present.
tests/Tests/Properties.hs+3−3
@@ -472,8 +472,8 @@ t_drop n = L.drop n `eqP` (unpackS . T.drop n)
472t_dropEnd n = (L.reverse . L.drop n . L.reverse) `eqP`
472t_dropEnd n = (L.reverse . L.drop n . L.reverse) `eqP`
473 (unpackS . T.dropEnd n)
473 (unpackS . T.dropEnd n)
474tl_drop n = L.drop n `eqP` (unpackS . TL.drop (fromIntegral n))
474tl_drop n = L.drop n `eqP` (unpackS . TL.drop (fromIntegral n))
475−--tl_dropEnd n = (L.reverse . L.drop n . L.reverse) `eqP`
475+tl_dropEnd n = (L.reverse . L.drop n . L.reverse) `eqP`
476−-- (unpackS . TL.dropEnd n)
476+ (unpackS . TL.dropEnd (fromIntegral n))
477s_take_drop m = (L.take n . L.drop n) `eqP` (unpackS . S.take n . S.drop n)
477s_take_drop m = (L.take n . L.drop n) `eqP` (unpackS . S.take n . S.drop n)
478 where n = small m
478 where n = small m
479s_take_drop_s m = (L.take n . L.drop n) `eqP`
479s_take_drop_s m = (L.take n . L.drop n) `eqP`
@@ -1067,7 +1067,7 @@ tests =
1067 testProperty "t_drop" t_drop,
1067 testProperty "t_drop" t_drop,
1068 testProperty "t_dropEnd" t_dropEnd,
1068 testProperty "t_dropEnd" t_dropEnd,
1069 testProperty "tl_drop" tl_drop,
1069 testProperty "tl_drop" tl_drop,
1070− -- testProperty "tl_dropEnd" tl_dropEnd,
1070+ testProperty "tl_dropEnd" tl_dropEnd,
1071 testProperty "s_take_drop" s_take_drop,
1071 testProperty "s_take_drop" s_take_drop,
1072 testProperty "s_take_drop_s" s_take_drop_s,
1072 testProperty "s_take_drop_s" s_take_drop_s,
1073 testProperty "s_takeWhile" s_takeWhile,
1073 testProperty "s_takeWhile" s_takeWhile,