2013-11-29 05:53:02 +01:00
|
|
|
;;; Hy anaphoric macros
|
|
|
|
;;
|
|
|
|
;; Copyright (c) 2013 James King <james@agentultra.com>
|
2013-12-25 18:11:25 +01:00
|
|
|
;; 2013 Paul R. Tagliamonte <tag@pault.ag>
|
2013-12-25 20:18:09 +01:00
|
|
|
;; 2013 Abhishek L <abhishek.lekshmanan@gmail.com>
|
2013-11-29 05:53:02 +01:00
|
|
|
;;
|
|
|
|
;; Permission is hereby granted, free of charge, to any person obtaining a
|
|
|
|
;; copy of this software and associated documentation files (the "Software"),
|
|
|
|
;; to deal in the Software without restriction, including without limitation
|
|
|
|
;; the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
|
|
;; and/or sell copies of the Software, and to permit persons to whom the
|
|
|
|
;; Software is furnished to do so, subject to the following conditions:
|
|
|
|
;;
|
|
|
|
;; The above copyright notice and this permission notice shall be included in
|
|
|
|
;; all copies or substantial portions of the Software.
|
|
|
|
;;
|
|
|
|
;; THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
|
;; IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
|
;; FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
|
|
|
;; THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
|
;; LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
|
|
;; FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
|
|
;; DEALINGS IN THE SOFTWARE.
|
|
|
|
;;
|
|
|
|
;;; These macros make writing functional programs more concise
|
|
|
|
|
|
|
|
|
2015-06-17 00:11:55 +02:00
|
|
|
(defmacro ap-if [test-form then-form &optional else-form]
|
|
|
|
`(let [it ~test-form]
|
|
|
|
(if it ~then-form ~else-form)))
|
2013-12-25 18:11:25 +01:00
|
|
|
|
|
|
|
|
2013-11-29 05:53:02 +01:00
|
|
|
(defmacro ap-each [lst &rest body]
|
2013-11-30 01:59:20 +01:00
|
|
|
"Evaluate the body form for each element in the list."
|
2013-12-31 19:35:31 +01:00
|
|
|
`(for [it ~lst] ~@body))
|
2013-11-29 05:53:02 +01:00
|
|
|
|
|
|
|
|
2013-11-30 01:59:20 +01:00
|
|
|
(defmacro ap-each-while [lst form &rest body]
|
2015-12-08 14:43:47 +01:00
|
|
|
"Evaluate the body form for each element in the list while the
|
2013-11-30 01:59:20 +01:00
|
|
|
predicate form evaluates to True."
|
2015-08-17 09:07:32 +02:00
|
|
|
`(let [p (lambda [it] ~form)]
|
2013-12-31 19:35:31 +01:00
|
|
|
(for [it ~lst]
|
2013-11-29 05:53:02 +01:00
|
|
|
(if (p it)
|
|
|
|
~@body
|
|
|
|
(break)))))
|
|
|
|
|
|
|
|
|
|
|
|
(defmacro ap-map [form lst]
|
2013-11-30 01:59:20 +01:00
|
|
|
"Yield elements evaluated in the form for each element in the list."
|
2015-08-17 09:07:32 +02:00
|
|
|
(let [v (gensym 'v)
|
|
|
|
f (gensym 'f)]
|
|
|
|
`(let [~f (lambda [it] ~form)]
|
2014-04-18 02:45:05 +02:00
|
|
|
(for [~v ~lst]
|
|
|
|
(yield (~f ~v))))))
|
2013-11-29 05:53:02 +01:00
|
|
|
|
|
|
|
|
|
|
|
(defmacro ap-map-when [predfn rep lst]
|
2013-11-30 01:59:20 +01:00
|
|
|
"Yield elements evaluated for each element in the list when the
|
|
|
|
predicate function returns True."
|
2015-08-17 09:07:32 +02:00
|
|
|
`(let [f (lambda [it] ~rep)]
|
2013-12-31 19:35:31 +01:00
|
|
|
(for [it ~lst]
|
2013-11-30 16:30:42 +01:00
|
|
|
(if (~predfn it)
|
2013-11-29 05:53:02 +01:00
|
|
|
(yield (f it))
|
|
|
|
(yield it)))))
|
|
|
|
|
|
|
|
|
|
|
|
(defmacro ap-filter [form lst]
|
2013-11-30 01:59:20 +01:00
|
|
|
"Yield elements returned when the predicate form evaluates to True."
|
2015-08-17 09:07:32 +02:00
|
|
|
`(let [pred (lambda [it] ~form)]
|
2013-12-31 19:35:31 +01:00
|
|
|
(for [val ~lst]
|
2013-11-29 05:53:02 +01:00
|
|
|
(if (pred val)
|
|
|
|
(yield val)))))
|
2013-12-25 20:18:09 +01:00
|
|
|
|
|
|
|
|
|
|
|
(defmacro ap-reject [form lst]
|
|
|
|
"Yield elements returned when the predicate form evaluates to False"
|
|
|
|
`(ap-filter (not ~form) ~lst))
|
|
|
|
|
|
|
|
|
|
|
|
(defmacro ap-dotimes [n &rest body]
|
|
|
|
"Execute body for side effects `n' times, with it bound from 0 to n-1"
|
|
|
|
(unless (numeric? n)
|
|
|
|
(raise (TypeError (.format "{0!r} is not a number" n))))
|
|
|
|
`(ap-each (range ~n) ~@body))
|
|
|
|
|
|
|
|
|
|
|
|
(defmacro ap-first [predfn lst]
|
|
|
|
"Yield the first element that passes `predfn`"
|
2014-05-15 20:11:24 +02:00
|
|
|
(with-gensyms [n]
|
2015-08-17 09:07:32 +02:00
|
|
|
`(let [~n None]
|
2014-05-15 20:11:24 +02:00
|
|
|
(ap-each ~lst (when ~predfn (setv ~n it) (break)))
|
|
|
|
~n)))
|
2013-12-25 20:18:09 +01:00
|
|
|
|
|
|
|
|
|
|
|
(defmacro ap-last [predfn lst]
|
|
|
|
"Yield the last element that passes `predfn`"
|
2014-05-15 20:11:24 +02:00
|
|
|
(with-gensyms [n]
|
2015-08-17 09:07:32 +02:00
|
|
|
`(let [~n None]
|
2014-05-15 20:11:24 +02:00
|
|
|
(ap-each ~lst (none? ~n)
|
|
|
|
(when ~predfn
|
|
|
|
(setv ~n it)))
|
|
|
|
~n)))
|
2013-12-25 20:18:09 +01:00
|
|
|
|
|
|
|
|
|
|
|
(defmacro ap-reduce [form lst &optional [initial-value None]]
|
|
|
|
"Anaphoric form of reduce, `acc' and `it' can be used for a form"
|
|
|
|
(if (none? initial-value)
|
2015-08-17 09:07:32 +02:00
|
|
|
`(let [acc (car ~lst)]
|
2013-12-25 20:18:09 +01:00
|
|
|
(ap-each (cdr ~lst) (setv acc ~form))
|
|
|
|
acc)
|
2015-08-17 09:07:32 +02:00
|
|
|
`(let [acc ~initial-value]
|
2013-12-25 20:18:09 +01:00
|
|
|
(ap-each ~lst (setv acc ~form))
|
|
|
|
acc)))
|
2015-04-03 23:02:39 +02:00
|
|
|
|
|
|
|
|
|
|
|
(defmacro ap-pipe [var &rest forms]
|
|
|
|
"Pushes a value through several forms.
|
|
|
|
(Anaphoric version of -> and ->>)"
|
|
|
|
(if (empty? forms) var
|
2015-08-17 09:07:32 +02:00
|
|
|
`(ap-pipe (let [it ~var] ~(first forms)) ~@(rest forms))))
|
2015-04-03 23:02:39 +02:00
|
|
|
|
|
|
|
|
|
|
|
(defmacro ap-compose [&rest forms]
|
|
|
|
"Returns a function which is the composition of several forms."
|
|
|
|
`(fn [var] (ap-pipe var ~@forms)))
|
2015-08-12 02:11:33 +02:00
|
|
|
|
2015-08-12 16:45:43 +02:00
|
|
|
(defmacro xi [&rest body]
|
2015-08-12 02:11:33 +02:00
|
|
|
"Returns a function with parameters implicitly determined by the presence in
|
|
|
|
the body of xi parameters. An xi symbol designates the ith parameter
|
|
|
|
(1-based, e.g. x1, x2, x3, etc.), or all remaining parameters for xi itself.
|
|
|
|
This is not a replacement for lambda. The xi forms cannot be nested. "
|
|
|
|
(setv flatbody (flatten body))
|
|
|
|
`(lambda [;; generate all xi symbols up to the maximum found in body
|
|
|
|
~@(genexpr (HySymbol (+ "x"
|
|
|
|
(str i)))
|
|
|
|
[i (range 1
|
|
|
|
;; find the maximum xi
|
2015-08-12 03:35:22 +02:00
|
|
|
(inc (max (+ (list-comp (int (cdr a))
|
|
|
|
[a flatbody]
|
|
|
|
(and (symbol? a)
|
|
|
|
(.startswith a 'x)
|
|
|
|
(.isdigit (cdr a))))
|
|
|
|
[0]))))])
|
2015-12-08 14:43:47 +01:00
|
|
|
;; generate the &rest parameter only if 'xi is present in body
|
2015-08-12 02:11:33 +02:00
|
|
|
~@(if (in 'xi flatbody)
|
|
|
|
'(&rest xi)
|
|
|
|
'())]
|
2015-08-12 16:45:43 +02:00
|
|
|
(~@body)))
|
2015-08-12 02:11:33 +02:00
|
|
|
|