2013-11-29 05:53:02 +01:00
|
|
|
;;; Hy anaphoric macros
|
2019-02-07 14:57:35 +01:00
|
|
|
;; Copyright 2019 the authors.
|
2017-04-27 23:16:57 +02:00
|
|
|
;; This file is part of Hy, which is free software licensed under the Expat
|
|
|
|
;; license. See the LICENSE.
|
|
|
|
|
2013-11-29 05:53:02 +01:00
|
|
|
;;; 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]
|
2017-02-04 02:03:06 +01:00
|
|
|
`(do
|
2017-10-26 20:53:08 +02:00
|
|
|
(setv 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."
|
2017-02-04 02:03:06 +01:00
|
|
|
(setv p (gensym))
|
|
|
|
`(do
|
|
|
|
(defn ~p [it] ~form)
|
2013-12-31 19:35:31 +01:00
|
|
|
(for [it ~lst]
|
2017-02-04 02:03:06 +01:00
|
|
|
(if (~p it)
|
2017-10-26 20:53:08 +02:00
|
|
|
~@body
|
|
|
|
(break)))))
|
2013-11-29 05:53:02 +01:00
|
|
|
|
|
|
|
|
|
|
|
(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."
|
2017-02-04 02:03:06 +01:00
|
|
|
(setv v (gensym 'v) f (gensym 'f))
|
|
|
|
`((fn []
|
2017-10-26 20:53:08 +02:00
|
|
|
(defn ~f [it] ~form)
|
|
|
|
(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."
|
2017-02-04 02:03:06 +01:00
|
|
|
(setv f (gensym))
|
|
|
|
`((fn []
|
2017-10-26 20:53:08 +02:00
|
|
|
(defn ~f [it] ~rep)
|
|
|
|
(for [it ~lst]
|
|
|
|
(if (~predfn it)
|
|
|
|
(yield (~f it))
|
|
|
|
(yield it))))))
|
2013-11-29 05:53:02 +01:00
|
|
|
|
|
|
|
|
|
|
|
(defmacro ap-filter [form lst]
|
2013-11-30 01:59:20 +01:00
|
|
|
"Yield elements returned when the predicate form evaluates to True."
|
2017-02-04 02:03:06 +01:00
|
|
|
(setv pred (gensym))
|
|
|
|
`((fn []
|
2017-10-26 20:53:08 +02:00
|
|
|
(defn ~pred [it] ~form)
|
|
|
|
(for [val ~lst]
|
|
|
|
(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"
|
|
|
|
`(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]
|
2017-02-04 02:03:06 +01:00
|
|
|
`(do
|
|
|
|
(setv ~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]
|
2017-02-04 02:03:06 +01:00
|
|
|
`(do
|
|
|
|
(setv ~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
|
|
|
|
|
|
|
|
2019-12-15 20:54:23 +01:00
|
|
|
(defmacro! ap-reduce [form o!lst &optional [initial-value None]]
|
2013-12-25 20:18:09 +01:00
|
|
|
"Anaphoric form of reduce, `acc' and `it' can be used for a form"
|
2017-02-04 02:03:06 +01:00
|
|
|
`(do
|
2019-12-15 20:54:23 +01:00
|
|
|
(setv acc ~(if (none? initial-value)
|
|
|
|
`(do
|
|
|
|
(setv ~g!lst (iter ~g!lst))
|
|
|
|
(next ~g!lst))
|
|
|
|
initial-value))
|
|
|
|
(ap-each ~g!lst (setv acc ~form))
|
2017-10-26 20:53:08 +02:00
|
|
|
acc))
|
2015-04-03 23:02:39 +02:00
|
|
|
|
|
|
|
|
2017-10-26 04:46:38 +02:00
|
|
|
(deftag % [expr]
|
|
|
|
"Makes an expression into a function with an implicit `%` parameter list.
|
2017-10-23 05:36:30 +02:00
|
|
|
|
2017-10-26 04:46:38 +02:00
|
|
|
A `%i` symbol designates the (1-based) ith parameter (such as `%3`).
|
|
|
|
Only the maximum `%i` determines the number of `%i` parameters--the
|
|
|
|
others need not appear in the expression.
|
2017-10-23 05:36:30 +02:00
|
|
|
`%*` and `%**` name the `&rest` and `&kwargs` parameters, respectively.
|
2017-10-26 04:46:38 +02:00
|
|
|
|
|
|
|
Nesting of `#%` forms is not recommended."
|
2018-06-07 21:18:49 +02:00
|
|
|
(setv %symbols (sfor a (flatten [expr])
|
|
|
|
:if (and (symbol? a)
|
|
|
|
(.startswith a '%))
|
|
|
|
a))
|
2017-10-26 04:46:38 +02:00
|
|
|
`(fn [;; generate all %i symbols up to the maximum found in expr
|
2018-06-07 21:18:49 +02:00
|
|
|
~@(gfor i (range 1 (-> (lfor a %symbols
|
|
|
|
:if (.isdigit (cut a 1))
|
|
|
|
(int (cut a 1)))
|
|
|
|
(or (, 0))
|
|
|
|
max
|
|
|
|
inc))
|
|
|
|
(HySymbol (+ "%" (str i))))
|
2017-10-26 04:46:38 +02:00
|
|
|
;; generate the &rest parameter only if '%* is present in expr
|
|
|
|
~@(if (in '%* %symbols)
|
2017-10-23 05:36:30 +02:00
|
|
|
'(&rest %*))
|
2017-10-26 04:46:38 +02:00
|
|
|
;; similarly for &kwargs and %**
|
|
|
|
~@(if (in '%** %symbols)
|
2017-10-23 05:36:30 +02:00
|
|
|
'(&kwargs %**))]
|
2017-10-26 04:46:38 +02:00
|
|
|
~expr))
|
|
|
|
|