source: project/release/4/srfi-4-comprehensions/srfi-4-comprehensions.scm @ 14560

Last change on this file since 14560 was 14560, checked in by Ivan Raikov, 10 years ago

srfi-4-comprehensions ported to Chicken 4

File size: 4.8 KB
Line 
1#|
2
3Derived (almost verbatim) from the code for vector comprehensions in
4the SRFI-42 reference implementation.  The copyright on that code is:
5
6Copyright (C) Sebastian Egner (2003). All Rights Reserved.
7
8Permission is hereby granted, free of charge, to any person obtaining
9a copy of this software and associated documentation
10files (the "Software"), to deal in the Software without restriction,
11including without limitation the rights to use, copy, modify, merge,
12publish, distribute, sublicense, and/or sell copies of the Software,
13and to permit persons to whom the Software is furnished to do so,
14subject to the following conditions:
15
16The above copyright notice and this permission notice shall be
17included in all copies or substantial portions of the Software.
18
19THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
20EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
21MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
22NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
23LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
24OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
25WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
26
27Modifications are copyright (C) Will M. Farr (2006), goverened by the
28same conditions.  Feel free to email comments to <farr@mit.edu>. 
29
30|#
31
32
33(module srfi-4-comprehensions 
34  (:s8vector :u8vector :s16vector :u16vector :s32vector 
35   :u32vector :f64vector :f32vector
36;   :s64vector :u64vector
37           
38   s8vector-ec u8vector-ec s16vector-ec u16vector-ec s32vector-ec 
39   u32vector-ec f64vector-ec f32vector-ec
40           
41   s8vector-of-length-ec u8vector-of-length-ec s16vector-of-length-ec 
42   u16vector-of-length-ec s32vector-of-length-ec 
43   u32vector-of-length-ec
44   f64vector-of-length-ec f32vector-of-length-ec)
45
46  (import scheme chicken srfi-4 srfi-13)
47  (require-extension srfi-42)
48
49  (define-syntax make/prefix
50    (lambda (x r c)
51      (let* ((pre-sym  (->string (cadr x)))
52             (vlength  (string->symbol (string-append pre-sym "vector-length")))
53             (vref     (string->symbol (string-append pre-sym "vector-ref")))
54             (vgen     (string->symbol (string-append ":" pre-sym "vector")))
55             (vfilter  (string->symbol (string-append "ec-:" pre-sym "vector-filter")))
56             (vmake    (string->symbol (string-append "make-" pre-sym "vector")))
57             (vset!    (string->symbol (string-append pre-sym "vector-set!")))
58             (list->v  (string->symbol (string-append "list->" pre-sym "vector")))
59             (v-ec     (string->symbol (string-append pre-sym "vector-ec")))
60             (v-of-length-ec (string->symbol (string-append pre-sym "vector-of-length-ec"))))
61        `(begin
62           (define-syntax ,vgen
63             (syntax-rules (index)
64               ((,vgen cc var arg)
65                (,vgen cc var (index i) arg) )
66               ((,vgen cc var (index i) arg)
67                (:do cc
68                     (let ((vec arg) (len 0)) 
69                       (set! len (,vlength vec)))
70                     ((i 0))
71                     (< i len)
72                     (let ((var (,vref vec i))))
73                     #t
74                     ((+ i 1)) ))
75               ((,vgen cc var (index i) arg1 arg2 arg (... ...))
76                (:parallel cc (,vgen cc var arg1 arg2 arg (... ...)) (:integers i)) )
77               ((,vgen cc var arg1 arg2 arg (... ...))
78                (:do cc
79                     (let ((vec #f)
80                           (len 0)
81                           (vecs (,vfilter (list arg1 arg2 arg (... ...)))) ))
82                     ((k 0))
83                     (if (< k len)
84                         #t
85                         (if (null? vecs)
86                             #f
87                             (begin (set! vec (car vecs))
88                                    (set! vecs (cdr vecs))
89                                    (set! len (,vlength vec))
90                                    (set! k 0)
91                                    #t )))
92                     (let ((var (,vref vec k))))
93                     #t
94                     ((+ k 1)) ))
95               ))
96
97           (define (,vfilter vecs)
98             (if (null? vecs)
99                 '()
100                 (if (zero? (,vlength (car vecs)))
101                     (,vfilter (cdr vecs))
102                     (cons (car vecs) (,vfilter (cdr vecs))) )))
103
104           (define-syntax ,v-ec
105             (syntax-rules ()
106               ((v-ec etc1 etc (... ...))
107                (,list->v (list-ec etc1 etc (... ...))) )))
108
109           (define-syntax ,v-of-length-ec
110             (syntax-rules (nested)
111               ((v-of-length-ec k (nested q1 (... ...)) q etc1 etc (... ...))
112                (v-of-length-ec k (nested q1 (... ...) q) etc1 etc (... ...)) )
113               ((v-of-length-ec k q1 q2             etc1 etc (... ...))
114                (v-of-length-ec k (nested q1 q2)    etc1 etc (... ...)) )
115               ((v-of-length-ec k expression)
116                (v-of-length-ec k (nested) expression) )
117               ((v-of-length-ec k qualifier expression)
118                (let ((len k))
119                  (let ((vec (,vmake len))
120                        (i 0) )
121                    (do-ec qualifier
122                           (if (< i len)
123                               (begin (,vset! vec i expression)
124                                      (set! i (+ i 1)) )
125                               (error "vector is too short for the comprehension") ))
126                    (if (= i len)
127                        vec
128                        (error "vector is too long for the comprehension") ))))))
129           )
130        )))
131
132   (make/prefix s8)
133   (make/prefix u8)
134   (make/prefix s16)
135   (make/prefix u16)
136   (make/prefix s32)
137   (make/prefix u32)
138   (make/prefix f32)
139   (make/prefix f64)
140;   (make/prefix s64)
141;   (make/prefix u64)
142)
Note: See TracBrowser for help on using the repository browser.