05dde8e · streamly · git.iko.soy

git.iko.soy/streamly

on master help sign in

Add readRev to mutable array module

05dde8eHarendra Kumar4 years ago

core/src/Streamly/Internal/Data/Array/Generic/Mut/Type.hs+8−1

@@ -92,7 +92,7 @@ module Streamly.Internal.Data.Array.Generic.Mut.Type

93 -- ** To containers

94 , toStreamD

95− -- , toStreamDRev

95+ , readRev

96 , toStreamK

97 -- , toStreamKRev

98 , toList

@@ -536,6 +536,7 @@ toStreamD :: MonadIO m => MutArray a -> D.Stream m a

536toStreamD arr@MutArray{..} =

537 D.mapM (`getIndexUnsafe` arr) $ D.enumerateFromToIntegral 0 (arrLen - 1)

538

539+-- Check equivalence with StreamK.fromStream . toStreamD and remove

539{-# INLINE toStreamK #-}

540{-# INLINE toStreamK #-}

540toStreamK :: MonadIO m => MutArray a -> K.StreamK m a

541toStreamK :: MonadIO m => MutArray a -> K.StreamK m a

541toStreamK arr@MutArray{..} = K.unfoldrM step 0

542toStreamK arr@MutArray{..} = K.unfoldrM step 0

@@ -548,6 +549,12 @@ toStreamK arr@MutArray{..} = K.unfoldrM step 0

548 x <- getIndexUnsafe i arr

549 x <- getIndexUnsafe i arr

549 return $ Just (x, i + 1)

550 return $ Just (x, i + 1)

550

551

552+{-# INLINE_NORMAL readRev #-}

553+readRev :: MonadIO m => MutArray a -> D.Stream m a

554+readRev arr@MutArray{..} =

555+ D.mapM (`getIndexUnsafe` arr)

556+ $ D.enumerateFromThenToIntegral (arrLen - 1) (arrLen - 2) 0

557+

551-------------------------------------------------------------------------------

558-------------------------------------------------------------------------------

552-- Folds

559-- Folds

553-------------------------------------------------------------------------------

560-------------------------------------------------------------------------------