8d3e03a913f7c349c0daccc3cc0cb0353f62274d
[fw/altos] / src / lisp / ao_lisp_make_const.c
1 /*
2  * Copyright © 2016 Keith Packard <keithp@keithp.com>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  */
14
15 #include "ao_lisp.h"
16 #include <stdlib.h>
17 #include <ctype.h>
18
19 static struct ao_lisp_builtin *
20 ao_lisp_make_builtin(enum ao_lisp_builtin_id func, int args) {
21         struct ao_lisp_builtin *b = ao_lisp_alloc(sizeof (struct ao_lisp_builtin));
22
23         b->type = AO_LISP_BUILTIN;
24         b->func = func;
25         b->args = args;
26         return b;
27 }
28
29 struct builtin_func {
30         char    *name;
31         int     args;
32         int     func;
33 };
34
35 struct builtin_func funcs[] = {
36         "car",          AO_LISP_LEXPR,  builtin_car,
37         "cdr",          AO_LISP_LEXPR,  builtin_cdr,
38         "cons",         AO_LISP_LEXPR,  builtin_cons,
39         "quote",        AO_LISP_NLAMBDA,builtin_quote,
40         "set",          AO_LISP_LEXPR,  builtin_set,
41         "setq",         AO_LISP_MACRO,  builtin_setq,
42         "print",        AO_LISP_LEXPR,  builtin_print,
43         "+",            AO_LISP_LEXPR,  builtin_plus,
44         "-",            AO_LISP_LEXPR,  builtin_minus,
45         "*",            AO_LISP_LEXPR,  builtin_times,
46         "/",            AO_LISP_LEXPR,  builtin_divide,
47         "%",            AO_LISP_LEXPR,  builtin_mod
48 };
49
50 #define N_FUNC (sizeof funcs / sizeof funcs[0])
51
52 int
53 main(int argc, char **argv)
54 {
55         int     f, o;
56         ao_poly atom, val;
57         struct ao_lisp_atom     *a;
58
59         for (f = 0; f < N_FUNC; f++) {
60                 struct ao_lisp_builtin  *b = ao_lisp_make_builtin(funcs[f].func, funcs[f].args);
61                 struct ao_lisp_atom     *a = ao_lisp_atom_intern(funcs[f].name);
62                 a->val = ao_lisp_builtin_poly(b);
63         }
64
65         for (;;) {
66                 atom = ao_lisp_read();
67                 if (!atom)
68                         break;
69                 val = ao_lisp_read();
70                 if (!val)
71                         break;
72                 if (ao_lisp_poly_type(atom) != AO_LISP_ATOM) {
73                         fprintf(stderr, "input must be atom val pairs\n");
74                         exit(1);
75                 }
76                 ao_lisp_poly_atom(atom)->val = val;
77         }
78
79         printf("/* constant objects, all referenced from atoms */\n\n");
80         printf("#define AO_LISP_POOL_CONST %d\n", ao_lisp_top);
81         printf("extern const uint8_t ao_lisp_const[AO_LISP_POOL_CONST] __attribute__((aligned(4)));\n");
82         printf("#define ao_builtin_atoms 0x%04x\n", ao_lisp_atom_poly(ao_lisp_atoms));
83
84         for (a = ao_lisp_atoms; a; a = ao_lisp_poly_atom(a->next)) {
85                 char    *n = a->name, c;
86                 printf ("#define _ao_lisp_atom_");
87                 while ((c = *n++)) {
88                         if (isalnum(c))
89                                 printf("%c", c);
90                         else
91                                 printf("%02x", c);
92                 }
93                 printf("  0x%04x\n", ao_lisp_atom_poly(a));
94         }
95         printf("#ifdef AO_LISP_CONST_BITS\n");
96         printf("const uint8_t ao_lisp_const[] = {");
97         for (o = 0; o < ao_lisp_top; o++) {
98                 uint8_t c;
99                 if ((o & 0xf) == 0)
100                         printf("\n\t");
101                 else
102                         printf(" ");
103                 c = ao_lisp_const[o];
104                 if (' ' < c && c <= '~' && c != '\'')
105                         printf (" '%c',", c);
106                 else
107                         printf("0x%02x,", c);
108         }
109         printf("\n};\n");
110         printf("#endif /* AO_LISP_CONST_BITS */\n");
111 }