1- {-# OPTIONS_GHC -Wno-deprecations #-}
21{-# OPTIONS_GHC -Wno-orphans #-}
32
43-- |
@@ -47,6 +46,7 @@ import Streamly.Internal.Data.MutArray (MutArray)
4746import qualified Streamly.Internal.Data.Array as Array
4847import qualified Streamly.Internal.Data.MutArray as MArray
4948import qualified Streamly.Internal.Data.Fold as Fold
49+ import qualified Streamly.Internal.Data.Scanl as Scanl
5050import qualified Streamly.Internal.Data.Stream as Stream
5151
5252import Test.Tasty.Bench
@@ -95,7 +95,7 @@ drain = Stream.fold Fold.drain
9595{-# INLINE sourceUnfoldr #-}
9696sourceUnfoldr :: MonadIO m => Int -> Int -> m (Stream Int )
9797sourceUnfoldr value n =
98- Stream. fold (MArray. writeN value) $ Stream. unfoldr step n
98+ Stream. fold (MArray. createOf value) $ Stream. unfoldr step n
9999
100100 where
101101
@@ -107,12 +107,12 @@ sourceUnfoldr value n =
107107{-# INLINE sourceIntFromTo #-}
108108sourceIntFromTo :: MonadIO m => Int -> Int -> m (Stream Int )
109109sourceIntFromTo value n =
110- Stream. fold (MArray. writeN value) $ Stream. enumerateFromTo n (n + value)
110+ Stream. fold (MArray. createOf value) $ Stream. enumerateFromTo n (n + value)
111111
112112{-# INLINE sourceFromList #-}
113113sourceFromList :: MonadIO m => Int -> Int -> m (Stream Int )
114114sourceFromList value n =
115- Stream. fold (MArray. writeN value) $ Stream. fromList [n .. n + value]
115+ Stream. fold (MArray. createOf value) $ Stream. fromList [n .. n + value]
116116
117117{-# INLINE sourceIntFromToFromList #-}
118118sourceIntFromToFromList :: MonadIO m => Int -> Int -> m (Stream Int )
@@ -121,7 +121,7 @@ sourceIntFromToFromList value n = MArray.fromListN value [n..n + value]
121121{-# INLINE sourceIntFromToFromStream #-}
122122sourceIntFromToFromStream :: MonadIO m => Int -> Int -> m (Stream Int )
123123sourceIntFromToFromStream value n =
124- Stream. fold MArray. write $ Stream. enumerateFromTo n (n + value)
124+ Stream. fold MArray. create $ Stream. enumerateFromTo n (n + value)
125125
126126{-# INLINE sourceUnfoldrM #-}
127127sourceUnfoldrM :: Monad m => Int -> Int -> Stream. Stream m Int
@@ -149,7 +149,7 @@ composeN n f x =
149149
150150{-# INLINE scanl' #-}
151151scanl' :: MonadIO m => Int -> Int -> Stream Int -> m (Stream Int )
152- scanl' value n = composeN n $ onArray value $ Stream. scan ( Fold. foldl' (+) 0 )
152+ scanl' value n = composeN n $ onArray value $ Stream. scanl ( Scanl. mkScanl (+) 0 )
153153
154154{-# INLINE scanl1' #-}
155155scanl1' :: MonadIO m => Int -> Int -> Stream Int -> m (Stream Int )
@@ -165,7 +165,7 @@ onArray
165165 -> Stream Int
166166 -> m (Stream Int )
167167onArray value f arr =
168- Stream. fold (MArray. writeN value) $ f $ Stream. unfold MArray. reader arr
168+ Stream. fold (MArray. createOf value) $ f $ Stream. unfold MArray. reader arr
169169
170170-------------------------------------------------------------------------------
171171-- Elimination
@@ -199,12 +199,12 @@ o_1_space_generation :: Int -> [Benchmark]
199199o_1_space_generation value =
200200 [ bgroup
201201 " generation"
202- [ benchIOSrc " writeN . intFromTo" (sourceIntFromTo value)
202+ [ benchIOSrc " createOf . intFromTo" (sourceIntFromTo value)
203203 , benchIOSrc
204204 " fromList . intFromTo"
205205 (sourceIntFromToFromList value)
206- , benchIOSrc " writeN . unfoldr" (sourceUnfoldr value)
207- , benchIOSrc " writeN . fromList" (sourceFromList value)
206+ , benchIOSrc " createOf . unfoldr" (sourceUnfoldr value)
207+ , benchIOSrc " createOf . fromList" (sourceFromList value)
208208 , benchIOSrc " write . intFromTo" (sourceIntFromToFromStream value)
209209 ]
210210 ]
@@ -226,7 +226,7 @@ o_n_heap_serial value =
226226 [ bgroup " elimination"
227227 [
228228 -- Converting the stream to an array
229- benchFold " writeN " (Stream. fold (MArray. writeN value))
229+ benchFold " createOf " (Stream. fold (MArray. createOf value))
230230 (sourceUnfoldrM value)
231231 ]
232232 ]
0 commit comments