{-# LANGUAGE RankNTypes #-} {-# LANGUAGE Safe #-} {-# LANGUAGE TypeOperators #-} {-| Module : Data.StableMemo Copyright : (c) 2012, 2014 Jake McArthur License : MIT Maintainer : Jake McArthur Stability : provisional Portability : GHC This module provides a small set of functions for memoization. Whereas most memo combinators memoize based on equality, these do it based on whether the exact same argument has been passed to the function before (that is, is the same argument in memory). * They only evaluate keys to WHNF. * Relative to value-base memoization, this can be more suitable for recursive functions over graphs with cycles. * These don't retain the keys they have seen so far, which allows mappings to be garbage collected if they will no longer be used. Finalizers are put in place to remove the corresponding entries from the memo table if this happens. * "Data.StableMemo.Weak" provides an alternative set of combinators that also avoid retaining the results of the function, only reusing results if they have not yet been garbage collected. * There is no type class constraint on the function's argument. These will not work for arguments which happen to have the same value but are not the same heap object. This rules out many candidates for memoization, such as the most common example, the naive Fibonacci implementation whose domain is machine Ints; it can still be made to work for some domains, though, such as the lazy naturals. > data Nat = Succ Nat | Zero > > fib :: Nat -> Integer > fib = memo fib' > where fib' Zero = 0 > fib' (Succ Zero) = 1 > fib' (Succ n1@(Succ n2)) = fib n1 + fib n2 Below is an implementation of map that preserves sharing of the spine for cyclic lists. It should even be safe to use this on arbitrarily long, acyclic lists since as long as the garbage collector is chasing you, the size of the memo table should stay under control, too. > map :: (a -> b) -> [a] -> [b] > map f = go > where go = memo map' > map' [] = [] > map' (x:xs) = f x : go xs -} module Data.StableMemo (memo, memo2, memo3, (-->) (), memoPoly) where import Control.Applicative import Data.Proxy import qualified Data.StableMemo.Internal as Internal import Data.StableMemo.Internal ((-->) ()) -- | Memoize a function with support for a certain form of polymorphic -- recursion. memoPoly :: (f --> g) -> (f --> g) {-# NOINLINE memoPoly #-} memoPoly = Internal.memo (Proxy :: Proxy Internal.Strong) -- | Memoize a unary function. memo :: (a -> b) -> (a -> b) memo f = getConst . memoPoly (Const . f . getConst) . Const -- | Curried memoization to share partial evaluation memo2 :: (a -> b -> c) -> (a -> b -> c) memo2 f = memo (memo . f) -- | Curried memoization to share partial evaluation memo3 :: (a -> b -> c -> d) -> (a -> b -> c -> d) memo3 f = memo (memo2 . f)