32fc13f617919063ff1b3494064aa357d41fce1a
[mes.git] / module / mes / scm.mes
1 ;;; -*-scheme-*-
2
3 ;;; Mes --- Maxwell Equations of Software
4 ;;; Copyright © 2016,2017,2018 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 (if (not (defined? 'peek-char))
86     (define (peek-char)
87       (integer->char (peek-byte))))
88
89 (if (not (defined? 'read-char))
90     (define (read-char)
91       (integer->char (read-byte))))
92
93 (if (not (defined? 'unread-char))
94     (define (unread-char c)
95       (integer->char (unread-byte (char->integer c)))))
96
97 (define (assq-set! alist key val)
98   (let ((entry (assq key alist)))
99     (if (not entry) (acons key val alist)
100         (let ((entry (set-cdr! entry val)))
101           alist))))
102
103 (define (assq-ref alist key)
104   (let ((entry (assq key alist)))
105     (if entry (cdr entry)
106         #f)))
107
108 (define assv assq)
109 (define assv-ref assq-ref)
110
111 (define (assoc key alist)
112   (if (null? alist) #f
113       (if (equal? key (caar alist)) (car alist)
114           (assoc key (cdr alist)))))
115
116 (define (assoc-ref alist key)
117   (let ((entry (assoc key alist)))
118     (if entry (cdr entry)
119         #f)))
120
121 (define (assoc-set! alist key value)
122   (let ((entry (assoc key alist)))
123     (if (not entry) (acons key value alist)
124         (let ((entry (set-cdr! entry value)))
125           alist))))
126
127 (define memv memq)
128
129 \f
130 ;;; Lists
131 (define (make-list n . x)
132   (let ((fill (if (pair? x) (car x) *unspecified*)))
133     (let loop ((n n))
134       (if (= 0 n) '()
135           (cons fill (loop (- n 1)))))))
136
137 (define (list-ref lst k)
138   (let loop ((lst lst) (k k))
139     (if (= 0 k) (car lst)
140         (loop (cdr lst) (- k 1)))))
141
142 (define (list-set! lst k v)
143   (let loop ((lst lst) (k k))
144     (if (= 0 k) (set-car! lst v)
145         (loop (cdr lst) (- k 1)))))
146
147 (define (list-head x n)
148   (if (= 0 n) '()
149       (cons (car x) (list-head (cdr x) (- n 1)))))
150
151 (define (list-tail x n)
152   (if (= 0 n) x
153       (list-tail (cdr x) (- n 1))))
154
155 (define (last-pair lst)
156   (let loop ((lst lst))
157     (if (or (null? lst) (null? (cdr lst))) lst
158         (loop (cdr lst)))))
159
160 (define (iota n)
161   (if (<= n 0) '()
162       (append2 (iota (- n 1)) (list (- n 1)))))
163
164 (define (reverse lst)
165   (if (null? lst) '()
166       (append (reverse (cdr lst)) (cons (car lst) '()))))
167
168 (define (filter pred lst)
169   (let loop ((lst lst))
170     (if (null? lst) '()
171         (if (pred (car lst))
172             (cons (car lst) (loop (cdr lst)))
173             (loop (cdr lst))))))
174
175 (define (delete x lst)
176   (filter (lambda (e) (not (equal? e x))) lst))
177
178 (define (delq x lst)
179   (filter (lambda (e) (not (eq? e x))) lst))
180
181 (define (compose proc . rest)
182   (if (null? rest) proc
183       (lambda args
184         (proc (apply (apply compose rest) args)))))
185
186 \f
187 ;; Vector
188 (define (vector . rest) (list->vector rest))
189 (define (make-vector n . x)
190   (if (null? x) (core:make-vector n)
191       (list->vector (apply make-list (cons n x)))))
192
193 (define (vector-copy x)
194   (list->vector (vector->list x)))
195
196 \f
197 ;;; Strings/srfi-13
198 (define (make-string n . fill)
199   (list->string (apply make-list n fill)))
200
201 (define (string-length s)
202   (length (string->list s)))
203
204 (define (string-ref s k)
205   (list-ref (string->list s) k))
206
207 (define (string-set! s k v)
208   (list->string (list-set! (string->list s) k v)))
209
210 (define (substring s start . rest)
211   (let* ((end (and (pair? rest) (car rest)))
212          (lst (list-tail (string->list s) start)))
213     (list->string (if (not end) lst
214                       (list-head lst (- end start))))))
215
216 (define (string-prefix? prefix string)
217   (let ((length (string-length string))
218         (prefix-length (string-length prefix)))
219   (and
220    (>= length prefix-length)
221    (equal? (substring string 0 prefix-length) prefix))))
222
223 (define (string-suffix? suffix string)
224   (let ((length (string-length string))
225         (suffix-length (string-length suffix)))
226     (and
227      (>= length suffix-length)
228      (equal? (substring string (- length suffix-length)) suffix))))
229
230 (define (string->number s . rest)
231   (let* ((radix (if (null? rest) 10 (car rest)))
232          (lst (string->list s))
233          (sign (if (char=? (car lst) #\-) -1 1))
234          (lst (if (= sign -1) (cdr lst) lst)))
235     (let loop ((lst lst) (n 0))
236       (if (null? lst) (* sign n)
237           (let ((i (char->integer (car lst))))
238             (loop (cdr lst) (+ (* n radix) (- i (if (<= i (char->integer #\9)) (char->integer #\0)
239                                                     (- (char->integer #\a) 10))))))))))
240
241 (define (number->string n . rest)
242   (let* ((radix (if (null? rest) 10 (car rest)))
243          (sign (if (< n 0) '(#\-) '())))
244     (let loop ((n (abs n)) (lst '()))
245       (let* ((i (remainder n radix))
246              (lst (cons (integer->char (+ i (if (< i 10) (char->integer #\0)
247                                                 (- (char->integer #\a) 10)))) lst))
248              (n (quotient n radix)))
249         (if (= 0 n) (list->string (append sign lst))
250             (loop n lst))))))
251
252 \f
253 ;;; Symbols
254 (define (symbol-prefix? prefix symbol)
255   (string-prefix? (symbol->string prefix) (symbol->string symbol)))
256
257 (define (symbol-append . rest)
258   (string->symbol (apply string-append (map symbol->string rest))))
259
260 (define gensym
261   (let ((counter 0))
262     (lambda (. rest)
263       (let ((value (number->string counter)))
264         (set! counter (+ counter 1))
265         (string->symbol (string-append "g" value))))))
266
267 \f
268 ;;; Keywords
269 (define (keyword->symbol s)
270   (list->symbol (keyword->list s)))
271
272 \f
273 ;;; Characters
274 (define (char=? x y)
275   (and (char? x) (char? y)
276        (eq? x y)))
277
278 (define (char<? a b) (< (char->integer a) (char->integer b)))
279 (define (char>? a b) (> (char->integer a) (char->integer b)))
280 (define (char<=? a b) (<= (char->integer a) (char->integer b)))
281 (define (char>=? a b) (>= (char->integer a) (char->integer b)))
282
283 (define (char-alphabetic? x)
284   (and (char? x)
285        (let ((i (char->integer x)))
286         (or (and (>= i (char->integer #\A)) (<= i (char->integer #\Z)))
287             (and (>= i (char->integer #\a)) (<= i (char->integer #\z)))))))
288
289 (define (char-numeric? x)
290   (and (char? x)
291        (let ((i (char->integer x)))
292          (and (>= i (char->integer #\0)) (<= i (char->integer #\9))))))
293
294 \f
295 ;;; Math
296 (define quotient /)
297
298 (define (<= . rest)
299   (or (apply < rest)
300       (apply = rest)))
301
302 (define (>= . rest)
303   (or (apply > rest)
304       (apply = rest)))
305
306 (define (remainder x y)
307   (- x (* (quotient x y) y)))
308
309 (define (even? x)
310   (= 0 (remainder x 2)))
311
312 (define (odd? x)
313   (= 1 (remainder x 2)))
314
315 (define (negative? x)
316   (< x 0))
317
318 (define (positive? x)
319   (> x 0))
320
321 (define (zero? x)
322   (= x 0))
323
324 (define (1+ x)
325   (+ x 1))
326
327 (define (1- x)
328   (- x 1))
329
330 (define (abs x)
331   (if (>= x 0) x (- x)))
332
333 (define (expt x y)
334   (let loop ((s 1) (count y))
335     (if (= 0 count) s
336         (loop (* s x) (- count 1)))))
337
338 (define (max x . rest)
339   (if (null? rest) x
340       (let ((y (car rest)))
341         (let ((z (if (> x y) x y)))
342           (apply max (cons z (cdr rest)))))))
343
344 (define (min x . rest)
345   (if (null? rest) x
346       (let ((y (car rest)))
347         (let ((z (if (< x y) x y)))
348           (apply min (cons z (cdr rest)))))))
349
350 (define (negate proc)
351   (lambda args
352     (not (apply proc args))))