@@ -1,6 +1,6 @@
11-- |
22-- Module : Streamly.Benchmark.Data.Stream.StreamK
3−-- Copyright : (c) 2018 Harendra Kumar
3+-- Copyright : (c) 2018 Composewell Technologies
44--
55-- License : BSD3
66-- Maintainer : streamly@composewell.com
@@ -8,6 +8,7 @@
88{-# LANGUAGE CPP #-}
99{-# LANGUAGE FlexibleContexts #-}
1010{-# LANGUAGE ScopedTypeVariables #-}
11+{-# OPTIONS_GHC -Wno-orphans #-}
1112
1213#ifdef __HADDOCK_VERSION__
1314#undef INSPECTION
@@ -20,20 +21,23 @@
2021
2122module Main (main) where
2223
24+#if !MIN_VERSION_base(4,18,0)
25+import Control.Applicative (liftA2)
26+#endif
2327import Control.Monad (when)
2428import Data.Maybe (isJust)
2529import System.Random (randomRIO)
2630import Prelude hiding
27− (tail, mapM_, foldl, last, map, mapM, concatMap, zip, init)
31+ ( Foldable(..), tail, mapM_, last, map, mapM, concatMap, zipWith, init, iterate
32+ , repeat, replicate
33+ )
2834
2935import qualified Prelude as P
3036import qualified Data.List as List
3137
32−import qualified Streamly.Internal.Data.Stream.StreamK as S
33−import qualified Streamly.Internal.Data.Stream.Prelude as SP
34−import qualified Streamly.Internal.Data.SVar as S
38+import qualified Streamly.Internal.Data.StreamK as S
3539
36−import Gauge (bench, nfIO, bgroup, Benchmark, defaultMain)
40+import Test.Tasty.Bench (bench, nfIO, bgroup, Benchmark, defaultMain)
3741
3842import Streamly.Benchmark.Common
3943
@@ -41,106 +45,163 @@ import Streamly.Benchmark.Common
4145import Test.Inspection
4246#endif
4347
44−value, value2, value3, value16, maxValue :: Int
45−value = 100000
46−value2 = round (P.fromIntegral value**(1/2::P.Double)) -- double nested loop
47−value3 = round (P.fromIntegral value**(1/3::P.Double)) -- triple nested loop
48−value16 = round (P.fromIntegral value**(1/16::P.Double)) -- triple nested loop
49−maxValue = value
50−
51−-------------------------------------------------------------------------------
52−-- Benchmark ops
53−-------------------------------------------------------------------------------
54−
55−{-# INLINE toNull #-}
56−{-# INLINE uncons #-}
57−{-# INLINE nullTail #-}
58−{-# INLINE headTail #-}
59−{-# INLINE zip #-}
60−toNull, uncons, nullTail, headTail, zip
61− :: Monad m
62− => Stream m Int -> m ()
63−
64−{-# INLINE toList #-}
65−toList :: Monad m => Stream m Int -> m [Int]
66−{-# INLINE foldl #-}
67−foldl :: Monad m => Stream m Int -> m Int
68−{-# INLINE last #-}
69−last :: Monad m => Stream m Int -> m (Maybe Int)
48+{-
49+
50+Benchmarks that need to be added
51+
52+-- fromList
53+
54+-- bindWith
55+-- concatPairsWith
56+-- apWith
57+-- apSerial
58+-- apSerialDiscardFst
59+-- apSerialDiscardSnd
60+
61+-- the
62+-- serial
63+-- consMStream
64+-- withLocal
65+-- mfix
66+
67+-- elem
68+-- notElem
69+-- all
70+-- any
71+-- minimum
72+-- minimumBy
73+-- maximum
74+-- maximumBy
75+-- findIndices
76+-- lookup
77+-- findM
78+-- find
79+-- (!!)
80+
81+-- foldlM'
82+-- foldlT
83+-- foldlx'
84+-- foldlMx'
85+-- fold
86+
87+-- sequence
88+
89+-- foldrSM
90+-- buildS
91+-- buildM
92+-- augmentS
93+-- augmentSM
94+-- foldr
95+-- foldr1
96+-- foldrM
97+-- foldrT
98+
99+-- intersperseM
100+-- insertBy
101+-- deleteBy
102+-- reverse
103+-- mapMaybe
104+
105+-- zipWithM
106+-- mergeBy
107+-- mergeByM
108+
109+-- toStreamK (Probably can be skipped)
110+-- hoist
111+
112+-- scanlx'
113+
114+-}
70115
71116-------------------------------------------------------------------------------
72117-- Stream generation and elimination
73118-------------------------------------------------------------------------------
74119
75−type Stream m a = S.Stream m a
120+type Stream m a = S.StreamK m a
76121
77−{-# INLINE sourceUnfoldr #-}
78−sourceUnfoldr :: Int -> Stream m Int
79−sourceUnfoldr n = S.unfoldr step n
122+{-# INLINE unfoldr #-}
123+unfoldr :: Int -> Int -> Stream m Int
124+unfoldr streamLen n = S.unfoldr step n
80125 where
81126 step cnt =
82− if cnt > n + value
127+ if cnt > n + streamLen
83128 then Nothing
84129 else Just (cnt, cnt + 1)
85130
86−{-# INLINE sourceUnfoldrN #-}
87−sourceUnfoldrN :: Int -> Int -> Stream m Int
88−sourceUnfoldrN m n = S.unfoldr step n
131+{-# INLINE unfoldrM #-}
132+unfoldrM :: Monad m => Int -> Int -> Stream m Int
133+unfoldrM streamLen n = S.unfoldrMWith S.consM step n
89134 where
90135 step cnt =
91− if cnt > n + m
92− then Nothing
93− else Just (cnt, cnt + 1)
94−
95−{-# INLINE sourceUnfoldrM #-}
96−sourceUnfoldrM :: S.MonadAsync m => Int -> Stream m Int
97−sourceUnfoldrM n = S.unfoldrM step n
98− where
99− step cnt =
100− if cnt > n + value
101− then return Nothing
102− else return (Just (cnt, cnt + 1))
103−
104−{-# INLINE sourceUnfoldrMN #-}
105−sourceUnfoldrMN :: S.MonadAsync m => Int -> Int -> Stream m Int
106−sourceUnfoldrMN m n = S.unfoldrM step n
107− where
108− step cnt =
109− if cnt > n + m
136+ if cnt > n + streamLen
110137 then return Nothing
111138 else return (Just (cnt, cnt + 1))
112139
113−{-# INLINE sourceFromFoldable #-}
114−sourceFromFoldable :: Int -> Stream m Int
115−sourceFromFoldable n = S.fromFoldable [n..n+value]
116−
117−{-# INLINE sourceFromFoldableM #-}
118−sourceFromFoldableM :: S.MonadAsync m => Int -> Stream m Int
119−sourceFromFoldableM n =
120− Prelude.foldr S.consM S.nil (Prelude.fmap return [n..n+value])
121−
122−{-# INLINE sourceFoldMapWith #-}
123−sourceFoldMapWith :: Int -> Stream m Int
124−sourceFoldMapWith n = SP.foldMapWith S.serial S.yield [n..n+value]
125−
126−{-# INLINE sourceFoldMapWithM #-}
127−sourceFoldMapWithM :: Monad m => Int -> Stream m Int
128−sourceFoldMapWithM n = SP.foldMapWith S.serial (S.yieldM . return) [n..n+value]
140+{-# INLINE repeat #-}
141+repeat :: Int -> Int -> Stream m Int
142+repeat streamLen = S.take streamLen . S.repeat
143+
144+{-# INLINE repeatM #-}
145+repeatM :: Monad m => Int -> Int -> Stream m Int
146+repeatM streamLen = S.take streamLen . S.repeatM . return
147+
148+{-# INLINE replicate #-}
149+replicate :: Int -> Int -> Stream m Int
150+replicate = S.replicate
151+
152+{-# INLINE replicateM #-}
153+replicateM :: Monad m => Int -> Int -> Stream m Int
154+replicateM streamLen = S.replicateMWith S.consM streamLen . return
155+
156+{-# INLINE iterate #-}
157+iterate :: Int -> Int -> Stream m Int
158+iterate streamLen = S.take streamLen . S.iterate (+1)
159+
160+{-# INLINE iterateM #-}
161+iterateM :: Monad m => Int -> Int -> Stream m Int
162+iterateM streamLen = S.take streamLen . S.iterateM (return . (+1)) . return
163+
164+{-# INLINE fromFoldable #-}
165+fromFoldable :: Int -> Int -> Stream m Int
166+fromFoldable streamLen n = S.fromFoldable [n..n+streamLen]
167+
168+{-# INLINE fromFoldableM #-}
169+fromFoldableM :: Monad m => Int -> Int -> Stream m Int
170+fromFoldableM streamLen n =
171+ List.foldr S.consM S.nil (Prelude.fmap return [n..n+streamLen])
172+
173+{-# INLINABLE concatMapFoldableWith #-}
174+concatMapFoldableWith :: P.Foldable f
175+ => (Stream m b -> Stream m b -> Stream m b)
176+ -> (a -> Stream m b)
177+ -> f a
178+ -> Stream m b
179+concatMapFoldableWith f g = P.foldr (f . g) S.nil
180+
181+{-# INLINE concatMapFoldableSerial #-}
182+concatMapFoldableSerial :: Int -> Int -> Stream m Int
183+concatMapFoldableSerial streamLen n =
184+ concatMapFoldableWith S.append S.fromPure [n..n+streamLen]
185+
186+{-# INLINE concatMapFoldableSerialM #-}
187+concatMapFoldableSerialM :: Monad m => Int -> Int -> Stream m Int
188+concatMapFoldableSerialM streamLen n =
189+ concatMapFoldableWith S.append (S.fromEffect . return) [n..n+streamLen]
129190
130191-------------------------------------------------------------------------------
131192-- Elimination
132193-------------------------------------------------------------------------------
133194
134−{-# INLINE runStream #-}
135−runStream :: Monad m => Stream m a -> m ()
136−runStream = S.drain
137−-- runStream = S.mapM_ (\_ -> return ())
195+{-# INLINE drain #-}
196+drain :: Monad m => Stream m a -> m ()
197+drain = S.drain
138198
139199{-# INLINE mapM_ #-}
140200mapM_ :: Monad m => Stream m a -> m ()
141201mapM_ = S.mapM_ (\_ -> return ())
142202
143−toNull = runStream
203+{-# INLINE uncons #-}
204+uncons :: Monad m => Stream m Int -> m ()
144205uncons s = do
145206 r <- S.uncons s
146207 case r of
@@ -148,167 +209,238 @@ uncons s = do
148209 Just (_, t) -> uncons t
149210
150211{-# INLINE init #-}
151−init :: (Monad m, S.IsStream t) => t m a -> m ()
212+init :: Monad m => Stream m a -> m ()
152213init s = do
153214 t <- S.init s
154215 P.mapM_ S.drain t
155216
156217{-# INLINE tail #-}
157−tail :: (Monad m, S.IsStream t) => t m a -> m ()
218+tail :: Monad m => Stream m a -> m ()
158219tail s = S.tail s >>= P.mapM_ tail
159220
221+{-# INLINE nullTail #-}
222+nullTail :: Monad m => Stream m Int -> m ()
160223nullTail s = do
161224 r <- S.null s
162225 when (not r) $ S.tail s >>= P.mapM_ nullTail
163226
227+{-# INLINE headTail #-}
228+headTail :: Monad m => Stream m Int -> m ()
164229headTail s = do
165230 h <- S.head s
166231 when (isJust h) $ S.tail s >>= P.mapM_ headTail
167232
233+{-# INLINE toList #-}
234+toList :: Monad m => Stream m Int -> m [Int]
168235toList = S.toList
169−foldl = S.foldl' (+) 0
170−last = S.last
236+
237+{-# INLINE foldl' #-}
238+foldl' :: Monad m => Stream m Int -> m Int
239+foldl' = S.foldl' (+) 0
240+
241+{-# INLINE last #-}
242+last :: Monad m => Stream m Int -> m (Maybe Int)
243+last = S.last
171244
172245-------------------------------------------------------------------------------
173246-- Transformation
174247-------------------------------------------------------------------------------
175248
176−{-# INLINE transform #-}
177−transform :: Monad m => Stream m a -> m ()
178−transform = runStream
179−
180249{-# INLINE composeN #-}
181250composeN
182251 :: Monad m
183252 => Int -> (Stream m Int -> Stream m Int) -> Stream m Int -> m ()
184253composeN n f =
185254 case n of
186− 1 -> transform . f
187− 2 -> transform . f . f
188− 3 -> transform . f . f . f
189− 4 -> transform . f . f . f . f
255+ 1 -> drain . f
256+ 2 -> drain . f . f
257+ 3 -> drain . f . f . f
258+ 4 -> drain . f . f . f . f
190259 _ -> undefined
191260
192−{-# INLINE scan #-}
261+{-# INLINE scanl' #-}
262+scanl' :: Monad m => Int -> Stream m Int -> m ()
263+scanl' n = composeN n $ S.scanl' (+) 0
264+
193265{-# INLINE map #-}
266+map :: Monad m => Int -> Stream m Int -> m ()
267+map n = composeN n $ S.map (+ 1)
268+
194269{-# INLINE fmapK #-}
270+fmapK :: Monad m => Int -> Stream m Int -> m ()
271+fmapK n = composeN n $ P.fmap (+ 1)
272+
273+{-# INLINE mapM #-}
274+mapM :: Monad m => Int -> Stream m Int -> m ()
275+mapM n = composeN n $ S.mapMWith S.consM return
276+
277+{-# INLINE mapMSerial #-}
278+mapMSerial :: Monad m => Int -> Stream m Int -> m ()
279+mapMSerial n = composeN n $ S.mapMSerial return
280+
195281{-# INLINE filterEven #-}
282+filterEven :: Monad m => Int -> Stream m Int -> m ()
283+filterEven n = composeN n $ S.filter even
284+
196285{-# INLINE filterAllOut #-}
286+filterAllOut :: Monad m => Int -> Int -> Stream m Int -> m ()
287+filterAllOut streamLen n = composeN n $ S.filter (> streamLen)
288+
197289{-# INLINE filterAllIn #-}
290+filterAllIn :: Monad m => Int -> Int -> Stream m Int -> m ()
291+filterAllIn streamLen n = composeN n $ S.filter (<= streamLen)
292+
198293{-# INLINE _takeOne #-}
294+_takeOne :: Monad m => Int -> Stream m Int -> m ()
295+_takeOne n = composeN n $ S.take 1
296+
199297{-# INLINE takeAll #-}
298+takeAll :: Monad m => Int -> Int -> Stream m Int -> m ()
299+takeAll streamLen n = composeN n $ S.take streamLen
300+
200301{-# INLINE takeWhileTrue #-}
302+takeWhileTrue :: Monad m => Int -> Int -> Stream m Int -> m ()
303+takeWhileTrue streamLen n = composeN n $ S.takeWhile (<= streamLen)
304+
201305{-# INLINE dropOne #-}
306+dropOne :: Monad m => Int -> Stream m Int -> m ()
307+dropOne n = composeN n $ S.drop 1
308+
202309{-# INLINE dropAll #-}
310+dropAll :: Monad m => Int -> Int -> Stream m Int -> m ()
311+dropAll streamLen n = composeN n $ S.drop streamLen
312+
203313{-# INLINE dropWhileTrue #-}
314+dropWhileTrue :: Monad m => Int -> Int -> Stream m Int -> m ()
315+dropWhileTrue streamLen n = composeN n $ S.dropWhile (<= streamLen)
316+
204317{-# INLINE dropWhileFalse #-}
318+dropWhileFalse :: Monad m => Int -> Stream m Int -> m ()
319+dropWhileFalse n = composeN n $ S.dropWhile (<= 1)
320+
205321{-# INLINE foldrS #-}
322+foldrS :: Monad m => Int -> Stream m Int -> m ()
323+foldrS n = composeN n $ S.foldrS S.cons S.nil
324+
206325{-# INLINE foldlS #-}
207−scan, map, fmapK, filterEven, filterAllOut,
208− filterAllIn, _takeOne, takeAll, takeWhileTrue, dropAll, dropOne,
209− dropWhileTrue, dropWhileFalse, foldrS, foldlS
210− :: Monad m
211− => Int -> Stream m Int -> m ()
326+foldlS :: Monad m => Int -> Stream m Int -> m ()
327+foldlS n = composeN n $ S.foldlS (flip S.cons) S.nil
212328
213−{-# INLINE mapM #-}
214−{-# INLINE mapMSerial #-}
215329{-# INLINE intersperse #-}
216−mapM, mapMSerial, intersperse
217− :: S.MonadAsync m => Int -> Stream m Int -> m ()
218−
219−scan n = composeN n $ S.scanl' (+) 0
220−map n = composeN n $ P.fmap (+1)
221−fmapK n = composeN n $ P.fmap (+1)
222−mapM n = composeN n $ S.mapM return
223−mapMSerial n = composeN n $ S.mapMSerial return
224−filterEven n = composeN n $ S.filter even
225−filterAllOut n = composeN n $ S.filter (> maxValue)
226−filterAllIn n = composeN n $ S.filter (<= maxValue)
227−_takeOne n = composeN n $ S.take 1
228−takeAll n = composeN n $ S.take maxValue
229−takeWhileTrue n = composeN n $ S.takeWhile (<= maxValue)
230−dropOne n = composeN n $ S.drop 1
231−dropAll n = composeN n $ S.drop maxValue
232−dropWhileTrue n = composeN n $ S.dropWhile (<= maxValue)
233−dropWhileFalse n = composeN n $ S.dropWhile (<= 1)
234−foldrS n = composeN n $ S.foldrS S.cons S.nil
235−foldlS n = composeN n $ S.foldlS (flip S.cons) S.nil
236−intersperse n = composeN n $ S.intersperse maxValue
330+intersperse :: Monad m => Int -> Int -> Stream m Int -> m ()
331+intersperse streamLen n = composeN n $ S.intersperse streamLen
237332
238333-------------------------------------------------------------------------------
239334-- Iteration
240335-------------------------------------------------------------------------------
241336
242−iterStreamLen, maxIters :: Int
243−iterStreamLen = 10
244−maxIters = 10000
245−
246337{-# INLINE iterateSource #-}
247338iterateSource
248− :: S.MonadAsync m
249− => (Stream m Int -> Stream m Int) -> Int -> Int -> Stream m Int
250−iterateSource g i n = f i (sourceUnfoldrMN iterStreamLen n)
339+ :: Monad m => Int -> (Stream m Int -> Stream m Int) -> Int -> Int -> Stream m Int
340+iterateSource iterStreamLen g i n = f i (unfoldrM iterStreamLen n)
251341 where
252342 f (0 :: Int) m = g m
253343 f x m = g (f (x P.- 1) m)
254344
255−{-# INLINE iterateMapM #-}
345+-- this is quadratic
256346{-# INLINE iterateScan #-}
347+iterateScan :: Monad m => Int -> Int -> Int -> Stream m Int
348+iterateScan iterStreamLen maxIters =
349+ iterateSource iterStreamLen (S.scanl' (+) 0) (maxIters `div` 10)
350+
351+-- this is quadratic
352+{-# INLINE iterateDropWhileFalse #-}
353+iterateDropWhileFalse :: Monad m => Int -> Int -> Int -> Int -> Stream m Int
354+iterateDropWhileFalse streamLen iterStreamLen maxIters =
355+ iterateSource iterStreamLen (S.dropWhile (> streamLen)) (maxIters `div` 10)
356+
357+{-# INLINE iterateMapM #-}
358+iterateMapM :: Monad m => Int -> Int -> Int -> Stream m Int
359+iterateMapM iterStreamLen =
360+ iterateSource iterStreamLen (S.mapMWith S.consM return)
361+
257362{-# INLINE iterateFilterEven #-}
363+iterateFilterEven :: Monad m => Int -> Int -> Int -> Stream m Int
364+iterateFilterEven iterStreamLen = iterateSource iterStreamLen (S.filter even)
365+
258366{-# INLINE iterateTakeAll #-}
259−{-# INLINE iterateDropOne #-}
260−{-# INLINE iterateDropWhileFalse #-}
261−{-# INLINE iterateDropWhileTrue #-}
262−iterateMapM, iterateScan, iterateFilterEven, iterateTakeAll, iterateDropOne,
263− iterateDropWhileFalse, iterateDropWhileTrue
264− :: S.MonadAsync m
265− => Int -> Stream m Int
367+iterateTakeAll :: Monad m => Int -> Int -> Int -> Int -> Stream m Int
368+iterateTakeAll streamLen iterStreamLen =
369+ iterateSource iterStreamLen (S.take streamLen)
266370
267−-- this is quadratic
268−iterateScan = iterateSource (S.scanl' (+) 0) (maxIters `div` 10)
269−iterateDropWhileFalse = iterateSource (S.dropWhile (> maxValue))
270− (maxIters `div` 10)
371+{-# INLINE iterateDropOne #-}
372+iterateDropOne :: Monad m => Int -> Int -> Int -> Stream m Int
373+iterateDropOne iterStreamLen = iterateSource iterStreamLen (S.drop 1)
271374
272−iterateMapM = iterateSource (S.mapM return) maxIters
273−iterateFilterEven = iterateSource (S.filter even) maxIters
274−iterateTakeAll = iterateSource (S.take maxValue) maxIters
275−iterateDropOne = iterateSource (S.drop 1) maxIters
276−iterateDropWhileTrue = iterateSource (S.dropWhile (<= maxValue)) maxIters
375+{-# INLINE iterateDropWhileTrue #-}
376+iterateDropWhileTrue ::
377+ Monad m => Int -> Int -> Int -> Int -> Stream m Int
378+iterateDropWhileTrue streamLen iterStreamLen =
379+ iterateSource iterStreamLen (S.dropWhile (<= streamLen))
277380
278381-------------------------------------------------------------------------------
279382-- Zipping
280383-------------------------------------------------------------------------------
281384
282−zip src = transform $ S.zipWith (,) src src
385+{-# INLINE zipWith #-}
386+zipWith :: Monad m => Stream m Int -> m ()
387+zipWith src = drain $ S.zipWith (,) src src
388+
389+{-# INLINE zipWithM #-}
390+zipWithM :: Monad m => Stream m Int -> m ()
391+zipWithM src = drain $ S.zipWithM (curry return) src src
392+
393+{-# INLINE sortByK #-}
394+sortByK :: (a -> a -> Ordering) -> Stream m a -> Stream m a
395+sortByK f = S.mergeMapWith (S.mergeBy f) S.fromPure
396+
397+{-# INLINE sortBy #-}
398+sortBy :: Monad m => Stream m Int -> m ()
399+sortBy = drain . sortByK compare
283400
284401-------------------------------------------------------------------------------
285402-- Mixed Composition
286403-------------------------------------------------------------------------------
287404
288405{-# INLINE scanMap #-}
406+scanMap :: Monad m => Int -> Stream m Int -> m ()
407+scanMap n = composeN n $ S.map (subtract 1) . S.scanl' (+) 0
408+
289409{-# INLINE dropMap #-}
410+dropMap :: Monad m => Int -> Stream m Int -> m ()
411+dropMap n = composeN n $ S.map (subtract 1) . S.drop 1
412+
290413{-# INLINE dropScan #-}
414+dropScan :: Monad m => Int -> Stream m Int -> m ()
415+dropScan n = composeN n $ S.scanl' (+) 0 . S.drop 1
416+
291417{-# INLINE takeDrop #-}
418+takeDrop :: Monad m => Int -> Int -> Stream m Int -> m ()
419+takeDrop streamLen n = composeN n $ S.drop 1 . S.take streamLen
420+
292421{-# INLINE takeScan #-}
422+takeScan :: Monad m => Int -> Int -> Stream m Int -> m ()
423+takeScan streamLen n = composeN n $ S.scanl' (+) 0 . S.take streamLen
424+
293425{-# INLINE takeMap #-}
426+takeMap :: Monad m => Int -> Int -> Stream m Int -> m ()
427+takeMap streamLen n = composeN n $ S.map (subtract 1) . S.take streamLen
428+
294429{-# INLINE filterDrop #-}
430+filterDrop :: Monad m => Int -> Int -> Stream m Int -> m ()
431+filterDrop streamLen n = composeN n $ S.drop 1 . S.filter (<= streamLen)
432+
295433{-# INLINE filterTake #-}
434+filterTake :: Monad m => Int -> Int -> Stream m Int -> m ()
435+filterTake streamLen n = composeN n $ S.take streamLen . S.filter (<= streamLen)
436+
296437{-# INLINE filterScan #-}
297−{-# INLINE filterMap #-}
298−scanMap, dropMap, dropScan, takeDrop, takeScan, takeMap, filterDrop,
299− filterTake, filterScan, filterMap
300− :: Monad m => Int -> Stream m Int -> m ()
301−
302−scanMap n = composeN n $ S.map (subtract 1) . S.scanl' (+) 0
303−dropMap n = composeN n $ S.map (subtract 1) . S.drop 1
304−dropScan n = composeN n $ S.scanl' (+) 0 . S.drop 1
305−takeDrop n = composeN n $ S.drop 1 . S.take maxValue
306−takeScan n = composeN n $ S.scanl' (+) 0 . S.take maxValue
307−takeMap n = composeN n $ S.map (subtract 1) . S.take maxValue
308−filterDrop n = composeN n $ S.drop 1 . S.filter (<= maxValue)
309−filterTake n = composeN n $ S.take maxValue . S.filter (<= maxValue)
438+filterScan :: Monad m => Int -> Stream m Int -> m ()
310439filterScan n = composeN n $ S.scanl' (+) 0 . S.filter (<= maxBound)
311−filterMap n = composeN n $ S.map (subtract 1) . S.filter (<= maxValue)
440+
441+{-# INLINE filterMap #-}
442+filterMap :: Monad m => Int -> Int -> Stream m Int -> m ()
443+filterMap streamLen n = composeN n $ S.map (subtract 1) . S.filter (<= streamLen)
312444
313445-------------------------------------------------------------------------------
314446-- ConcatMap
@@ -320,8 +452,8 @@ filterMap n = composeN n $ S.map (subtract 1) . S.filter (<= maxValue)
320452concatMap :: Int -> Int -> Int -> IO ()
321453concatMap outer inner n =
322454 S.drain $ S.concatMap
323− (\_ -> sourceUnfoldrMN inner n)
324− (sourceUnfoldrMN outer n)
455+ (\_ -> unfoldrM inner n)
456+ (unfoldrM outer n)
325457
326458#ifdef INSPECTION
327459inspect $ hasNoTypeClasses 'concatMap
@@ -333,8 +465,8 @@ inspect $ hasNoTypeClasses 'concatMap
333465concatMapPure :: Int -> Int -> Int -> IO ()
334466concatMapPure outer inner n =
335467 S.drain $ S.concatMap
336− (\_ -> sourceUnfoldrN inner n)
337− (sourceUnfoldrN outer n)
468+ (\_ -> unfoldr inner n)
469+ (unfoldr outer n)
338470
339471#ifdef INSPECTION
340472inspect $ hasNoTypeClasses 'concatMapPure
@@ -345,7 +477,7 @@ inspect $ hasNoTypeClasses 'concatMapPure
345477{-# INLINE concatMapRepl #-}
346478concatMapRepl :: Int -> Int -> Int -> IO ()
347479concatMapRepl outer inner n =
348− S.drain $ S.concatMap (S.replicate inner) (sourceUnfoldrMN outer n)
480+ S.drain $ S.concatMap (S.replicate inner) (unfoldrM outer n)
349481
350482#ifdef INSPECTION
351483inspect $ hasNoTypeClasses 'concatMapRepl
@@ -357,44 +489,70 @@ inspect $ hasNoTypeClasses 'concatMapRepl
357489sourceConcatMapId :: Monad m
358490 => Int -> Int -> Stream m (Stream m Int)
359491sourceConcatMapId val n =
360− S.fromFoldable $ fmap (S.yieldM . return) [n..n+val]
492+ S.fromFoldable $ fmap (S.fromEffect . return) [n..n+val]
361493
362−{-# INLINE concatStreamsWith #-}
363−concatStreamsWith :: Int -> Int -> Int -> IO ()
364−concatStreamsWith outer inner n =
365− S.drain $ S.concatMapBy S.serial
366− (sourceUnfoldrMN inner)
367− (sourceUnfoldrMN outer n)
494+{-# INLINE concatMapBySerial #-}
495+concatMapBySerial :: Int -> Int -> Int -> IO ()
496+concatMapBySerial outer inner n =
497+ S.drain $ S.concatMapWith S.append
498+ (unfoldrM inner)
499+ (unfoldrM outer n)
368500
369501-------------------------------------------------------------------------------
370502-- Nested Composition
371503-------------------------------------------------------------------------------
372504
373−{-# INLINE toNullApNested #-}
374−toNullApNested :: Monad m => Stream m Int -> m ()
375−toNullApNested s = runStream $ do
505+instance Monad m => Applicative (S.StreamK m) where
506+ {-# INLINE pure #-}
507+ pure = S.fromPure
508+
509+ {-# INLINE (<*>) #-}
510+ (<*>) = S.crossApply
511+
512+ {-# INLINE liftA2 #-}
513+ liftA2 f x = (<*>) (fmap f x)
514+
515+ {-# INLINE (*>) #-}
516+ (*>) = S.crossApplySnd
517+
518+ {-# INLINE (<*) #-}
519+ (<*) = S.crossApplyFst
520+
521+-- NOTE: even though concatMap for StreamD is 3x faster compared to StreamK,
522+-- the monad instance of StreamD is slower than StreamK after foldr/build
523+-- fusion.
524+instance Monad m => Monad (S.StreamK m) where
525+ {-# INLINE return #-}
526+ return = pure
527+
528+ {-# INLINE (>>=) #-}
529+ (>>=) = flip S.concatMap
530+
531+{-# INLINE drainApplicative #-}
532+drainApplicative :: Monad m => Stream m Int -> m ()
533+drainApplicative s = drain $ do
376534 (+) <$> s <*> s
377535
378−{-# INLINE toNullNested #-}
379−toNullNested :: Monad m => Stream m Int -> m ()
380−toNullNested s = runStream $ do
536+{-# INLINE drainMonad #-}
537+drainMonad :: Monad m => Stream m Int -> m ()
538+drainMonad s = drain $ do
381539 x <- s
382540 y <- s
383541 return $ x + y
384542
385−{-# INLINE toNullNested3 #-}
386−toNullNested3 :: Monad m => Stream m Int -> m ()
387−toNullNested3 s = runStream $ do
543+{-# INLINE drainMonad3 #-}
544+drainMonad3 :: Monad m => Stream m Int -> m ()
545+drainMonad3 s = drain $ do
388546 x <- s
389547 y <- s
390548 z <- s
391549 return $ x + y + z
392550
393−{-# INLINE filterAllOutNested #-}
394−filterAllOutNested
551+{-# INLINE filterAllOutMonad #-}
552+filterAllOutMonad
395553 :: Monad m
396554 => Stream m Int -> m ()
397−filterAllOutNested str = runStream $ do
555+filterAllOutMonad str = drain $ do
398556 x <- str
399557 y <- str
400558 let s = x + y
@@ -402,11 +560,11 @@ filterAllOutNested str = runStream $ do
402560 then return s
403561 else S.nil
404562
405−{-# INLINE filterAllInNested #-}
406−filterAllInNested
563+{-# INLINE filterAllInMonad #-}
564+filterAllInMonad
407565 :: Monad m
408566 => Stream m Int -> m ()
409−filterAllInNested str = runStream $ do
567+filterAllInMonad str = drain $ do
410568 x <- str
411569 y <- str
412570 let s = x + y
@@ -418,9 +576,14 @@ filterAllInNested str = runStream $ do
418576-- Nested Composition Pure lists
419577-------------------------------------------------------------------------------
420578
421−{-# INLINE sourceUnfoldrList #-}
422−sourceUnfoldrList :: Int -> Int -> [Int]
423−sourceUnfoldrList maxval n = List.unfoldr step n
579+-- There are several list benchmarks here for comparison with lists. It is easy
580+-- and convenient to see the comparisons when they are here, otherwise we'll
581+-- have to add a separate module for list benchmarks with the same names and
582+-- then add a comparison in bench.sh.
583+
584+{-# INLINE unfoldrList #-}
585+unfoldrList :: Int -> Int -> [Int]
586+unfoldrList maxval n = List.unfoldr step n
424587 where
425588 step cnt =
426589 if cnt > n + maxval
@@ -471,229 +634,301 @@ filterAllInNestedList str = do
471634-------------------------------------------------------------------------------
472635
473636moduleName :: String
474−moduleName = "Data.Stream.StreamK"
475−
476−o_1_space :: [Benchmark]
477−o_1_space =
478− [ bgroup (o_1_space_prefix moduleName)
479− [ bgroup "generation"
480− [ benchFold "unfoldr" toNull sourceUnfoldr
481− , benchFold "unfoldrM" toNull sourceUnfoldrM
482−
483− , benchFold "fromFoldable" toNull sourceFromFoldable
484− , benchFold "fromFoldableM" toNull sourceFromFoldableM
637+moduleName = "Data.StreamK"
638+
639+o_1_space_generation :: Int -> Benchmark
640+o_1_space_generation streamLen =
641+ bgroup "generation"
642+ [ benchFold "unfoldr" drain (unfoldr streamLen)
643+ , benchFold "unfoldrM" drain (unfoldrM streamLen)
644+ , benchFold "repeat" drain (repeat streamLen)
645+ , benchFold "repeatM" drain (repeatM streamLen)
646+ , benchFold "replicate" drain (replicate streamLen)
647+ , benchFold "replicateM" drain (replicateM streamLen)
648+ , benchFold "iterate" drain (iterate streamLen)
649+ , benchFold "iterateM" drain (iterateM streamLen)
650+
651+ , benchFold "fromFoldable" drain (fromFoldable streamLen)
652+ , benchFold "fromFoldableM" drain (fromFoldableM streamLen)
485653
486654 -- appends
487− , benchFold "foldMapWith" toNull sourceFoldMapWith
488− , benchFold "foldMapWithM" toNull sourceFoldMapWithM
655+ , benchFold "concatMapFoldableWith" drain (concatMapFoldableSerial streamLen)
656+ , benchFold "concatMapFoldableWithM" drain (concatMapFoldableSerialM streamLen)
489657 ]
490− , bgroup "elimination"
491− [ benchFold "toNull" toNull sourceUnfoldrM
492− , benchFold "mapM_" mapM_ sourceUnfoldrM
493− , benchFold "uncons" uncons sourceUnfoldrM
494− , benchFold "init" init sourceUnfoldrM
495− , benchFold "foldl'" foldl sourceUnfoldrM
496− , benchFold "last" last sourceUnfoldrM
658+
659+o_1_space_elimination :: Int -> Benchmark
660+o_1_space_elimination streamLen =
661+ bgroup "elimination"
662+ [ benchFold "toNull" drain (unfoldrM streamLen)
663+ , benchFold "mapM_" mapM_ (unfoldrM streamLen)
664+ , benchFold "uncons" uncons (unfoldrM streamLen)
665+ , benchFold "init" init (unfoldrM streamLen)
666+ , benchFold "foldl'" foldl' (unfoldrM streamLen)
667+ , benchFold "last" last (unfoldrM streamLen)
497668 ]
498− , bgroup "nested"
499− [ benchFold "toNullAp" toNullApNested (sourceUnfoldrMN value2)
500− , benchFold "toNull" toNullNested (sourceUnfoldrMN value2)
501− , benchFold "toNull3" toNullNested3 (sourceUnfoldrMN value3)
502− , benchFold "filterAllIn" filterAllInNested (sourceUnfoldrMN value2)
503− , benchFold "filterAllOut" filterAllOutNested (sourceUnfoldrMN value2)
504− , benchFold "toNullApPure" toNullApNested (sourceUnfoldrN value2)
505− , benchFold "toNullPure" toNullNested (sourceUnfoldrN value2)
506− , benchFold "toNull3Pure" toNullNested3 (sourceUnfoldrN value3)
507− , benchFold "filterAllInPure" filterAllInNested (sourceUnfoldrN value2)
508− , benchFold "filterAllOutPure" filterAllOutNested (sourceUnfoldrN value2)
669+
670+o_1_space_nested :: Int -> Benchmark
671+o_1_space_nested streamLen =
672+ bgroup "nested"
673+ [ benchFold "drainApplicative" drainApplicative (unfoldrM streamLen2)
674+ , benchFold "drainMonad" drainMonad (unfoldrM streamLen2)
675+ , benchFold "drainMonad3" drainMonad3 (unfoldrM streamLen3)
676+ , benchFold "filterAllInMonad" filterAllInMonad (unfoldrM streamLen2)
677+ , benchFold "filterAllOutMonad" filterAllOutMonad (unfoldrM streamLen2)
678+ , benchFold "drainApplicative (pure)" drainApplicative (unfoldr streamLen2)
679+ , benchFold "drainMonad (pure)" drainMonad (unfoldr streamLen2)
680+ , benchFold "drainMonad3 (pure)" drainMonad3 (unfoldr streamLen3)
681+ , benchFold "filterAllInMonad (pure)" filterAllInMonad (unfoldr streamLen2)
682+ , benchFold "filterAllOutMonad (pure)" filterAllOutMonad (unfoldr streamLen2)
509683 ]
510− , bgroup "transformation"
511− [ benchFold "foldrS" (foldrS 1) sourceUnfoldrM
512− , benchFold "scan" (scan 1) sourceUnfoldrM
513− , benchFold "map" (map 1) sourceUnfoldrM
514− , benchFold "fmap" (fmapK 1) sourceUnfoldrM
515− , benchFold "mapM" (mapM 1) sourceUnfoldrM
516− , benchFold "mapMSerial" (mapMSerial 1) sourceUnfoldrM
684+ where
685+ streamLen2 = round (P.fromIntegral streamLen**(1/2::P.Double)) -- double nested loop
686+ streamLen3 = round (P.fromIntegral streamLen**(1/3::P.Double)) -- triple nested loop
687+
688+o_1_space_transformation :: Int -> Benchmark
689+o_1_space_transformation streamLen =
690+ bgroup "transformation"
691+ [ benchFold "foldrS" (foldrS 1) (unfoldrM streamLen)
692+ , benchFold "scanl'" (scanl' 1) (unfoldrM streamLen)
693+ , benchFold "map" (map 1) (unfoldrM streamLen)
694+ , benchFold "fmap" (fmapK 1) (unfoldrM streamLen)
695+ , benchFold "mapM" (mapM 1) (unfoldrM streamLen)
696+ , benchFold "mapMSerial" (mapMSerial 1) (unfoldrM streamLen)
517697 ]
518− , bgroup "transformationX4"
519− [ benchFold "scan" (scan 4) sourceUnfoldrM
520− , benchFold "map" (map 4) sourceUnfoldrM
521− , benchFold "fmap" (fmapK 4) sourceUnfoldrM
522− , benchFold "mapM" (mapM 4) sourceUnfoldrM
523− , benchFold "mapMSerial" (mapMSerial 4) sourceUnfoldrM
698+
699+o_1_space_transformationX4 :: Int -> Benchmark
700+o_1_space_transformationX4 streamLen =
701+ bgroup "transformationX4"
702+ [ benchFold "scanl'" (scanl' 4) (unfoldrM streamLen)
703+ , benchFold "map" (map 4) (unfoldrM streamLen)
704+ , benchFold "fmap" (fmapK 4) (unfoldrM streamLen)
705+ , benchFold "mapM" (mapM 4) (unfoldrM streamLen)
706+ , benchFold "mapMSerial" (mapMSerial 4) (unfoldrM streamLen)
524707 -- XXX this is horribly slow
525− -- , benchFold "concatMap" (concatMap 4) (sourceUnfoldrMN value16)
708+ -- , benchFold "concatMap" (concatMap 4) (unfoldrM streamLen16)
526709 ]
527− , bgroup "concat"
710+
711+o_1_space_concat :: Int -> Benchmark
712+o_1_space_concat streamLen =
713+ bgroup "concat"
528714 [ benchIOSrc1 "concatMapPure (n of 1)"
529− (concatMapPure value 1)
715+ (concatMapPure streamLen 1)
530716 , benchIOSrc1 "concatMapPure (sqrt n of sqrt n)"
531− (concatMapPure value2 value2)
717+ (concatMapPure streamLen2 streamLen2)
532718 , benchIOSrc1 "concatMapPure (1 of n)"
533− (concatMapPure 1 value)
719+ (concatMapPure 1 streamLen)
534720
535721 , benchIOSrc1 "concatMap (n of 1)"
536− (concatMap value 1)
722+ (concatMap streamLen 1)
537723 , benchIOSrc1 "concatMap (sqrt n of sqrt n)"
538− (concatMap value2 value2)
724+ (concatMap streamLen2 streamLen2)
539725 , benchIOSrc1 "concatMap (1 of n)"
540− (concatMap 1 value)
726+ (concatMap 1 streamLen)
541727
542728 , benchIOSrc1 "concatMapRepl (sqrt n of sqrt n)"
543− (concatMapRepl value2 value2)
729+ (concatMapRepl streamLen2 streamLen2)
544730
545− -- This is for comparison with foldMapWith
731+ -- This is for comparison with concatMapFoldableWith
546732 , benchIOSrc1 "concatMapWithId (n of 1) (fromFoldable)"
547− (S.drain . S.concatMapBy S.serial id . sourceConcatMapId value)
548−
549− , benchIOSrc1 "concatMapWith (n of 1)"
550− (concatStreamsWith value 1)
551− , benchIOSrc1 "concatMapWith (sqrt n of sqrt n)"
552− (concatStreamsWith value2 value2)
553− , benchIOSrc1 "concatMapWith (1 of n)"
554− (concatStreamsWith 1 value)
733+ (S.drain
734+ . S.concatMapWith S.append id
735+ . sourceConcatMapId streamLen)
736+
737+ , benchIOSrc1 "concatMapBy serial (n of 1)"
738+ (concatMapBySerial streamLen 1)
739+ , benchIOSrc1 "concatMapBy serial (sqrt n of sqrt n)"
740+ (concatMapBySerial streamLen2 streamLen2)
741+ , benchIOSrc1 "concatMapBy serial (1 of n)"
742+ (concatMapBySerial 1 streamLen)
743+ , benchFold "sortBy" sortBy (unfoldrM streamLen)
555744 ]
556− , bgroup "filtering"
557− [ benchFold "filter-even" (filterEven 1) sourceUnfoldrM
558− , benchFold "filter-all-out" (filterAllOut 1) sourceUnfoldrM
559− , benchFold "filter-all-in" (filterAllIn 1) sourceUnfoldrM
560− , benchFold "take-all" (takeAll 1) sourceUnfoldrM
561− , benchFold "takeWhile-true" (takeWhileTrue 1) sourceUnfoldrM
562− , benchFold "drop-one" (dropOne 1) sourceUnfoldrM
563− , benchFold "drop-all" (dropAll 1) sourceUnfoldrM
564− , benchFold "dropWhile-true" (dropWhileTrue 1) sourceUnfoldrM
565− , benchFold "dropWhile-false" (dropWhileFalse 1) sourceUnfoldrM
745+ where
746+ streamLen2 = round (P.fromIntegral streamLen**(1/2::P.Double)) -- double nested loop
747+
748+o_1_space_filtering :: Int -> Benchmark
749+o_1_space_filtering streamLen =
750+ bgroup "filtering"
751+ [ benchFold "filter-even" (filterEven 1) (unfoldrM streamLen)
752+ , benchFold "filter-all-out" (filterAllOut streamLen 1) (unfoldrM streamLen)
753+ , benchFold "filter-all-in" (filterAllIn streamLen 1) (unfoldrM streamLen)
754+ , benchFold "take-all" (takeAll streamLen 1) (unfoldrM streamLen)
755+ , benchFold "takeWhile-true" (takeWhileTrue streamLen 1) (unfoldrM streamLen)
756+ , benchFold "drop-one" (dropOne 1) (unfoldrM streamLen)
757+ , benchFold "drop-all" (dropAll streamLen 1) (unfoldrM streamLen)
758+ , benchFold "dropWhile-true" (dropWhileTrue streamLen 1) (unfoldrM streamLen)
759+ , benchFold "dropWhile-false" (dropWhileFalse 1) (unfoldrM streamLen)
566760 ]
567− , bgroup "filteringX4"
568− [ benchFold "filter-even" (filterEven 4) sourceUnfoldrM
569− , benchFold "filter-all-out" (filterAllOut 4) sourceUnfoldrM
570− , benchFold "filter-all-in" (filterAllIn 4) sourceUnfoldrM
571− , benchFold "take-all" (takeAll 4) sourceUnfoldrM
572− , benchFold "takeWhile-true" (takeWhileTrue 4) sourceUnfoldrM
573− , benchFold "drop-one" (dropOne 4) sourceUnfoldrM
574− , benchFold "drop-all" (dropAll 4) sourceUnfoldrM
575− , benchFold "dropWhile-true" (dropWhileTrue 4) sourceUnfoldrM
576− , benchFold "dropWhile-false" (dropWhileFalse 4) sourceUnfoldrM
761+
762+o_1_space_filteringX4 :: Int -> Benchmark
763+o_1_space_filteringX4 streamLen =
764+ bgroup "filteringX4"
765+ [ benchFold "filter-even" (filterEven 4) (unfoldrM streamLen)
766+ , benchFold "filter-all-out" (filterAllOut streamLen 4) (unfoldrM streamLen)
767+ , benchFold "filter-all-in" (filterAllIn streamLen 4) (unfoldrM streamLen)
768+ , benchFold "take-all" (takeAll streamLen 4) (unfoldrM streamLen)
769+ , benchFold "takeWhile-true" (takeWhileTrue streamLen 4) (unfoldrM streamLen)
770+ , benchFold "drop-one" (dropOne 4) (unfoldrM streamLen)
771+ , benchFold "drop-all" (dropAll streamLen 4) (unfoldrM streamLen)
772+ , benchFold "dropWhile-true" (dropWhileTrue streamLen 4) (unfoldrM streamLen)
773+ , benchFold "dropWhile-false" (dropWhileFalse 4) (unfoldrM streamLen)
577774 ]
578− , bgroup "zipping"
579− [ benchFold "zip" zip sourceUnfoldrM
775+
776+o_1_space_zipping :: Int -> Benchmark
777+o_1_space_zipping streamLen =
778+ bgroup "zipping"
779+ [ benchFold "zipWith" zipWith (unfoldrM streamLen)
780+ , benchFold "zipWithM" zipWithM (unfoldrM streamLen)
580781 ]
581− , bgroup "mixed"
582− [ benchFold "scan-map" (scanMap 1) sourceUnfoldrM
583− , benchFold "drop-map" (dropMap 1) sourceUnfoldrM
584− , benchFold "drop-scan" (dropScan 1) sourceUnfoldrM
585− , benchFold "take-drop" (takeDrop 1) sourceUnfoldrM
586− , benchFold "take-scan" (takeScan 1) sourceUnfoldrM
587− , benchFold "take-map" (takeMap 1) sourceUnfoldrM
588− , benchFold "filter-drop" (filterDrop 1) sourceUnfoldrM
589− , benchFold "filter-take" (filterTake 1) sourceUnfoldrM
590− , benchFold "filter-scan" (filterScan 1) sourceUnfoldrM
591− , benchFold "filter-map" (filterMap 1) sourceUnfoldrM
782+
783+o_1_space_mixed :: Int -> Benchmark
784+o_1_space_mixed streamLen =
785+ bgroup "mixed"
786+ [ benchFold "scan-map" (scanMap 1) (unfoldrM streamLen)
787+ , benchFold "drop-map" (dropMap 1) (unfoldrM streamLen)
788+ , benchFold "drop-scan" (dropScan 1) (unfoldrM streamLen)
789+ , benchFold "take-drop" (takeDrop streamLen 1) (unfoldrM streamLen)
790+ , benchFold "take-scan" (takeScan streamLen 1) (unfoldrM streamLen)
791+ , benchFold "take-map" (takeMap streamLen 1) (unfoldrM streamLen)
792+ , benchFold "filter-drop" (filterDrop streamLen 1) (unfoldrM streamLen)
793+ , benchFold "filter-take" (filterTake streamLen 1) (unfoldrM streamLen)
794+ , benchFold "filter-scan" (filterScan 1) (unfoldrM streamLen)
795+ , benchFold "filter-map" (filterMap streamLen 1) (unfoldrM streamLen)
592796 ]
593− , bgroup "mixedX2"
594− [ benchFold "scan-map" (scanMap 2) sourceUnfoldrM
595− , benchFold "drop-map" (dropMap 2) sourceUnfoldrM
596− , benchFold "drop-scan" (dropScan 2) sourceUnfoldrM
597− , benchFold "take-drop" (takeDrop 2) sourceUnfoldrM
598− , benchFold "take-scan" (takeScan 2) sourceUnfoldrM
599− , benchFold "take-map" (takeMap 2) sourceUnfoldrM
600− , benchFold "filter-drop" (filterDrop 2) sourceUnfoldrM
601− , benchFold "filter-take" (filterTake 2) sourceUnfoldrM
602− , benchFold "filter-scan" (filterScan 2) sourceUnfoldrM
603− , benchFold "filter-map" (filterMap 2) sourceUnfoldrM
797+
798+o_1_space_mixedX2 :: Int -> Benchmark
799+o_1_space_mixedX2 streamLen =
800+ bgroup "mixedX2"
801+ [ benchFold "scan-map" (scanMap 2) (unfoldrM streamLen)
802+ , benchFold "drop-map" (dropMap 2) (unfoldrM streamLen)
803+ , benchFold "drop-scan" (dropScan 2) (unfoldrM streamLen)
804+ , benchFold "take-drop" (takeDrop streamLen 2) (unfoldrM streamLen)
805+ , benchFold "take-scan" (takeScan streamLen 2) (unfoldrM streamLen)
806+ , benchFold "take-map" (takeMap streamLen 2) (unfoldrM streamLen)
807+ , benchFold "filter-drop" (filterDrop streamLen 2) (unfoldrM streamLen)
808+ , benchFold "filter-take" (filterTake streamLen 2) (unfoldrM streamLen)
809+ , benchFold "filter-scan" (filterScan 2) (unfoldrM streamLen)
810+ , benchFold "filter-map" (filterMap streamLen 2) (unfoldrM streamLen)
604811 ]
605− , bgroup "mixedX4"
606− [ benchFold "scan-map" (scanMap 4) sourceUnfoldrM
607− , benchFold "drop-map" (dropMap 4) sourceUnfoldrM
608− , benchFold "drop-scan" (dropScan 4) sourceUnfoldrM
609− , benchFold "take-drop" (takeDrop 4) sourceUnfoldrM
610− , benchFold "take-scan" (takeScan 4) sourceUnfoldrM
611− , benchFold "take-map" (takeMap 4) sourceUnfoldrM
612− , benchFold "filter-drop" (filterDrop 4) sourceUnfoldrM
613− , benchFold "filter-take" (filterTake 4) sourceUnfoldrM
614− , benchFold "filter-scan" (filterScan 4) sourceUnfoldrM
615− , benchFold "filter-map" (filterMap 4) sourceUnfoldrM
812+
813+o_1_space_mixedX4 :: Int -> Benchmark
814+o_1_space_mixedX4 streamLen =
815+ bgroup "mixedX4"
816+ [ benchFold "scan-map" (scanMap 4) (unfoldrM streamLen)
817+ , benchFold "drop-map" (dropMap 4) (unfoldrM streamLen)
818+ , benchFold "drop-scan" (dropScan 4) (unfoldrM streamLen)
819+ , benchFold "take-drop" (takeDrop streamLen 4) (unfoldrM streamLen)
820+ , benchFold "take-scan" (takeScan streamLen 4) (unfoldrM streamLen)
821+ , benchFold "take-map" (takeMap streamLen 4) (unfoldrM streamLen)
822+ , benchFold "filter-drop" (filterDrop streamLen 4) (unfoldrM streamLen)
823+ , benchFold "filter-take" (filterTake streamLen 4) (unfoldrM streamLen)
824+ , benchFold "filter-scan" (filterScan 4) (unfoldrM streamLen)
825+ , benchFold "filter-map" (filterMap streamLen 4) (unfoldrM streamLen)
826+ ]
827+
828+o_1_space_list :: Int -> Benchmark
829+o_1_space_list streamLen =
830+ bgroup "list"
831+ [ bgroup "elimination"
832+ [ benchList "last" (\xs -> [List.last xs]) (unfoldrList streamLen)
833+ ]
834+ , bgroup "nested"
835+ [ benchList "toNullAp" toNullApNestedList (unfoldrList streamLen2)
836+ , benchList "toNull" toNullNestedList (unfoldrList streamLen2)
837+ , benchList "toNull3" toNullNestedList3 (unfoldrList streamLen3)
838+ , benchList "filterAllIn" filterAllInNestedList (unfoldrList streamLen2)
839+ , benchList "filterAllOut" filterAllOutNestedList (unfoldrList streamLen2)
616840 ]
617841 ]
618− ]
842+ where
843+
844+ streamLen2 = round (P.fromIntegral streamLen**(1/2::P.Double)) -- double nested loop
845+ streamLen3 = round (P.fromIntegral streamLen**(1/3::P.Double)) -- triple nested loop
846+
847+o_1_space :: Int -> Benchmark
848+o_1_space streamLen =
849+ bgroup (o_1_space_prefix moduleName)
850+ [ o_1_space_generation streamLen
851+ , o_1_space_elimination streamLen
852+ , o_1_space_nested streamLen
853+ , o_1_space_transformation streamLen
854+ , o_1_space_transformationX4 streamLen
855+ , o_1_space_concat streamLen
856+ , o_1_space_filtering streamLen
857+ , o_1_space_filteringX4 streamLen
858+ , o_1_space_zipping streamLen
859+ , o_1_space_mixed streamLen
860+ , o_1_space_mixedX2 streamLen
861+ , o_1_space_mixedX4 streamLen
862+ , o_1_space_list streamLen
863+ ]
619864
620−o_n_heap :: [Benchmark]
621−o_n_heap =
622− [ bgroup (o_n_heap_prefix moduleName)
865+o_n_heap :: Int -> Benchmark
866+o_n_heap streamLen =
867+ bgroup (o_n_heap_prefix moduleName)
623868 [ bgroup "transformation"
624− [ benchFold "foldlS" (foldlS 1) sourceUnfoldrM
869+ [ benchFold "foldlS" (foldlS 1) (unfoldrM streamLen)
625870 ]
626871 ]
627− ]
628872
629873{-# INLINE benchK #-}
630874benchK :: P.String -> (Int -> Stream P.IO Int) -> Benchmark
631−benchK name f = bench name $ nfIO $ randomRIO (1,1) >>= toNull . f
875+benchK name f = bench name $ nfIO $ randomRIO (1,1) >>= drain . f
632876
633−o_n_stack :: [Benchmark]
634−o_n_stack =
635− [ bgroup (o_n_stack_prefix moduleName)
877+o_n_stack :: Int -> Int -> Int -> Benchmark
878+o_n_stack streamLen iterStreamLen maxIters =
879+ bgroup (o_n_stack_prefix moduleName)
636880 [ bgroup "elimination"
637− [ benchFold "tail" tail sourceUnfoldrM
638− , benchFold "nullTail" nullTail sourceUnfoldrM
639− , benchFold "headTail" headTail sourceUnfoldrM
881+ [ benchFold "tail" tail (unfoldrM streamLen)
882+ , benchFold "nullTail" nullTail (unfoldrM streamLen)
883+ , benchFold "headTail" headTail (unfoldrM streamLen)
640884 ]
641885 , bgroup "transformation"
642886 [
643887 -- XXX why do these need so much stack
644− benchFold "intersperse" (intersperse 1) (sourceUnfoldrMN value2)
645− , benchFold "interspersePure" (intersperse 1) (sourceUnfoldrN value2)
888+ benchFold "intersperse" (intersperse streamLen 1) (unfoldrM streamLen2)
889+ , benchFold "interspersePure" (intersperse streamLen 1) (unfoldr streamLen2)
646890 ]
647891 , bgroup "transformationX4"
648892 [
649− benchFold "intersperse" (intersperse 4) (sourceUnfoldrMN value16)
893+ benchFold "intersperse" (intersperse streamLen 4) (unfoldrM streamLen16)
650894 ]
651895 , bgroup "iterated"
652− [ benchK "mapM" iterateMapM
653− , benchK "scan(1/10)" iterateScan
654− , benchK "filterEven" iterateFilterEven
655− , benchK "takeAll" iterateTakeAll
656− , benchK "dropOne" iterateDropOne
657− , benchK "dropWhileFalse(1/10)" iterateDropWhileFalse
658− , benchK "dropWhileTrue" iterateDropWhileTrue
896+ [ benchK "mapM" (iterateMapM iterStreamLen maxIters)
897+ , benchK "scan(1/10)" (iterateScan iterStreamLen maxIters)
898+ , benchK "filterEven" (iterateFilterEven iterStreamLen maxIters)
899+ , benchK "takeAll" (iterateTakeAll streamLen iterStreamLen maxIters)
900+ , benchK "dropOne" (iterateDropOne iterStreamLen maxIters)
901+ , benchK "dropWhileFalse(1/10)" (iterateDropWhileFalse streamLen iterStreamLen maxIters)
902+ , benchK "dropWhileTrue" (iterateDropWhileTrue streamLen iterStreamLen maxIters)
659903 ]
660904 ]
661− ]
905+ where
906+ streamLen2 = round (P.fromIntegral streamLen**(1/2::P.Double)) -- double nested loop
907+ streamLen16 = round (P.fromIntegral streamLen**(1/16::P.Double)) -- triple nested loop
662908
663−o_n_space :: [Benchmark]
664−o_n_space =
665− [ bgroup (o_n_space_prefix moduleName)
909+o_n_space :: Int -> Benchmark
910+o_n_space streamLen =
911+ bgroup (o_n_space_prefix moduleName)
666912 [ bgroup "elimination"
667− [ benchFold "toList" toList sourceUnfoldrM
913+ [ benchFold "toList" toList (unfoldrM streamLen)
668914 ]
669915 ]
670− ]
671916
672917{-# INLINE benchList #-}
673918benchList :: P.String -> ([Int] -> [Int]) -> (Int -> [Int]) -> Benchmark
674919benchList name run f = bench name $ nfIO $ randomRIO (1,1) >>= return . run . f
675920
676−o_1_space_list :: [Benchmark]
677−o_1_space_list =
678− [ bgroup "list"
679− [ bgroup "elimination"
680− [ benchList "last" (\xs -> [List.last xs]) (sourceUnfoldrList value)
681− ]
682− , bgroup "nested"
683− [ benchList "toNullAp" toNullApNestedList (sourceUnfoldrList value2)
684− , benchList "toNull" toNullNestedList (sourceUnfoldrList value2)
685− , benchList "toNull3" toNullNestedList3 (sourceUnfoldrList value3)
686− , benchList "filterAllIn" filterAllInNestedList (sourceUnfoldrList value2)
687− , benchList "filterAllOut" filterAllOutNestedList (sourceUnfoldrList value2)
921+main :: IO ()
922+main =
923+ defaultMain
924+ [ o_1_space streamLen
925+ , o_n_stack streamLen iterStreamLen maxIters
926+ , o_n_heap streamLen
927+ , o_n_space streamLen
688928 ]
689− ]
690− ]
691929
692−main :: IO ()
693−main = defaultMain $ concat
694− [ o_1_space
695− , [bgroup (o_1_space_prefix moduleName) o_1_space_list]
696− , o_n_stack
697− , o_n_heap
698− , o_n_space
699− ]
930+ where
931+
932+ streamLen = 100000
933+ maxIters = 10000
934+ iterStreamLen = 10