1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
(defmacro do-until [#* args]
(import
hy.model-patterns [whole FORM notpexpr dolike]
funcparserlib.parser [many])
(setv [body condition] (.parse
(whole [(many (notpexpr "until")) (dolike "until")])
args))
(setv g (hy.gensym))
`(do
(setv ~g True)
(while (or ~g (not (do ~@condition)))
~@body
(setv ~g False))))
(defn test-do-until []
(setv n 0 s "")
(do-until
(+= s "x")
(until (+= n 1) (>= n 3)))
(assert (= s "xxx"))
(do-until
(+= s "x")
(until (+= n 1) (>= n 3)))
(assert (= s "xxxx")))
(defmacro loop [#* args]
(import
hy.model-patterns [whole FORM sym SYM]
funcparserlib.parser [many])
(setv [loopers body] (.parse
(whole [
(many (|
(>> (+ (sym "while") FORM) (fn [x] [x]))
(+ (sym "for") SYM (sym "in") FORM)
(+ (sym "for") SYM (sym "from") FORM (sym "to") FORM)))
(sym "do")
(many FORM)])
args))
(defn f [loopers]
(setv head (if loopers (get loopers 0) None))
(setv tail (cut loopers 1 None))
(cond
(is head None)
`(do ~@body)
(= (len head) 1)
`(while ~@head ~(f tail))
(= (len head) 2)
`(for [~@head] ~(f tail))
True (do ; (= (len head) 3)
(setv [sym from to] head)
`(for [~sym (range ~from (+ ~to 1))] ~(f tail)))))
(f loopers))
(defn test-loop []
(setv l [])
(loop
for x in "abc"
do (.append l x))
(assert (= l ["a" "b" "c"]))
(setv l [] k 2)
(loop
while (> k 0)
for n from 1 to 3
for p in [k n (* 10 n)]
do (.append l p) (-= k 1))
(assert (= l [2 1 10 -1 2 20 -4 3 30])))
|
