vector-0.10.12.1: Efficient Arrays

Copyright(c) Roman Leshchinskiy 2009-2010
LicenseBSD-style
MaintainerRoman Leshchinskiy <rl@cse.unsw.edu.au>
Stabilityexperimental
Portabilitynon-portable
Safe HaskellNone
LanguageHaskell2010

Data.Vector.Unboxed

Contents

Description

Adaptive unboxed vectors. The implementation is based on type families and picks an efficient, specialised representation for every element type. In particular, unboxed vectors of pairs are represented as pairs of unboxed vectors.

Implementing unboxed vectors for new data types can be very easy. Here is how the library does this for Complex by simply wrapping vectors of pairs.

newtype instance MVector s (Complex a) = MV_Complex (MVector s (a,a))
newtype instance Vector    (Complex a) = V_Complex  (Vector    (a,a))

instance (RealFloat a, Unbox a) => MVector MVector (Complex a) where
  {-# INLINE basicLength #-}
  basicLength (MV_Complex v) = basicLength v
  ...

instance (RealFloat a, Unbox a) => Data.Vector.Generic.Vector Vector (Complex a) where
  {-# INLINE basicLength #-}
  basicLength (V_Complex v) = Data.Vector.Generic.basicLength v
  ...

instance (RealFloat a, Unbox a) => Unbox (Complex a)

Synopsis

Unboxed vectors

data family Vector a Source

Instances

Vector Vector Bool 
Vector Vector Char 
Vector Vector Double 
Vector Vector Float 
Vector Vector Int 
Vector Vector Int8 
Vector Vector Int16 
Vector Vector Int32 
Vector Vector Int64 
Vector Vector Word 
Vector Vector Word8 
Vector Vector Word16 
Vector Vector Word32 
Vector Vector Word64 
Vector Vector () 
(RealFloat a, Unbox a) => Vector Vector (Complex a) 
(Unbox a, Unbox b) => Vector Vector (a, b) 
(Unbox a, Unbox b, Unbox c) => Vector Vector (a, b, c) 
(Unbox a, Unbox b, Unbox c, Unbox d) => Vector Vector (a, b, c, d) 
(Unbox a, Unbox b, Unbox c, Unbox d, Unbox e) => Vector Vector (a, b, c, d, e) 
(Unbox a, Unbox b, Unbox c, Unbox d, Unbox e, Unbox f) => Vector Vector (a, b, c, d, e, f) 
Unbox e => IsList (Vector e) 
(Unbox a, Eq a) => Eq (Vector a) 
(Data a, Unbox a) => Data (Vector a) 
(Unbox a, Ord a) => Ord (Vector a) 
(Read a, Unbox a) => Read (Vector a) 
(Show a, Unbox a) => Show (Vector a) 
Unbox a => Monoid (Vector a) 
NFData (Vector a) 
Typeable (* -> *) Vector 
type Mutable Vector = MVector 
data Vector Bool = V_Bool (Vector Word8) 
data Vector Char = V_Char (Vector Char) 
data Vector Double = V_Double (Vector Double) 
data Vector Float = V_Float (Vector Float) 
data Vector Int = V_Int (Vector Int) 
data Vector Int8 = V_Int8 (Vector Int8) 
data Vector Int16 = V_Int16 (Vector Int16) 
data Vector Int32 = V_Int32 (Vector Int32) 
data Vector Int64 = V_Int64 (Vector Int64) 
data Vector Word = V_Word (Vector Word) 
data Vector Word8 = V_Word8 (Vector Word8) 
data Vector Word16 = V_Word16 (Vector Word16) 
data Vector Word32 = V_Word32 (Vector Word32) 
data Vector Word64 = V_Word64 (Vector Word64) 
data Vector () = V_Unit Int 
type Item (Vector e) = e 
data Vector (Complex a) = V_Complex (Vector (a, a)) 
data Vector (a, b) = V_2 !Int !(Vector a) !(Vector b) 
data Vector (a, b, c) = V_3 !Int !(Vector a) !(Vector b) !(Vector c) 
data Vector (a, b, c, d) = V_4 !Int !(Vector a) !(Vector b) !(Vector c) !(Vector d) 
data Vector (a, b, c, d, e) = V_5 !Int !(Vector a) !(Vector b) !(Vector c) !(Vector d) !(Vector e) 
data Vector (a, b, c, d, e, f) = V_6 !Int !(Vector a) !(Vector b) !(Vector c) !(Vector d) !(Vector e) !(Vector f) 

data family MVector s a Source

Instances

MVector MVector Bool 
MVector MVector Char 
MVector MVector Double 
MVector MVector Float 
MVector MVector Int 
MVector MVector Int8 
MVector MVector Int16 
MVector MVector Int32 
MVector MVector Int64 
MVector MVector Word 
MVector MVector Word8 
MVector MVector Word16 
MVector MVector Word32 
MVector MVector Word64 
MVector MVector () 
(RealFloat a, Unbox a) => MVector MVector (Complex a) 
(Unbox a, Unbox b) => MVector MVector (a, b) 
(Unbox a, Unbox b, Unbox c) => MVector MVector (a, b, c) 
(Unbox a, Unbox b, Unbox c, Unbox d) => MVector MVector (a, b, c, d) 
(Unbox a, Unbox b, Unbox c, Unbox d, Unbox e) => MVector MVector (a, b, c, d, e) 
(Unbox a, Unbox b, Unbox c, Unbox d, Unbox e, Unbox f) => MVector MVector (a, b, c, d, e, f) 
NFData (MVector s a) 
Typeable (* -> * -> *) MVector 
data MVector s Bool = MV_Bool (MVector s Word8) 
data MVector s Char = MV_Char (MVector s Char) 
data MVector s Double = MV_Double (MVector s Double) 
data MVector s Float = MV_Float (MVector s Float) 
data MVector s Word64 = MV_Word64 (MVector s Word64) 
data MVector s Word32 = MV_Word32 (MVector s Word32) 
data MVector s Word16 = MV_Word16 (MVector s Word16) 
data MVector s Word8 = MV_Word8 (MVector s Word8) 
data MVector s Word = MV_Word (MVector s Word) 
data MVector s Int64 = MV_Int64 (MVector s Int64) 
data MVector s Int32 = MV_Int32 (MVector s Int32) 
data MVector s Int16 = MV_Int16 (MVector s Int16) 
data MVector s Int8 = MV_Int8 (MVector s Int8) 
data MVector s Int = MV_Int (MVector s Int) 
data MVector s () = MV_Unit Int 
data MVector s (Complex a) = MV_Complex (MVector s (a, a)) 
data MVector s (a, b) = MV_2 !Int !(MVector s a) !(MVector s b) 
data MVector s (a, b, c) = MV_3 !Int !(MVector s a) !(MVector s b) !(MVector s c) 
data MVector s (a, b, c, d) = MV_4 !Int !(MVector s a) !(MVector s b) !(MVector s c) !(MVector s d) 
data MVector s (a, b, c, d, e) = MV_5 !Int !(MVector s a) !(MVector s b) !(MVector s c) !(MVector s d) !(MVector s e) 
data MVector s (a, b, c, d, e, f) = MV_6 !Int !(MVector s a) !(MVector s b) !(MVector s c) !(MVector s d) !(MVector s e) !(MVector s f) 

class (Vector Vector a, MVector MVector a) => Unbox a Source

Instances

Unbox Bool 
Unbox Char 
Unbox Double 
Unbox Float 
Unbox Int 
Unbox Int8 
Unbox Int16 
Unbox Int32 
Unbox Int64 
Unbox Word 
Unbox Word8 
Unbox Word16 
Unbox Word32 
Unbox Word64 
Unbox () 
(RealFloat a, Unbox a) => Unbox (Complex a) 
(Unbox a, Unbox b) => Unbox (a, b) 
(Unbox a, Unbox b, Unbox c) => Unbox (a, b, c) 
(Unbox a, Unbox b, Unbox c, Unbox d) => Unbox (a, b, c, d) 
(Unbox a, Unbox b, Unbox c, Unbox d, Unbox e) => Unbox (a, b, c, d, e) 
(Unbox a, Unbox b, Unbox c, Unbox d, Unbox e, Unbox f) => Unbox (a, b, c, d, e, f) 

Accessors

Length information

length :: Unbox a => Vector a -> Int Source

O(1) Yield the length of the vector.

null :: Unbox a => Vector a -> Bool Source

O(1) Test whether a vector if empty

Indexing

(!) :: Unbox a => Vector a -> Int -> a Source

O(1) Indexing

(!?) :: Unbox a => Vector a -> Int -> Maybe a Source

O(1) Safe indexing

head :: Unbox a => Vector a -> a Source

O(1) First element

last :: Unbox a => Vector a -> a Source

O(1) Last element

unsafeIndex :: Unbox a => Vector a -> Int -> a Source

O(1) Unsafe indexing without bounds checking

unsafeHead :: Unbox a => Vector a -> a Source

O(1) First element without checking if the vector is empty

unsafeLast :: Unbox a => Vector a -> a Source

O(1) Last element without checking if the vector is empty

Monadic indexing

indexM :: (Unbox a, Monad m) => Vector a -> Int -> m a Source

O(1) Indexing in a monad.

The monad allows operations to be strict in the vector when necessary. Suppose vector copying is implemented like this:

copy mv v = ... write mv i (v ! i) ...

For lazy vectors, v ! i would not be evaluated which means that mv would unnecessarily retain a reference to v in each element written.

With indexM, copying can be implemented like this instead:

copy mv v = ... do
                  x <- indexM v i
                  write mv i x

Here, no references to v are retained because indexing (but not the elements) is evaluated eagerly.

headM :: (Unbox a, Monad m) => Vector a -> m a Source

O(1) First element of a vector in a monad. See indexM for an explanation of why this is useful.

lastM :: (Unbox a, Monad m) => Vector a -> m a Source

O(1) Last element of a vector in a monad. See indexM for an explanation of why this is useful.

unsafeIndexM :: (Unbox a, Monad m) => Vector a -> Int -> m a Source

O(1) Indexing in a monad without bounds checks. See indexM for an explanation of why this is useful.

unsafeHeadM :: (Unbox a, Monad m) => Vector a -> m a Source

O(1) First element in a monad without checking for empty vectors. See indexM for an explanation of why this is useful.

unsafeLastM :: (Unbox a, Monad m) => Vector a -> m a Source

O(1) Last element in a monad without checking for empty vectors. See indexM for an explanation of why this is useful.

Extracting subvectors (slicing)

slice Source

Arguments

:: Unbox a 
=> Int

i starting index

-> Int

n length

-> Vector a 
-> Vector a 

O(1) Yield a slice of the vector without copying it. The vector must contain at least i+n elements.

init :: Unbox a => Vector a -> Vector a Source

O(1) Yield all but the last element without copying. The vector may not be empty.

tail :: Unbox a => Vector a -> Vector a Source

O(1) Yield all but the first element without copying. The vector may not be empty.

take :: Unbox a => Int -> Vector a -> Vector a Source

O(1) Yield at the first n elements without copying. The vector may contain less than n elements in which case it is returned unchanged.

drop :: Unbox a => Int -> Vector a -> Vector a Source

O(1) Yield all but the first n elements without copying. The vector may contain less than n elements in which case an empty vector is returned.

splitAt :: Unbox a => Int -> Vector a -> (Vector a, Vector a) Source

O(1) Yield the first n elements paired with the remainder without copying.

Note that splitAt n v is equivalent to (take n v, drop n v) but slightly more efficient.

unsafeSlice Source

Arguments

:: Unbox a 
=> Int

i starting index

-> Int

n length

-> Vector a 
-> Vector a 

O(1) Yield a slice of the vector without copying. The vector must contain at least i+n elements but this is not checked.

unsafeInit :: Unbox a => Vector a -> Vector a Source

O(1) Yield all but the last element without copying. The vector may not be empty but this is not checked.

unsafeTail :: Unbox a => Vector a -> Vector a Source

O(1) Yield all but the first element without copying. The vector may not be empty but this is not checked.

unsafeTake :: Unbox a => Int -> Vector a -> Vector a Source

O(1) Yield the first n elements without copying. The vector must contain at least n elements but this is not checked.

unsafeDrop :: Unbox a => Int -> Vector a -> Vector a Source

O(1) Yield all but the first n elements without copying. The vector must contain at least n elements but this is not checked.

Construction

Initialisation

empty :: Unbox a => Vector a Source

O(1) Empty vector

singleton :: Unbox a => a -> Vector a Source

O(1) Vector with exactly one element

replicate :: Unbox a => Int -> a -> Vector a Source

O(n) Vector of the given length with the same value in each position

generate :: Unbox a => Int -> (Int -> a) -> Vector a Source

O(n) Construct a vector of the given length by applying the function to each index

iterateN :: Unbox a => Int -> (a -> a) -> a -> Vector a Source

O(n) Apply function n times to value. Zeroth element is original value.

Monadic initialisation

replicateM :: (Monad m, Unbox a) => Int -> m a -> m (Vector a) Source

O(n) Execute the monadic action the given number of times and store the results in a vector.

generateM :: (Monad m, Unbox a) => Int -> (Int -> m a) -> m (Vector a) Source

O(n) Construct a vector of the given length by applying the monadic action to each index

create :: Unbox a => (forall s. ST s (MVector s a)) -> Vector a Source

Execute the monadic action and freeze the resulting vector.

create (do { v <- new 2; write v 0 'a'; write v 1 'b'; return v }) = <a,b>

Unfolding

unfoldr :: Unbox a => (b -> Maybe (a, b)) -> b -> Vector a Source

O(n) Construct a vector by repeatedly applying the generator function to a seed. The generator function yields Just the next element and the new seed or Nothing if there are no more elements.

unfoldr (\n -> if n == 0 then Nothing else Just (n,n-1)) 10
 = <10,9,8,7,6,5,4,3,2,1>

unfoldrN :: Unbox a => Int -> (b -> Maybe (a, b)) -> b -> Vector a Source

O(n) Construct a vector with at most n by repeatedly applying the generator function to the a seed. The generator function yields Just the next element and the new seed or Nothing if there are no more elements.

unfoldrN 3 (\n -> Just (n,n-1)) 10 = <10,9,8>

constructN :: Unbox a => Int -> (Vector a -> a) -> Vector a Source

O(n) Construct a vector with n elements by repeatedly applying the generator function to the already constructed part of the vector.

constructN 3 f = let a = f <> ; b = f <a> ; c = f <a,b> in f <a,b,c>

constructrN :: Unbox a => Int -> (Vector a -> a) -> Vector a Source

O(n) Construct a vector with n elements from right to left by repeatedly applying the generator function to the already constructed part of the vector.

constructrN 3 f = let a = f <> ; b = f<a> ; c = f <b,a> in f <c,b,a>

Enumeration

enumFromN :: (Unbox a, Num a) => a -> Int -> Vector a Source

O(n) Yield a vector of the given length containing the values x, x+1 etc. This operation is usually more efficient than enumFromTo.

enumFromN 5 3 = <5,6,7>

enumFromStepN :: (Unbox a, Num a) => a -> a -> Int -> Vector a Source

O(n) Yield a vector of the given length containing the values x, x+y, x+y+y etc. This operations is usually more efficient than enumFromThenTo.

enumFromStepN 1 0.1 5 = <1,1.1,1.2,1.3,1.4>

enumFromTo :: (Unbox a, Enum a) => a -> a -> Vector a Source

O(n) Enumerate values from x to y.

WARNING: This operation can be very inefficient. If at all possible, use enumFromN instead.

enumFromThenTo :: (Unbox a, Enum a) => a -> a -> a -> Vector a Source

O(n) Enumerate values from x to y with a specific step z.

WARNING: This operation can be very inefficient. If at all possible, use enumFromStepN instead.

Concatenation

cons :: Unbox a => a -> Vector a -> Vector a Source

O(n) Prepend an element

snoc :: Unbox a => Vector a -> a -> Vector a Source

O(n) Append an element

(++) :: Unbox a => Vector a -> Vector a -> Vector a infixr 5 Source

O(m+n) Concatenate two vectors

concat :: Unbox a => [Vector a] -> Vector a Source

O(n) Concatenate all vectors in the list

Restricting memory usage

force :: Unbox a => Vector a -> Vector a Source

O(n) Yield the argument but force it not to retain any extra memory, possibly by copying it.

This is especially useful when dealing with slices. For example:

force (slice 0 2 <huge vector>)

Here, the slice retains a reference to the huge vector. Forcing it creates a copy of just the elements that belong to the slice and allows the huge vector to be garbage collected.

Modifying vectors

Bulk updates

(//) Source

Arguments

:: Unbox a 
=> Vector a

initial vector (of length m)

-> [(Int, a)]

list of index/value pairs (of length n)

-> Vector a 

O(m+n) For each pair (i,a) from the list, replace the vector element at position i by a.

<5,9,2,7> // [(2,1),(0,3),(2,8)] = <3,9,8,7>

update Source

Arguments

:: Unbox a 
=> Vector a

initial vector (of length m)

-> Vector (Int, a)

vector of index/value pairs (of length n)

-> Vector a 

O(m+n) For each pair (i,a) from the vector of index/value pairs, replace the vector element at position i by a.

update <5,9,2,7> <(2,1),(0,3),(2,8)> = <3,9,8,7>

update_ Source

Arguments

:: Unbox a 
=> Vector a

initial vector (of length m)

-> Vector Int

index vector (of length n1)

-> Vector a

value vector (of length n2)

-> Vector a 

O(m+min(n1,n2)) For each index i from the index vector and the corresponding value a from the value vector, replace the element of the initial vector at position i by a.

update_ <5,9,2,7>  <2,0,2> <1,3,8> = <3,9,8,7>

The function update provides the same functionality and is usually more convenient.

update_ xs is ys = update xs (zip is ys)

unsafeUpd :: Unbox a => Vector a -> [(Int, a)] -> Vector a Source

Same as (//) but without bounds checking.

unsafeUpdate :: Unbox a => Vector a -> Vector (Int, a) -> Vector a Source

Same as update but without bounds checking.

unsafeUpdate_ :: Unbox a => Vector a -> Vector Int -> Vector a -> Vector a Source

Same as update_ but without bounds checking.

Accumulations

accum Source

Arguments

:: Unbox a 
=> (a -> b -> a)

accumulating function f

-> Vector a

initial vector (of length m)

-> [(Int, b)]

list of index/value pairs (of length n)

-> Vector a 

O(m+n) For each pair (i,b) from the list, replace the vector element a at position i by f a b.

accum (+) <5,9,2> [(2,4),(1,6),(0,3),(1,7)] = <5+3, 9+6+7, 2+4>

accumulate Source

Arguments

:: (Unbox a, Unbox b) 
=> (a -> b -> a)

accumulating function f

-> Vector a

initial vector (of length m)

-> Vector (Int, b)

vector of index/value pairs (of length n)

-> Vector a 

O(m+n) For each pair (i,b) from the vector of pairs, replace the vector element a at position i by f a b.

accumulate (+) <5,9,2> <(2,4),(1,6),(0,3),(1,7)> = <5+3, 9+6+7, 2+4>

accumulate_ Source

Arguments

:: (Unbox a, Unbox b) 
=> (a -> b -> a)

accumulating function f

-> Vector a

initial vector (of length m)

-> Vector Int

index vector (of length n1)

-> Vector b

value vector (of length n2)

-> Vector a 

O(m+min(n1,n2)) For each index i from the index vector and the corresponding value b from the the value vector, replace the element of the initial vector at position i by f a b.

accumulate_ (+) <5,9,2> <2,1,0,1> <4,6,3,7> = <5+3, 9+6+7, 2+4>

The function accumulate provides the same functionality and is usually more convenient.

accumulate_ f as is bs = accumulate f as (zip is bs)

unsafeAccum :: Unbox a => (a -> b -> a) -> Vector a -> [(Int, b)] -> Vector a Source

Same as accum but without bounds checking.

unsafeAccumulate :: (Unbox a, Unbox b) => (a -> b -> a) -> Vector a -> Vector (Int, b) -> Vector a Source

Same as accumulate but without bounds checking.

unsafeAccumulate_ :: (Unbox a, Unbox b) => (a -> b -> a) -> Vector a -> Vector Int -> Vector b -> Vector a Source

Same as accumulate_ but without bounds checking.

Permutations

reverse :: Unbox a => Vector a -> Vector a Source

O(n) Reverse a vector

backpermute :: Unbox a => Vector a -> Vector Int -> Vector a Source

O(n) Yield the vector obtained by replacing each element i of the index vector by xs!i. This is equivalent to map (xs!) is but is often much more efficient.

backpermute <a,b,c,d> <0,3,2,3,1,0> = <a,d,c,d,b,a>

unsafeBackpermute :: Unbox a => Vector a -> Vector Int -> Vector a Source

Same as backpermute but without bounds checking.

Safe destructive updates

modify :: Unbox a => (forall s. MVector s a -> ST s ()) -> Vector a -> Vector a Source

Apply a destructive operation to a vector. The operation will be performed in place if it is safe to do so and will modify a copy of the vector otherwise.

modify (\v -> write v 0 'x') (replicate 3 'a') = <'x','a','a'>

Elementwise operations

Indexing

indexed :: Unbox a => Vector a -> Vector (Int, a) Source

O(n) Pair each element in a vector with its index

Mapping

map :: (Unbox a, Unbox b) => (a -> b) -> Vector a -> Vector b Source

O(n) Map a function over a vector

imap :: (Unbox a, Unbox b) => (Int -> a -> b) -> Vector a -> Vector b Source

O(n) Apply a function to every element of a vector and its index

concatMap :: (Unbox a, Unbox b) => (a -> Vector b) -> Vector a -> Vector b Source

Map a function over a vector and concatenate the results.

Monadic mapping

mapM :: (Monad m, Unbox a, Unbox b) => (a -> m b) -> Vector a -> m (Vector b) Source

O(n) Apply the monadic action to all elements of the vector, yielding a vector of results

mapM_ :: (Monad m, Unbox a) => (a -> m b) -> Vector a -> m () Source

O(n) Apply the monadic action to all elements of a vector and ignore the results

forM :: (Monad m, Unbox a, Unbox b) => Vector a -> (a -> m b) -> m (Vector b) Source

O(n) Apply the monadic action to all elements of the vector, yielding a vector of results. Equvalent to flip mapM.

forM_ :: (Monad m, Unbox a) => Vector a -> (a -> m b) -> m () Source

O(n) Apply the monadic action to all elements of a vector and ignore the results. Equivalent to flip mapM_.

Zipping

zipWith :: (Unbox a, Unbox b, Unbox c) => (a -> b -> c) -> Vector a -> Vector b -> Vector c Source

O(min(m,n)) Zip two vectors with the given function.

zipWith3 :: (Unbox a, Unbox b, Unbox c, Unbox d) => (a -> b -> c -> d) -> Vector a -> Vector b -> Vector c -> Vector d Source

Zip three vectors with the given function.

zipWith4 :: (Unbox a, Unbox b, Unbox c, Unbox d, Unbox e) => (a -> b -> c -> d -> e) -> Vector a -> Vector b -> Vector c -> Vector d -> Vector e Source

zipWith5 :: (Unbox a, Unbox b, Unbox c, Unbox d, Unbox e, Unbox f) => (a -> b -> c -> d -> e -> f) -> Vector a -> Vector b -> Vector c -> Vector d -> Vector e -> Vector f Source

zipWith6 :: (Unbox a, Unbox b, Unbox c, Unbox d, Unbox e, Unbox f, Unbox g) => (a -> b -> c -> d -> e -> f -> g) -> Vector a -> Vector b -> Vector c -> Vector d -> Vector e -> Vector f -> Vector g Source

izipWith :: (Unbox a, Unbox b, Unbox c) => (Int -> a -> b -> c) -> Vector a -> Vector b -> Vector c Source

O(min(m,n)) Zip two vectors with a function that also takes the elements' indices.

izipWith3 :: (Unbox a, Unbox b, Unbox c, Unbox d) => (Int -> a -> b -> c -> d) -> Vector a -> Vector b -> Vector c -> Vector d Source

Zip three vectors and their indices with the given function.

izipWith4 :: (Unbox a, Unbox b, Unbox c, Unbox d, Unbox e) => (Int -> a -> b -> c -> d -> e) -> Vector a -> Vector b -> Vector c -> Vector d -> Vector e Source

izipWith5 :: (Unbox a, Unbox b, Unbox c, Unbox d, Unbox e, Unbox f) => (Int -> a -> b -> c -> d -> e -> f) -> Vector a -> Vector b -> Vector c -> Vector d -> Vector e -> Vector f Source

izipWith6 :: (Unbox a, Unbox b, Unbox c, Unbox d, Unbox e, Unbox f, Unbox g) => (Int -> a -> b -> c -> d -> e -> f -> g) -> Vector a -> Vector b -> Vector c -> Vector d -> Vector e -> Vector f -> Vector g Source

zip :: (Unbox a, Unbox b) => Vector a -> Vector b -> Vector (a, b) Source

O(1) Zip 2 vectors

zip3 :: (Unbox a, Unbox b, Unbox c) => Vector a -> Vector b -> Vector c -> Vector (a, b, c) Source

O(1) Zip 3 vectors

zip4 :: (Unbox a, Unbox b, Unbox c, Unbox d) => Vector a -> Vector b -> Vector c -> Vector d -> Vector (a, b, c, d) Source

O(1) Zip 4 vectors

zip5 :: (Unbox a, Unbox b, Unbox c, Unbox d, Unbox e) => Vector a -> Vector b -> Vector c -> Vector d -> Vector e -> Vector (a, b, c, d, e) Source

O(1) Zip 5 vectors

zip6 :: (Unbox a, Unbox b, Unbox c, Unbox d, Unbox e, Unbox f) => Vector a -> Vector b -> Vector c -> Vector d -> Vector e -> Vector f -> Vector (a, b, c, d, e, f) Source

O(1) Zip 6 vectors

Monadic zipping

zipWithM :: (Monad m, Unbox a, Unbox b, Unbox c) => (a -> b -> m c) -> Vector a -> Vector b -> m (Vector c) Source

O(min(m,n)) Zip the two vectors with the monadic action and yield a vector of results

zipWithM_ :: (Monad m, Unbox a, Unbox b) => (a -> b -> m c) -> Vector a -> Vector b -> m () Source

O(min(m,n)) Zip the two vectors with the monadic action and ignore the results

Unzipping

unzip :: (Unbox a, Unbox b) => Vector (a, b) -> (Vector a, Vector b) Source

O(1) Unzip 2 vectors

unzip3 :: (Unbox a, Unbox b, Unbox c) => Vector (a, b, c) -> (Vector a, Vector b, Vector c) Source

O(1) Unzip 3 vectors

unzip4 :: (Unbox a, Unbox b, Unbox c, Unbox d) => Vector (a, b, c, d) -> (Vector a, Vector b, Vector c, Vector d) Source

O(1) Unzip 4 vectors

unzip5 :: (Unbox a, Unbox b, Unbox c, Unbox d, Unbox e) => Vector (a, b, c, d, e) -> (Vector a, Vector b, Vector c, Vector d, Vector e) Source

O(1) Unzip 5 vectors

unzip6 :: (Unbox a, Unbox b, Unbox c, Unbox d, Unbox e, Unbox f) => Vector (a, b, c, d, e, f) -> (Vector a, Vector b, Vector c, Vector d, Vector e, Vector f) Source

O(1) Unzip 6 vectors

Working with predicates

Filtering

filter :: Unbox a => (a -> Bool) -> Vector a -> Vector a Source

O(n) Drop elements that do not satisfy the predicate

ifilter :: Unbox a => (Int -> a -> Bool) -> Vector a -> Vector a Source

O(n) Drop elements that do not satisfy the predicate which is applied to values and their indices

filterM :: (Monad m, Unbox a) => (a -> m Bool) -> Vector a -> m (Vector a) Source

O(n) Drop elements that do not satisfy the monadic predicate

takeWhile :: Unbox a => (a -> Bool) -> Vector a -> Vector a Source

O(n) Yield the longest prefix of elements satisfying the predicate without copying.

dropWhile :: Unbox a => (a -> Bool) -> Vector a -> Vector a Source

O(n) Drop the longest prefix of elements that satisfy the predicate without copying.

Partitioning

partition :: Unbox a => (a -> Bool) -> Vector a -> (Vector a, Vector a) Source

O(n) Split the vector in two parts, the first one containing those elements that satisfy the predicate and the second one those that don't. The relative order of the elements is preserved at the cost of a sometimes reduced performance compared to unstablePartition.

unstablePartition :: Unbox a => (a -> Bool) -> Vector a -> (Vector a, Vector a) Source

O(n) Split the vector in two parts, the first one containing those elements that satisfy the predicate and the second one those that don't. The order of the elements is not preserved but the operation is often faster than partition.

span :: Unbox a => (a -> Bool) -> Vector a -> (Vector a, Vector a) Source

O(n) Split the vector into the longest prefix of elements that satisfy the predicate and the rest without copying.

break :: Unbox a => (a -> Bool) -> Vector a -> (Vector a, Vector a) Source

O(n) Split the vector into the longest prefix of elements that do not satisfy the predicate and the rest without copying.

Searching

elem :: (Unbox a, Eq a) => a -> Vector a -> Bool infix 4 Source

O(n) Check if the vector contains an element

notElem :: (Unbox a, Eq a) => a -> Vector a -> Bool infix 4 Source

O(n) Check if the vector does not contain an element (inverse of elem)

find :: Unbox a => (a -> Bool) -> Vector a -> Maybe a Source

O(n) Yield Just the first element matching the predicate or Nothing if no such element exists.

findIndex :: Unbox a => (a -> Bool) -> Vector a -> Maybe Int Source

O(n) Yield Just the index of the first element matching the predicate or Nothing if no such element exists.

findIndices :: Unbox a => (a -> Bool) -> Vector a -> Vector Int Source

O(n) Yield the indices of elements satisfying the predicate in ascending order.

elemIndex :: (Unbox a, Eq a) => a -> Vector a -> Maybe Int Source

O(n) Yield Just the index of the first occurence of the given element or Nothing if the vector does not contain the element. This is a specialised version of findIndex.

elemIndices :: (Unbox a, Eq a) => a -> Vector a -> Vector Int Source

O(n) Yield the indices of all occurences of the given element in ascending order. This is a specialised version of findIndices.

Folding

foldl :: Unbox b => (a -> b -> a) -> a -> Vector b -> a Source

O(n) Left fold

foldl1 :: Unbox a => (a -> a -> a) -> Vector a -> a Source

O(n) Left fold on non-empty vectors

foldl' :: Unbox b => (a -> b -> a) -> a -> Vector b -> a Source

O(n) Left fold with strict accumulator

foldl1' :: Unbox a => (a -> a -> a) -> Vector a -> a Source

O(n) Left fold on non-empty vectors with strict accumulator

foldr :: Unbox a => (a -> b -> b) -> b -> Vector a -> b Source

O(n) Right fold

foldr1 :: Unbox a => (a -> a -> a) -> Vector a -> a Source

O(n) Right fold on non-empty vectors

foldr' :: Unbox a => (a -> b -> b) -> b -> Vector a -> b Source

O(n) Right fold with a strict accumulator

foldr1' :: Unbox a => (a -> a -> a) -> Vector a -> a Source

O(n) Right fold on non-empty vectors with strict accumulator

ifoldl :: Unbox b => (a -> Int -> b -> a) -> a -> Vector b -> a Source

O(n) Left fold (function applied to each element and its index)

ifoldl' :: Unbox b => (a -> Int -> b -> a) -> a -> Vector b -> a Source

O(n) Left fold with strict accumulator (function applied to each element and its index)

ifoldr :: Unbox a => (Int -> a -> b -> b) -> b -> Vector a -> b Source

O(n) Right fold (function applied to each element and its index)

ifoldr' :: Unbox a => (Int -> a -> b -> b) -> b -> Vector a -> b Source

O(n) Right fold with strict accumulator (function applied to each element and its index)

Specialised folds

all :: Unbox a => (a -> Bool) -> Vector a -> Bool Source

O(n) Check if all elements satisfy the predicate.

any :: Unbox a => (a -> Bool) -> Vector a -> Bool Source

O(n) Check if any element satisfies the predicate.

and :: Vector Bool -> Bool Source

O(n) Check if all elements are True

or :: Vector Bool -> Bool Source

O(n) Check if any element is True

sum :: (Unbox a, Num a) => Vector a -> a Source

O(n) Compute the sum of the elements

product :: (Unbox a, Num a) => Vector a -> a Source

O(n) Compute the produce of the elements

maximum :: (Unbox a, Ord a) => Vector a -> a Source

O(n) Yield the maximum element of the vector. The vector may not be empty.

maximumBy :: Unbox a => (a -> a -> Ordering) -> Vector a -> a Source

O(n) Yield the maximum element of the vector according to the given comparison function. The vector may not be empty.

minimum :: (Unbox a, Ord a) => Vector a -> a Source

O(n) Yield the minimum element of the vector. The vector may not be empty.

minimumBy :: Unbox a => (a -> a -> Ordering) -> Vector a -> a Source

O(n) Yield the minimum element of the vector according to the given comparison function. The vector may not be empty.

minIndex :: (Unbox a, Ord a) => Vector a -> Int Source

O(n) Yield the index of the minimum element of the vector. The vector may not be empty.

minIndexBy :: Unbox a => (a -> a -> Ordering) -> Vector a -> Int Source

O(n) Yield the index of the minimum element of the vector according to the given comparison function. The vector may not be empty.

maxIndex :: (Unbox a, Ord a) => Vector a -> Int Source

O(n) Yield the index of the maximum element of the vector. The vector may not be empty.

maxIndexBy :: Unbox a => (a -> a -> Ordering) -> Vector a -> Int Source

O(n) Yield the index of the maximum element of the vector according to the given comparison function. The vector may not be empty.

Monadic folds

foldM :: (Monad m, Unbox b) => (a -> b -> m a) -> a -> Vector b -> m a Source

O(n) Monadic fold

foldM' :: (Monad m, Unbox b) => (a -> b -> m a) -> a -> Vector b -> m a Source

O(n) Monadic fold with strict accumulator

fold1M :: (Monad m, Unbox a) => (a -> a -> m a) -> Vector a -> m a Source

O(n) Monadic fold over non-empty vectors

fold1M' :: (Monad m, Unbox a) => (a -> a -> m a) -> Vector a -> m a Source

O(n) Monadic fold over non-empty vectors with strict accumulator

foldM_ :: (Monad m, Unbox b) => (a -> b -> m a) -> a -> Vector b -> m () Source

O(n) Monadic fold that discards the result

foldM'_ :: (Monad m, Unbox b) => (a -> b -> m a) -> a -> Vector b -> m () Source

O(n) Monadic fold with strict accumulator that discards the result

fold1M_ :: (Monad m, Unbox a) => (a -> a -> m a) -> Vector a -> m () Source

O(n) Monadic fold over non-empty vectors that discards the result

fold1M'_ :: (Monad m, Unbox a) => (a -> a -> m a) -> Vector a -> m () Source

O(n) Monadic fold over non-empty vectors with strict accumulator that discards the result

Prefix sums (scans)

prescanl :: (Unbox a, Unbox b) => (a -> b -> a) -> a -> Vector b -> Vector a Source

O(n) Prescan

prescanl f z = init . scanl f z

Example: prescanl (+) 0 <1,2,3,4> = <0,1,3,6>

prescanl' :: (Unbox a, Unbox b) => (a -> b -> a) -> a -> Vector b -> Vector a Source

O(n) Prescan with strict accumulator

postscanl :: (Unbox a, Unbox b) => (a -> b -> a) -> a -> Vector b -> Vector a Source

O(n) Scan

postscanl f z = tail . scanl f z

Example: postscanl (+) 0 <1,2,3,4> = <1,3,6,10>

postscanl' :: (Unbox a, Unbox b) => (a -> b -> a) -> a -> Vector b -> Vector a Source

O(n) Scan with strict accumulator

scanl :: (Unbox a, Unbox b) => (a -> b -> a) -> a -> Vector b -> Vector a Source

O(n) Haskell-style scan

scanl f z <x1,...,xn> = <y1,...,y(n+1)>
  where y1 = z
        yi = f y(i-1) x(i-1)

Example: scanl (+) 0 <1,2,3,4> = <0,1,3,6,10>

scanl' :: (Unbox a, Unbox b) => (a -> b -> a) -> a -> Vector b -> Vector a Source

O(n) Haskell-style scan with strict accumulator

scanl1 :: Unbox a => (a -> a -> a) -> Vector a -> Vector a Source

O(n) Scan over a non-empty vector

scanl f <x1,...,xn> = <y1,...,yn>
  where y1 = x1
        yi = f y(i-1) xi

scanl1' :: Unbox a => (a -> a -> a) -> Vector a -> Vector a Source

O(n) Scan over a non-empty vector with a strict accumulator

prescanr :: (Unbox a, Unbox b) => (a -> b -> b) -> b -> Vector a -> Vector b Source

O(n) Right-to-left prescan

prescanr f z = reverse . prescanl (flip f) z . reverse

prescanr' :: (Unbox a, Unbox b) => (a -> b -> b) -> b -> Vector a -> Vector b Source

O(n) Right-to-left prescan with strict accumulator

postscanr :: (Unbox a, Unbox b) => (a -> b -> b) -> b -> Vector a -> Vector b Source

O(n) Right-to-left scan

postscanr' :: (Unbox a, Unbox b) => (a -> b -> b) -> b -> Vector a -> Vector b Source

O(n) Right-to-left scan with strict accumulator

scanr :: (Unbox a, Unbox b) => (a -> b -> b) -> b -> Vector a -> Vector b Source

O(n) Right-to-left Haskell-style scan

scanr' :: (Unbox a, Unbox b) => (a -> b -> b) -> b -> Vector a -> Vector b Source

O(n) Right-to-left Haskell-style scan with strict accumulator

scanr1 :: Unbox a => (a -> a -> a) -> Vector a -> Vector a Source

O(n) Right-to-left scan over a non-empty vector

scanr1' :: Unbox a => (a -> a -> a) -> Vector a -> Vector a Source

O(n) Right-to-left scan over a non-empty vector with a strict accumulator

Conversions

Lists

toList :: Unbox a => Vector a -> [a] Source

O(n) Convert a vector to a list

fromList :: Unbox a => [a] -> Vector a Source

O(n) Convert a list to a vector

fromListN :: Unbox a => Int -> [a] -> Vector a Source

O(n) Convert the first n elements of a list to a vector

fromListN n xs = fromList (take n xs)

Other vector types

convert :: (Vector v a, Vector w a) => v a -> w a Source

O(n) Convert different vector types

Mutable vectors

freeze :: (Unbox a, PrimMonad m) => MVector (PrimState m) a -> m (Vector a) Source

O(n) Yield an immutable copy of the mutable vector.

thaw :: (Unbox a, PrimMonad m) => Vector a -> m (MVector (PrimState m) a) Source

O(n) Yield a mutable copy of the immutable vector.

copy :: (Unbox a, PrimMonad m) => MVector (PrimState m) a -> Vector a -> m () Source

O(n) Copy an immutable vector into a mutable one. The two vectors must have the same length.

unsafeFreeze :: (Unbox a, PrimMonad m) => MVector (PrimState m) a -> m (Vector a) Source

O(1) Unsafe convert a mutable vector to an immutable one without copying. The mutable vector may not be used after this operation.

unsafeThaw :: (Unbox a, PrimMonad m) => Vector a -> m (MVector (PrimState m) a) Source

O(1) Unsafely convert an immutable vector to a mutable one without copying. The immutable vector may not be used after this operation.

unsafeCopy :: (Unbox a, PrimMonad m) => MVector (PrimState m) a -> Vector a -> m () Source

O(n) Copy an immutable vector into a mutable one. The two vectors must have the same length. This is not checked.