-- |
-- Extensions to the standard mutable Vector API.
module VectorBuilder.MVector
where

import VectorBuilder.Prelude
import Data.Vector.Generic.Mutable
import qualified VectorBuilder.Core.Builder as A
import qualified VectorBuilder.Core.Update as C


-- |
-- Construct a mutable vector from a builder.
-- 
-- Supports all kinds of vectors: boxed, unboxed, primitive, storable.
{-# INLINABLE build #-}
build :: MVector vector element => A.Builder element -> ST s (vector s element)
build :: Builder element -> ST s (vector s element)
build (A.Builder Int
size (C.Update forall s (vector :: * -> * -> *).
MVector vector element =>
vector s element -> Int -> ST s ()
update)) =
  do
    vector s element
vector <- Int -> ST s (vector (PrimState (ST s)) element)
forall (m :: * -> *) (v :: * -> * -> *) a.
(PrimMonad m, MVector v a) =>
Int -> m (v (PrimState m) a)
unsafeNew Int
size
    vector s element -> Int -> ST s ()
forall s (vector :: * -> * -> *).
MVector vector element =>
vector s element -> Int -> ST s ()
update vector s element
vector Int
0
    vector s element -> ST s (vector s element)
forall (m :: * -> *) a. Monad m => a -> m a
return vector s element
vector