source: project/chicken/branches/prerelease/synrules.scm @ 15101

Last change on this file since 15101 was 15101, checked in by felix winkelmann, 10 years ago

merged trunk changes from 14491:15100 into prerelease branch

File size: 11.1 KB
Line 
1;; Copyright (c) 1993-2001 by Richard Kelsey and Jonathan Rees.
2;; All rights reserved.
3
4;; Redistribution and use in source and binary forms, with or without
5;; modification, are permitted provided that the following conditions
6;; are met:
7;; 1. Redistributions of source code must retain the above copyright
8;;    notice, this list of conditions and the following disclaimer.
9;; 2. Redistributions in binary form must reproduce the above copyright
10;;    notice, this list of conditions and the following disclaimer in the
11;;    documentation and/or other materials provided with the distribution.
12;; 3. The name of the authors may not be used to endorse or promote products
13;;    derived from this software without specific prior written permission.
14
15;; THIS SOFTWARE IS PROVIDED BY THE AUTHORS ``AS IS'' AND ANY EXPRESS OR
16;; IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17;; OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18;; IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY DIRECT, INDIRECT,
19;; INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20;; NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21;; DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22;; THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23;; (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24;; THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25
26; The syntax-rules macro (new in R5RS)
27
28;;; [Hacked slightly by Taylor R. Campbell to make it work in his
29;;; macro expander `riaxpander'.]
30
31;; [Hacked even more by Felix L. Winkelmann to make it work in his
32;; Hi-Lo expander]
33
34; Example:
35;
36; (define-syntax or
37;   (syntax-rules ()
38;     ((or)          #f)
39;     ((or e)        e)
40;     ((or e1 e ...) (let ((temp e1))
41;                      (if temp temp (or e ...))))))
42
43
44(##sys#extend-macro-environment
45 'syntax-rules
46 '()
47 (##sys#er-transformer
48  (lambda (exp r c)
49    (##sys#check-syntax 'syntax-rules exp '#(_ 2))
50    (let ((subkeywords (cadr exp))
51          (rules (cddr exp))
52          (ellipsis '...))
53      (when (symbol? subkeywords)
54        (##sys#check-syntax 'syntax-rules exp '(_ _ list . #(_ 0)))
55        (set! ellipsis subkeywords)
56        (set! subkeywords (car rules))
57        (set! rules (cdr rules)))
58      (##sys#process-syntax-rules ellipsis rules subkeywords r c)))))
59
60
61(define (##sys#process-syntax-rules ellipsis rules subkeywords r c)
62
63  (define %append '##sys#append)
64  (define %apply '##sys#apply)
65  (define %and (r 'and))
66  (define %car '##sys#car)
67  (define %cdr '##sys#cdr)
68  (define %vector? '##sys#vector?)
69  (define %vector-length '##sys#vector-length)
70  (define %vector-ref '##sys#vector-ref)
71  (define %vector->list '##sys#vector->list)
72  (define %list->vector '##sys#list->vector)
73  (define %>= '##sys#>=)
74  (define %= '##sys#=)
75  (define %+ '##sys#+)
76  (define %i (r 'i))
77  (define %compare (r 'compare))
78  (define %cond (r 'cond))
79  (define %cons '##sys#cons)
80  (define %else (r 'else))
81  (define %eq? '##sys#eq?)
82  (define %equal? '##sys#equal?)
83  (define %input (r 'input))
84  (define %l (r 'l))
85  (define %lambda (r 'lambda))
86  (define %let (r 'let))
87  (define %let* (r 'let*))
88  (define %list? '##sys#list?)
89  (define %list (r 'list))
90  (define %loop (r 'loop))
91  (define %map1 '##sys#map)
92  (define %map '##sys#map-n)
93  (define %null? '##sys#null?)
94  (define %or (r 'or))
95  (define %pair? '##sys#pair?)
96  (define %quote (r 'quote))
97  (define %rename (r 'rename))
98  (define %tail (r 'tail))
99  (define %temp (r 'temp))
100  (define %syntax-error '##sys#syntax-error-hook)
101  (define %ellipsis (r ellipsis))
102
103  (define (ellipsis? x)
104    (c x %ellipsis))
105
106  (define (make-transformer rules)
107    `(,%lambda (,%input ,%rename ,%compare)
108               (,%let ((,%tail (,%cdr ,%input)))
109                      (,%cond ,@(map process-rule rules)
110                              (,%else 
111                               (##sys#syntax-rules-mismatch ,%input))))))
112
113  (define (process-rule rule)
114    (if (and (pair? rule)
115             (pair? (cdr rule))
116             (null? (cddr rule)))
117        (let ((pattern (cdar rule))
118              (template (cadr rule)))
119          `((,%and ,@(process-match %tail pattern))
120            (,%let* ,(process-pattern pattern
121                                      %tail
122                                      (lambda (x) x))
123                    ,(process-template template
124                                       0
125                                       (meta-variables pattern 0 '())))))
126        (##sys#syntax-error-hook "ill-formed syntax rule" rule)))
127
128  ;; Generate code to test whether input expression matches pattern
129
130  (define (process-match input pattern)
131    (cond ((symbol? pattern)
132           (if (memq pattern subkeywords)
133               `((,%compare ,input (,%rename (syntax ,pattern))))
134               `()))
135          ((segment-pattern? pattern)
136           (process-segment-match input (car pattern)))
137          ((pair? pattern)
138           `((,%let ((,%temp ,input))
139                    (,%and (,%pair? ,%temp)
140                           ,@(process-match `(,%car ,%temp) (car pattern))
141                           ,@(process-match `(,%cdr ,%temp) (cdr pattern))))))
142          ((vector? pattern)
143           (process-vector-match input pattern))
144          ((or (null? pattern) (boolean? pattern) (char? pattern))
145           `((,%eq? ,input ',pattern)))
146          (else
147           `((,%equal? ,input ',pattern)))))
148
149  (define (process-segment-match input pattern)
150    (let ((conjuncts (process-match `(,%car ,%l) pattern)))
151      (if (null? conjuncts)
152          `((,%list? ,input))           ;+++
153          `((,%let ,%loop ((,%l ,input))
154                   (,%or (,%null? ,%l)
155                         (,%and (,%pair? ,%l)
156                                ,@conjuncts
157                                (,%loop (,%cdr ,%l)))))))))
158
159   (define (process-vector-match input pattern)
160     (let* ((len (vector-length pattern))
161            (segment? (and (>= len 2)
162                           (ellipsis? (vector-ref pattern (- len 1))))))
163       `((,%let ((,%temp ,input))
164          (,%and (,%vector? ,%temp)
165                 ,(if segment?
166                      `(,%>= (,%vector-length ,%temp) ,(- len 2))
167                      `(,%= (,%vector-length ,%temp) ,len))
168                 ,@(let lp ((i 0))
169                     (cond
170                      ((>= i len)
171                       '())
172                      ((and (= i (- len 2)) segment?)
173                       `((,%let ,%loop ((,%i ,i))
174                            (,%or (,%>= ,%i ,len)
175                                  (,%and ,@(process-match
176                                            `(,%vector-ref ,%temp ,%i)
177                                            (vector-ref pattern (- len 2)))
178                                         (,%loop (,%+ ,%i 1)))))))
179                      (else
180                       (append (process-match `(,%vector-ref ,%temp ,i)
181                                              (vector-ref pattern i))
182                               (lp (+ i 1)))))))))))
183 
184  ;; Generate code to take apart the input expression
185  ;; This is pretty bad, but it seems to work (can't say why).
186
187  (define (process-pattern pattern path mapit)
188    (cond ((symbol? pattern)
189           (if (memq pattern subkeywords)
190               '()
191               (list (list pattern (mapit path)))))
192          ((segment-pattern? pattern)
193           (process-pattern (car pattern)
194                            %temp
195                            (lambda (x) ;temp is free in x
196                              (mapit (if (eq? %temp x)
197                                         path ;+++
198                                         `(,%map1 (,%lambda (,%temp) ,x)
199                                                  ,path))))))
200          ((pair? pattern)
201           (append (process-pattern (car pattern) `(,%car ,path) mapit)
202                   (process-pattern (cdr pattern) `(,%cdr ,path) mapit)))
203          ((vector? pattern)
204           (let* ((len (vector-length pattern))
205                  (segment? (and (>= len 2)
206                                 (ellipsis? (vector-ref pattern (- len 1))))))
207             (if segment?
208                 (process-pattern (vector->list pattern) 
209                                  `(,%vector->list ,path)
210                                  mapit)
211                 (let lp ((i 0))
212                   (cond
213                    ((>= i len)
214                     '())
215                    (else
216                     (append (process-pattern (vector-ref pattern i)
217                                              `(,%vector-ref ,path ,i)
218                                              mapit)
219                             (lp (+ i 1)))))))))
220          (else '())))
221
222  ;; Generate code to compose the output expression according to template
223
224  (define (process-template template dim env)
225    (cond ((symbol? template)
226           (let ((probe (assq template env)))
227             (if probe
228                 (if (<= (cdr probe) dim)
229                     template
230                     (##sys#syntax-error-hook "template dimension error (too few ellipses?)"
231                                              template))
232                 `(,%rename (syntax ,template)))))
233          ((segment-template? template)
234           (let* ((depth (segment-depth template))
235                  (seg-dim (+ dim depth))
236                  (vars
237                   (free-meta-variables (car template) seg-dim env '())))
238             (if (null? vars)
239                 (##sys#syntax-error-hook "too many ellipses" template)
240                 (let* ((x (process-template (car template)
241                                             seg-dim
242                                             env))
243                        (gen (if (and (pair? vars)
244                                      (null? (cdr vars))
245                                      (symbol? x)
246                                      (eq? x (car vars)))
247                                 x      ;+++
248                                 `(,%map (,%lambda ,vars ,x)
249                                         ,@vars)))
250                        (gen (do ((d depth (- d 1))
251                                  (gen gen `(,%apply ,%append ,gen)))
252                                 ((= d 1)
253                                  gen))))
254                   (if (null? (segment-tail template))
255                       gen              ;+++
256                       `(,%append ,gen ,(process-template (segment-tail template)
257                                                          dim env)))))))
258          ((pair? template)
259           `(,%cons ,(process-template (car template) dim env)
260                    ,(process-template (cdr template) dim env)))
261          ((vector? template)
262           `(,%list->vector
263             ,(process-template (vector->list template) dim env)))
264          (else
265           `(,%quote ,template))))
266
267  ;; Return an association list of (var . dim)
268
269  (define (meta-variables pattern dim vars)
270    (cond ((symbol? pattern)
271           (if (memq pattern subkeywords)
272               vars
273               (cons (cons pattern dim) vars)))
274          ((segment-pattern? pattern)
275           (meta-variables (car pattern) (+ dim 1) vars))
276          ((pair? pattern)
277           (meta-variables (car pattern) dim
278                           (meta-variables (cdr pattern) dim vars)))
279          ((vector? pattern)
280           (meta-variables (vector->list pattern) dim vars))
281          (else vars)))
282
283  ;; Return a list of meta-variables of given higher dim
284
285  (define (free-meta-variables template dim env free)
286    (cond ((symbol? template)
287           (if (and (not (memq template free))
288                    (let ((probe (assq template env)))
289                      (and probe (>= (cdr probe) dim))))
290               (cons template free)
291               free))
292          ((segment-template? template)
293           (free-meta-variables (car template)
294                                dim env
295                                (free-meta-variables (cddr template)
296                                                     dim env free)))
297          ((pair? template)
298           (free-meta-variables (car template)
299                                dim env
300                                (free-meta-variables (cdr template)
301                                                     dim env free)))
302          ((vector? template)
303           (free-meta-variables (vector->list template) dim env free))
304          (else free)))
305
306  (define (segment-pattern? pattern)
307    (and (segment-template? pattern)
308         (or (null? (cddr pattern))
309             (##sys#syntax-error-hook "segment matching not implemented" pattern))))
310
311  (define (segment-template? pattern)
312    (and (pair? pattern)
313         (pair? (cdr pattern))
314         (ellipsis? (cadr pattern))))
315
316  ;; Count the number of `...'s in PATTERN.
317
318  (define (segment-depth pattern)
319    (if (segment-template? pattern)
320        (+ 1 (segment-depth (cdr pattern)))
321        0))
322
323  ;; Get whatever is after the `...'s in PATTERN.
324
325  (define (segment-tail pattern)
326    (let loop ((pattern (cdr pattern)))
327      (if (and (pair? pattern)
328               (ellipsis? (car pattern)))
329          (loop (cdr pattern))
330          pattern)))
331
332  (make-transformer rules))
Note: See TracBrowser for help on using the repository browser.