Safe Haskell | None |
---|---|

Language | Haskell98 |

A time and space-efficient implementation of vectors using
packed arrays, suitable for high performance use, both in terms
of large data quantities, or high speed requirements. Vectors
are encoded as strict arrays, held in a `ForeignPtr`

,
and can be passed between C and Haskell with little effort.

This module is intended to be imported `qualified`

, to avoid name
clashes with Prelude functions. eg.

import qualified Data.StorableVector as V

Original GHC implementation by Bryan O'Sullivan. Rewritten to use UArray by Simon Marlow. Rewritten to support slices and use ForeignPtr by David Roundy. Polished and extended by Don Stewart. Generalized to any Storable value by Spencer Janssen. Chunky lazy stream, also with chunk pattern control, mutable access in ST monad, Builder monoid by Henning Thieleman.

- data Vector a
- empty :: Storable a => Vector a
- singleton :: Storable a => a -> Vector a
- pack :: Storable a => [a] -> Vector a
- unpack :: Storable a => Vector a -> [a]
- packN :: Storable a => Int -> [a] -> (Vector a, [a])
- packWith :: Storable b => (a -> b) -> [a] -> Vector b
- unpackWith :: Storable a => (a -> b) -> Vector a -> [b]
- cons :: Storable a => a -> Vector a -> Vector a
- snoc :: Storable a => Vector a -> a -> Vector a
- append :: Storable a => Vector a -> Vector a -> Vector a
- head :: Storable a => Vector a -> a
- last :: Storable a => Vector a -> a
- tail :: Storable a => Vector a -> Vector a
- init :: Vector a -> Vector a
- null :: Vector a -> Bool
- length :: Vector a -> Int
- viewL :: Storable a => Vector a -> Maybe (a, Vector a)
- viewR :: Storable a => Vector a -> Maybe (Vector a, a)
- switchL :: Storable a => b -> (a -> Vector a -> b) -> Vector a -> b
- switchR :: Storable a => b -> (Vector a -> a -> b) -> Vector a -> b
- map :: (Storable a, Storable b) => (a -> b) -> Vector a -> Vector b
- mapIndexed :: (Storable a, Storable b) => (Int -> a -> b) -> Vector a -> Vector b
- reverse :: Storable a => Vector a -> Vector a
- intersperse :: Storable a => a -> Vector a -> Vector a
- transpose :: Storable a => [Vector a] -> [Vector a]
- foldl :: Storable a => (b -> a -> b) -> b -> Vector a -> b
- foldl' :: Storable a => (b -> a -> b) -> b -> Vector a -> b
- foldl1 :: Storable a => (a -> a -> a) -> Vector a -> a
- foldl1' :: Storable a => (a -> a -> a) -> Vector a -> a
- foldr :: Storable a => (a -> b -> b) -> b -> Vector a -> b
- foldr1 :: Storable a => (a -> a -> a) -> Vector a -> a
- concat :: Storable a => [Vector a] -> Vector a
- concatMap :: (Storable a, Storable b) => (a -> Vector b) -> Vector a -> Vector b
- foldMap :: (Storable a, Monoid m) => (a -> m) -> Vector a -> m
- monoidConcatMap :: (Storable a, Monoid m) => (a -> m) -> Vector a -> m
- any :: Storable a => (a -> Bool) -> Vector a -> Bool
- all :: Storable a => (a -> Bool) -> Vector a -> Bool
- maximum :: (Storable a, Ord a) => Vector a -> a
- minimum :: (Storable a, Ord a) => Vector a -> a
- scanl :: (Storable a, Storable b) => (a -> b -> a) -> a -> Vector b -> Vector a
- scanl1 :: Storable a => (a -> a -> a) -> Vector a -> Vector a
- scanr :: (Storable a, Storable b) => (a -> b -> b) -> b -> Vector a -> Vector b
- scanr1 :: Storable a => (a -> a -> a) -> Vector a -> Vector a
- mapAccumL :: (Storable a, Storable b) => (acc -> a -> (acc, b)) -> acc -> Vector a -> (acc, Vector b)
- mapAccumR :: (Storable a, Storable b) => (acc -> a -> (acc, b)) -> acc -> Vector a -> (acc, Vector b)
- crochetL :: (Storable x, Storable y) => (x -> acc -> Maybe (y, acc)) -> acc -> Vector x -> Vector y
- crochetLResult :: (Storable x, Storable y) => (x -> acc -> Maybe (y, acc)) -> acc -> Vector x -> (Vector y, Maybe acc)
- replicate :: Storable a => Int -> a -> Vector a
- iterateN :: Storable a => Int -> (a -> a) -> a -> Vector a
- unfoldr :: Storable b => (a -> Maybe (b, a)) -> a -> Vector b
- unfoldrN :: Storable b => Int -> (a -> Maybe (b, a)) -> a -> (Vector b, Maybe a)
- unfoldrResultN :: Storable b => Int -> (a -> c) -> (a -> Either c (b, a)) -> a -> (Vector b, c)
- sample :: Storable a => Int -> (Int -> a) -> Vector a
- take :: Storable a => Int -> Vector a -> Vector a
- drop :: Storable a => Int -> Vector a -> Vector a
- splitAt :: Storable a => Int -> Vector a -> (Vector a, Vector a)
- takeWhile :: Storable a => (a -> Bool) -> Vector a -> Vector a
- dropWhile :: Storable a => (a -> Bool) -> Vector a -> Vector a
- span :: Storable a => (a -> Bool) -> Vector a -> (Vector a, Vector a)
- spanEnd :: Storable a => (a -> Bool) -> Vector a -> (Vector a, Vector a)
- break :: Storable a => (a -> Bool) -> Vector a -> (Vector a, Vector a)
- breakEnd :: Storable a => (a -> Bool) -> Vector a -> (Vector a, Vector a)
- group :: (Storable a, Eq a) => Vector a -> [Vector a]
- groupBy :: Storable a => (a -> a -> Bool) -> Vector a -> [Vector a]
- inits :: Storable a => Vector a -> [Vector a]
- tails :: Storable a => Vector a -> [Vector a]
- split :: (Storable a, Eq a) => a -> Vector a -> [Vector a]
- splitWith :: Storable a => (a -> Bool) -> Vector a -> [Vector a]
- tokens :: Storable a => (a -> Bool) -> Vector a -> [Vector a]
- sliceVertical :: Storable a => Int -> Vector a -> [Vector a]
- join :: Storable a => Vector a -> [Vector a] -> Vector a
- isPrefixOf :: (Storable a, Eq a) => Vector a -> Vector a -> Bool
- isSuffixOf :: (Storable a, Eq a) => Vector a -> Vector a -> Bool
- elem :: (Storable a, Eq a) => a -> Vector a -> Bool
- notElem :: (Storable a, Eq a) => a -> Vector a -> Bool
- find :: Storable a => (a -> Bool) -> Vector a -> Maybe a
- filter :: Storable a => (a -> Bool) -> Vector a -> Vector a
- index :: Storable a => Vector a -> Int -> a
- elemIndex :: (Storable a, Eq a) => a -> Vector a -> Maybe Int
- elemIndices :: (Storable a, Eq a) => a -> Vector a -> [Int]
- elemIndexEnd :: (Storable a, Eq a) => a -> Vector a -> Maybe Int
- findIndex :: Storable a => (a -> Bool) -> Vector a -> Maybe Int
- findIndices :: Storable a => (a -> Bool) -> Vector a -> [Int]
- count :: (Storable a, Eq a) => a -> Vector a -> Int
- findIndexOrEnd :: Storable a => (a -> Bool) -> Vector a -> Int
- zip :: (Storable a, Storable b) => Vector a -> Vector b -> [(a, b)]
- zipWith :: (Storable a, Storable b, Storable c) => (a -> b -> c) -> Vector a -> Vector b -> Vector c
- zipWith3 :: (Storable a, Storable b, Storable c, Storable d) => (a -> b -> c -> d) -> Vector a -> Vector b -> Vector c -> Vector d
- zipWith4 :: (Storable a, Storable b, Storable c, Storable d, Storable e) => (a -> b -> c -> d -> e) -> Vector a -> Vector b -> Vector c -> Vector d -> Vector e
- unzip :: (Storable a, Storable b) => [(a, b)] -> (Vector a, Vector b)
- copy :: Storable a => Vector a -> Vector a
- sieve :: Storable a => Int -> Vector a -> Vector a
- deinterleave :: Storable a => Int -> Vector a -> [Vector a]
- interleave :: Storable a => [Vector a] -> Vector a
- poke :: Storable a => Ptr a -> Vector a -> IO ()
- peek :: Storable a => Int -> Ptr a -> IO (Vector a)
- hGet :: Storable a => Handle -> Int -> IO (Vector a)
- hPut :: Storable a => Handle -> Vector a -> IO ()
- readFile :: Storable a => FilePath -> IO (Vector a)
- writeFile :: Storable a => FilePath -> Vector a -> IO ()
- appendFile :: Storable a => FilePath -> Vector a -> IO ()

# The `Vector`

type

A space-efficient representation of a vector, supporting many efficient operations.

Instances of Eq, Ord, Read, Show, Data, Typeable

# Introducing and eliminating `Vector`

s

singleton :: Storable a => a -> Vector a Source #

*O(1)* Construct a `Vector`

containing a single element

packN :: Storable a => Int -> [a] -> (Vector a, [a]) Source #

*O(n)* Convert first `n`

elements of a '[a]' into a 'Vector a'.

packWith :: Storable b => (a -> b) -> [a] -> Vector b Source #

*O(n)* Convert a list into a `Vector`

using a conversion function

unpackWith :: Storable a => (a -> b) -> Vector a -> [b] Source #

*O(n)* Convert a `Vector`

into a list using a conversion function

# Basic interface

cons :: Storable a => a -> Vector a -> Vector a Source #

*O(n)* `cons`

is analogous to (:) for lists, but of different
complexity, as it requires a memcpy.

snoc :: Storable a => Vector a -> a -> Vector a Source #

*O(n)* Append an element to the end of a `Vector`

# Transforming `Vector`

s

mapIndexed :: (Storable a, Storable b) => (Int -> a -> b) -> Vector a -> Vector b Source #

*O(n)* map functions, provided with the index at each position

reverse :: Storable a => Vector a -> Vector a Source #

*O(n)* `reverse`

`xs`

efficiently returns the elements of `xs`

in reverse order.

intersperse :: Storable a => a -> Vector a -> Vector a Source #

*O(n)* The `intersperse`

function takes a element and a
`Vector`

and `intersperses' that element between the elements of
the `Vector`

. It is analogous to the intersperse function on
Lists.

# Reducing `Vector`

s (folds)

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

'foldl\'' is like `foldl`

, but strict in the accumulator.

foldr :: Storable a => (a -> b -> b) -> b -> Vector a -> b Source #

`foldr`

, applied to a binary operator, a starting value
(typically the right-identity of the operator), and a `Vector`

,
reduces the `Vector`

using the binary operator, from right to left.
However, it is not the same as `foldl`

applied to the reversed vector.
Actually `foldr`

starts processing with the first element,
and thus can be used for efficiently building a singly linked list
by `foldr (:) [] vec`

.
Unfortunately `foldr`

is quite slow for low-level loops,
since GHC (up to 6.12.1) cannot detect the loop.

## Special folds

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

Map a function over a `Vector`

and concatenate the results

foldMap :: (Storable a, Monoid m) => (a -> m) -> Vector a -> m Source #

This is like `mconcat . map f`

,
but in many cases the result of `f`

will not be storable.

monoidConcatMap :: (Storable a, Monoid m) => (a -> m) -> Vector a -> m Source #

Deprecated: Use foldMap instead.

# Building `Vector`

s

## Scans

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

scanr is the right-to-left dual of scanl.

## Accumulating maps

mapAccumL :: (Storable a, Storable b) => (acc -> a -> (acc, b)) -> acc -> Vector a -> (acc, Vector b) Source #

mapAccumR :: (Storable a, Storable b) => (acc -> a -> (acc, b)) -> acc -> Vector a -> (acc, Vector b) Source #

crochetL :: (Storable x, Storable y) => (x -> acc -> Maybe (y, acc)) -> acc -> Vector x -> Vector y Source #

crochetLResult :: (Storable x, Storable y) => (x -> acc -> Maybe (y, acc)) -> acc -> Vector x -> (Vector y, Maybe acc) Source #

## Unfolding `Vector`

s

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

*O(n)*, where *n* is the length of the result. The `unfoldr`

function is analogous to the List 'unfoldr'. `unfoldr`

builds a
`Vector`

from a seed value. The function takes the element and
returns `Nothing`

if it is done producing the 'Vector or returns
`Just`

`(a,b)`

, in which case, `a`

is the next element in the `Vector`

,
and `b`

is the seed value for further production.

Examples:

unfoldr (\x -> if x <= 5 then Just (x, x + 1) else Nothing) 0 == pack [0, 1, 2, 3, 4, 5]

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

*O(n)* Like `unfoldr`

, `unfoldrN`

builds a `Vector`

from a seed
value. However, the length of the result is limited by the first
argument to `unfoldrN`

. This function is more efficient than `unfoldr`

when the maximum length of the result is known.

The following equation relates `unfoldrN`

and `unfoldr`

:

fst (unfoldrN n f s) == take n (unfoldr f s)

unfoldrResultN :: Storable b => Int -> (a -> c) -> (a -> Either c (b, a)) -> a -> (Vector b, c) Source #

*O(n)* Like `unfoldrN`

this function builds a `Vector`

from a seed value with limited size.
Additionally it returns a value, that depends on the state,
but is not necessarily the state itself.
If end of vector and end of the generator coincide,
then the result is as if only the end of vector is reached.

Example:

unfoldrResultN 30 Char.ord (\c -> if c>'z' then Left 1000 else Right (c, succ c)) 'a'

The following equation relates `unfoldrN`

and `unfoldrResultN`

:

unfoldrN n f s == unfoldrResultN n Just (maybe (Left Nothing) Right . f) s

It is not possible to express `unfoldrResultN`

in terms of `unfoldrN`

.

sample :: Storable a => Int -> (Int -> a) -> Vector a Source #

*O(n)*, where *n* is the length of the result.
This function constructs a vector by evaluating a function
that depends on the element index.
It is a special case of `unfoldrN`

and can in principle be parallelized.

Examples:

sample 26 (\x -> chr(ord 'a'+x)) == pack "abcdefghijklmnopqrstuvwxyz"

# Substrings

## Breaking strings

group :: (Storable a, Eq a) => Vector a -> [Vector a] Source #

The `group`

function takes a `Vector`

and returns a list of
`Vector`

s such that the concatenation of the result is equal to the
argument. Moreover, each sublist in the result contains only equal
elements. For example,

group "Mississippi" = ["M","i","ss","i","ss","i","pp","i"]

It is a special case of `groupBy`

, which allows the programmer to
supply their own equality test. It is about 40% faster than
*groupBy (==)*

inits :: Storable a => Vector a -> [Vector a] Source #

*O(n)* Return all initial segments of the given `Vector`

, shortest first.

tails :: Storable a => Vector a -> [Vector a] Source #

*O(n)* Return all final segments of the given `Vector`

, longest first.

## Breaking into many substrings

split :: (Storable a, Eq a) => a -> Vector a -> [Vector a] Source #

*O(n)* Break a `Vector`

into pieces separated by the
argument, consuming the delimiter. I.e.

split '\n' "a\nb\nd\ne" == ["a","b","d","e"] split 'a' "aXaXaXa" == ["","X","X","X"] split 'x' "x" == ["",""]

and

join [c] . split c == id split == splitWith . (==)

As for all splitting functions in this library, this function does
not copy the substrings, it just constructs new `Vector`

s that
are slices of the original.

splitWith :: Storable a => (a -> Bool) -> Vector a -> [Vector a] Source #

*O(n)* Splits a `Vector`

into components delimited by
separators, where the predicate returns True for a separator element.
The resulting components do not contain the separators. Two adjacent
separators result in an empty component in the output. eg.

splitWith (=='a') "aabbaca" == ["","","bb","c",""] splitWith (=='a') [] == []

tokens :: Storable a => (a -> Bool) -> Vector a -> [Vector a] Source #

Like `splitWith`

, except that sequences of adjacent separators are
treated as a single separator. eg.

tokens (=='a') "aabbaca" == ["bb","c"]

sliceVertical :: Storable a => Int -> Vector a -> [Vector a] Source #

`sliceVertical`

`n xs`

divides vector in chunks of size `n`

.
Requires time proportionally to length of result list,
i.e. `ceiling (length xs / n)`

.

## Joining strings

# Predicates

isPrefixOf :: (Storable a, Eq a) => Vector a -> Vector a -> Bool Source #

*O(n)* The `isPrefixOf`

function takes two `Vector`

and returns `True`

iff the first is a prefix of the second.

isSuffixOf :: (Storable a, Eq a) => Vector a -> Vector a -> Bool Source #

*O(n)* The `isSuffixOf`

function takes two `Vector`

s and returns `True`

iff the first is a suffix of the second.

The following holds:

isSuffixOf x y == reverse x `isPrefixOf` reverse y

# Searching `Vector`

s

## Searching by equality

## Searching with a predicate

# Indexing `Vector`

s

index :: Storable a => Vector a -> Int -> a Source #

*O(1)* `Vector`

index (subscript) operator, starting from 0.

elemIndices :: (Storable a, Eq a) => a -> Vector a -> [Int] Source #

*O(n)* The `elemIndices`

function extends `elemIndex`

, by returning
the indices of all elements equal to the query element, in ascending order.

elemIndexEnd :: (Storable a, Eq a) => a -> Vector a -> Maybe Int Source #

*O(n)* The `elemIndexEnd`

function returns the last index of the
element in the given `Vector`

which is equal to the query
element, or `Nothing`

if there is no such element. The following
holds:

elemIndexEnd c xs == (-) (length xs - 1) `fmap` elemIndex c (reverse xs)

findIndices :: Storable a => (a -> Bool) -> Vector a -> [Int] Source #

The `findIndices`

function extends `findIndex`

, by returning the
indices of all elements satisfying the predicate, in ascending order.

count :: (Storable a, Eq a) => a -> Vector a -> Int Source #

count returns the number of times its argument appears in the `Vector`

count = length . elemIndices

But more efficiently than using length on the intermediate list.

findIndexOrEnd :: Storable a => (a -> Bool) -> Vector a -> Int Source #

`findIndexOrEnd`

is a variant of findIndex, that returns the length
of the string if no element is found, rather than Nothing.

# Zipping and unzipping `Vector`

s

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

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

Like `zipWith`

but for three input vectors

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

Like `zipWith`

but for four input vectors
If you need even more input vectors,
you might write a function yourselve using unfoldrN and viewL.

# Interleaved `Vector`

s

deinterleave :: Storable a => Int -> Vector a -> [Vector a] Source #

*O(n)*
Returns n sieved vectors with successive starting elements.
`deinterleave 3 (pack ['a'..'k']) = [pack "adgj", pack "behk", pack "cfi"]`

This is the same as `sliceHorizontal`

.

interleave :: Storable a => [Vector a] -> Vector a Source #

*O(n)*
Almost the inverse of deinterleave.
Restriction is that all input vector must have equal length.
`interleave [pack "adgj", pack "behk", pack "cfil"] = pack ['a'..'l']`

# IO

poke :: Storable a => Ptr a -> Vector a -> IO () Source #

Write a `Vector`

to a contiguous piece of memory.

peek :: Storable a => Int -> Ptr a -> IO (Vector a) Source #

Read a `Vector`

from a contiguous piece of memory.