54cdbc92c5c5f589cc677faebf3289b1379fbf66
[mes.git] / src / lib.c
1 /* -*-comment-start: "//";comment-end:""-*-
2  * Mes --- Maxwell Equations of Software
3  * Copyright © 2016,2017,2018 Jan (janneke) 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 int g_depth;
22 SCM fdisplay_ (SCM, int, int);
23
24 SCM
25 display_helper (SCM x, int cont, char* sep, int fd, int write_p)
26 {
27   fputs (sep, fd);
28   if (g_depth == 0)
29     return cell_unspecified;
30   g_depth = g_depth - 1;
31   
32   switch (TYPE (x))
33     {
34     case TCHAR:
35       {
36         if (!write_p)
37           fputc (VALUE (x), fd);
38         else
39           {
40             fputs ("#\\", fd);
41             switch (VALUE (x))
42               {
43               case '\0': fputs ("nul", fd); break;
44               case '\a': fputs ("alarm", fd); break;
45               case '\b': fputs ("backspace", fd); break;
46               case '\t': fputs ("tab", fd); break;
47               case '\n': fputs ("newline", fd); break;
48               case '\v': fputs ("vtab", fd); break;
49               case '\f': fputs ("page", fd); break;
50               case '\r': fputs ("return", fd); break;
51               case ' ': fputs ("space", fd); break;
52               default: fputc (VALUE (x), fd);
53               }
54           }
55         break;
56       }
57     case TCLOSURE:
58       {
59         fputs ("#<closure ", fd);
60         display_helper (CDR (x), cont, "", fd, 0);
61         fputs (">", fd);
62         break;
63       }
64     case TFUNCTION:
65       {
66         fputs ("#<procedure ", fd);
67         char const *p = "?";
68         if (FUNCTION (x).name != 0)
69           p = FUNCTION (x).name;
70         fputs (p, fd);
71         fputs ("[", fd);
72         fputs (itoa (CDR (x)), fd);
73         fputs (",", fd);
74         fputs (itoa (x), fd);
75         fputs ("]>", fd);
76         break;
77       }
78     case TMACRO:
79       {
80         fputs ("#<macro ", fd);
81         display_helper (CDR (x), cont, "", fd, 0);
82         fputs (">", fd);
83         break;
84       }
85     case TVARIABLE:
86       {
87         fputs ("#<variable ", fd);
88         display_helper (CAR (VARIABLE (x)), cont, "", fd, 0);
89         fputs (">", fd);
90         break;
91       }
92     case TNUMBER:
93       {
94         fputs (itoa (VALUE (x)), fd);
95         break;
96       }
97     case TPAIR:
98       {
99         if (!cont)
100           fputs ("(", fd);
101         if (CAR (x) == cell_circular
102             && CADR (x) != cell_closure)
103           {
104             fputs ("(*circ* . ", fd);
105             int i = 0;
106             x = CDR (x);
107             while (x != cell_nil && i++ < 10)
108               {
109                 fdisplay_ (CAAR (x), fd, write_p); fputs (" ", fd);
110                 x = CDR (x);
111               }
112             fputs (" ...)", fd);
113           }
114         else
115           {
116             if (x && x != cell_nil)
117               fdisplay_ (CAR (x), fd, write_p);
118             if (CDR (x) && TYPE (CDR (x)) == TPAIR)
119               display_helper (CDR (x), 1, " ", fd, write_p);
120             else if (CDR (x) && CDR (x) != cell_nil)
121               {
122                 if (TYPE (CDR (x)) != TPAIR)
123                   fputs (" . ", fd);
124                 fdisplay_ (CDR (x), fd, write_p);
125               }
126           }
127         if (!cont)
128           fputs (")", fd);
129         break;
130       }
131     case TKEYWORD:
132     case TSPECIAL:
133     case TSTRING:
134     case TSYMBOL:
135       {
136         if (TYPE (x) == TKEYWORD)
137           fputs ("#:", fd);
138         if (write_p && TYPE (x) == TSTRING)
139           fputc ('"', fd);
140         SCM t = CAR (x);
141         while (t && t != cell_nil)
142           {
143             switch (write_p ? VALUE (CAR (t)) : 0)
144               {
145               case '\t': fputs ("\\t", fd); break;
146               case '\n': fputs ("\\n", fd); break;
147               case '\\': fputs ("\\\\", fd); break;
148               case '"': fputs ("\\\"", fd); break;
149               default: fputc (VALUE (CAR (t)), fd);
150               }
151             t = CDR (t);
152           }
153         if (write_p && TYPE (x) == TSTRING)
154           fputc ('"', fd);
155         break;
156       }
157     case TVECTOR:
158       {
159         fputs ("#(", fd);
160         SCM t = CAR (x);
161         for (int i = 0; i < LENGTH (x); i++)
162           {
163             if (i)
164               fputc (' ', fd);
165             fdisplay_ (VECTOR (x) + i, fd, write_p);
166           }
167         fputc (')', fd);
168         break;
169       }
170     default:
171       {
172         fputs ("<", fd);
173         fputs (itoa (TYPE (x)), fd);
174         fputs (":", fd);
175         fputs (itoa (x), fd);
176         fputs (">", fd);
177         break;
178       }
179     }
180   return 0;
181 }
182
183 SCM
184 display_ (SCM x)
185 {
186   g_depth = 5;
187   return display_helper (x, 0, "", g_stdout, 0);
188 }
189
190 SCM
191 display_error_ (SCM x)
192 {
193   g_depth = 5;
194   return display_helper (x, 0, "", STDERR, 0);
195 }
196
197 SCM
198 display_port_ (SCM x, SCM p)
199 {
200   assert (TYPE (p) == TNUMBER);
201   return fdisplay_ (x, VALUE (p), 0);
202 }
203
204 SCM
205 write_ (SCM x)
206 {
207   g_depth = 5;
208   return display_helper (x, 0, "", g_stdout, 1);
209 }
210
211 SCM
212 write_error_ (SCM x)
213 {
214   g_depth = 5;
215   return display_helper (x, 0, "", STDERR, 1);
216 }
217
218 SCM
219 write_port_ (SCM x, SCM p)
220 {
221   assert (TYPE (p) == TNUMBER);
222   return fdisplay_ (x, VALUE (p), 1);
223 }
224
225 SCM
226 fdisplay_ (SCM x, int fd, int write_p) ///((internal))
227 {
228   g_depth = 5;
229   return display_helper (x, 0, "", fd, write_p);
230 }
231
232 SCM
233 exit_ (SCM x) ///((name . "exit"))
234 {
235   assert (TYPE (x) == TNUMBER);
236   exit (VALUE (x));
237 }
238
239 SCM
240 xassq (SCM x, SCM a) ///for speed in core only
241 {
242   while (a != cell_nil && x != CDAR (a))
243     a = CDR (a);
244   return a != cell_nil ? CAR (a) : cell_f;
245 }
246
247 SCM
248 memq (SCM x, SCM a)
249 {
250   switch (TYPE (x))
251     {
252     case TCHAR:
253     case TNUMBER:
254       {
255         SCM v = VALUE (x);
256         while (a != cell_nil && v != VALUE (CAR (a)))
257           a = CDR (a);
258         break;
259       }
260     case TKEYWORD:
261       {
262         SCM v = STRING (x);
263         while (a != cell_nil && v != STRING (CAR (a)))
264           a = CDR (a);
265         break;
266       }
267       // case TSYMBOL:
268       // case TSPECIAL:
269     default:
270       while (a != cell_nil && x != CAR (a))
271         a = CDR (a);
272     }
273   return a != cell_nil ? a : cell_f;
274 }
275
276 SCM
277 equal2_p (SCM a, SCM b)
278 {
279   if (a == cell_nil && b == cell_nil)
280     return cell_t;
281   if (TYPE (a) == TPAIR && TYPE (b) == TPAIR)
282     return equal2_p (CAR (a), CAR (b)) == cell_t
283       && equal2_p (CDR (a), CDR (b)) == cell_t
284       ? cell_t : cell_f;
285   if (TYPE (a) == TSTRING && TYPE (b) == TSTRING)
286     return equal2_p (STRING (a), STRING (b));
287   if (TYPE (a) == TVECTOR && TYPE (b) == TVECTOR)
288     {
289       if (LENGTH (a) != LENGTH (b))
290         return cell_f;
291       for (int i=0; i < LENGTH (a); i++)
292         {
293           SCM ai = VECTOR (a) + i;
294           SCM bi = VECTOR (b) + i;
295           if (TYPE (ai) == TREF)
296             ai = REF (ai);
297           if (TYPE (bi) == TREF)
298             bi = REF (bi);
299           if (equal2_p (ai, bi) == cell_f)
300             return cell_f;
301         }
302       return cell_t;
303     }
304   return eq_p (a, b);
305 }