core: Support keywords.
[mes.git] / reader.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 SCM
22 peek_char ()
23 {
24   return make_char (peekchar ());
25 }
26
27 SCM
28 read_char ()
29 {
30   return make_char (getchar ());
31 }
32
33 SCM
34 unread_char (SCM c)
35 {
36   return ungetchar (VALUE (c));
37 }
38
39 int
40 read_block_comment (int c)
41 {
42   if (c == '!' && peekchar () == '#') return getchar ();
43   return read_block_comment (getchar ());
44 }
45
46 int
47 read_line_comment (int c)
48 {
49   if (c == '\n') return c;
50   return read_line_comment (getchar ());
51 }
52
53
54 SCM lookup_char (int c, SCM a);
55
56 SCM
57 read_word (int c, SCM w, SCM a)
58 {
59   if (c == EOF && w == cell_nil) return cell_nil;
60   if (c == '\t') return read_word ('\n', w, a);
61   if (c == '\f') return read_word ('\n', w, a);
62   if (c == '\n' && w == cell_nil) return read_word (getchar (), w, a);
63   if (c == '\n' && VALUE (car (w)) == '.' && cdr (w) == cell_nil) return cell_dot;
64   if (c == EOF || c == '\n') return lookup (w, a);
65   if (c == ' ') return read_word ('\n', w, a);
66   if (c == '(' && w == cell_nil) return read_list (a);
67   if (c == '(') {ungetchar (c); return lookup (w, a);}
68   if (c == ')' && w == cell_nil) {ungetchar (c); return cell_nil;}
69   if (c == ')') {ungetchar (c); return lookup (w, a);}
70   if (c == ';') {read_line_comment (c); return read_word ('\n', w, a);}
71 #if READER
72   if (c == '"' && w == cell_nil) return read_string ();
73   if (c == '"') {ungetchar (c); return lookup (w, a);}
74   if (c == ',' && peekchar () == '@') {getchar (); return cons (lookup (STRING (cell_symbol_unquote_splicing), a),
75                                                                    cons (read_word (getchar (), w, a),
76                                                                          cell_nil));}
77   if ((c == '\''
78        || c == '`'
79        || c == ',')
80       && w == cell_nil) {return cons (lookup_char (c, a),
81                                      cons (read_word (getchar (), w, a),
82                                            cell_nil));}
83   if (c == '#' && peekchar () == ',' && w == cell_nil) {
84     getchar ();
85     if (peekchar () == '@'){getchar (); return cons (lookup (STRING (cell_symbol_unsyntax_splicing), a),
86                                                      cons (read_word (getchar (), w, a),
87                                                            cell_nil));}
88     return cons (lookup (STRING (cell_symbol_unsyntax), a), cons (read_word (getchar (), w, a), cell_nil));
89   }
90   if (c == '#' && (peekchar () == '\'' || peekchar () == '`') && w == cell_nil) {
91     c = getchar ();
92     return cons (lookup (cons (make_char ('#'), cons (make_char (c), cell_nil)), a),
93                  cons (read_word (getchar (), w, a), cell_nil));}
94   if (c == '#' && peekchar () == 'x') {getchar (); return read_hex ();}
95   if (c == '#' && peekchar () == '\\') {getchar (); return read_character ();}
96   if (c == '#' && w == cell_nil && peekchar () == '(') {getchar (); return list_to_vector (read_list (a));}
97   if (c == '#' && peekchar () == '!') {getchar (); read_block_comment (getchar ()); return read_word (getchar (), w, a);}
98 #endif //READER
99   return read_word (getchar (), append2 (w, cons (make_char (c), cell_nil)), a);
100 }
101
102 SCM
103 read_character ()
104 {
105   int c = getchar ();
106   if (c >= '0' && c <= '7'
107       && peekchar () >= '0' && peekchar () <= '7') {
108     c = c - '0';
109     while (peekchar () >= '0' && peekchar () <= '7') {
110       c <<= 3;
111       c += getchar () - '0';
112     }
113   }
114   else if (c >= 'a' && c <= 'z'
115       && peekchar () >= 'a' && peekchar () <= 'z') {
116     char buf[10];
117     char *p = buf;
118     *p++ = c;
119     while (peekchar () >= 'a' && peekchar () <= 'z') {
120       *p++ = getchar ();
121     }
122     *p = 0;
123     if (!strcmp (buf, char_nul.name)) c = char_nul.value;
124     else if (!strcmp (buf, char_backspace.name)) c = char_backspace.value;
125     else if (!strcmp (buf, char_tab.name)) c = char_tab.value;
126     else if (!strcmp (buf, char_newline.name)) c = char_newline.value;
127     else if (!strcmp (buf, char_vt.name)) c = char_vt.value;
128     else if (!strcmp (buf, char_page.name)) c = char_page.value;
129     else if (!strcmp (buf, char_return.name)) c = char_return.value;
130     else if (!strcmp (buf, char_space.name)) c = char_space.value;
131     else {
132       fprintf (stderr, "char not supported: %s\n", buf);
133       assert (!"char not supported");
134     }
135   }
136   return make_char (c);
137 }
138
139 SCM
140 read_hex ()
141 {
142   int n = 0;
143   int c = peekchar ();
144   while ((c >= '0' && c <= '9')
145          || (c >= 'A' && c <= 'F')
146          || (c >= 'a' && c <= 'f')) {
147     n <<= 4;
148     if (c >= 'a') n += c - 'a' + 10;
149     else if (c >= 'A') n += c - 'A' + 10;
150     else n+= c - '0';
151     getchar ();
152     c = peekchar ();
153   }
154   return make_number (n);
155 }
156
157 SCM
158 append_char (SCM x, int i)
159 {
160   return append2 (x, cons (make_char (i), cell_nil));
161 }
162
163 SCM
164 read_string ()
165 {
166   SCM p = cell_nil;
167   int c = getchar ();
168   while (true) {
169     if (c == '"') break;
170     if (c == '\\' && peekchar () == '"') p = append_char (p, getchar ());
171     else if (c == '\\' && peekchar () == 'n') {getchar (); p = append_char (p, '\n');}
172     else if (c == EOF) assert (!"EOF in string");
173     else p = append_char (p, c);
174     c = getchar ();
175   }
176   return make_string (p);
177 }
178
179 int
180 eat_whitespace (int c)
181 {
182   while (c == ' ' || c == '\t' || c == '\n' || c == '\f') c = getchar ();
183   if (c == ';') return eat_whitespace (read_line_comment (c));
184 #if READER
185   if (c == '#' && peekchar () == '!') {getchar (); read_block_comment (getchar ()); return eat_whitespace (getchar ());}
186 #endif
187   return c;
188 }
189
190 SCM
191 read_list (SCM a)
192 {
193   int c = getchar ();
194   c = eat_whitespace (c);
195   if (c == ')') return cell_nil;
196   SCM w = read_word (c, cell_nil, a);
197   if (w == cell_dot)
198     return car (read_list (a));
199   return cons (w, read_list (a));
200 }
201
202 SCM
203 read_env (SCM a)
204 {
205   return read_word (getchar (), cell_nil, a);
206 }
207
208 SCM
209 lookup (SCM s, SCM a)
210 {
211   if (isdigit (VALUE (car (s))) || (VALUE (car (s)) == '-' && cdr (s) != cell_nil)) {
212     SCM p = s;
213     int sign = 1;
214     if (VALUE (car (s)) == '-') {
215       sign = -1;
216       p = cdr (s);
217     }
218     int n = 0;
219     while (p != cell_nil && isdigit (VALUE (car (p)))) {
220       n *= 10;
221       n += VALUE (car (p)) - '0';
222       p = cdr (p);
223     }
224     if (p == cell_nil) return make_number (n * sign);
225   }
226
227   if (VALUE (car (s)) == '#' && VALUE (cadr (s)) == ':') return make_keyword (cddr (s));
228
229   SCM x = internal_lookup_symbol (s);
230   if (x) return x;
231
232   if (cdr (s) == cell_nil) {
233     if (VALUE (car (s)) == '\'') return cell_symbol_quote;
234     if (VALUE (car (s)) == '`') return cell_symbol_quasiquote;
235     if (VALUE (car (s)) == ',') return cell_symbol_unquote;
236   }
237   else if (cddr (s) == cell_nil) {
238     if (VALUE (car (s)) == ',' && VALUE (cadr (s)) == '@') return cell_symbol_unquote_splicing;
239     if (VALUE (car (s)) == '#' && VALUE (cadr (s)) == '\'') return cell_symbol_syntax;
240     if (VALUE (car (s)) == '#' && VALUE (cadr (s)) == '`') return cell_symbol_quasisyntax;
241     if (VALUE (car (s)) == '#' && VALUE (cadr (s)) == ',') return cell_symbol_unsyntax;
242   }
243   else if (cdddr (s) == cell_nil) {
244     if (VALUE (car (s)) == '#' && VALUE (cadr (s)) == ',' && VALUE (caddr (s)) == '@') return cell_symbol_unsyntax_splicing;
245         if (VALUE (car (s)) == 'E' && VALUE (cadr (s)) == 'O' && VALUE (caddr (s)) == 'F') {
246       fprintf (stderr, "mes: got EOF\n");
247       return cell_nil; // `EOF': eval program, which may read stdin
248     }
249   }
250
251   return internal_make_symbol (s);
252 }
253
254 SCM
255 lookup_char (int c, SCM a)
256 {
257   return lookup (cons (make_char (c), cell_nil), a);
258 }
259
260 SCM
261 list_of_char_equal_p (SCM a, SCM b)
262 {
263   while (a != cell_nil && b != cell_nil && VALUE (car (a)) == VALUE (car (b))) {
264     assert (TYPE (car (a)) == CHAR);
265     assert (TYPE (car (b)) == CHAR);
266     a = cdr (a);
267     b = cdr (b);
268   }
269   return (a == cell_nil && b == cell_nil) ? cell_t : cell_f;
270 }
271
272 SCM
273 internal_lookup_symbol (SCM s)
274 {
275   SCM x = g_symbols;
276   while (x) {
277     // .string and .name is the same field; .name is used as a handy
278     // static field initializer.  A string can only be mistaken for a
279     // cell with type == PAIR for the one character long, zero-padded
280     // #\etx.
281     SCM p = STRING (car (x));
282     if (p < 0 || p >= g_free.value || TYPE (p) != PAIR)
283       STRING (car (x)) = cstring_to_list (NAME (car (x)));
284     if (list_of_char_equal_p (STRING (car (x)), s) == cell_t) break;
285     x = cdr (x);
286   }
287   if (x) x = car (x);
288   return x;
289 }