IntervalMap-0.4.0.0: Maps from Intervals to values, with efficient search.

Portabilityportable
Stabilityexperimental
Maintainerchbreitkopf@gmail.com
Safe HaskellSafe-Inferred

Data.IntervalMap.Generic.Strict

Contents

Description

An implementation of maps from intervals to values. The key intervals may overlap, and the implementation contains efficient search functions for all keys containing a point or overlapping an interval. Closed, open, and half-open intervals can be contained in the same map.

The functions in this module are strict in both the keys and the values. If you need value-lazy maps, use Data.IntervalMap.Lazy instead. The IntervalMap type itself is shared between the lazy and strict modules, meaning that the same IntervalMap value can be passed to functions in both modules (although that is rarely needed).

An IntervalMap cannot contain duplicate keys - if you need to map a key to multiple values, use a collection as the value type, for example: IntervalMap k [v].

It is an error to insert an empty interval into a map. This precondition is not checked by the various construction functions.

Since many function names (but not the type name) clash with Prelude names, this module is usually imported qualified, e.g.

  import Data.IntervalMap (IvMap)
  import qualified Data.IntervalMap as IvMap

It offers most of the same functions as Map, but uses Interval k instead of just k as the key type. Some of the functions need stricter type constraints to maintain the additional information for efficient interval searching, for example fromDistinctAscList needs an Ord k constraint. Also, some functions differ in asymptotic performance (for example size) or have not been tuned for efficiency as much as their equivalents in Map (in particular the various set functions).

In addition, there are functions specific to maps of intervals, for example to search for all keys containing a given point or contained in a given interval.

The implementation is a red-black tree augmented with the maximum upper bound of all keys.

Parts of this implementation are based on code from the Map implementation, (c) Daan Leijen 2002, (c) Andriy Palamarchuk 2008. The red-black tree deletion is based on code from llrbtree by Kazu Yamamoto. Of course, any errors are mine.

Synopsis

re-export

class Ord e => Interval i e | i -> e whereSource

Intervals with endpoints of type e. A minimal instance declaration needs to define con, lowerBound, and upperBound.

Methods

lowerBound, upperBound :: i -> eSource

lower and upper bound

leftClosed :: i -> BoolSource

Does the interval include its lower bound? Default is True for all values, i.e. closed intervals.

rightClosed :: i -> BoolSource

Does the interval include its upper bound bound? Default is True for all values, i.e. closed intervals.

overlaps, after, before, subsumes :: i -> i -> BoolSource

above, inside, below :: e -> i -> BoolSource

isEmpty :: i -> BoolSource

Instances

Ord a => Interval (Interval a) a 

Map type

data IntervalMap k v Source

A map from intervals of type k to values of type v.

Instances

Functor (IntervalMap k) 
Foldable (IntervalMap k) 
Traversable (IntervalMap k) 
(Eq k, Eq v) => Eq (IntervalMap k v) 
(Ord k, Ord v) => Ord (IntervalMap k v) 
(Ord k, Read k, Read e, Interval i k, Ord i, Read i) => Read (IntervalMap i e) 
(Show k, Show a) => Show (IntervalMap k a) 
(Interval i k, Ord i) => Monoid (IntervalMap i v) 
(NFData k, NFData a) => NFData (IntervalMap k a) 

Operators

(!) :: (Interval k e, Ord k) => IntervalMap k v -> k -> vSource

O(log n). Lookup value for given key. Calls error if the key is not in the map.

(\\) :: (Interval k e, Ord k) => IntervalMap k a -> IntervalMap k b -> IntervalMap k aSource

Same as difference.

Query

null :: IntervalMap k v -> BoolSource

O(1). Is the map empty?

size :: IntervalMap k v -> IntSource

O(n). Number of keys in the map.

Caution: unlike size, which takes constant time, this is linear in the number of keys!

member :: Ord k => k -> IntervalMap k v -> BoolSource

O(log n). Does the map contain the given key? See also notMember.

notMember :: Ord k => k -> IntervalMap k v -> BoolSource

O(log n). Does the map not contain the given key? See also member.

lookup :: Ord k => k -> IntervalMap k v -> Maybe vSource

O(log n). Look up the given key in the map, returning the value (Just value), or Nothing if the key is not in the map.

findWithDefault :: Ord k => a -> k -> IntervalMap k a -> aSource

O(log n). The expression (findWithDefault def k map) returns the value at key k or returns default value def when the key is not in the map.

 findWithDefault 'x' 1 (fromList [(5,'a'), (3,'b')]) == 'x'
 findWithDefault 'x' 5 (fromList [(5,'a'), (3,'b')]) == 'a'

Interval query

containing :: Interval k e => IntervalMap k v -> e -> [(k, v)]Source

Return all key/value pairs where the key intervals contain the given point. The elements are returned in ascending key order.

O(n), since potentially all keys could contain the point. O(log n) average case. This is also the worst case for maps containing no overlapping keys.

intersecting :: Interval k e => IntervalMap k v -> k -> [(k, v)]Source

Return all key/value pairs where the key intervals overlap (intersect) the given interval. The elements are returned in ascending key order.

O(n), since potentially all keys could intersect the interval. O(log n) average case, if few keys intersect the interval.

within :: Interval k e => IntervalMap k v -> k -> [(k, v)]Source

Return all key/value pairs where the key intervals are completely inside the given interval. The elements are returned in ascending key order.

O(n), since potentially all keys could be inside the interval. O(log n) average case, if few keys are inside the interval.

Construction

empty :: IntervalMap k vSource

O(1). The empty map.

singleton :: k -> v -> IntervalMap k vSource

O(1). A map with one entry.

Insertion

insert :: (Interval k e, Ord k) => k -> v -> IntervalMap k v -> IntervalMap k vSource

O(log n). Insert a new key/value pair. If the map already contains the key, its value is changed to the new value.

insertWith :: (Interval k e, Ord k) => (v -> v -> v) -> k -> v -> IntervalMap k v -> IntervalMap k vSource

O(log n). Insert with a function, combining new value and old value. insertWith f key value mp will insert the pair (key, value) into mp if key does not exist in the map. If the key does exist, the function will insert the pair (key, f new_value old_value).

insertWithKey :: (Interval k e, Ord k) => (k -> v -> v -> v) -> k -> v -> IntervalMap k v -> IntervalMap k vSource

O(log n). Insert with a function, combining key, new value and old value. insertWithKey f key value mp will insert the pair (key, value) into mp if key does not exist in the map. If the key does exist, the function will insert the pair (key, f key new_value old_value). Note that the key passed to f is the same key passed to insertWithKey.

insertLookupWithKey :: (Interval k e, Ord k) => (k -> v -> v -> v) -> k -> v -> IntervalMap k v -> (Maybe v, IntervalMap k v)Source

O(log n). Combine insert with old values retrieval.

Delete/Update

delete :: (Interval k e, Ord k) => k -> IntervalMap k v -> IntervalMap k vSource

O(log n). Delete a key from the map. If the map does not contain the key, it is returned unchanged.

adjust :: Ord k => (a -> a) -> k -> IntervalMap k a -> IntervalMap k aSource

O(log n). Update a value at a specific key with the result of the provided function. When the key is not a member of the map, the original map is returned.

adjustWithKey :: Ord k => (k -> a -> a) -> k -> IntervalMap k a -> IntervalMap k aSource

O(log n). Adjust a value at a specific key. When the key is not a member of the map, the original map is returned.

update :: (Interval k e, Ord k) => (a -> Maybe a) -> k -> IntervalMap k a -> IntervalMap k aSource

O(log n). The expression (update f k map) updates the value x at k (if it is in the map). If (f x) is Nothing, the element is deleted. If it is (Just y), the key k is bound to the new value y.

updateWithKey :: (Interval k e, Ord k) => (k -> a -> Maybe a) -> k -> IntervalMap k a -> IntervalMap k aSource

O(log n). The expression (updateWithKey f k map) updates the value x at k (if it is in the map). If (f k x) is Nothing, the element is deleted. If it is (Just y), the key k is bound to the new value y.

updateLookupWithKey :: (Interval k e, Ord k) => (k -> a -> Maybe a) -> k -> IntervalMap k a -> (Maybe a, IntervalMap k a)Source

O(log n). Lookup and update. See also updateWithKey. The function returns changed value, if it is updated. Returns the original key value if the map entry is deleted.

alter :: (Interval k e, Ord k) => (Maybe a -> Maybe a) -> k -> IntervalMap k a -> IntervalMap k aSource

O(log n). The expression (alter f k map) alters the value x at k, or absence thereof. alter can be used to insert, delete, or update a value in a Map. In short : lookup k (alter f k m) = f (lookup k m).

Combine

Union

union :: (Interval k e, Ord k) => IntervalMap k a -> IntervalMap k a -> IntervalMap k aSource

O(n+m). The expression (union t1 t2) takes the left-biased union of t1 and t2. It prefers t1 when duplicate keys are encountered, i.e. (union == unionWith const).

unionWith :: (Interval k e, Ord k) => (a -> a -> a) -> IntervalMap k a -> IntervalMap k a -> IntervalMap k aSource

O(n+m). Union with a combining function.

unionWithKey :: (Interval k e, Ord k) => (k -> a -> a -> a) -> IntervalMap k a -> IntervalMap k a -> IntervalMap k aSource

O(n+m). Union with a combining function.

unions :: (Interval k e, Ord k) => [IntervalMap k a] -> IntervalMap k aSource

The union of a list of maps: (unions == foldl union empty).

unionsWith :: (Interval k e, Ord k) => (a -> a -> a) -> [IntervalMap k a] -> IntervalMap k aSource

The union of a list of maps, with a combining operation: (unionsWith f == foldl (unionWith f) empty).

Difference

difference :: (Interval k e, Ord k) => IntervalMap k a -> IntervalMap k b -> IntervalMap k aSource

O(n+m). Difference of two maps. Return elements of the first map not existing in the second map.

differenceWith :: (Interval k e, Ord k) => (a -> b -> Maybe a) -> IntervalMap k a -> IntervalMap k b -> IntervalMap k aSource

O(n+m). Difference with a combining function. When two equal keys are encountered, the combining function is applied to the values of these keys. If it returns Nothing, the element is discarded (proper set difference). If it returns (Just y), the element is updated with a new value y.

differenceWithKey :: (Interval k e, Ord k) => (k -> a -> b -> Maybe a) -> IntervalMap k a -> IntervalMap k b -> IntervalMap k aSource

O(n+m). Difference with a combining function. When two equal keys are encountered, the combining function is applied to the key and both values. If it returns Nothing, the element is discarded (proper set difference). If it returns (Just y), the element is updated with a new value y.

Intersection

intersection :: (Interval k e, Ord k) => IntervalMap k a -> IntervalMap k b -> IntervalMap k aSource

O(n+m). Intersection of two maps. Return data in the first map for the keys existing in both maps. (intersection m1 m2 == intersectionWith const m1 m2).

intersectionWith :: (Interval k e, Ord k) => (a -> b -> c) -> IntervalMap k a -> IntervalMap k b -> IntervalMap k cSource

O(n+m). Intersection with a combining function.

intersectionWithKey :: (Interval k e, Ord k) => (k -> a -> b -> c) -> IntervalMap k a -> IntervalMap k b -> IntervalMap k cSource

O(n+m). Intersection with a combining function.

Traversal

Map

map :: (a -> b) -> IntervalMap k a -> IntervalMap k bSource

O(n). Map a function over all values in the map.

mapWithKey :: (k -> a -> b) -> IntervalMap k a -> IntervalMap k bSource

O(n). Map a function over all values in the map.

mapAccum :: (a -> b -> (a, c)) -> a -> IntervalMap k b -> (a, IntervalMap k c)Source

O(n). The function mapAccum threads an accumulating argument through the map in ascending order of keys.

 let f a b = (a ++ b, b ++ "X")
 mapAccum f "Everything: " (fromList [(5,"a"), (3,"b")]) == ("Everything: ba", fromList [(3, "bX"), (5, "aX")])

mapAccumWithKey :: (a -> k -> b -> (a, c)) -> a -> IntervalMap k b -> (a, IntervalMap k c)Source

O(n). The function mapAccumWithKey threads an accumulating argument through the map in ascending order of keys.

 let f a k b = (a ++ " " ++ (show k) ++ "-" ++ b, b ++ "X")
 mapAccumWithKey f "Everything:" (fromList [(5,"a"), (3,"b")]) == ("Everything: 3-b 5-a", fromList [(3, "bX"), (5, "aX")])

mapAccumRWithKey :: (a -> k -> b -> (a, c)) -> a -> IntervalMap k b -> (a, IntervalMap k c)Source

O(n). The function mapAccumRWithKey threads an accumulating argument through the map in descending order of keys.

mapKeys :: (Interval k2 e, Ord k2) => (k1 -> k2) -> IntervalMap k1 a -> IntervalMap k2 aSource

O(n log n). mapKeys f s is the map obtained by applying f to each key of s.

The size of the result may be smaller if f maps two or more distinct keys to the same new key. In this case the value at the smallest of these keys is retained.

mapKeysWith :: (Interval k2 e, Ord k2) => (a -> a -> a) -> (k1 -> k2) -> IntervalMap k1 a -> IntervalMap k2 aSource

O(n log n). mapKeysWith c f s is the map obtained by applying f to each key of s.

The size of the result may be smaller if f maps two or more distinct keys to the same new key. In this case the associated values will be combined using c.

mapKeysMonotonic :: (Interval k2 e, Ord k2) => (k1 -> k2) -> IntervalMap k1 a -> IntervalMap k2 aSource

O(n log n). mapKeysMonotonic f s == mapKeys f s, but works only when f is strictly monotonic. That is, for any values x and y, if x < y then f x < f y. The precondition is not checked.

Fold

foldr :: (a -> b -> b) -> b -> IntervalMap k a -> bSource

O(n). Fold the values in the map using the given right-associative binary operator, such that foldr f z == foldr f z . elems.

foldl :: (b -> a -> b) -> b -> IntervalMap k a -> bSource

O(n). Fold the values in the map using the given left-associative binary operator, such that foldl f z == foldl f z . elems.

foldrWithKey :: (k -> v -> a -> a) -> a -> IntervalMap k v -> aSource

O(n). Fold the keys and values in the map using the given right-associative binary operator, such that foldrWithKey f z == foldr (uncurry f) z . toAscList.

foldlWithKey :: (a -> k -> v -> a) -> a -> IntervalMap k v -> aSource

O(n). Fold the keys and values in the map using the given left-associative binary operator, such that foldlWithKey f z == foldl (\z' (kx, x) -> f z' kx x) z . toAscList.

Conversion

elems :: IntervalMap k v -> [v]Source

O(n). List of all values in the map, in ascending order of their keys.

keys :: IntervalMap k v -> [k]Source

O(n). List of all keys in the map, in ascending order.

keysSet :: Ord k => IntervalMap k v -> Set kSource

O(n). Set of the keys.

assocs :: IntervalMap k v -> [(k, v)]Source

Same as toAscList.

Lists

toList :: IntervalMap k v -> [(k, v)]Source

O(n). The list of all key/value pairs contained in the map, in no particular order.

fromList :: (Interval k e, Ord k) => [(k, v)] -> IntervalMap k vSource

O(n log n). Build a map from a list of key/value pairs. See also fromAscList. If the list contains more than one value for the same key, the last value for the key is retained.

fromListWith :: (Interval k e, Ord k) => (a -> a -> a) -> [(k, a)] -> IntervalMap k aSource

O(n log n). Build a map from a list of key/value pairs with a combining function. See also fromAscListWith.

fromListWithKey :: (Interval k e, Ord k) => (k -> a -> a -> a) -> [(k, a)] -> IntervalMap k aSource

O(n log n). Build a map from a list of key/value pairs with a combining function. See also fromAscListWith.

Ordered lists

toAscList :: IntervalMap k v -> [(k, v)]Source

O(n). The list of all key/value pairs contained in the map, in ascending order of keys.

toDescList :: IntervalMap k v -> [(k, v)]Source

O(n). The list of all key/value pairs contained in the map, in descending order of keys.

fromAscList :: (Interval k e, Eq k) => [(k, v)] -> IntervalMap k vSource

O(n). Build a map from an ascending list in linear time. The precondition (input list is ascending) is not checked.

fromAscListWith :: (Interval k e, Eq k) => (a -> a -> a) -> [(k, a)] -> IntervalMap k aSource

O(n). Build a map from an ascending list in linear time with a combining function for equal keys. The precondition (input list is ascending) is not checked.

fromAscListWithKey :: (Interval k e, Eq k) => (k -> a -> a -> a) -> [(k, a)] -> IntervalMap k aSource

O(n). Build a map from an ascending list in linear time with a combining function for equal keys. The precondition (input list is ascending) is not checked.

fromDistinctAscList :: Interval k e => [(k, v)] -> IntervalMap k vSource

O(n). Build a map from an ascending list of elements with distinct keys in linear time. The precondition is not checked.

Filter

filter :: Interval k e => (a -> Bool) -> IntervalMap k a -> IntervalMap k aSource

O(n). Filter values satisfying a predicate.

filterWithKey :: Interval k e => (k -> a -> Bool) -> IntervalMap k a -> IntervalMap k aSource

O(n). Filter keys/values satisfying a predicate.

partition :: Interval k e => (a -> Bool) -> IntervalMap k a -> (IntervalMap k a, IntervalMap k a)Source

O(n). Partition the map according to a predicate. The first map contains all elements that satisfy the predicate, the second all elements that fail the predicate. See also split.

partitionWithKey :: Interval k e => (k -> a -> Bool) -> IntervalMap k a -> (IntervalMap k a, IntervalMap k a)Source

O(n). Partition the map according to a predicate. The first map contains all elements that satisfy the predicate, the second all elements that fail the predicate. See also split.

mapMaybe :: Interval k e => (a -> Maybe b) -> IntervalMap k a -> IntervalMap k bSource

O(n). Map values and collect the Just results.

mapMaybeWithKey :: Interval k e => (k -> a -> Maybe b) -> IntervalMap k a -> IntervalMap k bSource

O(n). Map keys/values and collect the Just results.

mapEither :: Interval k e => (a -> Either b c) -> IntervalMap k a -> (IntervalMap k b, IntervalMap k c)Source

O(n). Map values and separate the Left and Right results.

mapEitherWithKey :: Interval k e => (k -> a -> Either b c) -> IntervalMap k a -> (IntervalMap k b, IntervalMap k c)Source

O(n). Map keys/values and separate the Left and Right results.

split :: (Interval i k, Ord i) => i -> IntervalMap i a -> (IntervalMap i a, IntervalMap i a)Source

O(n). The expression (split k map) is a pair (map1,map2) where the keys in map1 are smaller than k and the keys in map2 larger than k. Any key equal to k is found in neither map1 nor map2.

splitLookup :: (Interval i k, Ord i) => i -> IntervalMap i a -> (IntervalMap i a, Maybe a, IntervalMap i a)Source

O(n). The expression (splitLookup k map) splits a map just like split but also returns lookup k map.

Submap

isSubmapOf :: (Ord k, Eq a) => IntervalMap k a -> IntervalMap k a -> BoolSource

O(n+m). This function is defined as (isSubmapOf = isSubmapOfBy (==)).

isSubmapOfBy :: Ord k => (a -> b -> Bool) -> IntervalMap k a -> IntervalMap k b -> BoolSource

O(n+m). The expression (isSubmapOfBy f t1 t2) returns True if all keys in t1 are in tree t2, and f returns True when applied to their respective values.

isProperSubmapOf :: (Ord k, Eq a) => IntervalMap k a -> IntervalMap k a -> BoolSource

O(n+m). Is this a proper submap? (ie. a submap but not equal). Defined as (isProperSubmapOf = isProperSubmapOfBy (==)).

isProperSubmapOfBy :: Ord k => (a -> b -> Bool) -> IntervalMap k a -> IntervalMap k b -> BoolSource

O(n+m). Is this a proper submap? (ie. a submap but not equal). The expression (isProperSubmapOfBy f m1 m2) returns True when m1 and m2 are not equal, all keys in m1 are in m2, and when f returns True when applied to their respective values.

Min/Max

findMin :: IntervalMap k v -> (k, v)Source

O(log n). Returns the smallest key and its associated value. Calls error if the map is empty.

findMax :: IntervalMap k v -> (k, v)Source

O(log n). Returns the largest key and its associated value. Calls error if the map is empty.

findLast :: Interval k e => IntervalMap k v -> (k, v)Source

Returns the interval with the largest endpoint. If there is more than one interval with that endpoint, return the rightmost.

O(n), since all keys could have the same endpoint. O(log n) average case.

deleteMin :: (Interval k e, Ord k) => IntervalMap k v -> IntervalMap k vSource

O(log n). Remove the smallest key from the map. Return the empty map if the map is empty.

deleteMax :: (Interval k e, Ord k) => IntervalMap k v -> IntervalMap k vSource

O(log n). Remove the largest key from the map. Return the empty map if the map is empty.

deleteFindMin :: (Interval k e, Ord k) => IntervalMap k v -> ((k, v), IntervalMap k v)Source

O(log n). Delete and return the smallest key.

deleteFindMax :: (Interval k e, Ord k) => IntervalMap k v -> ((k, v), IntervalMap k v)Source

O(log n). Delete and return the largest key.

updateMin :: (Interval k e, Ord k) => (v -> Maybe v) -> IntervalMap k v -> IntervalMap k vSource

O(log n). Update or delete value at minimum key.

updateMax :: (Interval k e, Ord k) => (v -> Maybe v) -> IntervalMap k v -> IntervalMap k vSource

O(log n). Update or delete value at maximum key.

updateMinWithKey :: (Interval k e, Ord k) => (k -> v -> Maybe v) -> IntervalMap k v -> IntervalMap k vSource

O(log n). Update or delete value at minimum key.

updateMaxWithKey :: (Interval k e, Ord k) => (k -> v -> Maybe v) -> IntervalMap k v -> IntervalMap k vSource

O(log n). Update or delete value at maximum key.

minView :: (Interval k e, Ord k) => IntervalMap k a -> Maybe (a, IntervalMap k a)Source

O(log n). Retrieves the value associated with minimal key of the map, and the map stripped of that element, or Nothing if passed an empty map.

maxView :: (Interval k e, Ord k) => IntervalMap k a -> Maybe (a, IntervalMap k a)Source

O(log n). Retrieves the value associated with maximal key of the map, and the map stripped of that element, or Nothing if passed an empty map.

minViewWithKey :: (Interval k e, Ord k) => IntervalMap k a -> Maybe ((k, a), IntervalMap k a)Source

O(log n). Retrieves the minimal (key,value) pair of the map, and the map stripped of that element, or Nothing if passed an empty map.

 minViewWithKey (fromList [(5,"a"), (3,"b")]) == Just ((3,"b"), singleton 5 "a")
 minViewWithKey empty == Nothing

maxViewWithKey :: (Interval k e, Ord k) => IntervalMap k a -> Maybe ((k, a), IntervalMap k a)Source

O(log n). Retrieves the maximal (key,value) pair of the map, and the map stripped of that element, or Nothing if passed an empty map.

Debugging

valid :: (Interval i k, Ord i) => IntervalMap i v -> BoolSource

Check red-black-tree and interval search augmentation invariants. For testing/debugging only.

Testing

height :: IntervalMap k v -> IntSource

The height of the tree. For testing/debugging only.

maxHeight :: Int -> IntSource

The maximum height of a red-black tree with the given number of nodes. For testing/debugging only.

showStats :: IntervalMap k a -> (Int, Int, Int)Source

Tree statistics (size, height, maxHeight size). For testing/debugging only.