reverse (mutable only)
[darcs-mirrors/vector.git] / Data / Vector / MVector / Mut.hs
index 2b453a4..8c7cd1c 100644 (file)
@@ -3,7 +3,7 @@
 #include "phases.h"
 
 module Data.Vector.MVector.Mut (
-  Mut(..), run, unstream, map, update
+  Mut(..), run, unstream, update, reverse, map
 ) where
 
 import qualified Data.Vector.MVector as MVector
@@ -11,7 +11,7 @@ import           Data.Vector.MVector ( MVector )
 
 import           Data.Vector.Stream ( Stream )
 
-import Prelude hiding ( map )
+import Prelude hiding ( reverse, map )
 
 data Mut a = Mut (forall m mv. MVector mv m a => m (mv a))
 
@@ -19,15 +19,23 @@ run :: MVector mv m a => Mut a -> m (mv a)
 {-# INLINE run #-}
 run (Mut p) = p
 
+trans :: Mut a -> (forall m mv. MVector mv m a => mv a -> m ()) -> Mut a
+{-# INLINE trans #-}
+trans (Mut p) q = Mut (do { v <- p; q v; return v })
+
 unstream :: Stream a -> Mut a
 {-# INLINE_STREAM unstream #-}
 unstream s = Mut (MVector.unstream s)
 
-map :: (a -> a) -> Mut a -> Mut a
-{-# INLINE_STREAM map #-}
-map f (Mut p) = Mut (do { v <- p; MVector.map f v; return v })
-
 update :: Mut a -> Stream (Int, a) -> Mut a
 {-# INLINE_STREAM update #-}
-update (Mut p) s = Mut (do { v <- p; MVector.update v s; return v })
+update m s = trans m (\v -> MVector.update v s)
+
+reverse :: Mut a -> Mut a
+{-# INLINE_STREAM reverse #-}
+reverse m = trans m (MVector.reverse)
+
+map :: (a -> a) -> Mut a -> Mut a
+{-# INLINE_STREAM map #-}
+map f m = trans m (MVector.map f)