-
-
Save amalloy/1126157 to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
(defmacro fold | |
"sugar on reduce, similar to how 'for' is a nicer version of 'map'. | |
init is a symbol that evaluates to a value (local or var) or a vector of a symbol and an initial value. In the loop, init is bound to the result of the previous loop. | |
Binding can take any arguments supported by for, but should only bind one variable. | |
" | |
;; (fold r [i (range 10)] ;; existing symbol | |
;; (+ r i)) | |
;; (fold [r 0] [i (range 10)] ;; define new local | |
;; (+ r i)) | |
[initial binding & body] | |
(let [[init-sym init-val] (if (symbol? initial) | |
[initial initial] | |
initial)] | |
`(let [~init-sym ~init-val] | |
(reduce (fn [~init-sym ~(first binding)] | |
~@body) ~init-sym ~@(rest binding))))) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment