More error handling on apply_env.
[mes.git] / mes.c
1 /* -*-comment-start: "//";comment-end:""-*-
2  * Mes --- Maxwell Equations of Software
3  * Copyright © 2016 Jan Nieuwenhuizen <janneke@gnu.org>
4  *
5  * This file is part of Mes.
6  *
7  * Mes is free software; you can redistribute it and/or modify it
8  * under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 3 of the License, or (at
10  * your option) any later version.
11  *
12  * Mes is distributed in the hope that it will be useful, but
13  * WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with Mes.  If not, see <http://www.gnu.org/licenses/>.
19  */
20
21 #define _GNU_SOURCE
22 #include <assert.h>
23 #include <ctype.h>
24 #include <limits.h>
25 #include <stdio.h>
26 #include <string.h>
27 #include <stdlib.h>
28 #include <stdbool.h>
29
30 #define DEBUG 0
31 #define QUASIQUOTE 1
32
33 enum type {CHAR, MACRO, NUMBER, PAIR, SCM, STRING, SYMBOL, REF, VALUES, VECTOR,
34            FUNCTION0, FUNCTION1, FUNCTION2, FUNCTION3, FUNCTIONn};
35 struct scm_t;
36 typedef struct scm_t* (*function0_t) (void);
37 typedef struct scm_t* (*function1_t) (struct scm_t*);
38 typedef struct scm_t* (*function2_t) (struct scm_t*, struct scm_t*);
39 typedef struct scm_t* (*function3_t) (struct scm_t*, struct scm_t*, struct scm_t*);
40 typedef struct scm_t* (*functionn_t) (struct scm_t*);
41
42 typedef struct scm_t {
43   enum type type;
44   union {
45     char const *name;
46     struct scm_t* string;
47     struct scm_t* car;
48     struct scm_t* ref;
49     int length;
50   };
51   union {
52     int value;
53     function0_t function0;
54     function1_t function1;
55     function2_t function2;
56     function3_t function3;
57     functionn_t functionn;
58     struct scm_t* cdr;
59     struct scm_t* macro;
60     struct scm_t* vector;
61     int hits;
62   };
63 } scm;
64
65 #include "define.environment.h"
66 #include "lib.environment.h"
67 #include "math.environment.h"
68 #include "mes.environment.h"
69 #include "quasiquote.environment.h"
70 #include "string.environment.h"
71 #include "type.environment.h"
72
73 scm *display_ (FILE* f, scm *x);
74 scm *display_helper (FILE*, scm*, bool, char const*, bool);
75
76 scm scm_nil = {SCM, "()"};
77 scm scm_dot = {SCM, "."};
78 scm scm_f = {SCM, "#f"};
79 scm scm_t = {SCM, "#t"};
80 scm scm_undefined = {SCM, "*undefined*"};
81 scm scm_unspecified = {SCM, "*unspecified*"};
82 scm scm_closure = {SCM, "*closure*"};
83 scm scm_circular = {SCM, "*circular*"};
84 #if BOOT
85 scm scm_label = {
86   SCM, "label"};
87 #endif
88 scm scm_lambda = {SCM, "lambda"};
89
90 scm symbol_begin = {SCM, "begin"};
91 scm symbol_if = {SCM, "if"};
92 scm symbol_define = {SCM, "define"};
93 scm symbol_define_macro = {SCM, "define-macro"};
94 scm symbol_set_x = {SCM, "set!"};
95
96 scm symbol_quote = {SYMBOL, "quote"};
97 scm symbol_quasiquote = {SYMBOL, "quasiquote"};
98 scm symbol_unquote = {SYMBOL, "unquote"};
99 scm symbol_unquote_splicing = {SYMBOL, "unquote-splicing"};
100
101 scm symbol_sc_expand = {SYMBOL, "sc-expand"};
102 scm symbol_syntax = {SYMBOL, "syntax"};
103 scm symbol_quasisyntax = {SYMBOL, "quasisyntax"};
104 scm symbol_unsyntax = {SYMBOL, "unsyntax"};
105 scm symbol_unsyntax_splicing = {SYMBOL, "unsyntax-splicing"};
106
107 scm symbol_call_with_values = {SYMBOL, "call-with-values"};
108 scm symbol_current_module = {SYMBOL, "current-module"};
109
110
111 scm char_nul = {CHAR, .name="nul", .value=0};
112 scm char_backspace = {CHAR, .name="backspace", .value=8};
113 scm char_tab = {CHAR, .name="tab", .value=9};
114 scm char_newline = {CHAR, .name="newline", .value=10};
115 scm char_vt = {CHAR, .name="vt", .value=11};
116 scm char_page = {CHAR, .name="page", .value=12};
117 scm char_return = {CHAR, .name="return", .value=13};
118 scm char_space = {CHAR, .name="space", .value=32};
119
120 // PRIMITIVES
121
122 scm *
123 car (scm *x)
124 {
125   assert (x->type == PAIR);
126   return x->car;
127 }
128
129 scm *
130 cdr (scm *x)
131 {
132   assert (x->type == PAIR);
133   return x->cdr;
134 }
135
136 scm *
137 alloc (int n)
138 {
139   return (scm*)malloc (n * sizeof (scm));
140 }
141
142 scm *
143 make_cell (scm *type, scm *car, scm *cdr)
144 {
145   scm *x = alloc (1);
146   assert (type->type == NUMBER);
147   x->type = type->value;
148   if (type->value == CHAR || type->value == NUMBER) {
149     if (car) x->car = car->car;
150     if (cdr) x->cdr = cdr->cdr;
151   } else {
152     x->car = car;
153     x->cdr = cdr;
154   }
155   return x;
156 }
157
158 scm *
159 cons (scm *x, scm *y)
160 {
161   scm t = {NUMBER, .value=PAIR};
162   return make_cell (&t, x, y);
163 }
164
165 scm *
166 eq_p (scm *x, scm *y)
167 {
168   return (x == y
169           || (x->type == CHAR && y->type == CHAR
170               && x->value == y->value)
171           || (x->type == NUMBER && y->type == NUMBER
172               && x->value == y->value))
173     ? &scm_t : &scm_f;
174 }
175
176 scm *
177 set_car_x (scm *x, scm *e)
178 {
179   assert (x->type == PAIR);
180   x->car = e;
181   return &scm_unspecified;
182 }
183
184 scm *
185 set_cdr_x (scm *x, scm *e)
186 {
187   assert (x->type == PAIR);
188   cache_invalidate (x->cdr);
189   x->cdr = e;
190   return &scm_unspecified;
191 }
192
193 scm *
194 set_env_x (scm *x, scm *e, scm *a)
195 {
196   cache_invalidate (x);
197   scm *p = assq (x, a);
198   if (p->type != PAIR)
199     {
200       fprintf (stderr, "set!: unbound variable:");
201       display_ (stderr, x);
202       fprintf (stderr, "\n");
203       assert (!"unbound variable");
204     }
205   return set_cdr_x (p, e);
206 }
207
208 scm *
209 quote (scm *x)
210 {
211   return cons (&symbol_quote, x);
212 }
213
214 scm *
215 quasiquote (scm *x)
216 {
217   return cons (&symbol_quasiquote, x);
218 }
219
220 scm *
221 quasisyntax (scm *x)
222 {
223   return cons (&symbol_quasisyntax, x);
224 }
225
226 scm *
227 pairlis (scm *x, scm *y, scm *a)
228 {
229   if (x == &scm_nil)
230     return a;
231   if (pair_p (x) == &scm_f)
232     return cons (cons (x, y), a);
233   return cons (cons (car (x), car (y)),
234                pairlis (cdr (x), cdr (y), a));
235 }
236
237 scm *
238 assq (scm *x, scm *a)
239 {
240   while (a != &scm_nil && eq_p (x, a->car->car) == &scm_f) a = a->cdr;
241   return a != &scm_nil ? a->car : &scm_f;
242 }
243
244 #define ENV_CACHE 1
245 #define CACHE_SIZE 30
246 #define ENV_HEAD 15
247
248 #if !ENV_CACHE
249 scm *
250 assq_ref_cache (scm *x, scm *a)
251 {
252   x = assq (x, a);
253   if (x == &scm_f) return &scm_undefined;
254   return x->cdr;
255 }
256 scm*cache_invalidate (scm*x){}
257 scm*cache_invalidate_range (scm*p,scm*a){}
258 scm*cache_save (scm*p){}
259 scm*cache_lookup (scm*x){}
260
261 #else // ENV_CACHE
262
263 scm *env_cache_cars[CACHE_SIZE];
264 scm *env_cache_cdrs[CACHE_SIZE];
265 int cache_threshold = 0;
266 scm *
267 cache_save (scm *p)
268 {
269   int n = p->car->hits;
270   if (n < cache_threshold) return &scm_unspecified;
271   int j = -1;
272   for (int i=0; i < CACHE_SIZE; i++) {
273     if (!env_cache_cars[i]) {
274       j = i;
275       break;
276     }
277     if (env_cache_cars[i] == p->car) return &scm_unspecified;
278     if (n > env_cache_cars[i]->hits) {
279       n = env_cache_cars[i]->hits;
280       j = i;
281     }
282   }
283   if (j >= 0) {
284     cache_threshold = p->car->hits;
285     env_cache_cars[j] = p->car;
286     env_cache_cdrs[j] = p->cdr;
287   }
288   return &scm_unspecified;
289 }
290
291 scm *
292 cache_lookup (scm *x)
293 {
294   for (int i=0; i < CACHE_SIZE; i++) {
295     if (!env_cache_cars[i]) break;
296     if (env_cache_cars[i] == x) return env_cache_cdrs[i];
297   }
298   return &scm_undefined;
299 }
300
301 scm *
302 cache_invalidate (scm *x)
303 {
304   for (int i=0; i < CACHE_SIZE; i++) {
305     if (env_cache_cars[i] == x) {
306       env_cache_cars[i] = 0;
307       break;
308     }
309   }
310   return &scm_unspecified;
311 }
312
313 scm *
314 cache_invalidate_range (scm *p, scm *a)
315 {
316   do {
317     cache_invalidate (p->car->car);
318     p = p->cdr;
319   } while (p != a);
320   return &scm_unspecified;
321 }
322
323 scm *
324 assq_ref_cache (scm *x, scm *a)
325 {
326   x->hits++;
327   scm *c = cache_lookup (x);
328   if (c != &scm_undefined) return c;
329   int i = 0;
330   while (a != &scm_nil && x != a->car->car) {i++;a = a->cdr;}
331   if (a == &scm_nil) return &scm_undefined;
332   if (i>ENV_HEAD) cache_save (a->car);
333   return a->car->cdr;
334 }
335 #endif // ENV_CACHE
336
337 scm *
338 evlis_env (scm *m, scm *a)
339 {
340   if (m == &scm_nil) return &scm_nil;
341   if (m->type != PAIR) return builtin_eval (m, a);
342   scm *e = builtin_eval (car (m), a);
343   return cons (e, evlis_env (cdr (m), a));
344 }
345
346 scm *
347 apply_env (scm *fn, scm *x, scm *a)
348 {
349   if (fn->type != PAIR)
350     {
351       if (fn == &scm_car) return x->car->car;
352       if (fn == &scm_cdr) return x->car->cdr;
353       if (builtin_p (fn) == &scm_t)
354         return call (fn, x);
355       if (eq_p (fn, &symbol_call_with_values) == &scm_t)
356         return call (&scm_call_with_values_env, append2 (x, cons (a, &scm_nil)));
357       if (fn == &symbol_current_module) return a;
358     }
359   else if (fn->car == &scm_lambda) {
360     scm *p = pairlis (cadr (fn), x, a);
361     cache_invalidate_range (p, a->cdr);
362     scm *r = begin (cddr (fn), cons (cons (&scm_closure, p), p));
363     cache_invalidate_range (p, a->cdr);
364     return r;
365   }
366   else if (fn->car == &scm_closure) {
367     scm *args = caddr (fn);
368     scm *body = cdddr (fn);
369     a = cdadr (fn);
370     a = cdr (a);
371     scm *p = pairlis (args, x, a);
372     cache_invalidate_range (p, a->cdr);
373     scm *r = begin (body, cons (cons (&scm_closure, p), p));
374     cache_invalidate_range (p, a->cdr);
375     return r;
376   }
377 #if BOOT
378   else if (fn->car == &scm_label)
379     return apply_env (caddr (fn), x, cons (cons (cadr (fn), caddr (fn)), a));
380 #endif
381   scm *efn = builtin_eval (fn, a);
382   if (efn == &scm_f || efn == &scm_t) assert (!"apply bool");
383   if (efn->type == NUMBER) assert (!"apply number");
384   if (efn->type == STRING) assert (!"apply string");
385   if (efn == &scm_unspecified) assert (!"apply *unspecified*");
386   return apply_env (efn, x, a);
387 }
388
389 scm *
390 builtin_eval (scm *e, scm *a)
391 {
392   if (builtin_p (e) == &scm_t) return e;
393   if (e->type == SCM) return e;
394
395   e = expand_macro_env (e, a);
396
397   if (e->type == SYMBOL) {
398     scm *y = assq_ref_cache (e, a);
399     if (y == &scm_undefined) {
400       fprintf (stderr, "eval: unbound variable:");
401       display_ (stderr, e);
402       fprintf (stderr, "\n");
403       assert (!"unbound variable");
404     }
405     return y;
406   }
407   else if (e->type != PAIR)
408     return e;
409   else if (e->car->type != PAIR)
410     {
411       if (e->car == &symbol_quote)
412         return cadr (e);
413       if (e->car == &symbol_syntax)
414         return e;
415       if (e->car == &symbol_begin)
416         return begin (e, a);
417       if (e->car == &scm_lambda)
418         return make_closure (cadr (e), cddr (e), assq (&scm_closure, a));
419       if (e->car == &scm_closure)
420         return e;
421       if (e->car == &symbol_if)
422         return builtin_if (cdr (e), a);
423 #if !BOOT
424       if (e->car == &symbol_define)
425         return define (e, a);
426       if (e->car == &symbol_define_macro)
427         return define (e, a);
428 #else
429       if (e->car == &symbol_define) {
430         fprintf (stderr, "C DEFINE: ");
431         display_ (stderr,
432                   e->cdr->car->type == SYMBOL
433                   ? e->cdr->car->string
434                   : e->cdr->car->car->string);
435         fprintf (stderr, "\n");
436       }
437       assert (e->car != &symbol_define);
438       assert (e->car != &symbol_define_macro);
439 #endif
440       if (e->car == &symbol_set_x)
441         return set_env_x (cadr (e), builtin_eval (caddr (e), a), a);
442 #if QUASIQUOTE
443       if (e->car == &symbol_unquote)
444         return builtin_eval (cadr (e), a);
445       if (e->car == &symbol_quasiquote)
446         return eval_quasiquote (cadr (e), add_unquoters (a));
447       if (e->car == &symbol_unsyntax)
448         return builtin_eval (cadr (e), a);
449       if (e->car == &symbol_quasisyntax)
450         return eval_quasisyntax (cadr (e), add_unsyntaxers (a));
451 #endif //QUASIQUOTE
452     }
453   return apply_env (e->car, evlis_env (e->cdr, a), a);
454 }
455
456 scm *
457 expand_macro_env (scm *e, scm *a)
458 {
459   scm *macro;
460   if (e->type == PAIR
461       && (macro = lookup_macro (e->car, a)) != &scm_f)
462     return expand_macro_env (apply_env (macro, e->cdr, a), a);
463   return e;
464 }
465
466 scm *
467 begin (scm *e, scm *a)
468 {
469   scm *r = &scm_unspecified;
470   while (e != &scm_nil) {
471     r = builtin_eval (e->car, a);
472     e = e->cdr;
473   }
474   return r;
475 }
476
477 scm *
478 builtin_if (scm *e, scm *a)
479 {
480   if (builtin_eval (car (e), a) != &scm_f)
481     return builtin_eval (cadr (e), a);
482   if (cddr (e) != &scm_nil)
483     return builtin_eval (caddr (e), a);
484   return &scm_unspecified;
485 }
486
487 //Helpers
488
489 scm *
490 display (scm *x) ///((args . n))
491 {
492   scm *e = car (x);
493   scm *p = cdr (x);
494   int fd = 1;
495   if (p->type == PAIR && p->car->type == NUMBER) fd = p->car->hits;
496   FILE *f = fd == 1 ? stdout : stderr;
497   return display_helper (f, e, false, "", false);
498 }
499
500 scm *
501 display_ (FILE* f, scm *x)
502 {
503   return display_helper (f, x, false, "", false);
504 }
505
506 scm *
507 call (scm *fn, scm *x)
508 {
509   if (fn->type == FUNCTION0)
510     return fn->function0 ();
511   if (x != &scm_nil && x->car->type == VALUES)
512     x = cons (x->car->cdr->car, x->cdr);
513   if (fn->type == FUNCTION1)
514     return fn->function1 (car (x));
515   if (x != &scm_nil && x->cdr->car->type == VALUES)
516     x = cons (x->car, cons (x->cdr->car->cdr->car, x->cdr));
517   if (fn->type == FUNCTION2)
518     return fn->function2 (car (x), cadr (x));
519   if (fn->type == FUNCTION3)
520     return fn->function3 (car (x), cadr (x), caddr (x));
521   if (fn->type == FUNCTIONn)
522     return fn->functionn (x);
523   return &scm_unspecified;
524 }
525
526 scm *
527 append2 (scm *x, scm *y)
528 {
529   if (x == &scm_nil) return y;
530   assert (x->type == PAIR);
531   return cons (car (x), append2 (cdr (x), y));
532 }
533
534 scm *
535 append (scm *x) ///((args . n))
536  {
537   if (x == &scm_nil) return &scm_nil;
538   return append2 (car (x), append (cdr (x)));
539  }
540
541 scm *
542 make_char (int x)
543 {
544   scm t = {NUMBER, .value=CHAR};
545   scm n = {NUMBER, .value=x};  
546   return make_cell (&t, &n, &n);
547 }
548
549 scm *
550 make_macro (scm *name, scm *x)
551 {
552   scm t = {NUMBER, .value=MACRO};
553   return make_cell (&t, name->string, x);
554 }
555
556 scm *
557 make_number (int x)
558 {
559   scm t = {NUMBER, .value=NUMBER};
560   scm n = {NUMBER, .value=x};  
561   return make_cell (&t, &n, &n);
562 }
563
564 scm *
565 make_ref (scm *x)
566 {
567   scm t = {NUMBER, .value=REF};
568   return make_cell (&t, x, x);
569 }
570
571 scm *
572 make_string (scm *x)
573 {
574   scm t = {NUMBER, .value=STRING};
575   return make_cell (&t, x, 0);
576 }
577
578 scm *
579 cstring_to_list (char const* s)
580 {
581   scm *p = &scm_nil;
582   while (s && *s)
583     p = append2 (p, cons (make_char (*s++), &scm_nil));
584   return p;
585 }
586
587 scm *symbols = 0;
588
589 scm *
590 list_of_char_equal_p (scm *a, scm *b)
591 {
592   while (a != &scm_nil && b != &scm_nil && a->car->value == b->car->value) {
593     assert (a->car->type == CHAR);
594     assert (b->car->type == CHAR);
595     a = a->cdr;
596     b = b->cdr;
597   }
598   return (a == &scm_nil && b == &scm_nil) ? &scm_t : &scm_f;
599 }
600
601 scm *
602 internal_lookup_symbol (scm *s)
603 {
604   scm *x = symbols;
605   while (x) {
606     // .string and .name is the same field; .name is used as a handy
607     // static field initializer.  A string can only be mistaken for a
608     // cell with type == PAIR for the one character long, zero-padded
609     // #\etx.
610     if (x->car->string->type != PAIR)
611       x->car->string = cstring_to_list (x->car->name);
612     if (list_of_char_equal_p (x->car->string, s) == &scm_t) break;
613     x = x->cdr;
614   }
615   if (x) x = x->car;
616   return x;
617 }
618
619 scm *
620 internal_make_symbol (scm *s)
621 {
622   scm t = {NUMBER, .value=SYMBOL};
623   scm *x = make_cell (&t, s, 0);
624   symbols = cons (x, symbols);
625   return x;
626 }
627
628 scm *
629 make_symbol (scm *s)
630 {
631   scm *x = internal_lookup_symbol (s);
632   return x ? x : internal_make_symbol (s);
633 }
634
635 scm *
636 make_vector (scm *n)
637 {
638   scm t = {NUMBER, .value=VECTOR};
639   scm *v = alloc (n->value);
640   scm *x = make_cell (&t, (scm*)(long)n->value, v);
641   for (int i=0; i<n->value; i++) x->vector[i] = *vector_entry (&scm_unspecified);
642   return x;
643 }
644
645 scm *
646 values (scm *x) ///((args . n))
647 {
648   scm *v = cons (0, x);
649   v->type = VALUES;
650   return v;
651 }
652
653 scm *
654 call_with_values_env (scm *producer, scm *consumer, scm *a)
655 {
656   scm *v = apply_env (producer, &scm_nil, a);
657   if (v->type == VALUES)
658     v = v->cdr;
659   return apply_env (consumer, v, a);
660 }
661
662 scm *
663 vector_length (scm *x)
664 {
665   assert (x->type == VECTOR);
666   return make_number (x->length);
667 }
668
669 scm *
670 vector_ref (scm *x, scm *i)
671 {
672   assert (x->type == VECTOR);
673   assert (i->value < x->length);
674   scm *e = &x->vector[i->value];
675   if (e->type == REF) e = e->ref;
676   if (e->type == CHAR) e = make_char (e->value);
677   if (e->type == NUMBER) e = make_number (e->value);
678   return e;
679 }
680
681 scm *
682 vector_entry (scm *x) {
683   if (x->type == PAIR || x->type == SCM || x->type == STRING || x->type == SYMBOL || x->type == VECTOR) x = make_ref (x);
684   return x;
685 }
686
687 scm *
688 vector_set_x (scm *x, scm *i, scm *e)
689 {
690   assert (x->type == VECTOR);
691   assert (i->value < x->length);
692   x->vector[i->value] = *vector_entry (e);
693   return &scm_unspecified;
694 }
695
696 scm *
697 lookup (scm *s, scm *a)
698 {
699   if (isdigit (s->car->value) || (s->car->value == '-' && s->cdr != &scm_nil)) {
700     scm *p = s;
701     int sign = 1;
702     if (s->car->value == '-') {
703       sign = -1;
704       p = s->cdr;
705     }
706     int n = 0;
707     while (p != &scm_nil && isdigit (p->car->value)) {
708       n *= 10;
709       n += p->car->value - '0';
710       p = p->cdr;
711     }
712     if (p == &scm_nil) return make_number (n * sign);
713   }
714   
715   scm *x = internal_lookup_symbol (s);
716   if (x) return x;
717
718   if (s->cdr == &scm_nil) {
719     if (s->car->value == '\'') return &symbol_quote;
720     if (s->car->value == '`') return &symbol_quasiquote;
721     if (s->car->value == ',') return &symbol_unquote;
722   }
723   else if (s->cdr->cdr == &scm_nil) {
724     if (s->car->value == ',' && s->cdr->car->value == '@') return &symbol_unquote_splicing;
725     if (s->car->value == '#' && s->cdr->car->value == '\'') return &symbol_syntax;
726     if (s->car->value == '#' && s->cdr->car->value == '`') return &symbol_quasisyntax;
727     if (s->car->value == '#' && s->cdr->car->value == ',') return &symbol_unsyntax;
728   }
729   else if (s->cdr->cdr->cdr == &scm_nil) {
730     if (s->car->value == '#' && s->cdr->car->value == ',' && s->cdr->cdr->car->value == '@') return &symbol_unsyntax_splicing;
731     if (s->car->value == 'E' && s->cdr->car->value == 'O' && s->cdr->cdr->car->value == 'F') {
732       fprintf (stderr, "mes: got EOF\n");
733       return &scm_nil; // `EOF': eval program, which may read stdin
734     }
735   }
736
737   return internal_make_symbol (s);
738 }
739
740 scm *
741 lookup_char (int c, scm *a)
742 {
743   return lookup (cons (make_char (c), &scm_nil), a);
744 }
745
746 scm *
747 list_to_vector (scm *x)
748 {
749   scm n = {NUMBER, .value=length (x)->value};
750   scm *v = make_vector (&n);
751   scm *p = v->vector;
752   while (x != &scm_nil)
753     {
754       *p++ = *vector_entry (car (x));
755       x = cdr (x);
756     }
757   return v;
758 }
759
760 scm *
761 newline (scm *p) ///((args . n))
762 {
763   int fd = 1;
764   if (p->type == PAIR && p->car->type == NUMBER) fd = p->car->value;
765   FILE *f = fd == 1 ? stdout : stderr;
766   fputs ("\n", f);
767   return &scm_unspecified;
768 }
769
770 scm *
771 force_output (scm *p) ///((args . n))
772 {
773   int fd = 1;
774   if (p->type == PAIR && p->car->type == NUMBER) fd = p->car->value;
775   FILE *f = fd == 1 ? stdout : stderr;
776   fflush (f);
777 }
778
779 scm *
780 display_helper (FILE* f, scm *x, bool cont, char const *sep, bool quote)
781 {
782   scm *r;
783   fprintf (f, "%s", sep);
784   if (x->type == CHAR && x->value == char_nul.value) fprintf (f, "#\\%s", char_nul.name);
785   else if (x->type == CHAR && x->value == char_backspace.value) fprintf (f, "#\\%s", char_backspace.name);
786   else if (x->type == CHAR && x->value == char_tab.value) fprintf (f, "#\\%s", char_tab.name);
787   else if (x->type == CHAR && x->value == char_newline.value) fprintf (f, "#\\%s", char_newline.name);
788   else if (x->type == CHAR && x->value == char_vt.value) fprintf (f, "#\\%s", char_vt.name);
789   else if (x->type == CHAR && x->value == char_page.value) fprintf (f, "#\\%s", char_page.name);
790   else if (x->type == CHAR && x->value == char_return.value) fprintf (f, "#\\%s", char_return.name);
791   else if (x->type == CHAR && x->value == char_space.value) fprintf (f, "#\\%s", char_space.name);
792   else if (x->type == CHAR) fprintf (f, "#\\%c", x->value);
793   else if (x->type == MACRO) {
794     fprintf (f, "(*macro* ");
795     display_helper (f, x->macro, cont, sep, quote);
796     fprintf (f, ")");
797   }
798   else if (x->type == NUMBER) fprintf (f, "%d", x->value);
799   else if (x->type == PAIR) {
800     if (car (x) == &scm_circular) {
801       fprintf (f, "(*circ* . #-1#)");
802       return &scm_unspecified;
803     }
804     if (car (x) == &scm_closure) {
805       fprintf (f, "(*closure* . #-1#)");
806       return &scm_unspecified;
807     }
808     if (car (x) == &scm_quote) {
809       fprintf (f, "'");
810       return display_helper (f, car (cdr (x)), cont, "", true);
811     }
812     if (!cont) fprintf (f, "(");
813     display_ (f, car (x));
814     if (cdr (x)->type == PAIR)
815       display_helper (f, cdr (x), true, " ", false);
816     else if (cdr (x) != &scm_nil) {
817       fprintf (f, " . ");
818       display_ (f, cdr (x));
819     }
820     if (!cont) fprintf (f, ")");
821   }
822   else if (x->type == VECTOR) {
823     fprintf (f, "#(", x->length);
824     for (int i = 0; i < x->length; i++) {
825       if (x->vector[i].type == VECTOR
826           || (x->vector[i].type == REF
827               && x->vector[i].ref->type == VECTOR))
828         fprintf (f, "%s#(...)", i ? " " : "");
829       else
830         display_helper (f, &x->vector[i], false, i ? " " : "", false);
831     }
832     fprintf (f, ")");
833   }
834   else if (x->type == REF) display_helper (f, x->ref, cont, "", true);
835   else if (builtin_p (x) == &scm_t) fprintf (f, "#<procedure %s>", x->name);
836   else if (x->type != PAIR && x->string) {
837     scm *p = x->string;
838     assert (p);
839     while (p != &scm_nil) {
840       assert (p->car->type == CHAR);
841       fputc (p->car->value, f);
842       p = p->cdr;
843     }
844   }
845   else if (x->type != PAIR && x->name) fprintf (f, "%s", x->name);
846
847   return &scm_unspecified;
848 }
849
850 // READ
851
852 int
853 ungetchar (int c)
854 {
855   return ungetc (c, stdin);
856 }
857
858 int
859 peekchar ()
860 {
861   int c = getchar ();
862   ungetchar (c);
863   return c;
864 }
865
866 scm *
867 peek_char ()
868 {
869   return make_char (peekchar ());
870 }
871
872 scm *
873 read_char ()
874 {
875   return make_char (getchar ());
876 }
877
878 scm *
879 write_char (scm *x) ///((args . n))
880 {
881   scm *c = car (x);
882   scm *p = cdr (x);
883   int fd = 1;
884   if (p->type == PAIR && p->car->type == NUMBER) fd = p->car->value;
885   FILE *f = fd == 1 ? stdout : stderr;
886   assert (c->type == NUMBER || c->type == CHAR);
887   fputc (c->value, f);
888   return c;
889 }
890
891 scm *
892 unget_char (scm *c)
893 {
894   assert (c->type == NUMBER || c->type == CHAR);
895   ungetchar (c->value);
896   return c;
897 }
898
899 int
900 readcomment (int c)
901 {
902   if (c == '\n') return c;
903   return readcomment (getchar ());
904 }
905
906 int
907 readblock (int c)
908 {
909   if (c == '!' && peekchar () == '#') return getchar ();
910   return readblock (getchar ());
911 }
912
913 scm *
914 readword (int c, scm *w, scm *a)
915 {
916   if (c == EOF && w == &scm_nil) return &scm_nil;
917   if (c == '\n' && w == &scm_nil) return readword (getchar (), w, a);
918   if (c == '\n' && w->car->value == '.' && w->cdr == &scm_nil) return &scm_dot;
919   if (c == EOF || c == '\n') return lookup (w, a);
920   if (c == ' ') return readword ('\n', w, a);
921   if (c == '"' && w == &scm_nil) return readstring ();
922   if (c == '"') {ungetchar (c); return lookup (w, a);}
923   if (c == '(' && w == &scm_nil) return readlist (a);
924   if (c == '(') {ungetchar (c); return lookup (w, a);}
925   if (c == ')' && w == &scm_nil) {ungetchar (c); return &scm_nil;}
926   if (c == ')') {ungetchar (c); return lookup (w, a);}
927   if (c == ',' && peekchar () == '@') {getchar (); return cons (lookup (symbol_unquote_splicing.string, a),
928                                                                    cons (readword (getchar (), w, a),
929                                                                          &scm_nil));}
930   if ((c == '\''
931        || c == '`'
932        || c == ',')
933       && w == &scm_nil) {return cons (lookup_char (c, a),
934                                      cons (readword (getchar (), w, a),
935                                            &scm_nil));}
936   if (c == '#' && peekchar () == ',' && w == &scm_nil) {
937     getchar ();
938     if (peekchar () == '@'){getchar (); return cons (lookup (symbol_unsyntax_splicing.string, a),
939                                                      cons (readword (getchar (), w, a),
940                                                            &scm_nil));}
941     return cons (lookup (symbol_unsyntax.string, a), cons (readword (getchar (), w, a), &scm_nil));
942   }
943   if (c == '#'
944      && (peekchar () == '\''
945          || peekchar () == '`')
946       && w == &scm_nil) {return cons (lookup (cons (make_char ('#'), cons (make_char (getchar ()), &scm_nil)), a),
947                           cons (readword (getchar (), w, a),
948                                 &scm_nil));}
949   if (c == ';') {readcomment (c); return readword ('\n', w, a);}
950   if (c == '#' && peekchar () == 'x') {getchar (); return read_hex ();}
951   if (c == '#' && peekchar () == '\\') {getchar (); return read_character ();}
952   if (c == '#' && w == &scm_nil && peekchar () == '(') {getchar (); return list_to_vector (readlist (a));}
953   if (c == '#' && peekchar () == '(') {ungetchar (c); return lookup (w, a);}
954   if (c == '#' && peekchar () == '!') {getchar (); readblock (getchar ()); return readword (getchar (), w, a);}
955   return readword (getchar (), append2 (w, cons (make_char (c), &scm_nil)), a);
956 }
957
958 scm *
959 read_hex ()
960 {
961   int n = 0;
962   int c = peekchar ();
963   while ((c >= '0' && c <= '9')
964          || (c >= 'A' && c <= 'F')
965          || (c >= 'a' && c <= 'f')) {
966     n <<= 4;
967     if (c >= 'a') n += c - 'a' + 10;
968     else if (c >= 'A') n += c - 'A' + 10;
969     else n+= c - '0';
970     getchar ();
971     c = peekchar ();
972   }
973   return make_number (n);
974 }
975
976 scm *
977 read_character ()
978 {
979   int c = getchar ();
980   if (c >= '0' && c <= '7'
981       && peekchar () >= '0' && peekchar () <= '7') {
982     c = c - '0';
983     while (peekchar () >= '0' && peekchar () <= '7') {
984       c <<= 3;
985       c += getchar () - '0';
986     }
987   }
988   else if (c >= 'a' && c <= 'z'
989       && peekchar () >= 'a' && peekchar () <= 'z') {
990     char buf[10];
991     char *p = buf;
992     *p++ = c;
993     while (peekchar () >= 'a' && peekchar () <= 'z') {
994       *p++ = getchar ();
995     }
996     *p = 0;
997     if (!strcmp (buf, char_nul.name)) c = char_nul.value;
998     else if (!strcmp (buf, char_backspace.name)) c = char_backspace.value;
999     else if (!strcmp (buf, char_tab.name)) c = char_tab.value;
1000     else if (!strcmp (buf, char_newline.name)) c = char_newline.value;
1001     else if (!strcmp (buf, char_vt.name)) c = char_vt.value;
1002     else if (!strcmp (buf, char_page.name)) c = char_page.value;
1003     else if (!strcmp (buf, char_return.name)) c = char_return.value;
1004     else if (!strcmp (buf, char_space.name)) c = char_space.value;
1005     else {
1006       fprintf (stderr, "char not supported: %s\n", buf);
1007       assert (!"char not supported");
1008     }
1009   }
1010   return make_char (c);
1011 }
1012
1013 scm *
1014 append_char (scm *x, int i)
1015 {
1016   return append2 (x, cons (make_char (i), &scm_nil));
1017 }
1018
1019 scm *
1020 readstring ()
1021 {
1022   scm *p = &scm_nil;
1023   int c = getchar ();
1024   while (true) {
1025     if (c == '"') break;
1026     if (c == '\\' && peekchar () == '"') p = append_char (p, getchar ());
1027     else if (c == '\\' && peekchar () == 'n') {getchar (); p = append_char (p, '\n');}
1028     else if (c == EOF) assert (!"EOF in string");
1029     else p = append_char (p, c);
1030     c = getchar ();
1031   }
1032   return make_string (p);
1033 }
1034
1035 int
1036 eat_whitespace (int c)
1037 {
1038   while (c == ' ' || c == '\t' || c == '\n') c = getchar ();
1039   if (c == ';') return eat_whitespace (readcomment (c));
1040   if (c == '#' && peekchar () == '!') {getchar (); readblock (getchar ()); return eat_whitespace (getchar ());}
1041   return c;
1042 }
1043
1044 scm *
1045 readlist (scm *a)
1046 {
1047   int c = getchar ();
1048   c = eat_whitespace (c);
1049   if (c == ')') return &scm_nil;
1050   scm *w = readword (c, &scm_nil, a);
1051   if (w == &scm_dot)
1052     return car (readlist (a));
1053   return cons (w, readlist (a));
1054 }
1055
1056 scm *
1057 read_env (scm *a)
1058 {
1059   return readword (getchar (), &scm_nil, a);
1060 }
1061
1062 scm *
1063 add_environment (scm *a, char const *name, scm *x)
1064 {
1065   return cons (cons (make_symbol (cstring_to_list (name)), x), a);
1066 }
1067
1068 scm *
1069 mes_environment () ///((internal))
1070 {
1071   scm *a = &scm_nil;
1072
1073   #include "mes.symbols.i"
1074
1075 #if BOOT
1076   symbols = cons (&scm_label, symbols);
1077   a = cons (cons (&scm_label, &scm_t), a);
1078 #endif
1079
1080 #include "string.environment.i"
1081 #include "math.environment.i"
1082 #include "lib.environment.i"
1083 #include "mes.environment.i"
1084 #include "define.environment.i"
1085 #include "type.environment.i"
1086
1087   a = cons (cons (&scm_closure, a), a);
1088   return a;
1089 }
1090
1091 scm *
1092 make_lambda (scm *args, scm *body)
1093 {
1094   return cons (&scm_lambda, cons (args, body));
1095 }
1096
1097 scm *
1098 make_closure (scm *args, scm *body, scm *a)
1099 {
1100   return cons (&scm_closure, cons (cons (&scm_circular, a), cons (args, body)));
1101 }
1102
1103 scm *
1104 lookup_macro (scm *x, scm *a)
1105 {
1106   if (x->type != SYMBOL) return &scm_f;
1107   scm *m = assq_ref_cache (x, a);
1108   if (macro_p (m) == &scm_t) return m->macro;
1109   return &scm_f;
1110 }
1111
1112 scm *
1113 read_file_env (scm *e, scm *a)
1114 {
1115   if (e == &scm_nil) return e;
1116   return cons (e, read_file_env (read_env (a), a));
1117 }
1118
1119 scm *
1120 load_file_env (scm *a)
1121 {
1122   return begin (read_file_env (read_env (a), a), a);
1123 }
1124
1125 #include "type.c"
1126 #include "define.c"
1127 #include "lib.c"
1128 #include "math.c"
1129 #include "quasiquote.c"
1130 #include "string.c"
1131
1132 int
1133 main (int argc, char *argv[])
1134 {
1135   if (argc > 1 && !strcmp (argv[1], "--help")) return puts ("Usage: mes < FILE\n");
1136   if (argc > 1 && !strcmp (argv[1], "--version")) return puts ("Mes 0.1\n");
1137   scm *a = mes_environment ();
1138   display_ (stderr, load_file_env (a));
1139   fputs ("", stderr);
1140   return 0;
1141 }