56906640af5502934b5561148552501ad6c6de71
[mes.git] / module / mes / scm.mes
1 ;;; -*-scheme-*-
2
3 ;;; Mes --- Maxwell Equations of Software
4 ;;; Copyright © 2016,2017 Jan Nieuwenhuizen <janneke@gnu.org>
5 ;;;
6 ;;; This file is part of Mes.
7 ;;;
8 ;;; Mes is free software; you can redistribute it and/or modify it
9 ;;; under the terms of the GNU General Public License as published by
10 ;;; the Free Software Foundation; either version 3 of the License, or (at
11 ;;; your option) any later version.
12 ;;;
13 ;;; Mes is distributed in the hope that it will be useful, but
14 ;;; WITHOUT ANY WARRANTY; without even the implied warranty of
15 ;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16 ;;; GNU General Public License for more details.
17 ;;;
18 ;;; You should have received a copy of the GNU General Public License
19 ;;; along with Mes.  If not, see <http://www.gnu.org/licenses/>.
20
21 ;;; Commentary:
22
23 ;;; scm.mes is loaded after base, quasiquote and let.  It provides
24 ;;; basic Scheme functions bringing Mes close to basic RRS Scheme (no
25 ;;; labels, processes, fluids or throw/catch).
26
27 ;;; Code:
28
29 (mes-use-module (mes let))
30
31 (define (cadddr x) (car (cdddr x)))
32
33 (define-macro (case val . args)
34   (if (null? args) #f
35       (let ((clause (car args)))
36         (let ((pred (car clause)))
37           (let ((body (cdr clause)))
38            (if (pair? pred) `(if ,(if (null? (cdr pred))
39                                       `(eq? ,val ',(car pred))
40                                       `(member ,val ',pred))
41                                  (begin ,@body)
42                                  (case ,val ,@(cdr args)))
43                `(begin ,@body)))))))
44
45 (define-macro (when expr . body)
46   `(if ,expr
47        ((lambda () ,@body))))
48
49 (define-macro (unless expr . body)
50   `(if (not ,expr)
51        ((lambda () ,@body))))
52
53 (define-macro (do init test . body)
54   `(let loop ((,(caar init) ,(cadar init)))
55      (when (not ,@test)
56        ,@body
57        (loop ,@(cddar init)))))
58
59 (define (for-each f l . r)
60   (if (pair? l) (if (null? r) (begin (f (car l)) (for-each f (cdr l)))
61                     (if (null? (cdr r)) (begin (f (car l) (caar r)) (for-each f (cdr l) (cdar r)))))))
62
63 (define core:error error)
64
65 (define (error who . rest)
66   (display "error:" (current-error-port))
67   (display who (current-error-port))
68   (display ":" (current-error-port))
69   (display rest (current-error-port))
70   (newline (current-error-port))
71   (display "exiting...\n" (current-error-port))
72   (core:error (if (symbol? who) who 'error) (cons who rest)))
73
74 (define (syntax-error message . rest)
75   (display "syntax-error:" (current-error-port))
76   (display message (current-error-port))
77   (display ":" (current-error-port))
78   (display rest (current-error-port))
79   (newline (current-error-port))
80   (core:error 'syntax-error (cons message rest)))
81
82 \f
83 (define integer? number?)
84
85 (define (eof-object? x)
86   (or (and (number? x) (= x -1))
87       (and (char? x) (eof-object? (char->integer x)))))
88
89 (define (peek-char)
90   (integer->char (peek-byte)))
91
92 (define (read-char)
93   (integer->char (read-byte)))
94
95 (define (unread-char c)
96   (unread-byte (char->integer c))
97   c)
98
99 (define (assq-set! alist key val)
100   (let ((entry (assq key alist)))
101     (cond (entry (set-cdr! entry val)
102                  alist)
103           (#t (cons (cons key val) alist)))))
104
105 (define (assq-ref alist key)
106   (let ((entry (assq key alist)))
107     (if entry (cdr entry)
108         #f)))
109
110 (define assv assq)
111 (define assv-ref assq-ref)
112
113 (define (assoc key alist)
114   (if (null? alist) #f ;; IF
115       (if (equal? key (caar alist)) (car alist)
116           (assoc key (cdr alist)))))
117
118 (define (assoc-ref alist key)
119   (let ((entry (assoc key alist)))
120     (if entry (cdr entry)
121         #f)))
122
123 (define (memq x lst)
124   (if (null? lst) #f ;; IF
125       (if (eq? x (car lst)) lst
126           (memq x (cdr lst)))))
127 (define memv memq)
128
129 (define (member x lst)
130   (if (null? lst) #f ;; IF
131       (if (equal? x (car lst)) lst
132           (member x (cdr lst)))))
133
134 \f
135 ;;; Lists
136 (define (make-list n . x)
137   (let ((fill (if (pair? x) (car x) *unspecified*)))
138     (let loop ((n n))
139       (if (= 0 n) '()
140           (cons fill (loop (- n 1)))))))
141
142 (define (list-ref lst k)
143   (let loop ((lst lst) (k k))
144     (if (= 0 k) (car lst)
145         (loop (cdr lst) (- k 1)))))
146
147 (define (list-set! lst k v)
148   (let loop ((lst lst) (k k))
149     (if (= 0 k) (set-car! lst v)
150         (loop (cdr lst) (- k 1)))))
151
152 (define (list-head x n)
153   (if (= 0 n) '()
154       (cons (car x) (list-head (cdr x) (- n 1)))))
155
156 (define (list-tail x n)
157   (if (= 0 n) x
158       (list-tail (cdr x) (- n 1))))
159
160 (define (last-pair lst)
161   (let loop ((lst lst))
162     (if (or (null? lst) (null? (cdr lst))) lst
163         (loop (cdr lst)))))
164
165 (define (iota n)
166   (if (<= n 0) '()
167       (append2 (iota (- n 1)) (list (- n 1)))))
168
169 (define (reverse lst)
170   (if (null? lst) '()
171       (append (reverse (cdr lst)) (cons (car lst) '()))))
172
173 (define (filter pred lst)
174   (let loop ((lst lst))
175     (if (null? lst) '()
176         (if (pred (car lst))
177             (cons (car lst) (loop (cdr lst)))
178             (loop (cdr lst))))))
179
180 (define (delete x lst)
181   (filter (lambda (e) (not (equal? e x))) lst))
182
183 (define (delq x lst)
184   (filter (lambda (e) (not (eq? e x))) lst))
185
186 \f
187 ;; Vector
188 (define (vector . rest) (list->vector rest))
189 (define c:make-vector make-vector)
190 (define (make-vector n . x)
191   (if (null? x) (c:make-vector n)
192       (list->vector (apply make-list (cons n x)))))
193
194 (define (vector-copy x)
195   (list->vector (vector->list x)))
196
197 \f
198 ;;; Strings/srfi-13
199 (define (string-length s)
200   (length (string->list s)))
201
202 (define (string-ref s k)
203   (list-ref (string->list s) k))
204
205 (define (string-set! s k v)
206   (list->string (list-set! (string->list s) k v)))
207
208 (define (substring s start . rest)
209   (let* ((end (and (pair? rest) (car rest)))
210          (lst (list-tail (string->list s) start)))
211     (list->string (if (not end) lst
212                       (list-head lst (- end start))))))
213
214 (define (string-prefix? prefix string)
215   (and
216    (>= (string-length string) (string-length prefix))
217    (equal? (substring string 0 (string-length prefix)) prefix)))
218
219 (define (string->number s . rest)
220   (let* ((radix (if (null? rest) 10 (car rest)))
221          (lst (string->list s))
222          (sign (if (char=? (car lst) #\-) -1 1))
223          (lst (if (= sign -1) (cdr lst) lst)))
224     (let loop ((lst lst) (n 0))
225       (if (null? lst) (* sign n)
226           (let ((i (char->integer (car lst))))
227             (loop (cdr lst) (+ (* n radix) (- i (if (<= i (char->integer #\9)) (char->integer #\0)
228                                                     (- (char->integer #\a) 10))))))))))
229
230 (define (number->string n . rest)
231   (let* ((radix (if (null? rest) 10 (car rest)))
232          (sign (if (< n 0) '(#\-) '())))
233     (let loop ((n (abs n)) (lst '()))
234       (let* ((i (remainder n radix))
235              (lst (cons (integer->char (+ i (if (<= i 10) (char->integer #\0)
236                                                 (- (char->integer #\a) 10)))) lst))
237              (n (quotient n radix)))
238         (if (= 0 n) (list->string (append sign lst))
239             (loop n lst))))))
240
241 \f
242 ;;; Symbols
243 (define (symbol-prefix? prefix symbol)
244   (string-prefix? (symbol->string prefix) (symbol->string symbol)))
245
246 (define (symbol-append . rest)
247   (string->symbol (apply string-append (map symbol->string rest))))
248
249 (define gensym
250   (let ((counter 0))
251     (lambda (. rest)
252       (let ((value (number->string counter)))
253         (set! counter (+ counter 1))
254         (string->symbol (string-append "g" value))))))
255
256 \f
257 ;;; Keywords
258 (define (keyword->symbol s)
259   (list->symbol (keyword->list s)))
260
261 \f
262 ;;; Characters
263 (define (char=? x y)
264   (and (char? x) (char? y)
265        (eq? x y)))
266
267 (define (char<? a b) (< (char->integer a) (char->integer b)))
268 (define (char>? a b) (> (char->integer a) (char->integer b)))
269 (define (char<=? a b) (<= (char->integer a) (char->integer b)))
270 (define (char>=? a b) (>= (char->integer a) (char->integer b)))
271
272 (define (char-alphabetic? x)
273   (and (char? x)
274        (let ((i (char->integer x)))
275         (or (and (>= i (char->integer #\A)) (<= i (char->integer #\Z)))
276             (and (>= i (char->integer #\a)) (<= i (char->integer #\z)))))))
277
278 (define (char-numeric? x)
279   (and (char? x)
280        (let ((i (char->integer x)))
281          (and (>= i (char->integer #\0)) (<= i (char->integer #\9))))))
282
283 \f
284 ;;; Math
285 (define quotient /)
286
287 (define (<= . rest)
288   (or (apply < rest)
289       (apply = rest)))
290
291 (define (>= . rest)
292   (or (apply > rest)
293       (apply = rest)))
294
295 (define (remainder x y)
296   (- x (* (quotient x y) y)))
297
298 (define (even? x)
299   (= 0 (remainder x 2)))
300
301 (define (odd? x)
302   (= 1 (remainder x 2)))
303
304 (define (negative? x)
305   (< x 0))
306
307 (define (positive? x)
308   (> x 0))
309
310 (define (zero? x)
311   (= x 0))
312
313 (define (1+ x)
314   (+ x 1))
315
316 (define (1- x)
317   (- x 1))
318
319 (define (abs x)
320   (if (>= x 0) x (- x)))
321
322 (define (expt x y)
323   (let loop ((s 1) (count y))
324     (if (= 0 count) s
325         (loop (* s x) (- count 1)))))
326
327 (define (max x . rest)
328   (if (null? rest) x
329       (let ((y (car rest)))
330         (let ((z (if (> x y) x y)))
331           (apply max (cons z (cdr rest)))))))
332
333 (define (min x . rest)
334   (if (null? rest) x
335       (let ((y (car rest)))
336         (let ((z (if (< x y) x y)))
337           (apply min (cons z (cdr rest)))))))