Fix up

core/src/Streamly/Internal/Data/IORef/Unboxed.hs

@@ -25,19 +25,14 @@ module Streamly.Internal.Data.IORef.Unboxed
 
     -- * Construction
     , newIORef
-    , newIORefWithIndex
 
     -- * Write
     , writeIORef
-    , writeIORefWithIndex
     , modifyIORef'
-    , modifyIORefWithIndex'
 
     -- * Read
     , readIORef
-    , readIORefWithIndex
     , toStreamD
-    , toStreamDWithIndex
     )
 where
 
@@ -57,78 +52,38 @@ import qualified Streamly.Internal.Data.Stream.Type as D
 -- | An 'IORef' holds a single 'Unbox'-able value with offset.
 data IORef a = IORef !MutableByteArray !Int
 
--- | Create a new 'IORef' at index 'i'.
---
--- /Pre-release/
-{-# INLINE newIORefWithIndex #-}
-newIORefWithIndex :: forall a. Unbox a => Int -> a -> IO (IORef a)
-newIORefWithIndex i x = do
-    var <- newBytes (sizeOf (Proxy :: Proxy a))
-    pokeByteIndex i var x
-    return $ IORef var i
-
 -- | Create a new 'IORef' starting at index '0'.
 --
 -- /Pre-release/
 {-# INLINE newIORef #-}
 newIORef :: forall a. Unbox a => a -> IO (IORef a)
-newIORef = newIORefWithIndex 0
-
--- | Write a value to an 'IORef' at index 'i'.
---
--- /Pre-release/
-{-# INLINE writeIORefWithIndex #-}
-writeIORefWithIndex :: Unbox a => Int -> IORef a -> a -> IO ()
-writeIORefWithIndex i (IORef var _) = pokeByteIndex i var
+newIORef x = do
+    var <- newBytes (sizeOf (Proxy :: Proxy a))
+    pokeByteIndex 0 var x
+    return $ IORef var 0
 
 -- | Write a value to an 'IORef'.
 --
 -- /Pre-release/
 {-# INLINE writeIORef #-}
 writeIORef :: Unbox a => IORef a -> a -> IO ()
-writeIORef arr@(IORef _ o) = writeIORefWithIndex o arr
-
--- | Read a value from an 'IORef' at index 'i'.
---
--- /Pre-release/
-{-# INLINE readIORefWithIndex #-}
-readIORefWithIndex :: Unbox a => Int -> IORef a -> IO a
-readIORefWithIndex i (IORef var _) = peekByteIndex i var
+writeIORef (IORef var o) = pokeByteIndex o var
 
 -- | Read a value from an 'IORef'.
 --
 -- /Pre-release/
 {-# INLINE readIORef #-}
 readIORef :: Unbox a => IORef a -> IO a
-readIORef arr@(IORef _ o) = readIORefWithIndex o arr
-
--- | Modify the value of an 'IORef' at index 'i' using a function with strict application.
---
--- /Pre-release/
-{-# INLINE modifyIORefWithIndex' #-}
-modifyIORefWithIndex' :: Unbox a => Int -> IORef a -> (a -> a) -> IO ()
-modifyIORefWithIndex' i var g = do
-  x <- readIORefWithIndex i var
-  writeIORefWithIndex i var (g x)
+readIORef (IORef var o) = peekByteIndex o var
 
 -- | Modify the value of an 'IORef' using a function with strict application.
 --
 -- /Pre-release/
 {-# INLINE modifyIORef' #-}
 modifyIORef' :: Unbox a => IORef a -> (a -> a) -> IO ()
-modifyIORef' arr@(IORef _ o) = modifyIORefWithIndex' o arr
-
--- | Generate a stream by continuously reading the IORef at index 'i'.
---
--- /Pre-release/
-{-# INLINE_NORMAL toStreamDWithIndex #-}
-toStreamDWithIndex :: (MonadIO m, Unbox a) => Int -> IORef a -> D.Stream m a
-toStreamDWithIndex i var = D.Stream step ()
-
-    where
-
-    {-# INLINE_LATE step #-}
-    step _ () = liftIO (readIORefWithIndex i var) >>= \x -> return $ D.Yield x ()
+modifyIORef' var g = do
+    x <- readIORef var
+    writeIORef var (g x)
 
 -- | Generate a stream by continuously reading the IORef.
 --