Copyright	(c) Conal Elliott 2008-2016
License	BSD3
Maintainer	conal@conal.net
Stability	experimental
Safe Haskell	Safe
Language	Haskell2010

Data.MemoTrie

Description

Trie-based memoizer

Adapted from sjanssen's paste: "a lazy trie", which I think is based on Ralf Hinze's paper "Memo Functions, Polytypically!".

You can automatically derive generic instances. for example:

{--}
import Data.MemoTrie
import GHC.Generics (Generic) 

data Color = RGB Int Int Int
           | NamedColor String 
 deriving (Generic) 

instance HasTrie Color where
  newtype (Color :->: b) = ColorTrie { unColorTrie :: Reg Color :->: b } 
  trie = trieGeneric ColorTrie 
  untrie = untrieGeneric unColorTrie
  enumerate = enumerateGeneric unColorTrie

see examples/Generic.hs, which can be run with:

cabal configure -fexamples && cabal run generic

Synopsis

class HasTrie a where
- data (:->:) a :: * -> *
- trie :: (a -> b) -> a :->: b
- untrie :: (a :->: b) -> a -> b
- enumerate :: (a :->: b) -> [(a, b)]
domain :: HasTrie a => [a]
idTrie :: HasTrie a => a :->: a
(@.@) :: (HasTrie a, HasTrie b) => (b :->: c) -> (a :->: b) -> a :->: c
memo :: HasTrie t => (t -> a) -> t -> a
memo2 :: (HasTrie s, HasTrie t) => (s -> t -> a) -> s -> t -> a
memo3 :: (HasTrie r, HasTrie s, HasTrie t) => (r -> s -> t -> a) -> r -> s -> t -> a
mup :: HasTrie t => (b -> c) -> (t -> b) -> t -> c
inTrie :: (HasTrie a, HasTrie c) => ((a -> b) -> c -> d) -> (a :->: b) -> c :->: d
inTrie2 :: (HasTrie a, HasTrie c, HasTrie e) => ((a -> b) -> (c -> d) -> e -> f) -> (a :->: b) -> (c :->: d) -> e :->: f
inTrie3 :: (HasTrie a, HasTrie c, HasTrie e, HasTrie g) => ((a -> b) -> (c -> d) -> (e -> f) -> g -> h) -> (a :->: b) -> (c :->: d) -> (e :->: f) -> g :->: h
trieGeneric :: (Generic a, HasTrie (Reg a)) => ((Reg a :->: b) -> a :->: b) -> (a -> b) -> a :->: b
untrieGeneric :: (Generic a, HasTrie (Reg a)) => ((a :->: b) -> Reg a :->: b) -> (a :->: b) -> a -> b
enumerateGeneric :: (Generic a, HasTrie (Reg a)) => ((a :->: b) -> Reg a :->: b) -> (a :->: b) -> [(a, b)]
type Reg a = Rep a ()
memoFix :: HasTrie a => ((a -> b) -> a -> b) -> a -> b

Documentation

class HasTrie a where Source #

Mapping from all elements of a to the results of some function

Associated Types

data (:->:) a :: * -> * infixr 0 Source #

Representation of trie with domain type a

Methods

trie :: (a -> b) -> a :->: b Source #

Create the trie for the entire domain of a function

untrie :: (a :->: b) -> a -> b Source #

Convert a trie to a function, i.e., access a field of the trie

enumerate :: (a :->: b) -> [(a, b)] Source #

List the trie elements. Order of keys (:: a) is always the same.

Instances

HasTrie Bool Source #
Instance details Defined in Data.MemoTrie Associated Types data Bool :->: b :: Type Source # Methods trie :: (Bool -> b) -> Bool :->: b Source # untrie :: (Bool :->: b) -> Bool -> b Source # enumerate :: (Bool :->: b) -> [(Bool, b)] Source #
HasTrie Char Source #
Instance details Defined in Data.MemoTrie Associated Types data Char :->: b :: Type Source # Methods trie :: (Char -> b) -> Char :->: b Source # untrie :: (Char :->: b) -> Char -> b Source # enumerate :: (Char :->: b) -> [(Char, b)] Source #
HasTrie Int Source #
Instance details Defined in Data.MemoTrie Associated Types data Int :->: b :: Type Source # Methods trie :: (Int -> b) -> Int :->: b Source # untrie :: (Int :->: b) -> Int -> b Source # enumerate :: (Int :->: b) -> [(Int, b)] Source #
HasTrie Int8 Source #
Instance details Defined in Data.MemoTrie Associated Types data Int8 :->: b :: Type Source # Methods trie :: (Int8 -> b) -> Int8 :->: b Source # untrie :: (Int8 :->: b) -> Int8 -> b Source # enumerate :: (Int8 :->: b) -> [(Int8, b)] Source #
HasTrie Int16 Source #
Instance details Defined in Data.MemoTrie Associated Types data Int16 :->: b :: Type Source # Methods trie :: (Int16 -> b) -> Int16 :->: b Source # untrie :: (Int16 :->: b) -> Int16 -> b Source # enumerate :: (Int16 :->: b) -> [(Int16, b)] Source #
HasTrie Int32 Source #
Instance details Defined in Data.MemoTrie Associated Types data Int32 :->: b :: Type Source # Methods trie :: (Int32 -> b) -> Int32 :->: b Source # untrie :: (Int32 :->: b) -> Int32 -> b Source # enumerate :: (Int32 :->: b) -> [(Int32, b)] Source #
HasTrie Int64 Source #
Instance details Defined in Data.MemoTrie Associated Types data Int64 :->: b :: Type Source # Methods trie :: (Int64 -> b) -> Int64 :->: b Source # untrie :: (Int64 :->: b) -> Int64 -> b Source # enumerate :: (Int64 :->: b) -> [(Int64, b)] Source #
HasTrie Integer Source #
Instance details Defined in Data.MemoTrie Associated Types data Integer :->: b :: Type Source # Methods trie :: (Integer -> b) -> Integer :->: b Source # untrie :: (Integer :->: b) -> Integer -> b Source # enumerate :: (Integer :->: b) -> [(Integer, b)] Source #
HasTrie Word Source #
Instance details Defined in Data.MemoTrie Associated Types data Word :->: b :: Type Source # Methods trie :: (Word -> b) -> Word :->: b Source # untrie :: (Word :->: b) -> Word -> b Source # enumerate :: (Word :->: b) -> [(Word, b)] Source #
HasTrie Word8 Source #
Instance details Defined in Data.MemoTrie Associated Types data Word8 :->: b :: Type Source # Methods trie :: (Word8 -> b) -> Word8 :->: b Source # untrie :: (Word8 :->: b) -> Word8 -> b Source # enumerate :: (Word8 :->: b) -> [(Word8, b)] Source #
HasTrie Word16 Source #
Instance details Defined in Data.MemoTrie Associated Types data Word16 :->: b :: Type Source # Methods trie :: (Word16 -> b) -> Word16 :->: b Source # untrie :: (Word16 :->: b) -> Word16 -> b Source # enumerate :: (Word16 :->: b) -> [(Word16, b)] Source #
HasTrie Word32 Source #
Instance details Defined in Data.MemoTrie Associated Types data Word32 :->: b :: Type Source # Methods trie :: (Word32 -> b) -> Word32 :->: b Source # untrie :: (Word32 :->: b) -> Word32 -> b Source # enumerate :: (Word32 :->: b) -> [(Word32, b)] Source #
HasTrie Word64 Source #
Instance details Defined in Data.MemoTrie Associated Types data Word64 :->: b :: Type Source # Methods trie :: (Word64 -> b) -> Word64 :->: b Source # untrie :: (Word64 :->: b) -> Word64 -> b Source # enumerate :: (Word64 :->: b) -> [(Word64, b)] Source #
HasTrie () Source #
Instance details Defined in Data.MemoTrie Associated Types data () :->: b :: Type Source # Methods trie :: (() -> b) -> () :->: b Source # untrie :: (() :->: b) -> () -> b Source # enumerate :: (() :->: b) -> [((), b)] Source #
HasTrie Void Source #
Instance details Defined in Data.MemoTrie Associated Types data Void :->: b :: Type Source # Methods trie :: (Void -> b) -> Void :->: b Source # untrie :: (Void :->: b) -> Void -> b Source # enumerate :: (Void :->: b) -> [(Void, b)] Source #
HasTrie x => HasTrie [x] Source #
Instance details Defined in Data.MemoTrie Associated Types data [x] :->: b :: Type Source # Methods trie :: ([x] -> b) -> [x] :->: b Source # untrie :: ([x] :->: b) -> [x] -> b Source # enumerate :: ([x] :->: b) -> [([x], b)] Source #
HasTrie a => HasTrie (Maybe a) Source #
Instance details Defined in Data.MemoTrie Associated Types data (Maybe a) :->: b :: Type Source # Methods trie :: (Maybe a -> b) -> Maybe a :->: b Source # untrie :: (Maybe a :->: b) -> Maybe a -> b Source # enumerate :: (Maybe a :->: b) -> [(Maybe a, b)] Source #
(HasTrie a, HasTrie b) => HasTrie (Either a b) Source #
Instance details Defined in Data.MemoTrie Associated Types data (Either a b) :->: b :: Type Source # Methods trie :: (Either a b -> b0) -> Either a b :->: b0 Source # untrie :: (Either a b :->: b0) -> Either a b -> b0 Source # enumerate :: (Either a b :->: b0) -> [(Either a b, b0)] Source #
HasTrie (V1 x) Source #	just like `void`
Instance details Defined in Data.MemoTrie Associated Types data (V1 x) :->: b :: Type Source # Methods trie :: (V1 x -> b) -> V1 x :->: b Source # untrie :: (V1 x :->: b) -> V1 x -> b Source # enumerate :: (V1 x :->: b) -> [(V1 x, b)] Source #
HasTrie (U1 x) Source #	just like `()`
Instance details Defined in Data.MemoTrie Associated Types data (U1 x) :->: b :: Type Source # Methods trie :: (U1 x -> b) -> U1 x :->: b Source # untrie :: (U1 x :->: b) -> U1 x -> b Source # enumerate :: (U1 x :->: b) -> [(U1 x, b)] Source #
(HasTrie a, HasTrie b) => HasTrie (a, b) Source #
Instance details Defined in Data.MemoTrie Associated Types data (a, b) :->: b :: Type Source # Methods trie :: ((a, b) -> b0) -> (a, b) :->: b0 Source # untrie :: ((a, b) :->: b0) -> (a, b) -> b0 Source # enumerate :: ((a, b) :->: b0) -> [((a, b), b0)] Source #
(HasTrie a, HasTrie b, HasTrie c) => HasTrie (a, b, c) Source #
Instance details Defined in Data.MemoTrie Associated Types data (a, b, c) :->: b :: Type Source # Methods trie :: ((a, b, c) -> b0) -> (a, b, c) :->: b0 Source # untrie :: ((a, b, c) :->: b0) -> (a, b, c) -> b0 Source # enumerate :: ((a, b, c) :->: b0) -> [((a, b, c), b0)] Source #
HasTrie a => HasTrie (K1 i a x) Source #	wraps `a`
Instance details Defined in Data.MemoTrie Associated Types data (K1 i a x) :->: b :: Type Source # Methods trie :: (K1 i a x -> b) -> K1 i a x :->: b Source # untrie :: (K1 i a x :->: b) -> K1 i a x -> b Source # enumerate :: (K1 i a x :->: b) -> [(K1 i a x, b)] Source #
(HasTrie (f x), HasTrie (g x)) => HasTrie ((f :+: g) x) Source #	wraps `Either (f x) (g x)`
Instance details Defined in Data.MemoTrie Associated Types data ((f :+: g) x) :->: b :: Type Source # Methods trie :: ((f :+: g) x -> b) -> (f :+: g) x :->: b Source # untrie :: ((f :+: g) x :->: b) -> (f :+: g) x -> b Source # enumerate :: ((f :+: g) x :->: b) -> [((f :+: g) x, b)] Source #
(HasTrie (f x), HasTrie (g x)) => HasTrie ((f :*: g) x) Source #	wraps `(f x, g x)`
Instance details Defined in Data.MemoTrie Associated Types data ((f :: g) x) :->: b :: Type Source # Methods trie :: ((f :: g) x -> b) -> (f :: g) x :->: b Source # untrie :: ((f :: g) x :->: b) -> (f :: g) x -> b Source # enumerate :: ((f :: g) x :->: b) -> [((f :*: g) x, b)] Source #
HasTrie (f x) => HasTrie (M1 i t f x) Source #	wraps `f x`
Instance details Defined in Data.MemoTrie Associated Types data (M1 i t f x) :->: b :: Type Source # Methods trie :: (M1 i t f x -> b) -> M1 i t f x :->: b Source # untrie :: (M1 i t f x :->: b) -> M1 i t f x -> b Source # enumerate :: (M1 i t f x :->: b) -> [(M1 i t f x, b)] Source #

domain :: HasTrie a => [a] Source #

Domain elements of a trie

idTrie :: HasTrie a => a :->: a Source #

Identity trie

(@.@) :: (HasTrie a, HasTrie b) => (b :->: c) -> (a :->: b) -> a :->: c infixr 9 Source #

Trie composition

memo :: HasTrie t => (t -> a) -> t -> a Source #

Trie-based function memoizer

memo2 :: (HasTrie s, HasTrie t) => (s -> t -> a) -> s -> t -> a Source #

Memoize a binary function, on its first argument and then on its second. Take care to exploit any partial evaluation.

memo3 :: (HasTrie r, HasTrie s, HasTrie t) => (r -> s -> t -> a) -> r -> s -> t -> a Source #

Memoize a ternary function on successive arguments. Take care to exploit any partial evaluation.

mup :: HasTrie t => (b -> c) -> (t -> b) -> t -> c Source #

Lift a memoizer to work with one more argument.

inTrie :: (HasTrie a, HasTrie c) => ((a -> b) -> c -> d) -> (a :->: b) -> c :->: d Source #

Apply a unary function inside of a trie

inTrie2 :: (HasTrie a, HasTrie c, HasTrie e) => ((a -> b) -> (c -> d) -> e -> f) -> (a :->: b) -> (c :->: d) -> e :->: f Source #

Apply a binary function inside of a trie

inTrie3 :: (HasTrie a, HasTrie c, HasTrie e, HasTrie g) => ((a -> b) -> (c -> d) -> (e -> f) -> g -> h) -> (a :->: b) -> (c :->: d) -> (e :->: f) -> g :->: h Source #

Apply a ternary function inside of a trie

trieGeneric :: (Generic a, HasTrie (Reg a)) => ((Reg a :->: b) -> a :->: b) -> (a -> b) -> a :->: b Source #

Generic-friendly default for trie

untrieGeneric :: (Generic a, HasTrie (Reg a)) => ((a :->: b) -> Reg a :->: b) -> (a :->: b) -> a -> b Source #

Generic-friendly default for untrie

enumerateGeneric :: (Generic a, HasTrie (Reg a)) => ((a :->: b) -> Reg a :->: b) -> (a :->: b) -> [(a, b)] Source #

Generic-friendly default for enumerate

type Reg a = Rep a () Source #

the data type in a regular form. "unlifted" generic representation. (i.e. is a unary type constructor).

memoFix :: HasTrie a => ((a -> b) -> a -> b) -> a -> b Source #

Memoizing recursion. Use like fix.