1
1
mirror of https://github.com/kanaka/mal.git synced 2024-09-20 18:18:51 +03:00
mal/impls/lib/memoize.mal
Joel Martin 8a19f60386 Move implementations into impls/ dir
- Reorder README to have implementation list after "learning tool"
  bullet.

- This also moves tests/ and libs/ into impls. It would be preferrable
  to have these directories at the top level.  However, this causes
  difficulties with the wasm implementations which need pre-open
  directories and have trouble with paths starting with "../../". So
  in lieu of that, symlink those directories to the top-level.

- Move the run_argv_test.sh script into the tests directory for
  general hygiene.
2020-02-10 23:50:16 -06:00

26 lines
933 B
Plaintext

;; Memoize any function.
;; Implement `memoize` using an atom (`mem`) which holds the memoized results
;; (hash-map from the arguments to the result). When the function is called,
;; the hash-map is checked to see if the result for the given argument was already
;; calculated and stored. If this is the case, it is returned immediately;
;; otherwise, it is calculated and stored in `mem`.
;; For recursive functions, take care to store the wrapper under the
;; same name than the original computation with an assignment like
;; `(def! f (memoize f))`, so that intermediate results are memorized.
;; Adapted from http://clojure.org/atoms
(def! memoize
(fn* [f]
(let* [mem (atom {})]
(fn* [& args]
(let* [key (str args)]
(if (contains? @mem key)
(get @mem key)
(let* [ret (apply f args)]
(do
(swap! mem assoc key ret)
ret))))))))