altos/lisp: working on lexical scoping

[fw/altos] / src / lisp / ao_lisp_prim.c

diff --git a/src/lisp/ao_lisp_prim.c b/src/lisp/ao_lisp_prim.c
index 7f02505df7d3c7b2f95b5a780c2a2df1824fa8ce..bfd75ae3b629a1ad9e235277418f3d528e73d6a1 100644 (file)
--- a/src/lisp/ao_lisp_prim.c
+++ b/src/lisp/ao_lisp_prim.c
@@ -20,71 +20,67 @@
 #define DBG(...)
 #endif
 
-static void (*const ao_lisp_print_funcs[AO_LISP_NUM_TYPE])(ao_poly) = {
-       [AO_LISP_CONS] = ao_lisp_cons_print,
-       [AO_LISP_STRING] = ao_lisp_string_print,
-       [AO_LISP_INT] = ao_lisp_int_print,
-       [AO_LISP_ATOM] = ao_lisp_atom_print,
-       [AO_LISP_BUILTIN] = ao_lisp_builtin_print
+struct ao_lisp_funcs {
+       void (*print)(ao_poly);
+       void (*patom)(ao_poly);
 };
 
-ao_poly
-ao_lisp_poly_print(ao_poly p)
-{
-       void (*print)(ao_poly) = ao_lisp_print_funcs[ao_lisp_poly_type(p)];
-       if (print)
-               print(p);
-       return p;
-}
-
-static const struct ao_lisp_type const *ao_lisp_types[AO_LISP_NUM_TYPE] = {
-       [AO_LISP_CONS] = &ao_lisp_cons_type,
-       [AO_LISP_INT] = NULL,
-       [AO_LISP_STRING] = &ao_lisp_string_type,
-       [AO_LISP_OTHER] = (void *) 0x1,
-       [AO_LISP_ATOM] = &ao_lisp_atom_type,
-       [AO_LISP_BUILTIN] = &ao_lisp_builtin_type,
-       [AO_LISP_FRAME] = &ao_lisp_frame_type,
+static const struct ao_lisp_funcs ao_lisp_funcs[AO_LISP_NUM_TYPE] = {
+       [AO_LISP_CONS] = {
+               .print = ao_lisp_cons_print,
+               .patom = ao_lisp_cons_patom,
+       },
+       [AO_LISP_STRING] = {
+               .print = ao_lisp_string_print,
+               .patom = ao_lisp_string_patom,
+       },
+       [AO_LISP_INT] = {
+               .print = ao_lisp_int_print,
+               .patom = ao_lisp_int_print,
+       },
+       [AO_LISP_ATOM] = {
+               .print = ao_lisp_atom_print,
+               .patom = ao_lisp_atom_print,
+       },
+       [AO_LISP_BUILTIN] = {
+               .print = ao_lisp_builtin_print,
+               .patom = ao_lisp_builtin_print,
+       },
+       [AO_LISP_FRAME] = {
+               .print = ao_lisp_frame_print,
+               .patom = ao_lisp_frame_print,
+       },
+       [AO_LISP_LAMBDA] = {
+               .print = ao_lisp_lambda_print,
+               .patom = ao_lisp_lambda_print,
+       },
 };
 
-int
-ao_lisp_poly_mark(ao_poly p)
+static const struct ao_lisp_funcs *
+funcs(ao_poly p)
 {
-       const struct ao_lisp_type *lisp_type = ao_lisp_types[ao_lisp_poly_type(p)];
-       if (lisp_type)
-               return ao_lisp_mark(lisp_type, ao_lisp_ref(p));
-       return 1;
+       uint8_t type = ao_lisp_poly_type(p);
+
+       if (type < AO_LISP_NUM_TYPE)
+               return &ao_lisp_funcs[type];
+       return NULL;
 }
 
-int
-ao_lisp_poly_move(ao_poly *ref)
+void
+ao_lisp_poly_print(ao_poly p)
 {
-       uint8_t                         type;
-       ao_poly                         p = *ref;
-       const struct ao_lisp_type       *lisp_type;
-       int                             ret;
-       void                            *addr;
-
-       if (!p)
-               return 1;
+       const struct ao_lisp_funcs *f = funcs(p);
 
-       type = p & AO_LISP_TYPE_MASK;
-       if (type == AO_LISP_OTHER)
-               type = ao_lisp_other_type(ao_lisp_move_map(ao_lisp_poly_other(p)));
+       if (f && f->print)
+               f->print(p);
+}
 
-       if (type >= AO_LISP_NUM_TYPE)
-               abort();
+void
+ao_lisp_poly_patom(ao_poly p)
+{
+       const struct ao_lisp_funcs *f = funcs(p);
 
-       lisp_type = ao_lisp_types[type];
-       if (!lisp_type)
-               return 1;
-       addr = ao_lisp_ref(p);
-       ret = ao_lisp_move(lisp_type, &addr);
-       if (addr != ao_lisp_ref(p)) {
-               ao_poly np = ao_lisp_poly(addr, p & AO_LISP_TYPE_MASK);
-               DBG("poly %d moved %04x -> %04x\n",
-                   type, p, np);
-               *ref = np;
-       }
-       return ret;
+       if (f && f->patom)
+               f->patom(p);
 }
+