altos/lisp: add length, pack, unpack and flush
[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         "eval",         AO_LISP_FUNC_LAMBDA,    builtin_eval,
37         "read",         AO_LISP_FUNC_LAMBDA,    builtin_read,
38         "lambda",       AO_LISP_FUNC_NLAMBDA,   builtin_lambda,
39         "lexpr",        AO_LISP_FUNC_NLAMBDA,   builtin_lexpr,
40         "nlambda",      AO_LISP_FUNC_NLAMBDA,   builtin_nlambda,
41         "macro",        AO_LISP_FUNC_NLAMBDA,   builtin_macro,
42         "car",          AO_LISP_FUNC_LAMBDA,    builtin_car,
43         "cdr",          AO_LISP_FUNC_LAMBDA,    builtin_cdr,
44         "cons",         AO_LISP_FUNC_LAMBDA,    builtin_cons,
45         "last",         AO_LISP_FUNC_LAMBDA,    builtin_last,
46         "length",       AO_LISP_FUNC_LAMBDA,    builtin_length,
47         "quote",        AO_LISP_FUNC_NLAMBDA,   builtin_quote,
48         "set",          AO_LISP_FUNC_LAMBDA,    builtin_set,
49         "setq",         AO_LISP_FUNC_MACRO,     builtin_setq,
50         "cond",         AO_LISP_FUNC_NLAMBDA,   builtin_cond,
51         "progn",        AO_LISP_FUNC_NLAMBDA,   builtin_progn,
52         "while",        AO_LISP_FUNC_NLAMBDA,   builtin_while,
53         "print",        AO_LISP_FUNC_LEXPR,     builtin_print,
54         "patom",        AO_LISP_FUNC_LEXPR,     builtin_patom,
55         "+",            AO_LISP_FUNC_LEXPR,     builtin_plus,
56         "-",            AO_LISP_FUNC_LEXPR,     builtin_minus,
57         "*",            AO_LISP_FUNC_LEXPR,     builtin_times,
58         "/",            AO_LISP_FUNC_LEXPR,     builtin_divide,
59         "%",            AO_LISP_FUNC_LEXPR,     builtin_mod,
60         "=",            AO_LISP_FUNC_LEXPR,     builtin_equal,
61         "<",            AO_LISP_FUNC_LEXPR,     builtin_less,
62         ">",            AO_LISP_FUNC_LEXPR,     builtin_greater,
63         "<=",           AO_LISP_FUNC_LEXPR,     builtin_less_equal,
64         ">=",           AO_LISP_FUNC_LEXPR,     builtin_greater_equal,
65         "pack",         AO_LISP_FUNC_LAMBDA,    builtin_pack,
66         "unpack",       AO_LISP_FUNC_LAMBDA,    builtin_unpack,
67         "flush",        AO_LISP_FUNC_LAMBDA,    builtin_flush,
68         "delay",        AO_LISP_FUNC_LAMBDA,    builtin_delay,
69         "led",          AO_LISP_FUNC_LEXPR,     builtin_led,
70 };
71
72 #define N_FUNC (sizeof funcs / sizeof funcs[0])
73
74 struct ao_lisp_frame    *globals;
75
76 static int
77 is_atom(int offset)
78 {
79         struct ao_lisp_atom *a;
80
81         for (a = ao_lisp_atoms; a; a = ao_lisp_poly_atom(a->next))
82                 if (((uint8_t *) a->name - ao_lisp_const) == offset)
83                         return strlen(a->name);
84         return 0;
85 }
86
87 int
88 main(int argc, char **argv)
89 {
90         int     f, o, i;
91         ao_poly sexpr, val;
92         struct ao_lisp_atom     *a;
93         struct ao_lisp_builtin  *b;
94         int     in_atom;
95
96         printf("/*\n");
97         printf(" * Generated file, do not edit\n");
98         for (f = 0; f < N_FUNC; f++) {
99                 b = ao_lisp_make_builtin(funcs[f].func, funcs[f].args);
100                 a = ao_lisp_atom_intern(funcs[f].name);
101                 ao_lisp_atom_set(ao_lisp_atom_poly(a),
102                                  ao_lisp_builtin_poly(b));
103         }
104
105         /* boolean constants */
106         ao_lisp_atom_set(ao_lisp_atom_poly(ao_lisp_atom_intern("nil")),
107                          AO_LISP_NIL);
108         a = ao_lisp_atom_intern("t");
109         ao_lisp_atom_set(ao_lisp_atom_poly(a),
110                          ao_lisp_atom_poly(a));
111
112         /* end of file value */
113         a = ao_lisp_atom_intern("eof");
114         ao_lisp_atom_set(ao_lisp_atom_poly(a),
115                          ao_lisp_atom_poly(a));
116
117         ao_lisp_read_eval_print();
118
119         /* Reduce to referenced values */
120         ao_lisp_collect();
121         printf(" */\n");
122
123         printf("#define AO_LISP_POOL_CONST %d\n", ao_lisp_top);
124         printf("extern const uint8_t ao_lisp_const[AO_LISP_POOL_CONST] __attribute__((aligned(4)));\n");
125         printf("#define ao_builtin_atoms 0x%04x\n", ao_lisp_atom_poly(ao_lisp_atoms));
126         printf("#define ao_builtin_frame 0x%04x\n", ao_lisp_frame_poly(ao_lisp_frame_global));
127
128         for (a = ao_lisp_atoms; a; a = ao_lisp_poly_atom(a->next)) {
129                 char    *n = a->name, c;
130                 printf ("#define _ao_lisp_atom_");
131                 while ((c = *n++)) {
132                         if (isalnum(c))
133                                 printf("%c", c);
134                         else
135                                 printf("%02x", c);
136                 }
137                 printf("  0x%04x\n", ao_lisp_atom_poly(a));
138         }
139         printf("#ifdef AO_LISP_CONST_BITS\n");
140         printf("const uint8_t ao_lisp_const[] = {");
141         for (o = 0; o < ao_lisp_top; o++) {
142                 uint8_t c;
143                 if ((o & 0xf) == 0)
144                         printf("\n\t");
145                 else
146                         printf(" ");
147                 c = ao_lisp_const[o];
148                 if (!in_atom)
149                         in_atom = is_atom(o);
150                 if (in_atom) {
151                         printf (" '%c',", c);
152                         in_atom--;
153                 } else {
154                         printf("0x%02x,", c);
155                 }
156         }
157         printf("\n};\n");
158         printf("#endif /* AO_LISP_CONST_BITS */\n");
159 }