| Copyright | (c) Conal Elliott 2008-2016 |
|---|---|
| License | BSD3 |
| Maintainer | conal@conal.net |
| Stability | experimental |
| Safe Haskell | Safe-Inferred |
| 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:
{-# LANGUAGE DeriveGeneric, TypeOperators, TypeFamilies #-}
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 family (:->:) a :: Type -> Type
- 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 :: Type -> Type 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 Void Source # | |||||
| HasTrie Int16 Source # | |||||
| HasTrie Int32 Source # | |||||
| HasTrie Int64 Source # | |||||
| HasTrie Int8 Source # | |||||
| HasTrie Word16 Source # | |||||
| HasTrie Word32 Source # | |||||
| HasTrie Word64 Source # | |||||
| HasTrie Word8 Source # | |||||
| HasTrie Integer Source # | |||||
Defined in Data.MemoTrie Associated Types
| |||||
| HasTrie () Source # | |||||
| HasTrie Bool Source # | |||||
| HasTrie Char Source # | |||||
| HasTrie Int Source # | |||||
| HasTrie Word Source # | |||||
| HasTrie a => HasTrie (Maybe a) Source # | |||||
| HasTrie x => HasTrie [x] Source # | |||||
| (HasTrie a, HasTrie b) => HasTrie (Either a b) Source # | |||||
Defined in Data.MemoTrie Associated Types
| |||||
| HasTrie (U1 x) Source # | just like | ||||
| HasTrie (V1 x) Source # | just like | ||||
| (HasTrie a, HasTrie b) => HasTrie (a, b) Source # | |||||
Defined in Data.MemoTrie Associated Types
| |||||
| (HasTrie a, HasTrie b, HasTrie c) => HasTrie (a, b, c) Source # | |||||
Defined in Data.MemoTrie Associated Types
| |||||
| (HasTrie (f x), HasTrie (g x)) => HasTrie ((f :*: g) x) Source # | wraps | ||||
Defined in Data.MemoTrie Associated Types
| |||||
| (HasTrie (f x), HasTrie (g x)) => HasTrie ((f :+: g) x) Source # | wraps | ||||
Defined in Data.MemoTrie Associated Types
| |||||
| HasTrie a => HasTrie (K1 i a x) Source # | wraps | ||||
Defined in Data.MemoTrie Associated Types
| |||||
| HasTrie (f x) => HasTrie (M1 i t f x) Source # | wraps | ||||
Defined in Data.MemoTrie Associated Types
| |||||
data family (:->:) a :: Type -> Type infixr 0 Source #
Representation of trie with domain type a
Instances
| HasTrie a => Applicative ((:->:) a) Source # | |
| HasTrie a => Functor ((:->:) a) Source # | |
| HasTrie a => Monad ((:->:) a) Source # | |
| (HasTrie a, Monoid b) => Monoid (a :->: b) Source # | |
| (HasTrie a, Semigroup b) => Semigroup (a :->: b) Source # | |
| (HasTrie a, Show a, Show b) => Show (a :->: b) Source # | |
| (HasTrie a, Eq b) => Eq (a :->: b) Source # | |
| Newtype (Void :->: a) Source # | |
| Newtype (Either a b :->: x) Source # | |
| Newtype (Maybe a :->: x) Source # | |
| Newtype ((a, b) :->: x) Source # | |
| Newtype (() :->: a) Source # | |
| Newtype (Bool :->: a) Source # | |
| data Void :->: a Source # | |
Defined in Data.MemoTrie | |
| newtype Int16 :->: a Source # | |
Defined in Data.MemoTrie | |
| newtype Int32 :->: a Source # | |
Defined in Data.MemoTrie | |
| newtype Int64 :->: a Source # | |
Defined in Data.MemoTrie | |
| newtype Int8 :->: a Source # | |
Defined in Data.MemoTrie | |
| newtype Word16 :->: a Source # | |
Defined in Data.MemoTrie | |
| newtype Word32 :->: a Source # | |
Defined in Data.MemoTrie | |
| newtype Word64 :->: a Source # | |
Defined in Data.MemoTrie | |
| newtype Word8 :->: a Source # | |
Defined in Data.MemoTrie | |
| newtype Integer :->: a Source # | |
Defined in Data.MemoTrie | |
| newtype () :->: a Source # | |
Defined in Data.MemoTrie | |
| data Bool :->: x Source # | |
Defined in Data.MemoTrie | |
| newtype Char :->: a Source # | |
Defined in Data.MemoTrie | |
| newtype Int :->: a Source # | |
Defined in Data.MemoTrie | |
| newtype Word :->: a Source # | |
Defined in Data.MemoTrie | |
| data (Maybe a) :->: b Source # | |
Defined in Data.MemoTrie | |
| newtype [x] :->: a Source # | |
Defined in Data.MemoTrie | |
| data (Either a b) :->: x Source # | |
Defined in Data.MemoTrie | |
| newtype (U1 x) :->: b Source # | |
Defined in Data.MemoTrie | |
| data (V1 x) :->: b Source # | |
Defined in Data.MemoTrie | |
| newtype (a, b) :->: x Source # | |
Defined in Data.MemoTrie | |
| type O (Void :->: a) Source # | |
Defined in Data.MemoTrie | |
| type O (Either a b :->: x) Source # | |
Defined in Data.MemoTrie | |
| type O (Maybe a :->: x) Source # | |
Defined in Data.MemoTrie | |
| type O ((a, b) :->: x) Source # | |
Defined in Data.MemoTrie | |
| type O (() :->: a) Source # | |
Defined in Data.MemoTrie | |
| type O (Bool :->: a) Source # | |
Defined in Data.MemoTrie | |
| newtype (a, b, c) :->: x Source # | |
Defined in Data.MemoTrie | |
| newtype ((f :*: g) x) :->: b Source # | |
Defined in Data.MemoTrie | |
| newtype ((f :+: g) x) :->: b Source # | |
Defined in Data.MemoTrie | |
| newtype (K1 i a x) :->: b Source # | |
Defined in Data.MemoTrie | |
| newtype (M1 i t f x) :->: b Source # | |
Defined in Data.MemoTrie | |
(@.@) :: (HasTrie a, HasTrie b) => (b :->: c) -> (a :->: b) -> a :->: c infixr 9 Source #
Trie composition
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